Changeset 2693 for trunk/python/asapplotter.py

Timestamp:

12/12/12 20:01:06 (11 years ago)

Author:

Kana Sugimoto

Message:

New Development: Yes

JIRA Issue: Yes (Trac-287)

Ready for Test: Yes

Interface Changes: Yes

What Interface Changed: a new method, asapplotter.plotpointings2

Test Programs:

Put in Release Notes: No

Module(s): asapplotter, sdplot

Description:

a new method, asapplotter.plotpointings2, is created.
This method will be supposed to replace asapplotter.plotpointing
in near future if there's no issue found.
plotpointings2 plots pointing directions of a scantable.
It is possible to plot pointings of different source type, scanno,
ifno, polno, or beamno in different colors by setting the 'colorby'
paramter. When showline=True, the scan pattern is plotted in dotted
line.
This function alway plots pointings on ASAP plotter and works properly
with custom toolbar.

File:

• trunk/python/asapplotter.py (modified) (5 diffs)

trunk/python/asapplotter.py

@@ -1363 +1363 @@
            PL.savefig(outfile)
 
+
+    def plotpointing2(self, scan=None, colorby='', showline=False, projection=''):
+        """
+        plot telescope pointings
+        Parameters:
+            infile  : input filename or scantable instance
+            colorby : change color by either
+                      'type'(source type)|'scan'|'if'|'pol'|'beam'
+            showline : show dotted line
+            projection : projection type either
+                         ''(no projection [deg])|'coord'(not implemented)
+        """
+        from numpy import array, pi
+        from asap import scantable
+        # check for scantable
+        if isinstance(scan, scantable):
+            if self._data is not None:
+                if scan != self._data:
+                    self._data = scan
+                    # reset
+                    self._reset()
+            else:
+                self._data = scan
+                self._reset()
+        if not self._data:
+            msg = "Input is not a scantable"
+            raise TypeError(msg)
+        # check for color mode
+        validtypes=['type','scan','if','pol', 'beam']
+        stype = None
+        if (colorby in validtypes):
+            stype = colorby[0]
+        elif len(colorby) > 0:
+            msg = "Invalid choice of 'colorby' (choices: %s)" % str(validtypes)
+            raise ValueError(msg)
+        self._assert_plotter(action="reload")
+        self._plotter.hold()
+        if self.casabar_exists(): self._plotter.figmgr.casabar.disable_button()
+        # for now, only one plot
+        self._plotter.set_panels(rows=1,cols=1)
+        # first panel
+        self._plotter.subplot(0)
+        # first color and linestyles
+        self._plotter.palette(0)
+        self.gca().set_aspect('equal')
+        basesel = scan.get_selection()
+        marker = "+"
+        if showline:
+            basesel.set_order(["TIME"])
+            scan.set_selection(basesel)
+            if not (stype in ["t", "s"]):
+                marker = "+:"
+        if not stype:
+            selIds = [""] # cheating
+            sellab = "all points"
+        elif stype == 't':
+            selIds = range(15)
+            sellab = "src type "
+        else:
+            selIds = getattr(self._data,'get'+colorby+'nos')()
+            sellab = colorby.upper()
+        selFunc = "set_"+colorby+"s"
+        for idx in selIds:
+            sel = selector() + basesel
+            if stype:
+                bid = getattr(basesel,'get_'+colorby+"s")()
+                if (len(bid) > 0) and (not idx in bid):
+                    # base selection doesn't contain idx
+                    # Note summation of selector is logical sum if 
+                    continue
+                getattr(sel, selFunc)([idx])
+            if not sel.is_empty():
+                try:
+                    self._data.set_selection(sel)
+                except RuntimeError, instance:
+                    if stype == 't' and str(instance).startswith("Selection contains no data."):
+                        continue
+                    else:
+                        self._data.set_selection(basesel)
+                        raise RuntimeError, instance
+            if self._data.nrow() == 0:
+                self._data.set_selection(basesel)
+                continue
+            print "Plotting direction of %s = %s" % (colorby, str(idx))
+            dir = array(self._data.get_directionval()).transpose()
+            ra = dir[0]*180./pi
+            dec = dir[1]*180./pi
+            self._plotter.set_line(label=(sellab+str(idx)))
+            self._plotter.plot(ra,dec,marker)
+
+        # restore original selection
+        self._data.set_selection(basesel)
+        # need to plot scan pattern explicitly
+        if showline and (stype in ["t", "s"]):
+            dir = array(self._data.get_directionval()).transpose()
+            ra = dir[0]*180./pi
+            dec = dir[1]*180./pi
+            self._plotter.set_line(label="scan pattern")
+            self._plotter.plot(ra,dec,":")
+            
+        xlab = 'RA [deg.]'
+        ylab = 'Declination [deg.]'
+        self._plotter.set_axes('xlabel', xlab)
+        self._plotter.set_axes('ylabel', ylab)
+        self._plotter.set_axes('title', 'Telescope pointings')
+        if stype: self._plotter.legend(self._legendloc)
+        else: self._plotter.legend(None)
+        # reverse x-axis
+        xmin, xmax = self.gca().get_xlim()
+        self._plotter.set_limits(xlim=[xmax,xmin])
+
+        self._plotter.release()
+        self._plotter.show(hardrefresh=False)
+        return
+
     def plotpointing(self, scan=None, outfile=None):
         """
@@ -1407 +1522 @@
     @asaplog_post_dec
     def plottp(self, scan=None):
-        self._assert_plotter(action="reload")
-        self._plotter.hold()
-        self._plotter.clear()
         from asap import scantable
         if not self._data and not scan:
@@ -1430 +1542 @@
         #    self._datamask = None
 
+        self._assert_plotter(action="reload")
+        self._plotter.hold()
+        self._plotter.clear()
         # Adjust subplot margins
         if not self._margins or len(self._margins) !=6:
@@ -1741 +1856 @@
         self._assert_plotter(action="reload")
         self._plotter.hold()
-        self._plotter.clear()
+        #self._plotter.clear() #all artists are cleared at set_panels
         self._plotter.legend()
 
@@ -1748 +1863 @@
             self.set_margin(refresh=False)
         self._plotter.set_panels(rows=self._rows,cols=self._cols,
-                                 nplots=ntotpl,margin=self._margins,ganged=True)
+                                 nplots=ntotpl,margin=self._margins,ganged=True)        
         if self.casabar_exists():
             self._plotter.figmgr.casabar.set_pagecounter(1)