Changeset 8dfdd20 in sasview for guiframe/local_perspectives/plotting

Timestamp:

Jun 2, 2009 5:05:30 PM (16 years ago)

Author:

Gervaise Alina <gervyh@…>

Branches:

master, ESS_GUI, ESS_GUI_Docs, ESS_GUI_batch_fitting, ESS_GUI_bumps_abstraction, ESS_GUI_iss1116, ESS_GUI_iss879, ESS_GUI_iss959, ESS_GUI_opencl, ESS_GUI_ordering, ESS_GUI_sync_sascalc, costrafo411, magnetic_scatt, release-4.1.1, release-4.1.2, release-4.2.2, release_4.0.1, ticket-1009, ticket-1094-headless, ticket-1242-2d-resolution, ticket-1243, ticket-1249, ticket885, unittest-saveload

Children:

d483799

Parents:

e50f15b

Message:

change detector dialog to allow cmap selection

Location:

guiframe/local_perspectives/plotting

Files:

• Plotter2D.py (modified) (6 diffs)
• detector_dialog.py (modified) (11 diffs)

guiframe/local_perspectives/plotting/Plotter2D.py

@@ -82 +82 @@
         ## panel that contains result from slicer motion (ex: Boxsum info)
         self.panel_slicer=None
+        
         ## Graph        
         self.graph = Graph()
@@ -87 +88 @@
         self.graph.yaxis("\\rm{Intensity} ","cm^{-1}")
         self.graph.render(self)
-        
+        ## store default value of zmin and zmax 
+        self.default_zmin_ctl = self.zmin_2D
+        self.default_zmax_ctl = self.zmax_2D
         
     def _onEVT_1DREPLOT(self, event):
@@ -165 +168 @@
         self.graph.render(self)
         self.subplot.figure.canvas.draw_idle()
+        ## store default value of zmin and zmax 
+        self.default_zmin_ctl = self.zmin_2D
+        self.default_zmax_ctl = self.zmax_2D
 
 
@@ -263 +269 @@
         
         import detector_dialog
-        dialog = detector_dialog.DetectorDialog(self, -1,base=self.parent)
+       
+        dialog = detector_dialog.DetectorDialog(self, -1,base=self.parent,
+                       reset_zmin_ctl =self.default_zmin_ctl,
+                       reset_zmax_ctl = self.default_zmax_ctl,cmap=self.cmap)
         ## info of current detector and data2D
         xnpts = len(self.data2D.x_bins)
@@ -281 +290 @@
             self.zmin_2D = evt.zmin
             self.zmax_2D = evt.zmax
+            self.cmap= evt.cmap
        
         dialog.Destroy()
@@ -291 +301 @@
                    zmin= self.zmin_2D,
                    zmax= self.zmax_2D,
+                   cmap= self.cmap,
                    color=0,symbol=0,label=self.data2D.name)#'data2D')
         self.subplot.figure.canvas.draw_idle()

guiframe/local_perspectives/plotting/detector_dialog.py

@@ -6 +6 @@
 
 import wx
+import sys
 from sans.guiframe.utils import format_number
-from sans.guicomm.events import StatusEvent ,NewPlotEvent,SlicerEvent
-
-
+from sans.guicomm.events import StatusEvent ,NewPlotEvent
+
+import matplotlib 
+from matplotlib.backends.backend_wxagg import FigureCanvasWxAgg as Canvas
+from matplotlib import pyplot, mpl, pylab
+
+DEFAULT_CMAP= pylab.cm.jet
 class DetectorDialog(wx.Dialog):
     """
@@ -15 +20 @@
     """
     
-    def __init__(self,parent,id=1,base=None, *args, **kwds):
+    def __init__(self,parent,id=1,base=None,dpi = None,cmap=DEFAULT_CMAP,
+                 reset_zmin_ctl = None,reset_zmax_ctl = None, *args, **kwds):
 
         kwds["style"] = wx.DEFAULT_DIALOG_STYLE
@@ -21 +27 @@
     
         self.parent=base
+        self.dpi = dpi
+        self.cmap= cmap
+
+        self.reset_zmin_ctl= reset_zmin_ctl
+        self.reset_zmax_ctl= reset_zmax_ctl
+            
         self.label_xnpts = wx.StaticText(self, -1, "Detector width in pixels")
         self.label_ynpts = wx.StaticText(self, -1, "Detector Height in pixels")
@@ -34 +46 @@
         
         self.zmin_ctl = wx.TextCtrl(self, -1, size=(60,20))
+        self.zmin_ctl.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
         self.zmax_ctl = wx.TextCtrl(self, -1, size=(60,20))
+        self.zmax_ctl.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
     
         self.static_line_3 = wx.StaticLine(self, -1)
         
-        
+        self.button_Cancel = wx.Button(self, wx.ID_CANCEL, "Cancel")
+        self.button_reset = wx.Button(self, wx.NewId(),"Reset")
+        self.Bind(wx.EVT_BUTTON, self.resetValues, self.button_reset)
         self.button_OK = wx.Button(self, wx.ID_OK, "OK")
-        self.button_Cancel = wx.Button(self, wx.ID_CANCEL, "Cancel")
         self.Bind(wx.EVT_BUTTON, self.checkValues, self.button_OK)
 
@@ -55 +70 @@
         zmin = 0
         zmax = 0
+        cmap= None
         sym4 = False
+    
+    
+  
+    def onSetFocus(self, event):
+        """
+            Highlight the txtcrtl
+        """
+        # Get a handle to the TextCtrl
+        widget = event.GetEventObject()
+        # Select the whole control, after this event resolves
+        wx.CallAfter(widget.SetSelection, -1,-1)
+        return
+        
+    def resetValues(self, event):
+        """
+            reset detector info
+        """
+        try:
+            self.zmin_ctl.SetValue(str(float(self.reset_zmin_ctl)))
+            self.zmax_ctl.SetValue(str(float(self.reset_zmax_ctl)))
+            self.cmap = DEFAULT_CMAP
+            self.cmap_selector.SetValue(str(self.cmap.name))
+            self._on_select_cmap(event=None)
+        except:
+            msg ="error occurs while resetting Detector: %s"%sys.exc_value
+            wx.PostEvent(self.parent,StatusEvent(status= msg ))
+        
         
     def checkValues(self, event):
@@ -108 +151 @@
             @param sym:
         """
-        self.xnpts_ctl.SetLabel(str(format_number(xnpts)))
-        self.ynpts_ctl.SetLabel(str(format_number(ynpts)))
-        self.qmax_ctl.SetLabel(str(format_number(qmax)))
-        self.beam_ctl.SetLabel(str(format_number(beam)))
+        self.xnpts_ctl.SetLabel(str(float(xnpts)))
+        self.ynpts_ctl.SetLabel(str(float(ynpts)))
+        self.qmax_ctl.SetLabel(str(float(qmax)))
+        self.beam_ctl.SetLabel(str(float(beam)))
+        
+    
        
         if zmin !=None:
-            self.zmin_ctl.SetValue(str(format_number(zmin)))
+            self.zmin_ctl.SetValue(str(float(zmin)))
         if zmax !=None:
-            self.zmax_ctl.SetValue(str(format_number(zmax)))
+            self.zmax_ctl.SetValue(str(float(zmax)))
 
     def getContent(self):
@@ -143 +188 @@
         event.zmin = v_min
         event.zmax = v_max
+        event.cmap= self.cmap
         
         return event
@@ -161 +207 @@
         sizer_button = wx.BoxSizer(wx.HORIZONTAL)
         sizer_params = wx.GridBagSizer(5,5)
+        sizer_colormap = wx.BoxSizer(wx.VERTICAL)
+        sizer_selection= wx.BoxSizer(wx.HORIZONTAL)
+       
 
         iy = 0
@@ -181 +230 @@
         sizer_params.Add(self.zmax_ctl,   (iy,1), (1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
         iy += 1
-       
+        
+        
+       
+        self.fig = mpl.figure.Figure(dpi=self.dpi, figsize=(4,1))
+       
+        self.ax1 = self.fig.add_axes([0.05, 0.65, 0.9, 0.15])
+        
+        
+        self.norm = mpl.colors.Normalize(vmin=0, vmax=100)
+        self.cb1 = mpl.colorbar.ColorbarBase(self.ax1, cmap=self.cmap,
+                                           norm= self.norm,
+                                           orientation='horizontal')
+        self.cb1.set_label('Detector Colors')
+        self.canvas = Canvas(self, -1, self.fig)
+        sizer_colormap.Add(self.canvas,0, wx.LEFT | wx.EXPAND,5)
+      
+        self.cmap_selector = wx.ComboBox(self, -1)
+        self.cmap_selector.SetValue(str(self.cmap.name))
+        maps = sorted(m for m in pylab.cm.datad if not m.endswith("_r"))
+       
+        for i,m in enumerate(maps):
+            
+            self.cmap_selector.Append(str(m), pylab.get_cmap(m))
+        
+        wx.EVT_COMBOBOX(self.cmap_selector,-1, self._on_select_cmap)
+        sizer_selection.Add(wx.StaticText(self,-1,"Select Cmap: "),0, wx.LEFT|wx.ADJUST_MINSIZE,5) 
+        sizer_selection.Add(self.cmap_selector, 0, wx.EXPAND|wx.ALL, 10)
+        
         sizer_main.Add(sizer_params, 0, wx.EXPAND|wx.ALL, 10)
+        
+        sizer_main.Add(sizer_selection, 0, wx.EXPAND|wx.ALL, 10)
+        sizer_main.Add(sizer_colormap, 1, wx.EXPAND|wx.ALL, 10)
         sizer_main.Add(self.static_line_3, 0, wx.EXPAND, 0)
         
         
-        sizer_button.Add((20, 20), 1, wx.EXPAND|wx.ADJUST_MINSIZE, 0)
+        sizer_button.Add(self.button_reset,0, wx.LEFT|wx.ADJUST_MINSIZE, 100)
         sizer_button.Add(self.button_OK, 0, wx.LEFT|wx.ADJUST_MINSIZE, 10)
         sizer_button.Add(self.button_Cancel, 0, wx.LEFT|wx.RIGHT|wx.ADJUST_MINSIZE, 10)
@@ -197 +276 @@
         self.Centre()
         # end wxGlade
+        
+    def _on_select_cmap(self, event):
+        """
+             display a new cmap 
+        """
+        cmap_name= self.cmap_selector.GetCurrentSelection()
+        current_cmap= self.cmap_selector.GetClientData( cmap_name )
+        self.cmap= current_cmap
+        self.cb1 = mpl.colorbar.ColorbarBase(self.ax1, cmap=self.cmap,
+                                           norm= self.norm,
+                                           orientation='horizontal')
+        self.canvas.draw()
 
 
@@ -207 +298 @@
         dialog = DetectorDialog(None, -1, "")
         self.SetTopWindow(dialog)
-        dialog.setContent(128, 0.05)
+        dialog.setContent(xnpts=128,ynpts=128, qmax=20,
+                           beam=20,zmin=2,zmax=60, sym=False)
         print dialog.ShowModal()
         evt = dialog.getContent()
-        print evt.npts, evt.qmax
+        if hasattr(evt,"npts"):
+            print "number of point: ",evt.npts
+        if hasattr(evt,"qmax"): 
+            print "qmax: ",evt.qmax
         dialog.Destroy()
         return 1