summaryrefslogtreecommitdiff
path: root/lib/include/network.hpp
blob: 601222819df9af022359f48f7ca7b0abd24e0c1a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80

#pragma once

#include <vector>
#include <functional>
#include <utility>
#include <random>
#include <limits>

#include <cholmod.h>

#include "graph.hpp"
#include "hooks.hpp"
#include "current_info.hpp"
#include "array_hash.hpp"

#ifdef FRACTURE_LONGINT

#define CHOL_F(x) cholmod_l_##x
#define CHOL_INT long int

#else

#define CHOL_F(x) cholmod_##x
#define CHOL_INT int

#endif

class problem {
  private:
    const graph& G;
    cholmod_dense* b;
    cholmod_factor* factor;
    cholmod_sparse* voltcurmat;
    cholmod_common* c;
    unsigned axis;

  public:
    problem(const graph&, unsigned axis, cholmod_common*);
    problem(const graph&, unsigned axis, cholmod_sparse*, cholmod_common*);
    problem(const problem&);
    ~problem();
    current_info solve(std::vector<bool>& fuses);
    void break_edge(unsigned, bool unbreak = false);
};

class network {
  public:
    const graph& G;
    std::vector<bool> fuses;
    std::vector<long double> thresholds;

    network(const graph&);
    network(const network&);

    void set_thresholds(double, std::mt19937&);
};

class fuse_network : public network {
  public:
    problem ohm;

    fuse_network(const graph&, cholmod_common*);

    void fracture(hooks&, double cutoff = 1e-13);
    current_info get_current_info();
};

class elastic_network : public network {
  public:
    problem hook_x;
    problem hook_y;

    elastic_network(const graph&, cholmod_common*);
    elastic_network(const elastic_network&);

    void fracture(hooks&, double weight = 0.5, double cutoff = 1e-12);
    current_info get_current_info();
};