Changeset 1723 for branches/alma

Timestamp:

04/20/10 18:38:20 (15 years ago)

Author:

Kana Sugimoto

Message:

New Development: No

JIRA Issue: Yes (CAS-1801, CAS-2064)

Ready to Release: Yes

Interface Changes: Yes

What Interface Changed:
Three methods, print_headder(), set_panellayout() and _reset_panellayout()
are added to the class asapplotter.
A parameter layout (list) is added to set_layout() in asaplotbase class.

Test Programs: run sdplot with header & plotstyle=True + layout parameter set.

Put in Release Notes: Yes

Module(s): sdplot() task.

Description:

1. asapplotter.print_header(plot=True, fontsize=9, logger=False, selstr=, extrastr=): prints scantable header on the plot and/or logger (see help doc for parameter details). Note that the header information will not overlayed to the plot by plotazel() and plotpointing().
2. set_panellayout(layout=[],refresh=True): sets the layout of subplots. layout is a list of subplots layout in figure coordinate (0-1) in order of [left, bottom, right, top, horizontal space btw panels, vertical space btw panels]
3. _reset_panellayout(): resets the layout of subplots to matplotlib rc value.
4. a member variable 'asapplotter._panellayout' (list) is specified to store layout values.

Location:

branches/alma/python

Files:

• asaplotbase.py (modified) (4 diffs)
• asapplotter.py (modified) (7 diffs)

branches/alma/python/asaplotbase.py

@@ -570 +570 @@
 
 
-    def set_panels(self, rows=1, cols=0, n=-1, nplots=-1, ganged=True):
+    #def set_panels(self, rows=1, cols=0, n=-1, nplots=-1, ganged=True):
+    def set_panels(self, rows=1, cols=0, n=-1, nplots=-1, layout=None,ganged=True):
         """
         Set the panel layout.
@@ -593 +594 @@
             self.set_title()
 
+        if layout:
+            lef, bot, rig, top, wsp, hsp = layout
+            self.figure.subplots_adjust(
+                left=lef,bottom=bot,right=rig,top=top,wspace=wsp,hspace=hsp)
+            del lef,bot,rig,top,wsp,hsp
+
         if rows < 1: rows = 1
 
@@ -605 +612 @@
         if 0 <= n < rows*cols:
             i = len(self.subplots)
+
             self.subplots.append({})
 
@@ -666 +674 @@
                 self.cols = cols
             self.subplot(0)
+        del rows,cols,n,nplots,layout,ganged,i
 
     def tidy(self):

branches/alma/python/asapplotter.py

@@ -39 +39 @@
         self._selection = selector()
         self._hist = rcParams['plotter.histogram']
+        self._panellayout = self.set_panellayout(refresh=False)
 
     def _translate(self, instr):
@@ -476 +477 @@
             rcp('font', size=size)
         if refresh and self._data: self.plot(self._data)
+
+    def set_panellayout(self,layout=[],refresh=True):
+        """
+        Set the layout of subplots.
+        Parameters:
+            layout:   a list of subplots layout in figure coordinate (0-1),
+                      i.e., fraction of the figure width or height. 
+                      The order of elements should be:
+                      [left, bottom, right, top, horizontal space btw panels,
+                      vertical space btw panels]. 
+            refresh:  True (default) or False. If True, the plot is
+                      replotted based on the new parameter setting(s). 
+                      Otherwise,the parameter(s) are set without replotting.
+        Note
+        * When layout is not specified, the values are reset to the defaults
+          of matplotlib.
+        * If any element is set to be None, the current value is adopted. 
+        """
+        if layout == []: self._panellayout=self._reset_panellayout()
+        else: 
+            self._panellayout=[None]*6
+            self._panellayout[0:len(layout)]=layout
+        #print "panel layout set to ",self._panellayout
+        if refresh and self._data: self.plot(self._data)
+
+    def _reset_panellayout(self):
+        ks=map(lambda x: 'figure.subplot.'+x,
+               ['left','bottom','right','top','hspace','wspace'])
+        return map(matplotlib.rcParams.get,ks)
 
     def plot_lines(self, linecat=None, doppler=0.0, deltachan=10, rotate=90.0,
@@ -693 +723 @@
                 n = min(n,self._rows*self._cols)
                 self._plotter.set_panels(rows=self._rows,cols=self._cols,
-                                         nplots=n,ganged=ganged)
+#                                         nplots=n,ganged=ganged)
+                                         nplots=n,layout=self._panellayout,ganged=ganged)
             else:
-                self._plotter.set_panels(rows=n,cols=0,nplots=n,ganged=ganged)
+#                self._plotter.set_panels(rows=n,cols=0,nplots=n,ganged=ganged)
+                self._plotter.set_panels(rows=n,cols=0,nplots=n,layout=self._panellayout,ganged=ganged)
         else:
-            self._plotter.set_panels()
+#            self._plotter.set_panels()
+            self._plotter.set_panels(layout=self._panellayout)
         r=0
         nr = scan.nrow()
@@ -856 +889 @@
         #PL.ioff()
         PL.clf()
+        # Adjust subplot layouts
+        if len(self._panellayout) !=6: self.set_panellayout(refresh=False)
+        lef, bot, rig, top, wsp, hsp = self._panellayout
+        PL.gcf().subplots_adjust(left=lef,bottom=bot,right=rig,top=top,
+                                 wspace=wsp,hspace=hsp)
+        
         tdel = max(t) - min(t)
         ax = PL.subplot(2,1,1)
@@ -942 +981 @@
         #PL.ioff()
         PL.clf()
+        # Adjust subplot layouts
+        if len(self._panellayout) !=6: self.set_panellayout(refresh=False)
+        lef, bot, rig, top, wsp, hsp = self._panellayout
+        PL.gcf().subplots_adjust(left=lef,bottom=bot,right=rig,top=top,
+                                 wspace=wsp,hspace=hsp)
         ax = PL.axes([0.1,0.1,0.8,0.8])
         ax = PL.axes([0.1,0.1,0.8,0.8])
@@ -987 +1031 @@
         #    self._abcunit = self._data.get_unit()
         #    self._datamask = None
+
+        # Adjust subplot layouts
+        if len(self._panellayout) !=6: self.set_panellayout(refresh=False)
+        lef, bot, rig, top, wsp, hsp = self._panellayout
+        self._plotter.figure.subplots_adjust(
+            left=lef,bottom=bot,right=rig,top=top,wspace=wsp,hspace=hsp)
         self._plottp(self._data)
         if self._minmaxy is not None:
@@ -1052 +1102 @@
     # end matplotlib.Figure.text forwarding function
 
+
+    # printing header information
+    def print_header(self, plot=True, fontsize=9, logger=False, selstr='', extrastr=''):
+        """
+        print data (scantable) header on the plot and/or logger.
+        Parameters:
+            plot:      whether or not print header info on the plot. 
+            fontsize:  header font size (valid only plot=True)
+            autoscale: whether or not autoscale the plot (valid only plot=True)
+            logger:    whether or not print header info on the logger.
+            selstr:    additional selection string (not verified)
+            extrastr:  additional string to print (not verified)
+        """
+        if not plot and not logger: return
+        if not self._data: raise RuntimeError("No scantable has been set yet.")
+        # Now header will be printed on plot and/or logger. 
+        # Get header information and format it. 
+        ssum=self._data.__str__()
+        # Print Observation header to the upper-left corner of plot
+        if plot:
+            srest=ssum[ssum.find('Rest Freqs:'):ssum.find('Abcissa:')]
+            shead=ssum[ssum.find('Beams:'):ssum.find('Flux Unit:')]
+            headstr=shead.split('\n\n')
+            if extrastr != '': headstr[1]=extrastr+'\n'+headstr[1]
+            #headstr[1]='Data File:     '+(filestr or 'unknown')+'\n'+headstr[1]
+            headstr[0]=headstr[0]+'\n'+srest
+            headstr.reverse()
+            ssel='***Selections***\n'+(selstr+self._data.get_selection().__str__() or 'none')
+            headstr.append(ssel)
+            nstcol=len(headstr)
+            
+            self._plotter.hold()
+            for i in range(nstcol):
+                self._plotter.figure.text(0.03+float(i)/nstcol,0.98,
+                             headstr[i],
+                             horizontalalignment='left',
+                             verticalalignment='top',
+                             fontsize=fontsize)
+            import time
+            self._plotter.figure.text(0.99,0.0,
+                            time.strftime("%a %d %b %Y  %H:%M:%S %Z"),
+                            horizontalalignment='right',
+                            verticalalignment='bottom',fontsize=8)
+            self._plotter.release()
+            del srest, shead, headstr, ssel
+        if logger:
+            asaplog.push("----------------\n  Plot Summary\n----------------")
+            asaplog.push(extrastr)
+            asaplog.push(ssum[ssum.find('Beams:'):])
+            print_log()
+        del ssum