diff options
author | lim <lim@stsci.edu> | 2012-07-12 17:08:41 -0400 |
---|---|---|
committer | lim <lim@stsci.edu> | 2012-07-12 17:08:41 -0400 |
commit | 7c2f170c52c554326babd45badb4aa4852eef615 (patch) | |
tree | 34bbcc30facc19dcf736eb3b964a59c344fff0f2 /lib/skyline.py | |
parent | e0886647eb9f8073c214878498c4d422b0aa31ab (diff) | |
download | stsci.sphere-7c2f170c52c554326babd45badb4aa4852eef615.tar.gz |
fixed skyline bug, change polygon.to_radec output format
git-svn-id: http://svn.stsci.edu/svn/ssb/stsci_python/stsci_python/branches/sphere@17822 fe389314-cf27-0410-b35b-8c050e845b92
Former-commit-id: 80e7a2f26582c697cb6faf47418b58e879f94de0
Diffstat (limited to 'lib/skyline.py')
-rw-r--r-- | lib/skyline.py | 32 |
1 files changed, 21 insertions, 11 deletions
diff --git a/lib/skyline.py b/lib/skyline.py index 04c4afa..eb58a2f 100644 --- a/lib/skyline.py +++ b/lib/skyline.py @@ -89,7 +89,7 @@ SKYLINE_DEBUG = True __all__ = ['SkyLineMember', 'SkyLine'] __version__ = '0.4a' -__vdate__ = '11-Jul-2012' +__vdate__ = '12-Jul-2012' class SkyLineMember(object): """ @@ -248,6 +248,16 @@ class SkyLine(object): if v not in self._members: self._members.append(v) + def _indv_mem_wcslist(self): + """List of original HSTWCS from each EXT in each member.""" + wcs_list = [] + + for m in self.members: + for i in m.ext: + wcs_list.append(wcsutil.HSTWCS(m.fname, ext=i)) + + return wcs_list + def to_wcs(self): """ Combine `HSTWCS` objects from all `members` and return @@ -256,14 +266,13 @@ class SkyLine(object): .. warning:: This cannot return WCS of intersection. """ - wcs_list = [] - - for m in self.members: - for i in m.ext: - wcs_list.append(wcsutil.HSTWCS(m.fname, ext=i)) + wcs_list = self._indv_mem_wcslist() - if len(wcs_list) > 0: + n = len(wcs_list) + if n > 1: wcs = output_wcs(wcs_list) + elif n == 1: + wcs = wcs_list[0] else: wcs = None @@ -288,10 +297,11 @@ class SkyLine(object): **kwargs : Any plot arguments to pass to basemap """ - for m in self.members: - for i in m.ext: - poly = SphericalPolygon.from_wcs(wcsutil.HSTWCS(m.fname, ext=i)) - poly.draw(map, **kwargs) + wcs_list = self._indv_mem_wcslist() + + for wcs in wcs_list: + poly = SphericalPolygon.from_wcs(wcs) + poly.draw(map, **kwargs) def _find_members(self, given_members): """ |