From f8d5962c3a4b0f6850fbb392999cf3df6150163a Mon Sep 17 00:00:00 2001 From: dencheva Date: Mon, 14 Mar 2011 17:57:44 +0000 Subject: temporary work around pyfits problem #48 git-svn-id: http://svn.stsci.edu/svn/ssb/stsci_python/stsci_python/trunk/stwcs@12192 fe389314-cf27-0410-b35b-8c050e845b92 --- wcsutil/headerlet.py | 25 ++++++++++++++++++------- 1 file changed, 18 insertions(+), 7 deletions(-) (limited to 'wcsutil') diff --git a/wcsutil/headerlet.py b/wcsutil/headerlet.py index 70ebc7c..b3b7092 100644 --- a/wcsutil/headerlet.py +++ b/wcsutil/headerlet.py @@ -118,10 +118,8 @@ def createHeaderlet(fname, hdrname, destim=None, output=None): raise ValueError, 'Please provide a value for the DESTIM keyword' if hdrname is None: raise ValueError, "Please provide a name for the headerlet, HDRNAME is a required parameter." - if output is None: - output = hdrname+'_hdr.fits' - elif not output.endswith('_hdr.fits'): - output = output+'_hdr.fits' + + altkeys = altwcs.wcskeys(fobj[('SCI',1)].header) try: @@ -181,15 +179,28 @@ def createHeaderlet(fname, hdrname, destim=None, output=None): comment='Maximum error of D2IMARR')) hdu = pyfits.ImageHDU(header=pyfits.Header(h)) + # temporary fix for pyfits ticket # 48 + hdu._extver = e hdul.append(hdu) numwdvarr = countext(fname, 'WCSDVARR') numd2im = countext(fname, 'D2IMARR') for w in range(1, numwdvarr+1): - hdul.append(fobj[('WCSDVARR',w)].copy()) + hdu = fobj[('WCSDVARR',w)].copy() + # temporary fix for pyfits ticket # 48 + hdu._extver = w + hdul.append(hdu) for d in range(1, numd2im+1): - hdul.append(fobj[('D2IMARR',d)].copy()) - hdul.writeto(output,clobber=True) + hdu = fobj[('D2IMARR',d)].copy() + # temporary fix for pyfits ticket # 48 + hdu._extver = d + hdul.append(hdu) + if output is not None: + # write the headerlet to a file + if not output.endswith('_hdr.fits'): + output = output+'_hdr.fits' + hdul.writeto(output,clobber=True) fobj.close() + return Headerlet(hdul) def applyHeaderlet(hdrfile, destfile, destim=None, hdrname=None, createheaderlet=True): """ -- cgit