summaryrefslogtreecommitdiff
path: root/lib/stwcs/wcsutil/getinput.py
diff options
context:
space:
mode:
Diffstat (limited to 'lib/stwcs/wcsutil/getinput.py')
-rw-r--r--lib/stwcs/wcsutil/getinput.py62
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