Plotter1D.py @ 7ee79cb

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 7ee79cb was fec4939, checked in by Jae Cho <jhjcho@…>, 13 years ago
fixed an error that the 2Dmodel panel was not completely removed after deleting the data (broken 2D panel)
Property mode set to `100644`
File size: 25.9 KB

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

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

SasView

source: sasview/sansguiframe/src/sans/guiframe/local_perspectives/plotting/Plotter1D.py @ 7ee79cb

Download in other formats: