plotting.py @ 4ddf1067

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 4ddf1067 was 7764b41, checked in by Gervaise Alina <gervyh@…>, 16 years ago
modify panel for data2d purposed
Property mode set to `100644`
File size: 20.2 KB

Rev	Line
[41d466f]	1	"""
	2	This software was developed by the University of Tennessee as part of the
	3	Distributed Data Analysis of Neutron Scattering Experiments (DANSE)
	4	project funded by the US National Science Foundation.
	5
	6	See the license text in license.txt
	7
	8	copyright 2008, University of Tennessee
	9	"""
	10
	11
	12	import wx
	13	import sys
[1c9419f4]	14	import danse.common.plottools
	15	from danse.common.plottools.PlotPanel import PlotPanel
[7764b41]	16	from danse.common.plottools.plottables import Graph,Data1D
[41d466f]	17	from sans.guicomm.events import EVT_NEW_PLOT
	18
	19	class PanelMenu(wx.Menu):
	20	plots = None
[fc2b91a]	21	graph = None
[41d466f]	22
	23	def set_plots(self, plots):
	24	self.plots = plots
	25
[fc2b91a]	26	def set_graph(self, graph):
	27	self.graph = graph
[7764b41]	28
	29	class View1DPanel1D(PlotPanel):
[41d466f]	30	"""
	31	Plot panel for use with the GUI manager
	32	"""
	33
	34	## Internal name for the AUI manager
	35	window_name = "plotpanel"
	36	## Title to appear on top of the window
	37	window_caption = "Plot Panel"
	38	## Flag to tell the GUI manager that this panel is not
	39	# tied to any perspective
	40	ALWAYS_ON = True
[383189f9]	41	## Group ID
	42	group_id = None
[41d466f]	43
	44	def __init__(self, parent, id = -1, color = None,\
	45	dpi = None, style = wx.NO_FULL_REPAINT_ON_RESIZE, **kwargs):
	46	"""
	47	Initialize the panel
	48	"""
	49	PlotPanel.__init__(self, parent, id = id, style = style, **kwargs)
	50
	51	## Reference to the parent window
	52	self.parent = parent
	53	## Plottables
	54	self.plots = {}
	55
	56	## Unique ID (from gui_manager)
	57	self.uid = None
	58
[278cc25]	59	## Action IDs for internal call-backs
	60	self.action_ids = {}
	61
[41d466f]	62	## Graph
	63	self.graph = Graph()
	64	self.graph.xaxis("\\rm{Q}", 'A^{-1}')
	65	self.graph.yaxis("\\rm{Intensity} ","cm^{-1}")
	66	self.graph.render(self)
[49b3ddd]	67
[d0802c3]	68	def _reset(self):
	69	"""
	70	Resets internal data and graph
	71	"""
	72	self.graph.reset()
	73	self.plots = {}
	74	self.action_ids = {}
	75
[41d466f]	76	def _onEVT_1DREPLOT(self, event):
	77	"""
	78	Data is ready to be displayed
	79	@param event: data event
	80	"""
	81	#TODO: Check for existence of plot attribute
[c594365]	82
	83	# Check whether this is a replot. If we ask for a replot
	84	# and the plottable no longer exists, ignore the event.
	85	if hasattr(event, "update") and event.update==True \
	86	and event.plot.name not in self.plots.keys():
	87	return
[41d466f]	88
[d0802c3]	89	if hasattr(event, "reset"):
	90	self._reset()
	91
[41d466f]	92	is_new = True
	93	if event.plot.name in self.plots.keys():
[fc2b91a]	94	# Check whether the class of plottable changed
	95	if not event.plot.__class__==self.plots[event.plot.name].__class__:
	96	self.graph.delete(self.plots[event.plot.name])
	97	else:
	98	is_new = False
[41d466f]	99
	100	if is_new:
	101	self.plots[event.plot.name] = event.plot
	102	self.graph.add(self.plots[event.plot.name])
	103	else:
	104	self.plots[event.plot.name].x = event.plot.x
	105	self.plots[event.plot.name].y = event.plot.y
	106	self.plots[event.plot.name].dy = event.plot.dy
	107	if hasattr(event.plot, 'dx') and hasattr(self.plots[event.plot.name], 'dx'):
	108	self.plots[event.plot.name].dx = event.plot.dx
	109
	110
	111	# Check axis labels
	112	#TODO: Should re-factor this
[278cc25]	113	#if event.plot._xunit != self.graph.prop["xunit"]:
	114	self.graph.xaxis(event.plot._xaxis, event.plot._xunit)
[41d466f]	115
[278cc25]	116	#if event.plot._yunit != self.graph.prop["yunit"]:
	117	self.graph.yaxis(event.plot._yaxis, event.plot._yunit)
[41d466f]	118
[7bb61da]	119	# Set the view scale for all plots
	120	self._onEVT_FUNC_PROPERTY()
	121
[41d466f]	122	self.graph.render(self)
	123	self.subplot.figure.canvas.draw_idle()
	124
[9a3adab]	125	def onLeftDown(self,event):
	126	""" left button down and ready to drag"""
	127	from sans.guicomm.events import StatusEvent
	128	PlotPanel.onLeftDown(self, event)
	129	ax = event.inaxes
	130	if ax != None:
	131	position = "x: %8.3g y: %8.3g" % (event.xdata, event.ydata)
	132	wx.PostEvent(self.parent, StatusEvent(status=position))
	133
[6fc9a2a]	134	def _onRemove(self, event):
	135	"""
	136	"""
	137	if not self.graph.selected_plottable == None:
	138	print self.graph.selected_plottable
	139
	140
	141	self.graph.delete(self.plots[self.graph.selected_plottable])
	142	del self.plots[self.graph.selected_plottable]
	143	self.graph.render(self)
	144	self.subplot.figure.canvas.draw_idle()
	145
[9a3adab]	146
[41d466f]	147	def onContextMenu(self, event):
	148	"""
	149	1D plot context menu
	150	@param event: wx context event
	151	"""
	152	#slicerpop = wx.Menu()
	153	slicerpop = PanelMenu()
	154	slicerpop.set_plots(self.plots)
[fc2b91a]	155	slicerpop.set_graph(self.graph)
[41d466f]	156
[278cc25]	157	# Option to save the data displayed
[fc2b91a]	158
	159	#for plot in self.graph.plottables:
	160	if self.graph.selected_plottable in self.plots:
	161	plot = self.plots[self.graph.selected_plottable]
[96f13a9]	162	id = wx.NewId()
[278cc25]	163	name = plot.name
	164	slicerpop.Append(id, "&Save %s points" % name)
	165	self.action_ids[str(id)] = plot
	166	wx.EVT_MENU(self, id, self._onSave)
	167
[6fc9a2a]	168	# Option to delete plottable
	169	id = wx.NewId()
	170	slicerpop.Append(id, "Remove %s curve" % name)
	171	self.action_ids[str(id)] = plot
	172	wx.EVT_MENU(self, id, self._onRemove)
	173
	174	# Option to hide
	175	#TODO: implement functionality to hide a plottable (legend click)
	176	slicerpop.AppendSeparator()
	177
[41d466f]	178	# Various plot options
	179	id = wx.NewId()
	180	slicerpop.Append(id,'&Save image', 'Save image as PNG')
	181	wx.EVT_MENU(self, id, self.onSaveImage)
	182
[fc2b91a]	183
[2310d69]	184	item_list = self.parent.get_context_menu(self.graph)
[fc2b91a]	185	if (not item_list==None) and (not len(item_list)==0):
	186	slicerpop.AppendSeparator()
	187	for item in item_list:
	188	try:
	189	id = wx.NewId()
	190	slicerpop.Append(id, item[0], item[1])
	191	wx.EVT_MENU(self, id, item[2])
	192	except:
	193	print sys.exc_value
	194	print RuntimeError, "View1DPanel.onContextMenu: bad menu item"
[41d466f]	195
	196	slicerpop.AppendSeparator()
	197
[fc2b91a]	198	if self.graph.selected_plottable in self.plots:
	199	if self.plots[self.graph.selected_plottable].__class__.__name__=="Theory1D":
	200	id = wx.NewId()
	201	slicerpop.Append(id, '&Add errors to data')
	202	wx.EVT_MENU(self, id, self._on_add_errors)
[925155f]	203	else:
	204	id = wx.NewId()
	205	slicerpop.Append(id, '&Linear fit')
	206	wx.EVT_MENU(self, id, self.onFitting)
[1c9419f4]	207
[7764b41]	208
[fc2b91a]	209
[41d466f]	210	id = wx.NewId()
[2310d69]	211	slicerpop.Append(id, '&Change scale')
	212	wx.EVT_MENU(self, id, self._onProperties)
[96f13a9]	213
	214	id = wx.NewId()
[fc2b91a]	215	#slicerpop.AppendSeparator()
[96f13a9]	216	slicerpop.Append(id, '&Reset Graph')
	217	wx.EVT_MENU(self, id, self.onResetGraph)
[41d466f]	218
	219	pos = event.GetPosition()
	220	pos = self.ScreenToClient(pos)
	221	self.PopupMenu(slicerpop, pos)
	222
[7764b41]	223
[fc2b91a]	224	def _on_add_errors(self, evt):
	225	"""
	226	Compute reasonable errors for a data set without
	227	errors and transorm the plottable to a Data1D
	228	"""
	229	import math
	230	import numpy
[671d14e1]	231	import time
[fc2b91a]	232
	233	if not self.graph.selected_plottable == None:
	234	length = len(self.plots[self.graph.selected_plottable].x)
	235	dy = numpy.zeros(length)
	236	for i in range(length):
	237	dy[i] = math.sqrt(self.plots[self.graph.selected_plottable].y[i])
	238
	239	new_plot = Data1D(self.plots[self.graph.selected_plottable].x,
	240	self.plots[self.graph.selected_plottable].y,
	241	dy=dy)
	242	new_plot.interactive = True
[7764b41]	243	new_plot.name = self.plots[self.graph.selected_plottable].name
[671d14e1]	244	if hasattr(self.plots[self.graph.selected_plottable], "group_id"):
	245	new_plot.group_id = self.plots[self.graph.selected_plottable].group_id
	246	else:
	247	new_plot.group_id = str(time.time())
	248
[fc2b91a]	249	label, unit = self.plots[self.graph.selected_plottable].get_xaxis()
	250	new_plot.xaxis(label, unit)
	251	label, unit = self.plots[self.graph.selected_plottable].get_yaxis()
	252	new_plot.yaxis(label, unit)
	253
	254	self.graph.delete(self.plots[self.graph.selected_plottable])
	255
	256	self.graph.add(new_plot)
	257	self.plots[self.graph.selected_plottable]=new_plot
	258
	259	self.graph.render(self)
	260	self.subplot.figure.canvas.draw_idle()
	261
[278cc25]	262	def _onSave(self, evt):
	263	"""
	264	Save a data set to a text file
	265	@param evt: Menu event
	266	"""
	267	import os
	268	id = str(evt.GetId())
	269	if id in self.action_ids:
	270
	271	path = None
	272	dlg = wx.FileDialog(self, "Choose a file", os.getcwd(), "", "*.txt", wx.SAVE)
	273	if dlg.ShowModal() == wx.ID_OK:
	274	path = dlg.GetPath()
	275	mypath = os.path.basename(path)
	276	print path
	277	dlg.Destroy()
	278
	279	if not path == None:
	280	out = open(path, 'w')
	281	has_errors = True
	282	if self.action_ids[id].dy==None or self.action_ids[id].dy==[]:
	283	has_errors = False
	284
	285	# Sanity check
	286	if has_errors:
	287	try:
	288	if len(self.action_ids[id].y) != len(self.action_ids[id].dy):
	289	print "Y and dY have different lengths"
	290	has_errors = False
	291	except:
	292	has_errors = False
	293
	294	if has_errors:
	295	out.write("<X> <Y> <dY>\n")
	296	else:
	297	out.write("<X> <Y>\n")
	298
	299	for i in range(len(self.action_ids[id].x)):
	300	if has_errors:
	301	out.write("%g %g %g\n" % (self.action_ids[id].x[i],
	302	self.action_ids[id].y[i],
	303	self.action_ids[id].dy[i]))
	304	else:
	305	out.write("%g %g\n" % (self.action_ids[id].x[i],
	306	self.action_ids[id].y[i]))
	307
	308	out.close()
	309
	310
[41d466f]	311	def _onToggleScale(self, event):
	312	if self.get_yscale() == 'log':
	313	self.set_yscale('linear')
	314	else:
	315	self.set_yscale('log')
	316	self.subplot.figure.canvas.draw_idle()
[7764b41]	317
	318	class View1DPanel2D( View1DPanel1D):
	319	"""
	320	Plot panel for use with the GUI manager
	321	"""
	322
	323	## Internal name for the AUI manager
	324	window_name = "plotpanel"
	325	## Title to appear on top of the window
	326	window_caption = "Plot Panel"
	327	## Flag to tell the GUI manager that this panel is not
	328	# tied to any perspective
	329	ALWAYS_ON = True
	330	## Group ID
	331	group_id = None
	332
	333	def __init__(self, parent, id = -1, color = None,\
	334	dpi = None, style = wx.NO_FULL_REPAINT_ON_RESIZE, **kwargs):
	335	"""
	336	Initialize the panel
	337	"""
	338	View1DPanel1D.__init__(self, parent, id = id, style = style, **kwargs)
	339
	340	## Reference to the parent window
	341	self.parent = parent
	342	## Plottables
	343	self.plots = {}
	344
	345	## Unique ID (from gui_manager)
	346	self.uid = None
	347
	348	## Action IDs for internal call-backs
	349	self.action_ids = {}
	350
	351	## Graph
	352	self.graph = Graph()
	353	self.graph.xaxis("\\rm{Q}", 'A^{-1}')
	354	self.graph.yaxis("\\rm{Intensity} ","cm^{-1}")
	355	self.graph.render(self)
	356
	357	def _onEVT_1DREPLOT(self, event):
	358	"""
	359	Data is ready to be displayed
	360	@param event: data event
	361	"""
	362	#TODO: Check for existence of plot attribute
	363
	364	# Check whether this is a replot. If we ask for a replot
	365	# and the plottable no longer exists, ignore the event.
	366	if hasattr(event, "update") and event.update==True \
	367	and event.plot.name not in self.plots.keys():
	368	return
	369
	370	if hasattr(event, "reset"):
	371	self._reset()
	372
	373	is_new = True
	374	if event.plot.name in self.plots.keys():
	375	# Check whether the class of plottable changed
	376	if not event.plot.__class__==self.plots[event.plot.name].__class__:
	377	self.graph.delete(self.plots[event.plot.name])
	378	else:
	379	is_new = False
	380
	381	if is_new:
	382	self.plots[event.plot.name] = event.plot
	383	self.graph.add(self.plots[event.plot.name])
	384	else:
	385	self.plots[event.plot.name].x = event.plot.x
	386	self.plots[event.plot.name].y = event.plot.y
	387	self.plots[event.plot.name].dy = event.plot.dy
	388	if hasattr(event.plot, 'dx') and hasattr(self.plots[event.plot.name], 'dx'):
	389	self.plots[event.plot.name].dx = event.plot.dx
	390
	391
	392	# Check axis labels
	393	#TODO: Should re-factor this
	394	#if event.plot._xunit != self.graph.prop["xunit"]:
	395	self.graph.xaxis(event.plot._xaxis, event.plot._xunit)
	396
	397	#if event.plot._yunit != self.graph.prop["yunit"]:
	398	self.graph.yaxis(event.plot._yaxis, event.plot._yunit)
	399
	400
	401	self.graph.render(self)
	402	self.subplot.figure.canvas.draw_idle()
	403
	404
	405	def onContextMenu(self, event):
	406	"""
	407	1D plot context menu
	408	@param event: wx context event
	409	"""
	410	#slicerpop = wx.Menu()
	411	slicerpop = PanelMenu()
	412	slicerpop.set_plots(self.plots)
	413	slicerpop.set_graph(self.graph)
	414
	415	# Option to save the data displayed
	416
	417
	418	# Various plot options
	419	id = wx.NewId()
	420	slicerpop.Append(id,'&Save image', 'Save image as PNG')
	421	wx.EVT_MENU(self, id, self.onSaveImage)
	422
	423
	424	item_list = self.parent.get_context_menu(self.graph)
	425	if (not item_list==None) and (not len(item_list)==0):
	426	slicerpop.AppendSeparator()
	427	for item in item_list:
	428	try:
	429	id = wx.NewId()
	430	slicerpop.Append(id, item[0], item[1])
	431	wx.EVT_MENU(self, id, item[2])
	432	except:
	433	print sys.exc_value
	434	print RuntimeError, "View1DPanel2D.onContextMenu: bad menu item"
	435
	436	slicerpop.AppendSeparator()
	437
	438
	439	id = wx.NewId()
	440	slicerpop.Append(id, '&Toggle Linear/Log scale')
	441	wx.EVT_MENU(self, id, self._onToggleScale)
	442
	443	pos = event.GetPosition()
	444	pos = self.ScreenToClient(pos)
	445	self.PopupMenu(slicerpop, pos)
	446
[41d466f]	447
[7764b41]	448
	449
[41d466f]	450	class Plugin:
	451	"""
	452	Plug-in class to be instantiated by the GUI manager
	453	"""
	454
	455	def __init__(self):
	456	"""
	457	Initialize the plug-in
	458	"""
	459	## Plug-in name
	460	self.sub_menu = "Plotting"
	461
	462	## Reference to the parent window
	463	self.parent = None
	464
	465	## List of panels for the simulation perspective (names)
	466	self.perspective = []
	467
	468	## Plot panels
	469	self.plot_panels = []
	470
	471
	472	def populate_menu(self, id, parent):
	473	"""
	474	Create a 'Plot' menu to list the panels
	475	available for displaying
	476	@param id: next available unique ID for wx events
	477	@param parent: parent window
	478	"""
	479	self.menu = wx.Menu()
	480	return [(id, self.menu, "Plot")]
	481
	482
	483	def get_panels(self, parent):
	484	"""
	485	Create and return a list of panel objects
	486	"""
	487	## Save a reference to the parent
	488	self.parent = parent
	489	# Connect to plotting events
	490	self.parent.Bind(EVT_NEW_PLOT, self._on_plot_event)
	491
	492	# We have no initial panels for this plug-in
	493	return []
	494
	495	def get_perspective(self):
	496	"""
	497	Get the list of panel names for this perspective
	498	"""
	499	return self.perspective
	500
	501	def on_perspective(self, event):
	502	"""
	503	Call back function for the perspective menu item.
	504	We notify the parent window that the perspective
	505	has changed.
	506	@param event: menu event
	507	"""
	508	self.parent.set_perspective(self.perspective)
	509
	510	def post_init(self):
	511	"""
	512	Post initialization call back to close the loose ends
	513	[Somehow openGL needs this call]
	514	"""
	515	pass
	516
	517	def _on_show_panel(self, event):
	518	print "_on_show_panel"
	519
	520	def _on_plot_event(self, event):
	521	"""
	522	A new plottable is being shipped to the plotting plug-in.
	523	Check whether we have a panel to put in on, or create
	524	a new one
	525	@param event: EVT_NEW_PLOT event
	526	"""
	527	# Check whether we already have a graph with the same units
	528	# as the plottable we just received.
	529	is_available = False
	530	for panel in self.plot_panels:
[be72312c]	531	if event.plot._xunit == panel.graph.prop["xunit_base"] \
	532	and event.plot._yunit == panel.graph.prop["yunit_base"]:
[383189f9]	533	if hasattr(event.plot, "group_id"):
	534	if not event.plot.group_id==None \
	535	and event.plot.group_id==panel.group_id:
	536	is_available = True
	537	panel._onEVT_1DREPLOT(event)
	538	self.parent.show_panel(panel.uid)
	539	else:
	540	# Check that the plot panel has no group ID
	541	if panel.group_id==None:
	542	is_available = True
	543	panel._onEVT_1DREPLOT(event)
	544	self.parent.show_panel(panel.uid)
[41d466f]	545
	546	# Create a new plot panel if none was available
	547	if not is_available:
[7764b41]	548	if not hasattr(event.plot,'image'):
	549	new_panel = View1DPanel1D(self.parent, -1, style=wx.RAISED_BORDER)
	550	else:
	551	new_panel = View1DPanel2D(self.parent, -1, style=wx.RAISED_BORDER)
[383189f9]	552	# Set group ID if available
	553	group_id_str = ''
	554	if hasattr(event.plot, "group_id"):
	555	if not event.plot.group_id==None:
	556	new_panel.group_id = event.plot.group_id
	557	group_id_str = ' [%s]' % event.plot.group_id
	558
[41d466f]	559	if hasattr(event, "title"):
	560	new_panel.window_caption = event.title
	561	new_panel.window_name = event.title
[383189f9]	562	#new_panel.window_caption = event.title+group_id_str
	563	#new_panel.window_name = event.title+group_id_str
[41d466f]	564
	565	event_id = self.parent.popup_panel(new_panel)
	566	self.menu.Append(event_id, new_panel.window_caption,
	567	"Show %s plot panel" % new_panel.window_caption)
	568	# Set UID to allow us to reference the panel later
	569	new_panel.uid = event_id
	570	# Ship the plottable to its panel
	571	new_panel._onEVT_1DREPLOT(event)
	572	self.plot_panels.append(new_panel)
	573
	574	return
	575

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

SasView

source: sasview/guiframe/local_perspectives/plotting/plotting.py @ 4ddf1067

Download in other formats: