Plotter1D.py @ 8f12385

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 8f12385 was 6d727ae, checked in by Jae Cho <jhjcho@…>, 13 years ago
moving a new feature (fast reacting plot panels) from branch
Property mode set to `100644`
File size: 15.7 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
[a07e72f]	20
[1bf33c1]	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 RemoveDataEvent
[a45037aa]	29	from sans.guiframe.events import PanelOnFocusEvent
[52b8b74]	30	from sans.guiframe.events import EVT_NEW_LOADED_DATA
[0d9dae8]	31	from sans.guiframe.utils import PanelMenu
[4ac8556]	32	from sans.guiframe.dataFitting import Data1D
[691643c]	33	from sans.guiframe.panel_base import PanelBase
[1bf33c1]	34	from binder import BindArtist
	35
[0d9dae8]	36	DEFAULT_QMAX = 0.05
[1bf33c1]	37	DEFAULT_QSTEP = 0.001
	38	DEFAULT_BEAM = 0.005
[32c0841]	39	BIN_WIDTH = 1
[1bf33c1]	40
[0d9dae8]	41
[691643c]	42	class ModelPanel1D(PlotPanel, PanelBase):
[1bf33c1]	43	"""
[d955bf19]	44	Plot panel for use with the GUI manager
[1bf33c1]	45	"""
	46
	47	## Internal name for the AUI manager
	48	window_name = "plotpanel"
	49	## Title to appear on top of the window
	50	window_caption = "Plot Panel"
	51	## Flag to tell the GUI manager that this panel is not
	52	# tied to any perspective
	53	ALWAYS_ON = True
	54	## Group ID
	55	group_id = None
	56
[32c0841]	57	def __init__(self, parent, id=-1, color = None,
	58	dpi=None, style=wx.NO_FULL_REPAINT_ON_RESIZE, **kwargs):
	59	PlotPanel.__init__(self, parent, id=id, style=style, **kwargs)
[a45037aa]	60	PanelBase.__init__(self, parent)
[1bf33c1]	61	## Reference to the parent window
	62	self.parent = parent
	63	## Plottables
	64	self.plots = {}
[52b8b74]	65	#context menu
	66	self._slicerpop = None
[a07e72f]	67
[52b8b74]	68	self._available_data = []
	69	self._menu_add_ids = []
[a07e72f]	70	self._symbol_labels = self.get_symbol_label()
	71
	72	self.hide_menu = None
[1bf33c1]	73	## Unique ID (from gui_manager)
	74	self.uid = None
[a07e72f]	75
[6063b16]	76	## Default locations
	77	self._default_save_location = os.getcwd()
[1bf33c1]	78	## Graph
	79	self.graph = Graph()
	80	self.graph.xaxis("\\rm{Q}", 'A^{-1}')
[32c0841]	81	self.graph.yaxis("\\rm{Intensity} ", "cm^{-1}")
[1bf33c1]	82	self.graph.render(self)
[6d727ae]	83
	84	# In resizing event
	85	self.resizing = False
	86	self.canvas.set_resizing(self.resizing)
	87	self.Bind(wx.EVT_SIZE, self._OnReSize)
[a07e72f]	88
	89	def get_symbol_label(self):
[52b8b74]	90	"""
[a07e72f]	91	Associates label to symbol
[52b8b74]	92	"""
[a07e72f]	93	_labels = {}
	94	i = 0
[6d727ae]	95	_labels['Circle'] = i
[a07e72f]	96	i += 1
[6d727ae]	97	_labels['Cross X '] = i
[a07e72f]	98	i += 1
	99	_labels['Triangle Down'] = i
	100	i += 1
	101	_labels['Triangle Up'] = i
	102	i += 1
	103	_labels['Triangle Left'] = i
	104	i += 1
	105	_labels['Triangle Right'] = i
	106	i += 1
[6d727ae]	107	_labels['Cross +'] = i
[a07e72f]	108	i += 1
	109	_labels['Square'] = i
	110	i += 1
[6d727ae]	111	_labels['Diamond'] = i
[a07e72f]	112	i += 1
	113	_labels['Diamond'] = i
	114	i += 1
	115	_labels['Hexagon1'] = i
	116	i += 1
	117	_labels['Hexagon2'] = i
	118	i += 1
	119	_labels['Pentagon'] = i
	120	i += 1
[6d727ae]	121	_labels['Line'] = i
[a07e72f]	122	return _labels
	123
[52b8b74]	124
[32c0841]	125	def set_data(self, list=None):
[3c44c66]	126	"""
	127	"""
	128	pass
	129
[1bf33c1]	130	def _reset(self):
	131	"""
[d955bf19]	132	Resets internal data and graph
[1bf33c1]	133	"""
	134	self.graph.reset()
	135	self.plots = {}
[6d727ae]	136
	137	def _OnReSize(self, event):
	138	"""
	139	On response of the resize of a panel, set axes_visiable False
	140	"""
	141	# ready for another event
	142	event.Skip()
	143	# set the resizing flag
	144	self.resizing = True
	145	self.canvas.set_resizing(self.resizing)
	146	self.parent.set_schedule(True)
	147
	148
	149	def set_resizing(self, resizing=False):
	150	"""
	151	Set the resizing (True/False)
	152	"""
	153	self.resizing = resizing
	154	#self.canvas.set_resizing(resizing)
[1bf33c1]	155
[6d727ae]	156	def schedule_full_draw(self, func='append'):
	157	"""
	158	Put self in schedule to full redraw list
	159	"""
	160	# append/del this panel in the schedule list
	161	self.parent.set_schedule_full_draw(self, func)
	162
	163
[a07e72f]	164	def remove_data_by_id(self, id):
	165	"""'
	166	remove data from plot
	167	"""
	168	if id in self.plots.keys():
	169	data = self.plots[id]
	170	self.graph.delete(data)
	171	data_manager = self._manager.parent.get_data_manager()
[df22224]	172	data_list, theory_list = data_manager.get_by_id(id_list=[id])
	173
	174	if id in data_list.keys():
	175	data = data_list[id]
	176	else:
	177	data = theory_list[id]
	178
[a07e72f]	179	del self.plots[id]
	180	self.graph.render(self)
	181	self.subplot.figure.canvas.draw_idle()
[df22224]	182	event = RemoveDataEvent(data=data)
	183	wx.PostEvent(self.parent, event)
[a07e72f]	184	if len(self.graph.plottables) == 0:
[df22224]	185	#onRemove: graph is empty must be the panel must be destroyed
	186	self.parent.delete_panel(self.uid)
[a07e72f]	187
	188	def plot_data(self, data):
[1bf33c1]	189	"""
[d955bf19]	190	Data is ready to be displayed
	191
	192	:param event: data event
[1bf33c1]	193	"""
[a07e72f]	194	if data.id in self.plots.keys():
	195	#replace
	196	self.graph.replace(data)
	197	self.plots[data.id] = data
	198	else:
	199	self.plots[data.id] = data
	200	self.graph.add(self.plots[data.id])
[ffd23b5]	201
[a07e72f]	202	x_label, x_unit = data.get_xaxis()
	203	y_label, y_unit = data.get_yaxis()
	204	self.graph.xaxis(x_unit, x_label)
	205	self.graph.yaxis(y_unit, y_label)
[6c0568b]	206	## Set the view scale for all plots
[1bf33c1]	207	self._onEVT_FUNC_PROPERTY()
[6d727ae]	208	## render the graph<=No need this done in canvas
	209	#self.graph.render(self)
	210	#self.subplot.figure.canvas.draw_idle()
	211
	212	def draw_plot(self):
	213	"""
	214	Draw plot
	215	"""
	216	self.draw()
	217
[a07e72f]	218
	219
[1bf33c1]	220	def onLeftDown(self,event):
[6c0568b]	221	"""
[d955bf19]	222	left button down and ready to drag
	223	Display the position of the mouse on the statusbar
[6c0568b]	224	"""
[1bf33c1]	225	PlotPanel.onLeftDown(self, event)
	226	ax = event.inaxes
	227	if ax != None:
	228	position = "x: %8.3g y: %8.3g" % (event.xdata, event.ydata)
	229	wx.PostEvent(self.parent, StatusEvent(status=position))
[4ed210f4]	230
	231	#post nd event to notify guiframe that this panel is on focus
[a45037aa]	232	wx.PostEvent(self.parent, PanelOnFocusEvent(panel=self))
[a07e72f]	233	self._manager.set_panel_on_focus(self)
[4ed210f4]	234
[a07e72f]	235	def _ontoggle_hide_error(self, event):
	236	"""
	237	Toggle error display to hide or show
	238	"""
	239
	240	selected_plot = self.plots[self.graph.selected_plottable]
	241	if self.hide_menu.GetText() == "Hide Error":
	242	selected_plot.hide_error = True
	243	else:
	244	selected_plot.hide_error = False
	245	## increment graph color
	246	self.graph.render(self)
	247	self.subplot.figure.canvas.draw_idle()
	248
[1bf33c1]	249	def _onRemove(self, event):
	250	"""
[d955bf19]	251	Remove a plottable from the graph and render the graph
	252
	253	:param event: Menu event
	254
[1bf33c1]	255	"""
[6c0568b]	256	## Check if there is a selected graph to remove
[a07e72f]	257	if self.graph.selected_plottable in self.plots.keys():
	258	selected_plot = self.plots[self.graph.selected_plottable]
	259	id = self.graph.selected_plottable
[df22224]	260	self.remove_data_by_id(id)
	261
[1bf33c1]	262	def onContextMenu(self, event):
	263	"""
[d955bf19]	264	1D plot context menu
	265
	266	:param event: wx context event
	267
[1bf33c1]	268	"""
[52b8b74]	269	self._slicerpop = PanelMenu()
	270	self._slicerpop.set_plots(self.plots)
	271	self._slicerpop.set_graph(self.graph)
[9a585d0]	272	# Various plot options
	273	id = wx.NewId()
[6d727ae]	274	self._slicerpop.Append(id, '&Save Image', 'Save image as PNG')
[9a585d0]	275	wx.EVT_MENU(self, id, self.onSaveImage)
	276	id = wx.NewId()
[6d727ae]	277	self._slicerpop.Append(id, '&Print Image', 'Print image ')
[18eba35]	278	wx.EVT_MENU(self, id, self.onPrint)
	279	id = wx.NewId()
[6d727ae]	280	self._slicerpop.Append(id, '&Print Preview', 'Print preview')
[18eba35]	281	wx.EVT_MENU(self, id, self.onPrinterPreview)
[52b8b74]	282
[6d727ae]	283	id = wx.NewId()
	284	self._slicerpop.Append(id, '&Copy to Clipboard', 'Copy to the clipboard')
	285	wx.EVT_MENU(self, id, self.OnCopyFigureMenu)
	286
	287	self._slicerpop.AppendSeparator()
	288
[52b8b74]	289	#add menu of other plugins
[a07e72f]	290	item_list = self.parent.get_context_menu(self)
[6d727ae]	291
[32c0841]	292	if (not item_list == None) and (not len(item_list) == 0):
[9a585d0]	293	for item in item_list:
	294	try:
	295	id = wx.NewId()
[52b8b74]	296	self._slicerpop.Append(id, item[0], item[1])
[9a585d0]	297	wx.EVT_MENU(self, id, item[2])
	298	except:
[32c0841]	299	msg = "ModelPanel1D.onContextMenu: "
[a07e72f]	300	msg += "bad menu item %s" % sys.exc_value
[32c0841]	301	wx.PostEvent(self.parent, StatusEvent(status=msg))
[9a585d0]	302	pass
[52b8b74]	303	self._slicerpop.AppendSeparator()
[6d727ae]	304	#id = wx.NewId()
	305	#self._slicerpop.Append(id, '&Print image', 'Print image')
[1bf33c1]	306	if self.graph.selected_plottable in self.plots:
	307	plot = self.plots[self.graph.selected_plottable]
[6d727ae]	308
	309	id = wx.NewId()
	310	self._slicerpop.Append(id, '&Linear Fit')
	311	wx.EVT_MENU(self, id, self.onFitting)
	312	self._slicerpop.AppendSeparator()
	313	id = wx.NewId()
	314	name = plot.name
	315	self._slicerpop.Append(id, "&Save Points as a File")
	316	wx.EVT_MENU(self, id, self._onSave)
	317	id = wx.NewId()
	318	self._slicerpop.Append(id, "Remove %s Curve" % name)
	319	wx.EVT_MENU(self, id, self._onRemove)
[e6a93df]	320	if not plot.is_data:
	321	id = wx.NewId()
	322	self._slicerpop.Append(id, '&Freeze', 'Freeze')
	323	wx.EVT_MENU(self, id, self.onFreeze)
[a3c96f7a]	324	symbol_menu = wx.Menu()
	325	for label in self._symbol_labels:
	326	id = wx.NewId()
	327	symbol_menu.Append(id, str(label), str(label))
	328	wx.EVT_MENU(self, id, self.onChangeSymbol)
	329	id = wx.NewId()
	330	self._slicerpop.AppendMenu(id,'&Modify Symbol', symbol_menu)
	331	self._slicerpop.AppendSeparator()
[6d727ae]	332
[a07e72f]	333	id = wx.NewId()
	334	self.hide_menu = self._slicerpop.Append(id, "Hide Error")
	335	if plot.dy is not None or plot.dy != []:
	336	if plot.hide_error :
	337	self.hide_menu.SetText('Show Error')
	338	else:
	339	self.hide_menu.SetText('Hide Error')
	340	else:
	341	self.hide_menu.Disable()
	342	wx.EVT_MENU(self, id, self._ontoggle_hide_error)
	343
[52b8b74]	344	self._slicerpop.AppendSeparator()
[1bf33c1]	345	# Option to hide
	346	#TODO: implement functionality to hide a plottable (legend click)
[df22224]	347
[a07e72f]	348
[1bf33c1]	349	id = wx.NewId()
[52b8b74]	350	self._slicerpop.Append(id, '&Change scale')
[1bf33c1]	351	wx.EVT_MENU(self, id, self._onProperties)
	352	id = wx.NewId()
[52b8b74]	353	self._slicerpop.Append(id, '&Reset Graph')
[d468daa]	354	wx.EVT_MENU(self, id, self.onResetGraph)
[1bf33c1]	355	pos = event.GetPosition()
	356	pos = self.ScreenToClient(pos)
[52b8b74]	357	self.PopupMenu(self._slicerpop, pos)
[a07e72f]	358
[e6a93df]	359	def onFreeze(self, event):
	360	"""
	361	"""
	362	plot = self.plots[self.graph.selected_plottable]
	363	self.parent.onfreeze([plot.id])
	364
[a07e72f]	365	def onChangeSymbol(self, event):
[6c0568b]	366	"""
	367	"""
[a07e72f]	368	menu = event.GetEventObject()
	369	id = event.GetId()
	370	label = menu.GetLabel(id)
	371	selected_plot = self.plots[self.graph.selected_plottable]
	372	selected_plot.symbol = self._symbol_labels[label]
	373	## Set the view scale for all plots
	374	self._onEVT_FUNC_PROPERTY()
	375	## render the graph
[6d727ae]	376	#self.graph.render(self)
	377	#self.subplot.figure.canvas.draw_idle()
[c81140c]	378
[42d27f2]	379	def _onsaveTXT(self, path):
	380	"""
[d955bf19]	381	Save file as txt
[1abcb04]	382
[d955bf19]	383	:TODO: Refactor and remove this method. See TODO in _onSave.
	384
[42d27f2]	385	"""
	386	data = self.plots[self.graph.selected_plottable]
	387
	388	if not path == None:
	389	out = open(path, 'w')
	390	has_errors = True
[32c0841]	391	if data.dy == None or data.dy == []:
[42d27f2]	392	has_errors = False
	393	# Sanity check
	394	if has_errors:
	395	try:
	396	if len(data.y) != len(data.dy):
	397	has_errors = False
	398	except:
	399	has_errors = False
	400	if has_errors:
[6d727ae]	401	if data.dx != None:
	402	out.write("<X> <Y> <dY> <dX>\n")
	403	else:
	404	out.write("<X> <Y> <dY>\n")
[42d27f2]	405	else:
	406	out.write("<X> <Y>\n")
	407
	408	for i in range(len(data.x)):
	409	if has_errors:
[6d727ae]	410	if data.dx != None:
	411	out.write("%g %g %g %g\n" % (data.x[i],
	412	data.y[i],
	413	data.dy[i],
	414	data.dx[i]))
	415	else:
	416	out.write("%g %g %g\n" % (data.x[i],
	417	data.y[i],
	418	data.dy[i]))
[42d27f2]	419	else:
	420	out.write("%g %g\n" % (data.x[i],
	421	data.y[i]))
	422	out.close()
[6063b16]	423	try:
	424	self._default_save_location = os.path.dirname(path)
	425	except:
	426	pass
[8bd764d]	427
[1bf33c1]	428	def _onSave(self, evt):
	429	"""
[d955bf19]	430	Save a data set to a text file
	431
	432	:param evt: Menu event
	433
[1bf33c1]	434	"""
[a07e72f]	435
	436	path = None
	437	wildcard = "Text files (.txt)\|.txt\|"\
	438	"CanSAS 1D files(.xml)\|.xml"
	439	dlg = wx.FileDialog(self, "Choose a file",
	440	self._default_save_location,
	441	"", wildcard , wx.SAVE)
	442
	443	if dlg.ShowModal() == wx.ID_OK:
	444	path = dlg.GetPath()
	445	mypath = os.path.basename(path)
[1bf33c1]	446
[a07e72f]	447	#TODO: This is bad design. The DataLoader is designed
	448	#to recognize extensions.
	449	# It should be a simple matter of calling the .
	450	#save(file, data, '.xml') method
	451	# of the DataLoader.loader.Loader class.
	452	from DataLoader.loader import Loader
	453	#Instantiate a loader
	454	loader = Loader()
	455	data = self.plots[self.graph.selected_plottable]
	456	format = ".txt"
	457	if os.path.splitext(mypath)[1].lower() == format:
	458	self._onsaveTXT( path)
	459	format = ".xml"
	460	if os.path.splitext(mypath)[1].lower() == format:
	461	loader.save(path, data, format)
	462	try:
	463	self._default_save_location = os.path.dirname(path)
	464	except:
	465	pass
	466	dlg.Destroy()

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

SasView

source: sasview/guiframe/local_perspectives/plotting/Plotter1D.py @ 8f12385

Download in other formats: