[ModelicaSystem] add plot() function; see #144

Author: syntron

OMPython/ModelicaSystem.py

@@ -1132,6 +1132,28 @@ def simulate(
 
         self._simulated = True
 
+    def plot(
+            self,
+            plotdata: str,
+            resultfile: Optional[str] = None,
+    ) -> None:
+        """
+        Plot via OMC.
+        """
+
+        if resultfile is None:
+            # default result file generated by OM
+            plot_result_file = self.getWorkDirectory() / f"{self._model_name}_res.mat"
+        elif os.path.exists(resultfile):
+            plot_result_file = pathlib.Path(resultfile)
+        else:
+            ModelicaSystemError("No resultfile available - either run simulate() before plotting "
+                                "or provide a result file!")
+
+        expr = f'plot({plotdata}, fileName="{plot_result_file.as_posix()}")'
+
+        self.sendExpression(expr=expr)
+
     def getSolutions(self, varList: Optional[str | list[str]] = None, resultfile: Optional[str] = None) -> tuple[str] | np.ndarray:
         """Extract simulation results from a result data file.