some bugs cleaned, now the convolution works

litebird · Dec 6, 2024 · 98522ca · 98522ca
1 parent e38f8fd
commit 98522ca
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 12 deletions.
diff --git a/litebird_sim/__init__.py b/litebird_sim/__init__.py
@@ -18,7 +18,7 @@
     rle_compress,
     rle_decompress,
 )
-from .beam_convolution import add_convolved_sky_to_observations
+from .beam_convolution import add_convolved_sky_to_observations, BeamConvolutionParameters
 from .coordinates import (
     DEFAULT_COORDINATE_SYSTEM,
     DEFAULT_TIME_SCALE,
@@ -177,6 +177,7 @@ def destripe_with_toast2(*args, **kwargs):
     "PTEP_IMO_LOCATION",
     # beam_convolution.py
     "add_convolved_sky_to_observations",
+    "BeamConvolutionParameters",
     # compress.py
     "rle_compress",
     "rle_decompress",

diff --git a/litebird_sim/beam_convolution.py b/litebird_sim/beam_convolution.py
@@ -46,7 +46,18 @@ def add_convolved_sky_to_one_detector(
     """ """
 
     if not convolution_params:
-        default_lmax = min(len(sky_alms_det), len(beam_alms_det))
+        if sky_alms_det.ndim == 2:
+            sky_lamx = len(sky_alms_det[0])
+        else:
+            sky_lamx = len(sky_alms_det)
+
+        if beam_alms_det.ndim == 2:
+            beam_lamx = len(beam_alms_det[0])
+        else:
+            beam_lamx = len(beam_alms_det)
+
+
+        default_lmax = min(sky_lamx, beam_lamx)
         default_kmax = default_lmax
         logging.warning(
             (
@@ -58,7 +69,7 @@ def add_convolved_sky_to_one_detector(
         )
         convolution_params = BeamConvolutionParameters(
             lmax=default_lmax,
-            kmax=default_kmax,
+            kmax=default_kmax-4,
         )
 
     if (
@@ -76,16 +87,16 @@ def add_convolved_sky_to_one_detector(
         tod_det += inter.interpol(pointings_det)
     else:
         fullconv = MuellerConvolver(
-            convolution_params.lmax,
-            convolution_params.kmax,
-            sky_alms_det,
-            beam_alms_det,
-            mueller_matrix,
+            lmax = convolution_params.lmax,
+            kmax = convolution_params.kmax,
+            slm = sky_alms_det,
+            blm = beam_alms_det,
+            mueller = mueller_matrix,
             single_precision=convolution_params.single_precision,
             epsilon=convolution_params.epsilon,
             nthreads=nthreads,
         )
-        tod_det += fullconv.signal(pointings_det, hwp_angle)
+        tod_det += fullconv.signal(ptg=pointings_det, alpha=hwp_angle)
 
 
 def add_convolved_sky(
@@ -114,12 +125,12 @@ def add_convolved_sky(
         else:
             curr_pointings_det, hwp_angle = pointings(detector_idx)
 
-        if hwp_angle is None:
-            hwp_angle = 0
-
         if input_sky_alms_in_galactic:
             curr_pointings_det = rotate_coordinates_e2g(curr_pointings_det)
 
+        #FIXME: Fix this at some point ducc wants phi 0 -> 2pi
+        curr_pointings_det[:,1] = np.mod(curr_pointings_det[:,1], 2 * np.pi)
+
         if input_sky_names is None:
             sky_alms_det = sky_alms
         else: