def __init__(self, *args, **kwds):
PDFPanel.__init__(self)
# begin wxGlade: PlotPanel.__init__
kwds["style"] = wx.TAB_TRAVERSAL
wx.Panel.__init__(self, *args, **kwds)
self.sizer_4_staticbox = wx.StaticBox(self, -1, "Y")
self.sizer_3_staticbox = wx.StaticBox(self, -1, "X")
self.xDataCombo = wx.ComboBox(self, -1, choices=[], style=wx.CB_DROPDOWN|wx.CB_READONLY)
self.yDataList = KeyEventsListCtrl(self, -1, style=wx.LC_REPORT|wx.LC_NO_HEADER|wx.SUNKEN_BORDER)
self.offsetLabel = wx.StaticText(self, -1, "offset", style=wx.ALIGN_RIGHT)
self.offsetTextCtrl = wx.TextCtrl(self, -1, "-5", style=wx.TE_PROCESS_ENTER)
self.static_line_1 = wx.StaticLine(self, -1)
self.plotButton = wx.Button(self, -1, "Plot")
self.resetButton = wx.Button(self, -1, "Reset")
self.__set_properties()
self.__do_layout()
self.Bind(wx.EVT_TEXT_ENTER, self.onEnter, self.offsetTextCtrl)
self.Bind(wx.EVT_BUTTON, self.onPlot, self.plotButton)
self.Bind(wx.EVT_BUTTON, self.onReset, self.resetButton)
# end wxGlade
self.Bind(wx.EVT_COMBOBOX, self._check, self.xDataCombo)
self.Bind(wx.EVT_LIST_ITEM_SELECTED, self._check, self.yDataList)
self.Bind(wx.EVT_LIST_ITEM_DESELECTED, self._check, self.yDataList)
self.__customProperties()
def __init__(self, *args, **kwds):
PDFPanel.__init__(self)
# begin wxGlade: PlotPanel.__init__
kwds["style"] = wx.TAB_TRAVERSAL
wx.Panel.__init__(self, *args, **kwds)
self.sizer_4_staticbox = wx.StaticBox(self, -1, "Y")
self.sizer_3_staticbox = wx.StaticBox(self, -1, "X")
self.xDataCombo = wx.ComboBox(self, -1, choices=[], style=wx.CB_DROPDOWN|wx.CB_READONLY)
self.yDataList = KeyEventsListCtrl(self, -1, style=wx.LC_REPORT|wx.LC_NO_HEADER|wx.SUNKEN_BORDER)
self.offsetLabel = wx.StaticText(self, -1, "offset", style=wx.ALIGN_RIGHT)
self.offsetTextCtrl = wx.TextCtrl(self, -1, "-5", style=wx.TE_PROCESS_ENTER)
self.static_line_1 = wx.StaticLine(self, -1)
self.plotButton = wx.Button(self, -1, "Plot")
self.resetButton = wx.Button(self, -1, "Reset")
self.__set_properties()
self.__do_layout()
self.Bind(wx.EVT_TEXT_ENTER, self.onEnter, self.offsetTextCtrl)
self.Bind(wx.EVT_BUTTON, self.onPlot, self.plotButton)
self.Bind(wx.EVT_BUTTON, self.onReset, self.resetButton)
# end wxGlade
self.Bind(wx.EVT_COMBOBOX, self._check, self.xDataCombo)
self.Bind(wx.EVT_LIST_ITEM_SELECTED, self._check, self.yDataList)
self.Bind(wx.EVT_LIST_ITEM_DESELECTED, self._check, self.yDataList)
self.__customProperties()
class PlotPanel(wx.Panel, PDFPanel):
def __init__(self, *args, **kwds):
PDFPanel.__init__(self)
# begin wxGlade: PlotPanel.__init__
kwds["style"] = wx.TAB_TRAVERSAL
wx.Panel.__init__(self, *args, **kwds)
self.sizer_4_staticbox = wx.StaticBox(self, -1, "Y")
self.sizer_3_staticbox = wx.StaticBox(self, -1, "X")
self.xDataCombo = wx.ComboBox(self, -1, choices=[], style=wx.CB_DROPDOWN|wx.CB_READONLY)
self.yDataList = KeyEventsListCtrl(self, -1, style=wx.LC_REPORT|wx.LC_NO_HEADER|wx.SUNKEN_BORDER)
self.offsetLabel = wx.StaticText(self, -1, "offset", style=wx.ALIGN_RIGHT)
self.offsetTextCtrl = wx.TextCtrl(self, -1, "-5", style=wx.TE_PROCESS_ENTER)
self.static_line_1 = wx.StaticLine(self, -1)
self.plotButton = wx.Button(self, -1, "Plot")
self.resetButton = wx.Button(self, -1, "Reset")
self.__set_properties()
self.__do_layout()
self.Bind(wx.EVT_TEXT_ENTER, self.onEnter, self.offsetTextCtrl)
self.Bind(wx.EVT_BUTTON, self.onPlot, self.plotButton)
self.Bind(wx.EVT_BUTTON, self.onReset, self.resetButton)
# end wxGlade
self.Bind(wx.EVT_COMBOBOX, self._check, self.xDataCombo)
self.Bind(wx.EVT_LIST_ITEM_SELECTED, self._check, self.yDataList)
self.Bind(wx.EVT_LIST_ITEM_DESELECTED, self._check, self.yDataList)
self.__customProperties()
def __set_properties(self):
# begin wxGlade: PlotPanel.__set_properties
self.SetSize((456, 659))
self.offsetLabel.SetToolTipString("The vertical gap between stacked plots")
self.plotButton.SetToolTipString("Plot the selected data")
self.resetButton.SetToolTipString("Reset the plot configuration")
# end wxGlade
self.setToolTips(toolTips)
def __do_layout(self):
# begin wxGlade: PlotPanel.__do_layout
sizer_1 = wx.BoxSizer(wx.VERTICAL)
sizer_2 = wx.BoxSizer(wx.HORIZONTAL)
sizer_6 = wx.BoxSizer(wx.HORIZONTAL)
sizer_4 = wx.StaticBoxSizer(self.sizer_4_staticbox, wx.HORIZONTAL)
sizer_3 = wx.StaticBoxSizer(self.sizer_3_staticbox, wx.HORIZONTAL)
sizer_3.Add(self.xDataCombo, 1, wx.ALL, 5)
sizer_1.Add(sizer_3, 0, wx.EXPAND, 0)
sizer_4.Add(self.yDataList, 1, wx.ALL|wx.EXPAND, 5)
sizer_1.Add(sizer_4, 1, wx.EXPAND, 0)
sizer_6.Add(self.offsetLabel, 0, wx.ALL|wx.ALIGN_CENTER_VERTICAL, 5)
sizer_6.Add(self.offsetTextCtrl, 0, wx.ALL|wx.ALIGN_CENTER_VERTICAL, 5)
sizer_1.Add(sizer_6, 0, wx.EXPAND, 0)
sizer_1.Add(self.static_line_1, 0, wx.TOP|wx.BOTTOM|wx.EXPAND, 5)
sizer_2.Add(self.plotButton, 0, wx.ALL, 5)
sizer_2.Add(self.resetButton, 0, wx.ALL, 5)
sizer_1.Add(sizer_2, 0, wx.EXPAND, 0)
self.SetSizer(sizer_1)
# end wxGlade
# USER CONFIGURATION CODE #################################################
def __customProperties(self):
"""Custom Properties go here."""
self.yDataList.InsertColumn(0, "Y data")
self.offsetTextCtrl.SetValidator(TextValidator(FLOAT_ONLY,allowNeg=True))
# Testing Code. Comment or delete this block when finished.
#self.yDataList.InsertStringItem(sys.maxint, "y1")
#self.yDataList.InsertStringItem(sys.maxint, "y2")
#self.yDataList.InsertStringItem(sys.maxint, "y3")
#self.yDataList.InsertStringItem(sys.maxint, "y4")
#self.yDataList.InsertStringItem(sys.maxint, "y5")
# Initialize the sorter.
#self.yDataList.makeIDM()
#self.yDataList.initializeSorter()
return
def enableWidgets(self, on=True):
"""Enable or disable the widgets."""
self.xDataCombo.Enable(on)
self.yDataList.Enable(on)
self.offsetTextCtrl.Enable(on)
self.resetButton.Enable(on)
self.plotButton.Enable(on)
return
def updateWidgets(self):
"""Enable or disable certain widgets depending upon what is selected in
the tree and in the plotting widgets.
"""
# selections: selected nodes in treeCtrl
# fits: only different fittings
# refs: data item ids ( can be calculation, fit, structure and dataset
selections = self.treeCtrlMain.GetSelections()
# Only proceed if we have compatible items selected from the tree.
if not selections:
self.enableWidgets(False)
return
self.enableWidgets(True)
fits = dict.fromkeys([self.treeCtrlMain.GetControlData(self.treeCtrlMain.GetFitRoot(sel))
for sel in selections])
refs = [self.treeCtrlMain.GetControlData(sel) for sel in selections]
xdata = []
# step is added if selections include type other than calculation
for type in [ self.treeCtrlMain.GetNodeType(sel) for sel in selections ]:
if type != 'calculation':
xdata.append('step')
break
# index is added if mutiple selections are chosen from different fits
if len(fits) > 1:
xdata.append('index')
for ref in refs:
xdata.extend(ref.getXNames())
for fit in fits:
xdata.extend(fit.getMetaDataNames())
# also can plot y against y so add yNames as well
xdata.extend(fit.getYNames())
# reduce
xdata = dict.fromkeys(xdata).keys()
# Make the parameter entries a bit more presentable.
def _represent(mixedNames):
vals = ["@%i"%item for item in mixedNames if isinstance(item, int)]
others = [item for item in mixedNames if not isinstance(item, int)]
vals.extend(others)
numericStringSort(vals)
return vals
xvals = _represent(xdata)
try:
xvals.remove('rw')
except:
pass
numericStringSort(xvals)
# Fill the xDataCombo
if self.xDataCombo.GetCount():
current = self.xDataCombo.GetValue()
else:
current = None
self.xDataCombo.Clear()
for item in xvals:
self.xDataCombo.Append(item)
# Set default value for xDataCombo
# Either keep the current plot value selected, select 'r', or the
# first in the list.
defaultOrders = [ 'r', 'step', 'index']
if current:
defaultOrders.insert(0, current)
for item in defaultOrders:
if item in xvals:
self.xDataCombo.SetValue(item)
break
else:
self.xDataCombo.SetSelection(0)
# Y-DATA is the common subset of all data id
ydata = refs[0].getYNames()
for ref in refs[1:]:
for name in ydata[:]:
if name not in ref.getYNames(): ydata.remove(name)
yvals = _represent(ydata)
# Fill the List
self.yDataList.DeleteAllItems()
for val in yvals:
self.yDataList.InsertStringItem(sys.maxint, str(val))
self.yDataList.makeIDM()
self.yDataList.initializeSorter()
if yvals:
self.yDataList.Select(0)
#self.prevSelectionType = selectiontype
self._check(None)
return
def getSelectedYVals(self):
"""Get the y-values selected in the y-value ListCtrl."""
yvals = []
item = self.yDataList.GetFirstSelected()
first_item = item
while item != -1:
name = self.yDataList.GetItemText(item)
yvals.append(name)
item = self.yDataList.GetNextSelected(item)
return yvals
# EVENT CODE #############################################################
def onPlot(self, event): # wxGlade: PlotPanel.<event_handler>
"""Plot some stuff."""
self._plot(event)
return
def _plot (self,event):
"""This function is not wrapped"""
selections = self.treeCtrlMain.GetSelections()
refs = [self.treeCtrlMain.GetControlData(node) for node in selections]
xval = self.xDataCombo.GetValue()
if xval[0] == '@': xval = int(xval[1:])
temp = self.getSelectedYVals()
# Clean up some formatting so the control can understand this.
yvals = [ int(par[1:]) for par in temp if par[0] == '@']
yvals.extend([val for val in temp if val[0] != '@'])
offset = self.offsetTextCtrl.GetValue()
try:
offset = float(offset)
except ValueError: # offset can be empty string
offset = 0.0
self.mainFrame.control.plot(xval, yvals, refs, shift=offset, dry=(event is None))
return
def onEnter(self, event):
"""Reset plot."""
self.onPlot(event)
return
def onReset(self, event): # wxGlade: PlotPanel.<event_handler>
"""Reset everything."""
self.offsetTextCtrl.SetValue("-5")
self.refresh()
return
# Methods overloaded from PDFPanel
def refresh(self):
"""Refresh this panel."""
self.updateWidgets()
return
def _check(self, event):
try:
self._plot(None)
self.plotButton.Enable()
except ControlConfigError:
self.plotButton.Disable()
class PlotPanel(wx.Panel, PDFPanel):
def __init__(self, *args, **kwds):
PDFPanel.__init__(self)
# begin wxGlade: PlotPanel.__init__
kwds["style"] = wx.TAB_TRAVERSAL
wx.Panel.__init__(self, *args, **kwds)
self.sizer_4_staticbox = wx.StaticBox(self, -1, "Y")
self.sizer_3_staticbox = wx.StaticBox(self, -1, "X")
self.xDataCombo = wx.ComboBox(self, -1, choices=[], style=wx.CB_DROPDOWN|wx.CB_READONLY)
self.yDataList = KeyEventsListCtrl(self, -1, style=wx.LC_REPORT|wx.LC_NO_HEADER|wx.SUNKEN_BORDER)
self.offsetLabel = wx.StaticText(self, -1, "offset", style=wx.ALIGN_RIGHT)
self.offsetTextCtrl = wx.TextCtrl(self, -1, "-5", style=wx.TE_PROCESS_ENTER)
self.static_line_1 = wx.StaticLine(self, -1)
self.plotButton = wx.Button(self, -1, "Plot")
self.resetButton = wx.Button(self, -1, "Reset")
self.__set_properties()
self.__do_layout()
self.Bind(wx.EVT_TEXT_ENTER, self.onEnter, self.offsetTextCtrl)
self.Bind(wx.EVT_BUTTON, self.onPlot, self.plotButton)
self.Bind(wx.EVT_BUTTON, self.onReset, self.resetButton)
# end wxGlade
self.Bind(wx.EVT_COMBOBOX, self._check, self.xDataCombo)
self.Bind(wx.EVT_LIST_ITEM_SELECTED, self._check, self.yDataList)
self.Bind(wx.EVT_LIST_ITEM_DESELECTED, self._check, self.yDataList)
self.__customProperties()
def __set_properties(self):
# begin wxGlade: PlotPanel.__set_properties
self.SetSize((456, 659))
self.offsetLabel.SetToolTipString("The vertical gap between stacked plots")
self.plotButton.SetToolTipString("Plot the selected data")
self.resetButton.SetToolTipString("Reset the plot configuration")
# end wxGlade
self.setToolTips(toolTips)
def __do_layout(self):
# begin wxGlade: PlotPanel.__do_layout
sizer_1 = wx.BoxSizer(wx.VERTICAL)
sizer_2 = wx.BoxSizer(wx.HORIZONTAL)
sizer_6 = wx.BoxSizer(wx.HORIZONTAL)
sizer_4 = wx.StaticBoxSizer(self.sizer_4_staticbox, wx.HORIZONTAL)
sizer_3 = wx.StaticBoxSizer(self.sizer_3_staticbox, wx.HORIZONTAL)
sizer_3.Add(self.xDataCombo, 1, wx.ALL, 5)
sizer_1.Add(sizer_3, 0, wx.EXPAND, 0)
sizer_4.Add(self.yDataList, 1, wx.ALL|wx.EXPAND, 5)
sizer_1.Add(sizer_4, 1, wx.EXPAND, 0)
sizer_6.Add(self.offsetLabel, 0, wx.ALL|wx.ALIGN_CENTER_VERTICAL, 5)
sizer_6.Add(self.offsetTextCtrl, 0, wx.ALL|wx.ALIGN_CENTER_VERTICAL, 5)
sizer_1.Add(sizer_6, 0, wx.EXPAND, 0)
sizer_1.Add(self.static_line_1, 0, wx.TOP|wx.BOTTOM|wx.EXPAND, 5)
sizer_2.Add(self.plotButton, 0, wx.ALL, 5)
sizer_2.Add(self.resetButton, 0, wx.ALL, 5)
sizer_1.Add(sizer_2, 0, wx.EXPAND, 0)
self.SetSizer(sizer_1)
# end wxGlade
# USER CONFIGURATION CODE #################################################
def __customProperties(self):
"""Custom Properties go here."""
self.yDataList.InsertColumn(0, "Y data")
self.offsetTextCtrl.SetValidator(TextValidator(FLOAT_ONLY,allowNeg=True))
# Testing Code. Comment or delete this block when finished.
#self.yDataList.InsertStringItem(sys.maxint, "y1")
#self.yDataList.InsertStringItem(sys.maxint, "y2")
#self.yDataList.InsertStringItem(sys.maxint, "y3")
#self.yDataList.InsertStringItem(sys.maxint, "y4")
#self.yDataList.InsertStringItem(sys.maxint, "y5")
# Initialize the sorter.
#self.yDataList.makeIDM()
#self.yDataList.initializeSorter()
return
def enableWidgets(self, on=True):
"""Enable or disable the widgets."""
self.xDataCombo.Enable(on)
self.yDataList.Enable(on)
self.offsetTextCtrl.Enable(on)
self.resetButton.Enable(on)
self.plotButton.Enable(on)
return
def updateWidgets(self):
"""Enable or disable certain widgets depending upon what is selected in
the tree and in the plotting widgets.
"""
# selections: selected nodes in treeCtrl
# fits: only different fittings
# refs: data item ids ( can be calculation, fit, structure and dataset
selections = self.treeCtrlMain.GetSelections()
# Only proceed if we have compatible items selected from the tree.
if not selections:
self.enableWidgets(False)
return
self.enableWidgets(True)
fits = dict.fromkeys([self.treeCtrlMain.GetControlData(self.treeCtrlMain.GetFitRoot(sel))
for sel in selections])
refs = [self.treeCtrlMain.GetControlData(sel) for sel in selections]
xdata = []
# step is added if selections include type other than calculation
for type in [ self.treeCtrlMain.GetNodeType(sel) for sel in selections ]:
if type != 'calculation':
xdata.append('step')
break
# index is added if mutiple selections are chosen from different fits
if len(fits) > 1:
xdata.append('index')
for ref in refs:
xdata.extend(ref.getXNames())
for fit in fits:
xdata.extend(fit.getMetaDataNames())
# also can plot y against y so add yNames as well
xdata.extend(fit.getYNames())
# reduce
xdata = dict.fromkeys(xdata).keys()
# Make the parameter entries a bit more presentable.
def _represent(mixedNames):
vals = ["@%i"%item for item in mixedNames if isinstance(item, int)]
others = [item for item in mixedNames if not isinstance(item, int)]
vals.extend(others)
numericStringSort(vals)
return vals
xvals = _represent(xdata)
try:
xvals.remove('rw')
except:
pass
numericStringSort(xvals)
# Fill the xDataCombo
if self.xDataCombo.GetCount():
current = self.xDataCombo.GetValue()
else:
current = None
self.xDataCombo.Clear()
for item in xvals:
self.xDataCombo.Append(item)
# Set default value for xDataCombo
# Either keep the current plot value selected, select 'r', or the
# first in the list.
defaultOrders = [ 'r', 'step', 'index']
if current:
defaultOrders.insert(0, current)
for item in defaultOrders:
if item in xvals:
self.xDataCombo.SetValue(item)
break
else:
self.xDataCombo.SetSelection(0)
# Y-DATA is the common subset of all data id
ydata = refs[0].getYNames()
for ref in refs[1:]:
for name in ydata[:]:
if name not in ref.getYNames(): ydata.remove(name)
yvals = _represent(ydata)
# Fill the List
self.yDataList.DeleteAllItems()
for val in yvals:
self.yDataList.InsertStringItem(sys.maxint, str(val))
self.yDataList.makeIDM()
self.yDataList.initializeSorter()
if yvals:
self.yDataList.Select(0)
#self.prevSelectionType = selectiontype
self._check(None)
return
def getSelectedYVals(self):
"""Get the y-values selected in the y-value ListCtrl."""
yvals = []
item = self.yDataList.GetFirstSelected()
while item != -1:
name = self.yDataList.GetItemText(item)
yvals.append(name)
item = self.yDataList.GetNextSelected(item)
return yvals
# EVENT CODE #############################################################
def onPlot(self, event): # wxGlade: PlotPanel.<event_handler>
"""Plot some stuff."""
self._plot(event)
return
def _plot (self,event):
"""This function is not wrapped"""
selections = self.treeCtrlMain.GetSelections()
refs = [self.treeCtrlMain.GetControlData(node) for node in selections]
xval = self.xDataCombo.GetValue()
if xval[0] == '@': xval = int(xval[1:])
temp = self.getSelectedYVals()
# Clean up some formatting so the control can understand this.
yvals = [ int(par[1:]) for par in temp if par[0] == '@']
yvals.extend([val for val in temp if val[0] != '@'])
offset = self.offsetTextCtrl.GetValue()
try:
offset = float(offset)
except ValueError: # offset can be empty string
offset = 0.0
self.mainFrame.control.plot(xval, yvals, refs, shift=offset, dry=(event is None))
return
def onEnter(self, event):
"""Reset plot."""
self.onPlot(event)
return
def onReset(self, event): # wxGlade: PlotPanel.<event_handler>
"""Reset everything."""
self.offsetTextCtrl.SetValue("-5")
self.refresh()
return
# Methods overloaded from PDFPanel
def refresh(self):
"""Refresh this panel."""
self.updateWidgets()
return
def _check(self, event):
try:
self._plot(None)
self.plotButton.Enable()
except ControlConfigError:
self.plotButton.Disable()
