Add z_qso to tracer class

Author: andreicuceu

lya_2pt/read_io.py

@@ -33,6 +33,7 @@ def read_from_image(hdul, absorption_line, healpix_id, need_distortion=False, pr
     los_id_array = hdul["METADATA"]["LOS_ID"][:]
     ra_array = hdul["METADATA"]["RA"][:]
     dec_array = hdul["METADATA"]["DEC"][:]
+    z_qso_array = hdul["METADATA"]["Z"][:]
     dwave = hdul["LAMBDA"].read_header()['DELTA_LAMBDA']
 
     deltas_array = hdul["DELTA"].read().astype(float)
@@ -52,10 +53,12 @@ def read_from_image(hdul, absorption_line, healpix_id, need_distortion=False, pr
             "Did not find LOGLAM or LAMBDA in delta file")
 
     tracers = np.empty(los_id_array.shape, dtype=Tracer)
-    for i, (los_id, ra, dec) in enumerate(zip(los_id_array, ra_array, dec_array)):
+    for i, (los_id, ra, dec, z_qso) in enumerate(
+        zip(los_id_array, ra_array, dec_array, z_qso_array)
+    ):
         mask = ~np.isnan(deltas_array[i])
         tracers[i] = Tracer(
-            healpix_id, los_id, ra, dec, projection_order, deltas_array[i][mask],
+            healpix_id, los_id, ra, dec, z_qso, projection_order, deltas_array[i][mask],
             weights_array[i][mask], log_lambda[mask], z[mask], need_distortion
             )
 
@@ -97,6 +100,7 @@ def read_from_hdu(hdul, absorption_line, healpix_id, need_distortion=False, proj
         los_id = header["LOS_ID"]
         ra = header['RA']
         dec = header['DEC']
+        z_qso = header['Z']
 
         delta = hdu["DELTA"][:].astype(float)
         weights = hdu["WEIGHT"][:].astype(float)
@@ -114,7 +118,7 @@ def read_from_hdu(hdul, absorption_line, healpix_id, need_distortion=False, proj
                 "Did not find LOGLAM or LAMBDA in delta file")
 
         tracers.append(Tracer(
-            healpix_id, los_id, ra, dec, projection_order, delta, weights,
+            healpix_id, los_id, ra, dec, z_qso, projection_order, delta, weights,
             log_lambda, z, need_distortion
             ))
 

lya_2pt/tracer.py

@@ -75,7 +75,7 @@ class Tracer:
     project():
         Apply projection matrix to deltas
     """
-    def __init__(self, healpix_id, los_id, ra, dec, order,
+    def __init__(self, healpix_id, los_id, ra, dec, z_qso, order,
                  deltas, weights, log_lambda, z, need_distortion=False):
         """Initializes class instance
 
@@ -89,6 +89,8 @@ def __init__(self, healpix_id, los_id, ra, dec, order,
             Line of sight's right ascension
         dec: float
             Line of sight's declination
+        z_qso: float
+            Background quasar redshift
         order: int
             Order of polynomial used for the continuum fitting
         deltas: array of float
@@ -108,6 +110,7 @@ def __init__(self, healpix_id, los_id, ra, dec, order,
         self.los_id = los_id
         self.ra = ra
         self.dec = dec
+        self.z_qso = z_qso
         self.order = order
 
         self.x_cart = np.cos(ra) * np.cos(dec)