1 files changed, 40 insertions, 0 deletions
diff --git a/lib/net_voltages.c b/lib/net_voltages.c
new file mode 100644
index 0000000..7b07201
--- /dev/null
+++ b/lib/net_voltages.c
@@ -0,0 +1,40 @@
+
+#include "fracture.h"
+
+double *net_voltages(const net_t *net, cholmod_common *c) {
+	cholmod_dense *b = net->boundary_cond;
+	cholmod_factor *factor = net->factor;
+
+	cholmod_dense *x = CHOL_F(solve)(CHOLMOD_A, factor, b, c);
+
+	if (((double *)x->x)[0] != ((double *)x->x)[0]) {
+		printf("GET_VOLTAGE: value is NaN\n");
+		exit(EXIT_FAILURE);
+	}
+
+	double *t_voltages = (double *)x->x;
+	x->x = NULL;
+	CHOL_F(free_dense)(&x, c);
+
+	const graph_t *g = net->graph;
+
+	if (g->boundary == TORUS_BOUND) {
+		return t_voltages;
+	} else if (net->voltage_bound) {
+		double *voltages = (double *)malloc(g->nv * sizeof(double));
+		for (uint_t i = 0; i < g->nv - g->bi[g->nb]; i++) {
+			voltages[g->nbi[i]] = t_voltages[i];
+		}
+		for (uint_t i = 0; i < 2; i++) {
+			for (uint_t j = 0; j < g->bi[i + 1] - g->bi[i]; j++) {
+				voltages[g->b[g->bi[i] + j]] = 1 - i;
+			}
+		}
+
+		free(t_voltages);
+		return voltages;
+	} else {
+		return t_voltages;
+	}
+}
+