data_operator.py @ 786685e

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 786685e was 51f14603, checked in by Peter Parker, 11 years ago
Refs #202 - Fix Sphinx build errors (not including park-1.2.1/). Most warnings remain.
Property mode set to `100644`
File size: 33.7 KB

Rev	Line
[51f14603]	1	"""
	2	GUI for the data operation
	3	"""
	4	import wx
	5	import sys
	6	import time
	7	import numpy
	8	from sans.dataloader.data_info import Data1D
	9	from sans.plottools.PlotPanel import PlotPanel
	10	from sans.plottools.plottables import Graph
	11	from sans.plottools import transform
	12	from matplotlib.font_manager import FontProperties
	13	from sans.guiframe.events import StatusEvent
	14	from sans.perspectives.calculator import calculator_widgets as widget
	15	#Control panel width
	16	if sys.platform.count("win32") > 0:
	17	PANEL_WIDTH = 790
	18	PANEL_HEIGTH = 370
	19	FONT_VARIANT = 0
	20	_BOX_WIDTH = 200
	21	ON_MAC = False
	22	else:
	23	_BOX_WIDTH = 230
	24	PANEL_WIDTH = 900
	25	PANEL_HEIGTH = 430
	26	FONT_VARIANT = 1
	27	ON_MAC = True
	28
	29	class DataOperPanel(wx.ScrolledWindow):
	30	"""
	31	"""
	32	def __init__(self, parent, args, *kwds):
	33	kwds['name'] = "Data Operation"
	34	kwds["size"] = (PANEL_WIDTH, PANEL_HEIGTH)
	35	wx.ScrolledWindow.__init__(self, parent, args, *kwds)
	36	self.parent = parent
	37	#sizers etc.
	38	self.main_sizer = None
	39	self.name_sizer = None
	40	self.button_sizer = None
	41	self.data_namectr = None
	42	self.numberctr = None
	43	self.data1_cbox = None
	44	self.operator_cbox = None
	45	self.data2_cbox = None
	46	self.data_title_tcl = None
	47	self.out_pic = None
	48	self.equal_pic = None
	49	self.data1_pic = None
	50	self.operator_pic = None
	51	self.data2_pic = None
	52	self.output = None
	53	self._notes = None
	54	#text grayed color
	55	self.color = wx.SystemSettings.GetColour(wx.SYS_COLOUR_BACKGROUND)
	56	#data
	57	self._data = self.get_datalist()
	58	self._do_layout()
	59	self.fill_data_combox()
	60	self.fill_oprator_combox()
	61	self.Bind(wx.EVT_PAINT, self.set_panel_on_focus)
	62
	63	def _define_structure(self):
	64	"""
	65	define initial sizer
	66	"""
	67	self.main_sizer = wx.BoxSizer(wx.VERTICAL)
	68	title = "Data Operation "
	69	title += "[ + (add); - (subtract); "
	70	title += "* (multiply); / (divide); "
	71	title += "\| (append) ]"
	72	name_box = wx.StaticBox(self, -1, title)
	73	self.name_sizer = wx.StaticBoxSizer(name_box, wx.HORIZONTAL)
	74	self.button_sizer = wx.BoxSizer(wx.HORIZONTAL)
	75
	76	def _layout_name(self):
	77	"""
	78	Do the layout for data name related widgets
	79	"""
	80	new_data_sizer = wx.BoxSizer(wx.VERTICAL)
	81	equal_sizer = wx.BoxSizer(wx.VERTICAL)
	82	old_data1_sizer = wx.BoxSizer(wx.VERTICAL)
	83	operator_sizer = wx.BoxSizer(wx.VERTICAL)
	84	old_data2_sizer = wx.BoxSizer(wx.VERTICAL)
	85	data2_hori_sizer = wx.BoxSizer(wx.HORIZONTAL)
	86	data_name = wx.StaticText(self, -1, 'Output Data Name')
	87	equal_name = wx.StaticText(self, -1, ' =', size=(50, 25))
	88	data1_name = wx.StaticText(self, -1, 'Data1')
	89	operator_name = wx.StaticText(self, -1, 'Operator')
	90	data2_name = wx.StaticText(self, -1, 'Data2 (or Number)')
	91	self.data_namectr = wx.TextCtrl(self, -1, size=(_BOX_WIDTH, 25), style=wx.TE_PROCESS_ENTER)
	92	self.data_namectr.SetToolTipString("Hit 'Enter' key after typing.")
	93	self.data_namectr.SetValue(str('MyNewDataName'))
	94	self.numberctr = wx.TextCtrl(self, -1, size=(_BOX_WIDTH/3, 25), style=wx.TE_PROCESS_ENTER)
	95	self.numberctr.SetToolTipString("Hit 'Enter' key after typing.")
	96	self.numberctr.SetValue(str(1.0))
	97	self.data1_cbox = wx.ComboBox(self, -1, size=(_BOX_WIDTH, 25),
	98	style=wx.CB_READONLY)
	99	self.operator_cbox = wx.ComboBox(self, -1, size=(70, 25),
	100	style=wx.CB_READONLY)
	101	operation_tip = "Add: +, Subtract: -, "
	102	operation_tip += "Multiply: *, Divide: /, "
	103	operation_tip += "Append(Combine): \| "
	104	self.operator_cbox.SetToolTipString(operation_tip)
	105	self.data2_cbox = wx.ComboBox(self, -1, size=(_BOX_WIDTH*2/3, 25),
	106	style=wx.CB_READONLY)
	107
	108	self.out_pic = SmallPanel(self, -1, True,
	109	size=(_BOX_WIDTH, _BOX_WIDTH),
	110	style=wx.NO_BORDER)
	111	self.equal_pic = SmallPanel(self, -1, True, '=',
	112	size=(50, _BOX_WIDTH),
	113	style=wx.NO_BORDER)
	114	self.data1_pic = SmallPanel(self, -1, True,
	115	size=(_BOX_WIDTH, _BOX_WIDTH),
	116	style=wx.NO_BORDER)
	117	self.operator_pic = SmallPanel(self, -1, True, '+',
	118	size=(70, _BOX_WIDTH),
	119	style=wx.NO_BORDER)
	120	self.data2_pic = SmallPanel(self, -1, True,
	121	size=(_BOX_WIDTH, _BOX_WIDTH),
	122	style=wx.NO_BORDER)
	123	for ax in self.equal_pic.axes:
	124	ax.set_frame_on(False)
	125	for ax in self.operator_pic.axes:
	126	ax.set_frame_on(False)
	127
	128	new_data_sizer.AddMany([(data_name, 0, wx.LEFT, 3),
	129	(self.data_namectr, 0, wx.LEFT, 3),
	130	(self.out_pic, 0, wx.LEFT, 3)])
	131	equal_sizer.AddMany([(13, 13), (equal_name, 0, wx.LEFT, 3),
	132	(self.equal_pic, 0, wx.LEFT, 3)])
	133	old_data1_sizer.AddMany([(data1_name, 0, wx.LEFT, 3),
	134	(self.data1_cbox, 0, wx.LEFT, 3),
	135	(self.data1_pic, 0, wx.LEFT, 3)])
	136	operator_sizer.AddMany([(operator_name, 0, wx.LEFT, 3),
	137	(self.operator_cbox, 0, wx.LEFT, 3),
	138	(self.operator_pic, 0, wx.LEFT, 3)])
	139	data2_hori_sizer.AddMany([(self.data2_cbox, 0, wx.LEFT, 0),
	140	(self.numberctr, 0, wx.RIGHT, 0)])
	141	old_data2_sizer.AddMany([(data2_name, 0, wx.LEFT, 3),
	142	(data2_hori_sizer, 0, wx.LEFT, 3),
	143	(self.data2_pic, 0, wx.LEFT, 3)])
	144	self.name_sizer.AddMany([(new_data_sizer, 0, wx.LEFT\|wx.TOP, 5),
	145	(equal_sizer, 0, wx.TOP, 5),
	146	(old_data1_sizer, 0, wx.TOP, 5),
	147	(operator_sizer, 0, wx.TOP, 5),
	148	(old_data2_sizer, 0, wx.TOP, 5)])
	149	self.data2_cbox.Show(True)
	150
	151	self._show_numctrl(self.numberctr, False)
	152
	153	wx.EVT_TEXT_ENTER(self.data_namectr, -1, self.on_name)
	154	wx.EVT_TEXT_ENTER(self.numberctr, -1, self.on_number)
	155	wx.EVT_COMBOBOX(self.data1_cbox, -1, self.on_select_data1)
	156	wx.EVT_COMBOBOX(self.operator_cbox, -1, self.on_select_operator)
	157	wx.EVT_COMBOBOX(self.data2_cbox, -1, self.on_select_data2)
	158
	159	def _show_numctrl(self, ctrl, enable=True):
	160	"""
	161	Show/Hide on Win
	162	Enable/Disable on MAC
	163	"""
	164	if ON_MAC:
	165	ctrl.Enable(enable)
	166	children = ctrl.GetChildren()
	167	if len(children) > 0:
	168	ctrl.GetChildren()[0].SetBackGroundColour(self.color)
	169	if enable:
	170	wx.EVT_TEXT_ENTER(self.numberctr, -1, self.on_number)
	171	else:
	172	if not ctrl.IsEnabled():
	173	ctrl.Enable(True)
	174	ctrl.Show(enable)
	175
	176	def on_name(self, event=None):
	177	"""
	178	On data name typing
	179	"""
	180	if event != None:
	181	event.Skip()
	182	item = event.GetEventObject()
	183	if item.IsEnabled():
	184	self._set_textctrl_color(item, 'white')
	185	else:
	186	self._set_textctrl_color(item, self.color)
	187	text = item.GetValue().strip()
	188	self._check_newname(text)
	189
	190	def _check_newname(self, name=None):
	191	"""
	192	Check name ctr strings
	193	"""
	194	self.send_warnings('')
	195	msg = ''
	196	if name == None:
	197	text = self.data_namectr.GetValue().strip()
	198	else:
	199	text = name
	200	state_list = self.get_datalist().values()
	201	name_list = []
	202	for state in state_list:
	203	if state.data == None:
	204	theory_list = state.get_theory()
	205	theory, _ = theory_list.values()[0]
	206	d_name = str(theory.name)
	207	else:
	208	d_name = str(state.data.name)
	209	name_list.append(d_name)
	210	if text in name_list:
	211	self._set_textctrl_color(self.data_namectr, 'pink')
	212	msg = "DataOperation: The name already exists."
	213	if len(text) == 0:
	214	self._set_textctrl_color(self.data_namectr, 'pink')
	215	msg = "DataOperation: Type the data name first."
	216	if self._notes:
	217	self.send_warnings(msg, 'error')
	218	self.name_sizer.Layout()
	219	self.Refresh()
	220
	221	def _set_textctrl_color(self, ctrl, color):
	222	"""
	223	Set TextCtrl color
	224	"""
	225	if ON_MAC:
	226	children = ctrl.GetChildren()
	227	if len(children) > 0:
	228	children[0].SetBackgroundColour(color)
	229	else:
	230	ctrl.SetBackgroundColour(color)
	231	self.name_sizer.Layout()
	232
	233	def on_number(self, event=None):
	234	"""
	235	On selecting Number for Data2
	236	"""
	237	self.send_warnings('')
	238	item = event.GetEventObject()
	239	text = item.GetValue().strip()
	240	if self.numberctr.IsShown():
	241	if self.numberctr.IsEnabled():
	242	self._set_textctrl_color(self.numberctr, 'white')
	243	try:
	244	val = float(text)
	245	pos = self.data2_cbox.GetCurrentSelection()
	246	self.data2_cbox.SetClientData(pos, val)
	247	except:
	248	self._set_textctrl_color(self.numberctr, 'pink')
	249	msg = "DataOperation: Number requires a float number."
	250	self.send_warnings(msg, 'error')
	251	return
	252	else:
	253	self._set_textctrl_color(self.numberctr, self.color)
	254
	255	self.put_text_pic(self.data2_pic, content=str(val))
	256	self.check_data_inputs()
	257	if self.output != None:
	258	self.output.name = str(self.data_namectr.GetValue())
	259	self.draw_output(self.output)
	260	self.Refresh()
	261
	262	def on_select_data1(self, event=None):
	263	"""
	264	On select data1
	265	"""
	266	self.send_warnings('')
	267	item = event.GetEventObject()
	268	pos = item.GetCurrentSelection()
	269	data = item.GetClientData(pos)
	270	if data == None:
	271	content = "?"
	272	self.put_text_pic(self.data1_pic, content)
	273	else:
	274	self.data1_pic.add_image(data)
	275	self.check_data_inputs()
	276	if self.output != None:
	277	self.output.name = str(self.data_namectr.GetValue())
	278	self.draw_output(self.output)
	279
	280	def on_select_operator(self, event=None):
	281	"""
	282	On Select an Operator
	283	"""
	284	self.send_warnings('')
	285	item = event.GetEventObject()
	286	text = item.GetValue().strip()
	287	self.put_text_pic(self.operator_pic, content=text)
	288	self.check_data_inputs()
	289	if self.output != None:
	290	self.output.name = str(self.data_namectr.GetValue())
	291	self.draw_output(self.output)
	292
	293	def on_select_data2(self, event=None):
	294	"""
	295	On Selecting Data2
	296	"""
	297	self.send_warnings('')
	298	item = event.GetEventObject()
	299	text = item.GetValue().strip().lower()
	300	self._show_numctrl(self.numberctr, text=='number')
	301	pos = item.GetCurrentSelection()
	302	data = item.GetClientData(pos)
	303	content = "?"
	304	if not (self.numberctr.IsShown() and self.numberctr.IsEnabled()):
	305	if data == None:
	306	content = "?"
	307	self.put_text_pic(self.data2_pic, content)
	308	else:
	309	self.data2_pic.add_image(data)
	310	self.check_data_inputs()
	311	else:
	312	content = str(self.numberctr.GetValue().strip())
	313	try:
	314	content = float(content)
	315	data = content
	316	except:
	317	self._set_textctrl_color(self.numberctr, 'pink')
	318	content = "?"
	319	data = None
	320	item.SetClientData(pos, data)
	321	if data != None:
	322	self.check_data_inputs()
	323
	324	self.put_text_pic(self.data2_pic, content)
	325
	326	if self.output != None:
	327	self.output.name = str(self.data_namectr.GetValue())
	328	self.draw_output(self.output)
	329
	330	def put_text_pic(self, pic=None, content=''):
	331	"""
	332	Put text to the pic
	333	"""
	334	pic.set_content(content)
	335	pic.add_text()
	336	pic.draw()
	337
	338	def check_data_inputs(self):
	339	"""
	340	Check data1 and data2 whether or not they are ready for operation
	341	"""
	342	self._set_textctrl_color(self.data1_cbox, 'white')
	343	self._set_textctrl_color(self.data2_cbox, 'white')
	344	flag = False
	345	pos1 = self.data1_cbox.GetCurrentSelection()
	346	data1 = self.data1_cbox.GetClientData(pos1)
	347	if data1 == None:
	348	self.output = None
	349	return flag
	350	pos2 = self.data2_cbox.GetCurrentSelection()
	351	data2 = self.data2_cbox.GetClientData(pos2)
	352
	353	if data2 == None:
	354	self.output = None
	355	return flag
	356	if self.numberctr.IsShown():
	357	if self.numberctr.IsEnabled():
	358	self._set_textctrl_color(self.numberctr, 'white')
	359	try:
	360	float(data2)
	361	if self.operator_cbox.GetValue().strip() == '\|':
	362	msg = "DataOperation: This operation can not accept "
	363	msg += "a float number."
	364	self.send_warnings(msg, 'error')
	365	self._set_textctrl_color(self.numberctr, 'pink')
	366	self.output = None
	367	return flag
	368	except:
	369	msg = "DataOperation: Number requires a float number."
	370	self.send_warnings(msg, 'error')
	371	self._set_textctrl_color(self.numberctr, 'pink')
	372	self.output = None
	373	return flag
	374	else:
	375	self._set_textctrl_color(self.numberctr, self.color )
	376	elif data1.__class__.__name__ != data2.__class__.__name__:
	377	self._set_textctrl_color(self.data1_cbox, 'pink')
	378	self._set_textctrl_color(self.data2_cbox, 'pink')
	379	msg = "DataOperation: Data types must be same."
	380	self.send_warnings(msg, 'error')
	381	self.output = None
	382	return flag
	383	try:
	384	self.output = self.make_data_out(data1, data2)
	385	except:
	386	self._check_newname()
	387	self._set_textctrl_color(self.data1_cbox, 'pink')
	388	self._set_textctrl_color(self.data2_cbox, 'pink')
	389	msg = "DataOperation: Data types must be same."
	390	self.send_warnings(msg, 'error')
	391	self.output = None
	392	return flag
	393	return True
	394
	395	def make_data_out(self, data1, data2):
	396	"""
	397	Make a temp. data output set
	398	"""
	399	output = None
	400	pos = self.operator_cbox.GetCurrentSelection()
	401	operator = self.operator_cbox.GetClientData(pos)
	402	try:
	403	exec "output = data1 %s data2"% operator
	404	except:
	405	raise
	406	return output
	407
	408
	409	def draw_output(self, output):
	410	"""
	411	Draw output data(temp)
	412	"""
	413	out = self.out_pic
	414	if output == None:
	415	content = "?"
	416	self.put_text_pic(out, content)
	417	else:
	418	out.add_image(output)
	419	wx.CallAfter(self.name_sizer.Layout)
	420	self.Layout()
	421	self.Refresh()
	422
	423	def _layout_button(self):
	424	"""
	425	Do the layout for the button widgets
	426	"""
	427	self.bt_apply = wx.Button(self, -1, "Apply", size=(_BOX_WIDTH/2, -1))
	428	app_tip = "Generate the Data and send to Data Explorer."
	429	self.bt_apply.SetToolTipString(app_tip)
	430	self.bt_apply.Bind(wx.EVT_BUTTON, self.on_click_apply)
	431
	432	self.bt_close = wx.Button(self, -1, 'Close', size=(_BOX_WIDTH/2, -1))
	433	self.bt_close.Bind(wx.EVT_BUTTON, self.on_close)
	434	self.bt_close.SetToolTipString("Close this panel.")
	435
	436	self.button_sizer.AddMany([(PANEL_WIDTH/2, 25),
	437	(self.bt_apply, 0, wx.RIGHT, 10),
	438	(self.bt_close, 0, wx.RIGHT, 10)])
	439
	440	def _do_layout(self):
	441	"""
	442	Draw the current panel
	443	"""
	444	self._define_structure()
	445	self._layout_name()
	446	self._layout_button()
	447	self.main_sizer.AddMany([(self.name_sizer, 0, wx.EXPAND\|wx.ALL, 10),
	448	(self.button_sizer, 0,
	449	wx.EXPAND\|wx.TOP\|wx.BOTTOM, 5)])
	450	self.SetSizer(self.main_sizer)
	451	self.SetScrollbars(20, 20, 25, 65)
	452	self.SetAutoLayout(True)
	453
	454	def set_panel_on_focus(self, event):
	455	"""
	456	On Focus at this window
	457	"""
	458	if event != None:
	459	event.Skip()
	460	self._data = self.get_datalist()
	461	if ON_MAC:
	462	self.fill_data_combox()
	463	else:
	464	children = self.GetChildren()
	465	# update the list only when it is on the top
	466	if self.FindFocus() in children:
	467	self.fill_data_combox()
	468
	469	def fill_oprator_combox(self):
	470	"""
	471	fill the current combobox with the operator
	472	"""
	473	operator_list = [' +', ' -', ' *', " /", " \|"]
	474	for oper in operator_list:
	475	pos = self.operator_cbox.Append(str(oper))
	476	self.operator_cbox.SetClientData(pos, str(oper.strip()))
	477	self.operator_cbox.SetSelection(0)
	478
	479
	480	def fill_data_combox(self):
	481	"""
	482	fill the current combobox with the available data
	483	"""
	484	pos_pre1 = self.data1_cbox.GetCurrentSelection()
	485	pos_pre2 = self.data2_cbox.GetCurrentSelection()
	486	current1 = self.data1_cbox.GetLabel()
	487	current2 = self.data2_cbox.GetLabel()
	488	if pos_pre1 < 0:
	489	pos_pre1 = 0
	490	if pos_pre2 < 0:
	491	pos_pre2 = 0
	492	self.data1_cbox.Clear()
	493	self.data2_cbox.Clear()
	494
	495	if not self._data:
	496	pos = self.data1_cbox.Append('No Data Available')
	497	self.data1_cbox.SetSelection(pos)
	498	self.data1_cbox.SetClientData(pos, None)
	499	pos2 = self.data2_cbox.Append('No Data Available')
	500	self.data2_cbox.SetSelection(pos2)
	501	self.data2_cbox.SetClientData(pos2, None)
	502	return
	503	pos1 = self.data1_cbox.Append('Select Data')
	504	self.data1_cbox.SetSelection(pos1)
	505	self.data1_cbox.SetClientData(pos1, None)
	506	pos2 = self.data2_cbox.Append('Select Data')
	507	self.data2_cbox.SetSelection(pos2)
	508	self.data2_cbox.SetClientData(pos2, None)
	509	pos3 = self.data2_cbox.Append('Number')
	510	val = None
	511	if (self.numberctr.IsShown() and self.numberctr.IsEnabled()):
	512	try:
	513	val = float(self.numberctr.GetValue())
	514	except:
	515	val = None
	516	self.data2_cbox.SetClientData(pos3, val)
	517	dnames = []
	518	ids = self._data.keys()
	519	for id in ids:
	520	if id != None:
	521	if self._data[id].data != None:
	522	dnames.append(self._data[id].data.name)
	523	else:
	524	theory_list = self._data[id].get_theory()
	525	theory, _ = theory_list.values()[0]
	526	dnames.append(theory.name)
	527	ind = numpy.argsort(dnames)
	528	if len(ind) > 0:
	529	val_list = numpy.array(self._data.values())[ind]
	530	for datastate in val_list:
	531	data = datastate.data
	532	if data != None:
	533	name = data.name
	534	pos1 = self.data1_cbox.Append(str(name))
	535	self.data1_cbox.SetClientData(pos1, data)
	536	pos2 = self.data2_cbox.Append(str(name))
	537	self.data2_cbox.SetClientData(pos2, data)
	538	if str(current1) == str(name):
	539	pos_pre1 = pos1
	540	if str(current2) == str(name):
	541	pos_pre2 = pos2
	542	try:
	543	theory_list = datastate.get_theory()
	544	for theory, _ in theory_list.values():
	545	th_name = theory.name
	546	posth1 = self.data1_cbox.Append(str(th_name))
	547	self.data1_cbox.SetClientData(posth1, theory)
	548	posth2 = self.data2_cbox.Append(str(th_name))
	549	self.data2_cbox.SetClientData(posth2, theory)
	550	if str(current1) == str(th_name):
	551	pos_pre1 = posth1
	552	if str(current2) == str(th_name):
	553	pos_pre2 = posth2
	554	except:
	555	continue
	556	self.data1_cbox.SetSelection(pos_pre1)
	557	self.data2_cbox.SetSelection(pos_pre2)
	558
	559	def get_datalist(self):
	560	"""
	561	"""
	562	data_manager = self.parent.parent.get_data_manager()
	563	if data_manager != None:
	564	return data_manager.get_all_data()
	565	else:
	566	return {}
	567
	568	def on_click_apply(self, event):
	569	"""
	570	changes are saved in data object imported to edit
	571	"""
	572	self.send_warnings('')
	573	self.data_namectr.SetBackgroundColour('white')
	574	state_list = self.get_datalist().values()
	575	name = self.data_namectr.GetValue().strip()
	576	name_list = []
	577	for state in state_list:
	578	if state.data == None:
	579	theory_list = state.get_theory()
	580	theory, _ = theory_list.values()[0]
	581	d_name = str(theory.name)
	582	else:
	583	d_name = str(state.data.name)
	584	name_list.append(d_name)
	585	if name in name_list:
	586	self._set_textctrl_color(self.data_namectr, 'pink')
	587	msg = "The Output Data Name already exists... "
	588	wx.MessageBox(msg, 'Error')
	589	return
	590	if name == '':
	591	self._set_textctrl_color(self.data_namectr, 'pink')
	592	msg = "Please type the output data name first... "
	593	wx.MessageBox(msg, 'Error')
	594	return
	595	if self.output == None:
	596	msg = "No Output Data has been generated... "
	597	wx.MessageBox(msg, 'Error')
	598	return
	599	# send data to data manager
	600	self.output.name = name
	601	self.output.run = "Data Operation"
	602	self.output.instrument = "SansView"
	603	self.output.id = str(name) + str(time.time())
	604	data = {self.output.id :self.output}
	605	self.parent.parent.add_data(data)
	606	self.name_sizer.Layout()
	607	self.Refresh()
	608	#must post event here
	609	event.Skip()
	610
	611	def disconnect_panels(self):
	612	"""
	613	"""
	614	self.out_pic.connect.disconnect()
	615	self.equal_pic.connect.disconnect()
	616	self.data1_pic.connect.disconnect()
	617	self.operator_pic.connect.disconnect()
	618	self.data2_pic.connect.disconnect()
	619
	620	def on_close(self, event):
	621	"""
	622	leave data as it is and close
	623	"""
	624	self.parent.OnClose()
	625
	626	def set_plot_unfocus(self):
	627	"""
	628	Unfocus on right click
	629	"""
	630
	631	def send_warnings(self, msg='', info='info'):
	632	"""
	633	Send warning to status bar
	634	"""
	635	wx.PostEvent(self.parent.parent, StatusEvent(status=msg, info=info))
	636
	637	class SmallPanel(PlotPanel):
	638	"""
	639	PlotPanel for Quick plot and masking plot
	640	"""
	641	def __init__(self, parent, id=-1, is_number=False, content='?', **kwargs):
	642	"""
	643	"""
	644	PlotPanel.__init__(self, parent, id=id, **kwargs)
	645	self.is_number = is_number
	646	self.content = content
	647	self.point = None
	648	self.position = (0.4, 0.5)
	649	self.scale = 'linear'
	650	self.prevXtrans = "x"
	651	self.prevYtrans = "y"
	652	self.viewModel = "--"
	653	self.subplot.set_xticks([])
	654	self.subplot.set_yticks([])
	655	self.add_text()
	656	self.figure.subplots_adjust(left=0.1, bottom=0.1)
	657
	658	def set_content(self, content=''):
	659	"""
	660	Set text content
	661	"""
	662	self.content = str(content)
	663
	664	def add_toolbar(self):
	665	"""
	666	Add toolbar
	667	"""
	668	# Not implemented
	669	pass
	670
	671	def on_set_focus(self, event):
	672	"""
	673	send to the parenet the current panel on focus
	674	"""
	675	pass
	676
	677	def add_image(self, plot):
	678	"""
	679	Add Image
	680	"""
	681	self.content = ''
	682	self.textList = []
	683	self.plots = {}
	684	self.clear()
	685	self.point = plot
	686	try:
	687	self.figure.delaxes(self.figure.axes[0])
	688	self.subplot = self.figure.add_subplot(111)
	689	#self.figure.delaxes(self.figure.axes[1])
	690	except:
	691	pass
	692	try:
	693	name = plot.name
	694	except:
	695	name = plot.filename
	696	self.plots[name] = plot
	697
	698	#init graph
	699	self.graph = Graph()
	700
	701	#add plot
	702	self.graph.add(plot)
	703	#draw
	704	self.graph.render(self)
	705
	706	try:
	707	self.figure.delaxes(self.figure.axes[1])
	708	except:
	709	pass
	710	self.subplot.figure.canvas.resizing = False
	711	self.subplot.tick_params(axis='both', labelsize=9)
	712	# Draw zero axis lines
	713	self.subplot.axhline(linewidth = 1, color='r')
	714	self.subplot.axvline(linewidth = 1, color='r')
	715
	716	self.erase_legend()
	717	try:
	718	# mpl >= 1.1.0
	719	self.figure.tight_layout()
	720	except:
	721	self.figure.subplots_adjust(left=0.1, bottom=0.1)
	722	self.subplot.figure.canvas.draw()
	723
	724	def add_text(self):
	725	"""
	726	Text in the plot
	727	"""
	728	if not self.is_number:
	729	return
	730
	731	self.clear()
	732	try:
	733	self.figure.delaxes(self.figure.axes[0])
	734	self.subplot = self.figure.add_subplot(111)
	735	self.figure.delaxes(self.figure.axes[1])
	736	except:
	737	pass
	738	self.subplot.set_xticks([])
	739	self.subplot.set_yticks([])
	740	label = self.content
	741	FONT = FontProperties()
	742	xpos, ypos = (0.4, 0.5)
	743	font = FONT.copy()
	744	font.set_size(14)
	745
	746	self.textList = []
	747	self.subplot.set_xlim((0, 1))
	748	self.subplot.set_ylim((0, 1))
	749
	750	try:
	751	if self.content != '?':
	752	float(label)
	753	except:
	754	self.subplot.set_frame_on(False)
	755	try:
	756	# mpl >= 1.1.0
	757	self.figure.tight_layout()
	758	except:
	759	self.figure.subplots_adjust(left=0.1, bottom=0.1)
	760	if len(label) > 0 and xpos > 0 and ypos > 0:
	761	new_text = self.subplot.text(str(xpos), str(ypos), str(label),
	762	fontproperties=font)
	763	self.textList.append(new_text)
	764
	765	def erase_legend(self):
	766	"""
	767	Remove Legend
	768	"""
	769	#for ax in self.axes:
	770	self.remove_legend(self.subplot)
	771
	772	def onMouseMotion(self, event):
	773	"""
	774	Disable dragging 2D image
	775	"""
	776
	777	def onWheel(self, event):
	778	"""
	779	"""
	780
	781	def onLeftDown(self, event):
	782	"""
	783	Disables LeftDown
	784	"""
	785
	786	def onPick(self, event):
	787	"""
	788	Remove Legend
	789	"""
	790	for ax in self.axes:
	791	self.remove_legend(ax)
	792
	793
	794	def draw(self):
	795	"""
	796	Draw
	797	"""
	798	if self.dimension == 3:
	799	pass
	800	else:
	801	self.subplot.figure.canvas.resizing = False
	802	self.subplot.tick_params(axis='both', labelsize=9)
	803	self.erase_legend()
	804	self.subplot.figure.canvas.draw_idle()
	805	try:
	806	self.figure.delaxes(self.figure.axes[1])
	807	except:
	808	pass
	809
	810
	811	def onContextMenu(self, event):
	812	"""
	813	Default context menu for a plot panel
	814	"""
	815	id = wx.NewId()
	816	slicerpop = wx.Menu()
	817	data = self.point
	818	if issubclass(data.__class__, Data1D):
	819	slicerpop.Append(id, '&Change Scale')
	820	wx.EVT_MENU(self, id, self._onProperties)
	821	else:
	822	slicerpop.Append(id, '&Toggle Linear/Log Scale')
	823	wx.EVT_MENU(self, id, self.ontogglescale)
	824	try:
	825	# mouse event
	826	pos_evt = event.GetPosition()
	827	pos = self.ScreenToClient(pos_evt)
	828	except:
	829	# toolbar event
	830	pos_x, pos_y = self.toolbar.GetPositionTuple()
	831	pos = (pos_x, pos_y + 5)
	832	self.PopupMenu(slicerpop, pos)
	833
	834	def ontogglescale(self, event):
	835	"""
	836	On toggle 2d scale
	837	"""
	838	self._onToggleScale(event)
	839	try:
	840	# mpl >= 1.1.0
	841	self.figure.tight_layout()
	842	except:
	843	self.figure.subplots_adjust(left=0.1, bottom=0.1)
	844	try:
	845	self.figure.delaxes(self.figure.axes[1])
	846	except:
	847	pass
	848
	849	def _onProperties(self, event):
	850	"""
	851	when clicking on Properties on context menu ,
	852	The Property dialog is displayed
	853	The user selects a transformation for x or y value and
	854	a new plot is displayed
	855	"""
	856	list = []
	857	list = self.graph.returnPlottable()
	858	if len(list.keys()) > 0:
	859	first_item = list.keys()[0]
	860	if first_item.x != []:
	861	from sans.plottools.PropertyDialog import Properties
	862	dial = Properties(self, -1, 'Change Scale')
	863	# type of view or model used
	864	dial.xvalue.Clear()
	865	dial.yvalue.Clear()
	866	dial.view.Clear()
	867	dial.xvalue.Insert("x", 0)
	868	dial.xvalue.Insert("log10(x)", 1)
	869	dial.yvalue.Insert("y", 0)
	870	dial.yvalue.Insert("log10(y)", 1)
	871	dial.view.Insert("--", 0)
	872	dial.view.Insert("Linear y vs x", 1)
	873	dial.setValues(self.prevXtrans, self.prevYtrans, self.viewModel)
	874	dial.Update()
	875	if dial.ShowModal() == wx.ID_OK:
	876	self.xLabel, self.yLabel, self.viewModel = dial.getValues()
	877	if self.viewModel == "Linear y vs x":
	878	self.xLabel = "x"
	879	self.yLabel = "y"
	880	self.viewModel = "--"
	881	dial.setValues(self.xLabel, self.yLabel, self.viewModel)
	882	self._onEVT_FUNC_PROPERTY()
	883	dial.Destroy()
	884
	885	def _onEVT_FUNC_PROPERTY(self, remove_fit=True):
	886	"""
	887	Receive the x and y transformation from myDialog,
	888	Transforms x and y in View
	889	and set the scale
	890	"""
	891	list = []
	892	list = self.graph.returnPlottable()
	893	# Changing the scale might be incompatible with
	894	# currently displayed data (for instance, going
	895	# from ln to log when all plotted values have
	896	# negative natural logs).
	897	# Go linear and only change the scale at the end.
	898	self.set_xscale("linear")
	899	self.set_yscale("linear")
	900	_xscale = 'linear'
	901	_yscale = 'linear'
	902	for item in list:
	903	item.setLabel(self.xLabel, self.yLabel)
	904	# control axis labels from the panel itself
	905	yname, yunits = item.get_yaxis()
	906	xname, xunits = item.get_xaxis()
	907	# Goes through all possible scales
	908	# Goes through all possible scales
	909	if(self.xLabel == "x"):
	910	item.transformX(transform.toX, transform.errToX)
	911	self.graph._xaxis_transformed("%s" % xname, "%s" % xunits)
	912	if(self.xLabel == "log10(x)"):
	913	item.transformX(transform.toX_pos, transform.errToX_pos)
	914	_xscale = 'log'
	915	self.graph._xaxis_transformed("%s" % xname, "%s" % xunits)
	916	if(self.yLabel == "y"):
	917	item.transformY(transform.toX, transform.errToX)
	918	self.graph._yaxis_transformed("%s" % yname, "%s" % yunits)
	919	if(self.yLabel == "log10(y)"):
	920	item.transformY(transform.toX_pos, transform.errToX_pos)
	921	_yscale = 'log'
	922	self.graph._yaxis_transformed("%s" % yname, "%s" % yunits)
	923	item.transformView()
	924	self.prevXtrans = self.xLabel
	925	self.prevYtrans = self.yLabel
	926	self.set_xscale(_xscale)
	927	self.set_yscale(_yscale)
	928	self.draw()
	929
	930	class DataOperatorWindow(widget.CHILD_FRAME):
	931	def __init__(self, parent, manager, args, *kwds):
	932	kwds["size"] = (PANEL_WIDTH, PANEL_HEIGTH)
	933	widget.CHILD_FRAME.__init__(self, parent, args, *kwds)
	934	self.parent = parent
	935	self.manager = manager
	936	self.panel = DataOperPanel(parent=self)
	937	wx.EVT_CLOSE(self, self.OnClose)
	938	self.SetPosition((25, 150))
	939	self.Show()
	940
	941	def OnClose(self, event=None):
	942	"""
	943	On close event
	944	"""
	945	if self.manager != None:
	946	self.manager.data_operator_frame = None
	947	self.panel.disconnect_panels()
	948	self.Destroy()
	949
	950
	951	if __name__ == "__main__":
	952
	953	app = wx.App()
	954	widget.CHILD_FRAME = wx.Frame
	955	window = DataOperatorWindow(parent=None, data=[], title="Data Editor")
	956	app.MainLoop()
	957

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

SasView

source: sasview/src/sans/perspectives/calculator/data_operator.py @ 786685e

Download in other formats: