Plotter.py @ 0215e0a

ESS_GUIESS_GUI_DocsESS_GUI_batch_fittingESS_GUI_bumps_abstractionESS_GUI_iss1116ESS_GUI_iss879ESS_GUI_iss959ESS_GUI_openclESS_GUI_orderingESS_GUI_sync_sascalc

Last change on this file since 0215e0a was 83eb5208, checked in by Piotr Rozyczko <rozyczko@…>, 8 years ago
Putting files in more ordered fashion
Property mode set to `100644`
File size: 24.3 KB

Rev	Line
[8cb6cd6]	1	from PyQt4 import QtGui
[aadf0af1]	2	from PyQt4 import QtCore
	3	import functools
	4	import copy
[8cb6cd6]	5
	6	import matplotlib.pyplot as plt
[9290b1a]	7	from matplotlib.font_manager import FontProperties
[8cb6cd6]	8
[9290b1a]	9	from sas.sasgui.guiframe.dataFitting import Data1D
[83eb5208]	10	from sas.qtgui.Plotting.PlotterBase import PlotterBase
	11	import sas.qtgui.Utilities.GuiUtils as GuiUtils
	12	from sas.qtgui.Plotting.AddText import AddText
	13	from sas.qtgui.Plotting.SetGraphRange import SetGraphRange
	14	from sas.qtgui.Plotting.LinearFit import LinearFit
	15	from sas.qtgui.Plotting.PlotProperties import PlotProperties
	16	import sas.qtgui.Plotting.PlotUtilities as PlotUtilities
[8cb6cd6]	17
[416fa8f]	18	class PlotterWidget(PlotterBase):
[c4e5400]	19	"""
	20	1D Plot widget for use with a QDialog
[fecfe28]	21	"""
[416fa8f]	22	def __init__(self, parent=None, manager=None, quickplot=False):
	23	super(PlotterWidget, self).__init__(parent, manager=manager, quickplot=quickplot)
[570a58f9]	24
[27313b7]	25	self.parent = parent
[8cb6cd6]	26
[aadf0af1]	27	# Dictionary of {plot_id:Data1d}
	28	self.plot_dict = {}
	29
[fed94a2]	30	# Window for text add
	31	self.addText = AddText(self)
[aadf0af1]	32
[fed94a2]	33	# Log-ness of the axes
[570a58f9]	34	self.xLogLabel = "log10(x)"
	35	self.yLogLabel = "log10(y)"
	36
	37	# Data container for the linear fit
[fed94a2]	38	self.fit_result = Data1D(x=[], y=[], dy=None)
	39	self.fit_result.symbol = 13
	40	self.fit_result.name = "Fit"
[570a58f9]	41
	42	# Add a slot for receiving update signal from LinearFit
[fed94a2]	43	# NEW style signals
[570a58f9]	44	#self.updatePlot = QtCore.pyqtSignal(tuple)
[fed94a2]	45	# self.updatePlot.connect(self.onFitDisplay)
	46	# OLD style signals
[570a58f9]	47	QtCore.QObject.connect(self, QtCore.SIGNAL('updatePlot'), self.onFitDisplay)
	48
[31c5b58]	49	@property
	50	def data(self):
	51	return self._data
	52
	53	@data.setter
	54	def data(self, value):
[8cb6cd6]	55	""" data setter """
[31c5b58]	56	self._data = value
[6d05e1d]	57	self.xLabel = "%s(%s)"%(value._xaxis, value._xunit)
	58	self.yLabel = "%s(%s)"%(value._yaxis, value._yunit)
[aadf0af1]	59	self.title(title=value.name)
[8cb6cd6]	60
[87cc73a]	61	def plot(self, data=None, color=None, marker=None, hide_error=False):
[8cb6cd6]	62	"""
[aadf0af1]	63	Add a new plot of self._data to the chart.
[8cb6cd6]	64	"""
[9290b1a]	65	# Data1D
	66	if isinstance(data, Data1D):
	67	self.data = data
	68	assert(self._data)
	69
[87cc73a]	70	is_fit = (self.data.id=="fit")
[570a58f9]	71
[f182f93]	72	# Transform data if required.
	73	# TODO: it properly!
	74	#if data.xtransform is not None or data.ytransform is not None:
	75	# a, b, c, d = GuiUtils.xyTransform(self.data, self.data.xtransform, self.data.ytransform)
	76
[87cc73a]	77	# Shortcuts
[ef01be4]	78	ax = self.ax
[87cc73a]	79	x = self._data.view.x
	80	y = self._data.view.y
	81
	82	# Marker symbol. Passed marker is one of matplotlib.markers characters
	83	# Alternatively, picked up from Data1D as an int index of PlotUtilities.SHAPES dict
	84	if marker is None:
	85	marker = self.data.symbol
[6fd4e36]	86	# Try name first
	87	try:
	88	marker = PlotUtilities.SHAPES[marker]
	89	except KeyError:
	90	marker = PlotUtilities.SHAPES.values()[marker]
[87cc73a]	91
[6fd4e36]	92	assert marker is not None
[87cc73a]	93	# Plot name
[b789967]	94	if self.data.title:
	95	self.title(title=self.data.title)
	96	else:
	97	self.title(title=self.data.name)
[87cc73a]	98
	99	# Error marker toggle
	100	if hide_error is None:
	101	hide_error = self.data.hide_error
	102
	103	# Plot color
	104	if color is None:
	105	color = self.data.custom_color
	106
	107	color = PlotUtilities.getValidColor(color)
	108
	109	markersize = self._data.markersize
	110
[239214f]	111	# Draw non-standard markers
	112	l_width = markersize * 0.4
	113	if marker == '-' or marker == '--':
	114	line = self.ax.plot(x, y, color=color, lw=l_width, marker='',
	115	linestyle=marker, label=self._title, zorder=10)[0]
	116
	117	elif marker == 'vline':
	118	y_min = min(y)*9.0/10.0 if min(y) < 0 else 0.0
	119	line = self.ax.vlines(x=x, ymin=y_min, ymax=y, color=color,
	120	linestyle='-', label=self._title, lw=l_width, zorder=1)
	121
	122	elif marker == 'step':
	123	line = self.ax.step(x, y, color=color, marker='', linestyle='-',
	124	label=self._title, lw=l_width, zorder=1)[0]
[8cb6cd6]	125
[c4e5400]	126	else:
[87cc73a]	127	# plot data with/without errorbars
	128	if hide_error:
	129	line = ax.plot(x, y, marker=marker, color=color, markersize=markersize,
	130	linestyle='', label=self._title, picker=True)
	131	else:
	132	line = ax.errorbar(x, y,
	133	yerr=self._data.view.dy, xerr=None,
	134	capsize=2, linestyle='',
	135	barsabove=False,
	136	color=color,
	137	marker=marker,
	138	markersize=markersize,
	139	lolims=False, uplims=False,
	140	xlolims=False, xuplims=False,
	141	label=self._title,
	142	picker=True)
[8cb6cd6]	143
[aadf0af1]	144	# Update the list of data sets (plots) in chart
	145	self.plot_dict[self._data.id] = self.data
	146
[8cb6cd6]	147	# Now add the legend with some customizations.
[9290b1a]	148	self.legend = ax.legend(loc='upper right', shadow=True)
[b789967]	149	if self.legend:
	150	self.legend.set_picker(True)
[8cb6cd6]	151
[6d05e1d]	152	# Current labels for axes
[570a58f9]	153	if self.y_label and not is_fit:
	154	ax.set_ylabel(self.y_label)
	155	if self.x_label and not is_fit:
	156	ax.set_xlabel(self.x_label)
[6d05e1d]	157
	158	# Include scaling (log vs. linear)
	159	ax.set_xscale(self.xscale)
[3b7b218]	160	ax.set_yscale(self.yscale)
[8cb6cd6]	161
[257bd57]	162	# define the ranges
	163	self.setRange = SetGraphRange(parent=self,
	164	x_range=self.ax.get_xlim(), y_range=self.ax.get_ylim())
	165
[8cb6cd6]	166	# refresh canvas
	167	self.canvas.draw()
	168
[aadf0af1]	169	def createContextMenu(self):
[c4e5400]	170	"""
	171	Define common context menu and associated actions for the MPL widget
	172	"""
	173	self.defaultContextMenu()
	174
[aadf0af1]	175	# Separate plots
	176	self.addPlotsToContextMenu()
	177
[27313b7]	178	# Additional menu items
	179	self.contextMenu.addSeparator()
	180	self.actionAddText = self.contextMenu.addAction("Add Text")
	181	self.actionRemoveText = self.contextMenu.addAction("Remove Text")
	182	self.contextMenu.addSeparator()
	183	self.actionChangeScale = self.contextMenu.addAction("Change Scale")
	184	self.contextMenu.addSeparator()
	185	self.actionSetGraphRange = self.contextMenu.addAction("Set Graph Range")
	186	self.actionResetGraphRange =\
	187	self.contextMenu.addAction("Reset Graph Range")
	188	# Add the title change for dialogs
[aadf0af1]	189	#if self.parent:
	190	self.contextMenu.addSeparator()
	191	self.actionWindowTitle = self.contextMenu.addAction("Window Title")
[27313b7]	192
	193	# Define the callbacks
	194	self.actionAddText.triggered.connect(self.onAddText)
	195	self.actionRemoveText.triggered.connect(self.onRemoveText)
	196	self.actionChangeScale.triggered.connect(self.onScaleChange)
	197	self.actionSetGraphRange.triggered.connect(self.onSetGraphRange)
	198	self.actionResetGraphRange.triggered.connect(self.onResetGraphRange)
	199	self.actionWindowTitle.triggered.connect(self.onWindowsTitle)
[c4e5400]	200
[aadf0af1]	201	def addPlotsToContextMenu(self):
	202	"""
	203	Adds operations on all plotted sets of data to the context menu
	204	"""
	205	for id in self.plot_dict.keys():
	206	plot = self.plot_dict[id]
[b789967]	207
	208	name = plot.name if plot.name else plot.title
[aadf0af1]	209	plot_menu = self.contextMenu.addMenu('&%s' % name)
	210
	211	self.actionDataInfo = plot_menu.addAction("&DataInfo")
	212	self.actionDataInfo.triggered.connect(
	213	functools.partial(self.onDataInfo, plot))
	214
	215	self.actionSavePointsAsFile = plot_menu.addAction("&Save Points as a File")
	216	self.actionSavePointsAsFile.triggered.connect(
	217	functools.partial(self.onSavePoints, plot))
	218	plot_menu.addSeparator()
	219
	220	if plot.id != 'fit':
	221	self.actionLinearFit = plot_menu.addAction('&Linear Fit')
[570a58f9]	222	self.actionLinearFit.triggered.connect(
	223	functools.partial(self.onLinearFit, id))
[aadf0af1]	224	plot_menu.addSeparator()
	225
	226	self.actionRemovePlot = plot_menu.addAction("Remove")
	227	self.actionRemovePlot.triggered.connect(
	228	functools.partial(self.onRemovePlot, id))
	229
	230	if not plot.is_data:
	231	self.actionFreeze = plot_menu.addAction('&Freeze')
	232	self.actionFreeze.triggered.connect(
	233	functools.partial(self.onFreeze, id))
	234	plot_menu.addSeparator()
	235
	236	if plot.is_data:
	237	self.actionHideError = plot_menu.addAction("Hide Error Bar")
	238	if plot.dy is not None and plot.dy != []:
	239	if plot.hide_error:
	240	self.actionHideError.setText('Show Error Bar')
	241	else:
	242	self.actionHideError.setEnabled(False)
	243	self.actionHideError.triggered.connect(
	244	functools.partial(self.onToggleHideError, id))
	245	plot_menu.addSeparator()
	246
	247	self.actionModifyPlot = plot_menu.addAction('&Modify Plot Property')
[87cc73a]	248	self.actionModifyPlot.triggered.connect(
	249	functools.partial(self.onModifyPlot, id))
[aadf0af1]	250
	251	def createContextMenuQuick(self):
[6d05e1d]	252	"""
	253	Define context menu and associated actions for the quickplot MPL widget
	254	"""
[c4e5400]	255	# Default actions
	256	self.defaultContextMenu()
	257
	258	# Additional actions
[6d05e1d]	259	self.actionToggleGrid = self.contextMenu.addAction("Toggle Grid On/Off")
	260	self.contextMenu.addSeparator()
	261	self.actionChangeScale = self.contextMenu.addAction("Change Scale")
	262
	263	# Define the callbacks
	264	self.actionToggleGrid.triggered.connect(self.onGridToggle)
	265	self.actionChangeScale.triggered.connect(self.onScaleChange)
	266
	267	def onScaleChange(self):
	268	"""
	269	Show a dialog allowing axes rescaling
	270	"""
	271	if self.properties.exec_() == QtGui.QDialog.Accepted:
[570a58f9]	272	self.xLogLabel, self.yLogLabel = self.properties.getValues()
	273	self.xyTransform(self.xLogLabel, self.yLogLabel)
[6d05e1d]	274
[27313b7]	275	def onAddText(self):
	276	"""
	277	Show a dialog allowing adding custom text to the chart
	278	"""
[9290b1a]	279	if self.addText.exec_() == QtGui.QDialog.Accepted:
	280	# Retrieve the new text, its font and color
	281	extra_text = self.addText.text()
	282	extra_font = self.addText.font()
	283	extra_color = self.addText.color()
	284
	285	# Place the text on the screen at (0,0)
	286	pos_x = self.x_click
	287	pos_y = self.y_click
	288
	289	# Map QFont onto MPL font
	290	mpl_font = FontProperties()
	291	mpl_font.set_size(int(extra_font.pointSize()))
	292	mpl_font.set_family(str(extra_font.family()))
	293	mpl_font.set_weight(int(extra_font.weight()))
	294	# MPL style names
	295	styles = ['normal', 'italic', 'oblique']
	296	# QFont::Style maps directly onto the above
	297	try:
	298	mpl_font.set_style(styles[extra_font.style()])
	299	except:
	300	pass
	301
	302	if len(extra_text) > 0:
	303	new_text = self.ax.text(str(pos_x),
	304	str(pos_y),
	305	extra_text,
	306	color=extra_color,
	307	fontproperties=mpl_font)
	308	# Update the list of annotations
	309	self.textList.append(new_text)
	310	self.canvas.draw_idle()
[27313b7]	311
	312	def onRemoveText(self):
	313	"""
[9290b1a]	314	Remove the most recently added text
[27313b7]	315	"""
[d3ca363]	316	num_text = len(self.textList)
	317	if num_text < 1:
	318	return
	319	txt = self.textList[num_text - 1]
	320	text_remove = txt.get_text()
	321	txt.remove()
	322	self.textList.remove(txt)
	323
	324	self.canvas.draw_idle()
[27313b7]	325
	326	def onSetGraphRange(self):
	327	"""
	328	Show a dialog allowing setting the chart ranges
	329	"""
[d3ca363]	330	# min and max of data
	331	if self.setRange.exec_() == QtGui.QDialog.Accepted:
[257bd57]	332	x_range = self.setRange.xrange()
	333	y_range = self.setRange.yrange()
	334	if x_range is not None and y_range is not None:
	335	self.ax.set_xlim(x_range)
	336	self.ax.set_ylim(y_range)
	337	self.canvas.draw_idle()
[27313b7]	338
	339	def onResetGraphRange(self):
	340	"""
[d3ca363]	341	Resets the chart X and Y ranges to their original values
[27313b7]	342	"""
[d3ca363]	343	x_range = (self.data.x.min(), self.data.x.max())
	344	y_range = (self.data.y.min(), self.data.y.max())
[257bd57]	345	if x_range is not None and y_range is not None:
	346	self.ax.set_xlim(x_range)
	347	self.ax.set_ylim(y_range)
	348	self.canvas.draw_idle()
[27313b7]	349
[570a58f9]	350	def onLinearFit(self, id):
[aadf0af1]	351	"""
	352	Creates and displays a simple linear fit for the selected plot
	353	"""
[570a58f9]	354	selected_plot = self.plot_dict[id]
	355
	356	maxrange = (min(selected_plot.x), max(selected_plot.x))
	357	fitrange = self.ax.get_xlim()
	358
	359	fit_dialog = LinearFit(parent=self,
	360	data=selected_plot,
	361	max_range=maxrange,
	362	fit_range=fitrange,
	363	xlabel=self.xLogLabel,
	364	ylabel=self.yLogLabel)
	365	if fit_dialog.exec_() == QtGui.QDialog.Accepted:
	366	return
[aadf0af1]	367
[87cc73a]	368	def replacePlot(self, id, new_plot):
	369	"""
	370	Remove plot 'id' and add 'new_plot' to the chart.
	371	This effectlvely refreshes the chart with changes to one of its plots
	372	"""
	373	self.removePlot(id)
	374	self.plot(data=new_plot)
	375
[aadf0af1]	376	def onRemovePlot(self, id):
	377	"""
[570a58f9]	378	Responds to the plot delete action
	379	"""
	380	self.removePlot(id)
	381
	382	if len(self.plot_dict) == 0:
	383	# last plot: graph is empty must be the panel must be destroyed
	384	self.parent.close()
	385
	386	def removePlot(self, id):
	387	"""
[aadf0af1]	388	Deletes the selected plot from the chart
	389	"""
[570a58f9]	390	if id not in self.plot_dict:
	391	return
	392
[aadf0af1]	393	selected_plot = self.plot_dict[id]
	394
	395	plot_dict = copy.deepcopy(self.plot_dict)
	396
[b46f285]	397	# Labels might have been changed
	398	xl = self.ax.xaxis.label.get_text()
	399	yl = self.ax.yaxis.label.get_text()
	400
[aadf0af1]	401	self.plot_dict = {}
	402
	403	plt.cla()
	404	self.ax.cla()
	405
	406	for ids in plot_dict:
	407	if ids != id:
[b46f285]	408	self.plot(data=plot_dict[ids], hide_error=plot_dict[ids].hide_error)
	409
	410	# Reset the labels
	411	self.ax.set_xlabel(xl)
	412	self.ax.set_ylabel(yl)
	413	self.canvas.draw()
[aadf0af1]	414
	415	def onFreeze(self, id):
	416	"""
	417	Freezes the selected plot to a separate chart
	418	"""
	419	plot = self.plot_dict[id]
	420	self.manager.add_data(data_list=[plot])
	421
[87cc73a]	422	def onModifyPlot(self, id):
[aadf0af1]	423	"""
	424	Allows for MPL modifications to the selected plot
	425	"""
[87cc73a]	426	selected_plot = self.plot_dict[id]
	427
	428	# Old style color - single integer for enum color
	429	# New style color - #hhhhhh
	430	color = selected_plot.custom_color
	431	# marker symbol and size
	432	marker = selected_plot.symbol
	433	marker_size = selected_plot.markersize
	434	# plot name
	435	legend = selected_plot.title
	436
	437	plotPropertiesWidget = PlotProperties(self,
	438	color=color,
	439	marker=marker,
	440	marker_size=marker_size,
	441	legend=legend)
	442	if plotPropertiesWidget.exec_() == QtGui.QDialog.Accepted:
	443	# Update Data1d
[0f3c22d]	444	selected_plot.markersize = plotPropertiesWidget.markersize()
	445	selected_plot.custom_color = plotPropertiesWidget.color()
	446	selected_plot.symbol = plotPropertiesWidget.marker()
	447	selected_plot.title = plotPropertiesWidget.legend()
[87cc73a]	448
	449	# Redraw the plot
	450	self.replacePlot(id, selected_plot)
[aadf0af1]	451
	452	def onToggleHideError(self, id):
	453	"""
	454	Toggles hide error/show error menu item
	455	"""
	456	selected_plot = self.plot_dict[id]
	457	current = selected_plot.hide_error
	458
	459	# Flip the flag
	460	selected_plot.hide_error = not current
	461
	462	plot_dict = copy.deepcopy(self.plot_dict)
	463	self.plot_dict = {}
	464
	465	# Clean the canvas
	466	plt.cla()
	467	self.ax.cla()
	468
	469	# Recreate the plots but reverse the error flag for the current
	470	for ids in plot_dict:
	471	if ids == id:
	472	self.plot(data=plot_dict[ids], hide_error=(not current))
	473	else:
	474	self.plot(data=plot_dict[ids], hide_error=plot_dict[ids].hide_error)
	475
[6d05e1d]	476	def xyTransform(self, xLabel="", yLabel=""):
	477	"""
	478	Transforms x and y in View and set the scale
	479	"""
[570a58f9]	480	# Transform all the plots on the chart
	481	for id in self.plot_dict.keys():
	482	current_plot = self.plot_dict[id]
	483	if current_plot.id == "fit":
	484	self.removePlot(id)
	485	continue
[fed94a2]	486
[570a58f9]	487	new_xlabel, new_ylabel, xscale, yscale =\
	488	GuiUtils.xyTransform(current_plot, xLabel, yLabel)
	489	self.xscale = xscale
	490	self.yscale = yscale
[b46f285]	491
[570a58f9]	492	# Plot the updated chart
	493	self.removePlot(id)
[b46f285]	494
	495	# This assignment will wrap the label in Latex "$"
	496	self.xLabel = new_xlabel
	497	self.yLabel = new_ylabel
[a66ff280]	498	# Directly overwrite the data to avoid label reassignment
	499	self._data = current_plot
[87cc73a]	500	self.plot()
[570a58f9]	501
	502	pass # debug hook
	503
[fed94a2]	504	def onFitDisplay(self, fit_data):
	505	"""
	506	Add a linear fitting line to the chart
	507	"""
	508	# Create new data structure with fitting result
	509	tempx = fit_data[0]
	510	tempy = fit_data[1]
	511	self.fit_result.x = []
	512	self.fit_result.y = []
	513	self.fit_result.x = tempx
	514	self.fit_result.y = tempy
	515	self.fit_result.dx = None
	516	self.fit_result.dy = None
	517
	518	#Remove another Fit, if exists
	519	self.removePlot("fit")
	520
	521	self.fit_result.reset_view()
	522	#self.offset_graph()
	523
	524	# Set plot properties
	525	self.fit_result.id = 'fit'
	526	self.fit_result.title = 'Fit'
	527	self.fit_result.name = 'Fit'
	528
	529	# Plot the line
[87cc73a]	530	self.plot(data=self.fit_result, marker='-', hide_error=True)
[fed94a2]	531
[3bdbfcc]	532	def onMplMouseDown(self, event):
	533	"""
	534	Left button down and ready to drag
	535	"""
	536	# Check that the LEFT button was pressed
[0268aed]	537	if event.button != 1:
	538	return
	539
	540	self.leftdown = True
	541	for text in self.textList:
	542	if text.contains(event)[0]: # If user has clicked on text
	543	self.selectedText = text
	544	return
	545	if event.inaxes is None:
	546	return
	547	try:
	548	self.x_click = float(event.xdata) # / size_x
	549	self.y_click = float(event.ydata) # / size_y
	550	except:
	551	self.position = None
[3bdbfcc]	552
	553	def onMplMouseUp(self, event):
	554	"""
	555	Set the data coordinates of the click
	556	"""
	557	self.x_click = event.xdata
	558	self.y_click = event.ydata
	559
	560	# Check that the LEFT button was released
	561	if event.button == 1:
	562	self.leftdown = False
	563	self.selectedText = None
	564
	565	#release the legend
	566	if self.gotLegend == 1:
	567	self.gotLegend = 0
	568
	569	def onMplMouseMotion(self, event):
	570	"""
	571	Check if the left button is press and the mouse in moving.
	572	Compute delta for x and y coordinates and then perform the drag
	573	"""
	574	if self.gotLegend == 1 and self.leftdown:
	575	self.onLegendMotion(event)
	576	return
	577
[0268aed]	578	#if self.leftdown and self.selectedText is not None:
[9f25bce]	579	if not self.leftdown or self.selectedText is None:
[3bdbfcc]	580	return
[0268aed]	581	# User has clicked on text and is dragging
[9f25bce]	582	if event.inaxes is None:
[0268aed]	583	# User has dragged outside of axes
	584	self.selectedText = None
	585	else:
	586	# Only move text if mouse is within axes
	587	self.selectedText.set_position((event.xdata, event.ydata))
	588	self.canvas.draw_idle()
	589	return
[3bdbfcc]	590
	591	def onMplPick(self, event):
	592	"""
	593	On pick legend
	594	"""
	595	legend = self.legend
[0268aed]	596	if event.artist != legend:
	597	return
	598	# Get the box of the legend.
	599	bbox = self.legend.get_window_extent()
	600	# Get mouse coordinates at time of pick.
	601	self.mouse_x = event.mouseevent.x
	602	self.mouse_y = event.mouseevent.y
	603	# Get legend coordinates at time of pick.
	604	self.legend_x = bbox.xmin
	605	self.legend_y = bbox.ymin
	606	# Indicate we picked up the legend.
	607	self.gotLegend = 1
	608
	609	#self.legend.legendPatch.set_alpha(0.5)
[3bdbfcc]	610
	611	def onLegendMotion(self, event):
	612	"""
	613	On legend in motion
	614	"""
	615	ax = event.inaxes
	616	if ax == None:
	617	return
	618	# Event occurred inside a plotting area
	619	lo_x, hi_x = ax.get_xlim()
	620	lo_y, hi_y = ax.get_ylim()
	621	# How much the mouse moved.
	622	x = mouse_diff_x = self.mouse_x - event.x
	623	y = mouse_diff_y = self.mouse_y - event.y
	624	# Put back inside
	625	if x < lo_x:
	626	x = lo_x
	627	if x > hi_x:
	628	x = hi_x
	629	if y < lo_y:
	630	y = lo_y
	631	if y > hi_y:
	632	y = hi_y
	633	# Move the legend from its previous location by that same amount
	634	loc_in_canvas = self.legend_x - mouse_diff_x, \
	635	self.legend_y - mouse_diff_y
	636	# Transform into legend coordinate system
	637	trans_axes = self.legend.parent.transAxes.inverted()
	638	loc_in_norm_axes = trans_axes.transform_point(loc_in_canvas)
	639	self.legend_pos_loc = tuple(loc_in_norm_axes)
	640	self.legend._loc = self.legend_pos_loc
	641	# self.canvas.draw()
	642	self.canvas.draw_idle()
	643
	644	def onMplWheel(self, event):
	645	"""
	646	Process mouse wheel as zoom events
	647	"""
	648	ax = event.inaxes
	649	step = event.step
	650
	651	if ax != None:
	652	# Event occurred inside a plotting area
	653	lo, hi = ax.get_xlim()
	654	lo, hi = PlotUtilities.rescale(lo, hi, step,
	655	pt=event.xdata, scale=ax.get_xscale())
	656	if not self.xscale == 'log' or lo > 0:
	657	self._scale_xlo = lo
	658	self._scale_xhi = hi
	659	ax.set_xlim((lo, hi))
	660
	661	lo, hi = ax.get_ylim()
	662	lo, hi = PlotUtilities.rescale(lo, hi, step, pt=event.ydata,
	663	scale=ax.get_yscale())
	664	if not self.yscale == 'log' or lo > 0:
	665	self._scale_ylo = lo
	666	self._scale_yhi = hi
	667	ax.set_ylim((lo, hi))
	668	else:
	669	# Check if zoom happens in the axes
	670	xdata, ydata = None, None
	671	x, y = event.x, event.y
	672
	673	for ax in self.axes:
	674	insidex, _ = ax.xaxis.contains(event)
	675	if insidex:
	676	xdata, _ = ax.transAxes.inverted().transform_point((x, y))
	677	insidey, _ = ax.yaxis.contains(event)
	678	if insidey:
	679	_, ydata = ax.transAxes.inverted().transform_point((x, y))
	680	if xdata is not None:
	681	lo, hi = ax.get_xlim()
	682	lo, hi = PlotUtilities.rescale(lo, hi, step,
	683	bal=xdata, scale=ax.get_xscale())
	684	if not self.xscale == 'log' or lo > 0:
	685	self._scale_xlo = lo
	686	self._scale_xhi = hi
	687	ax.set_xlim((lo, hi))
	688	if ydata is not None:
	689	lo, hi = ax.get_ylim()
	690	lo, hi = PlotUtilities.rescale(lo, hi, step, bal=ydata,
	691	scale=ax.get_yscale())
	692	if not self.yscale == 'log' or lo > 0:
	693	self._scale_ylo = lo
	694	self._scale_yhi = hi
	695	ax.set_ylim((lo, hi))
	696	self.canvas.draw_idle()
	697
[c4e5400]	698
[416fa8f]	699	class Plotter(QtGui.QDialog, PlotterWidget):
	700	def __init__(self, parent=None, quickplot=False):
	701
	702	QtGui.QDialog.__init__(self)
[aadf0af1]	703	PlotterWidget.__init__(self, parent=self, manager=parent, quickplot=quickplot)
[c4e5400]	704	icon = QtGui.QIcon()
	705	icon.addPixmap(QtGui.QPixmap(":/res/ball.ico"), QtGui.QIcon.Normal, QtGui.QIcon.Off)
	706	self.setWindowIcon(icon)
	707
[416fa8f]	708

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

SasView

source: sasview/src/sas/qtgui/Plotting/Plotter.py @ 0215e0a

Download in other formats: