From f295c5599888f33ea53c82d522505d0c441939b7 Mon Sep 17 00:00:00 2001 From: dencheva Date: Fri, 24 Apr 2009 21:27:53 +0000 Subject: More formatting corrections git-svn-id: http://svn.stsci.edu/svn/ssb/stsci_python/development/trunk/updatewcs@7948 fe389314-cf27-0410-b35b-8c050e845b92 --- wcsutil/__init__.py | 24 ++++++++++++++++++++---- 1 file changed, 20 insertions(+), 4 deletions(-) (limited to 'wcsutil/__init__.py') diff --git a/wcsutil/__init__.py b/wcsutil/__init__.py index cdc847d..aa67b9a 100644 --- a/wcsutil/__init__.py +++ b/wcsutil/__init__.py @@ -237,7 +237,11 @@ class HSTWCS(WCS): def all_pix2sky(self, *args, **kwargs): origin = self.get_origin(*args) if self.det2imext != None and self.d2imerr > self.minerr: - return WCS.all_pix2sky(self, self.det2im(*args),origin ) + img = WCS.all_pix2sky(self, self.det2im(*args),origin ) + if len(args) == 3: + return [img[:, i] for i in range(img.shape[1])] + else: + return img else: return WCS.all_pix2sky(self, *args) all_pix2sky.__doc__ = """ @@ -250,7 +254,11 @@ class HSTWCS(WCS): def pix2foc(self, *args, **kwargs): origin = self.get_origin(*args) if self.det2imext != None and self.d2imerr > self.minerr: - return WCS.pix2foc(self, self.det2im(*args), origin) + img = WCS.pix2foc(self, self.det2im(*args), origin) + if len(args)==3: + return [img[:, i] for i in range(img.shape[1])] + else: + return img else: return WCS.pix2foc(self, *args) @@ -264,7 +272,11 @@ class HSTWCS(WCS): def p4_pix2foc(self, *args, **kwargs): origin = self.get_origin(*args) if self.det2imext != None and self.d2imerr > self.minerr: - return WCS.p4_pix2foc(self, self.det2im(*args), origin) + img = WCS.p4_pix2foc(self, self.det2im(*args), origin) + if len(args)==3: + return [img[:, i] for i in range(img.shape[1])] + else: + return img else: return WCS.p4_pix2foc(self, *args) @@ -278,7 +290,11 @@ class HSTWCS(WCS): def sip_pix2foc(self, *args, **kwargs): origin = self.get_origin(*args) if self.det2imext != None and self.d2imerr > self.minerr: - return WCS.sip_pix2foc(self, self.det2im(*args), origin) + img = WCS.sip_pix2foc(self, self.det2im(*args), origin) + if len(args)==3: + return [img[:, i] for i in range(img.shape[1])] + else: + return img else: return WCS.sip_pix2foc(self, *args) -- cgit