diff options
Diffstat (limited to 'lib/stwcs/updatewcs/npol.py')
| -rw-r--r-- | lib/stwcs/updatewcs/npol.py | 15 |
1 files changed, 14 insertions, 1 deletions
diff --git a/lib/stwcs/updatewcs/npol.py b/lib/stwcs/updatewcs/npol.py index acef50b..18f73b0 100644 --- a/lib/stwcs/updatewcs/npol.py +++ b/lib/stwcs/updatewcs/npol.py @@ -233,6 +233,7 @@ class NPOLCorr(object): if the science image is a subarray or binned image. """ npl = pyfits.open(npolfile) + npol_phdr = npl[0].header for ext in npl: try: nplextname = ext.header['EXTNAME'] @@ -294,7 +295,19 @@ class NPOLCorr(object): cdl.append(pyfits.Card(key=key, value=kw_val0[key], comment=kw_comm0[key])) for key in kw_comm1.keys(): cdl.append(pyfits.Card(key=key, value=kw_val1[key], comment=kw_comm1[key])) - + # Now add keywords from NPOLFILE header to document source of calibration + # include all keywords after and including 'FILENAME' from header + start_indx = -1 + end_indx = 0 + for c,i in zip(npol_phdr,range(len(npol_phdr))): + if c == 'FILENAME': + start_indx = i + if c == '': # remove blanks from end of header + end_indx = i+1 + break + if start_indx >= 0: + for card in npol_phdr[start_indx:end_indx]: + cdl.append(card) hdr = pyfits.Header(cards=cdl) |