source: sasview/src/sas/sasgui/guiframe/local_perspectives/plotting/parameters_panel_slicer.py @ 982577b

import wx
import wx.lib.newevent
import time
from sas.sascalc.dataloader.readers.cansas_reader import Reader
from sas.sasgui.guiframe.events import EVT_SLICER_PARS
from sas.sasgui.guiframe.utils import format_number
from sas.sasgui.guiframe.events import EVT_SLICER
from sas.sasgui.guiframe.events import SlicerParameterEvent, SlicerEvent
from Plotter1D import ModelPanel1D
from Plotter2D import ModelPanel2D
from sas.sascalc.dataloader.data_info import Data1D, Data2D
apply_params, EVT_APPLY_PARAMS = wx.lib.newevent.NewEvent()
auto_save, EVT_AUTO_SAVE = wx.lib.newevent.NewEvent()
auto_close, EVT_ON_CLOSE = wx.lib.newevent.NewEvent()


class SlicerParameterPanel(wx.Dialog):
    """
    Panel class to show the slicer parameters
    """
    # TODO: show units
    # TODO: order parameters properly

    def __init__(self, parent, *args, **kwargs):
        """
        Dialog window that allow to edit parameters slicer
        by entering new values
        """
        wx.Dialog.__init__(self, parent, *args, **kwargs)
        self.params = {}
        self.parent = parent
        self.type = None
        self.listeners = []
        self.parameters = []
        self.bck = wx.GridBagSizer(5, 5)
        self.SetSizer(self.bck)
        self.auto_save = None
        self.path = None
        self.type_list = ["SectorInteractor", "AnnulusInteractor",
                          "BoxInteractorX", "BoxInteractorY"]
        self.type_select = wx.ComboBox(parent=self, choices=self.type_list)
        self.append_name = wx.TextCtrl(parent=self, id=wx.NewId(),
                                       name="Append to file name:")
        self.data_list = 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)
        # Bindings
        self.parent.Bind(EVT_SLICER, self.onEVT_SLICER)
        self.parent.Bind(EVT_SLICER_PARS, self.onParamChange)
        self.Bind(EVT_APPLY_PARAMS, self.apply_params_list_and_process)
        self.Bind(EVT_AUTO_SAVE, self.save_files)
        self.Bind(EVT_ON_CLOSE, self.on_close)

    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 is 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.bck.Add((5, 5), (0, 0), (1, 1),
                     wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 5)
        self.type = type
        if type is None:
            label = "Right-click on 2D plot for slicer options"
            title = wx.StaticText(self, -1, label, style=wx.ALIGN_LEFT)
            self.bck.Add(title, (1, 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, (1, 0), (1, 2),
                         flag=wx.LEFT | wx.ALIGN_CENTER_VERTICAL, border=15)
            iy = 1
            self.parameters = []
            keys = params.keys()
            keys.sort()
            for item in keys:
                iy += 1
                ix = 0
                if not item in ["count", "errors"]:
                    text = wx.StaticText(self, -1, item, style=wx.ALIGN_LEFT)
                    self.bck.Add(text, (iy, ix), (1, 1),
                                 wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 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)
                    ix = 1
                    ctl.SetValue(format_number(str(params[item])))
                    self.Bind(wx.EVT_TEXT_ENTER, self.onTextEnter)
                    self.parameters.append([item, ctl])
                    self.bck.Add(ctl, (iy, ix), (1, 1),
                                 wx.EXPAND | wx.ADJUST_MINSIZE, 0)
                    ix = 3
                    self.bck.Add((20, 20), (iy, ix), (1, 1),
                                 wx.EXPAND | wx.ADJUST_MINSIZE, 0)
                else:
                    text = wx.StaticText(self, -1, item + " : ",
                                         style=wx.ALIGN_LEFT)
                    self.bck.Add(text, (iy, ix), (1, 1),
                                 wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)
                    ctl = wx.StaticText(self, -1,
                                        format_number(str(params[item])),
                                        style=wx.ALIGN_LEFT)
                    ix = 1
                    self.bck.Add(ctl, (iy, ix), (1, 1),
                                 wx.EXPAND | wx.ADJUST_MINSIZE, 0)

            # Change slicer within the window
            ix = 0
            iy += 1
            txt = "Slicer type:"
            text = wx.StaticText(self, -1, txt, style=wx.ALIGN_LEFT)
            self.bck.Add(text, (iy, ix), (1, 1),
                         wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)
            self.Bind(wx.EVT_COMBOBOX, self.onChangeSlicer)
            index = self.type_select.FindString(type)
            self.type_select.SetSelection(index)
            self.bck.Add(self.type_select, (iy, 1), (1, 1),
                         wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)

            # batch slicing parameters
            title_text = "Batch Slicing Options:"
            title = wx.StaticText(self, -1, title_text, style=wx.ALIGN_LEFT)
            iy += 1
            line = wx.StaticLine(self, -1, style=wx.LI_VERTICAL)
            line.SetSize((60, 60))
            self.bck.Add(line, (iy, ix), (1, 2),
                         wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)
            iy += 1
            self.bck.Add(title, (iy, ix), (1, 1),
                         wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)

            # Create a list box with all of the 2D plots
            iy += 1
            self.process_list()
            self.bck.Add(self.data_list, (iy, ix), (1, 1),
                         wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)

            # Checkbox for autosaving data
            iy += 1

            self.auto_save = wx.CheckBox(parent=self, id=wx.NewId(),
                                         label="Auto save generated 1D:")
            self.Bind(wx.EVT_CHECKBOX, self.on_auto_save_checked)
            self.bck.Add(self.auto_save, (iy, ix), (1, 1),
                         wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)
            iy += 1
            # TODO: Get list of loaded data, not plots - plot again
            # TODO: try/catch block to catch wx._core.PyDeadObjectError (pass)
            # File browser
            save_to = "Save files to:"
            save = wx.StaticText(self, -1, save_to, style=wx.ALIGN_LEFT)
            self.path = wx.DirPickerCtrl(self, id=wx.NewId(), path="",
                                         message=save_to)
            self.path.Enable(False)
            self.bck.Add(save, (iy, ix), (1, 1),
                         wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)
            self.bck.Add(self.path, (iy, 1), (1, 1),
                         wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)
            # Append to file
            iy += 1
            default_value = "_{0}".format(self.type)
            for key in params:
                default_value += "_%d.2" % params[key]
            append_text = "Append to file name:"
            append = wx.StaticText(self, -1, append_text, style=wx.ALIGN_LEFT)
            self.append_name.SetValue(default_value)
            self.append_name.Enable(False)
            self.bck.Add(append, (iy, ix), (1, 1),
                         wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)
            self.bck.Add(self.append_name, (iy, 1), (1, 1),
                         wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)

            # TODO: Fix fitting options combobox/radiobox
            # Combobox for selecting fitting options
            # iy += 1
            # self.fitting_radio = wx.RadioBox(parent=self, id=wx.NewId(),
            #                                  size=(4,1))
            # self.fitting_radio.SetString(0, "No fitting")
            # self.fitting_radio.SetString(1, "Batch Fitting")
            # self.fitting_radio.SetString(2, "Fitting")
            # self.fitting_radio.SetString(3, "Simultaneous and Constrained Fit")
            # self.fitting_radio.SetValue(0)
            # self.bck.Add(self.fitting_radio, (iy, ix), (1, 1),
            #              wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)

            # Button to start batch slicing
            iy += 1
            button_label = "Apply Slicer to Selected Plots"
            self.batch_slicer_button = wx.Button(parent=self,
                                                 label=button_label)
            self.Bind(wx.EVT_BUTTON, self.on_batch_slicer)
            self.bck.Add(self.batch_slicer_button, (iy, ix), (1, 1),
                         wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)
            iy += 1
            self.bck.Add((5, 5), (iy, ix), (1, 1),
                         wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 5)
        self.bck.Layout()
        self.bck.Fit(self)
        self.parent.GetSizer().Layout()

    def onParamChange(self, evt):
        """
        receive an event end reset value text fields
        inside self.parameters
        """
        evt.Skip()
        if evt.type == "UPDATE":
            for item in self.parameters:
                if item[0] in evt.params:
                    item[1].SetValue("%-5.3g" % 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 not has_error:
            # Post parameter event
            # parent here is plotter2D
            event = SlicerParameterEvent(type=self.type, params=params)
            wx.PostEvent(self, event)

    def on_batch_slicer(self, evt=None):
        """
        Method invoked with batch slicing button is pressed
        :param evt: Event triggering hide/show of the batch slicer parameters
        """
        apply_to_list = []
        spp = self.parent.parent
        params = self.parent.slicer.get_params()
        type = self.type_select.GetStringSelection()
        save = self.auto_save.IsChecked()
        append = self.append_name.GetValue()
        path = self.path.GetPath()

        # Find desired 2D data panels
        for key, mgr in spp.plot_panels.iteritems():
            if mgr.graph.prop['title'] in self.data_list.CheckedStrings:
                apply_to_list.append(mgr)

        # Apply slicer type to selected panels
        for item in apply_to_list:
            self._apply_slicer_to_plot(item, type)

        # Post an event to apply appropriate slicer params to each slicer
        # Event needed due to how apply_slicer_to_plot works
        event = apply_params(params=params, plot_list=apply_to_list,
                             auto_save=save, append=append,
                             path=path)
        wx.PostEvent(self, event)
        event = auto_close()
        wx.PostEvent(self, event)

    def onChangeSlicer(self, evt):
        """
        Event driven slicer change when self.type_select changes
        :param evt: Event triggering this change
        """
        self._apply_slicer_to_plot(self.parent)

    def _apply_slicer_to_plot(self, plot, type=None):
        """
        Apply a slicer to *any* plot window, not just parent window
        :param plot: 2D plot panel to apply a slicer to
        :param type: The type of slicer to apply to the panel
        """
        if type is None:
            type = self.type_select.GetStringSelection()
        if type == "SectorInteractor":
            plot.onSectorQ(None)
        elif type == "AnnulusInteractor":
            plot.onSectorPhi(None)
        elif type == "BoxInteractorX":
            plot.onBoxavgX(None)
        elif type == "BoxInteractorY":
            plot.onBoxavgY(None)

    def process_list(self):
        """
        Populate the check list from the currently plotted 2D data
        """
        self.checkme = None
        main_window = self.parent.parent
        self.loaded_data = []
        id = wx.NewId()
        # Iterate over the loaded plots and find all 2D panels
        for key, value in main_window.plot_panels.iteritems():
            if isinstance(value, ModelPanel2D):
                self.loaded_data.append(value.data2D.name)
                if value.data2D.id == self.parent.data2D.id:
                    # Set current plot panel as uncheckable
                    self.checkme = self.loaded_data.index(value.data2D.name)
        self.data_list = wx.CheckListBox(parent=self, id=id,
                                         choices=self.loaded_data,
                                         name="Apply Slicer to 2D Plots:")
        # Check all items by default
        for item in range(len(self.data_list.Items)):
            self.data_list.Check(item)
        self.data_list.Bind(wx.EVT_CHECKLISTBOX, self.on_check_box_list)

    def on_check_box_list(self, evt=None):
        """
        Prevent a checkbox item from being unchecked
        :param e: Event triggered when a checkbox list item is checked
        """
        if evt is None:
            return
        index = evt.GetSelection()
        if index == self.checkme:
            self.data_list.Check(index)

    def apply_params_list_and_process(self, evt=None):
        """
        Event based parameter setting.
        :param evt: Event triggered to apply parameters to a list of plots
                    evt should have attrs plot_list and params
        """
        # Apply parameter list to each plot as desired
        for item in evt.plot_list:
            item.slicer.set_params(evt.params)
            item.slicer.base.update()
        # Post an event to save each data set to file
        if evt.auto_save:
            event = auto_save(append_to_name=evt.append,
                              file_list=evt.plot_list,
                              path=evt.path)
            wx.PostEvent(self, event)

    def save_files(self, evt=None):
        """
        Automatically save the sliced data to file.
        :param evt: Event that triggered the call to the method
        """
        if evt is None:
            return
        writer = Reader()
        main_window = self.parent.parent
        data_dic = {}
        append = evt.append_to_name
        for key, plot in main_window.plot_panels.iteritems():
            if not hasattr(plot, "data2D"):
                for item in plot.plots:
                    data_dic[item] = plot.plots[item]
        for item, data1d in data_dic.iteritems():
            base = item.split(".")[0]
            save_to = evt.path + "\\" + base + append + ".xml"
            writer.write(save_to, data1d)
        # TODO: save all files

    def on_auto_save_checked(self, evt=None):
        """
        Enable/Disable auto append when checkbox is checked
        :param evt: Event
        """
        self.append_name.Enable(self.auto_save.IsChecked())
        self.path.Enable(self.auto_save.IsChecked())
    
    def on_close(self, evt=None):
        """
        Auto close the panel
        """
        self.Destroy()