Merge pull request #10 from HERA-Team/lines

Lines
HERA-Team · Sep 15, 2021 · 9218ea6 · 9218ea6
2 parents 82507d1 + 0997c23
commit 9218ea6
Showing 1 changed file with 22 additions and 1 deletion.
diff --git a/hera_cc_utils/survey.py b/hera_cc_utils/survey.py
@@ -7,6 +7,7 @@
 import numpy as np
 import matplotlib.pyplot as plt
 from astropy import constants as const
+from matplotlib.patches import Rectangle
 from astropy.cosmology import FlatLambdaCDM
 
 from .line import Line
@@ -50,6 +51,20 @@
         "freq_range": [200000.0, 360000.0],
         "angular_res": 5.8178 * 1e-5,
     },
+    "roman": {
+        "target_lines": ["Lya"],
+        "rval": 461,  # Really 461 * wave, with `wave` in microns.
+        "hemisphere": "S",
+        "lambda_range": [1, 1.93],
+        "angular_res": 5.333e-07,  # 0.11 arcsec/pixel
+    },
+    "euclid": {
+        "target_lines": ["Lya"],
+        "rval": 250,
+        "hemisphere": "S",
+        "lambda_range": [0.92, 2.0],
+        "angular_res": 4.848e-07,  # 0.1 arcsec/pixel
+    },
 }
 
 survey_registry["ccatp"] = survey_registry["fyst"]
@@ -419,7 +434,13 @@ def plot_coverage_k(self, ax=None, fig=1, **kwargs):
         kperp = self.get_kperp_range()
 
         for line in self.get_target_lines():
-            ax.fill_between(kperp[line], *kpara[line], **kwargs)
+            rect = Rectangle(
+                (kperp[line][0], kpara[line][0]),
+                np.diff(kperp[line])[0],
+                np.diff(kpara[line])[0],
+                **kwargs
+            )
+            ax.add_patch(rect)
 
         ax.set_xscale("log")
         ax.set_yscale("log")