Plotter1D.py @ 340c2b3

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 340c2b3 was 691643c, checked in by Gervaise Alina <gervyh@…>, 14 years ago
make panels inheriting from panelbase
Property mode set to `100644`
File size: 20.0 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
[1bf33c1]	20	import danse.common.plottools
	21	from danse.common.plottools.PlotPanel import PlotPanel
[e5664f2]	22	from danse.common.plottools.plottables import Graph
[3b69ca6]	23	from sans.guiframe import dataFitting
[1bf33c1]	24	from sans.guicomm.events import EVT_NEW_PLOT
[32c0841]	25	from sans.guicomm.events import StatusEvent
	26	from sans.guicomm.events import NewPlotEvent
	27	from sans.guicomm.events import SlicerEvent
	28	from sans.guicomm.events import ErrorDataEvent
	29	from sans.guicomm.events import RemoveDataEvent
	30	from sans.guicomm.events import AddManyDataEvent
[0d9dae8]	31	from sans.guiframe.utils import PanelMenu
[4ac8556]	32	from sans.guiframe.dataFitting import Data1D
[e5664f2]	33	from sans.guiframe.dataFitting import Theory1D
[691643c]	34	from sans.guiframe.panel_base import PanelBase
[1bf33c1]	35	from binder import BindArtist
	36
[0d9dae8]	37	DEFAULT_QMAX = 0.05
[1bf33c1]	38	DEFAULT_QSTEP = 0.001
	39	DEFAULT_BEAM = 0.005
[32c0841]	40	BIN_WIDTH = 1
[1bf33c1]	41
[0d9dae8]	42
[691643c]	43	class ModelPanel1D(PlotPanel, PanelBase):
[1bf33c1]	44	"""
[d955bf19]	45	Plot panel for use with the GUI manager
[1bf33c1]	46	"""
	47
	48	## Internal name for the AUI manager
	49	window_name = "plotpanel"
	50	## Title to appear on top of the window
	51	window_caption = "Plot Panel"
	52	## Flag to tell the GUI manager that this panel is not
	53	# tied to any perspective
	54	ALWAYS_ON = True
	55	## Group ID
	56	group_id = None
	57
[32c0841]	58	def __init__(self, parent, id=-1, color = None,
	59	dpi=None, style=wx.NO_FULL_REPAINT_ON_RESIZE, **kwargs):
	60	PlotPanel.__init__(self, parent, id=id, style=style, **kwargs)
[691643c]	61	PanelBase.__init__(self)
[1bf33c1]	62	## Reference to the parent window
	63	self.parent = parent
	64	## Plottables
	65	self.plots = {}
[869c368]	66	## save errors dy for each data plotted
[32c0841]	67	self.err_dy = {}
[6c0568b]	68	## flag to determine if the hide or show context menu item should
	69	## be displayed
[32c0841]	70	self.errors_hide = False
[1bf33c1]	71	## Unique ID (from gui_manager)
	72	self.uid = None
	73	## Action IDs for internal call-backs
	74	self.action_ids = {}
[6063b16]	75	## Default locations
	76	self._default_save_location = os.getcwd()
[1bf33c1]	77	## Graph
	78	self.graph = Graph()
	79	self.graph.xaxis("\\rm{Q}", 'A^{-1}')
[32c0841]	80	self.graph.yaxis("\\rm{Intensity} ", "cm^{-1}")
[1bf33c1]	81	self.graph.render(self)
[4ed210f4]	82
[32c0841]	83	def set_data(self, list=None):
[3c44c66]	84	"""
	85	"""
	86	pass
	87
[1bf33c1]	88	def _reset(self):
	89	"""
[d955bf19]	90	Resets internal data and graph
[1bf33c1]	91	"""
	92	self.graph.reset()
	93	self.plots = {}
	94	self.action_ids = {}
	95
	96	def _onEVT_1DREPLOT(self, event):
	97	"""
[d955bf19]	98	Data is ready to be displayed
	99
	100	:param event: data event
[1bf33c1]	101	"""
	102	#TODO: Check for existence of plot attribute
	103	# Check whether this is a replot. If we ask for a replot
	104	# and the plottable no longer exists, ignore the event.
[32c0841]	105	if hasattr(event, "update") and event.update == True \
[1bf33c1]	106	and event.plot.name not in self.plots.keys():
	107	return
	108	if hasattr(event, "reset"):
	109	self._reset()
[f5fda87]	110	# Check whether the plottable is empty
[32c0841]	111	is_empty = len(event.plot.x) == 0
[f5fda87]	112
[1bf33c1]	113	is_new = True
	114	if event.plot.name in self.plots.keys():
[f5fda87]	115	# If the plottable is empty, just remove the plottable from the graph
	116	if is_empty:
	117	self.graph.delete(self.plots[event.plot.name])
[21d99c2]	118	del self.plots[event.plot.name]
	119	else:
	120	# Check whether the class of plottable changed
[32c0841]	121	if not event.plot.__class__ == self.plots[event.plot.name].__class__:
[21d99c2]	122	#overwrite a plottable using the same name
	123	self.graph.delete(self.plots[event.plot.name])
	124	else:
	125	# plottable is already draw on the panel
	126	is_new = False
[ffd23b5]	127
[f5fda87]	128	if not is_empty:
	129	if is_new:
	130	# a new plottable overwrites a plotted one using the same id
	131	for plottable in self.plots.itervalues():
[32c0841]	132	if hasattr(event.plot, "id") and hasattr(plottable, "id"):
[0b57a57]	133	#remove the existing plot and same id and same name
[32c0841]	134	if event.plot.id == plottable.id and \
	135	event.plot.name == plottable.name:
[f5fda87]	136	self.graph.delete(plottable)
	137
	138	self.plots[event.plot.name] = event.plot
	139	self.graph.add(self.plots[event.plot.name])
	140	else:
	141	#replot the graph
	142	self.plots[event.plot.name].x = event.plot.x
	143	self.plots[event.plot.name].y = event.plot.y
	144	self.plots[event.plot.name].dy = event.plot.dy
[32c0841]	145	if hasattr(event.plot, 'dx') and \
	146	hasattr(self.plots[event.plot.name], 'dx'):
[f5fda87]	147	self.plots[event.plot.name].dx = event.plot.dx
[31482fc]	148
[1bf33c1]	149	#TODO: Should re-factor this
[6c0568b]	150	## for all added plot the option to hide error show be displayed first
[31482fc]	151	#self.errors_hide = 0
[6c0568b]	152	## Set axis labels
[1bf33c1]	153	self.graph.xaxis(event.plot._xaxis, event.plot._xunit)
	154	self.graph.yaxis(event.plot._yaxis, event.plot._yunit)
[6c0568b]	155	## Set the view scale for all plots
[1bf33c1]	156	self._onEVT_FUNC_PROPERTY()
[6c0568b]	157	## render the graph
[1bf33c1]	158	self.graph.render(self)
	159	self.subplot.figure.canvas.draw_idle()
[0b57a57]	160	#if self.errors_hide:
	161	# self._on_remove_errors(evt=None)
	162	#else:
	163	# self._on_add_errors( evt=None)
[31482fc]	164
[1bf33c1]	165	def onLeftDown(self,event):
[6c0568b]	166	"""
[d955bf19]	167	left button down and ready to drag
	168	Display the position of the mouse on the statusbar
[6c0568b]	169	"""
[1bf33c1]	170	PlotPanel.onLeftDown(self, event)
	171	ax = event.inaxes
	172	if ax != None:
	173	position = "x: %8.3g y: %8.3g" % (event.xdata, event.ydata)
	174	wx.PostEvent(self.parent, StatusEvent(status=position))
[4ed210f4]	175
	176	#post nd event to notify guiframe that this panel is on focus
	177	wx.PostEvent(self.parent, AddManyDataEvent(panel=self))
	178
[1bf33c1]	179	def _onRemove(self, event):
	180	"""
[d955bf19]	181	Remove a plottable from the graph and render the graph
	182
	183	:param event: Menu event
	184
[1bf33c1]	185	"""
[32c0841]	186	selected_plot = self.plots[self.graph.selected_plottable]
[6c0568b]	187	## Check if there is a selected graph to remove
[1debb29]	188	if not self.graph.selected_plottable == None and\
	189	self.graph.selected_plottable in self.plots.keys():
[32c0841]	190	color = self.graph.plottables[selected_plot]
	191	event = RemoveDataEvent(data=selected_plot)
[1debb29]	192	wx.PostEvent(self.parent, event)
[32c0841]	193	self.graph.delete(selected_plot)
	194	del selected_plot
[1debb29]	195	## increment graph color
	196	self.graph.color += color
[1bf33c1]	197	self.graph.render(self)
	198	self.subplot.figure.canvas.draw_idle()
[1debb29]	199
[1bf33c1]	200	def onContextMenu(self, event):
	201	"""
[d955bf19]	202	1D plot context menu
	203
	204	:param event: wx context event
	205
[1bf33c1]	206	"""
	207	slicerpop = PanelMenu()
	208	slicerpop.set_plots(self.plots)
[32c0841]	209	slicerpop.set_graph(self.graph)
[9a585d0]	210	# Various plot options
	211	id = wx.NewId()
[32c0841]	212	slicerpop.Append(id, '&Save image', 'Save image as PNG')
[9a585d0]	213	wx.EVT_MENU(self, id, self.onSaveImage)
	214	id = wx.NewId()
[32c0841]	215	slicerpop.Append(id, '&Print image', 'Print image ')
[18eba35]	216	wx.EVT_MENU(self, id, self.onPrint)
	217	id = wx.NewId()
[32c0841]	218	slicerpop.Append(id, '&Print Preview', 'image preview for print')
[18eba35]	219	wx.EVT_MENU(self, id, self.onPrinterPreview)
[9a585d0]	220	slicerpop.AppendSeparator()
	221	item_list = self.parent.get_context_menu(self.graph)
[32c0841]	222	if (not item_list == None) and (not len(item_list) == 0):
[9a585d0]	223	for item in item_list:
	224	try:
	225	id = wx.NewId()
	226	slicerpop.Append(id, item[0], item[1])
	227	wx.EVT_MENU(self, id, item[2])
	228	except:
[32c0841]	229	msg = "ModelPanel1D.onContextMenu: "
	230	msg += "bad menu item %s"%sys.exc_value
	231	wx.PostEvent(self.parent, StatusEvent(status=msg))
[9a585d0]	232	pass
	233	slicerpop.AppendSeparator()
[6c0568b]	234
[1bf33c1]	235	if self.graph.selected_plottable in self.plots:
	236	plot = self.plots[self.graph.selected_plottable]
	237	id = wx.NewId()
	238	name = plot.name
[42d27f2]	239	slicerpop.Append(id, "&Save points" )
[1bf33c1]	240	self.action_ids[str(id)] = plot
	241	wx.EVT_MENU(self, id, self._onSave)
	242	id = wx.NewId()
	243	slicerpop.Append(id, "Remove %s curve" % name)
	244	self.action_ids[str(id)] = plot
	245	wx.EVT_MENU(self, id, self._onRemove)
[9a585d0]	246	slicerpop.AppendSeparator()
[1bf33c1]	247	# Option to hide
	248	#TODO: implement functionality to hide a plottable (legend click)
	249	if self.graph.selected_plottable in self.plots:
[32c0841]	250	selected_plot = self.plots[self.graph.selected_plottable]
[3cc533e]	251	id = wx.NewId()
	252	slicerpop.Append(id, '&Linear fit')
	253	wx.EVT_MENU(self, id, self.onFitting)
[9a585d0]	254	slicerpop.AppendSeparator()
[1bf33c1]	255	id = wx.NewId()
	256	slicerpop.Append(id, '&Change scale')
	257	wx.EVT_MENU(self, id, self._onProperties)
	258	id = wx.NewId()
	259	slicerpop.Append(id, '&Reset Graph')
[d468daa]	260	wx.EVT_MENU(self, id, self.onResetGraph)
[1bf33c1]	261	pos = event.GetPosition()
	262	pos = self.ScreenToClient(pos)
	263	self.PopupMenu(slicerpop, pos)
[18eba35]	264
[869c368]	265	def _on_remove_errors(self, evt):
[6c0568b]	266	"""
[d955bf19]	267	Save name and dy of data in dictionary self.err_dy
	268	Create a new data1D with the same x, y
	269	vector and dy with zeros.
	270	post self.err_dy as event (ErrorDataEvent) for any object
	271	which wants to reconstruct the initial data.
	272
	273	:param evt: Menu event
	274
[6c0568b]	275	"""
[869c368]	276	if not self.graph.selected_plottable == None:
[6c0568b]	277	## store existing dy
[869c368]	278	name =self.plots[self.graph.selected_plottable].name
	279	dy = self.plots[self.graph.selected_plottable].dy
[32c0841]	280	self.err_dy[name] = dy
[6c0568b]	281	## Create a new dy for a new plottable
[32c0841]	282	#import numpy
[d955bf19]	283	dy = numpy.zeros(len(self.plots[self.graph.selected_plottable].y))
	284	selected_plot = self.plots[self.graph.selected_plottable]
[c81140c]	285
[32c0841]	286	if selected_plot.__class__.__name__ == "Data1D":
[8068b52]	287	# Make sure that we can pass a basic Data1D
	288	dxl = None
	289	dxw = None
	290	if hasattr(selected_plot, "dxl"):
	291	dxl = selected_plot.dxl
	292	if hasattr(selected_plot, "dxw"):
	293	dxw = selected_plot.dxw
[32c0841]	294	new_plot = Data1D(x=selected_plot.x, y=selected_plot.y,
	295	dx=selected_plot.dx, dy=dy)
	296	new_plot.dxl = dxl
	297	new_plot.dxw = dxw
[0b16ee3]	298	else:
[32c0841]	299	new_plot = Theory1D(x=selected_plot.x,
	300	y=selected_plot.y, dy=dy)
[869c368]	301	new_plot.interactive = True
[31482fc]	302	self.errors_hide = True
[869c368]	303	new_plot.name = self.plots[self.graph.selected_plottable].name
	304	if hasattr(self.plots[self.graph.selected_plottable], "group_id"):
	305	new_plot.group_id = self.plots[self.graph.selected_plottable].group_id
[32c0841]	306	if hasattr(self.plots[self.graph.selected_plottable], "id"):
[ba535a6]	307	new_plot.id = self.plots[self.graph.selected_plottable].id
	308	else:
	309	new_plot.id = str(time.time())
[869c368]	310	else:
	311	new_plot.group_id = str(time.time())
	312	new_plot.id = str(time.time())
	313	label, unit = self.plots[self.graph.selected_plottable].get_xaxis()
	314	new_plot.xaxis(label, unit)
	315	label, unit = self.plots[self.graph.selected_plottable].get_yaxis()
	316	new_plot.yaxis(label, unit)
[6c0568b]	317	## save the color of the selected plottable before it is deleted
[32c0841]	318	color = self.graph.plottables[self.plots[self.graph.selected_plottable]]
[18eba35]	319	self.graph.delete(self.plots[self.graph.selected_plottable])
[6c0568b]	320	## add newly created plottable to the graph with the save color
[0b16ee3]	321	self.graph.color += color
[32c0841]	322	self.graph.add(new_plot, color)
	323	## transforming the view of the new data into the same of
	324	#the previous data
[869c368]	325	self._onEVT_FUNC_PROPERTY()
[6c0568b]	326	## save the plot
[32c0841]	327	self.plots[self.graph.selected_plottable] = new_plot
[6c0568b]	328	## Render the graph
[869c368]	329	self.graph.render(self)
	330	self.subplot.figure.canvas.draw_idle()
[c81140c]	331	event = ErrorDataEvent(err_dy=self.err_dy)
	332	wx.PostEvent(self.parent, event)
[d955bf19]	333
[1bf33c1]	334	def _on_add_errors(self, evt):
	335	"""
[d955bf19]	336	create a new data1D witht the errors saved in self.err_dy
	337	to show errors of the plot.
	338	Compute reasonable errors for a data set without
	339	errors and transorm the plottable to a Data1D
	340
	341	:param evt: Menu event
[c81140c]	342
[d955bf19]	343	"""
[1debb29]	344	if not self.graph.selected_plottable == None \
	345	and self.graph.selected_plottable in self.plots.keys():
[6c0568b]	346	##Reset the flag to display the hide option on the context menu
[31482fc]	347	self.errors_hide = False
[6c0568b]	348	## restore dy
[1bf33c1]	349	length = len(self.plots[self.graph.selected_plottable].x)
	350	dy = numpy.zeros(length)
[32c0841]	351	selected_plot = self.plots[self.graph.selected_plottable]
[869c368]	352	try:
[c81140c]	353	dy = self.err_dy[selected_plot.name]
[869c368]	354	except:
[32c0841]	355	if hasattr(selected_plot, "dy"):
	356	dy = selected_plot.dy
[0b16ee3]	357	else:
	358	dy = numpy.zeros(selected_plot.dy)
[6c0568b]	359	## Create a new plottable data1D
[32c0841]	360	if selected_plot.__class__.__name__ == "Data1D":
[8068b52]	361	# Make sure that we can pass a basic Data1D
	362	dxl = None
	363	dxw = None
	364	if hasattr(selected_plot, "dxl"):
	365	dxl = selected_plot.dxl
	366	if hasattr(selected_plot, "dxw"):
	367	dxw = selected_plot.dxw
[32c0841]	368	new_plot = Data1D(x=selected_plot.x,
	369	y=selected_plot.y,
[3b69ca6]	370	dx=selected_plot.dx,
[4ac8556]	371	dy=dy)
	372	new_plot.dxl = dxl
[32c0841]	373	new_plot.dxw = dxw
[0b16ee3]	374	else:
	375	## Create a new plottable Theory1D
[32c0841]	376	new_plot = Theory1D(x=selected_plot.x, y=selected_plot.y, dy=dy)
[1bf33c1]	377	new_plot.interactive = True
	378	new_plot.name = self.plots[self.graph.selected_plottable].name
	379	if hasattr(self.plots[self.graph.selected_plottable], "group_id"):
	380	new_plot.group_id = self.plots[self.graph.selected_plottable].group_id
[ba535a6]	381	if hasattr(self.plots[self.graph.selected_plottable],"id"):
	382	new_plot.id = self.plots[self.graph.selected_plottable].id
	383	else:
	384	new_plot.id = str(time.time())
[1bf33c1]	385	else:
	386	new_plot.group_id = str(time.time())
[d1dd9d4]	387	new_plot.id = str(time.time())
[1bf33c1]	388
	389	label, unit = self.plots[self.graph.selected_plottable].get_xaxis()
	390	new_plot.xaxis(label, unit)
	391	label, unit = self.plots[self.graph.selected_plottable].get_yaxis()
	392	new_plot.yaxis(label, unit)
[6c0568b]	393	## save the color of the selected plottable before it is deleted
[18eba35]	394	color=self.graph.plottables[self.plots[self.graph.selected_plottable]]
	395	self.graph.delete(self.plots[self.graph.selected_plottable])
[0b16ee3]	396	self.graph.color += color
[6c0568b]	397	## add newly created plottable to the graph with the save color
[18eba35]	398	self.graph.add(new_plot, color)
[32c0841]	399	## transforming the view of the new data into the same of
	400	# the previous data
[ffd23b5]	401	self._onEVT_FUNC_PROPERTY()
[6c0568b]	402	## save the plot
[1bf33c1]	403	self.plots[self.graph.selected_plottable]=new_plot
[6c0568b]	404	## render the graph with its new content
[1bf33c1]	405	self.graph.render(self)
[8bd764d]	406	self.subplot.figure.canvas.draw_idle()
	407
[6c0568b]	408
[42d27f2]	409	def _onsaveTXT(self, path):
	410	"""
[d955bf19]	411	Save file as txt
[1abcb04]	412
[d955bf19]	413	:TODO: Refactor and remove this method. See TODO in _onSave.
	414
[42d27f2]	415	"""
	416	data = self.plots[self.graph.selected_plottable]
	417
	418	if not path == None:
	419	out = open(path, 'w')
	420	has_errors = True
[32c0841]	421	if data.dy == None or data.dy == []:
[42d27f2]	422	has_errors = False
	423	# Sanity check
	424	if has_errors:
	425	try:
	426	if len(data.y) != len(data.dy):
	427	has_errors = False
	428	except:
	429	has_errors = False
	430	if has_errors:
	431	out.write("<X> <Y> <dY>\n")
	432	else:
	433	out.write("<X> <Y>\n")
	434
	435	for i in range(len(data.x)):
	436	if has_errors:
	437	out.write("%g %g %g\n" % (data.x[i],
	438	data.y[i],
	439	data.dy[i]))
	440	else:
	441	out.write("%g %g\n" % (data.x[i],
	442	data.y[i]))
	443	out.close()
[6063b16]	444	try:
	445	self._default_save_location = os.path.dirname(path)
	446	except:
	447	pass
[8bd764d]	448
[1bf33c1]	449	def _onSave(self, evt):
	450	"""
[d955bf19]	451	Save a data set to a text file
	452
	453	:param evt: Menu event
	454
[1bf33c1]	455	"""
	456	id = str(evt.GetId())
	457	if id in self.action_ids:
	458
	459	path = None
[5fe5871c]	460	wildcard = "Text files (.txt)\|.txt\|"\
[7959f297]	461	"CanSAS 1D files(.xml)\|.xml"
[6063b16]	462	dlg = wx.FileDialog(self, "Choose a file",
[32c0841]	463	self._default_save_location,
	464	"", wildcard , wx.SAVE)
[42d27f2]	465
[1bf33c1]	466	if dlg.ShowModal() == wx.ID_OK:
	467	path = dlg.GetPath()
	468	mypath = os.path.basename(path)
[1abcb04]	469
[32c0841]	470	#TODO: This is bad design. The DataLoader is designed
	471	#to recognize extensions.
	472	# It should be a simple matter of calling the .
	473	#save(file, data, '.xml') method
[f0a9a3cc]	474	# of the DataLoader.loader.Loader class.
	475	from DataLoader.loader import Loader
	476	#Instantiate a loader
	477	loader = Loader()
	478	data = self.plots[self.graph.selected_plottable]
[32c0841]	479	format = ".txt"
[f0a9a3cc]	480	if os.path.splitext(mypath)[1].lower() == format:
	481	self._onsaveTXT( path)
[32c0841]	482	format = ".xml"
	483	if os.path.splitext(mypath)[1].lower() == format:
	484	loader.save(path, data, format)
[f0a9a3cc]	485	try:
	486	self._default_save_location = os.path.dirname(path)
	487	except:
	488	pass
[1bf33c1]	489	dlg.Destroy()

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

SasView

source: sasview/guiframe/local_perspectives/plotting/Plotter1D.py @ 340c2b3

Download in other formats: