Add ability to make html reports.

transientlunatic · Sep 3, 2024 · d8ff2ed · d8ff2ed
1 parent e20cfb2
commit d8ff2ed
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 11 deletions.
diff --git a/minke/injection.py b/minke/injection.py
@@ -10,6 +10,8 @@
 
 import astropy.units as u
 
+import otter
+
 from .models.lalsimulation import SEOBNRv3, IMRPhenomPv2
 from .models.lalnoise import KNOWN_PSDS
 from .detector import KNOWN_IFOS
@@ -39,7 +41,6 @@ def make_injection(
 
     waveform_model = waveform()
 
-
     injections = {}
     for detector, psd_model in detectors.items():
         logger.info(f"Making injection for {detector}")
@@ -151,10 +152,18 @@ def injection(settings):
         }
 
         logging.basicConfig(level=LOGGER_LEVELS[level])
-
+    settings_ = settings
     settings = settings["injection"]
     parameters = injection_parameters_add_units(settings["parameters"])
 
+    report = otter.Otter(os.path.join(settings_.get("pages directory", "."), "injection.html")
+                         title="Minke Injection"
+                         )
+
+    with report:
+        report + "# Injection frames"
+        report + settings
+
     detector_dict = {
         settings["interferometers"][ifo]: settings["psds"][ifo]
         for ifo in settings["interferometers"]
@@ -171,3 +180,8 @@ def injection(settings):
         psdfile="psd",
     )
     data = injections
+
+    for injection in injections:
+        f = injection.plot()
+        with report:
+            report + f
diff --git a/minke/models/lalsimulation.py b/minke/models/lalsimulation.py
@@ -179,13 +179,6 @@ def time_domain(self, parameters=None, times=None, **kwargs):
             self._cache = WaveformDict(parameters=parameters, plus=hp_ts, cross=hx_ts)
 
         return self._cache
-
-class IMRPhenomPv1(LALSimulationApproximant):
-    def __init__(self):
-        super().__init__()
-        self._args["approximant"] = lalsimulation.GetApproximantFromString(
-            "IMRPhenomP"
-        )
 
 class IMRPhenomPv2(LALSimulationApproximant):
     def __init__(self):
@@ -203,5 +196,3 @@ class SEOBNRv3(LALSimulationApproximant):
     def __init__(self):
         super().__init__()
         self._args["approximant"] = lalsimulation.GetApproximantFromString("SEOBNRv3")
-
-