Plotter1D.py @ 8960479

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 8960479 was a436b2e, checked in by Jae Cho <jhjcho@…>, 13 years ago
Still working on SLDpanel
Property mode set to `100644`
File size: 16.6 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
[a45037aa]	28	from sans.guiframe.events import PanelOnFocusEvent
[52b8b74]	29	from sans.guiframe.events import EVT_NEW_LOADED_DATA
[0d9dae8]	30	from sans.guiframe.utils import PanelMenu
[4ac8556]	31	from sans.guiframe.dataFitting import Data1D
[691643c]	32	from sans.guiframe.panel_base import PanelBase
[1bf33c1]	33	from binder import BindArtist
	34
[0d9dae8]	35	DEFAULT_QMAX = 0.05
[1bf33c1]	36	DEFAULT_QSTEP = 0.001
	37	DEFAULT_BEAM = 0.005
[32c0841]	38	BIN_WIDTH = 1
[1bf33c1]	39
[0d9dae8]	40
[691643c]	41	class ModelPanel1D(PlotPanel, PanelBase):
[1bf33c1]	42	"""
[d955bf19]	43	Plot panel for use with the GUI manager
[1bf33c1]	44	"""
	45
	46	## Internal name for the AUI manager
	47	window_name = "plotpanel"
	48	## Title to appear on top of the window
	49	window_caption = "Plot Panel"
	50	## Flag to tell the GUI manager that this panel is not
	51	# tied to any perspective
	52	ALWAYS_ON = True
	53	## Group ID
	54	group_id = None
	55
[32c0841]	56	def __init__(self, parent, id=-1, color = None,
	57	dpi=None, style=wx.NO_FULL_REPAINT_ON_RESIZE, **kwargs):
	58	PlotPanel.__init__(self, parent, id=id, style=style, **kwargs)
[a45037aa]	59	PanelBase.__init__(self, parent)
[1bf33c1]	60	## Reference to the parent window
	61	self.parent = parent
	62	## Plottables
	63	self.plots = {}
[52b8b74]	64	#context menu
	65	self._slicerpop = None
[a07e72f]	66
[52b8b74]	67	self._available_data = []
	68	self._menu_add_ids = []
[a07e72f]	69	self._symbol_labels = self.get_symbol_label()
	70
	71	self.hide_menu = None
[1bf33c1]	72	## Unique ID (from gui_manager)
	73	self.uid = None
[a07e72f]	74
[6063b16]	75	## Default locations
[adf44c2]	76	self._default_save_location = os.getcwd()
	77	self.size = None
[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)
[adf44c2]	147	pos_x, pos_y = self.GetPositionTuple()
	148	if pos_x != 0 and pos_y != 0:
	149	self.size, _ = self.GetClientSizeTuple()
[6d727ae]	150
	151	def set_resizing(self, resizing=False):
	152	"""
	153	Set the resizing (True/False)
	154	"""
	155	self.resizing = resizing
	156	#self.canvas.set_resizing(resizing)
[1bf33c1]	157
[6d727ae]	158	def schedule_full_draw(self, func='append'):
	159	"""
	160	Put self in schedule to full redraw list
	161	"""
	162	# append/del this panel in the schedule list
	163	self.parent.set_schedule_full_draw(self, func)
	164
	165
[a07e72f]	166	def remove_data_by_id(self, id):
	167	"""'
	168	remove data from plot
	169	"""
	170	if id in self.plots.keys():
	171	data = self.plots[id]
	172	self.graph.delete(data)
	173	data_manager = self._manager.parent.get_data_manager()
[df22224]	174	data_list, theory_list = data_manager.get_by_id(id_list=[id])
	175
	176	if id in data_list.keys():
	177	data = data_list[id]
[248b918]	178	if id in theory_list.keys():
[df22224]	179	data = theory_list[id]
	180
[a07e72f]	181	del self.plots[id]
	182	self.graph.render(self)
	183	self.subplot.figure.canvas.draw_idle()
	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))
[0275276]	230	# unfocus all
	231	self.parent.set_plot_unfocus()
[4ed210f4]	232	#post nd event to notify guiframe that this panel is on focus
[a45037aa]	233	wx.PostEvent(self.parent, PanelOnFocusEvent(panel=self))
[df68da1]	234
[4ed210f4]	235
[a07e72f]	236	def _ontoggle_hide_error(self, event):
	237	"""
	238	Toggle error display to hide or show
	239	"""
	240
	241	selected_plot = self.plots[self.graph.selected_plottable]
	242	if self.hide_menu.GetText() == "Hide Error":
	243	selected_plot.hide_error = True
	244	else:
	245	selected_plot.hide_error = False
	246	## increment graph color
	247	self.graph.render(self)
	248	self.subplot.figure.canvas.draw_idle()
	249
[1bf33c1]	250	def _onRemove(self, event):
	251	"""
[d955bf19]	252	Remove a plottable from the graph and render the graph
	253
	254	:param event: Menu event
	255
[1bf33c1]	256	"""
[6c0568b]	257	## Check if there is a selected graph to remove
[a07e72f]	258	if self.graph.selected_plottable in self.plots.keys():
	259	selected_plot = self.plots[self.graph.selected_plottable]
	260	id = self.graph.selected_plottable
[df22224]	261	self.remove_data_by_id(id)
	262
[1bf33c1]	263	def onContextMenu(self, event):
	264	"""
[d955bf19]	265	1D plot context menu
	266
	267	:param event: wx context event
	268
[1bf33c1]	269	"""
[52b8b74]	270	self._slicerpop = PanelMenu()
	271	self._slicerpop.set_plots(self.plots)
	272	self._slicerpop.set_graph(self.graph)
[9a585d0]	273	# Various plot options
	274	id = wx.NewId()
[6d727ae]	275	self._slicerpop.Append(id, '&Save Image', 'Save image as PNG')
[9a585d0]	276	wx.EVT_MENU(self, id, self.onSaveImage)
	277	id = wx.NewId()
[6d727ae]	278	self._slicerpop.Append(id, '&Print Image', 'Print image ')
[18eba35]	279	wx.EVT_MENU(self, id, self.onPrint)
	280	id = wx.NewId()
[6d727ae]	281	self._slicerpop.Append(id, '&Print Preview', 'Print preview')
[18eba35]	282	wx.EVT_MENU(self, id, self.onPrinterPreview)
[52b8b74]	283
[6d727ae]	284	id = wx.NewId()
	285	self._slicerpop.Append(id, '&Copy to Clipboard', 'Copy to the clipboard')
	286	wx.EVT_MENU(self, id, self.OnCopyFigureMenu)
	287
	288	self._slicerpop.AppendSeparator()
	289
[52b8b74]	290	#add menu of other plugins
[a07e72f]	291	item_list = self.parent.get_context_menu(self)
[6d727ae]	292
[32c0841]	293	if (not item_list == None) and (not len(item_list) == 0):
[9a585d0]	294	for item in item_list:
	295	try:
	296	id = wx.NewId()
[52b8b74]	297	self._slicerpop.Append(id, item[0], item[1])
[9a585d0]	298	wx.EVT_MENU(self, id, item[2])
	299	except:
[32c0841]	300	msg = "ModelPanel1D.onContextMenu: "
[a07e72f]	301	msg += "bad menu item %s" % sys.exc_value
[32c0841]	302	wx.PostEvent(self.parent, StatusEvent(status=msg))
[9a585d0]	303	pass
[52b8b74]	304	self._slicerpop.AppendSeparator()
[6d727ae]	305	#id = wx.NewId()
	306	#self._slicerpop.Append(id, '&Print image', 'Print image')
[1bf33c1]	307	if self.graph.selected_plottable in self.plots:
	308	plot = self.plots[self.graph.selected_plottable]
[a436b2e]	309
[6d727ae]	310	id = wx.NewId()
	311	name = plot.name
	312	self._slicerpop.Append(id, "&Save Points as a File")
[a436b2e]	313	self._slicerpop.AppendSeparator()
	314	if plot.name != 'SLD':
	315	wx.EVT_MENU(self, id, self._onSave)
[e6a93df]	316	id = wx.NewId()
[a436b2e]	317	self._slicerpop.Append(id, '&Linear Fit')
	318	wx.EVT_MENU(self, id, self.onFitting)
	319	self._slicerpop.AppendSeparator()
	320
[a3c96f7a]	321	id = wx.NewId()
[a436b2e]	322	self._slicerpop.Append(id, "Remove %s Curve" % name)
	323	wx.EVT_MENU(self, id, self._onRemove)
	324	if not plot.is_data:
	325	id = wx.NewId()
	326	self._slicerpop.Append(id, '&Freeze', 'Freeze')
	327	wx.EVT_MENU(self, id, self.onFreeze)
	328	symbol_menu = wx.Menu()
	329	for label in self._symbol_labels:
	330	id = wx.NewId()
	331	symbol_menu.Append(id, str(label), str(label))
	332	wx.EVT_MENU(self, id, self.onChangeSymbol)
	333	id = wx.NewId()
	334	self._slicerpop.AppendMenu(id,'&Modify Symbol', symbol_menu)
	335	self._slicerpop.AppendSeparator()
	336
	337	id = wx.NewId()
	338	self.hide_menu = self._slicerpop.Append(id, "Hide Error")
	339
	340	if plot.dy is not None and plot.dy != []:
	341	if plot.hide_error :
	342	self.hide_menu.SetText('Show Error')
	343	else:
	344	self.hide_menu.SetText('Hide Error')
[a07e72f]	345	else:
[a436b2e]	346	self.hide_menu.Enable(False)
	347	wx.EVT_MENU(self, id, self._ontoggle_hide_error)
	348
	349	self._slicerpop.AppendSeparator()
	350	# Option to hide
	351	#TODO: implement functionality to hide a plottable (legend click)
[df22224]	352
[a07e72f]	353
[1bf33c1]	354	id = wx.NewId()
[52b8b74]	355	self._slicerpop.Append(id, '&Change scale')
[1bf33c1]	356	wx.EVT_MENU(self, id, self._onProperties)
	357	id = wx.NewId()
[52b8b74]	358	self._slicerpop.Append(id, '&Reset Graph')
[d468daa]	359	wx.EVT_MENU(self, id, self.onResetGraph)
[1bf33c1]	360	pos = event.GetPosition()
	361	pos = self.ScreenToClient(pos)
[52b8b74]	362	self.PopupMenu(self._slicerpop, pos)
[a07e72f]	363
[e6a93df]	364	def onFreeze(self, event):
	365	"""
	366	"""
	367	plot = self.plots[self.graph.selected_plottable]
	368	self.parent.onfreeze([plot.id])
	369
[a07e72f]	370	def onChangeSymbol(self, event):
[6c0568b]	371	"""
	372	"""
[a07e72f]	373	menu = event.GetEventObject()
	374	id = event.GetId()
	375	label = menu.GetLabel(id)
	376	selected_plot = self.plots[self.graph.selected_plottable]
	377	selected_plot.symbol = self._symbol_labels[label]
	378	## Set the view scale for all plots
	379	self._onEVT_FUNC_PROPERTY()
	380	## render the graph
[6d727ae]	381	#self.graph.render(self)
	382	#self.subplot.figure.canvas.draw_idle()
[c81140c]	383
[42d27f2]	384	def _onsaveTXT(self, path):
	385	"""
[d955bf19]	386	Save file as txt
[1abcb04]	387
[d955bf19]	388	:TODO: Refactor and remove this method. See TODO in _onSave.
	389
[42d27f2]	390	"""
	391	data = self.plots[self.graph.selected_plottable]
	392
	393	if not path == None:
	394	out = open(path, 'w')
	395	has_errors = True
[32c0841]	396	if data.dy == None or data.dy == []:
[42d27f2]	397	has_errors = False
	398	# Sanity check
	399	if has_errors:
	400	try:
	401	if len(data.y) != len(data.dy):
	402	has_errors = False
	403	except:
	404	has_errors = False
	405	if has_errors:
[6d727ae]	406	if data.dx != None:
	407	out.write("<X> <Y> <dY> <dX>\n")
	408	else:
	409	out.write("<X> <Y> <dY>\n")
[42d27f2]	410	else:
	411	out.write("<X> <Y>\n")
	412
	413	for i in range(len(data.x)):
	414	if has_errors:
[6d727ae]	415	if data.dx != None:
	416	out.write("%g %g %g %g\n" % (data.x[i],
	417	data.y[i],
	418	data.dy[i],
	419	data.dx[i]))
	420	else:
	421	out.write("%g %g %g\n" % (data.x[i],
	422	data.y[i],
	423	data.dy[i]))
[42d27f2]	424	else:
	425	out.write("%g %g\n" % (data.x[i],
	426	data.y[i]))
	427	out.close()
[6063b16]	428	try:
	429	self._default_save_location = os.path.dirname(path)
	430	except:
	431	pass
[8bd764d]	432
[1bf33c1]	433	def _onSave(self, evt):
	434	"""
[d955bf19]	435	Save a data set to a text file
	436
	437	:param evt: Menu event
	438
[1bf33c1]	439	"""
[a07e72f]	440
	441	path = None
	442	wildcard = "Text files (.txt)\|.txt\|"\
	443	"CanSAS 1D files(.xml)\|.xml"
	444	dlg = wx.FileDialog(self, "Choose a file",
	445	self._default_save_location,
	446	"", wildcard , wx.SAVE)
	447
	448	if dlg.ShowModal() == wx.ID_OK:
	449	path = dlg.GetPath()
[ad1e49c]	450	# ext_num = 0 for .txt, ext_num = 1 for .xml
	451	# This is MAC Fix
	452	ext_num = dlg.GetFilterIndex()
	453	if ext_num == 0:
	454	format = '.txt'
	455	else:
	456	format = '.xml'
	457	path = os.path.splitext(path)[0] + format
[a07e72f]	458	mypath = os.path.basename(path)
[1bf33c1]	459
[a07e72f]	460	#TODO: This is bad design. The DataLoader is designed
	461	#to recognize extensions.
	462	# It should be a simple matter of calling the .
	463	#save(file, data, '.xml') method
	464	# of the DataLoader.loader.Loader class.
	465	from DataLoader.loader import Loader
	466	#Instantiate a loader
	467	loader = Loader()
	468	data = self.plots[self.graph.selected_plottable]
	469	format = ".txt"
	470	if os.path.splitext(mypath)[1].lower() == format:
[959981b]	471	# Make sure the ext included in the file name
	472	# especially on MAC
	473	fName = os.path.splitext(path)[0] + format
	474	self._onsaveTXT(fName)
[a07e72f]	475	format = ".xml"
	476	if os.path.splitext(mypath)[1].lower() == format:
[959981b]	477	# Make sure the ext included in the file name
	478	# especially on MAC
	479	fName = os.path.splitext(path)[0] + format
	480	loader.save(fName, data, format)
[a07e72f]	481	try:
	482	self._default_save_location = os.path.dirname(path)
	483	except:
	484	pass
	485	dlg.Destroy()

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

SasView

source: sasview/guiframe/local_perspectives/plotting/Plotter1D.py @ 8960479

Download in other formats: