data_panel.py @ af7ba61

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.1.1release-4.1.2release-4.2.2release_4.0.1ticket-1009ticket-1094-headlessticket-1242-2d-resolutionticket-1243ticket-1249ticket885unittest-saveload

Last change on this file since af7ba61 was af7ba61, checked in by Jae Cho <jhjcho@…>, 13 years ago
check the wx version on MAC
Property mode set to `100644`
File size: 45.0 KB

Line
1	################################################################################
2	#This software was developed by the University of Tennessee as part of the
3	#Distributed Data Analysis of Neutron Scattering Experiments (DANSE)
4	#project funded by the US National Science Foundation.
5	#
6	#See the license text in license.txt
7	#
8	#copyright 2010, University of Tennessee
9	################################################################################
10	"""
11	This module provides Graphic interface for the data_manager module.
12	"""
13	import os
14	import wx
15	# Check version
16	toks = wx.__version__.split('.')
17	if int(toks[1]) < 9:
18	wx_version = 8
19	else:
20	wx_version = 9
21	import sys
22	import warnings
23	import logging
24	from wx.lib.scrolledpanel import ScrolledPanel
25	import wx.lib.agw.customtreectrl as CT
26	from sans.guiframe.dataFitting import Data1D
27	from sans.guiframe.dataFitting import Data2D
28	from sans.guiframe.panel_base import PanelBase
29	from sans.guiframe.events import StatusEvent
30	from sans.guiframe.events import EVT_DELETE_PLOTPANEL
31	from sans.guiframe.events import NewLoadDataEvent
32	from sans.guiframe.events import NewPlotEvent
33	from sans.guiframe.gui_style import GUIFRAME
34	from DataLoader.loader import Loader
35
36	try:
37	# Try to find a local config
38	import imp
39	path = os.getcwd()
40	if(os.path.isfile("%s/%s.py" % (path, 'local_config'))) or \
41	(os.path.isfile("%s/%s.pyc" % (path, 'local_config'))):
42	fObj, path, descr = imp.find_module('local_config', [path])
43	config = imp.load_module('local_config', fObj, path, descr)
44	else:
45	# Try simply importing local_config
46	import local_config as config
47	except:
48	# Didn't find local config, load the default
49	import config
50
51	extension_list = []
52	if config.APPLICATION_STATE_EXTENSION is not None:
53	extension_list.append(config.APPLICATION_STATE_EXTENSION)
54	EXTENSIONS = config.PLUGIN_STATE_EXTENSIONS + extension_list
55	PLUGINS_WLIST = config.PLUGINS_WLIST
56	APPLICATION_WLIST = config.APPLICATION_WLIST
57
58	#Control panel width
59	if sys.platform.count("win32") > 0:
60	PANEL_WIDTH = 235
61	PANEL_HEIGHT = 700
62	CBOX_WIDTH = 140
63	BUTTON_WIDTH = 80
64	FONT_VARIANT = 0
65	IS_MAC = False
66	else:
67	PANEL_WIDTH = 255
68	PANEL_HEIGHT = 750
69	CBOX_WIDTH = 155
70	BUTTON_WIDTH = 100
71	FONT_VARIANT = 1
72	IS_MAC = True
73
74	STYLE_FLAG =wx.RAISED_BORDER\|CT.TR_HAS_BUTTONS\| CT.TR_HIDE_ROOT\|\
75	wx.WANTS_CHARS\|CT.TR_HAS_VARIABLE_ROW_HEIGHT
76
77
78	class DataTreeCtrl(CT.CustomTreeCtrl):
79	"""
80	Check list control to be used for Data Panel
81	"""
82	def __init__(self, parent,args, *kwds):
83	#agwstyle is introduced in wx.2.8.11 but is not working for mac
84	if IS_MAC and wx_version < 9:
85	try:
86	kwds['style'] = STYLE_FLAG
87	CT.CustomTreeCtrl.__init__(self, parent, args, *kwds)
88	except:
89	del kwds['style']
90	CT.CustomTreeCtrl.__init__(self, parent, args, *kwds)
91	else:
92	#agwstyle is introduced in wx.2.8.11 .argument working only for windows
93	try:
94	kwds['agwStyle'] = STYLE_FLAG
95	CT.CustomTreeCtrl.__init__(self, parent, args, *kwds)
96	except:
97	try:
98	del kwds['agwStyle']
99	kwds['style'] = STYLE_FLAG
100	CT.CustomTreeCtrl.__init__(self, parent, args, *kwds)
101	except:
102	del kwds['style']
103	CT.CustomTreeCtrl.__init__(self, parent, args, *kwds)
104	self.root = self.AddRoot("Available Data")
105
106	class DataPanel(ScrolledPanel, PanelBase):
107	"""
108	This panel displays data available in the application and widgets to
109	interact with data.
110	"""
111	## Internal name for the AUI manager
112	window_name = "Data Panel"
113	## Title to appear on top of the window
114	window_caption = "Data Explorer"
115	#type of window
116	window_type = "Data Panel"
117	## Flag to tell the GUI manager that this panel is not
118	# tied to any perspective
119	#ALWAYS_ON = True
120	def __init__(self, parent,
121	list=None,
122	size=(PANEL_WIDTH, PANEL_HEIGHT),
123	list_of_perspective=None, manager=None, args, *kwds):
124	kwds['size']= size
125	kwds['style'] = STYLE_FLAG
126	ScrolledPanel.__init__(self, parent=parent, args, *kwds)
127	PanelBase.__init__(self)
128	self.SetupScrolling()
129	#Set window's font size
130	self.SetWindowVariant(variant=FONT_VARIANT)
131	self.loader = Loader()
132	#Default location
133	self._default_save_location = None
134	self.all_data1d = True
135	self.parent = parent
136	self.manager = manager
137	if list is None:
138	list = []
139	self.list_of_data = list
140	if list_of_perspective is None:
141	list_of_perspective = []
142	self.list_of_perspective = list_of_perspective
143	self.list_rb_perspectives= []
144	self.list_cb_data = {}
145	self.list_cb_theory = {}
146	self.tree_ctrl = None
147	self.tree_ctrl_theory = None
148	self.perspective_cbox = None
149
150	self.owner = None
151	self.do_layout()
152	self.fill_cbox_analysis(self.list_of_perspective)
153	self.Bind(wx.EVT_SHOW, self.on_close_page)
154	if self.parent is not None:
155	self.parent.Bind(EVT_DELETE_PLOTPANEL, self._on_delete_plot_panel)
156
157
158	def do_layout(self):
159	"""
160	"""
161	self.define_panel_structure()
162	self.layout_selection()
163	self.layout_data_list()
164	self.layout_button()
165	#self.layout_batch()
166
167	def define_panel_structure(self):
168	"""
169	Define the skeleton of the panel
170	"""
171	w, h = self.parent.GetSize()
172	self.vbox = wx.BoxSizer(wx.VERTICAL)
173	self.sizer1 = wx.BoxSizer(wx.VERTICAL)
174	self.sizer1.SetMinSize((w/13, h*2/5))
175
176	self.sizer2 = wx.BoxSizer(wx.VERTICAL)
177	self.sizer3 = wx.FlexGridSizer(7, 2, 4, 1)
178	self.sizer4 = wx.BoxSizer(wx.HORIZONTAL)
179	self.sizer5 = wx.BoxSizer(wx.VERTICAL)
180
181	self.vbox.Add(self.sizer5, 0, wx.EXPAND\|wx.ALL,1)
182	self.vbox.Add(self.sizer1, 0, wx.EXPAND\|wx.ALL,0)
183	self.vbox.Add(self.sizer2, 0, wx.EXPAND\|wx.ALL,1)
184	self.vbox.Add(self.sizer3, 0, wx.EXPAND\|wx.ALL,5)
185	self.vbox.Add(self.sizer4, 0, wx.EXPAND\|wx.ALL,5)
186
187	self.SetSizer(self.vbox)
188
189	def layout_selection(self):
190	"""
191	"""
192	select_txt = wx.StaticText(self, -1, 'Selection Options')
193	select_txt.SetForegroundColour('blue')
194	self.selection_cbox = wx.ComboBox(self, -1, style=wx.CB_READONLY)
195	list_of_options = ['Select all Data',
196	'Unselect all Data',
197	'Select all Data 1D',
198	'Unselect all Data 1D',
199	'Select all Data 2D',
200	'Unselect all Data 2D' ]
201	for option in list_of_options:
202	self.selection_cbox.Append(str(option))
203	self.selection_cbox.SetValue('Select all Data')
204	wx.EVT_COMBOBOX(self.selection_cbox,-1, self._on_selection_type)
205	self.sizer5.AddMany([(select_txt,0, wx.ALL,5),
206	(self.selection_cbox,0, wx.ALL,5)])
207	self.enable_selection()
208
209
210	def _on_selection_type(self, event):
211	"""
212	Select data according to patterns
213	"""
214
215	list_of_options = ['Select all Data',
216	'Unselect all Data',
217	'Select all Data 1D',
218	'Unselect all Data 1D',
219	'Select all Data 2D',
220	'Unselect all Data 2D' ]
221	option = self.selection_cbox.GetValue()
222
223	pos = self.selection_cbox.GetSelection()
224	if pos == wx.NOT_FOUND:
225	return
226	option = self.selection_cbox.GetString(pos)
227	for item in self.list_cb_data.values():
228	data_ctrl, _, _, _,_, _ = item
229	data_id, data_class, _ = self.tree_ctrl.GetItemPyData(data_ctrl)
230	if option == 'Select all Data':
231	self.tree_ctrl.CheckItem(data_ctrl, True)
232	elif option == 'Unselect all Data':
233	self.tree_ctrl.CheckItem(data_ctrl, False)
234	elif option == 'Select all Data 1D':
235	if data_class == 'Data1D':
236	self.tree_ctrl.CheckItem(data_ctrl, True)
237	elif option == 'Unselect all Data 1D':
238	if data_class == 'Data1D':
239	self.tree_ctrl.CheckItem(data_ctrl, False)
240	elif option == 'Select all Data 1D':
241	if data_class == 'Data1D':
242	self.tree_ctrl.CheckItem(data_ctrl, True)
243	elif option == 'Select all Data 2D':
244	if data_class == 'Data2D':
245	self.tree_ctrl.CheckItem(data_ctrl, True)
246	elif option == 'Unselect all Data 2D':
247	if data_class == 'Data2D':
248	self.tree_ctrl.CheckItem(data_ctrl, False)
249	self.enable_append()
250	self.enable_freeze()
251	self.enable_plot()
252	self.enable_import()
253	self.enable_remove()
254
255	def layout_button(self):
256	"""
257	Layout widgets related to buttons
258	"""
259	w, _ = self.GetSize()
260
261	self.bt_add = wx.Button(self, wx.NewId(), "Load Data",
262	size=(BUTTON_WIDTH, -1))
263	self.bt_add.SetToolTipString("Load data files")
264	wx.EVT_BUTTON(self, self.bt_add.GetId(), self._load_data)
265	self.bt_remove = wx.Button(self, wx.NewId(), "Delete Data",
266	size=(BUTTON_WIDTH, -1))
267	self.bt_remove.SetToolTipString("Delete data from the application")
268	wx.EVT_BUTTON(self, self.bt_remove.GetId(), self.on_remove)
269	self.bt_import = wx.Button(self, wx.NewId(), "Send To",
270	size=(BUTTON_WIDTH, -1))
271	self.bt_import.SetToolTipString("Send set of Data to active perspective")
272	wx.EVT_BUTTON(self, self.bt_import.GetId(), self.on_import)
273	self.perspective_cbox = wx.ComboBox(self, -1,
274	style=wx.CB_READONLY)
275	#self.perspective_cbox.SetMinSize((CBOX_WIDTH, -1))
276	wx.EVT_COMBOBOX(self.perspective_cbox,-1,
277	self._on_perspective_selection)
278
279	self.bt_append_plot = wx.Button(self, wx.NewId(), "Append Plot To",
280	size=(BUTTON_WIDTH, -1))
281	self.bt_append_plot.SetToolTipString("Plot the selected data in the active panel")
282	wx.EVT_BUTTON(self, self.bt_append_plot.GetId(), self.on_append_plot)
283
284	self.bt_plot = wx.Button(self, wx.NewId(), "New Plot",
285	size=(BUTTON_WIDTH, -1))
286	self.bt_plot.SetToolTipString("To trigger plotting")
287	wx.EVT_BUTTON(self, self.bt_plot.GetId(), self.on_plot)
288
289	self.bt_freeze = wx.Button(self, wx.NewId(), "Freeze Theory",
290	size=(BUTTON_WIDTH, -1))
291	self.bt_freeze.SetToolTipString("To trigger freeze a theory")
292	wx.EVT_BUTTON(self, self.bt_freeze.GetId(), self.on_freeze)
293	#hide plot
294	#self.bt_close_plot = wx.Button(self, wx.NewId(), "Delete Plot",
295	# size=(BUTTON_WIDTH, -1))
296	#self.bt_close_plot.SetToolTipString("Delete the plot panel on focus")
297	#wx.EVT_BUTTON(self, self.bt_close_plot.GetId(), self.on_close_plot)
298
299	self.cb_plotpanel = wx.ComboBox(self, -1,
300	style=wx.CB_READONLY\|wx.CB_SORT)
301	#self.cb_plotpanel.SetMinSize((CBOX_WIDTH, -1))
302	wx.EVT_COMBOBOX(self.cb_plotpanel,-1, self._on_plot_selection)
303	self.cb_plotpanel.Disable()
304
305	self.sizer3.AddMany([(self.bt_add),
306	((10, 10)),
307	(self.bt_remove),
308	((10, 10)),
309	(self.bt_import, 0, wx.EXPAND\|wx.RIGHT, 5),
310	(self.perspective_cbox, wx.EXPAND\|wx.ADJUST_MINSIZE, 5),
311	(self.bt_append_plot),
312	(self.cb_plotpanel, wx.EXPAND\|wx.ADJUST_MINSIZE, 5),
313	(self.bt_plot),
314	((10, 10)),
315	(self.bt_freeze),
316	#((10, 10)),
317	#(self.bt_close_plot),
318	((10, 10))])
319
320	self.sizer3.AddGrowableCol(1, 1)
321	self.show_data_button()
322	self.enable_remove()
323	self.enable_import()
324	self.enable_plot()
325	self.enable_append()
326	self.enable_freeze()
327	self.enable_remove_plot()
328
329	def layout_batch(self):
330	"""
331	"""
332	self.rb_single_mode = wx.RadioButton(self, -1, 'Single Mode',
333	style=wx.RB_GROUP)
334	self.rb_batch_mode = wx.RadioButton(self, -1, 'Batch Mode')
335
336	self.rb_single_mode.SetValue(True)
337	self.rb_batch_mode.SetValue(False)
338	self.sizer4.AddMany([(self.rb_single_mode,0, wx.ALL,5),
339	(self.rb_batch_mode,0, wx.ALL,5)])
340
341	def layout_data_list(self):
342	"""
343	Add a listcrtl in the panel
344	"""
345	tree_ctrl_label = wx.StaticText(self, -1, "Data")
346	tree_ctrl_label.SetForegroundColour('blue')
347	self.tree_ctrl = DataTreeCtrl(parent=self)
348	self.tree_ctrl.Bind(CT.EVT_TREE_ITEM_CHECKING, self.on_check_item)
349	tree_ctrl_theory_label = wx.StaticText(self, -1, "Theory")
350	tree_ctrl_theory_label.SetForegroundColour('blue')
351	self.tree_ctrl_theory = DataTreeCtrl(parent=self)
352	self.tree_ctrl_theory.Bind(CT.EVT_TREE_ITEM_CHECKING, self.on_check_item)
353	self.sizer1.Add(tree_ctrl_label, 0, wx.LEFT, 10)
354	self.sizer1.Add(self.tree_ctrl, 1, wx.EXPAND\|wx.ALL, 10)
355	self.sizer1.Add(tree_ctrl_theory_label, 0, wx.LEFT, 10)
356	self.sizer1.Add(self.tree_ctrl_theory, 1, wx.EXPAND\|wx.ALL, 10)
357
358	def onContextMenu(self, event):
359	"""
360	Retrieve the state selected state
361	"""
362	# Skipping the save state functionality for release 0.9.0
363	#return
364	pos = event.GetPosition()
365	pos = self.ScreenToClient(pos)
366	self.PopupMenu(self.popUpMenu, pos)
367
368
369	def on_check_item(self, event):
370	"""
371	"""
372	item = event.GetItem()
373	item.Check(not item.IsChecked())
374	self.enable_append()
375	self.enable_freeze()
376	self.enable_plot()
377	self.enable_import()
378	self.enable_remove()
379	event.Skip()
380
381	def fill_cbox_analysis(self, plugin):
382	"""
383	fill the combobox with analysis name
384	"""
385	self.list_of_perspective = plugin
386	if self.parent is None or \
387	not hasattr(self.parent, "get_current_perspective") or \
388	len(self.list_of_perspective) == 0:
389	return
390	if self.parent is not None and self.perspective_cbox is not None:
391	for plug in self.list_of_perspective:
392	if plug.get_perspective():
393	self.perspective_cbox.Append(plug.sub_menu, plug)
394
395	curr_pers = self.parent.get_current_perspective()
396	self.perspective_cbox.SetStringSelection(curr_pers.sub_menu)
397	self.enable_import()
398
399	def load_data_list(self, list):
400	"""
401	add need data with its theory under the tree
402	"""
403	if list:
404	for state_id, dstate in list.iteritems():
405	data = dstate.get_data()
406	theory_list = dstate.get_theory()
407	if data is not None:
408	data_name = data.name
409	data_class = data.__class__.__name__
410	path = dstate.get_path()
411	process_list = data.process
412	data_id = data.id
413
414	if state_id not in self.list_cb_data:
415	#new state
416	data_c = self.tree_ctrl.InsertItem(self.tree_ctrl.root,0,
417	data_name, ct_type=1,
418	data=(data_id, data_class, state_id))
419	data_c.Check(True)
420	d_i_c = self.tree_ctrl.AppendItem(data_c, 'Info')
421	i_c_c = self.tree_ctrl.AppendItem(d_i_c,
422	'Type: %s' % data_class)
423	p_c_c = self.tree_ctrl.AppendItem(d_i_c,
424	'Path: %s' % str(path))
425	d_p_c = self.tree_ctrl.AppendItem(d_i_c, 'Process')
426
427	for process in process_list:
428	i_t_c = self.tree_ctrl.AppendItem(d_p_c,
429	process.__str__())
430	theory_child = self.tree_ctrl.AppendItem(data_c, "THEORIES")
431
432	self.list_cb_data[state_id] = [data_c,
433	d_i_c,
434	i_c_c,
435	p_c_c,
436	d_p_c,
437	theory_child]
438	else:
439	data_ctrl_list = self.list_cb_data[state_id]
440	#This state is already display replace it contains
441	data_c, d_i_c, i_c_c, p_c_c, d_p_c, t_c = data_ctrl_list
442	self.tree_ctrl.SetItemText(data_c, data_name)
443	temp = (data_id, data_class, state_id)
444	self.tree_ctrl.SetItemPyData(data_c, temp)
445	self.tree_ctrl.SetItemText(i_c_c, 'Type: %s' % data_class)
446	self.tree_ctrl.SetItemText(p_c_c, 'Path: %s' % str(path))
447	self.tree_ctrl.DeleteChildren(d_p_c)
448	for process in process_list:
449	i_t_c = self.tree_ctrl.AppendItem(d_p_c,
450	process.__str__())
451	self.append_theory(state_id, theory_list)
452	self.enable_remove()
453	self.enable_import()
454	self.enable_plot()
455	self.enable_freeze()
456	self.enable_selection()
457
458	def _uncheck_all(self):
459	"""
460	Uncheck all check boxes
461	"""
462	for item in self.list_cb_data.values():
463	data_ctrl, _, _, _,_, _ = item
464	self.tree_ctrl.CheckItem(data_ctrl, False)
465	self.enable_append()
466	self.enable_freeze()
467	self.enable_plot()
468	self.enable_import()
469	self.enable_remove()
470
471	def append_theory(self, state_id, theory_list):
472	"""
473	append theory object under data from a state of id = state_id
474	replace that theory if already displayed
475	"""
476	if not theory_list:
477	return
478	if state_id not in self.list_cb_data.keys():
479	root = self.tree_ctrl_theory.root
480	tree = self.tree_ctrl_theory
481	else:
482	item = self.list_cb_data[state_id]
483	data_c, _, _, _, _, _ = item
484	root = data_c
485	tree = self.tree_ctrl
486	if root is not None:
487	self.append_theory_helper(tree=tree, root=root,
488	state_id=state_id,
489	theory_list=theory_list)
490
491
492	def append_theory_helper(self, tree, root, state_id, theory_list):
493	"""
494	"""
495	if state_id in self.list_cb_theory.keys():
496	#update current list of theory for this data
497	theory_list_ctrl = self.list_cb_theory[state_id]
498
499	for theory_id, item in theory_list.iteritems():
500	theory_data, theory_state = item
501	if theory_data is None:
502	name = "Unknown"
503	theory_class = "Unknown"
504	theory_id = "Unknown"
505	temp = (None, None, None)
506	else:
507	name = theory_data.name
508	theory_class = theory_data.__class__.__name__
509	theory_id = theory_data.id
510	#if theory_state is not None:
511	# name = theory_state.model.name
512	temp = (theory_id, theory_class, state_id)
513	if theory_id not in theory_list_ctrl:
514	#add new theory
515	t_child = tree.AppendItem(root,
516	name, ct_type=1, data=temp)
517	t_i_c = tree.AppendItem(t_child, 'Info')
518	i_c_c = tree.AppendItem(t_i_c,
519	'Type: %s' % theory_class)
520	t_p_c = tree.AppendItem(t_i_c, 'Process')
521
522	for process in theory_data.process:
523	i_t_c = tree.AppendItem(t_p_c,
524	process.__str__())
525	theory_list_ctrl[theory_id] = [t_child,
526	i_c_c,
527	t_p_c]
528	else:
529	#replace theory
530	t_child, i_c_c, t_p_c = theory_list_ctrl[theory_id]
531	tree.SetItemText(t_child, name)
532	tree.SetItemPyData(t_child, temp)
533	tree.SetItemText(i_c_c, 'Type: %s' % theory_class)
534	tree.DeleteChildren(t_p_c)
535	for process in theory_data.process:
536	i_t_c = tree.AppendItem(t_p_c,
537	process.__str__())
538
539	else:
540	#data didn't have a theory associated it before
541	theory_list_ctrl = {}
542	for theory_id, item in theory_list.iteritems():
543	theory_data, theory_state = item
544	if theory_data is not None:
545	name = theory_data.name
546	theory_class = theory_data.__class__.__name__
547	theory_id = theory_data.id
548	#if theory_state is not None:
549	# name = theory_state.model.name
550	temp = (theory_id, theory_class, state_id)
551	t_child = tree.AppendItem(root,
552	name, ct_type=1,
553	data=(theory_data.id, theory_class, state_id))
554	t_i_c = tree.AppendItem(t_child, 'Info')
555	i_c_c = tree.AppendItem(t_i_c,
556	'Type: %s' % theory_class)
557	t_p_c = tree.AppendItem(t_i_c, 'Process')
558
559	for process in theory_data.process:
560	i_t_c = tree.AppendItem(t_p_c,
561	process.__str__())
562
563	theory_list_ctrl[theory_id] = [t_child, i_c_c, t_p_c]
564	#self.list_cb_theory[data_id] = theory_list_ctrl
565	self.list_cb_theory[state_id] = theory_list_ctrl
566
567
568
569	def set_data_helper(self):
570	"""
571	"""
572	data_to_plot = []
573	state_to_plot = []
574	theory_to_plot = []
575	for value in self.list_cb_data.values():
576	item, _, _, _, _, _ = value
577	if item.IsChecked():
578	data_id, _, state_id = self.tree_ctrl.GetItemPyData(item)
579	data_to_plot.append(data_id)
580	if state_id not in state_to_plot:
581	state_to_plot.append(state_id)
582
583	for theory_dict in self.list_cb_theory.values():
584	for key, value in theory_dict.iteritems():
585	item, _, _ = value
586	if item.IsChecked():
587	theory_id, _, state_id = self.tree_ctrl.GetItemPyData(item)
588	theory_to_plot.append(theory_id)
589	if state_id not in state_to_plot:
590	state_to_plot.append(state_id)
591	return data_to_plot, theory_to_plot, state_to_plot
592
593	def remove_by_id(self, id):
594	"""
595	"""
596	for item in self.list_cb_data.values():
597	data_c, _, _, _, _, theory_child = item
598	data_id, _, state_id = self.tree_ctrl.GetItemPyData(data_c)
599	if id == data_id:
600	self.tree_ctrl.Delete(data_c)
601	del self.list_cb_data[state_id]
602	del self.list_cb_theory[data_id]
603
604	def load_error(self, error=None):
605	"""
606	Pop up an error message.
607
608	:param error: details error message to be displayed
609	"""
610	if error is not None or str(error).strip() != "":
611	dial = wx.MessageDialog(self.parent, str(error), 'Error Loading File',
612	wx.OK \| wx.ICON_EXCLAMATION)
613	dial.ShowModal()
614
615	def _load_data(self, event):
616	"""
617	send an event to the parent to trigger load from plugin module
618	"""
619	if self.parent is not None:
620	wx.PostEvent(self.parent, NewLoadDataEvent())
621
622
623	def on_remove(self, event):
624	"""
625	Get a list of item checked and remove them from the treectrl
626	Ask the parent to remove reference to this item
627	"""
628	data_to_remove, theory_to_remove, _ = self.set_data_helper()
629	data_key = []
630	theory_key = []
631	#remove data from treectrl
632	for d_key, item in self.list_cb_data.iteritems():
633	data_c, d_i_c, i_c_c, p_c_c, d_p_c, t_c = item
634	if data_c.IsChecked():
635	self.tree_ctrl.Delete(data_c)
636	data_key.append(d_key)
637	if d_key in self.list_cb_theory.keys():
638	theory_list_ctrl = self.list_cb_theory[d_key]
639	theory_to_remove += theory_list_ctrl.keys()
640	# Remove theory from treectrl
641	for t_key, theory_dict in self.list_cb_theory.iteritems():
642	for key, value in theory_dict.iteritems():
643	item, _, _ = value
644	if item.IsChecked():
645	try:
646	self.tree_ctrl.Delete(item)
647	except:
648	pass
649	theory_key.append(key)
650
651	#Remove data and related theory references
652	for key in data_key:
653	del self.list_cb_data[key]
654	if key in theory_key:
655	del self.list_cb_theory[key]
656	#remove theory references independently of data
657	for key in theory_key:
658	for t_key, theory_dict in self.list_cb_theory.iteritems():
659	if key in theory_dict:
660	for key, value in theory_dict.iteritems():
661	item, _, _ = value
662	if item.IsChecked():
663	try:
664	self.tree_ctrl_theory.Delete(item)
665	except:
666	pass
667	del theory_dict[key]
668
669
670	self.parent.remove_data(data_id=data_to_remove,
671	theory_id=theory_to_remove)
672	self.enable_remove()
673	self.enable_freeze()
674	self.enable_remove_plot()
675
676	def on_import(self, event=None):
677	"""
678	Get all select data and set them to the current active perspetive
679	"""
680	data_id, theory_id, state_id = self.set_data_helper()
681	temp = data_id + state_id
682	self.parent.set_data(data_id=temp, theory_id=theory_id)
683
684	def on_append_plot(self, event=None):
685	"""
686	append plot to plot panel on focus
687	"""
688	self._on_plot_selection()
689	data_id, theory_id, state_id = self.set_data_helper()
690	self.parent.plot_data(data_id=data_id,
691	state_id=state_id,
692	theory_id=theory_id,
693	append=True)
694
695	def on_plot(self, event=None):
696	"""
697	Send a list of data names to plot
698	"""
699	data_id, theory_id, state_id = self.set_data_helper()
700	self.parent.plot_data(data_id=data_id,
701	state_id=state_id,
702	theory_id=theory_id,
703	append=False)
704	self.enable_remove_plot()
705
706	def on_close_page(self, event=None):
707	"""
708	On close
709	"""
710	if event != None:
711	event.Skip()
712	# send parent to update menu with no show nor hide action
713	self.parent.show_data_panel(action=False)
714
715	def on_freeze(self, event):
716	"""
717	"""
718	_, theory_id, state_id = self.set_data_helper()
719	self.parent.freeze(data_id=state_id, theory_id=theory_id)
720
721	def set_active_perspective(self, name):
722	"""
723	set the active perspective
724	"""
725	self.perspective_cbox.SetStringSelection(name)
726	self.enable_import()
727
728	def _on_delete_plot_panel(self, event):
729	"""
730	get an event with attribute name and caption to delete existing name
731	from the combobox of the current panel
732	"""
733	name = event.name
734	caption = event.caption
735	if self.cb_plotpanel is not None:
736	pos = self.cb_plotpanel.FindString(str(caption))
737	if pos != wx.NOT_FOUND:
738	self.cb_plotpanel.Delete(pos)
739	self.enable_append()
740
741	def set_panel_on_focus(self, name=None):
742	"""
743	set the plot panel on focus
744	"""
745	for key, value in self.parent.plot_panels.iteritems():
746	name_plot_panel = str(value.window_caption)
747	if name_plot_panel not in self.cb_plotpanel.GetItems():
748	self.cb_plotpanel.Append(name_plot_panel, value)
749	if name != None and name == name_plot_panel:
750	self.cb_plotpanel.SetStringSelection(name_plot_panel)
751	break
752	self.enable_append()
753	self.enable_remove_plot()
754
755	def _on_perspective_selection(self, event=None):
756	"""
757	select the current perspective for guiframe
758	"""
759	selection = self.perspective_cbox.GetSelection()
760
761	if self.perspective_cbox.GetValue() != 'None':
762	perspective = self.perspective_cbox.GetClientData(selection)
763	perspective.on_perspective(event=None)
764
765	def _on_plot_selection(self, event=None):
766	"""
767	On source combobox selection
768	"""
769	if event != None:
770	combo = event.GetEventObject()
771	event.Skip()
772	else:
773	combo = self.cb_plotpanel
774	selection = combo.GetSelection()
775
776	if combo.GetValue() != 'None':
777	panel = combo.GetClientData(selection)
778	self.parent.on_set_plot_focus(panel)
779
780	def on_close_plot(self, event):
781	"""
782	clseo the panel on focus
783	"""
784	self.enable_append()
785	selection = self.cb_plotpanel.GetSelection()
786	if self.cb_plotpanel.GetValue() != 'None':
787	panel = self.cb_plotpanel.GetClientData(selection)
788	if self.parent is not None and panel is not None:
789	wx.PostEvent(self.parent,
790	NewPlotEvent(group_id=panel.group_id,
791	action="delete"))
792	self.enable_remove_plot()
793
794	def enable_remove_plot(self):
795	"""
796	enable remove plot button if there is a plot panel on focus
797	"""
798	pass
799	#if self.cb_plotpanel.GetCount() == 0:
800	# self.bt_close_plot.Disable()
801	#else:
802	# self.bt_close_plot.Enable()
803
804	def enable_remove(self):
805	"""
806	enable or disable remove button
807	"""
808	n_t = self.tree_ctrl.GetCount()
809	n_t_t = self.tree_ctrl_theory.GetCount()
810	if n_t + n_t_t <= 0:
811	self.bt_remove.Disable()
812	else:
813	self.bt_remove.Enable()
814
815	def enable_import(self):
816	"""
817	enable or disable send button
818	"""
819	n_t = 0
820	if self.tree_ctrl != None:
821	n_t = self.tree_ctrl.GetCount()
822	if n_t > 0 and len(self.list_of_perspective) > 0:
823	self.bt_import.Enable()
824	else:
825	self.bt_import.Disable()
826	if len(self.list_of_perspective) <= 0 or \
827	self.perspective_cbox.GetValue() in ["None",
828	"No Active Application"]:
829	self.perspective_cbox.Disable()
830	else:
831	self.perspective_cbox.Enable()
832
833	def enable_plot(self):
834	"""
835	enable or disable plot button
836	"""
837	n_t = 0
838	n_t_t = 0
839	if self.tree_ctrl != None:
840	n_t = self.tree_ctrl.GetCount()
841	if self.tree_ctrl_theory != None:
842	n_t_t = self.tree_ctrl_theory.GetCount()
843	if n_t + n_t_t <= 0:
844	self.bt_plot.Disable()
845	else:
846	self.bt_plot.Enable()
847	self.enable_append()
848
849	def enable_append(self):
850	"""
851	enable or disable append button
852	"""
853	n_t = 0
854	n_t_t = 0
855	if self.tree_ctrl != None:
856	n_t = self.tree_ctrl.GetCount()
857	if self.tree_ctrl_theory != None:
858	n_t_t = self.tree_ctrl_theory.GetCount()
859	if n_t + n_t_t <= 0:
860	self.bt_append_plot.Disable()
861	self.cb_plotpanel.Disable()
862	elif self.cb_plotpanel.GetCount() <= 0:
863	self.cb_plotpanel.Disable()
864	self.bt_append_plot.Disable()
865	else:
866	self.bt_append_plot.Enable()
867	self.cb_plotpanel.Enable()
868
869	def check_theory_to_freeze(self):
870	"""
871	"""
872	def enable_freeze(self):
873	"""
874	enable or disable the freeze button
875	"""
876	n_t_t = 0
877	n_l = 0
878	if self.tree_ctrl_theory != None:
879	n_t_t = self.tree_ctrl_theory.GetCount()
880	n_l = len(self.list_cb_theory)
881	if (n_t_t + n_l > 0):
882	self.bt_freeze.Enable()
883	else:
884	self.bt_freeze.Disable()
885
886	def enable_selection(self):
887	"""
888	enable or disable combobo box selection
889	"""
890	n_t = 0
891	n_t_t = 0
892	if self.tree_ctrl != None:
893	n_t = self.tree_ctrl.GetCount()
894	if self.tree_ctrl_theory != None:
895	n_t_t = self.tree_ctrl_theory.GetCount()
896	if n_t + n_t_t > 0 and self.selection_cbox != None:
897	self.selection_cbox.Enable()
898	else:
899	self.selection_cbox.Disable()
900
901	def show_data_button(self):
902	"""
903	show load data and remove data button if
904	dataloader on else hide them
905	"""
906	try:
907	gui_style = self.parent.get_style()
908	style = gui_style & GUIFRAME.DATALOADER_ON
909	if style == GUIFRAME.DATALOADER_ON:
910	#self.bt_remove.Show(True)
911	self.bt_add.Show(True)
912	else:
913	#self.bt_remove.Hide()
914	self.bt_add.Hide()
915	except:
916	#self.bt_remove.Hide()
917	self.bt_add.Hide()
918
919
920
921	WIDTH = 400
922	HEIGHT = 300
923
924
925	class DataDialog(wx.Dialog):
926	"""
927	Allow file selection at loading time
928	"""
929	def __init__(self, data_list, parent=None, text='', args, *kwds):
930	wx.Dialog.__init__(self, parent, args, *kwds)
931	self.SetTitle("Data Selection")
932	self.SetSize((WIDTH, HEIGHT))
933	self.list_of_ctrl = []
934	if not data_list:
935	return
936	self._sizer_main = wx.BoxSizer(wx.VERTICAL)
937	self._sizer_txt = wx.BoxSizer(wx.VERTICAL)
938	self._sizer_button = wx.BoxSizer(wx.HORIZONTAL)
939	self.sizer = wx.GridBagSizer(5, 5)
940	self._panel = ScrolledPanel(self, style=wx.RAISED_BORDER,
941	size=(WIDTH-20, HEIGHT-50))
942	self._panel.SetupScrolling()
943	self.__do_layout(data_list, text=text)
944
945	def __do_layout(self, data_list, text=''):
946	"""
947	layout the dialog
948	"""
949	if not data_list or len(data_list) <= 1:
950	return
951	#add text
952
953	text = "Deleting these file reset some panels.\n"
954	text += "Do you want to proceed?\n"
955	text_ctrl = wx.StaticText(self, -1, str(text))
956	self._sizer_txt.Add(text_ctrl)
957	iy = 0
958	ix = 0
959	data_count = 0
960	for (data_name, in_use, sub_menu) in range(len(data_list)):
961	if in_use == True:
962	ctrl_name = wx.StaticBox(self, -1, str(data_name))
963	ctrl_in_use = wx.StaticBox(self, -1, " is used by ")
964	plug_name = str(sub_menu) + "\n"
965	ctrl_sub_menu = wx.StaticBox(self, -1, plug_name)
966	self.sizer.Add(ctrl_name, (iy, ix),
967	(1, 1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
968	ix += 1
969	self._sizer_button.Add(ctrl_in_use, 1,
970	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
971	ix += 1
972	self._sizer_button.Add(plug_name, 1,
973	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
974	iy += 1
975	self._panel.SetSizer(self.sizer)
976	#add sizer
977	self._sizer_button.Add((20, 20), 1, wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
978	button_cancel = wx.Button(self, wx.ID_CANCEL, "Cancel")
979	self._sizer_button.Add(button_cancel, 0,
980	wx.LEFT\|wx.RIGHT\|wx.ADJUST_MINSIZE, 10)
981	button_OK = wx.Button(self, wx.ID_OK, "Ok")
982	button_OK.SetFocus()
983	self._sizer_button.Add(button_OK, 0,
984	wx.LEFT\|wx.RIGHT\|wx.ADJUST_MINSIZE, 10)
985	static_line = wx.StaticLine(self, -1)
986
987	self._sizer_txt.Add(self._panel, 1, wx.EXPAND\|wx.LEFT\|wx.RIGHT, 5)
988	self._sizer_main.Add(self._sizer_txt, 1, wx.EXPAND\|wx.ALL, 10)
989	self._sizer_main.Add(self._data_text_ctrl, 0,
990	wx.EXPAND\|wx.LEFT\|wx.RIGHT, 10)
991	self._sizer_main.Add(static_line, 0, wx.EXPAND, 0)
992	self._sizer_main.Add(self._sizer_button, 0, wx.EXPAND\|wx.ALL, 10)
993	self.SetSizer(self._sizer_main)
994	self.Layout()
995
996	def get_data(self):
997	"""
998	return the selected data
999	"""
1000	temp = []
1001	for item in self.list_of_ctrl:
1002	cb, data = item
1003	if cb.GetValue():
1004	temp.append(data)
1005	return temp
1006
1007	def _count_selected_data(self, event):
1008	"""
1009	count selected data
1010	"""
1011	if event.GetEventObject().GetValue():
1012	self._nb_selected_data += 1
1013	else:
1014	self._nb_selected_data -= 1
1015	select_data_text = " %s Data selected.\n" % str(self._nb_selected_data)
1016	self._data_text_ctrl.SetLabel(select_data_text)
1017	if self._nb_selected_data <= self._max_data:
1018	self._data_text_ctrl.SetForegroundColour('blue')
1019	else:
1020	self._data_text_ctrl.SetForegroundColour('red')
1021
1022
1023
1024	class DataFrame(wx.Frame):
1025	## Internal name for the AUI manager
1026	window_name = "Data Panel"
1027	## Title to appear on top of the window
1028	window_caption = "Data Panel"
1029	## Flag to tell the GUI manager that this panel is not
1030	# tied to any perspective
1031	ALWAYS_ON = True
1032
1033	def __init__(self, parent=None, owner=None, manager=None,size=(300, 800),
1034	list_of_perspective=[],list=[], args, *kwds):
1035	kwds['size'] = size
1036	kwds['id'] = -1
1037	kwds['title']= "Loaded Data"
1038	wx.Frame.__init__(self, parent=parent, args, *kwds)
1039	self.parent = parent
1040	self.owner = owner
1041	self.manager = manager
1042	self.panel = DataPanel(parent=self,
1043	#size=size,
1044	list_of_perspective=list_of_perspective)
1045
1046	def load_data_list(self, list=[]):
1047	"""
1048	Fill the list inside its panel
1049	"""
1050	self.panel.load_data_list(list=list)
1051
1052
1053
1054	from dataFitting import Data1D
1055	from dataFitting import Data2D, Theory1D
1056	from data_state import DataState
1057	import sys
1058	class State():
1059	def __init__(self):
1060	self.msg = ""
1061	def __str__(self):
1062	self.msg = "model mane : model1\n"
1063	self.msg += "params : \n"
1064	self.msg += "name value\n"
1065	return msg
1066	def set_data_state(data=None, path=None, theory=None, state=None):
1067	dstate = DataState(data=data)
1068	dstate.set_path(path=path)
1069	dstate.set_theory(theory, state)
1070
1071	return dstate
1072	"""'
1073	data_list = [1:('Data1', 'Data1D', '07/01/2010', "theory1d", "state1"),
1074	('Data2', 'Data2D', '07/03/2011', "theory2d", "state1"),
1075	('Data3', 'Theory1D', '06/01/2010', "theory1d", "state1"),
1076	('Data4', 'Theory2D', '07/01/2010', "theory2d", "state1"),
1077	('Data5', 'Theory2D', '07/02/2010', "theory2d", "state1")]
1078	"""
1079	if __name__ == "__main__":
1080
1081	app = wx.App()
1082	try:
1083	list_of_perspective = [('perspective2', False), ('perspective1', True)]
1084	data_list = {}
1085	# state 1
1086	data = Data2D()
1087	data.name = "data2"
1088	data.id = 1
1089	data.append_empty_process()
1090	process = data.process[len(data.process)-1]
1091	process.data = "07/01/2010"
1092	theory = Data2D()
1093	theory.id = 34
1094	theory.name = "theory1"
1095	path = "path1"
1096	state = State()
1097	data_list['1']=set_data_state(data, path,theory, state)
1098	#state 2
1099	data = Data2D()
1100	data.name = "data2"
1101	data.id = 76
1102	theory = Data2D()
1103	theory.id = 78
1104	theory.name = "CoreShell 07/24/25"
1105	path = "path2"
1106	#state3
1107	state = State()
1108	data_list['2']=set_data_state(data, path,theory, state)
1109	data = Data1D()
1110	data.id = 3
1111	data.name = "data2"
1112	theory = Theory1D()
1113	theory.name = "CoreShell"
1114	theory.id = 4
1115	theory.append_empty_process()
1116	process = theory.process[len(theory.process)-1]
1117	process.description = "this is my description"
1118	path = "path3"
1119	data.append_empty_process()
1120	process = data.process[len(data.process)-1]
1121	process.data = "07/22/2010"
1122	data_list['4']=set_data_state(data, path,theory, state)
1123	#state 4
1124	temp_data_list = {}
1125	data.name = "data5 erasing data2"
1126	temp_data_list['4'] = set_data_state(data, path,theory, state)
1127	#state 5
1128	data = Data2D()
1129	data.name = "data3"
1130	data.id = 5
1131	data.append_empty_process()
1132	process = data.process[len(data.process)-1]
1133	process.data = "07/01/2010"
1134	theory = Theory1D()
1135	theory.name = "Cylinder"
1136	path = "path2"
1137	state = State()
1138	dstate= set_data_state(data, path,theory, state)
1139	theory = Theory1D()
1140	theory.id = 6
1141	theory.name = "CoreShell"
1142	dstate.set_theory(theory)
1143	theory = Theory1D()
1144	theory.id = 6
1145	theory.name = "CoreShell replacing coreshell in data3"
1146	dstate.set_theory(theory)
1147	data_list['3'] = dstate
1148	#state 6
1149	data_list['6']=set_data_state(None, path,theory, state)
1150	data_list['6']=set_data_state(theory=theory, state=None)
1151	theory = Theory1D()
1152	theory.id = 7
1153	data_list['6']=set_data_state(theory=theory, state=None)
1154	data_list['7']=set_data_state(theory=theory, state=None)
1155	window = DataFrame(list=data_list)
1156	window.load_data_list(list=data_list)
1157	window.Show(True)
1158	window.load_data_list(list=temp_data_list)
1159	except:
1160	#raise
1161	print "error",sys.exc_value
1162
1163	app.MainLoop()
1164
1165

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

SasView

source: sasview/guiframe/data_panel.py @ af7ba61

Download in other formats: