def offset_wcs(slit_wcs):
"""
Prepend a Shift transform to the slit WCS to account for subarrays.
Parameters
----------
slit_wcs : `~gwcs.wcs.WCS`
The WCS for this slit.
slit_name : str
The name of the slit.
"""
xlo, xhi = _toindex(slit_wcs.bounding_box[0])
ylo, yhi = _toindex(slit_wcs.bounding_box[1])
# Add the slit offset to each slit WCS object
tr = slit_wcs.get_transform('detector', 'sca')
tr = Shift(xlo) & Shift(ylo) | tr
slit_wcs.set_transform('detector', 'sca', tr.rename('dms2sca'))
return xlo, xhi, ylo, yhi
def offset_wcs(slit_wcs):
"""
Prepend a Shift transform to the slit WCS to account for subarrays.
Parameters
----------
slit_wcs : `~gwcs.wcs.WCS`
The WCS for this slit.
slit_name : str
The name of the slit.
"""
xlo, xhi = _toindex(slit_wcs.bounding_box[0])
ylo, yhi = _toindex(slit_wcs.bounding_box[1])
# Add the slit offset to each slit WCS object
tr = slit_wcs.get_transform('detector', 'sca')
tr = Shift(xlo) & Shift(ylo) | tr
slit_wcs.set_transform('detector', 'sca', tr.rename('dms2sca'))
return xlo, xhi, ylo, yhi
def offset_wcs(slit_wcs, slit_name):
"""
Prepend a Shift to the slit WCS to account for subarrays.
Parameters
----------
slit_wcs : `~gwcs.wcs.WCS`
The WCS for this slit.
slit_name : str
The name of the slit.
"""
xlo, xhi = _toindex(slit_wcs.bounding_box[0])
ylo, yhi = _toindex(slit_wcs.bounding_box[1])
# Add the slit offset to each slit WCS object
tr = slit_wcs.get_transform('detector', 'sca')
tr = Shift(xlo) & Shift(ylo) | tr
slit_wcs.set_transform('detector', 'sca', tr.rename('dms2sca'))
log.info('Name of subarray extracted: %s', slit_name)
log.info('Subarray x-extents are: %s %s', xlo, xhi)
log.info('Subarray y-extents are: %s %s', ylo, yhi)
return xlo, xhi, ylo, yhi
def extract2d(input_model, which_subarray=None):
supported_modes = [
'NRS_FIXEDSLIT', 'NRS_MSASPEC', 'NRS_BRIGHTOBJ', 'NRS_LAMP'
]
exp_type = input_model.meta.exposure.type.upper()
log.info('EXP_TYPE is {0}'.format(exp_type))
if exp_type not in supported_modes:
input_model.meta.cal_step.extract_2d = 'SKIPPED'
return input_model
else:
output_model = datamodels.MultiSlitModel()
output_model.update(input_model)
if exp_type in supported_modes:
slit2msa = input_model.meta.wcs.get_transform('slit_frame',
'msa_frame')
# This is a cludge but will work for now.
# This model keeps open_slits as an attribute.
open_slits = slit2msa[1].slits[:]
if which_subarray is not None:
open_slits = [
sub for sub in open_slits if sub.name == which_subarray
]
log.debug('open slits {0}'.format(open_slits))
for slit in open_slits:
slit_wcs = nirspec.nrs_wcs_set_input(input_model, slit.name)
xlo, xhi = _toindex(slit_wcs.bounding_box[0])
ylo, yhi = _toindex(slit_wcs.bounding_box[1])
# Add the slit offset to each slit WCS object
tr = slit_wcs.get_transform('detector', 'sca')
tr = Shift(xlo) & Shift(ylo) | tr
slit_wcs.set_transform('detector', 'sca', tr.rename('dms2sca'))
log.info('Name of subarray extracted: %s', slit.name)
log.info('Subarray x-extents are: %s %s', xlo, xhi)
log.info('Subarray y-extents are: %s %s', ylo, yhi)
ext_data = input_model.data[ylo:yhi + 1, xlo:xhi + 1].copy()
ext_err = input_model.err[ylo:yhi + 1, xlo:xhi + 1].copy()
ext_dq = input_model.dq[ylo:yhi + 1, xlo:xhi + 1].copy()
new_model = datamodels.ImageModel(data=ext_data,
err=ext_err,
dq=ext_dq)
shape = ext_data.shape
bounding_box = ((0, shape[1] - 1), (0, shape[0] - 1))
slit_wcs.bounding_box = bounding_box
new_model.meta.wcs = slit_wcs
output_model.slits.append(new_model)
# set x/ystart values relative to the image (screen) frame.
# The overall subarray offset is recorded in model.meta.subarray.
nslit = len(output_model.slits) - 1
xlo_ind, xhi_ind, ylo_ind, yhi_ind = _toindex(
(xlo, xhi, ylo, yhi)).astype(np.int16)
output_model.slits[nslit].name = str(slit.name)
output_model.slits[nslit].xstart = xlo_ind + 1
output_model.slits[nslit].xsize = (xhi_ind - xlo_ind) + 1
output_model.slits[nslit].ystart = ylo_ind + 1
output_model.slits[nslit].ysize = (yhi_ind - ylo_ind) + 1
if exp_type.lower() == 'nrs_msaspec':
output_model.slits[nslit].source_id = int(slit.source_id)
output_model.slits[nslit].source_name = slit.source_name
output_model.slits[nslit].source_alias = slit.source_alias
output_model.slits[nslit].catalog_id = slit.catalog_id
output_model.slits[nslit].stellarity = float(slit.stellarity)
output_model.slits[nslit].source_xpos = float(slit.source_xpos)
output_model.slits[nslit].source_ypos = float(slit.source_ypos)
output_model.slits[nslit].slitlet_id = int(slit.name)
# for pathloss correction
output_model.slits[nslit].nshutters = int(slit.nshutters)
del input_model
# Set the step status to COMPLETE
output_model.meta.cal_step.extract_2d = 'COMPLETE'
return output_model
