Plotter1D.py @ 64beda4

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 64beda4 was f468791, checked in by Mathieu Doucet <doucetm@…>, 11 years ago
Move plottools under sans
Property mode set to `100644`
File size: 33.2 KB

Line
1
2	################################################################################
3	#This software was developed by the University of Tennessee as part of the
4	#Distributed Data Analysis of Neutron Scattering Experiments (DANSE)
5	#project funded by the US National Science Foundation.
6	#
7	#See the license text in license.txt
8	#
9	#copyright 2008, University of Tennessee
10	################################################################################
11
12
13	import wx
14	import sys
15	import math
16	import numpy
17
18	from sans.plottools.PlotPanel import PlotPanel
19	from sans.guiframe.events import StatusEvent
20	from sans.guiframe.events import PanelOnFocusEvent
21	from sans.guiframe.utils import PanelMenu
22	from sans.guiframe.panel_base import PanelBase
23	from sans.guiframe.gui_style import GUIFRAME_ICON
24	from appearanceDialog import appearanceDialog
25	from graphAppearance import graphAppearance
26
27	DEFAULT_QMAX = 0.05
28	DEFAULT_QSTEP = 0.001
29	DEFAULT_BEAM = 0.005
30	BIN_WIDTH = 1
31	IS_MAC = (sys.platform == 'darwin')
32
33
34	def find_key(dic, val):
35	"""return the key of dictionary dic given the value"""
36	return [k for k, v in dic.iteritems() if v == val][0]
37
38
39
40	class ModelPanel1D(PlotPanel, PanelBase):
41	"""
42	Plot panel for use with the GUI manager
43	"""
44
45	## Internal name for the AUI manager
46	window_name = "plotpanel"
47	## Title to appear on top of the window
48	window_caption = "Graph"
49	## Flag to tell the GUI manager that this panel is not
50	# tied to any perspective
51	ALWAYS_ON = True
52	## Group ID
53	group_id = None
54
55	def __init__(self, parent, id=-1, color = None,
56	dpi=None, style=wx.NO_FULL_REPAINT_ON_RESIZE, **kwargs):
57	PlotPanel.__init__(self, parent, id=id, style=style, **kwargs)
58	PanelBase.__init__(self, parent)
59	## Reference to the parent window
60	self.parent = parent
61	if hasattr(parent, "parent"):
62	self.parent = self.parent.parent
63	## Plottables
64	self.plots = {}
65	self.frame = None
66	#context menu
67	self._slicerpop = None
68
69	self._available_data = []
70	self._menu_add_ids = []
71	self._symbol_labels = self.get_symbol_label()
72	self._color_labels = self.get_color_label()
73	self.currColorIndex = ""
74	self._is_changed_legend_label = False
75	self.is_xtick = False
76	self.is_ytick = False
77
78	self.hide_menu = None
79	## Unique ID (from gui_manager)
80	self.uid = None
81	self.x_size = None
82	## Default locations
83	#self._default_save_location = os.getcwd()
84	self.size = None
85	self.vl_ind = 0
86	## Graph
87	#self.graph = Graph()
88	self.graph.xaxis("\\rm{Q}", 'A^{-1}')
89	self.graph.yaxis("\\rm{Intensity} ", "cm^{-1}")
90	self.graph.render(self)
91	self.cursor_id = None
92
93	# In resizing event
94	self.resizing = False
95	self.canvas.set_resizing(self.resizing)
96	self.Bind(wx.EVT_SIZE, self._OnReSize)
97	self._add_more_tool()
98	self.parent.SetFocus()
99
100
101	def get_symbol_label(self):
102	"""
103	Associates label to symbol
104	"""
105	_labels = {}
106	i = 0
107	_labels['Circle'] = i
108	i += 1
109	_labels['Cross X '] = i
110	i += 1
111	_labels['Triangle Down'] = i
112	i += 1
113	_labels['Triangle Up'] = i
114	i += 1
115	_labels['Triangle Left'] = i
116	i += 1
117	_labels['Triangle Right'] = i
118	i += 1
119	_labels['Cross +'] = i
120	i += 1
121	_labels['Square'] = i
122	i += 1
123	_labels['diamond'] = i
124	i += 1
125	_labels['Diamond'] = i
126	i += 1
127	_labels['Hexagon1'] = i
128	i += 1
129	_labels['Hexagon2'] = i
130	i += 1
131	_labels['Pentagon'] = i
132	i += 1
133	_labels['Line'] = i
134	i += 1
135	_labels['Dash'] = i
136	i += 1
137	_labels['Vline'] = i
138	i += 1
139	_labels['Step'] = i
140	return _labels
141
142	def get_color_label(self):
143	"""
144	Associates label to a specific color
145	"""
146	_labels = {}
147	i = 0
148	_labels['Blue'] = i
149	i += 1
150	_labels['Green'] = i
151	i += 1
152	_labels['Red'] = i
153	i += 1
154	_labels['Cyan'] = i
155	i += 1
156	_labels['Magenta'] = i
157	i += 1
158	_labels['Yellow'] = i
159	i += 1
160	_labels['Black'] = i
161	return _labels
162
163
164	def set_data(self, list=None):
165	"""
166	"""
167	pass
168
169	def _reset(self):
170	"""
171	Resets internal data and graph
172	"""
173	self.graph.reset()
174	self.plots = {}
175	if self.is_zoomed:
176	self.is_zoomed = False
177
178	def _OnReSize(self, event):
179	"""
180	On response of the resize of a panel, set axes_visiable False
181	"""
182	# It was found that wx >= 2.9.3 sends an event even if no size changed.
183	# So manually recode the size (=x_size) and compare here.
184	# Massy code to work around:<
185	if self.parent._mgr != None:
186	max_panel = self.parent._mgr.GetPane(self)
187	if max_panel.IsMaximized():
188	self.parent._mgr.RestorePane(max_panel)
189	max_panel.Maximize()
190	if self.x_size != None:
191	if self.x_size == self.GetSize():
192	self.resizing = False
193	self.canvas.set_resizing(self.resizing)
194	return
195	self.x_size = self.GetSize()
196
197	# Ready for another event
198	# Do not remove this Skip. Otherwise it will get runtime error on wx>=2.9.
199	event.Skip()
200	# set the resizing flag
201	self.resizing = True
202	self.canvas.set_resizing(self.resizing)
203	self.parent.set_schedule(True)
204	pos_x, pos_y = self.GetPositionTuple()
205	if pos_x != 0 and pos_y != 0:
206	self.size, _ = self.GetClientSizeTuple()
207	self.SetSizer(self.sizer)
208	wx.CallAfter(self.parent.disable_app_menu,self)
209
210	def on_plot_qrange(self, event=None):
211	"""
212	On Qmin Qmax vertical line event
213	"""
214	if event == None:
215	return
216	event.Skip()
217	active_ctrl = event.active
218	if active_ctrl == None:
219	return
220	if event.id in self.plots.keys():
221	# Set line position and color
222	colors = ['red', 'purple']
223	self.cursor_id = event.id
224	ctrl = event.ctrl
225	if self.ly == None:
226	self.ly = []
227	for ind_ly in range(len(colors)):
228	self.ly.append(self.subplot.axvline(color=colors[ind_ly],
229	lw=2.5, alpha=0.7))
230	self.ly[ind_ly].set_rasterized(True)
231	try:
232	# Display x,y in the status bar if possible
233	xval = float(active_ctrl.GetValue())
234	position = self.get_data_xy_vals(xval)
235	if position != None:
236	wx.PostEvent(self.parent, StatusEvent(status=position))
237	except:
238	pass
239	if not event.leftdown:
240	# text event
241	try:
242	is_moved = False
243	for idx in range(len(self.ly)):
244	val = float(ctrl[idx].GetValue())
245	# check if vline moved
246	if self.ly[idx].get_xdata() != val:
247	self.ly[idx].set_xdata(val)
248	is_moved = True
249	if is_moved:
250	self.canvas.draw()
251	except:
252	pass
253	event.Skip()
254	return
255	self.q_ctrl = ctrl
256	try:
257	pos_x_min = float(self.q_ctrl[0].GetValue())
258	except:
259	pos_x_min = xmin
260	try:
261	pos_x_max = float(self.q_ctrl[1].GetValue())
262	except:
263	pos_x_max = xmax
264	pos_x = [pos_x_min, pos_x_max]
265	for ind_ly in range(len(colors)):
266	self.ly[ind_ly].set_color(colors[ind_ly])
267	self.ly[ind_ly].set_xdata(pos_x[ind_ly])
268	self.canvas.draw()
269	else:
270	self.q_ctrl = None
271
272	def get_data_xy_vals(self, xval):
273	"""
274	Get x, y data values near x = x_val
275	"""
276	try:
277	x_data = self.plots[self.cursor_id].x
278	y_data = self.plots[self.cursor_id].y
279	indx = self._find_nearest(x_data, xval)
280	pos_x = x_data[indx]
281	pos_y = y_data[indx]
282	position = str(pos_x), str(pos_y)
283	return position
284	except:
285	return None
286
287	def _find_nearest(self, array, value):
288	"""
289	Find and return the nearest value in array to the value.
290	Used in cusor_line()
291	:Param array: numpy array
292	:Param value: float
293	"""
294	idx = (numpy.abs(array - value)).argmin()
295	return int(idx)#array.flat[idx]
296
297	def _check_line_positions(self, pos_x=None, nop=None):
298	"""
299	Check vertical line positions
300	:Param pos_x: position of the current line [float]
301	:Param nop: number of plots [int]
302	"""
303	ly = self.ly
304	ly0x = ly[0].get_xdata()
305	ly1x = ly[1].get_xdata()
306	self.q_ctrl[0].SetBackgroundColour('white')
307	self.q_ctrl[1].SetBackgroundColour('white')
308	if ly0x >= ly1x:
309	if self.vl_ind == 0:
310	ly[1].set_xdata(pos_x)
311	ly[1].set_zorder(nop)
312	self.q_ctrl[1].SetValue(str(pos_x))
313	self.q_ctrl[0].SetBackgroundColour('pink')
314	elif self.vl_ind == 1:
315	ly[0].set_xdata(pos_x)
316	ly[0].set_zorder(nop)
317	self.q_ctrl[0].SetValue(str(pos_x))
318	self.q_ctrl[1].SetBackgroundColour('pink')
319
320	def _get_cusor_lines(self, event):
321	"""
322	Revmove or switch cursor line if drawn
323	:Param event: LeftClick mouse event
324	"""
325	ax = event.inaxes
326	if hasattr(event, "action"):
327	dclick = event.action == 'dclick'
328	if ax == None or dclick:
329	# remove the vline
330	self._check_zoom_plot()
331	self.canvas.draw()
332	self.q_ctrl = None
333	return
334	if self.ly != None and event.xdata != None:
335	# Selecting a new line if cursor lines are displayed already
336	dqmin = math.fabs(event.xdata - self.ly[0].get_xdata())
337	dqmax = math.fabs(event.xdata - self.ly[1].get_xdata())
338	is_qmax = dqmin > dqmax
339	if is_qmax:
340	self.vl_ind = 1
341	else:
342	self.vl_ind = 0
343
344	def cusor_line(self, event):
345	"""
346	Move the cursor line to write Q range
347	"""
348	if self.q_ctrl == None:
349	return
350	#release a q range vline
351	if self.ly != None and not self.leftdown:
352	for ly in self.ly:
353	ly.set_alpha(0.7)
354	self.canvas.draw()
355	return
356	ax = event.inaxes
357	if ax == None or not hasattr(event, 'action'):
358	return
359	end_drag = event.action != 'drag' and event.xdata != None
360	nop = len(self.plots)
361	pos_x, pos_y = float(event.xdata), float(event.ydata)
362	try:
363	ly = self.ly
364	ly0x = ly[0].get_xdata()
365	ly1x = ly[1].get_xdata()
366	if ly0x == ly1x:
367	if ly[0].get_zorder() > ly[1].get_zorder():
368	self.vl_ind = 0
369	else:
370	self.vl_ind = 1
371	vl_ind = self.vl_ind
372	x_data = self.plots[self.cursor_id].x
373	y_data = self.plots[self.cursor_id].y
374	xmin = x_data.min()
375	xmax = x_data.max()
376	indx = self._find_nearest(x_data, pos_x)
377	#pos_x = self._find_nearest(x_data, pos_x)
378	#indx = int(numpy.searchsorted(x_data, [pos_x])[0])
379	# Need to hold LeftButton to drag
380	if end_drag:
381	if event.button:
382	self._check_line_positions(pos_x, nop)
383	return
384	if indx >= len(x_data):
385	indx = len(x_data) - 1
386	pos_x = x_data[indx]
387	pos_y = y_data[indx]
388	if xmin == ly1x:
389	vl_ind = 1
390	elif xmax == ly0x:
391	vl_ind = 0
392	else:
393	ly[vl_ind].set_xdata(pos_x)
394	ly[vl_ind].set_zorder(nop + 1)
395	self._check_line_positions(pos_x, nop)
396	ly[vl_ind].set_xdata(pos_x)
397	ly[vl_ind].set_alpha(1.0)
398	ly[vl_ind].set_zorder(nop + 1)
399	self.canvas.draw()
400	self.q_ctrl[vl_ind].SetValue(str(pos_x))
401	except:
402	pass
403
404	def set_resizing(self, resizing=False):
405	"""
406	Set the resizing (True/False)
407	"""
408	self.resizing = resizing
409	#self.canvas.set_resizing(resizing)
410
411	def schedule_full_draw(self, func='append'):
412	"""
413	Put self in schedule to full redraw list
414	"""
415	# append/del this panel in the schedule list
416	self.parent.set_schedule_full_draw(self, func)
417
418
419	def remove_data_by_id(self, id):
420	"""'
421	remove data from plot
422	"""
423	if id in self.plots.keys():
424	data = self.plots[id]
425	self.graph.delete(data)
426	data_manager = self._manager.parent.get_data_manager()
427	data_list, theory_list = data_manager.get_by_id(id_list=[id])
428
429	if id in data_list.keys():
430	data = data_list[id]
431	if id in theory_list.keys():
432	data = theory_list[id]
433	# Update Graph menu and help string
434	#h_id = self.parent._window_menu.FindItem(self.window_caption)
435	if data != None:
436	if data.__class__.__name__ == 'list':
437	label = data[0].label
438	else:
439	label = data.label
440	else:
441	label = '???'
442	#helpString = self.parent._window_menu.GetHelpString(h_id)
443	d_string = (' ' + str(label) +';')
444	#new_tip = helpString.replace(d_string, '')
445	#self.parent._window_menu.SetHelpString(h_id, new_tip)
446
447	del self.plots[id]
448	self.graph.render(self)
449	self.subplot.figure.canvas.draw_idle()
450	if len(self.graph.plottables) == 0:
451	#onRemove: graph is empty must be the panel must be destroyed
452	self.parent.delete_panel(self.uid)
453
454
455	def plot_data(self, data):
456	"""
457	Data is ready to be displayed
458
459	:param event: data event
460	"""
461	if data.__class__.__name__ == 'Data2D':
462	return
463	plot_keys = self.plots.keys()
464	if data.id in plot_keys:
465	#Recover panel prop.s
466	xlo, xhi = self.subplot.get_xlim()
467	ylo, yhi = self.subplot.get_ylim()
468	old_data = self.plots[data.id]
469	if self._is_changed_legend_label:
470	data.label = old_data.label
471	if old_data.__class__.__name__ == 'Data1D':
472	data.custom_color = old_data.custom_color
473	data.symbol = old_data.symbol
474	data.markersize = old_data.markersize
475	data.zorder = len(plot_keys)
476	# Replace data
477	self.graph.replace(data)
478	self.plots[data.id] = data
479	## Set the view scale for all plots
480	try:
481	self._onEVT_FUNC_PROPERTY()
482	except:
483	msg=" Encountered singular points..."
484	wx.PostEvent(self.parent, StatusEvent(status=\
485	"Plotting Error: %s"% msg, info="error"))
486	# Check if zoomed
487	toolbar_zoomed = self.toolbar.GetToolEnabled(self.toolbar._NTB2_BACK)
488	if self.is_zoomed or toolbar_zoomed:
489	# Recover the x,y limits
490	self.subplot.set_xlim((xlo, xhi))
491	self.subplot.set_ylim((ylo, yhi))
492	else:
493	self.plots[data.id] = data
494	self.graph.add(self.plots[data.id])
495	data.zorder = len(plot_keys)
496	## Set the view scale for all plots
497	try:
498	self._onEVT_FUNC_PROPERTY()
499	if IS_MAC:
500	# MAC: forcing to plot 2D avg
501	self.canvas._onDrawIdle()
502	except:
503	msg=" Encountered singular points..."
504	wx.PostEvent(self.parent, StatusEvent(status=\
505	"Plotting Error: %s"% msg, info="error"))
506	self.toolbar.update()
507	if self.is_zoomed:
508	self.is_zoomed = False
509	# Update Graph menu and help string
510	#pos = self.parent._window_menu.FindItem(self.window_caption)
511	helpString = 'Show/Hide Graph: '
512	for plot in self.plots.itervalues():
513	helpString += (' ' + str(plot.label) +';')
514	#self.parent._window_menu.SetHelpString(pos, helpString)
515
516	def draw_plot(self):
517	"""
518	Draw plot
519	"""
520	self.draw()
521
522	def onLeftDown(self,event):
523	"""
524	left button down and ready to drag
525	Display the position of the mouse on the statusbar
526	"""
527	#self.parent.set_plot_unfocus()
528	self._get_cusor_lines(event)
529	ax = event.inaxes
530	PlotPanel.onLeftDown(self, event)
531	if ax != None:
532	try:
533	pos_x = float(event.xdata)# / size_x
534	pos_y = float(event.ydata)# / size_y
535	pos_x = "%8.3g"% pos_x
536	pos_y = "%8.3g"% pos_y
537	self.position = str(pos_x), str(pos_y)
538	wx.PostEvent(self.parent, StatusEvent(status=self.position))
539	except:
540	self.position = None
541	# unfocus all
542	self.parent.set_plot_unfocus()
543	#post nd event to notify guiframe that this panel is on focus
544	wx.PostEvent(self.parent, PanelOnFocusEvent(panel=self))
545
546
547	def _ontoggle_hide_error(self, event):
548	"""
549	Toggle error display to hide or show
550	"""
551	menu = event.GetEventObject()
552	id = event.GetId()
553	self.set_selected_from_menu(menu, id)
554	# Check zoom
555	xlo, xhi = self.subplot.get_xlim()
556	ylo, yhi = self.subplot.get_ylim()
557
558	selected_plot = self.plots[self.graph.selected_plottable]
559	if self.hide_menu.GetText() == "Hide Error Bar":
560	selected_plot.hide_error = True
561	else:
562	selected_plot.hide_error = False
563	## increment graph color
564	self.graph.render(self)
565	self.subplot.figure.canvas.draw_idle()
566	# Check if zoomed
567	toolbar_zoomed = self.toolbar.GetToolEnabled(self.toolbar._NTB2_BACK)
568	if self.is_zoomed or toolbar_zoomed:
569	# Recover the x,y limits
570	self.subplot.set_xlim((xlo, xhi))
571	self.subplot.set_ylim((ylo, yhi))
572
573
574	def _onRemove(self, event):
575	"""
576	Remove a plottable from the graph and render the graph
577
578	:param event: Menu event
579
580	"""
581	menu = event.GetEventObject()
582	id = event.GetId()
583	self.set_selected_from_menu(menu, id)
584	## Check if there is a selected graph to remove
585	if self.graph.selected_plottable in self.plots.keys():
586	selected_plot = self.plots[self.graph.selected_plottable]
587	id = self.graph.selected_plottable
588	self.remove_data_by_id(id)
589
590	def onContextMenu(self, event):
591	"""
592	1D plot context menu
593
594	:param event: wx context event
595
596	"""
597	self._slicerpop = PanelMenu()
598	self._slicerpop.set_plots(self.plots)
599	self._slicerpop.set_graph(self.graph)
600	if not self.graph.selected_plottable in self.plots:
601	# Various plot options
602	id = wx.NewId()
603	self._slicerpop.Append(id, '&Save Image', 'Save image as PNG')
604	wx.EVT_MENU(self, id, self.onSaveImage)
605	id = wx.NewId()
606	self._slicerpop.Append(id, '&Print Image', 'Print image ')
607	wx.EVT_MENU(self, id, self.onPrint)
608	id = wx.NewId()
609	self._slicerpop.Append(id, '&Print Preview', 'Print preview')
610	wx.EVT_MENU(self, id, self.onPrinterPreview)
611
612	id = wx.NewId()
613	self._slicerpop.Append(id, '&Copy to Clipboard',
614	'Copy to the clipboard')
615	wx.EVT_MENU(self, id, self.OnCopyFigureMenu)
616
617	self._slicerpop.AppendSeparator()
618
619	for plot in self.plots.values():
620	#title = plot.title
621	name = plot.name
622	plot_menu = wx.Menu()
623	if self.graph.selected_plottable:
624	if not self.graph.selected_plottable in self.plots.keys():
625	continue
626	if plot != self.plots[self.graph.selected_plottable]:
627	continue
628
629	id = wx.NewId()
630	plot_menu.Append(id, "&DataInfo", name)
631	wx.EVT_MENU(self, id, self. _onDataShow)
632	id = wx.NewId()
633	plot_menu.Append(id, "&Save Points as a File", name)
634	wx.EVT_MENU(self, id, self._onSave)
635	plot_menu.AppendSeparator()
636
637	#add menu of other plugins
638	item_list = self.parent.get_current_context_menu(self)
639
640	if (not item_list == None) and (not len(item_list) == 0):
641	for item in item_list:
642
643	try:
644	id = wx.NewId()
645	plot_menu.Append(id, item[0], name)
646	wx.EVT_MENU(self, id, item[2])
647	except:
648	msg = "ModelPanel1D.onContextMenu: "
649	msg += "bad menu item %s" % sys.exc_value
650	wx.PostEvent(self.parent, StatusEvent(status=msg))
651	pass
652	plot_menu.AppendSeparator()
653
654	if self.parent.ClassName.count('wxDialog') == 0:
655	id = wx.NewId()
656	plot_menu.Append(id, '&Linear Fit', name)
657	wx.EVT_MENU(self, id, self.onFitting)
658	plot_menu.AppendSeparator()
659
660	id = wx.NewId()
661	plot_menu.Append(id, "Remove", name)
662	wx.EVT_MENU(self, id, self._onRemove)
663	if not plot.is_data:
664	id = wx.NewId()
665	plot_menu.Append(id, '&Freeze', name)
666	wx.EVT_MENU(self, id, self.onFreeze)
667	plot_menu.AppendSeparator()
668	symbol_menu = wx.Menu()
669
670	if plot.is_data:
671	id = wx.NewId()
672	self.hide_menu = plot_menu.Append(id,
673	"Hide Error Bar", name)
674
675	if plot.dy is not None and plot.dy != []:
676	if plot.hide_error :
677	self.hide_menu.SetText('Show Error Bar')
678	else:
679	self.hide_menu.SetText('Hide Error Bar')
680	else:
681	self.hide_menu.Enable(False)
682	wx.EVT_MENU(self, id, self._ontoggle_hide_error)
683
684	plot_menu.AppendSeparator()
685
686	id = wx.NewId()
687	plot_menu.Append(id, '&Modify Plot Property', name)
688	wx.EVT_MENU(self, id, self.createAppDialog)
689
690
691
692	id = wx.NewId()
693	#plot_menu.SetTitle(name)
694	self._slicerpop.AppendMenu(id, '&%s'% name, plot_menu)
695	# Option to hide
696	#TODO: implement functionality to hide a plottable (legend click)
697	if not self.graph.selected_plottable in self.plots:
698	self._slicerpop.AppendSeparator()
699	loc_menu = wx.Menu()
700	for label in self._loc_labels:
701	id = wx.NewId()
702	loc_menu.Append(id, str(label), str(label))
703	wx.EVT_MENU(self, id, self.onChangeLegendLoc)
704
705	id = wx.NewId()
706	self._slicerpop.Append(id, '&Modify Graph Appearance',
707	'Modify graph appearance')
708	wx.EVT_MENU(self, id, self.modifyGraphAppearance)
709	self._slicerpop.AppendSeparator()
710
711
712	if self.position != None:
713	id = wx.NewId()
714	self._slicerpop.Append(id, '&Add Text')
715	wx.EVT_MENU(self, id, self._on_addtext)
716	id = wx.NewId()
717	self._slicerpop.Append(id, '&Remove Text')
718	wx.EVT_MENU(self, id, self._on_removetext)
719	self._slicerpop.AppendSeparator()
720	id = wx.NewId()
721	self._slicerpop.Append(id, '&Change Scale')
722	wx.EVT_MENU(self, id, self._onProperties)
723	self._slicerpop.AppendSeparator()
724	id = wx.NewId()
725	self._slicerpop.Append(id, '&Reset Graph Range')
726	wx.EVT_MENU(self, id, self.onResetGraph)
727
728	if self.parent.ClassName.count('wxDialog') == 0:
729	self._slicerpop.AppendSeparator()
730	id = wx.NewId()
731	self._slicerpop.Append(id, '&Window Title')
732	wx.EVT_MENU(self, id, self.onChangeCaption)
733	try:
734	pos_evt = event.GetPosition()
735	pos = self.ScreenToClient(pos_evt)
736	except:
737	pos_x, pos_y = self.toolbar.GetPositionTuple()
738	pos = (pos_x, pos_y + 5)
739	self.PopupMenu(self._slicerpop, pos)
740
741	def onFreeze(self, event):
742	"""
743	on Freeze data
744	"""
745	menu = event.GetEventObject()
746	id = event.GetId()
747	self.set_selected_from_menu(menu, id)
748	plot = self.plots[self.graph.selected_plottable]
749	self.parent.onfreeze([plot.id])
750
751
752	def _onSave(self, evt):
753	"""
754	Save a data set to a text file
755
756	:param evt: Menu event
757
758	"""
759	menu = evt.GetEventObject()
760	id = evt.GetId()
761	self.set_selected_from_menu(menu, id)
762	data = self.plots[self.graph.selected_plottable]
763	default_name = data.label
764	if default_name.count('.') > 0:
765	default_name = default_name.split('.')[0]
766	default_name += "_out"
767	if self.parent != None:
768	self.parent.save_data1d(data, default_name)
769
770
771	def _onDataShow(self, evt):
772	"""
773	Show the data set in text
774
775	:param evt: Menu event
776
777	"""
778	menu = evt.GetEventObject()
779	id = evt.GetId()
780	self.set_selected_from_menu(menu, id)
781	data = self.plots[self.graph.selected_plottable]
782	default_name = data.label
783	if default_name.count('.') > 0:
784	default_name = default_name.split('.')[0]
785	#default_name += "_out"
786	if self.parent != None:
787	self.parent.show_data1d(data, default_name)
788
789	def _add_more_tool(self):
790	"""
791	Add refresh, add/hide button in the tool bar
792	"""
793	return
794	if self.parent.__class__.__name__ != 'ViewerFrame':
795	return
796	self.toolbar.AddSeparator()
797	id_hide = wx.NewId()
798	hide = wx.Bitmap(GUIFRAME_ICON.HIDE_ID_PATH, wx.BITMAP_TYPE_PNG)
799	self.toolbar.AddSimpleTool(id_hide, hide, 'Hide', 'Hide')
800	self.toolbar.Realize()
801	wx.EVT_TOOL(self, id_hide, self._on_hide)
802
803	def _on_hide(self, event):
804	"""
805	Hides the plot when button is pressed
806	"""
807	if self.parent is not None:
808	self.parent.hide_panel(self.uid)
809
810	def on_close(self, event):
811	"""
812	On Close Event
813	"""
814	ID = self.uid
815	self.parent.delete_panel(ID)
816
817	def createAppDialog(self, event):
818	"""
819	Create the custom dialog for fit appearance modification
820	"""
821	menu = event.GetEventObject()
822	id = event.GetId()
823	self.set_selected_from_menu(menu, id)
824	self.appearance_selected_plot = \
825	self.plots[self.graph.selected_plottable]
826	# find current properties
827	curr_color = self.appearance_selected_plot.custom_color
828	curr_symbol = self.appearance_selected_plot.symbol
829	curr_size = self.appearance_selected_plot.markersize
830	curr_label = self.appearance_selected_plot.label
831
832	if curr_color == None:
833	curr_color = self._color_labels['Blue']
834	curr_symbol = 13
835
836	self.appD = appearanceDialog(self, 'Modify Plot Property')
837	icon = self.parent.GetIcon()
838	self.appD.SetIcon(icon)
839	self.appD.set_defaults(float(curr_size), int(curr_color),
840	str(appearanceDialog.find_key(self.get_symbol_label(),
841	int(curr_symbol))), curr_label)
842	self.appD.Bind(wx.EVT_CLOSE, self.on_AppDialog_close)
843
844	def on_AppDialog_close(self, event):
845	"""
846	on_Modify Plot Property_close
847	"""
848	if(self.appD.okay_clicked == True):
849	# returns (size,color,symbol,datalabel)
850	info = self.appD.get_current_values()
851	self.appearance_selected_plot.custom_color = \
852	self._color_labels[info[1].encode('ascii', 'ignore')]
853
854	self.appearance_selected_plot.markersize = float(info[0])
855	self.appearance_selected_plot.symbol = \
856	self.get_symbol_label()[info[2]]
857	self.appearance_selected_plot.label = str(info[3])
858
859	#pos = self.parent._window_menu.FindItem(self.window_caption)
860	#helpString = 'Show/Hide Graph: '
861	#for plot in self.plots.itervalues():
862	# helpString += (' ' + str(plot.label) + ';')
863	# self.parent._window_menu.SetHelpString(pos, helpString)
864	# self._is_changed_legend_label = True
865
866	self.appD.Destroy()
867	self._check_zoom_plot()
868
869
870	def modifyGraphAppearance(self, event):
871	"""
872	On Modify Graph Appearance
873	"""
874	self.graphApp = graphAppearance(self, 'Modify Graph Appearance')
875	icon = self.parent.GetIcon()
876	self.graphApp.SetIcon(icon)
877	self.graphApp.setDefaults(self.grid_on, self.legend_on,
878	self.xaxis_label, self.yaxis_label,
879	self.xaxis_unit, self.yaxis_unit,
880	self.xaxis_font, self.yaxis_font,
881	find_key(self.get_loc_label(),
882	self.legendLoc),
883	self.xcolor, self.ycolor,
884	self.is_xtick, self.is_ytick)
885	self.graphApp.Bind(wx.EVT_CLOSE, self.on_graphApp_close)
886
887
888	def on_graphApp_close(self, event):
889	"""
890	Gets values from graph appearance dialog and sends them off
891	to modify the plot
892	"""
893	graph_app = self.graphApp
894	toggle_grid = graph_app.get_togglegrid()
895	legend_loc = graph_app.get_legend_loc()
896	toggle_legend = graph_app.get_togglelegend()
897
898	self.onGridOnOff(toggle_grid )
899	self.ChangeLegendLoc(legend_loc)
900	self.onLegend(toggle_legend)
901
902	self.xaxis_label = graph_app.get_xlab()
903	self.yaxis_label = graph_app.get_ylab()
904	self.xaxis_unit = graph_app.get_xunit()
905	self.yaxis_unit = graph_app.get_yunit()
906	self.xaxis_font = graph_app.get_xfont()
907	self.yaxis_font = graph_app.get_yfont()
908	self.is_xtick = graph_app.get_xtick_check()
909	self.is_ytick = graph_app.get_ytick_check()
910	if self.is_xtick:
911	self.xaxis_tick = self.xaxis_font
912	if self.is_ytick:
913	self.yaxis_tick = self.yaxis_font
914
915	self.xaxis(self.xaxis_label, self.xaxis_unit,
916	graph_app.get_xfont(), graph_app.get_xcolor(),
917	self.xaxis_tick)
918	self.yaxis(self.yaxis_label, self.yaxis_unit,
919	graph_app.get_yfont(), graph_app.get_ycolor(),
920	self.yaxis_tick)
921
922	graph_app.Destroy()

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

SasView

source: sasview/src/sans/guiframe/local_perspectives/plotting/Plotter1D.py @ 64beda4

Download in other formats: