data_panel.py @ ba64348

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 ba64348 was 1b1bbf9, checked in by Jae Cho <jhjcho@…>, 14 years ago
focus color (plots) and adjust panel sizes a bit
Property mode set to `100644`
File size: 32.8 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 wx
14	import sys
15	import warnings
16	from wx.lib.scrolledpanel import ScrolledPanel
17	import wx.lib.agw.customtreectrl as CT
18	from sans.guiframe.dataFitting import Data1D
19	from sans.guiframe.dataFitting import Data2D
20	from sans.guiframe.panel_base import PanelBase
21
22	PANEL_WIDTH = 180
23	#PANEL_HEIGHT = 560
24	PANEL_HEIGHT = 700
25	STYLE_FLAG =wx.RAISED_BORDER\|CT.TR_HAS_BUTTONS\| CT.TR_HIDE_ROOT\|\
26	wx.WANTS_CHARS\|CT.TR_HAS_VARIABLE_ROW_HEIGHT
27
28
29	class DataTreeCtrl(CT.CustomTreeCtrl):
30	"""
31	Check list control to be used for Data Panel
32	"""
33	def __init__(self, parent,args, *kwds):
34	#agwstyle is introduced in wx.2.8.11 but is not working for mac
35	if sys.platform.count("darwin") != 0:
36	try:
37	kwds['style'] = STYLE_FLAG
38	except:
39	raise
40	else:
41	#agwstyle is introduced in wx.2.8.11 .argument working only for windows
42	try:
43	kwds['agwStyle'] = STYLE_FLAG
44	except:
45	try:
46	kwds['style'] = STYLE_FLAG
47	except:
48	raise
49
50	CT.CustomTreeCtrl.__init__(self, parent, args, *kwds)
51	self.root = self.AddRoot("Available Data")
52
53	class DataPanel(ScrolledPanel, PanelBase):
54	"""
55	This panel displays data available in the application and widgets to
56	interact with data.
57	"""
58	## Internal name for the AUI manager
59	window_name = "Data Panel"
60	## Title to appear on top of the window
61	window_caption = "Data Explorer"
62	#type of window
63	window_type = "Data Panel"
64	## Flag to tell the GUI manager that this panel is not
65	# tied to any perspective
66	#ALWAYS_ON = True
67	def __init__(self, parent, list=[],list_of_perspective=[],
68	size=(PANEL_WIDTH,PANEL_HEIGHT), manager=None, args, *kwds):
69	kwds['size']= size
70	kwds['style'] = STYLE_FLAG
71	ScrolledPanel.__init__(self, parent=parent, args, *kwds)
72	PanelBase.__init__(self)
73	self.SetupScrolling()
74	self.all_data1d = True
75	self.parent = parent
76	self.manager = manager
77	self.list_of_data = list
78	self.list_of_perspective = list_of_perspective
79	self.list_rb_perspectives= []
80	self.list_cb_data = {}
81	self.list_cb_theory = {}
82
83	self.owner = None
84	self.do_layout()
85	self.Bind(wx.EVT_SHOW, self.on_close_page)
86
87	def do_layout(self):
88	"""
89	"""
90	self.define_panel_structure()
91	self.layout_selection()
92	self.layout_data_list()
93	self.layout_button()
94	self.layout_batch()
95
96	def define_panel_structure(self):
97	"""
98	Define the skeleton of the panel
99	"""
100	w, h = self.parent.GetSize()
101	self.vbox = wx.BoxSizer(wx.VERTICAL)
102	self.sizer1 = wx.BoxSizer(wx.VERTICAL)
103	self.sizer1.SetMinSize((w/13, h*2/5))
104
105	self.sizer2 = wx.BoxSizer(wx.VERTICAL)
106	self.sizer3 = wx.GridBagSizer(5,5)
107	self.sizer4 = wx.BoxSizer(wx.HORIZONTAL)
108	self.sizer5 = wx.BoxSizer(wx.VERTICAL)
109
110	self.vbox.Add(self.sizer5, 0,wx.EXPAND\|wx.ALL,10)
111	self.vbox.Add(self.sizer1, 0,wx.EXPAND\|wx.ALL,0)
112	self.vbox.Add(self.sizer2, 0,wx.EXPAND\|wx.ALL,10)
113	self.vbox.Add(self.sizer3, 0,wx.EXPAND\|wx.ALL,10)
114	self.vbox.Add(self.sizer4, 0,wx.EXPAND\|wx.ALL,10)
115
116	self.SetSizer(self.vbox)
117
118	def layout_selection(self):
119	"""
120	"""
121	select_txt = wx.StaticText(self, -1, 'Selection Options')
122	select_txt.SetForegroundColour('blue')
123	self.selection_cbox = wx.ComboBox(self, -1, style=wx.CB_READONLY)
124	list_of_options = ['Select all Data',
125	'Unselect all Data',
126	'Select all Data 1D',
127	'Unselect all Data 1D',
128	'Select all Data 2D',
129	'Unselect all Data 2D' ]
130	for option in list_of_options:
131	self.selection_cbox.Append(str(option))
132	self.selection_cbox.SetValue('Select all Data')
133	wx.EVT_COMBOBOX(self.selection_cbox,-1, self._on_selection_type)
134	self.sizer5.AddMany([(select_txt,0, wx.ALL,5),
135	(self.selection_cbox,0, wx.ALL,5)])
136	def layout_perspective(self, list_of_perspective=[]):
137	"""
138	Layout widgets related to the list of plug-ins of the gui_manager
139	"""
140	if len(list_of_perspective)==0:
141	return
142	w, h = self.parent.GetSize()
143	box_description_2= wx.StaticBox(self, -1, "Set Active Perspective")
144	self.boxsizer_2 = wx.StaticBoxSizer(box_description_2, wx.HORIZONTAL)
145	self.sizer_perspective = wx.GridBagSizer(5,5)
146	self.boxsizer_2.Add(self.sizer_perspective)
147	self.sizer2.Add(self.boxsizer_2,1, wx.ALL, 10)
148	self.list_of_perspective = list_of_perspective
149	self.sizer_perspective.Clear(True)
150	self.list_rb_perspectives = []
151
152	nb_active_perspective = 0
153	if list_of_perspective:
154	ix = 0
155	iy = 0
156	for perspective_name, is_active in list_of_perspective:
157
158	if is_active:
159	nb_active_perspective += 1
160	if nb_active_perspective == 1:
161	rb = wx.RadioButton(self, -1, perspective_name,
162	style=wx.RB_GROUP)
163	rb.SetToolTipString("Data will be applied to this perspective")
164	rb.SetValue(is_active)
165	else:
166	rb = wx.RadioButton(self, -1, perspective_name)
167	rb.SetToolTipString("Data will be applied to this perspective")
168	#only one perpesctive can be active
169	rb.SetValue(False)
170
171	self.Bind(wx.EVT_RADIOBUTTON, self.on_set_active_perspective,
172	id=rb.GetId())
173	self.list_rb_perspectives.append(rb)
174	self.sizer_perspective.Add(rb,(iy, ix),(1,1),
175	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 10)
176	iy += 1
177	else:
178	rb = wx.RadioButton(self, -1, 'No Perspective',
179	style=wx.RB_GROUP)
180	rb.SetValue(True)
181	rb.Disable()
182	ix = 0
183	iy = 0
184	self.sizer_perspective.Add(rb,(iy, ix),(1,1),
185	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 10)
186
187	def _on_selection_type(self, event):
188	"""
189	Select data according to patterns
190	"""
191
192	list_of_options = ['Select all Data',
193	'Unselect all Data',
194	'Select all Data 1D',
195	'Unselect all Data 1D',
196	'Select all Data 2D',
197	'Unselect all Data 2D' ]
198	option = self.selection_cbox.GetValue()
199
200	pos = self.selection_cbox.GetSelection()
201	if pos == wx.NOT_FOUND:
202	return
203	option = self.selection_cbox.GetString(pos)
204	for item in self.list_cb_data.values():
205	data_ctrl, _, _, _,_, _ = item
206	data_id, data_class, _ = self.tree_ctrl.GetItemPyData(data_ctrl)
207	if option == 'Select all Data':
208	self.tree_ctrl.CheckItem(data_ctrl, True)
209	elif option == 'Unselect all Data':
210	self.tree_ctrl.CheckItem(data_ctrl, False)
211	elif option == 'Select all Data 1D':
212	if data_class == 'Data1D':
213	self.tree_ctrl.CheckItem(data_ctrl, True)
214	elif option == 'Unselect all Data 1D':
215	if data_class == 'Data1D':
216	self.tree_ctrl.CheckItem(data_ctrl, False)
217	elif option == 'Select all Data 1D':
218	if data_class == 'Data1D':
219	self.tree_ctrl.CheckItem(data_ctrl, True)
220	elif option == 'Select all Data 2D':
221	if data_class == 'Data2D':
222	self.tree_ctrl.CheckItem(data_ctrl, True)
223	elif option == 'Unselect all Data 2D':
224	if data_class == 'Data2D':
225	self.tree_ctrl.CheckItem(data_ctrl, False)
226
227	def on_set_active_perspective(self, event):
228	"""
229	Select the active perspective
230	"""
231	ctrl = event.GetEventObject()
232
233	def layout_button(self):
234	"""
235	Layout widgets related to buttons
236	"""
237	self.bt_import = wx.Button(self, wx.NewId(), "Send To")
238	self.bt_import.SetToolTipString("Send set of Data to active perspective")
239	wx.EVT_BUTTON(self, self.bt_import.GetId(), self.on_import)
240
241	self.bt_append_plot = wx.Button(self, wx.NewId(), "Append Plot To")
242	self.bt_append_plot.SetToolTipString("Plot the selected data in the active panel")
243	wx.EVT_BUTTON(self, self.bt_append_plot.GetId(), self.on_append_plot)
244
245	self.bt_plot = wx.Button(self, wx.NewId(), "New Plot")
246	self.bt_plot.SetToolTipString("To trigger plotting")
247	wx.EVT_BUTTON(self, self.bt_plot.GetId(), self.on_plot)
248
249	self.bt_freeze = wx.Button(self, wx.NewId(), "Freeze")
250	self.bt_freeze.SetToolTipString("To trigger freeze")
251	wx.EVT_BUTTON(self, self.bt_freeze.GetId(), self.on_freeze)
252
253	self.bt_remove = wx.Button(self, wx.NewId(), "Remove Data")
254	self.bt_remove.SetToolTipString("Remove data from the application")
255	wx.EVT_BUTTON(self, self.bt_remove.GetId(), self.on_remove)
256
257	self.tctrl_perspective = wx.StaticText(self, -1, 'No Active Application')
258	self.tctrl_perspective.SetToolTipString("Active Application")
259	self.tctrl_plotpanel = wx.StaticText(self, -1, 'No Plot panel on focus')
260	self.tctrl_plotpanel.SetToolTipString("Active Plot Panel")
261
262	ix = 0
263	iy = 0
264	self.sizer3.Add(self.bt_import,( iy, ix),(1,1),
265	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
266	ix += 1
267	self.sizer3.Add(self.tctrl_perspective,(iy, ix),(1,1),
268	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
269	ix = 0
270	iy += 1
271	self.sizer3.Add(self.bt_append_plot,( iy, ix),(1,1),
272	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
273	ix += 1
274	self.sizer3.Add(self.tctrl_plotpanel,(iy, ix),(1,1),
275	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
276	ix = 0
277	iy += 1
278	self.sizer3.Add(self.bt_plot,( iy, ix),(1,1),
279	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
280	ix = 0
281	iy += 1
282	self.sizer3.Add(self.bt_freeze,( iy, ix),(1,1),
283	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
284	ix = 0
285	iy += 1
286	self.sizer3.Add(self.bt_remove,( iy, ix),(1,1),
287	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
288
289
290
291	def layout_batch(self):
292	"""
293	"""
294
295	self.rb_single_mode = wx.RadioButton(self, -1, 'Single Mode',
296	style=wx.RB_GROUP)
297	self.rb_batch_mode = wx.RadioButton(self, -1, 'Batch Mode')
298
299	self.rb_single_mode.SetValue(True)
300	self.rb_batch_mode.SetValue(False)
301	self.sizer4.AddMany([(self.rb_single_mode,0, wx.ALL,5),
302	(self.rb_batch_mode,0, wx.ALL,5)])
303
304	def old_layout_data_list(self):
305	"""
306	Add a listcrtl in the panel
307	"""
308	self.tree_ctrl = DataTreeCtrl(parent=self)
309	self.tree_ctrl.Bind(CT.EVT_TREE_ITEM_CHECKING, self.on_check_item)
310	self.sizer1.Add(self.tree_ctrl,1, wx.EXPAND\|wx.ALL, 10)
311	self.theory_root = self.tree_ctrl.InsertItem(self.tree_ctrl.root,0,
312	"THEORIES", ct_type=0)
313
314	def layout_data_list(self):
315	"""
316	Add a listcrtl in the panel
317	"""
318	tree_ctrl_label = wx.StaticText(self, -1, "Data")
319	tree_ctrl_label.SetForegroundColour('blue')
320	self.tree_ctrl = DataTreeCtrl(parent=self)
321	self.tree_ctrl.Bind(CT.EVT_TREE_ITEM_CHECKING, self.on_check_item)
322	tree_ctrl_theory_label = wx.StaticText(self, -1, "Theory")
323	tree_ctrl_theory_label.SetForegroundColour('blue')
324	self.tree_ctrl_theory = DataTreeCtrl(parent=self)
325	self.tree_ctrl_theory.Bind(CT.EVT_TREE_ITEM_CHECKING, self.on_check_item)
326	self.sizer1.Add(tree_ctrl_label, 0, wx.LEFT, 10)
327	self.sizer1.Add(self.tree_ctrl, 1, wx.EXPAND\|wx.ALL, 10)
328	self.sizer1.Add(tree_ctrl_theory_label, 0, wx.LEFT, 10)
329	self.sizer1.Add(self.tree_ctrl_theory, 1, wx.EXPAND\|wx.ALL, 10)
330
331	def onContextMenu(self, event):
332	"""
333	Retrieve the state selected state
334	"""
335	# Skipping the save state functionality for release 0.9.0
336	#return
337	pos = event.GetPosition()
338	pos = self.ScreenToClient(pos)
339	self.PopupMenu(self.popUpMenu, pos)
340
341
342	def on_check_item(self, event):
343	"""
344	"""
345	item = event.GetItem()
346	item.Check(not item.IsChecked())
347	self.enable_button(item)
348	event.Skip()
349
350	def enable_button(self, item):
351	"""
352	"""
353	_, data_class, _= self.tree_ctrl.GetItemPyData(item)
354	if item.IsChecked():
355	self.all_data1d &= (data_class != "Data2D")
356	if self.all_data1d:
357	self.bt_freeze.Enable()
358	else:
359	self.bt_freeze.Disable()
360	else:
361	self.all_data1d \|= True
362	self.all_data1d &= (data_class != "Data2D")
363	if self.all_data1d:
364	self.bt_freeze.Enable()
365	else:
366	self.bt_freeze.Disable()
367
368	def load_data_list(self, list):
369	"""
370	add need data with its theory under the tree
371	"""
372	if not list:
373	return
374
375	for state_id, dstate in list.iteritems():
376	data = dstate.get_data()
377	theory_list = dstate.get_theory()
378	if data is not None:
379	data_name = data.name
380	data_class = data.__class__.__name__
381	path = dstate.get_path()
382	process_list = data.process
383	data_id = data.id
384
385	if state_id not in self.list_cb_data:
386	#new state
387	data_c = self.tree_ctrl.InsertItem(self.tree_ctrl.root,0,
388	data_name, ct_type=1,
389	data=(data_id, data_class, state_id))
390	data_c.Check(True)
391	self.enable_button(data_c)
392	d_i_c = self.tree_ctrl.AppendItem(data_c, 'Info')
393	i_c_c = self.tree_ctrl.AppendItem(d_i_c,
394	'Type: %s' % data_class)
395	p_c_c = self.tree_ctrl.AppendItem(d_i_c,
396	'Path: %s' % str(path))
397	d_p_c = self.tree_ctrl.AppendItem(d_i_c, 'Process')
398
399	for process in process_list:
400	i_t_c = self.tree_ctrl.AppendItem(d_p_c,
401	process.__str__())
402	theory_child = self.tree_ctrl.AppendItem(data_c, "THEORIES")
403
404	self.list_cb_data[state_id] = [data_c,
405	d_i_c,
406	i_c_c,
407	p_c_c,
408	d_p_c,
409	theory_child]
410	else:
411	data_ctrl_list = self.list_cb_data[state_id]
412	#This state is already display replace it contains
413	data_c, d_i_c, i_c_c, p_c_c, d_p_c, t_c = data_ctrl_list
414	self.tree_ctrl.SetItemText(data_c, data_name)
415	temp = (data_id, data_class, state_id)
416	self.tree_ctrl.SetItemPyData(data_c, temp)
417	self.tree_ctrl.SetItemText(i_c_c, 'Type: %s' % data_class)
418	self.tree_ctrl.SetItemText(p_c_c, 'Path: %s' % str(path))
419	self.tree_ctrl.DeleteChildren(d_p_c)
420	for process in process_list:
421	i_t_c = self.tree_ctrl.AppendItem(d_p_c,
422	process.__str__())
423	self.append_theory(state_id, theory_list)
424
425
426	def old_append_theory(self, state_id, theory_list):
427	"""
428	append theory object under data from a state of id = state_id
429	replace that theory if already displayed
430	"""
431	if not theory_list:
432	return
433	if state_id not in self.list_cb_data.keys():
434	root = self.theory_root
435	else:
436	item = self.list_cb_data[state_id]
437	data_c, _, _, _, _, _ = item
438	root = data_c
439	if root is not None:
440	self.append_theory_helper(root=root,
441	state_id=state_id,
442	theory_list=theory_list)
443	def append_theory(self, state_id, theory_list):
444	"""
445	append theory object under data from a state of id = state_id
446	replace that theory if already displayed
447	"""
448	if not theory_list:
449	return
450	if state_id not in self.list_cb_data.keys():
451	root = self.tree_ctrl_theory.root
452	tree = self.tree_ctrl_theory
453	else:
454	item = self.list_cb_data[state_id]
455	data_c, _, _, _, _, _ = item
456	root = data_c
457	tree = self.tree_ctrl
458	if root is not None:
459	self.append_theory_helper(tree=tree, root=root,
460	state_id=state_id,
461	theory_list=theory_list)
462
463
464	def append_theory_helper(self, tree, root, state_id, theory_list):
465	"""
466	"""
467	if state_id in self.list_cb_theory.keys():
468	#update current list of theory for this data
469	theory_list_ctrl = self.list_cb_theory[state_id]
470
471	for theory_id, item in theory_list.iteritems():
472	theory_data, theory_state = item
473	if theory_data is None:
474	name = "Unknown"
475	theory_class = "Unknown"
476	theory_id = "Unknown"
477	temp = (None, None, None)
478	else:
479	name = theory_data.name
480	theory_class = theory_data.__class__.__name__
481	theory_id = theory_data.id
482	#if theory_state is not None:
483	# name = theory_state.model.name
484	temp = (theory_id, theory_class, state_id)
485	if theory_id not in theory_list_ctrl:
486	#add new theory
487	t_child = tree.AppendItem(root,
488	name, ct_type=1, data=temp)
489	t_i_c = tree.AppendItem(t_child, 'Info')
490	i_c_c = tree.AppendItem(t_i_c,
491	'Type: %s' % theory_class)
492	t_p_c = tree.AppendItem(t_i_c, 'Process')
493
494	for process in theory_data.process:
495	i_t_c = tree.AppendItem(t_p_c,
496	process.__str__())
497	theory_list_ctrl[theory_id] = [t_child,
498	i_c_c,
499	t_p_c]
500	else:
501	#replace theory
502	t_child, i_c_c, t_p_c = theory_list_ctrl[theory_id]
503	tree.SetItemText(t_child, name)
504	tree.SetItemPyData(t_child, temp)
505	tree.SetItemText(i_c_c, 'Type: %s' % theory_class)
506	tree.DeleteChildren(t_p_c)
507	for process in theory_data.process:
508	i_t_c = tree.AppendItem(t_p_c,
509	process.__str__())
510
511	else:
512	#data didn't have a theory associated it before
513	theory_list_ctrl = {}
514	for theory_id, item in theory_list.iteritems():
515	theory_data, theory_state = item
516	if theory_data is not None:
517	name = theory_data.name
518	theory_class = theory_data.__class__.__name__
519	theory_id = theory_data.id
520	#if theory_state is not None:
521	# name = theory_state.model.name
522	temp = (theory_id, theory_class, state_id)
523	t_child = tree.AppendItem(root,
524	name, ct_type=1,
525	data=(theory_data.id, theory_class, state_id))
526	t_i_c = tree.AppendItem(t_child, 'Info')
527	i_c_c = tree.AppendItem(t_i_c,
528	'Type: %s' % theory_class)
529	t_p_c = tree.AppendItem(t_i_c, 'Process')
530
531	for process in theory_data.process:
532	i_t_c = tree.AppendItem(t_p_c,
533	process.__str__())
534
535	theory_list_ctrl[theory_id] = [t_child, i_c_c, t_p_c]
536	#self.list_cb_theory[data_id] = theory_list_ctrl
537	self.list_cb_theory[state_id] = theory_list_ctrl
538
539
540
541	def set_data_helper(self):
542	"""
543	"""
544	data_to_plot = []
545	state_to_plot = []
546	theory_to_plot = []
547	for value in self.list_cb_data.values():
548	item, _, _, _, _, _ = value
549	if item.IsChecked():
550	data_id, _, state_id = self.tree_ctrl.GetItemPyData(item)
551	data_to_plot.append(data_id)
552	if state_id not in state_to_plot:
553	state_to_plot.append(state_id)
554
555	for theory_dict in self.list_cb_theory.values():
556	for key, value in theory_dict.iteritems():
557	item, _, _ = value
558	if item.IsChecked():
559	theory_id, _, state_id = self.tree_ctrl.GetItemPyData(item)
560	theory_to_plot.append(theory_id)
561	if state_id not in state_to_plot:
562	state_to_plot.append(state_id)
563	return data_to_plot, theory_to_plot, state_to_plot
564
565	def remove_by_id(self, id):
566	"""
567	"""
568	for item in self.list_cb_data.values():
569	data_c, _, _, _, _, theory_child = item
570	data_id, _, state_id = self.tree_ctrl.GetItemPyData(data_c)
571	if id == data_id:
572	self.tree_ctrl.Delete(data_c)
573	del self.list_cb_data[state_id]
574	del self.list_cb_theory[data_id]
575
576	def on_remove(self, event):
577	"""
578	Get a list of item checked and remove them from the treectrl
579	Ask the parent to remove reference to this item
580	"""
581	data_to_remove, theory_to_remove, _ = self.set_data_helper()
582	data_key = []
583	theory_key = []
584	#remove data from treectrl
585	for d_key, item in self.list_cb_data.iteritems():
586	data_c, d_i_c, i_c_c, p_c_c, d_p_c, t_c = item
587	if data_c.IsChecked():
588	self.tree_ctrl.Delete(data_c)
589	data_key.append(d_key)
590	if d_key in self.list_cb_theory.keys():
591	theory_list_ctrl = self.list_cb_theory[d_key]
592	theory_to_remove += theory_list_ctrl.keys()
593	# Remove theory from treectrl
594	for t_key, theory_dict in self.list_cb_theory.iteritems():
595	for key, value in theory_dict.iteritems():
596	item, _, _ = value
597	if item.IsChecked():
598	self.tree_ctrl.Delete(item)
599	theory_key.append(key)
600	#Remove data and related theory references
601	for key in data_key:
602	del self.list_cb_data[key]
603	if key in theory_key:
604	del self.list_cb_theory[key]
605	#remove theory references independently of data
606	for key in theory_key:
607	for t_key, theory_dict in self.list_cb_theory.iteritems():
608	if key in theory_dict:
609	del theory_dict[key]
610
611	self.parent.remove_data(data_id=data_to_remove,
612	theory_id=theory_to_remove)
613
614	def on_import(self, event=None):
615	"""
616	Get all select data and set them to the current active perspetive
617	"""
618	data_id, theory_id, state_id = self.set_data_helper()
619	self.parent.set_data(data_id)
620	self.parent.set_data(data_id=state_id, theory_id=theory_id)
621
622	def on_append_plot(self, event=None):
623	"""
624	append plot to plot panel on focus
625	"""
626	data_id, theory_id, state_id = self.set_data_helper()
627	self.parent.plot_data(data_id=data_id,
628	state_id=state_id,
629	theory_id=theory_id,
630	append=True)
631
632	def on_plot(self, event=None):
633	"""
634	Send a list of data names to plot
635	"""
636	data_id, theory_id, state_id = self.set_data_helper()
637	self.parent.plot_data(data_id=data_id,
638	state_id=state_id,
639	theory_id=theory_id,
640	append=False)
641
642	def on_close_page(self, event=None):
643	"""
644	On close
645	"""
646	if event != None:
647	event.Skip()
648
649	# send parent to update menu with no show nor hide action
650	self.parent.show_data_panel(action=False)
651
652
653	def on_freeze(self, event):
654	"""
655	"""
656	_, theory_id, state_id = self.set_data_helper()
657	self.parent.freeze(data_id=state_id, theory_id=theory_id)
658
659	def set_active_perspective(self, name):
660	"""
661	set the active perspective
662	"""
663	self.tctrl_perspective.SetLabel(str(name))
664
665	def set_panel_on_focus(self, name):
666	"""
667	set the plot panel on focus
668	"""
669	self.tctrl_plotpanel.SetLabel(str(name))
670
671
672
673
674	class DataFrame(wx.Frame):
675	## Internal name for the AUI manager
676	window_name = "Data Panel"
677	## Title to appear on top of the window
678	window_caption = "Data Panel"
679	## Flag to tell the GUI manager that this panel is not
680	# tied to any perspective
681	ALWAYS_ON = True
682
683	def __init__(self, parent=None, owner=None, manager=None,size=(200, 800),
684	list_of_perspective=[],list=[], args, *kwds):
685	kwds['size'] = size
686	kwds['id'] = -1
687	kwds['title']= "Loaded Data"
688	wx.Frame.__init__(self, parent=parent, args, *kwds)
689	self.parent = parent
690	self.owner = owner
691	self.manager = manager
692	self.panel = DataPanel(parent=self,
693	#size=size,
694	list_of_perspective=list_of_perspective)
695
696	def load_data_list(self, list=[]):
697	"""
698	Fill the list inside its panel
699	"""
700	self.panel.load_data_list(list=list)
701
702	def layout_perspective(self, list_of_perspective=[]):
703	"""
704	"""
705	self.panel.layout_perspective(list_of_perspective=list_of_perspective)
706
707
708
709
710	from dataFitting import Data1D
711	from dataFitting import Data2D, Theory1D
712	from data_state import DataState
713	import sys
714	class State():
715	def __init__(self):
716	self.msg = ""
717	def __str__(self):
718	self.msg = "model mane : model1\n"
719	self.msg += "params : \n"
720	self.msg += "name value\n"
721	return msg
722	def set_data_state(data=None, path=None, theory=None, state=None):
723	dstate = DataState(data=data)
724	dstate.set_path(path=path)
725	dstate.set_theory(theory, state)
726
727	return dstate
728	"""'
729	data_list = [1:('Data1', 'Data1D', '07/01/2010', "theory1d", "state1"),
730	('Data2', 'Data2D', '07/03/2011', "theory2d", "state1"),
731	('Data3', 'Theory1D', '06/01/2010', "theory1d", "state1"),
732	('Data4', 'Theory2D', '07/01/2010', "theory2d", "state1"),
733	('Data5', 'Theory2D', '07/02/2010', "theory2d", "state1")]
734	"""
735	if __name__ == "__main__":
736
737	app = wx.App()
738	try:
739	list_of_perspective = [('perspective2', False), ('perspective1', True)]
740	data_list = {}
741	# state 1
742	data = Data2D()
743	data.name = "data2"
744	data.id = 1
745	data.append_empty_process()
746	process = data.process[len(data.process)-1]
747	process.data = "07/01/2010"
748	theory = Data2D()
749	theory.id = 34
750	theory.name = "theory1"
751	path = "path1"
752	state = State()
753	data_list['1']=set_data_state(data, path,theory, state)
754	#state 2
755	data = Data2D()
756	data.name = "data2"
757	data.id = 76
758	theory = Data2D()
759	theory.id = 78
760	theory.name = "CoreShell 07/24/25"
761	path = "path2"
762	#state3
763	state = State()
764	data_list['2']=set_data_state(data, path,theory, state)
765	data = Data1D()
766	data.id = 3
767	data.name = "data2"
768	theory = Theory1D()
769	theory.name = "CoreShell"
770	theory.id = 4
771	theory.append_empty_process()
772	process = theory.process[len(theory.process)-1]
773	process.description = "this is my description"
774	path = "path3"
775	data.append_empty_process()
776	process = data.process[len(data.process)-1]
777	process.data = "07/22/2010"
778	data_list['4']=set_data_state(data, path,theory, state)
779	#state 4
780	temp_data_list = {}
781	data.name = "data5 erasing data2"
782	temp_data_list['4'] = set_data_state(data, path,theory, state)
783	#state 5
784	data = Data2D()
785	data.name = "data3"
786	data.id = 5
787	data.append_empty_process()
788	process = data.process[len(data.process)-1]
789	process.data = "07/01/2010"
790	theory = Theory1D()
791	theory.name = "Cylinder"
792	path = "path2"
793	state = State()
794	dstate= set_data_state(data, path,theory, state)
795	theory = Theory1D()
796	theory.id = 6
797	theory.name = "CoreShell"
798	dstate.set_theory(theory)
799	theory = Theory1D()
800	theory.id = 6
801	theory.name = "CoreShell replacing coreshell in data3"
802	dstate.set_theory(theory)
803	data_list['3'] = dstate
804	#state 6
805	data_list['6']=set_data_state(None, path,theory, state)
806	data_list['6']=set_data_state(theory=theory, state=None)
807	theory = Theory1D()
808	theory.id = 7
809	data_list['6']=set_data_state(theory=theory, state=None)
810	data_list['7']=set_data_state(theory=theory, state=None)
811	window = DataFrame(list=data_list)
812	window.load_data_list(list=data_list)
813	#window.layout_perspective(list_of_perspective=list_of_perspective)
814	window.Show(True)
815	window.load_data_list(list=temp_data_list)
816	except:
817	#raise
818	print "error",sys.exc_value
819
820	app.MainLoop()
821
822

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

SasView

source: sasview/guiframe/data_panel.py @ ba64348

Download in other formats: