#include "fracture.h"

double *net_currents(const net_t *net, const double *voltages, cholmod_common *c) {
	uint_t ne = net->graph->ne;
	uint_t dim = net->graph->break_dim;
	cholmod_sparse *voltcurmat = net->graph->voltcurmat;

	cholmod_dense *x = CHOL_F(allocate_dense)(dim, 1, dim, CHOLMOD_REAL, c);

	double *tmp_x = x->x;
	x->x = (void *)voltages;

	cholmod_dense *y = CHOL_F(allocate_dense)(ne, 1, ne, CHOLMOD_REAL, c);

	double alpha[2] = {1, 0};
	double beta[2] = {0, 0};
	CHOL_F(sdmult)(voltcurmat, 0, alpha, beta, x, y, c);

	double *currents = (double *)malloc(ne * sizeof(double));

	for (int i = 0; i < ne; i++) {
		if (net->fuses[i]) {
			currents[i] = 0;
		} else {
			currents[i] = ((double *)y->x)[i];
		}
	}

	x->x = tmp_x;
	CHOL_F(free_dense)(&x, c);
	CHOL_F(free_dense)(&y, c);

	return currents;
}