Plotter1D.py @ 1fc5e10a

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 1fc5e10a was 1fc5e10a, checked in by Jae Cho <jhjcho@…>, 13 years ago
another temp commit
Property mode set to `100644`
File size: 27.4 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 os
16	import pylab
17	import math
18	import numpy
19	import time
20
21	from danse.common.plottools.PlotPanel import PlotPanel
22	from danse.common.plottools.SizeDialog import SizeDialog
23	from danse.common.plottools.LabelDialog import LabelDialog
24	#from danse.common.plottools.plottables import Graph
25	from sans.guiframe import dataFitting
26	from sans.guiframe.events import EVT_NEW_PLOT
27	from sans.guiframe.events import StatusEvent
28	from sans.guiframe.events import NewPlotEvent
29	from sans.guiframe.events import NewColorEvent
30	from sans.guiframe.events import SlicerEvent
31	from sans.guiframe.events import PanelOnFocusEvent
32	from sans.guiframe.events import EVT_NEW_LOADED_DATA
33	from sans.guiframe.utils import PanelMenu
34	from sans.guiframe.dataFitting import Data1D
35	from sans.guiframe.panel_base import PanelBase
36	from binder import BindArtist
37
38	DEFAULT_QMAX = 0.05
39	DEFAULT_QSTEP = 0.001
40	DEFAULT_BEAM = 0.005
41	BIN_WIDTH = 1
42
43
44	class ModelPanel1D(PlotPanel, PanelBase):
45	"""
46	Plot panel for use with the GUI manager
47	"""
48
49	## Internal name for the AUI manager
50	window_name = "plotpanel"
51	## Title to appear on top of the window
52	window_caption = "Graph"
53	## Flag to tell the GUI manager that this panel is not
54	# tied to any perspective
55	ALWAYS_ON = True
56	## Group ID
57	group_id = None
58
59	def __init__(self, parent, id=-1, color = None,
60	dpi=None, style=wx.NO_FULL_REPAINT_ON_RESIZE, **kwargs):
61	PlotPanel.__init__(self, parent, id=id, style=style, **kwargs)
62	PanelBase.__init__(self, parent)
63	## Reference to the parent window
64	self.parent = parent
65	## Plottables
66	self.plots = {}
67	#context menu
68	self._slicerpop = None
69
70	self._available_data = []
71	self._menu_add_ids = []
72	self._symbol_labels = self.get_symbol_label()
73	self._color_labels = self.get_color_label()
74	self.currColorIndex = ""
75	self._is_changed_legend_label = False
76
77	self.hide_menu = None
78	## Unique ID (from gui_manager)
79	self.uid = None
80	self.x_size = None
81	## Default locations
82	#self._default_save_location = os.getcwd()
83	self.size = None
84	## Graph
85	#self.graph = Graph()
86	self.graph.xaxis("\\rm{Q}", 'A^{-1}')
87	self.graph.yaxis("\\rm{Intensity} ", "cm^{-1}")
88	self.graph.render(self)
89
90	# In resizing event
91	self.resizing = False
92	self.canvas.set_resizing(self.resizing)
93	self.Bind(wx.EVT_SIZE, self._OnReSize)
94	self._add_more_tool()
95	self.parent.SetFocus()
96
97	def get_symbol_label(self):
98	"""
99	Associates label to symbol
100	"""
101	_labels = {}
102	i = 0
103	_labels['Circle'] = i
104	i += 1
105	_labels['Cross X '] = i
106	i += 1
107	_labels['Triangle Down'] = i
108	i += 1
109	_labels['Triangle Up'] = i
110	i += 1
111	_labels['Triangle Left'] = i
112	i += 1
113	_labels['Triangle Right'] = i
114	i += 1
115	_labels['Cross +'] = i
116	i += 1
117	_labels['Square'] = i
118	i += 1
119	_labels['Diamond'] = i
120	i += 1
121	_labels['Diamond'] = i
122	i += 1
123	_labels['Hexagon1'] = i
124	i += 1
125	_labels['Hexagon2'] = i
126	i += 1
127	_labels['Pentagon'] = i
128	i += 1
129	_labels['Line'] = i
130	return _labels
131
132	def get_color_label(self):
133	"""
134	Associates label to a specific color
135	"""
136	_labels = {}
137	i = 0
138	_labels['Blue'] = i
139	i += 1
140	_labels['Green'] = i
141	i += 1
142	_labels['Red'] = i
143	i += 1
144	_labels['Cyan'] = i
145	i += 1
146	_labels['Magenta'] = i
147	i += 1
148	_labels['Yellow'] = i
149	i += 1
150	_labels['Black'] = i
151	return _labels
152
153
154	def set_data(self, list=None):
155	"""
156	"""
157	pass
158
159	def _reset(self):
160	"""
161	Resets internal data and graph
162	"""
163	self.graph.reset()
164	self.plots = {}
165	if self.is_zoomed:
166	self.is_zoomed = False
167
168	def _OnReSize(self, event):
169	"""
170	On response of the resize of a panel, set axes_visiable False
171	"""
172	# It was found that wx >= 2.9.3 sends an event even if no size changed.
173	# So manually recode the size (=x_size) and compare here.
174	if self.x_size != None:
175	if self.x_size == self.GetSize():
176	self.resizing = False
177	self.canvas.set_resizing(self.resizing)
178	return
179	self.x_size = self.GetSize()
180
181	# Ready for another event
182	# Do not remove this Skip. Otherwise it will get runtime error on wx>=2.9.
183	event.Skip()
184	# set the resizing flag
185	self.resizing = True
186	self.canvas.set_resizing(self.resizing)
187	self.parent.set_schedule(True)
188	pos_x, pos_y = self.GetPositionTuple()
189	if pos_x != 0 and pos_y != 0:
190	self.size, _ = self.GetClientSizeTuple()
191	self.SetSizer(self.sizer)
192
193	try:
194	self.figure.Update()
195	self.canvas.Update()
196	self.Update()
197	except:
198	pass
199
200	def set_resizing(self, resizing=False):
201	"""
202	Set the resizing (True/False)
203	"""
204	self.resizing = resizing
205	#self.canvas.set_resizing(resizing)
206
207	def schedule_full_draw(self, func='append'):
208	"""
209	Put self in schedule to full redraw list
210	"""
211	# append/del this panel in the schedule list
212	self.parent.set_schedule_full_draw(self, func)
213
214
215	def remove_data_by_id(self, id):
216	"""'
217	remove data from plot
218	"""
219	if id in self.plots.keys():
220	data = self.plots[id]
221	self.graph.delete(data)
222	data_manager = self._manager.parent.get_data_manager()
223	data_list, theory_list = data_manager.get_by_id(id_list=[id])
224
225	if id in data_list.keys():
226	data = data_list[id]
227	if id in theory_list.keys():
228	data = theory_list[id]
229
230	del self.plots[id]
231	self.graph.render(self)
232	self.subplot.figure.canvas.draw_idle()
233	if len(self.graph.plottables) == 0:
234	#onRemove: graph is empty must be the panel must be destroyed
235	self.parent.delete_panel(self.uid)
236
237
238	def plot_data(self, data):
239	"""
240	Data is ready to be displayed
241
242	:param event: data event
243	"""
244	if data.id in self.plots.keys():
245	#Recover panel prop.s
246	xlo, xhi = self.subplot.get_xlim()
247	ylo, yhi = self.subplot.get_ylim()
248	old_data = self.plots[data.id]
249	if self._is_changed_legend_label:
250	data.label = old_data.label
251	data.custom_color = old_data.custom_color
252	data.symbol = old_data.symbol
253	data.markersize = old_data.markersize
254	# Replace data
255	self.graph.replace(data)
256	self.plots[data.id] = data
257	## Set the view scale for all plots
258	try:
259	self._onEVT_FUNC_PROPERTY()
260	except:
261	msg=" Encountered singular points..."
262	wx.PostEvent(self.parent, StatusEvent(status=\
263	"Plotting Erorr: %s"% msg, info="error"))
264	# Check if zoomed
265	toolbar_zoomed = self.toolbar.GetToolEnabled(self.toolbar._NTB2_BACK)
266	if self.is_zoomed or toolbar_zoomed:
267	# Recover the x,y limits
268	self.subplot.set_xlim((xlo, xhi))
269	self.subplot.set_ylim((ylo, yhi))
270	# Update Graph menu and help string
271	pos = self.parent._window_menu.FindItem(self.window_caption)
272	helpString = 'Show/Hide Graph: '
273	for plot in self.plots.itervalues():
274	helpString += (' ' + str(plot.label) +';')
275	self.parent._window_menu.SetHelpString(pos, helpString)
276	else:
277	self.plots[data.id] = data
278	self.graph.add(self.plots[data.id])
279	## Set the view scale for all plots
280	try:
281	self._onEVT_FUNC_PROPERTY()
282	except:
283	msg=" Encountered singular points..."
284	wx.PostEvent(self.parent, StatusEvent(status=\
285	"Plotting Erorr: %s"% msg, info="error"))
286	self.toolbar.update()
287	if self.is_zoomed:
288	self.is_zoomed = False
289
290
291	def draw_plot(self):
292	"""
293	Draw plot
294	"""
295	self.draw()
296
297	def onLeftDown(self,event):
298	"""
299	left button down and ready to drag
300	Display the position of the mouse on the statusbar
301	"""
302	PlotPanel.onLeftDown(self, event)
303	ax = event.inaxes
304	if ax != None:
305	try:
306	pos_x = float(event.xdata)# / size_x
307	pos_y = float(event.ydata)# / size_y
308	pos_x = "%8.3g"% pos_x
309	pos_y = "%8.3g"% pos_y
310	self.position = str(pos_x), str(pos_y)
311	wx.PostEvent(self.parent, StatusEvent(status=self.position))
312	except:
313	self.position = None
314	# unfocus all
315	self.parent.set_plot_unfocus()
316	#post nd event to notify guiframe that this panel is on focus
317	wx.PostEvent(self.parent, PanelOnFocusEvent(panel=self))
318
319
320	def _ontoggle_hide_error(self, event):
321	"""
322	Toggle error display to hide or show
323	"""
324	# Check zoom
325	xlo, xhi = self.subplot.get_xlim()
326	ylo, yhi = self.subplot.get_ylim()
327
328	selected_plot = self.plots[self.graph.selected_plottable]
329	if self.hide_menu.GetText() == "Hide Error Bar":
330	selected_plot.hide_error = True
331	else:
332	selected_plot.hide_error = False
333	## increment graph color
334	self.graph.render(self)
335	self.subplot.figure.canvas.draw_idle()
336	# Check if zoomed
337	toolbar_zoomed = self.toolbar.GetToolEnabled(self.toolbar._NTB2_BACK)
338	if self.is_zoomed or toolbar_zoomed:
339	# Recover the x,y limits
340	self.subplot.set_xlim((xlo, xhi))
341	self.subplot.set_ylim((ylo, yhi))
342
343
344	def _onRemove(self, event):
345	"""
346	Remove a plottable from the graph and render the graph
347
348	:param event: Menu event
349
350	"""
351	## Check if there is a selected graph to remove
352	if self.graph.selected_plottable in self.plots.keys():
353	selected_plot = self.plots[self.graph.selected_plottable]
354	id = self.graph.selected_plottable
355	self.remove_data_by_id(id)
356
357	def onContextMenu(self, event):
358	"""
359	1D plot context menu
360
361	:param event: wx context event
362
363	"""
364	self._slicerpop = PanelMenu()
365	self._slicerpop.set_plots(self.plots)
366	self._slicerpop.set_graph(self.graph)
367	# Various plot options
368	id = wx.NewId()
369	self._slicerpop.Append(id, '&Save Image', 'Save image as PNG')
370	wx.EVT_MENU(self, id, self.onSaveImage)
371	id = wx.NewId()
372	self._slicerpop.Append(id, '&Print Image', 'Print image ')
373	wx.EVT_MENU(self, id, self.onPrint)
374	id = wx.NewId()
375	self._slicerpop.Append(id, '&Print Preview', 'Print preview')
376	wx.EVT_MENU(self, id, self.onPrinterPreview)
377
378	id = wx.NewId()
379	self._slicerpop.Append(id, '&Copy to Clipboard', 'Copy to the clipboard')
380	wx.EVT_MENU(self, id, self.OnCopyFigureMenu)
381
382	self._slicerpop.AppendSeparator()
383
384	#add menu of other plugins
385	item_list = self.parent.get_context_menu(self)
386
387	if (not item_list == None) and (not len(item_list) == 0):
388	for item in item_list:
389	try:
390	id = wx.NewId()
391	self._slicerpop.Append(id, item[0], item[1])
392	wx.EVT_MENU(self, id, item[2])
393	except:
394	msg = "ModelPanel1D.onContextMenu: "
395	msg += "bad menu item %s" % sys.exc_value
396	wx.PostEvent(self.parent, StatusEvent(status=msg))
397	pass
398	self._slicerpop.AppendSeparator()
399	#id = wx.NewId()
400	#self._slicerpop.Append(id, '&Print image', 'Print image')
401	if self.graph.selected_plottable in self.plots:
402	plot = self.plots[self.graph.selected_plottable]
403
404	id = wx.NewId()
405	name = plot.name
406	self._slicerpop.Append(id, "&Save Points as a File")
407	wx.EVT_MENU(self, id, self._onSave)
408	self._slicerpop.AppendSeparator()
409	if self.parent.ClassName.count('wxDialog') == 0:
410	id = wx.NewId()
411	self._slicerpop.Append(id, '&Linear Fit')
412	wx.EVT_MENU(self, id, self.onFitting)
413	self._slicerpop.AppendSeparator()
414
415	id = wx.NewId()
416	self._slicerpop.Append(id, "Remove %s Curve" % name)
417	wx.EVT_MENU(self, id, self._onRemove)
418	if not plot.is_data:
419	id = wx.NewId()
420	self._slicerpop.Append(id, '&Freeze', 'Freeze')
421	wx.EVT_MENU(self, id, self.onFreeze)
422	self._slicerpop.AppendSeparator()
423	symbol_menu = wx.Menu()
424	for label in self._symbol_labels:
425	id = wx.NewId()
426	symbol_menu.Append(id, str(label), str(label))
427	wx.EVT_MENU(self, id, self.onChangeSymbol)
428	id = wx.NewId()
429	self._slicerpop.AppendMenu(id,'&Modify Symbol', symbol_menu)
430
431	color_menu = wx.Menu()
432	for label in self._color_labels:
433	id = wx.NewId()
434	color_menu.Append(id, str(label), str(label))
435	wx.EVT_MENU(self, id, self.onChangeColor)
436	id = wx.NewId()
437	self._slicerpop.AppendMenu(id, '&Modify Symbol Color', color_menu)
438
439	size_menu = wx.Menu()
440	for i in range(10):
441	id = wx.NewId()
442	size_menu.Append(id, str(i), str(i))
443	wx.EVT_MENU(self, id, self.onChangeSize)
444	id = wx.NewId()
445	size_menu.Append(id, '&Custom', 'Custom')
446	wx.EVT_MENU(self, id, self.onChangeSize)
447	id = wx.NewId()
448	self._slicerpop.AppendMenu(id, '&Modify Symbol Size', size_menu)
449
450	self.hide_menu = self._slicerpop.Append(id, "Hide Error Bar")
451
452	if plot.dy is not None and plot.dy != []:
453	if plot.hide_error :
454	self.hide_menu.SetText('Show Error Bar')
455	else:
456	self.hide_menu.SetText('Hide Error Bar')
457	else:
458	self.hide_menu.Enable(False)
459	wx.EVT_MENU(self, id, self._ontoggle_hide_error)
460
461	self._slicerpop.AppendSeparator()
462	id = wx.NewId()
463	self._slicerpop.Append(id, '&Edit Legend Label', 'Edit Legend Label')
464	wx.EVT_MENU(self, id, self.onEditLabels)
465	# Option to hide
466	#TODO: implement functionality to hide a plottable (legend click)
467
468	loc_menu = wx.Menu()
469	for label in self._loc_labels:
470	id = wx.NewId()
471	loc_menu.Append(id, str(label), str(label))
472	wx.EVT_MENU(self, id, self.onChangeLegendLoc)
473	id = wx.NewId()
474	self._slicerpop.AppendMenu(id, '&Modify Legend Location', loc_menu)
475
476	id = wx.NewId()
477	self._slicerpop.Append(id, '&Toggle Legend On/Off', 'Toggle Legend On/Off')
478	wx.EVT_MENU(self, id, self.onLegend)
479	self._slicerpop.AppendSeparator()
480
481	id = wx.NewId()
482	self._slicerpop.Append(id, '&Edit Y Axis Label')
483	wx.EVT_MENU(self, id, self._on_yaxis_label)
484	id = wx.NewId()
485	self._slicerpop.Append(id, '&Edit X Axis Label')
486	wx.EVT_MENU(self, id, self._on_xaxis_label)
487
488	id = wx.NewId()
489	self._slicerpop.Append(id, '&Toggle Grid On/Off', 'Toggle Grid On/Off')
490	wx.EVT_MENU(self, id, self.onGridOnOff)
491	self._slicerpop.AppendSeparator()
492
493	if self.position != None:
494	id = wx.NewId()
495	self._slicerpop.Append(id, '&Add Text')
496	wx.EVT_MENU(self, id, self._on_addtext)
497	id = wx.NewId()
498	self._slicerpop.Append(id, '&Remove Text')
499	wx.EVT_MENU(self, id, self._on_removetext)
500	self._slicerpop.AppendSeparator()
501	id = wx.NewId()
502	self._slicerpop.Append(id, '&Change Scale')
503	wx.EVT_MENU(self, id, self._onProperties)
504	self._slicerpop.AppendSeparator()
505	id = wx.NewId()
506	self._slicerpop.Append(id, '&Reset Graph Range')
507	wx.EVT_MENU(self, id, self.onResetGraph)
508	try:
509	pos_evt = event.GetPosition()
510	pos = self.ScreenToClient(pos_evt)
511	except:
512	pos_x, pos_y = self.toolbar.GetPositionTuple()
513	pos = (pos_x, pos_y + 5)
514
515	if self.parent.ClassName.count('wxDialog') == 0:
516	self._slicerpop.AppendSeparator()
517	id = wx.NewId()
518	self._slicerpop.Append(id, '&Window Title')
519	wx.EVT_MENU(self, id, self.onChangeCaption)
520
521	self.PopupMenu(self._slicerpop, pos)
522
523	def onFreeze(self, event):
524	"""
525	"""
526	plot = self.plots[self.graph.selected_plottable]
527	self.parent.onfreeze([plot.id])
528
529	def onEditLabels(self, event):
530	"""
531	Edit legend label
532	"""
533	selected_plot = self.plots[self.graph.selected_plottable]
534	label = selected_plot.label
535	dial = LabelDialog(None, -1, 'Change Legend Label', label)
536	if dial.ShowModal() == wx.ID_OK:
537	newLabel = dial.getText()
538	selected_plot.label = newLabel
539	# Updata Graph menu help string
540	pos = self.parent._window_menu.FindItem(self.window_caption)
541	helpString = 'Show/Hide Graph: '
542	for plot in self.plots.itervalues():
543	helpString += (' ' + str(plot.label) +';')
544	self.parent._window_menu.SetHelpString(pos, helpString)
545	self._is_changed_legend_label = True
546	#break
547	dial.Destroy()
548
549	## render the graph
550	try:
551	self._onEVT_FUNC_PROPERTY()
552	except:
553	msg=" Encountered singular points..."
554	wx.PostEvent(self.parent, StatusEvent(status=\
555	"Plotting Erorr: %s"% msg, info="error"))
556
557	def onChangeColor(self, event):
558	"""
559	Changes the color of the graph when selected
560	"""
561	menu = event.GetEventObject()
562	id = event.GetId()
563	label = menu.GetLabel(id)
564	selected_plot = self.plots[self.graph.selected_plottable]
565	selected_plot.custom_color = self._color_labels[label]
566	## Set the view scale for all plots
567	self._check_zoom_plot()
568	#self._onEVT_FUNC_PROPERTY()
569	#wx.PostEvent(self.parent,
570	# NewColorEvent(color=selected_plot.custom_color,
571	# id=selected_plot.id))
572
573	def onChangeSize(self, event):
574
575	menu = event.GetEventObject()
576	id = event.GetId()
577	label = menu.GetLabel(id)
578	selected_plot = self.plots[self.graph.selected_plottable]
579
580	if label == "&Custom":
581	sizedial = SizeDialog(None, -1, 'Change Marker Size')
582	if sizedial.ShowModal() == wx.ID_OK:
583	try:
584	label = sizedial.getText()
585	selected_plot.markersize = int(label)
586	self._check_zoom_plot()
587	except:
588	msg = 'Symbol Size: Got an invalid Value.'
589	wx.PostEvent( self.parent,
590	StatusEvent(status= msg, info='error'))
591	sizedial.Destroy()
592
593
594
595	def onChangeSymbol(self, event):
596	"""
597	"""
598	menu = event.GetEventObject()
599	id = event.GetId()
600	label = menu.GetLabel(id)
601	selected_plot = self.plots[self.graph.selected_plottable]
602	selected_plot.symbol = self._symbol_labels[label]
603	## Set the view scale for all plots
604	self._check_zoom_plot()
605	#self._onEVT_FUNC_PROPERTY()
606	## render the graph
607	#self.graph.render(self)
608	#self.subplot.figure.canvas.draw_idle()
609
610
611
612	def _onsaveTXT(self, path):
613	"""
614	Save file as txt
615
616	:TODO: Refactor and remove this method. See TODO in _onSave.
617
618	"""
619	data = self.plots[self.graph.selected_plottable]
620
621	if not path == None:
622	out = open(path, 'w')
623	has_errors = True
624	if data.dy == None or data.dy == []:
625	has_errors = False
626	# Sanity check
627	if has_errors:
628	try:
629	if len(data.y) != len(data.dy):
630	has_errors = False
631	except:
632	has_errors = False
633	if has_errors:
634	if data.dx != None or data.dx != []:
635	out.write("<X> <Y> <dY> <dX>\n")
636	else:
637	out.write("<X> <Y> <dY>\n")
638	else:
639	out.write("<X> <Y>\n")
640
641	for i in range(len(data.x)):
642	if has_errors:
643	if data.dx != None or data.dx != []:
644	out.write("%g %g %g %g\n" % (data.x[i],
645	data.y[i],
646	data.dy[i],
647	data.dx[i]))
648	else:
649	out.write("%g %g %g\n" % (data.x[i],
650	data.y[i],
651	data.dy[i]))
652	else:
653	out.write("%g %g\n" % (data.x[i],
654	data.y[i]))
655	out.close()
656	try:
657	self._default_save_location = os.path.dirname(path)
658	self.parent._default_save_location = self._default_save_location
659	except:
660	pass
661
662	def _onSave(self, evt):
663	"""
664	Save a data set to a text file
665
666	:param evt: Menu event
667
668	"""
669
670	path = None
671	wildcard = "Text files (.txt)\|.txt\|"\
672	"CanSAS 1D files(.xml)\|.xml"
673	default_name = self.plots[self.graph.selected_plottable].label
674	if default_name.count('.') > 0:
675	default_name = default_name.split('.')[0]
676	default_name += "_out"
677	if self.parent != None:
678	self._default_save_location = self.parent._default_save_location
679	dlg = wx.FileDialog(self, "Choose a file",
680	self._default_save_location,
681	default_name, wildcard , wx.SAVE)
682
683	if dlg.ShowModal() == wx.ID_OK:
684	path = dlg.GetPath()
685	# ext_num = 0 for .txt, ext_num = 1 for .xml
686	# This is MAC Fix
687	ext_num = dlg.GetFilterIndex()
688	if ext_num == 0:
689	format = '.txt'
690	else:
691	format = '.xml'
692	path = os.path.splitext(path)[0] + format
693	mypath = os.path.basename(path)
694
695	#TODO: This is bad design. The DataLoader is designed
696	#to recognize extensions.
697	# It should be a simple matter of calling the .
698	#save(file, data, '.xml') method
699	# of the sans.dataloader.loader.Loader class.
700	from sans.dataloader.loader import Loader
701	#Instantiate a loader
702	loader = Loader()
703	data = self.plots[self.graph.selected_plottable]
704	format = ".txt"
705	if os.path.splitext(mypath)[1].lower() == format:
706	# Make sure the ext included in the file name
707	# especially on MAC
708	fName = os.path.splitext(path)[0] + format
709	self._onsaveTXT(fName)
710	format = ".xml"
711	if os.path.splitext(mypath)[1].lower() == format:
712	# Make sure the ext included in the file name
713	# especially on MAC
714	fName = os.path.splitext(path)[0] + format
715	loader.save(fName, data, format)
716	try:
717	self._default_save_location = os.path.dirname(path)
718	self.parent._default_save_location = self._default_save_location
719	except:
720	pass
721	dlg.Destroy()
722
723	def _add_more_tool(self):
724	"""
725	Add refresh, add/delete button in the tool bar
726	"""
727	if self.parent.__class__.__name__ != 'ViewerFrame':
728	return
729	self.toolbar.AddSeparator()
730	id_delete = wx.NewId()
731	delete = wx.ArtProvider.GetBitmap(wx.ART_DELETE, wx.ART_TOOLBAR)
732	self.toolbar.AddSimpleTool(id_delete, delete,
733	'Delete', 'permanently Delete')
734
735	self.toolbar.Realize()
736	wx.EVT_TOOL(self, id_delete, self._on_delete)
737
738	"""
739	self.toolbar.AddSeparator()
740	id_text = wx.NewId()
741	text = wx.ArtProvider.GetBitmap(wx.ART_CUT, wx.ART_TOOLBAR)
742	self.toolbar.AddSimpleTool(id_text, text,
743	'Remove Text from Plot', 'Removes text from plot')
744
745	self.toolbar.Realize()
746	wx.EVT_TOOL(self, id_text, self._on_removetext)
747	"""
748	def _on_delete(self, event):
749	"""
750	Refreshes the plotpanel on refresh tollbar button
751	"""
752
753	if self.parent is not None:
754	wx.PostEvent(self.parent,
755	NewPlotEvent(group_id=self.group_id,
756	action="delete"))
757
758

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

SasView

source: sasview/sansguiframe/src/sans/guiframe/local_perspectives/plotting/Plotter1D.py @ 1fc5e10a

Download in other formats: