Merge branch 'ModelicaSystem_check' into MERGE-4.1.0.rc2

Author: syntron

OMPython/ModelicaSystem.py

@@ -52,6 +52,7 @@
     OMCSessionRunData,
     OMCSessionZMQ,
     OMCProcess,
+    OMCProcessLocal,
     OMCPath,
 )
 
@@ -1141,6 +1142,36 @@ def simulate(
 
         self._simulated = True
 
+    def plot(
+            self,
+            plotdata: str,
+            resultfile: Optional[str | os.PathLike] = None,
+    ) -> None:
+        """
+        Plot a variable using OMC; this will work for local OMC usage only (OMCProcessLocal).
+        """
+
+        if not isinstance(self._session.omc_process, OMCProcessLocal):
+            raise ModelicaSystemError("Plot is using the OMC plot functionality; "
+                                      "thus, it is only working if OMC is running locally!")
+
+        plot_result_file = None
+        if resultfile is not None:
+            plot_result_file = self._session.omcpath(resultfile)
+        elif self._result_file is not None:
+            plot_result_file = self._session.omcpath(self._result_file)
+        else:
+            ModelicaSystemError("No resultfile available - either run simulate() before plotting "
+                                "or provide a result file!")
+
+        if plot_result_file is None:
+            ModelicaSystemError("No resultfile defined!")
+        elif not plot_result_file.is_file():
+            ModelicaSystemError(f"Provided resultfile {repr(plot_result_file.as_posix())} does not exists!")
+        else:
+            expr = f'plot({plotdata}, fileName="{plot_result_file.as_posix()}")'
+            self.sendExpression(expr=expr)
+
     def getSolutions(
             self,
             varList: Optional[str | list[str]] = None,