JP-3436 Empty wavelength array in s2d products

CHANGES.rst

@@ -92,6 +92,8 @@ resample
 - Remove sleep in median combination added in 8305 as it did not address
   the issue in operation [#8419]
 
+- Populated the wavelength attribute in the result model. [#8374]
+
 residual_fringe
 ---------------
 

jwst/regtest/test_miri_lrs_slit_spec3.py

@@ -1,6 +1,8 @@
 """ Test of the spec3 pipeline using MIRI LRS fixed-slit exposures.
     This takes an association and generates the level 3 products."""
 import pytest
+import numpy as np
+from gwcs import wcstools
 
 import asdf
 from astropy.io.fits.diff import FITSDiff
@@ -76,3 +78,26 @@ def test_miri_lrs_slit_spec3(run_pipeline, rtdata_module, fitsdiff_default_kwarg
     # Compare the results
     diff = FITSDiff(rtdata.output, rtdata.truth, **fitsdiff_default_kwargs)
     assert diff.identical, diff.report()
+
+    if "s2d" in output:
+        # Compare the calculated wavelengths
+        tolerance = 1e-03
+        dmt = datamodels.open(rtdata.truth)
+        dmr = datamodels.open(rtdata.output)
+        if isinstance(dmt, datamodels.MultiSlitModel):
+            names = [s.name for s in dmt.slits]
+            for name in names:
+                st_idx = [(s.wcs, s.wavelength) for s in dmt.slits if s.name==name]
+                w = dmt.slits[st_idx].meta.wcs
+                x, y = wcstools.grid_from_bounding_box(w.bounding_box, step=(1, 1), center=True)
+                _, _, wave = w(x, y)
+                sr_idx = [(s.wcs, s.wavelength) for s in dmr.slits if s.name==name]
+                wlr = dmr.slits[sr_idx].wavelength
+                assert np.all(np.isclose(wave, wlr, atol=tolerance))
+        else:
+            w = dmt.meta.wcs
+            x, y = wcstools.grid_from_bounding_box(w.bounding_box, step=(1, 1), center=True)
+            _, _, wave = w(x, y)
+            wlr = dmr.wavelength
+            assert np.all(np.isclose(wave, wlr, atol=tolerance))
+

jwst/regtest/test_nirspec_fs_spec3.py

@@ -1,7 +1,10 @@
 from astropy.io.fits.diff import FITSDiff
 import pytest
+import numpy as np
+from gwcs import wcstools
 
 from jwst.stpipe import Step
+from stdatamodels.jwst import datamodels
 
 
 @pytest.fixture(scope="module")
@@ -42,3 +45,25 @@ def test_nirspec_fs_spec3(run_pipeline, rtdata_module, fitsdiff_default_kwargs,
     # Compare the results
     diff = FITSDiff(rtdata.output, rtdata.truth, **fitsdiff_default_kwargs)
     assert diff.identical, diff.report()
+
+    if "s2d" in output:
+        # Compare the calculated wavelengths
+        tolerance = 1e-03
+        dmt = datamodels.open(rtdata.truth)
+        dmr = datamodels.open(rtdata.output)
+        if isinstance(dmt, datamodels.MultiSlitModel):
+            names = [s.name for s in dmt.slits]
+            for name in names:
+                st_idx = [(s.wcs, s.wavelength) for s in dmt.slits if s.name==name]
+                w = dmt.slits[st_idx].meta.wcs
+                x, y = wcstools.grid_from_bounding_box(w.bounding_box, step=(1, 1), center=True)
+                _, _, wave = w(x, y)
+                sr_idx = [(s.wcs, s.wavelength) for s in dmr.slits if s.name==name]
+                wlr = dmr.slits[sr_idx].wavelength
+                assert np.all(np.isclose(wave, wlr, atol=tolerance))
+        else:
+            w = dmt.meta.wcs
+            x, y = wcstools.grid_from_bounding_box(w.bounding_box, step=(1, 1), center=True)
+            _, _, wave = w(x, y)
+            wlr = dmr.wavelength
+            assert np.all(np.isclose(wave, wlr, atol=tolerance))

jwst/regtest/test_nirspec_mos_spec3.py

@@ -1,7 +1,10 @@
 import pytest
 from astropy.io.fits.diff import FITSDiff
+import numpy as np
+from gwcs import wcstools
 
 from jwst.stpipe import Step
+from stdatamodels.jwst import datamodels
 
 
 @pytest.fixture(scope="module")
@@ -36,3 +39,17 @@ def test_nirspec_mos_spec3(run_pipeline, suffix, source_id, fitsdiff_default_kwa
 
     diff = FITSDiff(rtdata.output, rtdata.truth, **fitsdiff_default_kwargs)
     assert diff.identical, diff.report()
+
+    if "s2d" in output:
+        # Compare the calculated wavelengths
+        dmt = datamodels.open(rtdata.truth)
+        dmr = datamodels.open(rtdata.output)
+        names = [s.name for s in dmt.slits]
+        for name in names:
+            st_idx = [(s.wcs, s.wavelength) for s in dmt.slits if s.name==name]
+            w = dmt.slits[st_idx].meta.wcs
+            x, y = wcstools.grid_from_bounding_box(w.bounding_box, step=(1, 1), center=True)
+            _, _, wave = w(x, y)
+            sr_idx = [(s.wcs, s.wavelength) for s in dmr.slits if s.name==name]
+            wlr = dmr.slits[sr_idx].wavelength
+            assert np.all(np.isclose(wave, wlr, atol=1e-03))

jwst/resample/resample_spec_step.py

@@ -7,6 +7,8 @@
 from . import resample_spec, ResampleStep
 from ..exp_to_source import multislit_to_container
 from ..assign_wcs.util import update_s_region_spectral
+from jwst.lib.wcs_utils import get_wavelengths
+
 
 # Force use of all DQ flagged data except for DO_NOT_USE and NON_SCIENCE
 GOOD_BITS = '~DO_NOT_USE+NON_SCIENCE'
@@ -109,6 +111,16 @@ def process(self, input):
         result.meta.asn.table_name = input_models[0].meta.asn.table_name
         result.meta.asn.pool_name = input_models[0].meta.asn.pool_name
 
+        # populate the result wavelength attribute for MultiSlitModel
+        if isinstance(result, MultiSlitModel):
+            for slit_idx, slit in enumerate(result.slits):
+                wl_array = get_wavelengths(result.slits[slit_idx])
+                result.slits[slit_idx].wavelength = wl_array
+        else:
+            # populate the result wavelength attribute for SlitModel
+            wl_array = get_wavelengths(result)
+            result.wavelength = wl_array
+
         return result
 
     def _process_multislit(self, input_models):