1 files changed, 17 insertions, 32 deletions

diff --git a/lib/include/network.hpp b/lib/include/network.hpp
index 8211cfc..bf9015c 100644
--- a/lib/include/network.hpp
+++ b/lib/include/network.hpp
@@ -14,53 +14,38 @@
 #include "hooks.hpp"
 #include "current_info.hpp"
 #include "array_hash.hpp"
+#include "clusters.hpp"
 
 #define CURRENT_CUTOFF 1e-10
 
-#include <boost/graph/adjacency_list.hpp>
-#include <boost/graph/connected_components.hpp>
-#include <boost/graph/depth_first_search.hpp>
-#include <boost/range/combine.hpp>
-#include <boost/foreach.hpp>
-#include <boost/graph/graph_utility.hpp>
-#include <boost/graph/incremental_components.hpp>
-#include <boost/pending/disjoint_sets.hpp>
-
-struct EdgeProperties {
-  unsigned index;
-};
+class network {
+  private:
+    problem px;
+    problem py;
+    std::unordered_map<std::array<unsigned, 2>, bool> seen_pairs;
 
-typedef boost::adjacency_list<boost::listS, boost::vecS, boost::undirectedS, boost::no_property, EdgeProperties> Graph;
-typedef boost::graph_traits<Graph>::vertex_descriptor Vertex;
-typedef boost::graph_traits<Graph>::vertices_size_type VertexIndex;
+    void update_backbone(const std::vector<double>& c);
+    void break_edge(unsigned, bool unbreak = false);
+    void get_cycle_edges_helper(std::set<unsigned>&, std::set<unsigned>&, unsigned, unsigned) const;
+    std::set<unsigned> get_cycle_edges(unsigned) const;
+    std::array<unsigned, 2> find_cycle(const std::set<unsigned>&, unsigned, unsigned) const;
 
-class network {
   public:
     const graph& G;
 
-    Graph bG;
-    Graph dG;
-    std::vector<VertexIndex> rank;
-    std::vector<Vertex> parent;
-    boost::disjoint_sets<VertexIndex*, Vertex*> ds;
-    std::unordered_map<std::array<unsigned, 2>, bool> seen_pairs;
+    ClusterTree C;
 
     std::vector<bool> fuses;
     std::vector<bool> backbone;
     std::vector<long double> thresholds;
 
-    problem px;
-    problem py;
-
     network(const graph&, cholmod_common*);
     network(const network&);
 
     void set_thresholds(double, std::mt19937&);
-    void fracture(hooks&, bool one_axis = false);
-    void update_backbone(const std::vector<double>& c);
+    void fracture(hooks&, bool one_axis = true);
 
-    void break_edge(unsigned, bool unbreak = false);
-    virtual current_info get_current_info() {current_info empty; return empty;};
+    virtual current_info get_current_info() const {current_info empty; return empty;};
 };
 
 class fuse_network : public network {
@@ -71,7 +56,7 @@ class fuse_network : public network {
     fuse_network(const graph&, cholmod_common*, double weight = 1.0);
     fuse_network(const fuse_network&);
 
-    current_info get_current_info() override;
+    current_info get_current_info() const override;
 };
 
 class elastic_network : public network {
@@ -82,7 +67,7 @@ class elastic_network : public network {
     elastic_network(const graph&, cholmod_common*, double weight = 0.5);
     elastic_network(const elastic_network&);
 
-    current_info get_current_info() override;
+    current_info get_current_info() const override;
 };
 
 class percolation_network : public network {
@@ -90,6 +75,6 @@ class percolation_network : public network {
     percolation_network(const graph&, cholmod_common*);
     percolation_network(const percolation_network&);
 
-    current_info get_current_info() override;
+    current_info get_current_info() const override;
 };