Changeset 2150 for trunk/python

Timestamp:

04/22/11 18:21:29 (14 years ago)

Author:

Kana Sugimoto

Message:

New Development: No

JIRA Issue: No (a fix)

Ready for Test: Yes

Interface Changes: added a new method new_asaplot() to module asapplotter which returns asaplot instance based on your backend selection.

What Interface Changed:

Test Programs: sdaverage, sdsmooth with verify=True and plotlevel > 0

Put in Release Notes: No

Module(s):

Description: proper handling of plotting in non-TkAgg backend.

Location:

trunk/python

Files:

• asapfitter.py (modified) (1 diff)
• asapmath.py (modified) (7 diffs)
• asapplotter.py (modified) (2 diffs)
• scantable.py (modified) (2 diffs)

trunk/python/asapfitter.py

@@ -538 +538 @@
             return
         if not self._p or self._p.is_dead:
-            if rcParams['plotter.gui']:
-                from asap.asaplotgui import asaplotgui as asaplot
-            else:
-                from asap.asaplot import asaplot
-            self._p = asaplot()
+            #if rcParams['plotter.gui']:
+            #    from asap.asaplotgui import asaplotgui as asaplot
+            #else:
+            #    from asap.asaplot import asaplot
+            #self._p = asaplot()
+            from asap.asapplotter import new_asaplot
+            self._p = new_asaplot(rcParams['plotter.gui'])
         self._p.hold()
         self._p.clear()

trunk/python/asapmath.py

@@ -3 +3 @@
 from asap.logging import asaplog, asaplog_post_dec
 from asap.selector import selector
-from asap import asaplotgui
+#from asap import asaplotgui
+from asap.asapplotter import new_asaplot
 
 @asaplog_post_dec
@@ -311 +312 @@
         asaplog.push('Plot only first spectrum for each [if,pol] pairs to verify calibration.')
         asaplog.post('WARN')
-        p=asaplotgui.asaplotgui()
+        #p=asaplotgui.asaplotgui()
+        p=new_asaplot()
         #nr=min(6,len(ifnos)*len(polnos))
         nr=len(ifnos)*len(polnos)
@@ -388 +390 @@
         x=raw_input('Accept calibration ([y]/n): ' )
         if x.upper() == 'N':
-            p.unmap()
+            p.quit()
             del p
             return scabtab
-        p.unmap()
+        p.quit()
         del p
     ###
@@ -530 +532 @@
         asaplog.push('Plot only first spectrum for each [if,pol] pairs to verify calibration.')
         asaplog.post('WARN')
-        p=asaplotgui.asaplotgui()
+        #p=asaplotgui.asaplotgui()
+        p=new_asaplot()
         #nr=min(6,len(ifnos)*len(polnos))
         nr=len(ifnos)*len(polnos)
@@ -607 +610 @@
         x=raw_input('Accept calibration ([y]/n): ' )
         if x.upper() == 'N':
-            p.unmap()
+            p.quit()
             del p
             return scabtab
-        p.unmap()
+        p.quit()
         del p
     ###
@@ -742 +745 @@
         asaplog.push('Plot only first spectrum for each [if,pol] pairs to verify calibration.')
         asaplog.post('WARN')
-        p=asaplotgui.asaplotgui()
+        #p=asaplotgui.asaplotgui()
+        p=new_asaplot()
         #nr=min(6,len(ifnos)*len(polnos))
         nr=len(ifnos)/2*len(polnos)
@@ -811 +815 @@
         x=raw_input('Accept calibration ([y]/n): ' )
         if x.upper() == 'N':
-            p.unmap()
+            p.quit()
             del p
             return scabtab
-        p.unmap()
+        p.quit()
         del p
     ###

trunk/python/asapplotter.py

@@ -8 +8 @@
 
 import re
+
+def new_asaplot(visible=None,**kwargs):
+    """
+    Returns a new asaplot instance based on the backend settings.
+    """
+    if visible == None:
+        visible = rcParams['plotter.gui']
+
+    backend=matplotlib.get_backend()
+    if not visible:
+        from asap.asaplot import asaplot
+    elif backend == 'TkAgg':
+        from asap.asaplotgui import asaplotgui as asaplot
+    elif backend == 'Qt4Agg':
+        from asap.asaplotgui_qt4 import asaplotgui as asaplot
+    elif backend == 'GTkAgg':
+        from asap.asaplotgui_gtk import asaplotgui as asaplot
+    else:
+        from asap.asaplot import asaplot
+    return asaplot(**kwargs)
 
 class asapplotter:
@@ -66 +86 @@
 
     def _newplotter(self, **kwargs):
-        backend=matplotlib.get_backend()
-        if not self._visible:
-            from asap.asaplot import asaplot
-        elif backend == 'TkAgg':
-            from asap.asaplotgui import asaplotgui as asaplot
-        elif backend == 'Qt4Agg':
-            from asap.asaplotgui_qt4 import asaplotgui as asaplot
-        elif backend == 'GTkAgg':
-            from asap.asaplotgui_gtk import asaplotgui as asaplot
-        else:
-            from asap.asaplot import asaplot
-        return asaplot(**kwargs)
+        return new_asaplot(self._visible,**kwargs)
 
     def _newcasabar(self):

trunk/python/scantable.py

@@ -2030 +2030 @@
 
         if plot:
-            if rcParams['plotter.gui']:
-                from asap.asaplotgui import asaplotgui as asaplot
-            else:
-                from asap.asaplot import asaplot
-            self._p=asaplot()
-            self._p.set_panels()
+            from asap.asapplotter import new_asaplot
+            theplot = new_asaplot(rcParams['plotter.gui'])
+            theplot.set_panels()
             ylab=s._get_ordinate_label()
-            #self._p.palette(0,["#777777","red"])
+            #theplot.palette(0,["#777777","red"])
             for r in xrange(s.nrow()):
                 xsm=s._getabcissa(r)
@@ -2043 +2040 @@
                 xorg=orgscan._getabcissa(r)
                 yorg=orgscan._getspectrum(r)
-                self._p.clear()
-                self._p.hold()
-                self._p.set_axes('ylabel',ylab)
-                self._p.set_axes('xlabel',s._getabcissalabel(r))
-                self._p.set_axes('title',s._getsourcename(r))
-                self._p.set_line(label='Original',color="#777777")
-                self._p.plot(xorg,yorg)
-                self._p.set_line(label='Smoothed',color="red")
-                self._p.plot(xsm,ysm)
+                theplot.clear()
+                theplot.hold()
+                theplot.set_axes('ylabel',ylab)
+                theplot.set_axes('xlabel',s._getabcissalabel(r))
+                theplot.set_axes('title',s._getsourcename(r))
+                theplot.set_line(label='Original',color="#777777")
+                theplot.plot(xorg,yorg)
+                theplot.set_line(label='Smoothed',color="red")
+                theplot.plot(xsm,ysm)
                 ### Ugly part for legend
                 for i in [0,1]:
-                    self._p.subplots[0]['lines'].append([self._p.subplots[0]['axes'].lines[i]])
-                self._p.release()
+                    theplot.subplots[0]['lines'].append([theplot.subplots[0]['axes'].lines[i]])
+                theplot.release()
                 ### Ugly part for legend
-                self._p.subplots[0]['lines']=[]
+                theplot.subplots[0]['lines']=[]
                 res = raw_input("Accept smoothing ([y]/n): ")
                 if res.upper() == 'N':
                     s._setspectrum(yorg, r)
-            self._p.unmap()
-            self._p = None
+            theplot.quit()
+            del theplot
             del orgscan