diff options
Diffstat (limited to 'updatewcs/det2im.py')
| -rw-r--r-- | updatewcs/det2im.py | 20 |
1 files changed, 11 insertions, 9 deletions
diff --git a/updatewcs/det2im.py b/updatewcs/det2im.py index 90fe339..c8e9ab8 100644 --- a/updatewcs/det2im.py +++ b/updatewcs/det2im.py @@ -42,8 +42,8 @@ class DET2IMCorr(object): getAxisCorr = classmethod(getAxisCorr) def applyDet2ImCorr(cls,fobj, axiscorr): - d2imfile = fileutil.osfn(fobj[0].header['D2IMFILE']) - hdu = cls.createDgeoHDU(d2imfile, axiscorr) + + hdu = cls.createDgeoHDU(fobj, axiscorr) d2imarr_ind = cls.getD2imIndex(fobj) if d2imarr_ind: fobj[d2imarr_ind] = hdu @@ -63,30 +63,32 @@ class DET2IMCorr(object): return index getD2imIndex = classmethod(getD2imIndex) - def createDgeoHDU(cls, d2imfile, axiscorr): - + def createDgeoHDU(cls, fobj, axiscorr): + d2imfile = fileutil.osfn(fobj[0].header['D2IMFILE']) d2im_data = pyfits.getdata(d2imfile, ext=1) - d2im_hdr = cls.createDet2ImHdr(d2im_data.shape, axiscorr) + sci_hdr = fobj['sci',1].header + d2im_hdr = cls.createDet2ImHdr(sci_hdr, d2im_data.shape, axiscorr) hdu = pyfits.ImageHDU(header=d2im_hdr, data=d2im_data) return hdu createDgeoHDU = classmethod(createDgeoHDU) - def createDet2ImHdr(cls, data_shape, axiscorr): + def createDet2ImHdr(cls, sci_hdr, data_shape, axiscorr): """ Creates a header for the D2IMARR extension based on the reference file recorded in D2IMFILE keyword in the primary header. """ - + ltv1 = sci_hdr.get('LTV1', 0.0) + ltv2 = sci_hdr.get('LTV2', 0.0) naxis1 = data_shape[0] naxis2 = 0 crpix1 = 0.0 crpix2 = 0.0 cdelt1 = 1.0 cdelt2 = 1.0 - crval1 = 0.0 - crval2 = 0.0 + crval1 = 0.0 + ltv1 + crval2 = 0.0 + ltv2 keys = ['XTENSION','BITPIX','NAXIS','NAXIS1','NAXIS2', 'EXTNAME','EXTVER','PCOUNT','GCOUNT','CRPIX1', 'CDELT1','CRVAL1','CRPIX2','CDELT2','CRVAL2', 'AXISCORR'] |