Plotter1D.py @ 65e61c1

Last change on this file since 65e61c1 was 9a5097c, checked in by andyfaff, 8 years ago
MAINT: import numpy as np
Property mode set to `100644`
File size: 31.6 KB

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

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

SasView

source: sasview/src/sas/sasgui/guiframe/local_perspectives/plotting/Plotter1D.py @ 65e61c1

Download in other formats: