diff options
author | Nadia Dencheva <nadia.dencheva@gmail.com> | 2016-08-04 17:42:01 -0400 |
---|---|---|
committer | Nadia Dencheva <nadia.dencheva@gmail.com> | 2016-08-04 17:42:01 -0400 |
commit | 4b65b226085ccb9e665a5866023d8114f7438188 (patch) | |
tree | 7183ed93c0624164930069bf5fedcd582dce84b6 /lib/stwcs/wcsutil/getinput.py | |
parent | 86d1bc5a77491770d45b86e5cf18b79ded68fb9b (diff) | |
download | stwcs_hcf-4b65b226085ccb9e665a5866023d8114f7438188.tar.gz |
restructure and add stwcs tests
Diffstat (limited to 'lib/stwcs/wcsutil/getinput.py')
-rw-r--r-- | lib/stwcs/wcsutil/getinput.py | 62 |
1 files changed, 0 insertions, 62 deletions
diff --git a/lib/stwcs/wcsutil/getinput.py b/lib/stwcs/wcsutil/getinput.py deleted file mode 100644 index 8ee1123..0000000 --- a/lib/stwcs/wcsutil/getinput.py +++ /dev/null @@ -1,62 +0,0 @@ -from astropy.io import fits -from stsci.tools import irafglob, fileutil, parseinput - -def parseSingleInput(f=None, ext=None): - if isinstance(f, str): - # create an HSTWCS object from a filename - if ext != None: - filename = f - if isinstance(ext,tuple): - if ext[0] == '': - extnum = ext[1] # handle ext=('',1) - else: - extnum = ext - else: - extnum = int(ext) - elif ext == None: - filename, ext = fileutil.parseFilename(f) - ext = fileutil.parseExtn(ext) - if ext[0] == '': - extnum = int(ext[1]) #handle ext=('',extnum) - else: - extnum = ext - phdu = fits.open(filename) - hdr0 = phdu[0].header - try: - ehdr = phdu[extnum].header - except (IndexError, KeyError) as e: - raise e.__class__('Unable to get extension %s.' % extnum) - - elif isinstance(f, fits.HDUList): - phdu = f - if ext == None: - extnum = 0 - else: - extnum = ext - ehdr = f[extnum].header - hdr0 = f[0].header - filename = hdr0.get('FILENAME', "") - - else: - raise ValueError('Input must be a file name string or a' - '`astropy.io.fits.HDUList` object') - - return filename, hdr0, ehdr, phdu - - -def parseMultipleInput(input): - if isinstance(input, str): - if input[0] == '@': - # input is an @ file - filelist = irafglob.irafglob(input) - else: - try: - filelist, output = parseinput.parseinput(input) - except IOError: raise - elif isinstance(input, list): - if isinstance(input[0], wcsutil.HSTWCS): - # a list of HSTWCS objects - return input - else: - filelist = input[:] - return filelist |