parameters_panel_slicer.py @ dfa1579

ESS_GUIESS_GUI_DocsESS_GUI_batch_fittingESS_GUI_bumps_abstractionESS_GUI_iss1116ESS_GUI_iss879ESS_GUI_iss959ESS_GUI_openclESS_GUI_orderingESS_GUI_sync_sascalccostrafo411magnetic_scattrelease-4.2.2ticket-1009ticket-1094-headlessticket-1242-2d-resolutionticket-1243ticket-1249ticket885unittest-saveload

Last change on this file since dfa1579 was dfa1579, checked in by krzywon, 7 years ago
Remove 1D data from batch slice list. Keep initial data set checked. Remove unused parameter.
Property mode set to `100644`
File size: 11.4 KB

Line
1
2
3	import wx
4	import wx.lib.newevent
5	#from copy import deepcopy
6	from sas.sasgui.guiframe.events import EVT_SLICER_PARS
7	from sas.sasgui.guiframe.utils import format_number
8	from sas.sasgui.guiframe.events import EVT_SLICER
9	from sas.sasgui.guiframe.events import SlicerParameterEvent, SlicerEvent
10	from Plotter2D import ModelPanel2D
11	from sas.sascalc.dataloader.data_info import Data1D, Data2D
12
13
14	class SlicerParameterPanel(wx.Dialog):
15	"""
16	Panel class to show the slicer parameters
17	"""
18	#TODO: show units
19	#TODO: order parameters properly
20
21	def __init__(self, parent, args, *kwargs):
22	"""
23	Dialog window that allow to edit parameters slicer
24	by entering new values
25	"""
26	wx.Dialog.__init__(self, parent, args, *kwargs)
27	self.params = {}
28	self.parent = parent
29	self.type = None
30	self.listeners = []
31	self.parameters = []
32	self.bck = wx.GridBagSizer(5, 5)
33	self.SetSizer(self.bck)
34	label = "Right-click on 2D plot for slicer options"
35	title = wx.StaticText(self, -1, label, style=wx.ALIGN_LEFT)
36	self.bck.Add(title, (0, 0), (1, 2),
37	flag=wx.LEFT \| wx.ALIGN_CENTER_VERTICAL, border=15)
38	# Bindings
39	self.parent.Bind(EVT_SLICER, self.onEVT_SLICER)
40	self.parent.Bind(EVT_SLICER_PARS, self.onParamChange)
41
42	def onEVT_SLICER(self, event):
43	"""
44	Process EVT_SLICER events
45	When the slicer changes, update the panel
46
47	:param event: EVT_SLICER event
48	"""
49	event.Skip()
50	if event.obj_class is None:
51	self.set_slicer(None, None)
52	else:
53	self.set_slicer(event.type, event.params)
54
55	def set_slicer(self, type, params):
56	"""
57	Rebuild the panel
58	"""
59	self.bck.Clear(True)
60	self.bck.Add((5, 5), (0, 0), (1, 1),
61	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 5)
62	self.type = type
63	if type is None:
64	label = "Right-click on 2D plot for slicer options"
65	title = wx.StaticText(self, -1, label, style=wx.ALIGN_LEFT)
66	self.bck.Add(title, (1, 0), (1, 2),
67	flag=wx.LEFT \| wx.ALIGN_CENTER_VERTICAL, border=15)
68	else:
69	title = wx.StaticText(self, -1,
70	"Slicer Parameters:", style=wx.ALIGN_LEFT)
71	self.bck.Add(title, (1, 0), (1, 2),
72	flag=wx.LEFT \| wx.ALIGN_CENTER_VERTICAL, border=15)
73	ix = 0
74	iy = 1
75	self.parameters = []
76	keys = params.keys()
77	keys.sort()
78	for item in keys:
79	iy += 1
80	ix = 0
81	if not item in ["count", "errors"]:
82	text = wx.StaticText(self, -1, item, style=wx.ALIGN_LEFT)
83	self.bck.Add(text, (iy, ix), (1, 1),
84	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 15)
85	ctl = wx.TextCtrl(self, -1, size=(80, 20),
86	style=wx.TE_PROCESS_ENTER)
87	hint_msg = "Modify the value of %s to change" % item
88	hint_msg += " the 2D slicer"
89	ctl.SetToolTipString(hint_msg)
90	ix = 1
91	ctl.SetValue(format_number(str(params[item])))
92	self.Bind(wx.EVT_TEXT_ENTER, self.onTextEnter)
93	self.parameters.append([item, ctl])
94	self.bck.Add(ctl, (iy, ix), (1, 1),
95	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
96	ix = 3
97	self.bck.Add((20, 20), (iy, ix), (1, 1),
98	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
99	else:
100	text = wx.StaticText(self, -1, item + " : ",
101	style=wx.ALIGN_LEFT)
102	self.bck.Add(text, (iy, ix), (1, 1),
103	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 15)
104	ctl = wx.StaticText(self, -1,
105	format_number(str(params[item])),
106	style=wx.ALIGN_LEFT)
107	ix = 1
108	self.bck.Add(ctl, (iy, ix), (1, 1),
109	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
110	ix = 0
111	iy += 1
112
113	# Change slicer within the window
114	txt = "Slicer"
115	text = wx.StaticText(self, -1, txt, style=wx.ALIGN_LEFT)
116	self.bck.Add(text, (iy, ix), (1, 1),
117	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 15)
118	type_list = ["SectorInteractor", "AnnulusInteractor",
119	"BoxInteractorX", "BoxInteractorY"]
120	self.type_select = wx.ComboBox(parent=self, choices=type_list)
121	self.Bind(wx.EVT_COMBOBOX, self.onChangeSlicer)
122	index = self.type_select.FindString(self.type)
123	self.type_select.SetSelection(index)
124	self.bck.Add(self.type_select, (iy, 1), (1, 1),
125	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 15)
126
127	# batch slicing parameters
128	title_text = "Batch Slicing Options:"
129	title = wx.StaticText(self, -1, title_text, style=wx.ALIGN_LEFT)
130	iy += 1
131	ln = wx.StaticLine(self, -1, style=wx.LI_VERTICAL)
132	ln.SetSize((60,60))
133	self.bck.Add(ln, (iy, ix), (1, 2),
134	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 15)
135	iy += 1
136	self.bck.Add(title, (iy, ix), (1, 1),
137	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 15)
138	iy += 1
139	self.process_list()
140	self.bck.Add(self.data_list, (iy, ix), (1, 1),
141	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 15)
142	iy += 1
143	button_label = "Apply Slicer to Selected Files"
144	self.batch_slicer_button = wx.Button(parent=self,
145	label=button_label)
146	self.Bind(wx.EVT_BUTTON, self.onBatchSlice)
147	self.bck.Add(self.batch_slicer_button, (iy, ix), (1, 1),
148	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 15)
149	# TODO: Check box for saving file
150	# TODO: append to file information and file type
151	# TODO: Send to fitting options
152
153	iy += 1
154	self.bck.Add((5, 5), (iy, ix), (1, 1),
155	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 5)
156	self.bck.Layout()
157	self.bck.Fit(self)
158	self.parent.GetSizer().Layout()
159
160	def onParamChange(self, evt):
161	"""
162	receive an event end reset value text fields
163	inside self.parameters
164	"""
165	evt.Skip()
166	if evt.type == "UPDATE":
167	for item in self.parameters:
168	if item[0] in evt.params:
169	item[1].SetValue("%-5.3g" % evt.params[item[0]])
170	item[1].Refresh()
171
172	def onTextEnter(self, evt):
173	"""
174	Parameters have changed
175	"""
176	params = {}
177	has_error = False
178	for item in self.parameters:
179	try:
180	params[item[0]] = float(item[1].GetValue())
181	item[1].SetBackgroundColour(
182	wx.SystemSettings_GetColour(wx.SYS_COLOUR_WINDOW))
183	item[1].Refresh()
184	except:
185	has_error = True
186	item[1].SetBackgroundColour("pink")
187	item[1].Refresh()
188
189	if not has_error:
190	# Post parameter event
191	# parent here is plotter2D
192	event = SlicerParameterEvent(type=self.type, params=params)
193	wx.PostEvent(self.parent, event)
194
195	def onBatchSlice(self, evt=None):
196	"""
197	Batch slicing button is pushed
198	:param evt: Event triggering hide/show of the batch slicer parameters
199	"""
200	# Process each data file individually
201	for item in self.data_list.CheckedStrings:
202	# Get data_id
203	num = len(item)
204	spp = self.parent.parent
205	data_panel = spp._data_panel
206	data_list = data_panel.list_cb_data
207	for key in data_list:
208	loaded_key = (key[:num]) if len(key) > num else key
209	if loaded_key == item:
210	selection = key
211	break
212	# Check the data checkbox
213	data_ctrl = data_list[selection][0]
214	self.check_item_and_children(data_ctrl=data_ctrl, check_value=True)
215	# Plot all checked data
216	data_panel.on_plot()
217	panel = plot_mgr = None
218	plot_mgrs = spp.plot_panels
219	for key, mgr in plot_mgrs.iteritems():
220	if isinstance(mgr, ModelPanel2D):
221	plot_mgr = mgr
222	break
223	if plot_mgr is not None:
224	# TODO: find proper plot window
225	panels = plot_mgr._manager.plot_panels
226	for _, pane in panels.iteritems():
227	if pane.window_name == item:
228	panel = pane
229	# TODO: apply slicer
230	print(panels)
231	if panel is not None:
232	self._apply_slicer_to_plot(panel)
233	# TODO: save file (if desired)
234	# TODO: send to fitting (if desired)
235
236	def onChangeSlicer(self, evt):
237	"""
238	Event driven slicer change when self.type_select changes
239	:param evt: Event triggering this change
240	"""
241	self._apply_slicer_to_plot(self.parent)
242
243	def _apply_slicer_to_plot(self, plot):
244	"""
245	Apply a slicer to any plot window, not just parent window
246	:param plot:
247	:return:
248	"""
249	type = self.type_select.GetStringSelection()
250	if self.type != type:
251	if type == "SectorInteractor":
252	plot.onSectorQ(None)
253	elif type == "AnnulusInteractor":
254	plot.onSectorPhi(None)
255	elif type == "BoxInteractorX":
256	plot.onBoxavgX(None)
257	elif type == "BoxInteractorY":
258	plot.onBoxavgY(None)
259
260	def check_item_and_children(self, data_ctrl, check_value=True):
261	self.parent.parent._data_panel.tree_ctrl.CheckItem(data_ctrl, check_value)
262	if data_ctrl.HasChildren():
263	if check_value and not data_ctrl.IsExpanded():
264	# Only select children if control is expanded
265	# Always deselect children, regardless (see ticket #259)
266	return
267	for child_ctrl in data_ctrl.GetChildren():
268	self.tree_ctrl.CheckItem(child_ctrl, check_value)
269
270	def process_list(self):
271	main_window = self.parent.parent
272	self.loaded_data = []
273	id = wx.NewId()
274	for key, value in main_window._data_manager.stored_data.iteritems():
275	if isinstance(value.data, Data2D):
276	self.loaded_data.append(value.data.name)
277	if key == self.parent.data2D.id:
278	self.checkme = self.loaded_data.index(value.data.name)
279	self.data_list = wx.CheckListBox(parent=self, id=id,
280	choices=self.loaded_data,
281	name="Apply Slicer to Data Sets:")
282	self.data_list.Check(self.checkme)
283	self.data_list.Bind(wx.EVT_CHECKLISTBOX, self.onCheckBoxList)
284
285	def onCheckBoxList(self, e):
286	index = e.GetSelection()
287	if index == self.checkme:
288	self.data_list.Check(index)

Note: See TracBrowser for help on using the repository browser.

SasView

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

Download in other formats: