revamped and simplied fracture code with c++

3 files changed, 89 insertions, 0 deletions

diff --git a/lib/include/graph.hpp b/lib/include/graph.hpp
new file mode 100644
index 0000000..80cdd66
--- /dev/null
+++ b/lib/include/graph.hpp
@@ -0,0 +1,29 @@
+
+#pragma once
+
+#include <cmath>
+#include <vector>
+#include <array>
+
+class graph {
+  public:
+    typedef struct coordinate {
+      double x;
+      double y;
+    } coordinate;
+
+    typedef struct vertex {
+      coordinate r;
+    } vertex;
+
+    typedef std::array<unsigned int, 2> edge;
+
+    std::vector<vertex> vertices;
+    std::vector<edge> edges;
+
+    std::vector<vertex> dual_vertices;
+    std::vector<edge> dual_edges;
+
+    graph(unsigned int L);
+};
+
diff --git a/lib/include/hooks.hpp b/lib/include/hooks.hpp
new file mode 100644
index 0000000..350f318
--- /dev/null
+++ b/lib/include/hooks.hpp
@@ -0,0 +1,12 @@
+
+#pragma once
+
+class network;
+
+class hooks {
+  public:
+    virtual void pre_fracture(const network&) {};
+    virtual void bond_broken(const network&, const std::pair<double, std::vector<double>>&, unsigned int) {};
+    virtual void post_fracture(network&) {}; // post fracture hook can be destructive
+};
+
diff --git a/lib/include/network.hpp b/lib/include/network.hpp
new file mode 100644
index 0000000..abf88cd
--- /dev/null
+++ b/lib/include/network.hpp
@@ -0,0 +1,48 @@
+
+#pragma once
+
+#include <vector>
+#include <functional>
+#include <utility>
+#include <random>
+
+#include "cholmod.h"
+
+#include "graph.hpp"
+#include "hooks.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 network {
+  private:
+    cholmod_dense *b;
+    cholmod_factor *factor;
+    cholmod_sparse *voltcurmat;
+    cholmod_common *c;
+
+  public:
+    const graph& G;
+    std::vector<bool> fuses;
+    std::vector<long double> thresholds;
+
+    network(const graph&, cholmod_common*);
+    network(const network &other);
+    ~network();
+
+    void set_thresholds(double, std::mt19937&);
+    void break_edge(unsigned int);
+    void add_edge(unsigned int);
+    std::pair<double, std::vector<double>> currents();
+    void fracture(hooks&, double cutoff = 1e-13);
+};
+