Plotter1D.py @ 3d2d7f60

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 3d2d7f60 was df7046f, checked in by Gervaise Alina <gervyh@…>, 14 years ago
working on loading multiple file
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
[df7046f]	24	from sans.guiframe.events import EVT_NEW_PLOT
	25	from sans.guiframe.events import StatusEvent
	26	from sans.guiframe.events import NewPlotEvent
	27	from sans.guiframe.events import SlicerEvent
	28	from sans.guiframe.events import ErrorDataEvent
	29	from sans.guiframe.events import RemoveDataEvent
	30	from sans.guiframe.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 @ 3d2d7f60

Download in other formats: