source: sasview/prview/perspectives/pr/pr_widgets.py @ cacbd7d

################################################################################
#This software was developed by the University of Tennessee as part of the
#Distributed Data Analysis of Neutron Scattering Experiments (DANSE)
#project funded by the US National Science Foundation. 
#
#See the license text in license.txt
#
#copyright 2009, University of Tennessee
################################################################################

"""
Text controls for input/output of the main PrView panel
"""

import wx
import os
from wx.lib.scrolledpanel import ScrolledPanel
class PrTextCtrl(wx.TextCtrl):
    """
    Text control for model and fit parameters.
    Binds the appropriate events for user interactions.
    """
    def __init__(self, *args, **kwds):
        
        wx.TextCtrl.__init__(self, *args, **kwds)
        
        ## Set to True when the mouse is clicked while the whole string is selected
        full_selection = False
        ## Call back for EVT_SET_FOCUS events
        _on_set_focus_callback = None
        # Bind appropriate events
        self.Bind(wx.EVT_LEFT_UP, self._highlight_text)
        self.Bind(wx.EVT_SET_FOCUS, self._on_set_focus)

    def _on_set_focus(self, event):
        """
        Catch when the text control is set in focus to highlight the whole
        text if necessary
        
        :param event: mouse event
        
        """
        event.Skip()
        self.full_selection = True
        
    def _highlight_text(self, event):
        """
        Highlight text of a TextCtrl only of no text has be selected
        
        :param event: mouse event
        
        """
        # Make sure the mouse event is available to other listeners
        event.Skip()
        control  = event.GetEventObject()
        if self.full_selection:
            self.full_selection = False
            # Check that we have a TextCtrl
            if issubclass(control.__class__, wx.TextCtrl):
                # Check whether text has been selected, 
                # if not, select the whole string
                (start, end) = control.GetSelection()
                if start==end:
                    control.SetSelection(-1,-1)

class OutputTextCtrl(wx.TextCtrl):
    """
    Text control used to display outputs.
    No editing allowed. The background is 
    grayed out. User can't select text.
    """
    def __init__(self, *args, **kwds):
        """
        """
        wx.TextCtrl.__init__(self, *args, **kwds)
        self.SetEditable(False)
        self.SetBackgroundColour(self.GetParent().GetBackgroundColour())
        
        # Bind to mouse event to avoid text highlighting
        # The event will be skipped once the call-back
        # is called.
        self.Bind(wx.EVT_MOUSE_EVENTS, self._click)
        
    def _click(self, event):
        """
        Prevent further handling of the mouse event
        by not calling Skip().
        """ 
        pass
        

class DataFileTextCtrl(OutputTextCtrl):
    """
    Text control used to display only the file name
    given a full path.
     
    :TODO: now that we no longer choose the data file from the panel,
        it's no longer necessary to pass around the file path. That code
        should be refactored away and simplified. 
    """
    def __init__(self, *args, **kwds):
        """
        """
        OutputTextCtrl.__init__(self, *args, **kwds)
        self._complete_path = None
    
    def SetValue(self, value):
        """
        Sets the file name given a path
        """
        self._complete_path = str(value)
        file = os.path.basename(self._complete_path)
        OutputTextCtrl.SetValue(self, file)
        
    def GetValue(self):
        """
        Return the full path
        """
        return self._complete_path
    
class DataDialog(wx.Dialog):
    """
    Allow file selection at loading time
    """
    def __init__(self, data_list, parent=None, text='', *args, **kwds):
        wx.Dialog.__init__(self, parent, *args, **kwds)
        self.list_of_ctrl = []
        self._do_layout(data_list, text=text)
        
    def _do_layout(self, data_list, text=''):
        """
        layout the dialog
        """
        if not data_list or len(data_list) <= 1:
            return 
        # Dialog box properties
        self.SetTitle("Data Selection")
        w = 400
        h = 200
        self.SetSize((w, h))
        sizer_main = wx.BoxSizer(wx.VERTICAL)
        sizer_txt = wx.BoxSizer(wx.VERTICAL)
        sizer_button = wx.BoxSizer(wx.HORIZONTAL)
        choice_sizer = wx.GridBagSizer(5, 5)
        #choice_sizer.SetMinSize((w-20, h-50))
        panel = ScrolledPanel(self, style=wx.RAISED_BORDER, size=(w-20, h-50))
        panel.SetupScrolling()
        #add text
        if text.strip() == "":
            text = "This Perspective does not allow multiple data !\n"
            text += "Please select only one Data.\n"
        text_ctrl = wx.StaticText(self, -1, str(text))
        sizer_txt.Add(text_ctrl)
        iy = 0
        ix = 0
        rbox = wx.RadioButton(panel, -1, str(data_list[0].name), 
                                  (10, 10), style= wx.RB_GROUP)
        rbox.SetValue(True)
        self.list_of_ctrl.append((rbox, data_list[0]))
        choice_sizer.Add(rbox, (iy, ix), (1, 1),
                         wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
        for i in range(1, len(data_list)):
            iy += 1
            rbox = wx.RadioButton(panel, -1, str(data_list[i].name), (10, 10))
            rbox.SetValue(False)
            self.list_of_ctrl.append((rbox, data_list[i]))
            choice_sizer.Add(rbox, (iy, ix),
                           (1, 1), wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
        panel.SetSizer(choice_sizer)
        #add sizer
        sizer_button.Add((20, 20), 1, wx.EXPAND|wx.ADJUST_MINSIZE, 0)
        button_cancel = wx.Button(self, wx.ID_CANCEL, "Cancel")
        sizer_button.Add(button_cancel, 0,
                          wx.LEFT|wx.RIGHT|wx.ADJUST_MINSIZE, 10)
        button_OK = wx.Button(self, wx.ID_OK, "Ok")
        button_OK.SetFocus()
        sizer_button.Add(button_OK, 0, wx.LEFT|wx.RIGHT|wx.ADJUST_MINSIZE, 10)
        static_line = wx.StaticLine(self, -1)
        
        sizer_txt.Add(panel, 1, wx.EXPAND|wx.LEFT|wx.RIGHT, 5)
        sizer_main.Add(sizer_txt, 1, wx.EXPAND|wx.ALL, 10)
        sizer_main.Add(static_line, 0, wx.EXPAND, 0)
        sizer_main.Add(sizer_button, 0, wx.EXPAND|wx.ALL, 10)
        self.SetSizer(sizer_main)
        self.Layout()
        
    def get_data(self):
        """
        return the selected data
        """
        for item in self.list_of_ctrl:
            rbox, data = item
            if rbox.GetValue():
                return data