data_panel.py @ 5385a9e

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 5385a9e was 32c0841, checked in by Gervaise Alina <gervyh@…>, 14 years ago
working on pylint
Property mode set to `100644`
File size: 18.7 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	"""
	13	import wx
	14	import sys
[32c0841]	15	#import warnings
[3c44c66]	16	from wx.lib.scrolledpanel import ScrolledPanel
	17	import wx.lib.mixins.listctrl as listmix
[32c0841]	18	#from sans.guicomm.events import NewPlotEvent
[3c44c66]	19
	20	class CheckListCtrl(wx.ListCtrl, listmix.CheckListCtrlMixin,
	21	listmix.ListCtrlAutoWidthMixin):
	22	"""
	23	Check list control to be used for Data Panel
	24	"""
	25	def __init__(self, parent, args, *kwds):
[32c0841]	26	kwds['style'] = wx.LC_REPORT\|wx.SUNKEN_BORDER
[3c44c66]	27	wx.ListCtrl.__init__(self, parent, -1, args, *kwds)
	28	listmix.CheckListCtrlMixin.__init__(self)
	29	listmix.ListCtrlAutoWidthMixin.__init__(self)
	30
	31	class DataPanel(ScrolledPanel):
	32	"""
	33	This panel displays data available in the application and widgets to
	34	interact with data.
	35	"""
[32c0841]	36	def __init__(self, parent, list=None, list_of_perspective=[],
	37	args, *kwds):
[3c44c66]	38	ScrolledPanel.__init__(self, parent=parent, args, *kwds)
	39	self.SetupScrolling()
	40	self.parent = parent
	41	self.manager = None
	42	self.owner = None
[32c0841]	43	if list is None:
	44	list = []
[3c44c66]	45	self.list_of_data = list
	46	self.perspectives = []
[32c0841]	47	#layout widgets
	48	self.sizer4 = None
	49	self.sizer5 = None
	50	self.sizer1 = None
	51	self.sizer2 = None
	52	self.sizer3 = None
	53	self.sizer4 = None
	54	self.sizer5 = None
	55	self.vbox = None
	56	self.list_ctrl = None
	57	self.boxsizer_2_2 = None
	58	self.cb_select_data1d = None
	59	self.cb_select_data2d = None
	60	self.cb_select_all = None
	61	self.cb_theory = None
	62	self.bt_import = None
	63	self.bt_plot = None
	64	self.bt_close = None
[3c44c66]	65
	66	self.define_panel_structure()
	67	self.layout_list()
	68	self.layout_selection()
	69	self.layout_perspective(list_of_perspective=list_of_perspective)
	70	self.load_list()
	71	self.layout_theory()
	72	self.layout_button()
	73
	74	def define_panel_structure(self):
	75	"""
	76	Define the skeleton of the panel
	77	"""
	78	self.vbox = wx.BoxSizer(wx.VERTICAL)
	79	self.sizer1 = wx.BoxSizer(wx.VERTICAL)
	80
	81	box_description_2 = wx.StaticBox(self, -1, "Selection Patterns")
	82	self.boxsizer_2 = wx.StaticBoxSizer(box_description_2, wx.HORIZONTAL)
	83	box_description_2_2 = wx.StaticBox(self, -1, "Set Active Perspective")
[32c0841]	84	self.boxsizer_2_2 = wx.StaticBoxSizer(box_description_2_2,
	85	wx.HORIZONTAL)
[3c44c66]	86
	87	w, h = self.parent.GetSize()
	88	self.sizer2 = wx.BoxSizer(wx.HORIZONTAL)
[32c0841]	89	self.sizer2.Add(self.boxsizer_2, 1, wx.ALL, 10)
	90	self.sizer2.Add(self.boxsizer_2_2, 1, wx.ALL, 10)
[3c44c66]	91
[32c0841]	92	box_description_3 = wx.StaticBox(self, -1,
	93	"Import to Active perspective")
[3c44c66]	94	self.boxsizer_3 = wx.StaticBoxSizer(box_description_3, wx.HORIZONTAL)
	95	self.sizer3 = wx.BoxSizer(wx.HORIZONTAL)
[32c0841]	96	self.sizer3.Add(self.boxsizer_3, 1, wx.ALL, 10)
[3c44c66]	97
	98	self.sizer4 = wx.BoxSizer(wx.HORIZONTAL)
	99	self.sizer5 = wx.BoxSizer(wx.HORIZONTAL)
	100	self.sizer1.SetMinSize((w-10, h/3))
	101	self.sizer2.SetMinSize((w-10, -1))
	102	self.sizer3.SetMinSize((w-10, -1))
	103	self.sizer4.SetMinSize((w-10, -1))
	104	self.sizer5.SetMinSize((w-10, -1))
	105	self.vbox.Add(self.sizer1)
	106	self.vbox.Add(self.sizer2)
	107	self.vbox.Add(self.sizer3)
[32c0841]	108	self.vbox.Add(wx.StaticLine(self, -1), 0, wx.EXPAND, 0)
[3c44c66]	109	self.vbox.Add(self.sizer4)
	110	self.vbox.Add(self.sizer5)
	111	self.SetSizer(self.vbox)
	112
	113	def GetListCtrl(self):
[32c0841]	114	"""
	115	"""
[3c44c66]	116	return self.list_ctrl
	117
	118	def layout_list(self):
	119	"""
	120	Add a listcrtl in the panel
	121	"""
	122	self.list_ctrl = CheckListCtrl(parent=self)
	123
	124	self.list_ctrl.InsertColumn(0, 'Name')
	125	self.list_ctrl.InsertColumn(1, 'Type')
	126	self.list_ctrl.InsertColumn(2, 'Date Modified')
	127	self.sizer1.Add(self.list_ctrl, 1, wx.EXPAND\|wx.ALL, 10)
	128
[32c0841]	129	def layout_perspective(self, list_of_perspective=None):
[3c44c66]	130	"""
	131	Layout widgets related to the list of plug-ins of the gui_manager
	132	"""
[32c0841]	133	if list_of_perspective is None:
	134	list_of_perspective = []
[3c44c66]	135	self.boxsizer_2_2.Clear(True)
	136	self.perspectives = []
[32c0841]	137	sizer = wx.GridBagSizer(5, 5)
[3c44c66]	138
	139	if list_of_perspective:
	140	item = list_of_perspective[0].sub_menu
	141	rb = wx.RadioButton(self, -1, item, style=wx.RB_GROUP)
	142	rb.SetToolTipString("Data will be applied to this perspective")
[32c0841]	143	if hasattr(item, "set_default_perspective"):
[3c44c66]	144	if item.set_default_perspective():
	145	rb.SetValue(item.set_default_perspective())
	146	self.Bind(wx.EVT_RADIOBUTTON, self.on_set_active_perspective,
	147	id=rb.GetId())
	148	self.perspectives.append(rb)
	149	ix = 0
	150	iy = 0
[32c0841]	151	sizer.Add(rb, (iy, ix), (1, 1),
[3c44c66]	152	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 10)
	153	for index in range(1, len(list_of_perspective)):
	154	item = list_of_perspective[index].sub_menu
[32c0841]	155	rb = wx.RadioButton(self, -1, item)
[3c44c66]	156	rb.SetToolTipString("Data will be applied to this perspective")
	157	self.Bind(wx.EVT_RADIOBUTTON, self.on_set_active_perspective,
	158	id=rb.GetId())
	159	self.perspectives.append(rb)
	160	iy += 1
[32c0841]	161	sizer.Add(rb, (iy, ix), (1, 1),
[3c44c66]	162	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 10)
	163	if hasattr(item,"set_default_perspective"):
	164	if item.set_default_perspective():
	165	rb.SetValue(item.set_default_perspective())
	166	else:
	167	rb = wx.RadioButton(self, -1, 'No Perspective',
	168	style=wx.RB_GROUP)
	169	rb.SetValue(True)
	170	rb.Disable()
	171	ix = 0
	172	iy = 0
[32c0841]	173	sizer.Add(rb, (iy, ix), (1, 1),
[3c44c66]	174	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 10)
	175	self.boxsizer_2_2.Add(sizer)
	176
	177	def layout_selection(self):
	178	"""
	179	Layout widgets related to selection patterns
	180	"""
[32c0841]	181	sizer = wx.GridBagSizer(5, 5)
	182	self.cb_select_data1d = wx.CheckBox(self, -1,
	183	"Select/Unselect Data 1D", (10, 10))
[3c44c66]	184	msg_data1d = "To check/uncheck to select/unselect all Data 1D"
	185	self.cb_select_data1d.SetToolTipString(msg_data1d)
	186	wx.EVT_CHECKBOX(self, self.cb_select_data1d.GetId(),
	187	self.on_select_all_data1d)
	188	self.cb_select_data2d = wx.CheckBox(self, -1,
	189	"Select/Unselect all Data 2D", (10, 10))
	190	msg_data2d = "To check/uncheck to select/unselect all Data 2D"
	191	self.cb_select_data2d.SetToolTipString(msg_data2d)
	192	wx.EVT_CHECKBOX(self, self.cb_select_data2d.GetId(),
	193	self.on_select_all_data2d)
	194	self.cb_select_theory1d = wx.CheckBox(self, -1,
[32c0841]	195	"Select/Unselect all Theory 1D", (10, 10))
[3c44c66]	196	msg_theory1d = "To check/uncheck to select/unselect all Theory 1D"
	197	self.cb_select_theory1d.SetToolTipString(msg_theory1d)
	198	wx.EVT_CHECKBOX(self, self.cb_select_theory1d.GetId(),
	199	self.on_select_all_theory1d)
	200	self.cb_select_theory2d = wx.CheckBox(self, -1,
[32c0841]	201	"Select/Unselect all Theory 2D", (10, 10))
[3c44c66]	202	msg_theory2d = "To check/uncheck to select/unselect all Theory 2D"
	203	self.cb_select_theory2d.SetToolTipString(msg_theory2d)
	204	wx.EVT_CHECKBOX(self, self.cb_select_theory2d.GetId(),
	205	self.on_select_all_theory2d)
[32c0841]	206	self.cb_select_all = wx.CheckBox(self, -1, "Select/Unselect all",
	207	(10, 10))
[3c44c66]	208	msg_select_all = "To check/uncheck to select/unselect all"
	209	self.cb_select_all.SetToolTipString(msg_select_all)
	210	wx.EVT_CHECKBOX(self, self.cb_select_all.GetId(), self.on_select_all)
	211	iy = 0
	212	ix = 0
[32c0841]	213	sizer.Add(self.cb_select_data1d, (iy, ix), (1, 1),
[3c44c66]	214	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 10)
	215	iy += 1
[32c0841]	216	sizer.Add(self.cb_select_data2d, (iy, ix), (1, 1),
[3c44c66]	217	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 10)
	218	iy += 1
[32c0841]	219	sizer.Add(self.cb_select_theory1d, (iy, ix), (1, 1),
[3c44c66]	220	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 10)
	221	iy += 1
[32c0841]	222	sizer.Add( self.cb_select_theory2d, (iy, ix), (1, 1),
[3c44c66]	223	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 10)
	224	iy += 1
[32c0841]	225	sizer.Add(self.cb_select_all,(iy, ix), (1, 1),
[3c44c66]	226	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 10)
	227	self.boxsizer_2.Add(sizer)
	228
	229	def layout_theory(self):
	230	"""
	231	Layout widget related to theory to import
	232	"""
	233	msg = "Import the Selected Theory\n"
	234	msg += "to active perspective."
	235	st_description = wx.StaticText(self, -1, msg)
	236	self.cb_theory = wx.ComboBox(self, -1)
	237	wx.EVT_COMBOBOX(self.cb_theory,-1, self.on_select_theory)
	238
[32c0841]	239	self.boxsizer_3.AddMany([(st_description, 0, wx.ALL, 10),
	240	(self.cb_theory, 0, wx.ALL, 10)])
[3c44c66]	241	self.load_theory(list=[])
	242
	243	def layout_button(self):
	244	"""
	245	Layout widgets related to buttons
	246	"""
	247	self.bt_import = wx.Button(self, wx.NewId(), "Import", (30, 10))
[32c0841]	248	hint_msg = "Import set of Data to active perspective"
	249	self.bt_import.SetToolTipString(hint_msg)
[3c44c66]	250	wx.EVT_BUTTON(self, self.bt_import.GetId(), self.on_import)
	251
	252	self.bt_plot = wx.Button(self, wx.NewId(), "Plot", (30, 10))
	253	self.bt_plot.SetToolTipString("To trigger plotting")
	254	wx.EVT_BUTTON(self, self.bt_plot.GetId(), self.on_plot)
	255
	256	self.bt_close = wx.Button(self, wx.NewId(), "Close", (30, 10))
	257	self.bt_close.SetToolTipString("close the current window")
	258	wx.EVT_BUTTON(self, self.bt_close.GetId(), self.on_close)
	259
[32c0841]	260	self.sizer5.AddMany([((40, 40), 0, wx.LEFT\|wx.ADJUST_MINSIZE, 180),
[3c44c66]	261	(self.bt_import, 0, wx.ALL,5),
	262	(self.bt_plot, 0, wx.ALL,5),
	263	(self.bt_close, 0, wx.ALL, 5 )])
	264
	265	def set_manager(self, manager):
	266	"""
	267	:param manager: object responsible of filling on empty the listcrtl of
	268	this panel. for sansview manager is data_manager
	269	"""
	270	self.manager = manager
	271
	272	def set_owner(self, owner):
	273	"""
	274	:param owner: is the main widget creating this frame
	275	for sansview owner is gui_manager
	276	"""
	277	self.owner = owner
	278
[32c0841]	279	def load_theory(self, list=None):
[3c44c66]	280	"""
	281	Recieve a list of theory name and fill the combobox with that list
	282	"""
[32c0841]	283	if list is None:
	284	list = []
[3c44c66]	285	for theory in list:
	286	self.cb_theory.Append(theory)
	287	if list:
	288	self.cb_theory.Enable()
	289	else:
	290	self.cb_theory.Disable()
	291
[32c0841]	292	def load_list(self, list=None):
[3c44c66]	293	"""
	294	Get a list of turple and store each string in these turples in
	295	the column of the listctrl.
	296
	297	:param list: list of turples containing string only.
	298	"""
[32c0841]	299	if list is None:
	300	return
[3c44c66]	301	for i in list:
	302	index = self.list_ctrl.InsertStringItem(sys.maxint, i[0])
	303	self.list_ctrl.SetStringItem(index, 1, i[1])
	304	self.list_ctrl.SetStringItem(index, 2, i[2])
	305
	306	def set_perspective(self, sub_menu):
	307	"""
	308	Receive the name of the current perspective and set
	309	the active perspective
	310	"""
	311	for item in self.perspectives:
	312	if item.GetLabelText()== sub_menu:
	313	item.SetValue(True)
	314	else:
	315	item.SetValue(False)
	316
	317	def select_data_type(self, type='Data1D', check=False):
	318	"""
	319	check item in the list according to a type.
	320	:param check: if check true set checkboxes toTrue else to False
	321	:param type: type of data to select
	322	"""
	323	num = self.list_ctrl.GetItemCount()
	324	for index in range(num):
	325	if self.list_ctrl.GetItem(index, 1).GetText() == type:
	326	self.list_ctrl.CheckItem(index, check)
	327
	328	def on_select_all_data1d(self, event):
	329	"""
	330	check/ uncheck list of all data 1D
	331	"""
	332	ctrl = event.GetEventObject()
	333	self.select_data_type(type='Data1D', check=ctrl.GetValue())
	334
	335	def on_select_all_data2d(self, event):
	336	"""
	337	check/ uncheck list of all data 2D
	338	"""
	339	ctrl = event.GetEventObject()
	340	self.select_data_type(type='Data2D', check=ctrl.GetValue())
	341
	342	def on_select_all_theory1d(self, event):
	343	"""
	344	check/ uncheck list of all theory 1D
	345	"""
	346	ctrl = event.GetEventObject()
	347	self.select_data_type(type='Theory1D', check=ctrl.GetValue())
	348
	349	def on_select_all_theory2d(self, event):
	350	"""
	351	check/ uncheck list of all theory 2D
	352	"""
	353	ctrl = event.GetEventObject()
	354	self.select_data_type(type='Theory2D', check=ctrl.GetValue())
	355
	356	def on_select_all(self, event):
	357	"""
	358	Check or uncheck all data listed
	359	"""
	360	ctrl = event.GetEventObject()
	361	self.cb_select_data1d.SetValue(ctrl.GetValue())
	362	self.cb_select_data2d.SetValue(ctrl.GetValue())
	363	self.cb_select_theory1d.SetValue(ctrl.GetValue())
	364	self.cb_select_theory2d.SetValue(ctrl.GetValue())
	365	num = self.list_ctrl.GetItemCount()
	366	for i in range(num):
	367	self.list_ctrl.CheckItem(i, ctrl.GetValue())
	368
	369	def on_select_theory(self, event):
	370	"""
	371	Select the theory to import in the active perspective
	372	"""
	373
	374	def on_set_active_perspective(self, event):
	375	"""
	376	Select the active perspective
	377	"""
	378	ctrl = event.GetEventObject()
[32c0841]	379
[3c44c66]	380	def set_data_helper(self):
	381	"""
	382	"""
	383	data_to_plot = []
	384
	385	num = self.list_ctrl.GetItemCount()
	386	for index in range(num):
	387	if self.list_ctrl.IsChecked(index):
[32c0841]	388	data_to_plot.append(self.list_ctrl.GetItemText(index))
[3c44c66]	389	return data_to_plot
	390
	391	def on_import(self, event=None):
	392	"""
	393	Get all select data and set them to the current active perspetive
	394	"""
	395	data_to_plot = self.set_data_helper()
	396	current_perspective = None
	397	if self.perspectives:
	398	for item in self.perspectives:
	399	if item.GetValue():
	400	current_perspective = item.GetLabelText()
	401	if self.manager is not None:
	402	self.manager.post_data(data_name_list=data_to_plot,
	403	perspective=current_perspective, plot=False)
	404
	405	def on_plot(self, event=None):
	406	"""
	407	Send a list of data names to plot
	408	"""
	409	data_to_plot = self.set_data_helper()
	410	if self.manager is not None:
	411	self.manager.post_data(data_name_list=data_to_plot, plot=True)
	412
	413	def on_close(self, event):
	414	"""
	415	Close the current panel's parent
	416	"""
	417	self.parent._onClose()
	418
	419
	420	data_list = [('Data1', 'Data1D', '07/01/2010'),
	421	('Data2', 'Data2D', '07/03/2011'),
	422	('Data3', 'Theory1D', '06/01/2010'),
	423	('Data4', 'Theory2D', '07/01/2010'),
	424	('Data5', 'Theory2D', '07/02/2010')]
	425
	426	class DataFrame(wx.Frame):
[32c0841]	427	def __init__(self, parent=None, owner=None,
	428	list_of_perspective=None, list=None,
	429	manager=None, args, *kwds):
[3c44c66]	430	kwds['size'] = (500, 500)
	431	kwds['id'] = -1
[32c0841]	432	kwds['title'] = "Loaded Data"
[3c44c66]	433	wx.Frame.__init__(self, parent=parent, args, *kwds)
	434	self.parent = parent
	435	self.owner = owner
	436	self.manager = manager
[32c0841]	437	self.panel = DataPanel(parent=self,
	438	list_of_perspective=list_of_perspective)
[3c44c66]	439	self.panel.load_list(list=list)
	440	wx.EVT_CLOSE(self, self._onClose)
	441
	442	def set_owner(self, owner):
	443	"""
	444	:param owner: is the main widget creating this frame
	445	for sansview owner is gui_manager
	446	"""
	447	self.owner = owner
	448	self.panel.set_owner(owner=self.owner)
	449
	450	def set_manager(self, manager):
	451	"""
	452	:param manager: object responsible of filling on empty the listcrtl of
	453	this panel. for sansview manager is data_manager
	454
	455	"""
	456	self.manager = manager
	457	self.panel.set_manager(manager=self.manager)
	458
[32c0841]	459	def load_list(self, list=None):
[3c44c66]	460	"""
	461	Fill the list inside its panel
	462	"""
	463	self.panel.load_list(list=list)
	464
[32c0841]	465	def layout_perspective(self, list_of_perspective=None):
[3c44c66]	466	"""
	467	fill the panel with list of perspective
	468	"""
	469	self.panel.layout_perspective(list_of_perspective=list_of_perspective)
	470
	471	def set_perspective(self, sub_menu):
	472	"""
	473	Receive the name of the current perspective and set
	474	the active perspective
	475	"""
	476	self.panel.set_perspective(sub_menu=sub_menu)
[32c0841]	477
[3c44c66]	478	def _onClose(self, event=None):
	479	"""
	480	this frame can only be hidden unless the application destroys it
	481	"""
	482	self.Hide()
	483
	484	if __name__ == "__main__":
	485	app = wx.App()
[32c0841]	486	window = DataFrame(list=data_list)
[3c44c66]	487	window.Show(True)
	488	app.MainLoop()
	489
	490

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

SasView

source: sasview/guiframe/data_panel.py @ 5385a9e

Download in other formats: