Add Observatory.from_json and to_json

src/adam_test_data/main.py

@@ -18,7 +18,7 @@
 from adam_core.time import Timestamp
 
 from .noise import generate_noise
-from .observatory import Observatory, observatory_to_sorcha_config
+from .observatories import Observatory, observatory_to_sorcha_config
 from .pointings import Pointings
 from .populations import (
     SmallBodies,

src/adam_test_data/noise.py

@@ -20,7 +20,7 @@
 from jax import Array
 from scipy.stats import skewnorm
 
-from .observatory import Observatory
+from .observatories import Observatory
 from .pointings import Pointings
 
 

src/adam_test_data/observatories/__init__.py

@@ -0,0 +1,13 @@
+from .observatory import (
+    FieldOfView,
+    Observatory,
+    Simulation,
+    observatory_to_sorcha_config,
+)
+
+__all__ = [
+    "FieldOfView",
+    "Observatory",
+    "Simulation",
+    "observatory_to_sorcha_config",
+]

src/adam_test_data/observatory.py → src/adam_test_data/observatories/observatory.py

@@ -1,3 +1,4 @@
+import json
 from dataclasses import dataclass
 from typing import Literal, Optional
 
@@ -69,6 +70,75 @@ class Observatory:
     fov: FieldOfView
     simulation: Simulation
 
+    @classmethod
+    def from_json(cls, file: str) -> "Observatory":
+        """
+        Load an observatofy from a JSON file.
+
+        Parameters
+        ----------
+        file : str
+            The path to the JSON file.
+
+        Returns
+        -------
+        Observatory
+            The observatory object.
+        """
+        with open(file, "r") as f:
+            data = json.load(f)
+
+        code = data["code"]
+        filters = data["filters"]
+        main_filter = data["main_filter"]
+        bright_limit = data["bright_limit"]
+        fov = FieldOfView(
+            camera_model=data["fov"]["camera_model"],
+            circle_radius=data["fov"]["circle_radius"],
+            fill_factor=data["fov"]["fill_factor"],
+        )
+        simulation = Simulation(
+            ang_fov=data["simulation"]["ang_fov"],
+            fov_buffer=data["simulation"]["fov_buffer"],
+        )
+
+        return Observatory(
+            code=code,
+            filters=filters,
+            main_filter=main_filter,
+            bright_limit=bright_limit,
+            fov=fov,
+            simulation=simulation,
+        )
+
+    def to_json(self, file: str) -> None:
+        """
+        Save the observatory to a JSON file.
+
+        Parameters
+        ----------
+        file : str
+            The path to the JSON file.
+        """
+        data = {
+            "code": self.code,
+            "filters": self.filters,
+            "main_filter": self.main_filter,
+            "bright_limit": self.bright_limit,
+            "fov": {
+                "camera_model": self.fov.camera_model,
+                "circle_radius": self.fov.circle_radius,
+                "fill_factor": self.fov.fill_factor,
+            },
+            "simulation": {
+                "ang_fov": self.simulation.ang_fov,
+                "fov_buffer": self.simulation.fov_buffer,
+            },
+        }
+
+        with open(file, "w") as f:
+            json.dump(data, f, indent=4)
+
 
 def observatory_to_sorcha_config(
     observatory: Observatory,

src/adam_test_data/tests/test_observatory.py → src/adam_test_data/observatories/tests/test_observatory.py

@@ -1,3 +1,5 @@
+import tempfile
+
 import pytest
 
 from ..observatory import (
@@ -181,3 +183,22 @@ def test_observatory_to_sorcha_config_raises() -> None:
             main_filter="r",
         )
         observatory_to_sorcha_config(obs)
+
+
+def test_Observatory_to_from_json() -> None:
+    # Test that we can round trip an Observatory object to and from a JSON file.
+    with tempfile.NamedTemporaryFile(suffix=".json") as temp_file:
+        obs = Observatory(
+            code="X05",
+            filters=["u", "g", "r", "i", "z", "y"],
+            bright_limit=[16, 16, 16, 16, 16, 16],
+            fov=FieldOfView(camera_model="circle", fill_factor=0.9, circle_radius=3),
+            simulation=Simulation(ang_fov=1.0, fov_buffer=0.1),
+            main_filter="r",
+        )
+
+        obs.to_json(temp_file.name)
+
+        obs2 = Observatory.from_json(temp_file.name)
+
+        assert obs == obs2

src/adam_test_data/tests/test_main.py

@@ -12,7 +12,7 @@
 from adam_core.time import Timestamp
 
 from ..main import generate_test_data, sorcha, write_sorcha_inputs
-from ..observatory import FieldOfView, Observatory, Simulation
+from ..observatories import FieldOfView, Observatory, Simulation
 from ..pointings import Pointings
 from ..populations import PhotometricProperties, SmallBodies