??? run_doe

Author: syntron

tools/run_doe.py

@@ -0,0 +1,122 @@
+import numpy as np
+import OMPython
+import pathlib
+import logging
+
+logging.basicConfig(level=logging.WARNING)  # DEBUG)
+
+logger = logging.getLogger(__name__)
+
+def model_doe(tmp_path: pathlib.Path) -> pathlib.Path:
+    # see: https://trac.openmodelica.org/OpenModelica/ticket/4052
+    mod = tmp_path / "M_local.mo"
+    mod.write_text("""
+model M
+  parameter Integer p=1;
+  parameter Integer q=1;
+  parameter Real a = -1;
+  parameter Real b = -1;
+  Real x[p];
+  Real y[q];
+equation
+  der(x) = a * fill(1.0, p);
+  der(y) = b * fill(1.0, q);
+end M;
+""")
+    return mod
+
+
+def param_doe() -> dict[str, list]:
+    param = {
+        # structural
+        'p': [1, 2],
+        'q': [3, 4],
+        # simple
+        'a': [5, 6],
+        'b': [7, 8],
+    }
+    return param
+
+
+def test_ModelicaSystemDoE(tmp_path, model_doe, param_doe):
+    tmpdir = tmp_path / 'DoE'
+    tmpdir.mkdir(exist_ok=True)
+
+    # om = OMPython.OMCSessionZMQ()
+    omcp = OMPython.OMCProcessDocker(docker="openmodelica/openmodelica:v1.25.0-minimal")
+    om = OMPython.OMCSessionZMQ(omc_process=omcp)
+
+    model_path = om.omcpath_tempdir()
+    model_file = model_path / 'M.mo'
+    model_file.write_text(model_doe.read_text())
+
+    model_path_doe = model_path / 'simDoE'
+    model_path_doe.mkdir(exist_ok=True)
+
+    doe_mod = OMPython.ModelicaSystemDoE(
+        fileName=model_file,
+        modelName="M",
+        parameters=param_doe,
+        resultpath=result_path_doe,
+        simargs={"override": {'stopTime': 1.0}},
+        omc_process=om.omc_process,
+    )
+    doe_count = doe_mod.prepare()
+    assert doe_count == 16
+
+    doe_dict = doe_mod.get_doe()
+    assert isinstance(doe_dict, dict)
+    assert len(doe_dict.keys()) == 16
+
+    doe_status = doe_mod.simulate()
+    assert doe_status is True
+
+    doe_sol = doe_mod.get_solutions()
+
+    for resultfilename in doe_dict:
+        row = doe_dict[resultfilename]
+
+        assert resultfilename in doe_sol
+        sol = doe_sol[resultfilename]
+
+        var_dict = {
+            # simple / non-structural parameters
+            'a': float(row['a']),
+            'b': float(row['b']),
+            # structural parameters
+            'p': float(row['p']),
+            'q': float(row['q']),
+            # variables using the structural parameters
+            f"x[{row['p']}]": float(row['a']),
+            f"y[{row['p']}]": float(row['b']),
+        }
+
+        for var in var_dict:
+            assert var in sol['data']
+            assert np.isclose(sol['data'][var][-1], var_dict[var])
+
+
+if __name__ == "__main__":
+    # om = OMPython.OMCSessionZMQ()
+    omcp = OMPython.OMCProcessDocker(docker="openmodelica/openmodelica:v1.25.0-minimal")
+    om = OMPython.OMCSessionZMQ(omc_process=omcp)
+
+    tempdir_local = pathlib.Path('.') / 'DoE'
+    tempdir_local = tempdir_local.resolve().absolute()
+    tempdir_local.mkdir(exist_ok=True)
+    tempdir_omcpath = om.omcpath_tempdir()
+
+    model_file_local = model_doe(tempdir_local)
+    model_file_omcpath = tempdir_omcpath / 'M.mo'
+    model_file_omcpath.write_text(model_file_local.read_text())
+
+    param = param_doe()
+
+    test_ModelicaSystemDoE(
+        tmp_path=tempdir_omcpath,
+        model_doe=model_file_omcpath,
+        param_doe=param,
+        result_path_doe=tempdir_omcpath,
+    )
+
+    print("DONE")