From 37eedebd45b9ee0985d2e05b05b5942be4404569 Mon Sep 17 00:00:00 2001
From: Valentin Boettcher <hiro@protagon.space>
Date: Mon, 21 Mar 2022 14:58:57 +0100
Subject: [PATCH] add some auxiliary functionality

---
 model_auxiliary.py | 53 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 53 insertions(+)
 create mode 100644 model_auxiliary.py

diff --git a/model_auxiliary.py b/model_auxiliary.py
new file mode 100644
index 0000000..dccd029
--- /dev/null
+++ b/model_auxiliary.py
@@ -0,0 +1,53 @@
+"""Functionality to integrate :any:`Model` instances and analyze the results."""
+
+from model_base import Model
+from hops.core.integration import HOPSSupervisor
+from contextlib import contextmanager
+from utility import JSONEncoder, object_hook
+from filelock import FileLock
+from pathlib import Path
+
+
+@contextmanager
+def model_db(data_path: str = "."):
+    """
+    Opens the model database json file in the folder ``data_path`` as
+    a dictionary.
+
+    Mutations will be synchronized to the file.  Access is managed via
+    a lock file.
+    """
+    db_path = Path(data_path) / "model_data.json"
+    db_lock = Path(data_path) / "model_data.json.lock"
+
+    lock = FileLock(db_lock)
+    with lock:
+        with db_path.open("rw") as f:
+            db = JSONEncoder.loads(f.read())
+
+            yield db
+
+            f.write(JSONEncoder.dumps(db))
+
+
+def integrate(model: Model, n: int, data_path: str = "."):
+    """Integrate the hops equations for the model.
+
+    A call to :any:`ray.init` may be required.
+
+    :param n: The number of samples to be integrated.
+    """
+
+    hash = model.__hash__()
+    supervisor = HOPSSupervisor(
+        model.hops_config, n, data_path=data_path, data_name=str(hash)
+    )
+
+    supervisor.integrate()
+
+    with supervisor.get_data(True) as data:
+        with model_db(data_path) as db:
+            db[str(hash)] = {
+                "config": model.to_json(),
+                "data_path": str(Path(data.hdf5_name).relative_to(data_path)),
+            }