Updated matrix handler for saving figures to file

neuromllite/MatrixHandler.py

@@ -11,16 +11,19 @@
 from neuromllite.NetworkGenerator import _get_rng_for_network
 
 import numpy as np
+import os
 
 
 class MatrixHandler(ConnectivityHandler):
  colormaps_used = []
 
  weight_arrays_to_show = {}
 
- def __init__(self, level=10, nl_network=None):
+ def __init__(self, level=10, nl_network=None, show_already=True, save_figs_to_dir=None):
  self.nl_network = nl_network
  self.level = level
+ self.show_already = show_already
+ self.save_figs_to_dir=save_figs_to_dir
 
  self.rng, seed = _get_rng_for_network(self.nl_network)
 
@@ -34,11 +37,6 @@ def print_settings(self):
  print_v("* level: %s" % self.level)
  print_v("* is_cell_level: %s" % self.is_cell_level())
  print_v("* CUTOFF_INH_SYN_MV: %s" % self.CUTOFF_INH_SYN_MV)
- # print_v('* include_inputs: %s'%self.include_inputs)
- # print_v('* scale_by_post_pop_size: %s'%self.scale_by_post_pop_size)
- # print_v('* scale_by_post_pop_cond: %s'%self.scale_by_post_pop_cond)
- # print_v('* min_weight_to_show: %s'%self.min_weight_to_show)
- # print_v('* min_weight_to_show: %s'%self.min_weight_to_show)
  print_v("*")
  print_v("* Used values: ")
  print_v("* colormaps_used: %s" % self.colormaps_used)
@@ -68,6 +66,9 @@ def finalise_document(self):
  for v in self.proj_post_pops.values():
  all_pops.append(v)
 
+ self.weight_matrices_generated = {}
+ self.weight_matrices_saved = {}
+
  for pop in all_pops:
  if self.is_cell_level():
  for i in range(self.pop_sizes[pop]):
@@ -376,10 +377,19 @@ def finalise_document(self):
  if proj_type in cbar_labels:
  cbar.set_label(cbar_labels[proj_type])
 
+ if self.save_figs_to_dir:
+ safe = proj_type.replace(' ','_').replace('(','-').replace(')','-').replace('*','_').replace('/','_')
+ save_figure_to = os.path.join(self.save_figs_to_dir, '%s_%s.png'%(self.network_id, safe))
+ plt.savefig(save_figure_to, bbox_inches="tight")
+ print_v("Saved image to %s of plot: %s" % (save_figure_to, title))
+
  print_v("Generating matrix for: %s" % self.network_id)
+
+
  self.print_settings()
 
- plt.show()
+ if self.show_already:
+ plt.show()
 
  def handle_population(
  self,
@@ -535,3 +545,31 @@ def finalise_projection(
 
  def finalise_input_source(self, inputListId):
  pass
+
+if __name__ == "__main__":
+
+ from neuromllite.utils import load_network_json
+
+ tests = ['/Users/padraig/neuroConstruct/osb/cerebral_cortex/networks/del-Molino2017/NeuroML/delMolinoEtAl_low_baseline.json',
+ '/Users/padraig/NeuroMLlite/examples/Example12_MultiComp.json']
+
+ for test in tests:
+
+ network = load_network_json(test)
+
+ from neuromllite.NetworkGenerator import generate_network
+
+ level = 1
+ handler = MatrixHandler(level, 
+ nl_network=network, 
+ show_already=True,
+ save_figs_to_dir='.')
+
+ generate_network(
+ network, handler, always_include_props=True, base_dir=os.path.dirname(test)
+ )
+
+ print_v("Done with MatrixHandler...")
+
+ for w in handler.weight_arrays_to_show:
+ print('%s:\n%s'%(w, handler.weight_arrays_to_show[w]))