From d93a10017d62f39d80167b45c1044a5e113f5994 Mon Sep 17 00:00:00 2001
From: embray <embray@stsci.edu>
Date: Wed, 22 Jun 2011 23:24:07 +0000
Subject: Redoing the r13221-13223 merge in the actual trunk now.  This updates
 trunk to the setup_refactoring branch (however, coords, pysynphot, and pywcs
 are still being pulled from the astrolib setup_refactoring branch.  Will have
 to do that separately then update the svn:externals)

git-svn-id: http://svn.stsci.edu/svn/ssb/stsci_python/stsci_python/trunk/stwcs@13225 fe389314-cf27-0410-b35b-8c050e845b92
---
 lib/stwcs/wcsutil/getinput.py | 62 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 62 insertions(+)
 create mode 100644 lib/stwcs/wcsutil/getinput.py

(limited to 'lib/stwcs/wcsutil/getinput.py')

diff --git a/lib/stwcs/wcsutil/getinput.py b/lib/stwcs/wcsutil/getinput.py
new file mode 100644
index 0000000..2f64f46
--- /dev/null
+++ b/lib/stwcs/wcsutil/getinput.py
@@ -0,0 +1,62 @@
+import pyfits
+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 = pyfits.open(filename)
+        hdr0 = pyfits.getheader(filename)
+        try:
+            ehdr = pyfits.getheader(filename, ext=extnum)
+        except (IndexError,KeyError):
+            print 'Unable to get extension.', extnum
+            raise
+
+    elif isinstance(f, pyfits.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 pyfits file 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
\ No newline at end of file
-- 
cgit