Plotting signals in besim

src/bedit_gui/simulation_models.py

@@ -25,6 +25,22 @@ class CompiledModel:
         return {"model_name": self.model_name, "executable": self.executable, "working_directory": self.working_directory, "output": self.output, "errors": self.errors}
 
 
+@dataclass
+class SimulationPlotTab:
+    name: str
+    signals: list[str] = field(default_factory=list)
+
+    @classmethod
+    def from_data(cls, data: Mapping[str, Any]) -> SimulationPlotTab:
+        raw_signals = data.get("signals", [])
+        if not isinstance(raw_signals, list):
+            raise TypeError("plot tab signals must be a list")
+        return cls(name=str(data.get("name", "Plot")), signals=[str(signal) for signal in raw_signals])
+
+    def to_data(self) -> dict[str, Any]:
+        return {"name": self.name, "signals": self.signals}
+
+
 @dataclass
 class SimulationRoot:
     format_version: int
@@ -36,6 +52,7 @@ class SimulationRoot:
     settings: Simulation
     current_end_time: float | None = None
     results: list[SimulationResult] = field(default_factory=list)
+    plot_tabs: list[SimulationPlotTab] = field(default_factory=lambda: [SimulationPlotTab("Plot 1")])
 
     @classmethod
     def from_data(cls, data: Mapping[str, Any]) -> SimulationRoot:
@@ -51,6 +68,7 @@ class SimulationRoot:
             settings=Simulation.from_data(data["settings"]),
             current_end_time=float(data["current_end_time"]) if data.get("current_end_time") is not None else None,
             results=[_result_from_data(result) for result in data.get("results", [])],
+            plot_tabs=[SimulationPlotTab.from_data(tab) for tab in data["plot_tabs"]] if "plot_tabs" in data else [SimulationPlotTab("Plot 1")],
         )
 
     def to_data(self) -> dict[str, Any]:
@@ -65,6 +83,7 @@ class SimulationRoot:
             "settings": self.settings.to_data(),
             "current_end_time": self.current_end_time,
             "results": [_result_to_data(result) for result in self.results],
+            "plot_tabs": [tab.to_data() for tab in self.plot_tabs],
         }