From 19d657b60b22159359f7a229f5a5b73e729cff79 Mon Sep 17 00:00:00 2001 From: Jaron Kent-Dobias Date: Wed, 6 Nov 2019 13:55:38 -0500 Subject: added support for saving networks to the library and revamped the sample method --- src/sample.cpp | 58 +++++++++------------------------------ src/sample.hpp | 4 ++- src/sample_fracture.cpp | 73 ++++++++++++++++++++++++++++++++++++------------- 3 files changed, 70 insertions(+), 65 deletions(-) (limited to 'src') diff --git a/src/sample.cpp b/src/sample.cpp index febff19..e7ca09b 100644 --- a/src/sample.cpp +++ b/src/sample.cpp @@ -3,8 +3,13 @@ #include #include -sample::sample(double Lx, double Ly, double beta) { - std::string filename = "fracture_" + std::to_string(Lx) + "_" + std::to_string(Ly) + "_" + std::to_string(beta) + "_sample.dat"; +sample::sample(unsigned n, double a, double beta) { + std::string filename = "sample_" + std::to_string(n) + "_" + std::to_string(a) + "_" + std::to_string(beta) + ".dat"; + sample_file.open(filename, std::ifstream::app); +} + +sample::sample(unsigned Lx, unsigned Ly, double beta) { + std::string filename = "sample_" + std::to_string(Lx) + "_" + std::to_string(Ly) + "_" + std::to_string(beta) + ".dat"; sample_file.open(filename, std::ifstream::app); } @@ -14,56 +19,19 @@ sample::~sample() { void sample::pre_fracture(const network& n) { sample_file << "<|"; - sample_file << "\"vr\"->{"; - for (const graph::vertex &v : n.G.vertices) { - sample_file <{"; - for (const graph::vertex &v : n.G.vertices) { - sample_file << "{"; - for (const graph::coordinate &r : v.polygon) { - sample_file <{"; - for (const graph::vertex &v : n.G.dual_vertices) { - sample_file <{"; - for (const graph::vertex &v : n.G.dual_vertices) { - sample_file << "{"; - for (const graph::coordinate &r : v.polygon) { - sample_file <{"; - for (const graph::edge &e : n.G.edges) { - sample_file << "{" << e.v[0] << "," << e.v[1] << "},"; - } - sample_file << "},\"ec\"->{"; - for (const graph::edge &e : n.G.edges) { - sample_file << "{" << e.crossings[0] << "," << e.crossings[1] << "},"; - } - sample_file << "},\"d\"->{"; - for (const graph::edge &e : n.G.dual_edges) { - sample_file << "{" << e.v[0] << "," << e.v[1] << "},"; - } - sample_file << "},\"dc\"->{"; - for (const graph::edge &e : n.G.dual_edges) { - sample_file << "{" << e.crossings[0] << "," << e.crossings[1] << "},"; - } - sample_file << "},\"data\"->{"; + sample_file << n.G.write(); + sample_file << ",\"data\"->{"; } void sample::bond_broken(const network& net, const current_info& cur, unsigned i) { long double c = logl(cur.conductivity[0] / fabs(cur.currents[i])) + net.thresholds[i]; - sample_file << "{" << i << "," << c << "," << cur.conductivity[0] << "},"; + sample_string += "{" + std::to_string(i) + "," + std::to_string(c) + "," + std::to_string(cur.conductivity[0]) + "},"; } void sample::post_fracture(network &n) { - sample_file << "}"; + sample_string.pop_back(); + sample_file << sample_string + "}"; sample_file << "|>\n"; + sample_string.clear(); } diff --git a/src/sample.hpp b/src/sample.hpp index d5d6fe5..c21daca 100644 --- a/src/sample.hpp +++ b/src/sample.hpp @@ -18,9 +18,11 @@ class sample : public hooks { // private: std::ofstream sample_file; + std::string sample_string; public: - sample(double, double, double); + sample(unsigned, double, double); + sample(unsigned, unsigned, double); ~sample(); void pre_fracture(const network &) override; diff --git a/src/sample_fracture.cpp b/src/sample_fracture.cpp index eaba69e..b01b4d6 100644 --- a/src/sample_fracture.cpp +++ b/src/sample_fracture.cpp @@ -16,28 +16,43 @@ #include int main(int argc, char* argv[]) { - int opt; unsigned N = 1; unsigned Lx = 16; - double Ly = 16; + unsigned Ly = 16; double beta = 0.5; - while ((opt = getopt(argc, argv, "N:X:Y:b:")) != -1) { + unsigned n = 128; + double a = 1.0; + bool use_aN = false; + double w = 0.5; + + while ((opt = getopt(argc, argv, "N:X:Y:b:n:a:w:")) != -1) { switch (opt) { case 'N': N = (unsigned)atof(optarg); break; case 'X': - Lx = atof(optarg); + Lx = atoi(optarg); break; case 'Y': - Ly = atof(optarg); + Ly = atoi(optarg); break; case 'b': beta = atof(optarg); break; + case 'n': + n = (unsigned)atof(optarg); + use_aN = true; + break; + case 'a': + a = atof(optarg); + use_aN = true; + break; + case 'w': + w = atof(optarg); + break; default: exit(1); } @@ -46,23 +61,43 @@ int main(int argc, char* argv[]) { cholmod_common c; CHOL_F(start)(&c); - sample meas(Lx, Ly, beta); - //sample meas2(Ly, Lx, beta); - randutils::auto_seed_128 seeds; std::mt19937 rng{seeds}; - for (unsigned trial = 0; trial < N; trial++) { - graph G(Lx, 1.0, rng); - elastic_network network(G, &c); - network.set_thresholds(beta, rng); - network.fracture(meas); - - /*graph G2 = G.rotate(); - class network network2(G2, &c); - network2.thresholds = network.thresholds; - network2.fracture(meas2); - */ + if (use_aN) { + sample meas(n, a, beta); + + for (unsigned trial = 0; trial < N; trial++) { + while (true) { + try { + graph G(n, a, rng); + elastic_network fuse_network(G, &c, w); + fuse_network.set_thresholds(beta, rng); + fuse_network.fracture(meas); + break; + } catch (std::exception &e) { + std::cout << e.what() << '\n'; + } + } + } + } else { + sample meas(Lx, Ly, beta); + + const graph G(Lx, Ly); + const elastic_network fuse_network(G, &c, w); + + for (unsigned trial = 0; trial < N; trial++) { + while (true) { + try { + elastic_network net = fuse_network; + net.set_thresholds(beta, rng); + net.fracture(meas); + break; + } catch (std::exception &e) { + std::cout << e.what() << '\n'; + } + } + } } CHOL_F(finish)(&c); -- cgit v1.2.3-54-g00ecf