data_panel.py @ 1c6389c

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 1c6389c was c5a769e, checked in by Jae Cho <jhjcho@…>, 13 years ago
more touch: move delete plot to p_panel from data_panel…
Property mode set to `100644`
File size: 44.8 KB

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

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

SasView

source: sasview/guiframe/data_panel.py @ 1c6389c

Download in other formats: