Changeset 1446 for branches/alma/python/asapplotter.py

Timestamp:

11/12/08 17:04:01 (16 years ago)

Author:

TakTsutsumi

Message:

Merged recent updates (since 2007) from nrao-asap

File:

• branches/alma/python/asapplotter.py (modified) (9 diffs)

branches/alma/python/asapplotter.py

@@ -605 +605 @@
         if n > 1:
             ganged = rcParams['plotter.ganged']
+            ###Start Mod: 2008.09.22 kana ###
+            if self._panelling == 'i':
+                ganged = False
+            ###End Mod#######################
             if self._rows and self._cols:
                 n = min(n,self._rows*self._cols)
@@ -688 +692 @@
                 if stackcount == nstack:
                     allxlim.sort()
-                    self._plotter.axes.set_xlim([allxlim[0],allxlim[-1]])
+                    ###Start Mod: 2008.09.22 kana ###
+                    #self._plotter.axes.set_xlim([allxlim[0],allxlim[-1]])
+                    self._plotter.subplots[panelcount-1]['axes'].set_xlim([allxlim[0],allxlim[-1]])
+                    ###End Mod#######################
                     # clear
                     allxlim =[]
@@ -741 +748 @@
         return userlabel or d[mode]
 
-    def plotazel(self, scan=None):
+    def plotazel(self, scan=None, outfile=None):
         """
         plot azimuth and elevation  versus time of a scantable
@@ -750 +757 @@
         from matplotlib.numerix import array, pi
         self._data = scan
+        self._outfile = outfile
         dates = self._data.get_time()
         t = PL.date2num(dates)
         tz = timezone('UTC')
         PL.cla()
-        PL.ioff()
+        #PL.ioff()
         PL.clf()
         tdel = max(t) - min(t)
@@ -772 +780 @@
             minloc = MinuteLocator(20)
         PL.title(dstr)
-        PL.plot_date(t,el,'b,', tz=tz)
-        #ax.grid(True)
+
+        if tdel == 0.0:
+            th = (t - PL.floor(t))*24.0
+            PL.plot(th,el,'o',markersize=2, markerfacecolor='b', markeredgecolor='b')
+        else:
+            PL.plot_date(t,el,'o', markersize=2, markerfacecolor='b', markeredgecolor='b',tz=tz)
+            #ax.grid(True)
+            ax.xaxis.set_major_formatter(timefmt)
+            ax.xaxis.set_major_locator(majloc)
+            ax.xaxis.set_minor_locator(minloc)
         ax.yaxis.grid(True)
         yloc = MultipleLocator(30)
         ax.set_ylim(0,90)
-        ax.xaxis.set_major_formatter(timefmt)
-        ax.xaxis.set_major_locator(majloc)
-        ax.xaxis.set_minor_locator(minloc)
         ax.yaxis.set_major_locator(yloc)
         if tdel > 1.0:
@@ -785 +798 @@
         #    PL.setp(labels, fontsize=10, rotation=45)
             PL.setp(labels, fontsize=10)
+
         # Az plot
         az = array(self._data.get_azimuth())*180./pi
@@ -792 +806 @@
 
         ax = PL.subplot(2,1,2)
-        PL.xlabel('Time (UT)')
+        #PL.xlabel('Time (UT [hour])')
         PL.ylabel('Az [deg.]')
-        PL.plot_date(t,az,'b,', tz=tz)
+        if tdel == 0.0:
+            PL.plot(th,az,'o',markersize=2, markeredgecolor='b',markerfacecolor='b')
+        else:
+            PL.plot_date(t,az,'o', markersize=2,markeredgecolor='b',markerfacecolor='b',tz=tz)
+            ax.xaxis.set_major_formatter(timefmt)
+            ax.xaxis.set_major_locator(majloc)
+            ax.xaxis.set_minor_locator(minloc)
+        #ax.grid(True)
         ax.set_ylim(0,360)
-        #ax.grid(True)
         ax.yaxis.grid(True)
         #hfmt = DateFormatter('%H')
         #hloc = HourLocator()
         yloc = MultipleLocator(60)
-        ax.xaxis.set_major_formatter(timefmt)
-        ax.xaxis.set_major_locator(majloc)
-        ax.xaxis.set_minor_locator(minloc)
         ax.yaxis.set_major_locator(yloc)
         if tdel > 1.0:
             labels = ax.get_xticklabels()
             PL.setp(labels, fontsize=10)
-        PL.ion()
+            PL.xlabel('Time (UT [day])')
+        else:
+            PL.xlabel('Time (UT [hour])')
+
+        #PL.ion()
         PL.draw()
-
-    def plotpointing(self, scan=None):
+        if (self._outfile is not None):
+           PL.savefig(self._outfile)
+
+    def plotpointing(self, scan=None, outfile=None):
         """
         plot telescope pointings
@@ -820 +843 @@
         from matplotlib.numerix import array, pi, zeros
         self._data = scan
+        self._outfile = outfile
         dir = array(self._data.get_directionval()).transpose()
         ra = dir[0]*180./pi
         dec = dir[1]*180./pi
         PL.cla()
-        PL.ioff()
+        #PL.ioff()
         PL.clf()
         ax = PL.axes([0.1,0.1,0.8,0.8])
@@ -835 +859 @@
         [xmin,xmax,ymin,ymax] = PL.axis()
         PL.axis([xmax,xmin,ymin,ymax])
-        PL.ion()
+        #PL.ion()
         PL.draw()
-
+        if (self._outfile is not None):
+           PL.savefig(self._outfile)
+
+    # plot total power data
+    # plotting in time is not yet implemented..
+    def plottp(self, scan=None, outfile=None):
+        if self._plotter.is_dead:
+            self._plotter = self._newplotter()
+        self._plotter.hold()
+        self._plotter.clear()
+        from asap import scantable
+        if not self._data and not scan:
+            msg = "Input is not a scantable"
+            if rcParams['verbose']:
+                print msg
+                return
+            raise TypeError(msg)
+        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()
+        # ranges become invalid when abcissa changes?
+        #if self._abcunit and self._abcunit != self._data.get_unit():
+        #    self._minmaxx = None
+        #    self._minmaxy = None
+        #    self._abcunit = self._data.get_unit()
+        #    self._datamask = None
+        self._plottp(self._data)
+        if self._minmaxy is not None:
+            self._plotter.set_limits(ylim=self._minmaxy)
+        self._plotter.release()
+        self._plotter.tidy()
+        self._plotter.show(hardrefresh=False)
+        print_log()
+        return
+
+    def _plottp(self,scan):
+        """
+        private method for plotting total power data
+        """
+        from matplotlib.numerix import ma, array, arange, logical_not
+        r=0
+        nr = scan.nrow()
+        a0,b0 = -1,-1
+        allxlim = []
+        allylim = []
+        y=[]
+        self._plotter.set_panels()
+        self._plotter.palette(0)
+        #title
+        #xlab = self._abcissa and self._abcissa[panelcount] \
+        #       or scan._getabcissalabel()
+        #ylab = self._ordinate and self._ordinate[panelcount] \
+        #       or scan._get_ordinate_label()
+        xlab = self._abcissa or 'row number' #or Time
+        ylab = self._ordinate or scan._get_ordinate_label()
+        self._plotter.set_axes('xlabel',xlab)
+        self._plotter.set_axes('ylabel',ylab)
+        lbl = self._get_label(scan, r, 's', self._title)
+        if isinstance(lbl, list) or isinstance(lbl, tuple):
+        #    if 0 <= panelcount < len(lbl):
+        #        lbl = lbl[panelcount]
+        #    else:
+                # get default label
+             lbl = self._get_label(scan, r, self._panelling, None)
+        self._plotter.set_axes('title',lbl)
+        y=array(scan._get_column(scan._getspectrum,-1))
+        m = array(scan._get_column(scan._getmask,-1))
+        y = ma.masked_array(y,mask=logical_not(array(m,copy=False)))
+        x = arange(len(y))
+        # try to handle spectral data somewhat...
+        l,m = y.shape
+        if m > 1:
+            y=y.mean(axis=1)
+        plotit = self._plotter.plot
+        llbl = self._get_label(scan, r, self._stacking, None)
+        self._plotter.set_line(label=llbl)
+        if len(x) > 0:
+            plotit(x,y)
+
+
+    # forwards to matplotlib.Figure.text
+    def figtext(self, *args, **kwargs):
+        """
+        Add text to figure at location x,y (relative 0-1 coords).
+        This method forwards *args and **kwargs to a Matplotlib method,
+        matplotlib.Figure.text.
+        See the method help for detailed information.
+        """
+        self._plotter.text(*args, **kwargs)
+    # end matplotlib.Figure.text forwarding function
+