Write Cloud Model Simulation configuration choices into FITS header

NuSpaceSim · Aug 16, 2023 · b3dde41 · b3dde41
1 parent 1a0280d
commit b3dde41
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/src/nuspacesim/config.py b/src/nuspacesim/config.py
@@ -248,6 +248,7 @@ def __call__(self) -> dict[str, tuple[Union[int, float, str], str]]:
         }
 
         d.update(self.spectrum())
+        d.update(self.cloud_model())
 
         return d
 

diff --git a/src/nuspacesim/types/cloud_types.py b/src/nuspacesim/types/cloud_types.py
@@ -43,14 +43,21 @@
 
 @dataclass
 class NoCloud:
-    pass
+    def __call__(self) -> dict:
+        return {"cloudmdl": ("NoCloud", "Simulation: No Cloud Model")}
 
 
 @dataclass
 class MonoCloud:
     altitude: float = -np.inf
     """Altitude of monoheight cloud."""
 
+    def __call__(self) -> dict:
+        return {
+            "cloudmdl": ("MonoCloud", "Simulation: Uniform Height Cloud Model"),
+            "cloudtop": (self.altitude, "Simulation: Uniform Cloud Top Height"),
+        }
+
 
 @dataclass
 class PressureMapCloud:
@@ -60,6 +67,16 @@ class PressureMapCloud:
     version: str = "0"
     """Cloud Map File Version."""
 
+    def __call__(self) -> dict:
+        return {
+            "cloudmdl": (
+                "PressureMapCloud",
+                "Simulation: Builtin PressureMap Cloud Model",
+            ),
+            "cloudMth": (self.month, "Simulation: Month of Model"),
+            "cloudver": (self.version, "Simulation: Version of Model"),
+        }
+
 
 def parse_month(date: str | int | datetime) -> int:
     if isinstance(date, datetime):