visualise intermediate params in json

Author: hmonroedd

examples/sparging_standard.py

@@ -40,6 +40,8 @@
     dispersion_on=True,
 )
 
+standard_input.to_json(FOLDER / "intermediate_params.json")
+breakpoint()
 if __name__ == "__main__":
     # my_simulation.exports = ["pressure", "J_T2"]
     output = my_simulation.solve(fast_solve=True)

src/sparging/inputs.py

@@ -1,5 +1,7 @@
 from __future__ import annotations
 from dataclasses import dataclass, replace
+from datetime import datetime
+from sparging import helpers
 from sparging.correlations import Correlation, all_correlations
 import pint
 from typing import List
@@ -201,15 +203,29 @@ def __post_init__(self):
     def to_json(self, path: str):
         import json
 
+        output = {
+            "metadata": {
+                "git_commit": helpers.get_git_hash(),
+                "date": datetime.now().isoformat(),
+            },
+        }
+
+        output["intermediate parameters"] = {}
+        for node in self.graph.nodes:
+            output["intermediate parameters"][node] = {
+                "value": str(self.graph.nodes[node]["value"]),
+                "origin": self.graph.nodes[node]["origin"],
+            }
         with open(path, "w") as f:
             json.dump(
-                {
-                    key: str(value)
-                    for key, value in self.__dict__.items()
-                    if isinstance(value, pint.Quantity)
-                },
+                output,
+                # {
+                #     key: str(value)
+                #     for key, value in self.__dict__.items()
+                #     if isinstance(value, pint.Quantity)
+                # },
                 f,
-                indent=2,
+                indent=4,
             )
 
     @classmethod