source: sasview/test/guiframe/local_perspectives/plotting/slicerpanel.py @ 01dc067

import wx
import wx.lib.newevent
#from copy import deepcopy
from sans.guiframe.utils import format_number
from sans.guiframe.events import SlicerParameterEvent
from sans.guiframe.events import EVT_SLICER_PARS
from sans.guiframe.events import EVT_SLICER

from sans.guiframe.panel_base import PanelBase

class SlicerPanel(wx.Panel, PanelBase):
    """
    Panel class to show the slicer parameters 
    """
    #TODO: show units
    #TODO: order parameters properly
     ## Internal name for the AUI manager
    window_name = "Slicer panel"
    ## Title to appear on top of the window
    window_caption = "Slicer Panel"
    CENTER_PANE = False
    
    def __init__(self, parent, id=-1, type=None, base=None,
                params=None, *args, **kwargs):
        wx.Panel.__init__(self, parent, id, *args, **kwargs)
        PanelBase.__init__(self)
        ##  Initialization of the class     
        self.base = base
        if params is None:
            params = {}
        self.params = params
        self.parent = parent
        self.type = type
        self.listeners = []
        self.parameters = []
        self.bck = wx.GridBagSizer(5, 5)
        self.SetSizer(self.bck)
        if type == None and params == None:  
            label = "Right-click on 2D plot for slicer options"  
            title = wx.StaticText(self, -1, label, style=wx.ALIGN_LEFT)
            self.bck.Add(title, (0, 0), (1, 2), 
                         flag=wx.LEFT|wx.ALIGN_CENTER_VERTICAL, border=15)
        else:
            self.set_slicer(type, params)
        ## Bindings
        self.parent.Bind(EVT_SLICER, self.onEVT_SLICER)
        self.parent.Bind(EVT_SLICER_PARS, self.onParamChange)

    def onEVT_SLICER(self, event):
        """
        Process EVT_SLICER events
        When the slicer changes, update the panel
        
        :param event: EVT_SLICER event
        
        """
        event.Skip()
        if event.obj_class == None:
            self.set_slicer(None, None) 
        else:
            self.set_slicer(event.type, event.params)
        
    def set_slicer(self, type, params):
        """
        Rebuild the panel
        """
        self.bck.Clear(True)  
        self.type = type  
        if type == None:
            label = "Right-click on 2D plot for slicer options"
            title = wx.StaticText(self, -1, label, style=wx.ALIGN_LEFT)
            self.bck.Add(title, (0, 0), (1, 2), 
                         flag=wx.LEFT|wx.ALIGN_CENTER_VERTICAL, border=15)
        else:
            title = wx.StaticText(self, -1, "Slicer Parameters", 
                                  style=wx.ALIGN_LEFT)
            self.bck.Add(title, (0, 0), (1, 2), 
                         flag=wx.LEFT|wx.ALIGN_CENTER_VERTICAL, border=15)
            n = 1
            self.parameters = []
            keys = params.keys()
            keys.sort()
            for item in keys:
                if not item.lower() in ["errors", "count"]:
                    n += 1
                    text = wx.StaticText(self, -1, item, style=wx.ALIGN_LEFT)
                    self.bck.Add(text, (n-1, 0), 
                            flag=wx.LEFT|wx.ALIGN_CENTER_VERTICAL, border=15)
                    ctl = wx.TextCtrl(self, -1, size=(80, 20), 
                                      style=wx.TE_PROCESS_ENTER)
                    hint_msg = "Modify the value of %s to change " % item
                    hint_msg += "the 2D slicer"
                    ctl.SetToolTipString(hint_msg)
                    ctl.SetValue(str(format_number(params[item])))
                    self.Bind(wx.EVT_TEXT_ENTER, self.onTextEnter)
                    ctl.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
                    ctl.Bind(wx.EVT_KILL_FOCUS, self.onTextEnter)
                    self.parameters.append([item, ctl])
                    self.bck.Add(ctl, (n-1, 1), flag=wx.TOP|wx.BOTTOM, border=0)
            for item in keys:
                if  item.lower() in ["errors", "count"]:
                    n += 1
                    text = wx.StaticText(self, -1, item + ": ", 
                                         style=wx.ALIGN_LEFT)
                    self.bck.Add(text, (n-1, 0), 
                                 flag=wx.LEFT|wx.ALIGN_CENTER_VERTICAL, 
                                 border=15)
                    ctl = wx.StaticText(self, -1, 
                                        str(format_number(params[item])),
                                        style=wx.ALIGN_LEFT)
                    ctl.SetToolTipString("Result %s" % item)
                    self.bck.Add(ctl, (n-1, 1), flag=wx.TOP|wx.BOTTOM, border=0)       
        self.bck.Layout()
        #self.bck.Fit(self)
        self.Layout()
        self.parent.GetSizer().Layout()
        
    def onSetFocus(self, evt):
        """
        Hightlight the textcrtl
        """
        # Get a handle to the TextCtrl
        widget = evt.GetEventObject()
        # Select the whole control, after this event resolves
        wx.CallAfter(widget.SetSelection, -1, -1)
        return
    
    def onParamChange(self, evt):
        """
        Receive and event and reset the text field contained in self.parameters
            
        """
        evt.Skip()
        for item in self.parameters:              
            if item[0] in evt.params:
                item[1].SetValue(format_number(evt.params[item[0]]))
                item[1].Refresh()
   
    def onTextEnter(self, evt): 
        """
        Parameters have changed
        """ 
        params = {}
        has_error = False
        for item in self.parameters:
            try:
                params[item[0]] = float(item[1].GetValue())
                item[1].SetBackgroundColour(
                        wx.SystemSettings_GetColour(wx.SYS_COLOUR_WINDOW))
                item[1].Refresh()
            except:
                has_error = True
                item[1].SetBackgroundColour("pink")
                item[1].Refresh()
            
        if has_error == False:
            # Post parameter event
            ## base is guiframe is this case
            event = SlicerParameterEvent(type=self.type, params=params)
            wx.PostEvent(self.base, event)