diff --git a/spaceKLIP/database.py b/spaceKLIP/database.py index 99e4aa82..f70f4627 100644 --- a/spaceKLIP/database.py +++ b/spaceKLIP/database.py @@ -109,6 +109,7 @@ def read_jwst_s012_data(self, datapaths, psflibpaths=None, bgpaths=None, + cr_from_saif=False, assoc_using_targname=True): """ Read JWST stage 0 (*uncal), 1 (*rate or *rateints), or 2 (*cal or @@ -274,10 +275,12 @@ def read_jwst_s012_data(self, BLURFWHM += [head.get('BLURFWHM', np.nan)] head = hdul['SCI'].header BUNIT += [head.get('BUNIT', 'NONE')] - # CRPIX1 += [head.get('CRPIX1', np.nan)] - # CRPIX2 += [head.get('CRPIX2', np.nan)] - CRPIX1 += [ap.XSciRef] - CRPIX2 += [ap.YSciRef] + if cr_from_saif: + CRPIX1 += [ap.XSciRef] + CRPIX2 += [ap.YSciRef] + else: + CRPIX1 += [head.get('CRPIX1', np.nan)] + CRPIX2 += [head.get('CRPIX2', np.nan)] VPARITY += [head.get('VPARITY', -1)] V3I_YANG += [head.get('V3I_YANG', 0.)] RA_REF += [head.get('RA_REF', np.nan)] @@ -573,7 +576,8 @@ def read_jwst_s012_data(self, pass def read_jwst_s3_data(self, - datapaths): + datapaths, + cr_from_saif=False): """ Read JWST stage 3 data (this can be *i2d data from the official JWST pipeline, or data products from the pyKLIP and classical PSF @@ -720,10 +724,12 @@ def read_jwst_s3_data(self, if TYPE[-1] == 'CORON3': head = hdul['SCI'].header BUNIT += [head.get('BUNIT', 'NONE')] - # CRPIX1 += [head.get('CRPIX1', np.nan)] - # CRPIX2 += [head.get('CRPIX2', np.nan)] - CRPIX1 += [ap.XSciRef] - CRPIX2 += [ap.YSciRef] + if cr_from_saif: + CRPIX1 += [ap.XSciRef] + CRPIX2 += [ap.YSciRef] + else: + CRPIX1 += [head.get('CRPIX1', np.nan)] + CRPIX2 += [head.get('CRPIX2', np.nan)] HASH += [TELESCOP[-1] + '_' + INSTRUME[-1] + '_' + DETECTOR[-1] + '_' + FILTER[-1] + '_' + PUPIL[-1] + '_' + CORONMSK[-1] + '_' + SUBARRAY[-1]] hdul.close() TYPE = np.array(TYPE)