plottable_interactor.py @ 286c757

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.2ticket-1009ticket-1094-headlessticket-1242-2d-resolutionticket-1243ticket-1249ticket885unittest-saveload

Last change on this file since 286c757 was d7bb526, checked in by Piotr Rozyczko <piotr.rozyczko@…>, 9 years ago
Refactored plottools into sasgui
Property mode set to `100644`
File size: 8.9 KB

Rev	Line
[a9d5684]	1	"""
	2	This module allows more interaction with the plot
	3	"""
	4	from BaseInteractor import _BaseInteractor
	5
	6	class PointInteractor(_BaseInteractor):
	7	"""
	8	"""
	9	def __init__(self, base, axes, color='black', zorder=3, id=''):
	10	"""
	11	"""
	12	_BaseInteractor.__init__(self, base, axes, color=color)
	13	self.zorder = zorder
	14	self.id = id
	15	self.color = color
	16	self.colorlist = ['b', 'g', 'r', 'c', 'm', 'y', 'k']
	17	self.symbollist = ['o', 'x', '^', 'v', '<', '>',
	18	'+', 's', 'd', 'D', 'h', 'H', 'p', '-', '--',
	19	'vline', 'step']
	20	self.markersize = None
	21	self.marker = None
	22	self.marker2 = None
	23	self._button_down = False
	24	self._context_menu = False
	25	self._dragged = False
	26	self.connect_markers([self.axes])
[3477478]	27
[a9d5684]	28	def _color(self, c):
	29	"""Return a particular colour"""
	30	return self.colorlist[c % len(self.colorlist)]
[3477478]	31
[a9d5684]	32	def _symbol(self, s):
	33	"""Return a particular symbol"""
	34	return self.symbollist[s % len(self.symbollist)]
	35
[3477478]	36	def points(self, x, y, dx=None, dy=None, color=0, symbol=0, zorder=1,
[a9d5684]	37	markersize=5, label=None, hide_error=False):
	38	"""
	39	"""
[3477478]	40	# Draw curve
[a9d5684]	41	if self._symbol(symbol) == '-' or self._symbol(symbol) == '--':
	42	l_width = markersize * 0.4
[3477478]	43	return self.curve(x=x, y=y, color=color, symbol=symbol,
[a9d5684]	44	label=label, width=l_width)
[3477478]	45	# return
[a9d5684]	46	if self._symbol(symbol) == 'vline':
	47	l_width = markersize * 0.4
[3477478]	48	return self.vline(x=x, y=y, color=color, label=label, width=l_width)
[a9d5684]	49	if self._symbol(symbol) == 'step':
	50	l_width = markersize * 0.4
[3477478]	51	return self.step(x=x, y=y, color=color, label=label, width=l_width)
[a9d5684]	52	if not self.marker == None:
	53	self.base.connect.clear([self.marker])
	54	self.color = self._color(color)
	55	if self.markersize != None:
	56	markersize = self.markersize
	57	# Convert tuple (lo,hi) to array [(x-lo),(hi-x)]
	58	if dx != None and type(dx) == type(()):
[3477478]	59	dx = nx.vstack((x - dx[0], dx[1] - x)).transpose()
[a9d5684]	60	if dy != None and type(dy) == type(()):
[3477478]	61	dy = nx.vstack((y - dy[0], dy[1] - y)).transpose()
	62
[a9d5684]	63	if dx == None and dy == None:
[3477478]	64	# zorder = 1
[a9d5684]	65	self.marker = self.axes.plot(x, y, color=self.color,
	66	marker=self._symbol(symbol),
	67	markersize=markersize,
	68	linestyle='', label=label,
	69	zorder=zorder)[0]
	70	else:
[3477478]	71
[a9d5684]	72	if hide_error:
[3477478]	73	# zorder = 1
[a9d5684]	74	self.marker = self.axes.plot(x, y, color=self.color,
	75	marker=self._symbol(symbol),
	76	markersize=markersize,
	77	linestyle='', label=label,
	78	zorder=1)[0]
	79	else:
[3477478]	80	# zorder = 2
[a9d5684]	81	self.marker = self.axes.errorbar(x, y, yerr=dy,
	82	xerr=None,
	83	ecolor=self.color,
	84	color=self.color,
	85	capsize=2,
	86	linestyle='',
	87	barsabove=False,
	88	marker=self._symbol(symbol),
	89	markersize=markersize,
	90	lolims=False, uplims=False,
	91	xlolims=False, xuplims=False,
	92	label=label,
	93	zorder=1)[0]
[3477478]	94
[a9d5684]	95	self.connect_markers([self.marker])
	96	self.update()
[3477478]	97
	98	def curve(self, x, y, dy=None, color=0, symbol=0, zorder=10,
[a9d5684]	99	label=None, width=2.0):
	100	"""
	101	"""
	102	if not self.marker == None:
	103	self.base.connect.clear([self.marker])
	104	self.color = self._color(color)
	105	self.marker = self.axes.plot(x, y, color=self.color, lw=width,
	106	marker='', linestyle=self._symbol(symbol),
	107	label=label, zorder=zorder)[0]
[3477478]	108
[a9d5684]	109	self.connect_markers([self.marker])
	110	self.update()
	111
[3477478]	112
	113	def vline(self, x, y, dy=None, color=0, symbol=0, zorder=1,
[a9d5684]	114	label=None, width=2.0):
	115	"""
	116	"""
	117	if not self.marker == None:
	118	self.base.connect.clear([self.marker])
	119	self.color = self._color(color)
	120	if min(y) < 0:
	121	y_min = 0.0
	122	else:
[3477478]	123	y_min = min(y) * 9 / 10
	124	self.marker = self.axes.vlines(x=x, ymin=y_min, ymax=y,
	125	color=self.color,
	126	linestyle='-', label=label,
	127	lw=width, zorder=zorder)
[a9d5684]	128	self.connect_markers([self.marker])
	129	self.update()
	130
	131	def step(self, x, y, dy=None, color=0, symbol=0, zorder=1,
[3477478]	132	label=None, width=2.0):
[a9d5684]	133	"""
	134	"""
	135	if not self.marker == None:
	136	self.base.connect.clear([self.marker])
	137	self.color = self._color(color)
	138	self.marker = self.axes.step(x, y, color=self.color,
[3477478]	139	marker='',
	140	linestyle='-', label=label,
	141	lw=width, zorder=zorder)[0]
[a9d5684]	142	self.connect_markers([self.marker])
	143	self.update()
[3477478]	144
[a9d5684]	145	def connect_markers(self, markers):
	146	"""
	147	Connect markers to callbacks
	148	"""
	149	for h in markers:
	150	connect = self.base.connect
	151	connect('enter', h, self._on_enter)
	152	connect('leave', h, self._on_leave)
	153	connect('click', h, self._on_click)
	154	connect('release', h, self._on_release)
	155	connect('key', h, self.onKey)
	156
	157	def clear(self):
	158	print "plottable_interactor.clear()"
[3477478]	159
[a9d5684]	160	def _on_click(self, evt):
	161	"""
	162	Called when a mouse button is clicked
	163	from within the boundaries of an artist.
	164	"""
	165	if self._context_menu == True:
	166	self._context_menu = False
	167	evt.artist = self.marker
	168	self._on_leave(evt)
	169
	170	def _on_release(self, evt):
	171	"""
	172	Called when a mouse button is released
	173	within the boundaries of an artist
	174	"""
	175	# Check to see whether we are about to pop
	176	# the context menu up
	177	if evt.button == 3:
	178	self._context_menu = True
[3477478]	179
[a9d5684]	180	def _on_enter(self, evt):
	181	"""
	182	Called when we are entering the boundaries
	183	of an artist.
	184	"""
	185	if not evt.artist.__class__.__name__ == "AxesSubplot":
	186	self.base.plottable_selected(self.id)
[3477478]	187
[a9d5684]	188	if evt.artist.get_color() == 'y':
	189	try:
	190	evt.artist.set_color('b')
	191	except:
	192	evt.artist.set_color_cycle('b')
	193	if hasattr(evt.artist, "set_facecolor"):
	194	evt.artist.set_facecolor('b')
	195	if hasattr(evt.artist, "set_edgecolor"):
	196	evt.artist.set_edgecolor('b')
	197	else:
	198	try:
	199	evt.artist.set_color('y')
	200	except:
	201	evt.artist.set_color_cycle('y')
	202	if hasattr(evt.artist, "set_facecolor"):
	203	evt.artist.set_facecolor('y')
	204	if hasattr(evt.artist, "set_edgecolor"):
	205	evt.artist.set_edgecolor('y')
	206
	207	self.axes.figure.canvas.draw_idle()
[3477478]	208
[a9d5684]	209	def _on_leave(self, evt):
	210	"""
	211	Called when we are leaving the boundaries
	212	of an artist.
	213	"""
	214	if not evt.artist.__class__.__name__ == "AxesSubplot":
	215	if self._context_menu == False:
	216	self.base.plottable_selected(None)
	217	try:
	218	evt.artist.set_color(self.color)
	219	except:
	220	evt.artist.set_color_cycle(self.color)
	221	if hasattr(evt.artist, "set_facecolor"):
	222	evt.artist.set_facecolor(self.color)
	223	if hasattr(evt.artist, "set_edgecolor"):
	224	evt.artist.set_edgecolor(self.color)
	225	self.axes.figure.canvas.draw_idle()
[3477478]	226
[a9d5684]	227	def update(self):
	228	"""
	229	Update
	230	"""
	231	pass

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

SasView

source: sasview/src/sas/sasgui/plottools/plottable_interactor.py @ 286c757

Download in other formats: