source: sasview/guiframe/local_perspectives/plotting/slicerpanel.py @ ac8671e

"""
    Panel class to show the slicer parameters 
"""

import wx
import wx.lib.newevent
from copy import deepcopy

from sans.guiframe.utils import format_number
from sans.guicomm.events  import SlicerParameterEvent,EVT_SLICER_PARS,EVT_SLICER



class SlicerPanel(wx.Panel):
    #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={}, *args, **kwargs):
        wx.Panel.__init__(self, parent,id, *args, **kwargs)
        ##  Initialization of the class     
        self.base= base
        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:      
            title = wx.StaticText(self, -1, "Right-click on 2D plot for slicer options", 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:
            title = wx.StaticText(self, -1, "Right-click on 2D plot for slicer options", 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)
                    ctl.SetToolTipString("Modify the value of %s to change the 2D slicer" % item)
                    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)