Plotter1D.py @ 7d89e50

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 7d89e50 was f8be87d, checked in by Kieran Campbell <kieranrcampbell@…>, 12 years ago
Changes to include 1D appearance dialog
Property mode set to `100644`
File size: 23.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
[6e75ed0]	36	from sans.guiframe.gui_style import GUIFRAME_ICON
[1bf33c1]	37	from binder import BindArtist
[f8be87d]	38	from appearanceDialog import appearanceDialog
[1bf33c1]	39
[0d9dae8]	40	DEFAULT_QMAX = 0.05
[1bf33c1]	41	DEFAULT_QSTEP = 0.001
	42	DEFAULT_BEAM = 0.005
[32c0841]	43	BIN_WIDTH = 1
[940aca7]	44	IS_MAC = (sys.platform == 'darwin')
[0d9dae8]	45
[691643c]	46	class ModelPanel1D(PlotPanel, PanelBase):
[1bf33c1]	47	"""
[d955bf19]	48	Plot panel for use with the GUI manager
[1bf33c1]	49	"""
	50
	51	## Internal name for the AUI manager
	52	window_name = "plotpanel"
	53	## Title to appear on top of the window
[37c36d9]	54	window_caption = "Graph"
[1bf33c1]	55	## Flag to tell the GUI manager that this panel is not
	56	# tied to any perspective
	57	ALWAYS_ON = True
	58	## Group ID
	59	group_id = None
	60
[32c0841]	61	def __init__(self, parent, id=-1, color = None,
	62	dpi=None, style=wx.NO_FULL_REPAINT_ON_RESIZE, **kwargs):
	63	PlotPanel.__init__(self, parent, id=id, style=style, **kwargs)
[a45037aa]	64	PanelBase.__init__(self, parent)
[1bf33c1]	65	## Reference to the parent window
	66	self.parent = parent
	67	## Plottables
	68	self.plots = {}
[52b8b74]	69	#context menu
	70	self._slicerpop = None
[a07e72f]	71
[52b8b74]	72	self._available_data = []
	73	self._menu_add_ids = []
[a07e72f]	74	self._symbol_labels = self.get_symbol_label()
[2636188]	75	self._color_labels = self.get_color_label()
	76	self.currColorIndex = ""
[36288ca]	77	self._is_changed_legend_label = False
[a07e72f]	78
	79	self.hide_menu = None
[1bf33c1]	80	## Unique ID (from gui_manager)
	81	self.uid = None
[66718a1]	82	self.x_size = None
[6063b16]	83	## Default locations
[c553b18]	84	#self._default_save_location = os.getcwd()
[adf44c2]	85	self.size = None
[1bf33c1]	86	## Graph
[9f5c8bb]	87	#self.graph = Graph()
[1bf33c1]	88	self.graph.xaxis("\\rm{Q}", 'A^{-1}')
[32c0841]	89	self.graph.yaxis("\\rm{Intensity} ", "cm^{-1}")
[1bf33c1]	90	self.graph.render(self)
[ef4a4ea]	91
[6d727ae]	92	# In resizing event
	93	self.resizing = False
	94	self.canvas.set_resizing(self.resizing)
	95	self.Bind(wx.EVT_SIZE, self._OnReSize)
[c5a769e]	96	self._add_more_tool()
[6e75ed0]	97	self.parent.SetFocus()
	98
[140fad00]	99
[a07e72f]	100	def get_symbol_label(self):
[52b8b74]	101	"""
[a07e72f]	102	Associates label to symbol
[52b8b74]	103	"""
[a07e72f]	104	_labels = {}
	105	i = 0
[6d727ae]	106	_labels['Circle'] = i
[a07e72f]	107	i += 1
[6d727ae]	108	_labels['Cross X '] = i
[a07e72f]	109	i += 1
	110	_labels['Triangle Down'] = i
	111	i += 1
	112	_labels['Triangle Up'] = i
	113	i += 1
	114	_labels['Triangle Left'] = i
	115	i += 1
	116	_labels['Triangle Right'] = i
	117	i += 1
[6d727ae]	118	_labels['Cross +'] = i
[a07e72f]	119	i += 1
	120	_labels['Square'] = i
	121	i += 1
[6d727ae]	122	_labels['Diamond'] = i
[a07e72f]	123	i += 1
	124	_labels['Diamond'] = i
	125	i += 1
	126	_labels['Hexagon1'] = i
	127	i += 1
	128	_labels['Hexagon2'] = i
	129	i += 1
	130	_labels['Pentagon'] = i
	131	i += 1
[6d727ae]	132	_labels['Line'] = i
[a07e72f]	133	return _labels
[2636188]	134
	135	def get_color_label(self):
	136	"""
	137	Associates label to a specific color
	138	"""
	139	_labels = {}
	140	i = 0
	141	_labels['Blue'] = i
	142	i += 1
	143	_labels['Green'] = i
	144	i += 1
	145	_labels['Red'] = i
	146	i += 1
	147	_labels['Cyan'] = i
	148	i += 1
	149	_labels['Magenta'] = i
	150	i += 1
	151	_labels['Yellow'] = i
[ad019b83]	152	i += 1
	153	_labels['Black'] = i
[2636188]	154	return _labels
[a07e72f]	155
[52b8b74]	156
[32c0841]	157	def set_data(self, list=None):
[3c44c66]	158	"""
	159	"""
	160	pass
	161
[1bf33c1]	162	def _reset(self):
	163	"""
[d955bf19]	164	Resets internal data and graph
[1bf33c1]	165	"""
	166	self.graph.reset()
	167	self.plots = {}
[7022fdc]	168	if self.is_zoomed:
[fd51a7c]	169	self.is_zoomed = False
[6d727ae]	170
	171	def _OnReSize(self, event):
	172	"""
	173	On response of the resize of a panel, set axes_visiable False
	174	"""
[66718a1]	175	# It was found that wx >= 2.9.3 sends an event even if no size changed.
[294efd5]	176	# So manually recode the size (=x_size) and compare here.
[940aca7]	177	# Massy code to work around:<
	178	if self.parent._mgr != None:
	179	max_panel = self.parent._mgr.GetPane(self)
	180	if max_panel.IsMaximized():
	181	self.parent._mgr.RestorePane(max_panel)
	182	max_panel.Maximize()
[66718a1]	183	if self.x_size != None:
	184	if self.x_size == self.GetSize():
[71fa9bb9]	185	self.resizing = False
	186	self.canvas.set_resizing(self.resizing)
[66718a1]	187	return
[294efd5]	188	self.x_size = self.GetSize()
	189
[ad9e5e1]	190	# Ready for another event
[4520830]	191	# Do not remove this Skip. Otherwise it will get runtime error on wx>=2.9.
[ad9e5e1]	192	event.Skip()
[6d727ae]	193	# set the resizing flag
	194	self.resizing = True
	195	self.canvas.set_resizing(self.resizing)
	196	self.parent.set_schedule(True)
[adf44c2]	197	pos_x, pos_y = self.GetPositionTuple()
	198	if pos_x != 0 and pos_y != 0:
[294efd5]	199	self.size, _ = self.GetClientSizeTuple()
[2b29bb4]	200	self.SetSizer(self.sizer)
[7fb69f26]	201	wx.CallAfter(self.parent.disable_app_menu,self)
[5112838]	202
[6d727ae]	203	def set_resizing(self, resizing=False):
	204	"""
	205	Set the resizing (True/False)
	206	"""
	207	self.resizing = resizing
	208	#self.canvas.set_resizing(resizing)
[1bf33c1]	209
[6d727ae]	210	def schedule_full_draw(self, func='append'):
	211	"""
	212	Put self in schedule to full redraw list
	213	"""
	214	# append/del this panel in the schedule list
	215	self.parent.set_schedule_full_draw(self, func)
	216
	217
[a07e72f]	218	def remove_data_by_id(self, id):
	219	"""'
	220	remove data from plot
	221	"""
	222	if id in self.plots.keys():
	223	data = self.plots[id]
	224	self.graph.delete(data)
	225	data_manager = self._manager.parent.get_data_manager()
[df22224]	226	data_list, theory_list = data_manager.get_by_id(id_list=[id])
	227
	228	if id in data_list.keys():
	229	data = data_list[id]
[248b918]	230	if id in theory_list.keys():
[df22224]	231	data = theory_list[id]
[53a6d09]	232	# Update Graph menu and help string
	233	h_id = self.parent._window_menu.FindItem(self.window_caption)
[e032dfc]	234	if data != None:
[78919f45]	235	if data.__class__.__name__ == 'list':
	236	label = data[0].label
	237	else:
	238	label = data.label
[e032dfc]	239	else:
	240	label = '???'
[53a6d09]	241	helpString = self.parent._window_menu.GetHelpString(h_id)
[e032dfc]	242	d_string = (' ' + str(label) +';')
[53a6d09]	243	new_tip = helpString.replace(d_string, '')
	244	self.parent._window_menu.SetHelpString(h_id, new_tip)
	245
[a07e72f]	246	del self.plots[id]
	247	self.graph.render(self)
	248	self.subplot.figure.canvas.draw_idle()
	249	if len(self.graph.plottables) == 0:
[df22224]	250	#onRemove: graph is empty must be the panel must be destroyed
	251	self.parent.delete_panel(self.uid)
[53a6d09]	252
[a07e72f]	253
	254	def plot_data(self, data):
[1bf33c1]	255	"""
[d955bf19]	256	Data is ready to be displayed
	257
	258	:param event: data event
[1bf33c1]	259	"""
[b837e1f]	260	if data.__class__.__name__ == 'Data2D':
	261	return
[a07e72f]	262	if data.id in self.plots.keys():
[e4a703a]	263	#Recover panel prop.s
	264	xlo, xhi = self.subplot.get_xlim()
	265	ylo, yhi = self.subplot.get_ylim()
[5eede4e]	266	old_data = self.plots[data.id]
[36288ca]	267	if self._is_changed_legend_label:
	268	data.label = old_data.label
[c9579e0]	269	if old_data.__class__.__name__ == 'Data1D':
	270	data.custom_color = old_data.custom_color
	271	data.symbol = old_data.symbol
	272	data.markersize = old_data.markersize
[e4a703a]	273	# Replace data
[a07e72f]	274	self.graph.replace(data)
	275	self.plots[data.id] = data
[e4a703a]	276	## Set the view scale for all plots
[a4964b8e]	277	try:
	278	self._onEVT_FUNC_PROPERTY()
	279	except:
[4904062]	280	msg=" Encountered singular points..."
	281	wx.PostEvent(self.parent, StatusEvent(status=\
[91a3860]	282	"Plotting Error: %s"% msg, info="error"))
[7022fdc]	283	# Check if zoomed
	284	toolbar_zoomed = self.toolbar.GetToolEnabled(self.toolbar._NTB2_BACK)
	285	if self.is_zoomed or toolbar_zoomed:
	286	# Recover the x,y limits
	287	self.subplot.set_xlim((xlo, xhi))
	288	self.subplot.set_ylim((ylo, yhi))
[a07e72f]	289	else:
	290	self.plots[data.id] = data
	291	self.graph.add(self.plots[data.id])
[e4a703a]	292	## Set the view scale for all plots
[a4964b8e]	293	try:
	294	self._onEVT_FUNC_PROPERTY()
[940aca7]	295	if IS_MAC:
	296	# MAC: forcing to plot 2D avg
	297	self.canvas._onDrawIdle()
[a4964b8e]	298	except:
[4904062]	299	msg=" Encountered singular points..."
	300	wx.PostEvent(self.parent, StatusEvent(status=\
[91a3860]	301	"Plotting Error: %s"% msg, info="error"))
[fe48fcc]	302	self.toolbar.update()
[7022fdc]	303	if self.is_zoomed:
	304	self.is_zoomed = False
[d6e39a8e]	305	# Update Graph menu and help string
	306	pos = self.parent._window_menu.FindItem(self.window_caption)
	307	helpString = 'Show/Hide Graph: '
	308	for plot in self.plots.itervalues():
	309	helpString += (' ' + str(plot.label) +';')
	310	self.parent._window_menu.SetHelpString(pos, helpString)
[e4a703a]	311
[6d727ae]	312	def draw_plot(self):
	313	"""
	314	Draw plot
	315	"""
	316	self.draw()
	317
[1bf33c1]	318	def onLeftDown(self,event):
[6c0568b]	319	"""
[d955bf19]	320	left button down and ready to drag
	321	Display the position of the mouse on the statusbar
[6c0568b]	322	"""
[e85e2bc]	323	PlotPanel.onLeftDown(self, event)
	324	ax = event.inaxes
	325	if ax != None:
[902f373]	326	try:
	327	pos_x = float(event.xdata)# / size_x
	328	pos_y = float(event.ydata)# / size_y
	329	pos_x = "%8.3g"% pos_x
	330	pos_y = "%8.3g"% pos_y
	331	self.position = str(pos_x), str(pos_y)
	332	wx.PostEvent(self.parent, StatusEvent(status=self.position))
	333	except:
	334	self.position = None
[0275276]	335	# unfocus all
	336	self.parent.set_plot_unfocus()
[4ed210f4]	337	#post nd event to notify guiframe that this panel is on focus
[a45037aa]	338	wx.PostEvent(self.parent, PanelOnFocusEvent(panel=self))
[df68da1]	339
[4ed210f4]	340
[a07e72f]	341	def _ontoggle_hide_error(self, event):
	342	"""
	343	Toggle error display to hide or show
	344	"""
[70ec588d]	345	menu = event.GetEventObject()
	346	id = event.GetId()
	347	self.set_selected_from_menu(menu, id)
[ebf422a]	348	# Check zoom
	349	xlo, xhi = self.subplot.get_xlim()
	350	ylo, yhi = self.subplot.get_ylim()
	351
[a07e72f]	352	selected_plot = self.plots[self.graph.selected_plottable]
[5637362]	353	if self.hide_menu.GetText() == "Hide Error Bar":
[a07e72f]	354	selected_plot.hide_error = True
	355	else:
	356	selected_plot.hide_error = False
	357	## increment graph color
	358	self.graph.render(self)
	359	self.subplot.figure.canvas.draw_idle()
[ebf422a]	360	# Check if zoomed
	361	toolbar_zoomed = self.toolbar.GetToolEnabled(self.toolbar._NTB2_BACK)
	362	if self.is_zoomed or toolbar_zoomed:
	363	# Recover the x,y limits
	364	self.subplot.set_xlim((xlo, xhi))
	365	self.subplot.set_ylim((ylo, yhi))
	366
[a07e72f]	367
[1bf33c1]	368	def _onRemove(self, event):
	369	"""
[d955bf19]	370	Remove a plottable from the graph and render the graph
	371
	372	:param event: Menu event
	373
[1bf33c1]	374	"""
[70ec588d]	375	menu = event.GetEventObject()
	376	id = event.GetId()
	377	self.set_selected_from_menu(menu, id)
[6c0568b]	378	## Check if there is a selected graph to remove
[a07e72f]	379	if self.graph.selected_plottable in self.plots.keys():
	380	selected_plot = self.plots[self.graph.selected_plottable]
	381	id = self.graph.selected_plottable
[df22224]	382	self.remove_data_by_id(id)
	383
[1bf33c1]	384	def onContextMenu(self, event):
	385	"""
[d955bf19]	386	1D plot context menu
	387
	388	:param event: wx context event
	389
[1bf33c1]	390	"""
[52b8b74]	391	self._slicerpop = PanelMenu()
	392	self._slicerpop.set_plots(self.plots)
[70ec588d]	393	self._slicerpop.set_graph(self.graph)
	394	if not self.graph.selected_plottable in self.plots:
	395	# Various plot options
	396	id = wx.NewId()
	397	self._slicerpop.Append(id, '&Save Image', 'Save image as PNG')
	398	wx.EVT_MENU(self, id, self.onSaveImage)
	399	id = wx.NewId()
	400	self._slicerpop.Append(id, '&Print Image', 'Print image ')
	401	wx.EVT_MENU(self, id, self.onPrint)
	402	id = wx.NewId()
	403	self._slicerpop.Append(id, '&Print Preview', 'Print preview')
	404	wx.EVT_MENU(self, id, self.onPrinterPreview)
[a436b2e]	405
[6d727ae]	406	id = wx.NewId()
[70ec588d]	407	self._slicerpop.Append(id, '&Copy to Clipboard', 'Copy to the clipboard')
	408	wx.EVT_MENU(self, id, self.OnCopyFigureMenu)
	409
	410	self._slicerpop.AppendSeparator()
	411
	412	for plot in self.plots.values():
	413	#title = plot.title
[6d727ae]	414	name = plot.name
[70ec588d]	415	plot_menu = wx.Menu()
[242aff5]	416	if self.graph.selected_plottable:
[4a9bce1]	417	if not self.graph.selected_plottable in self.plots.keys():
	418	continue
[242aff5]	419	if plot != self.plots[self.graph.selected_plottable]:
	420	continue
[940aca7]	421
	422	id = wx.NewId()
	423	plot_menu.Append(id, "&DataInfo", name)
	424	wx.EVT_MENU(self, id, self. _onDataShow)
	425	id = wx.NewId()
	426	plot_menu.Append(id, "&Save Points as a File", name)
	427	wx.EVT_MENU(self, id, self._onSave)
	428	plot_menu.AppendSeparator()
	429
[70ec588d]	430	#add menu of other plugins
[940aca7]	431	item_list = self.parent.get_current_context_menu(self)
[242aff5]	432
[70ec588d]	433	if (not item_list == None) and (not len(item_list) == 0):
	434	for item in item_list:
[242aff5]	435
[70ec588d]	436	try:
	437	id = wx.NewId()
	438	plot_menu.Append(id, item[0], name)
	439	wx.EVT_MENU(self, id, item[2])
	440	except:
	441	msg = "ModelPanel1D.onContextMenu: "
	442	msg += "bad menu item %s" % sys.exc_value
	443	wx.PostEvent(self.parent, StatusEvent(status=msg))
	444	pass
	445	plot_menu.AppendSeparator()
[940aca7]	446
[73eb92fc]	447	if self.parent.ClassName.count('wxDialog') == 0:
[e6a93df]	448	id = wx.NewId()
[70ec588d]	449	plot_menu.Append(id, '&Linear Fit', name)
[a436b2e]	450	wx.EVT_MENU(self, id, self.onFitting)
[70ec588d]	451	plot_menu.AppendSeparator()
[a436b2e]	452
[a3c96f7a]	453	id = wx.NewId()
[70ec588d]	454	plot_menu.Append(id, "Remove", name)
[a436b2e]	455	wx.EVT_MENU(self, id, self._onRemove)
	456	if not plot.is_data:
	457	id = wx.NewId()
[70ec588d]	458	plot_menu.Append(id, '&Freeze', name)
[a436b2e]	459	wx.EVT_MENU(self, id, self.onFreeze)
[70ec588d]	460	plot_menu.AppendSeparator()
[a436b2e]	461	symbol_menu = wx.Menu()
[2636188]	462
[70ec588d]	463	if plot.is_data:
[f8be87d]	464	id = wx.NewId()
[70ec588d]	465	self.hide_menu = plot_menu.Append(id,
	466	"Hide Error Bar", name)
	467
	468	if plot.dy is not None and plot.dy != []:
	469	if plot.hide_error :
	470	self.hide_menu.SetText('Show Error Bar')
	471	else:
	472	self.hide_menu.SetText('Hide Error Bar')
[a436b2e]	473	else:
[70ec588d]	474	self.hide_menu.Enable(False)
	475	wx.EVT_MENU(self, id, self._ontoggle_hide_error)
[a436b2e]	476
[f8be87d]	477	plot_menu.AppendSeparator()
	478
[e1714a9]	479	id = wx.NewId()
[f8be87d]	480	plot_menu.Append(id, '&Modify graph properties', name)
	481	wx.EVT_MENU(self, id, self.createAppDialog)
	482
	483
	484
[70ec588d]	485	id = wx.NewId()
	486	#plot_menu.SetTitle(name)
	487	self._slicerpop.AppendMenu(id, '&%s'% name, plot_menu)
[a436b2e]	488	# Option to hide
	489	#TODO: implement functionality to hide a plottable (legend click)
[70ec588d]	490	if not self.graph.selected_plottable in self.plots:
	491	self._slicerpop.AppendSeparator()
	492	loc_menu = wx.Menu()
	493	for label in self._loc_labels:
	494	id = wx.NewId()
	495	loc_menu.Append(id, str(label), str(label))
	496	wx.EVT_MENU(self, id, self.onChangeLegendLoc)
[e1714a9]	497	id = wx.NewId()
[70ec588d]	498	self._slicerpop.AppendMenu(id, '&Modify Legend Location', loc_menu)
	499
	500	id = wx.NewId()
	501	self._slicerpop.Append(id, '&Toggle Legend On/Off', 'Toggle Legend On/Off')
	502	wx.EVT_MENU(self, id, self.onLegend)
	503	self._slicerpop.AppendSeparator()
	504
	505	id = wx.NewId()
	506	self._slicerpop.Append(id, '&Edit Y Axis Label')
	507	wx.EVT_MENU(self, id, self._on_yaxis_label)
	508	id = wx.NewId()
	509	self._slicerpop.Append(id, '&Edit X Axis Label')
	510	wx.EVT_MENU(self, id, self._on_xaxis_label)
	511
[ef4a4ea]	512	id = wx.NewId()
[70ec588d]	513	self._slicerpop.Append(id, '&Toggle Grid On/Off', 'Toggle Grid On/Off')
	514	wx.EVT_MENU(self, id, self.onGridOnOff)
	515	self._slicerpop.AppendSeparator()
	516
	517	if self.position != None:
	518	id = wx.NewId()
	519	self._slicerpop.Append(id, '&Add Text')
	520	wx.EVT_MENU(self, id, self._on_addtext)
	521	id = wx.NewId()
	522	self._slicerpop.Append(id, '&Remove Text')
	523	wx.EVT_MENU(self, id, self._on_removetext)
	524	self._slicerpop.AppendSeparator()
[ef4a4ea]	525	id = wx.NewId()
[70ec588d]	526	self._slicerpop.Append(id, '&Change Scale')
	527	wx.EVT_MENU(self, id, self._onProperties)
[857d00f]	528	self._slicerpop.AppendSeparator()
[70ec588d]	529	id = wx.NewId()
	530	self._slicerpop.Append(id, '&Reset Graph Range')
	531	wx.EVT_MENU(self, id, self.onResetGraph)
	532
	533	if self.parent.ClassName.count('wxDialog') == 0:
	534	self._slicerpop.AppendSeparator()
	535	id = wx.NewId()
	536	self._slicerpop.Append(id, '&Window Title')
	537	wx.EVT_MENU(self, id, self.onChangeCaption)
[dc51a7f]	538	try:
	539	pos_evt = event.GetPosition()
	540	pos = self.ScreenToClient(pos_evt)
	541	except:
	542	pos_x, pos_y = self.toolbar.GetPositionTuple()
	543	pos = (pos_x, pos_y + 5)
[52b8b74]	544	self.PopupMenu(self._slicerpop, pos)
[70ec588d]	545
[e6a93df]	546	def onFreeze(self, event):
	547	"""
	548	"""
[70ec588d]	549	menu = event.GetEventObject()
	550	id = event.GetId()
	551	self.set_selected_from_menu(menu, id)
[e6a93df]	552	plot = self.plots[self.graph.selected_plottable]
	553	self.parent.onfreeze([plot.id])
[d7d1255e]	554
[176fbf1]	555
[1bf33c1]	556	def _onSave(self, evt):
	557	"""
[d955bf19]	558	Save a data set to a text file
	559
	560	:param evt: Menu event
	561
[1bf33c1]	562	"""
[70ec588d]	563	menu = evt.GetEventObject()
	564	id = evt.GetId()
	565	self.set_selected_from_menu(menu, id)
[176fbf1]	566	data = self.plots[self.graph.selected_plottable]
	567	default_name = data.label
[c553b18]	568	if default_name.count('.') > 0:
	569	default_name = default_name.split('.')[0]
	570	default_name += "_out"
	571	if self.parent != None:
[176fbf1]	572	self.parent.save_data1d(data, default_name)
[c5a769e]	573
[0aca693]	574
	575	def _onDataShow(self, evt):
	576	"""
	577	Show the data set in text
	578
	579	:param evt: Menu event
	580
	581	"""
	582	menu = evt.GetEventObject()
	583	id = evt.GetId()
	584	self.set_selected_from_menu(menu, id)
	585	data = self.plots[self.graph.selected_plottable]
	586	default_name = data.label
	587	if default_name.count('.') > 0:
	588	default_name = default_name.split('.')[0]
	589	#default_name += "_out"
	590	if self.parent != None:
	591	self.parent.show_data1d(data, default_name)
	592
[c5a769e]	593	def _add_more_tool(self):
	594	"""
[6e75ed0]	595	Add refresh, add/hide button in the tool bar
[c5a769e]	596	"""
	597	if self.parent.__class__.__name__ != 'ViewerFrame':
	598	return
	599	self.toolbar.AddSeparator()
[d6e39a8e]	600	id_hide = wx.NewId()
	601	hide = wx.Bitmap(GUIFRAME_ICON.HIDE_ID_PATH, wx.BITMAP_TYPE_PNG)
	602	self.toolbar.AddSimpleTool(id_hide, hide, 'Hide', 'Hide')
[c5a769e]	603	self.toolbar.Realize()
[d6e39a8e]	604	wx.EVT_TOOL(self, id_hide, self._on_hide)
[c5a769e]	605
[6e75ed0]	606	def _on_hide(self, event):
	607	"""
	608	Hides the plot when button is pressed
	609	"""
[c5a769e]	610	if self.parent is not None:
[6e75ed0]	611	self.parent.hide_panel(self.uid)
[f8be87d]	612
	613	def createAppDialog(self, event):
	614	"""
	615	Create the custom dialog for fit appearance modification
	616	"""
	617	menu = event.GetEventObject()
	618	id = event.GetId()
	619	self.set_selected_from_menu(menu,id)
	620	self.appearance_selected_plot = self.plots[self.graph.selected_plottable]
	621
	622	# find current properties
	623	curr_color = self.appearance_selected_plot.custom_color
	624	curr_symbol = self.appearance_selected_plot.symbol
	625	curr_size = self.appearance_selected_plot.markersize
	626	curr_label = self.appearance_selected_plot.label
	627
	628	if curr_color == None:
	629	curr_color = self._color_labels['Blue']
	630	curr_symbol = 13
	631
	632	self.appD = appearanceDialog(self,'Modify graph properties')
	633	self.appD.setDefaults(float(curr_size),int(curr_color),str(appearanceDialog.find_key(self.get_symbol_label(),int(curr_symbol))),curr_label)
	634	self.appD.Bind(wx.EVT_CLOSE, self.on_AppDialog_close)
	635
	636	def on_AppDialog_close(self,e):
	637	if(self.appD.okay_clicked == True):
	638	info = self.appD.getCurrentValues() # returns (size,color,symbol,datalabel)
	639	self.appearance_selected_plot.custom_color = self._color_labels[info[1].encode('ascii','ignore')]
	640
	641	self.appearance_selected_plot.markersize = float(info[0])
	642	self.appearance_selected_plot.symbol = self.get_symbol_label()[info[2]] # self._symbol_labels[info[2].encode('ascii','ignore')]
	643	self.appearance_selected_plot.label = str(info[3])
	644
	645	pos = self.parent._window_menu.FindItem(self.window_caption)
	646	helpString = 'Show/Hide Graph: '
	647	for plot in self.plots.itervalues():
	648	helpString += (' ' + str(plot.label) +';')
	649	self.parent._window_menu.SetHelpString(pos, helpString)
	650	self._is_changed_legend_label = True
	651
	652	self.appD.Destroy()
	653	self._check_zoom_plot()

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

SasView

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

Download in other formats: