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 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() 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) 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 # 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 = "" for key in params: default_value += "_{0}-".format(key) default_value += "{:.5f}".format(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, type=type) 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 """ # Skip redrawing the current plot if no change if self.parent == plot and self.type == type: return 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: event = SlicerParameterEvent(type=evt.type, params=evt.params) wx.PostEvent(item, event) # 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, type=evt.type) 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 names = [] convert_dict = {"SectorInteractor": "SectorQ", "AnnulusInteractor": "AnnulusPhi", "BoxInteractorX": "SlabX", "BoxInteractorY": "SlabY"} for f_name in evt.file_list: names.append(f_name.data2D.label) for key, plot in main_window.plot_panels.iteritems(): if not hasattr(plot, "data2D"): for item in plot.plots: base = item.replace(convert_dict[evt.type], "") if base in names: 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) 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())