Harrison 3029 - PMCL - CoDICE Lo direct events: Convert spin_sector from float to int when reading L2 to workaround a bug

Author: Menlo Innovations - CAVA Project

imap_l3_processing/codice/l3/lo/models.py

@@ -128,7 +128,7 @@ def read_from_cdf(cls, l2_direct_event_cdf: Path):
                 num_events=read_variable_and_mask_fill_values(cdf["num_events"])[:, :CODICE_LO_L2_NUM_PRIORITIES, ...],
                 spin_angle=read_variable_and_mask_fill_values(cdf["spin_angle"])[:, :CODICE_LO_L2_NUM_PRIORITIES, ...],
                 spin_sector=read_variable_and_mask_fill_values(cdf["spin_sector"])[:, :CODICE_LO_L2_NUM_PRIORITIES,
-                            ...],
+                            ...].astype(int),
                 elevation_angle=read_variable_and_mask_fill_values(cdf["elevation_angle"])[:,
                                 :CODICE_LO_L2_NUM_PRIORITIES, ...],
                 tof=read_variable_and_mask_fill_values(cdf["tof"])[:, :CODICE_LO_L2_NUM_PRIORITIES, ...],

tests/codice/l3/lo/test_models.py

@@ -547,7 +547,7 @@ def test_codice_lo_l2_direct_events_reads_from_correct_float_data(self):
                 cdf["energy_step"] = rng.random(cdf["energy_step"].shape)
                 cdf["energy_per_charge"] = rng.random(cdf["energy_per_charge"].shape)
                 cdf["spin_angle"] = rng.random(cdf["spin_angle"].shape)
-                cdf["spin_sector"] = rng.random(cdf["spin_sector"].shape)
+                cdf["spin_sector"] =  rng.integers(0, 24, cdf["spin_sector"].shape)
                 cdf["elevation_angle"] = rng.random(cdf["elevation_angle"].shape)
                 cdf["position"] = rng.random(cdf["position"].shape)
 
@@ -561,6 +561,22 @@ def test_codice_lo_l2_direct_events_reads_from_correct_float_data(self):
                 np.testing.assert_array_equal(l2_direct_event.spin_angle, cdf["spin_angle"][:, :7, ...])
                 np.testing.assert_array_equal(l2_direct_event.elevation_angle, cdf["elevation_angle"][:, :7, ...])
 
+    def test_workaround_to_convert_float_spin_sector_to_int(self):
+        all_fill_l2_cdf_path = get_test_data_path('codice/imap_codice_l2_lo-direct-events_20260307_v003-all-fill.cdf')
+
+        rng = np.random.default_rng()
+        with tempfile.TemporaryDirectory() as tmpdir:
+            tmpdir = Path(tmpdir)
+
+            test_cdf_path = tmpdir / "test_cdf.cdf"
+            with CDF(str(test_cdf_path), masterpath=str(all_fill_l2_cdf_path)) as cdf:
+                expected_spin_sector = rng.integers(0, 24, cdf["spin_sector"].shape, dtype=int)
+                cdf["spin_sector"] = expected_spin_sector.astype(float)
+
+            l2_direct_event = CodiceLoL2DirectEventData.read_from_cdf(test_cdf_path)
+
+            np.testing.assert_array_equal(l2_direct_event.spin_sector, expected_spin_sector[:, :7, ...], strict=True)
+
     def test_codice_lo_l2_direct_events_read_from_cdf_handles_fill_value(self):
         all_fill_l2_cdf_path = get_test_data_path('codice/imap_codice_l2_lo-direct-events_20260307_v003-all-fill.cdf')
         l2_direct_event = CodiceLoL2DirectEventData.read_from_cdf(all_fill_l2_cdf_path)