def get_leak_matrix(internal_population, sparse=False):
internal_population_copy = internal_population.copy()
network = Network([internal_population_copy], [])
network.dt = 1.
network.t0 = 1.
network.ti = 1.
internal_population_copy.initialize()
if sparse == False:
return internal_population_copy.leak_flux_matrix_dict['dense'].copy()
else:
return spsp.csc_matrix(internal_population_copy.leak_flux_matrix_dict['dense'])
def get_connection_flux_matrices(connection, sparse=False):
target_population_copy = connection.target.copy()
network = Network([target_population_copy], [])
network.dt = 1.
network.t0 = 1.
network.ti = 1.
target_population_copy.initialize()
conn_dist = ConnectionDistribution(target_population_copy.edges, connection.weights, connection.probs)
conn_dist.initialize()
if sparse == False:
flux_matrix = conn_dist.flux_matrix_dict['dense'].copy()
else:
flux_matrix = spsp.csc_matrix(conn_dist.flux_matrix_dict['dense'])
flux_vector = conn_dist.threshold_flux_vector.copy()
return flux_matrix, flux_vector
