Serialize graph sources

src/miv_simulator/interface/neuroh5_graph.py

@@ -18,6 +18,7 @@ def graph(self) -> None:
     def __call__(self) -> None:
         print("Merging H5 data")
         self.network = None
+        self.distances = None
         self.synapses = {}
         self.synapse_forest = {}
         self.connections = {}
@@ -26,6 +27,8 @@ def __call__(self) -> None:
             name = u.module.replace("miv_simulator.interface.", "")
             if name == "network_architecture":
                 self.network = u
+            elif name == "distances":
+                self.distances = u
             elif name == "connections":
                 for p in read_population_names(u.config.forest_filepath):
                     populations.append(p)
@@ -69,6 +72,24 @@ def __call__(self) -> None:
         for p, c in self.connections.items():
             self.graph.import_projections(p, c.output_filepath)
 
+        # serialize sources
+        self.save_file(
+            "graph.json",
+            {
+                "network": self.network.serialize(),
+                "distances": self.distances.serialize(),
+                "connections": {
+                    k: v.serialize() for k, v in self.connections.items()
+                },
+                "synapse_forest": {
+                    k: v.serialize() for k, v in self.synapse_forest.items()
+                },
+                "synapses": {
+                    k: v.serialize() for k, v in self.synapses.items()
+                },
+            },
+        )
+
     def on_compute_predicate(self):
         def generate_uid(use):
             if getattr(use, "refreshed_at", None) is not None: