Plotter2D.py @ b796c72

ESS_GUIESS_GUI_DocsESS_GUI_batch_fittingESS_GUI_bumps_abstractionESS_GUI_iss1116ESS_GUI_iss879ESS_GUI_iss959ESS_GUI_openclESS_GUI_orderingESS_GUI_sync_sascalcmagnetic_scattrelease-4.2.2ticket-1009ticket-1094-headlessticket-1242-2d-resolutionticket-1243ticket-1249ticket885unittest-saveload

Last change on this file since b796c72 was 3e5648b, checked in by krzywon, 8 years ago
Resolve merge conflicts and 2 small fixes to slicer panel. #467
Property mode set to `100644`
File size: 28.7 KB

Rev	Line
[959eb01]	1
	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	################################################################################
	11
	12
	13	import wx
	14	import sys
	15	import math
	16	import numpy as np
	17	import logging
	18	from sas.sasgui.plottools.PlotPanel import PlotPanel
	19	from sas.sasgui.plottools.plottables import Graph
	20	from sas.sasgui.plottools.TextDialog import TextDialog
	21	from sas.sasgui.guiframe.events import StatusEvent
	22	from sas.sasgui.guiframe.events import NewPlotEvent
	23	from sas.sasgui.guiframe.events import PanelOnFocusEvent
	24	from sas.sasgui.guiframe.events import SlicerEvent
	25	from sas.sasgui.guiframe.utils import PanelMenu
	26	from sas.sasgui.guiframe.local_perspectives.plotting.binder import BindArtist
	27	from Plotter1D import ModelPanel1D
	28	from sas.sasgui.plottools.toolbar import NavigationToolBar
	29	from matplotlib.font_manager import FontProperties
	30	from graphAppearance import graphAppearance
	31	(InternalEvent, EVT_INTERNAL) = wx.lib.newevent.NewEvent()
	32
	33	logger = logging.getLogger(__name__)
	34
	35	DEFAULT_QMAX = 0.05
	36	DEFAULT_QSTEP = 0.001
	37	DEFAULT_BEAM = 0.005
	38	BIN_WIDTH = 1.0
	39
	40
	41	def find_key(dic, val):
	42	"""return the key of dictionary dic given the value"""
	43	return [k for k, v in dic.iteritems() if v == val][0]
	44
	45
	46	class NavigationToolBar2D(NavigationToolBar):
	47	"""
	48	"""
	49	def __init__(self, canvas, parent=None):
	50	NavigationToolBar.__init__(self, canvas=canvas, parent=parent)
	51
	52	def delete_option(self):
	53	"""
	54	remove default toolbar item
	55	"""
	56	#delete reset button
	57	self.DeleteToolByPos(0)
	58	#delete dragging
	59	self.DeleteToolByPos(2)
	60	#delete unwanted button that configures subplot parameters
	61	self.DeleteToolByPos(4)
	62
	63	def add_option(self):
	64	"""
	65	add item to the toolbar
	66	"""
	67	#add button
	68	id_context = wx.NewId()
	69	context_tip = 'Graph Menu'
	70	context = wx.ArtProvider.GetBitmap(wx.ART_LIST_VIEW, wx.ART_TOOLBAR)
	71	self.InsertSimpleTool(0, id_context, context, context_tip, context_tip)
	72	wx.EVT_TOOL(self, id_context, self.parent.onToolContextMenu)
	73	self.InsertSeparator(1)
	74	#add print button
	75	id_print = wx.NewId()
	76	print_bmp = wx.ArtProvider.GetBitmap(wx.ART_PRINT, wx.ART_TOOLBAR)
	77	self.AddSimpleTool(id_print, print_bmp, 'Print', 'Activate printing')
	78	wx.EVT_TOOL(self, id_print, self.on_print)
	79
	80
	81	class ModelPanel2D(ModelPanel1D):
	82	"""
	83	Plot panel for use with the GUI manager
	84	"""
	85
	86	## Internal name for the AUI manager
	87	window_name = "plotpanel"
	88	## Title to appear on top of the window
	89	window_caption = "Plot Panel"
	90	## Flag to tell the GUI manager that this panel is not
	91	# tied to any perspective
	92	ALWAYS_ON = True
	93	## Group ID
	94	group_id = None
	95
	96
	97	def __init__(self, parent, id=-1, data2d=None, color=None,
	98	dpi=None, style=wx.NO_FULL_REPAINT_ON_RESIZE, **kwargs):
	99	"""
	100	Initialize the panel
	101	"""
	102	ModelPanel1D.__init__(self, parent, id=id, style=style, **kwargs)
	103	self.parent = parent
	104	## Reference to the parent window
	105	if hasattr(parent, "parent"):
	106	self.parent = self.parent.parent
	107	## Dictionary containing Plottables
	108	self.plots = {}
	109	## Save reference of the current plotted
	110	self.data2D = data2d
	111	## Unique ID (from gui_manager)
	112	self.uid = None
	113	## Action IDs for internal call-backs
	114	self.action_ids = {}
	115	## Create Artist and bind it
	116	self.connect = BindArtist(self.subplot.figure)
	117	## Beam stop
	118	self.beamstop_radius = DEFAULT_BEAM
	119	## to set the order of lines drawn first.
	120	self.slicer_z = 5
	121	## Reference to the current slicer
	122	self.slicer = None
	123	## event to send slicer info
	124	self.Bind(EVT_INTERNAL, self._onEVT_INTERNAL)
	125
	126	self.axes_frozen = False
	127	## panel that contains result from slicer motion (ex: Boxsum info)
	128	self.panel_slicer = None
	129	self.title_label = None
	130	self.title_font = None
	131	self.title_color = 'black'
	132	## Graph
	133	self.graph = Graph()
	134	self.graph.xaxis("\\rm{Q}", 'A^{-1}')
	135	self.graph.yaxis("\\rm{Intensity} ", "cm^{-1}")
	136	self.graph.render(self)
	137	## store default value of zmin and zmax
	138	self.default_zmin_ctl = self.zmin_2D
	139	self.default_zmax_ctl = self.zmax_2D
	140
	141	def on_plot_qrange(self, event=None):
	142	"""
	143	On Qmin Qmax vertical line event
	144	"""
	145	# Not implemented
[235f514]	146	if event is None:
[959eb01]	147	return
	148	event.Skip()
	149
	150	def onLeftDown(self, event):
	151	"""
	152	left button down and ready to drag
	153
	154	"""
	155	# Check that the LEFT button was pressed
	156	PlotPanel.onLeftDown(self, event)
	157	ax = event.inaxes
[7432acb]	158	if ax is not None:
[959eb01]	159	# data coordinate position
	160	pos_x = "%8.3g" % event.xdata
	161	pos_y = "%8.3g" % event.ydata
	162	position = "x: %s y: %s" % (pos_x, pos_y)
	163	wx.PostEvent(self.parent, StatusEvent(status=position))
	164	self.plottable_selected(self.data2D.id)
	165	self._manager.set_panel_on_focus(self)
	166	wx.PostEvent(self.parent, PanelOnFocusEvent(panel=self))
	167
	168	def add_toolbar(self):
	169	"""
	170	add toolbar
	171	"""
	172	self.enable_toolbar = True
	173	self.toolbar = NavigationToolBar2D(parent=self, canvas=self.canvas)
	174	self.toolbar.Realize()
	175	# On Windows platform, default window size is incorrect, so set
	176	# toolbar width to figure width.
	177	_, th = self.toolbar.GetSizeTuple()
	178	fw, _ = self.canvas.GetSizeTuple()
	179	self.toolbar.SetSize(wx.Size(fw, th))
	180	self.sizer.Add(self.toolbar, 0, wx.LEFT \| wx.EXPAND)
	181	# update the axes menu on the toolbar
	182	self.toolbar.update()
	183
	184	def plot_data(self, data):
	185	"""
	186	Data is ready to be displayed
	187
	188	:TODO: this name should be changed to something more appropriate
	189	Don't forget that changing this name will mean changing code
	190	in plotting.py
	191
	192	:param event: data event
	193	"""
	194	xlo = None
	195	xhi = None
	196	ylo = None
	197	yhi = None
	198	if data.__class__.__name__ == 'Data1D':
	199	return
	200	## Update self.data2d with the current plot
	201	self.data2D = data
	202	if data.id in self.plots.keys():
	203	#replace
	204	xlo, xhi = self.subplot.get_xlim()
	205	ylo, yhi = self.subplot.get_ylim()
	206	self.graph.replace(data)
	207	self.plots[data.id] = data
	208	else:
	209	self.plots[data.id] = data
	210	self.graph.add(self.plots[data.id])
	211	# update qmax with the new xmax of data plotted
	212	self.qmax = data.xmax
	213	self.slicer = None
	214	# Check axis labels
	215	#TODO: Should re-factor this
	216	## render the graph with its new content
	217	#data2D: put 'Pixel (Number)' for axis title and unit in case of having no detector info and none in _units
	218	if len(data.detector) < 1:
	219	if len(data._xunit) < 1 and len(data._yunit) < 1:
	220	data._xaxis = '\\rm{x}'
	221	data._yaxis = '\\rm{y}'
	222	data._xunit = 'pixel'
	223	data._yunit = 'pixel'
	224	# graph properties
	225	self.graph.xaxis(data._xaxis, data._xunit)
	226	self.graph.yaxis(data._yaxis, data._yunit)
	227	if self._is_changed_legend_label:
	228	data.label = self.title_label
[235f514]	229	if data.label is None:
[959eb01]	230	data.label = data.name
	231	if not self.title_font:
	232	self.graph.title(data.label)
	233	self.graph.render(self)
	234	# Set the axis labels on subplot
	235	self._set_axis_labels()
	236	self.draw_plot()
	237	else:
	238	self.graph.render(self)
	239	self.draw_plot()
	240	self.subplot.set_title(label=data.label,
	241	fontproperties=self.title_font,
	242	color=self.title_color)
	243	self.subplot.figure.canvas.draw_idle()
	244	## store default value of zmin and zmax
	245	self.default_zmin_ctl = self.zmin_2D
	246	self.default_zmax_ctl = self.zmax_2D
	247	if not self.is_zoomed:
	248	return
	249	# Recover the x,y limits
	250	if xlo and xhi and ylo and yhi:
	251	if xlo > data.xmin and xhi < data.xmax and \
	252	ylo > data.ymin and yhi < data.ymax:
	253	self.subplot.set_xlim((xlo, xhi))
	254	self.subplot.set_ylim((ylo, yhi))
	255	else:
	256	self.toolbar.update()
	257	self.is_zoomed = False
	258
	259	def _set_axis_labels(self):
	260	"""
	261	Set axis labels
	262	"""
	263	data = self.data2D
	264	# control axis labels from the panel itself
	265	yname, yunits = data.get_yaxis()
[7432acb]	266	if self.yaxis_label is not None:
[959eb01]	267	yname = self.yaxis_label
	268	yunits = self.yaxis_unit
	269	else:
	270	self.yaxis_label = yname
	271	self.yaxis_unit = yunits
	272	xname, xunits = data.get_xaxis()
[7432acb]	273	if self.xaxis_label is not None:
[959eb01]	274	xname = self.xaxis_label
	275	xunits = self.xaxis_unit
	276	else:
	277	self.xaxis_label = xname
	278	self.xaxis_unit = xunits
	279	self.xaxis(xname, xunits, self.xaxis_font,
	280	self.xaxis_color, self.xaxis_tick)
	281	self.yaxis(yname, yunits, self.yaxis_font,
	282	self.yaxis_color, self.yaxis_tick)
	283
	284	def onContextMenu(self, event):
	285	"""
	286	2D plot context menu
	287
	288	:param event: wx context event
	289
	290	"""
	291	ids = iter(self._menu_ids)
	292	slicerpop = PanelMenu()
	293	slicerpop.set_plots(self.plots)
	294	slicerpop.set_graph(self.graph)
	295
	296	wx_id = ids.next()
	297	slicerpop.Append(wx_id, '&Save Image', 'Save image as png')
	298	wx.EVT_MENU(self, wx_id, self.onSaveImage)
	299
	300	wx_id = ids.next()
	301	slicerpop.Append(wx_id, '&Print Image', 'Print image')
	302	wx.EVT_MENU(self, wx_id, self.onPrint)
	303
	304	wx_id = ids.next()
	305	slicerpop.Append(wx_id, '&Copy to Clipboard', 'Copy to the clipboard')
	306	wx.EVT_MENU(self, wx_id, self.OnCopyFigureMenu)
	307	slicerpop.AppendSeparator()
	308	# saving data
	309	plot = self.data2D
	310	wx_id = ids.next()
	311	slicerpop.Append(wx_id, "&Data Info")
	312	wx.EVT_MENU(self, wx_id, self._onDataShow)
	313
	314	wx_id = ids.next()
	315	slicerpop.Append(wx_id, "&Save as a File (DAT)")
	316	self.action_ids[str(wx_id)] = plot
	317	wx.EVT_MENU(self, wx_id, self._onSave)
	318
	319	slicerpop.AppendSeparator()
	320	if len(self.data2D.detector) <= 1:
	321	item_list = self.parent.get_current_context_menu(self)
[ac07a3a]	322	if ((item_list is not None) and len(item_list) and
	323	self.data2D.name.split(" ")[0] != 'Residuals'):
[959eb01]	324	for item, wx_id in zip(item_list, [ids.next() for i in range(len(item_list))]):
	325	try:
	326	slicerpop.Append(wx_id, item[0], item[1])
	327	wx.EVT_MENU(self, wx_id, item[2])
	328	except:
	329	msg = "ModelPanel1D.onContextMenu: "
	330	msg += "bad menu item %s" % sys.exc_value
	331	wx.PostEvent(self.parent, StatusEvent(status=msg))
	332	slicerpop.AppendSeparator()
	333
	334	wx_id = ids.next()
	335	slicerpop.Append(wx_id, '&Perform Circular Average')
	336	wx.EVT_MENU(self, wx_id, self.onCircular) \
	337	# For Masked Data
	338	if not plot.mask.all():
	339	wx_id = ids.next()
	340	slicerpop.Append(wx_id, '&Masked Circular Average')
	341	wx.EVT_MENU(self, wx_id, self.onMaskedCircular)
	342	wx_id = ids.next()
	343	slicerpop.Append(wx_id, '&Sector [Q View]')
	344	wx.EVT_MENU(self, wx_id, self.onSectorQ)
	345	wx_id = ids.next()
	346	slicerpop.Append(wx_id, '&Annulus [Phi View ]')
	347	wx.EVT_MENU(self, wx_id, self.onSectorPhi)
	348	wx_id = ids.next()
	349	slicerpop.Append(wx_id, '&Box Sum')
	350	wx.EVT_MENU(self, wx_id, self.onBoxSum)
	351	wx_id = ids.next()
	352	slicerpop.Append(wx_id, '&Box Averaging in Qx')
	353	wx.EVT_MENU(self, wx_id, self.onBoxavgX)
	354	wx_id = ids.next()
	355	slicerpop.Append(wx_id, '&Box Averaging in Qy')
	356	wx.EVT_MENU(self, wx_id, self.onBoxavgY)
[7432acb]	357	if self.slicer is not None:
[959eb01]	358	wx_id = ids.next()
	359	slicerpop.Append(wx_id, '&Clear Slicer')
	360	wx.EVT_MENU(self, wx_id, self.onClearSlicer)
	361	if self.slicer.__class__.__name__ != "BoxSum":
	362	wx_id = ids.next()
[3e5648b]	363	name = '&Edit Slicer Parameters and Batch Slicing'
	364	slicerpop.Append(wx_id, name)
[959eb01]	365	wx.EVT_MENU(self, wx_id, self._onEditSlicer)
	366	slicerpop.AppendSeparator()
	367
	368	wx_id = ids.next()
	369	slicerpop.Append(wx_id, '&Edit Graph Label', 'Edit Graph Label')
	370	wx.EVT_MENU(self, wx_id, self.onEditLabels)
	371	slicerpop.AppendSeparator()
	372
	373	# ILL mod here
	374
	375	wx_id = ids.next()
	376	slicerpop.Append(wx_id, '&Modify graph appearance', 'Modify graph appearance')
	377	wx.EVT_MENU(self, wx_id, self.modifyGraphAppearance)
	378	slicerpop.AppendSeparator()
	379
	380	wx_id = ids.next()
	381	slicerpop.Append(wx_id, '&2D Color Map')
	382	wx.EVT_MENU(self, wx_id, self._onEditDetector)
	383	slicerpop.AppendSeparator()
	384
	385	wx_id = ids.next()
	386	slicerpop.Append(wx_id, '&Toggle Linear/Log Scale')
	387	wx.EVT_MENU(self, wx_id, self._onToggleScale)
	388
	389	slicerpop.AppendSeparator()
	390	wx_id = ids.next()
	391	slicerpop.Append(wx_id, '&Window Title')
	392	wx.EVT_MENU(self, wx_id, self.onChangeCaption)
	393
	394	try:
	395	pos_evt = event.GetPosition()
	396	pos = self.ScreenToClient(pos_evt)
	397	except:
	398	pos_x, pos_y = self.toolbar.GetPositionTuple()
	399	pos = (pos_x, pos_y + 5)
	400	self.PopupMenu(slicerpop, pos)
	401
	402	def onEditLabels(self, event):
	403	"""
	404	Edit legend label
	405	"""
	406	try:
	407	selected_plot = self.plots[self.graph.selected_plottable]
	408	except:
	409	selected_plot = self.plots[self.data2D.id]
	410	label = selected_plot.label
	411	dial = TextDialog(None, -1, 'Change Label', label)
	412	if dial.ShowModal() == wx.ID_OK:
	413	try:
	414	FONT = FontProperties()
	415	newlabel = dial.getText()
	416	font = FONT.copy()
	417	font.set_size(dial.getSize())
	418	font.set_family(dial.getFamily())
	419	font.set_style(dial.getStyle())
	420	font.set_weight(dial.getWeight())
	421	colour = dial.getColor()
	422	if len(newlabel) > 0:
	423	# update Label
	424	selected_plot.label = newlabel
	425	self.graph.title(newlabel)
	426	self.title_label = selected_plot.label
	427	self.title_font = font
	428	self.title_color = colour
	429	## render the graph
	430	self.subplot.set_title(label=self.title_label,
	431	fontproperties=self.title_font,
	432	color=self.title_color)
	433	self._is_changed_legend_label = True
	434	self.subplot.figure.canvas.draw_idle()
	435	except:
	436	msg = "Add Text: Error. Check your property values..."
	437	logger.error(msg)
[7432acb]	438	if self.parent is not None:
[959eb01]	439	wx.PostEvent(self.parent, StatusEvent(status=msg))
	440	dial.Destroy()
	441
	442	def _onEditDetector(self, event):
	443	"""
	444	Allow to view and edits detector parameters
	445
	446	:param event: wx.menu event
	447
	448	"""
	449	import detector_dialog
	450	dialog = detector_dialog.DetectorDialog(self, -1, base=self.parent,
	451	reset_zmin_ctl=self.default_zmin_ctl,
	452	reset_zmax_ctl=self.default_zmax_ctl, cmap=self.cmap)
	453	## info of current detector and data2D
	454	xnpts = len(self.data2D.x_bins)
	455	ynpts = len(self.data2D.y_bins)
	456	xmax = max(self.data2D.xmin, self.data2D.xmax)
	457	ymax = max(self.data2D.ymin, self.data2D.ymax)
	458	qmax = math.sqrt(math.pow(xmax, 2) + math.pow(ymax, 2))
	459	## set dialog window content
	460	dialog.setContent(xnpts=xnpts, ynpts=ynpts, qmax=qmax,
	461	beam=self.data2D.xmin,
	462	zmin=self.zmin_2D,
	463	zmax=self.zmax_2D)
	464	if dialog.ShowModal() == wx.ID_OK:
	465	evt = dialog.getContent()
	466	self.zmin_2D = evt.zmin
	467	self.zmax_2D = evt.zmax
	468	self.cmap = evt.cmap
	469	dialog.Destroy()
	470	## Redraw the current image
	471	self.image(data=self.data2D.data,
	472	qx_data=self.data2D.qx_data,
	473	qy_data=self.data2D.qy_data,
	474	xmin=self.data2D.xmin,
	475	xmax=self.data2D.xmax,
	476	ymin=self.data2D.ymin,
	477	ymax=self.data2D.ymax,
	478	zmin=self.zmin_2D,
	479	zmax=self.zmax_2D,
	480	cmap=self.cmap,
	481	color=0, symbol=0, label=self.data2D.name)
	482	self.subplot.figure.canvas.draw_idle()
	483
	484	def freeze_axes(self):
	485	"""
	486	"""
	487	self.axes_frozen = True
	488
	489	def thaw_axes(self):
	490	"""
	491	"""
	492	self.axes_frozen = False
	493
	494	def onMouseMotion(self, event):
	495	"""
	496	"""
	497	pass
	498
	499	def onWheel(self, event):
	500	"""
	501	"""
	502	pass
	503
	504	def update(self, draw=True):
	505	"""
	506	Respond to changes in the model by recalculating the
	507	profiles and resetting the widgets.
	508	"""
	509	self.draw_plot()
	510
	511	def _getEmptySlicerEvent(self):
	512	"""
	513	create an empty slicervent
	514	"""
	515	return SlicerEvent(type=None, params=None, obj_class=None)
	516
	517	def _onEVT_INTERNAL(self, event):
	518	"""
	519	Draw the slicer
	520
	521	:param event: wx.lib.newevent (SlicerEvent) containing slicer
	522	parameter
	523
	524	"""
	525	self._setSlicer(event.slicer)
	526
	527	def _setSlicer(self, slicer):
	528	"""
	529	Clear the previous slicer and create a new one.Post an internal
	530	event.
	531
	532	:param slicer: slicer class to create
	533
	534	"""
[3e5648b]	535	# Clear current slicer
[ac07a3a]	536	if self.slicer is not None:
[959eb01]	537	self.slicer.clear()
[3e5648b]	538	# Create a new slicer
[959eb01]	539	self.slicer_z += 1
	540	self.slicer = slicer(self, self.subplot, zorder=self.slicer_z)
	541	self.subplot.set_ylim(self.data2D.ymin, self.data2D.ymax)
	542	self.subplot.set_xlim(self.data2D.xmin, self.data2D.xmax)
[3e5648b]	543	# Draw slicer
[959eb01]	544	self.update()
	545	self.slicer.update()
	546	msg = "Plotter2D._setSlicer %s" % self.slicer.__class__.__name__
	547	wx.PostEvent(self.parent, StatusEvent(status=msg))
	548	# Post slicer event
	549	event = self._getEmptySlicerEvent()
	550	event.type = self.slicer.__class__.__name__
	551	event.obj_class = self.slicer.__class__
	552	event.params = self.slicer.get_params()
	553	wx.PostEvent(self, event)
	554
	555	def onMaskedCircular(self, event):
	556	"""
	557	perform circular averaging on Data2D with mask if it exists
	558
	559	:param event: wx.menu event
	560
	561	"""
	562	self.onCircular(event, True)
	563
	564	def onCircular(self, event, ismask=False):
	565	"""
	566	perform circular averaging on Data2D
	567
	568	:param event: wx.menu event
	569
	570	"""
	571	# Find the best number of bins
	572	npt = math.sqrt(len(self.data2D.data[np.isfinite(self.data2D.data)]))
	573	npt = math.floor(npt)
	574	from sas.sascalc.dataloader.manipulations import CircularAverage
[3e5648b]	575	# compute the maximum radius of data2D
[959eb01]	576	self.qmax = max(math.fabs(self.data2D.xmax),
	577	math.fabs(self.data2D.xmin))
	578	self.ymax = max(math.fabs(self.data2D.ymax),
	579	math.fabs(self.data2D.ymin))
	580	self.radius = math.sqrt(math.pow(self.qmax, 2) + math.pow(self.ymax, 2))
[3e5648b]	581	# Compute beam width
[959eb01]	582	bin_width = (self.qmax + self.qmax) / npt
[3e5648b]	583	# Create data1D circular average of data2D
[959eb01]	584	Circle = CircularAverage(r_min=0, r_max=self.radius,
	585	bin_width=bin_width)
	586	circ = Circle(self.data2D, ismask=ismask)
	587	from sas.sasgui.guiframe.dataFitting import Data1D
	588	if hasattr(circ, "dxl"):
	589	dxl = circ.dxl
	590	else:
	591	dxl = None
	592	if hasattr(circ, "dxw"):
	593	dxw = circ.dxw
	594	else:
	595	dxw = None
	596
	597	new_plot = Data1D(x=circ.x, y=circ.y, dy=circ.dy, dx=circ.dx)
	598	new_plot.dxl = dxl
	599	new_plot.dxw = dxw
	600	new_plot.name = "Circ avg " + self.data2D.name
	601	new_plot.source = self.data2D.source
[3e5648b]	602	# new_plot.info = self.data2D.info
[959eb01]	603	new_plot.interactive = True
	604	new_plot.detector = self.data2D.detector
	605
[3e5648b]	606	# If the data file does not tell us what the axes are, just assume...
[959eb01]	607	new_plot.xaxis("\\rm{Q}", "A^{-1}")
	608	if hasattr(self.data2D, "scale") and \
	609	self.data2D.scale == 'linear':
	610	new_plot.ytransform = 'y'
	611	new_plot.yaxis("\\rm{Residuals} ", "normalized")
	612	else:
	613	new_plot.yaxis("\\rm{Intensity} ", "cm^{-1}")
	614
	615	new_plot.group_id = "2daverage" + self.data2D.name
	616	new_plot.id = "Circ avg " + self.data2D.name
	617	new_plot.is_data = True
[3e5648b]	618	self.parent.update_theory(data_id=self.data2D.id, theory=new_plot)
[959eb01]	619	wx.PostEvent(self.parent,
	620	NewPlotEvent(plot=new_plot, title=new_plot.name))
	621
	622	def _onEditSlicer(self, event):
	623	"""
	624	Is available only when a slicer is drawn.Create a dialog
	625	window where the user can enter value to reset slicer
	626	parameters.
	627
	628	:param event: wx.menu event
	629
	630	"""
[7432acb]	631	if self.slicer is not None:
[3e5648b]	632	from parameters_panel_slicer import SlicerParameterPanel
[959eb01]	633	dialog = SlicerParameterPanel(self, -1, "Slicer Parameters")
	634	dialog.set_slicer(self.slicer.__class__.__name__,
	635	self.slicer.get_params())
	636	if dialog.ShowModal() == wx.ID_OK:
	637	dialog.Destroy()
	638
	639	def onSectorQ(self, event):
	640	"""
	641	Perform sector averaging on Q and draw sector slicer
	642	"""
	643	from SectorSlicer import SectorInteractor
	644	self.onClearSlicer(event)
	645	wx.PostEvent(self, InternalEvent(slicer=SectorInteractor))
	646
	647	def onSectorPhi(self, event):
	648	"""
	649	Perform sector averaging on Phi and draw annulus slicer
	650	"""
	651	from AnnulusSlicer import AnnulusInteractor
	652	self.onClearSlicer(event)
	653	wx.PostEvent(self, InternalEvent(slicer=AnnulusInteractor))
	654
	655	def onBoxSum(self, event):
	656	"""
	657	"""
	658	from sas.sasgui.guiframe.gui_manager import MDIFrame
	659	from boxSum import BoxSum
	660	self.onClearSlicer(event)
	661	self.slicer_z += 1
	662	self.slicer = BoxSum(self, self.subplot, zorder=self.slicer_z)
	663	self.subplot.set_ylim(self.data2D.ymin, self.data2D.ymax)
	664	self.subplot.set_xlim(self.data2D.xmin, self.data2D.xmax)
	665	self.update()
	666	self.slicer.update()
	667	## Value used to initially set the slicer panel
	668	params = self.slicer.get_params()
	669	## Create a new panel to display results of summation of Data2D
[3e5648b]	670	from parameters_panel_boxsum import SlicerPanel
[959eb01]	671	win = MDIFrame(self.parent, None, 'None', (100, 200))
	672	new_panel = SlicerPanel(parent=win, id=-1,
	673	base=self, type=self.slicer.__class__.__name__,
	674	params=params, style=wx.RAISED_BORDER)
	675
	676	new_panel.window_caption = self.slicer.__class__.__name__ + " " + \
	677	str(self.data2D.name)
	678	new_panel.window_name = self.slicer.__class__.__name__ + " " + \
	679	str(self.data2D.name)
	680	## Store a reference of the new created panel
	681
	682	## save the window_caption of the new panel in the current slicer
	683	self.slicer.set_panel_name(name=new_panel.window_caption)
	684	## post slicer panel to guiframe to display it
	685	from sas.sasgui.guiframe.events import SlicerPanelEvent
	686
	687	win.set_panel(new_panel)
	688	new_panel.frame = win
	689	wx.PostEvent(self.parent, SlicerPanelEvent(panel=new_panel,
	690	main_panel=self))
	691	wx.CallAfter(new_panel.frame.Show)
	692	self.panel_slicer = new_panel
	693
	694	def onBoxavgX(self, event):
	695	"""
	696	Perform 2D data averaging on Qx
	697	Create a new slicer .
	698
	699	:param event: wx.menu event
	700	"""
	701	from boxSlicer import BoxInteractorX
	702	self.onClearSlicer(event)
	703	wx.PostEvent(self, InternalEvent(slicer=BoxInteractorX))
	704
	705	def onBoxavgY(self, event):
	706	"""
	707	Perform 2D data averaging on Qy
	708	Create a new slicer .
	709
	710	:param event: wx.menu event
	711
	712	"""
	713	from boxSlicer import BoxInteractorY
	714	self.onClearSlicer(event)
	715	wx.PostEvent(self, InternalEvent(slicer=BoxInteractorY))
	716
	717	def onClearSlicer(self, event):
	718	"""
	719	Clear the slicer on the plot
	720	"""
[ac07a3a]	721	if self.slicer is not None:
[959eb01]	722	self.slicer.clear()
	723	self.subplot.figure.canvas.draw()
	724	self.slicer = None
	725	# Post slicer None event
	726	event = self._getEmptySlicerEvent()
	727	wx.PostEvent(self, event)
	728
	729	def _onSave(self, evt):
	730	"""
	731	Save a data set to a dat(text) file
	732
	733	:param evt: Menu event
	734
	735	"""
	736	event_id = str(evt.GetId())
[7432acb]	737	if self.parent is not None:
[959eb01]	738	self._default_save_location = self.parent._default_save_location
	739	default_name = self.plots[self.graph.selected_plottable].label
	740	if default_name.count('.') > 0:
	741	default_name = default_name.split('.')[0]
	742	default_name += "_out"
	743	if event_id in self.action_ids:
	744	self.parent.save_data2d(self.data2D, default_name)
	745
	746	def _onDataShow(self, evt):
	747	"""
	748	Show the data set in text
	749
	750	:param evt: Menu event
	751
	752	"""
	753	menu = evt.GetEventObject()
	754	event_id = evt.GetId()
	755	self.set_selected_from_menu(menu, event_id)
	756	data = self.plots[self.graph.selected_plottable]
	757	default_name = data.label
	758	if default_name.count('.') > 0:
	759	default_name = default_name.split('.')[0]
[7432acb]	760	if self.parent is not None:
[959eb01]	761	self.parent.show_data2d(data, default_name)
	762
	763	def modifyGraphAppearance(self, e):
[3e5648b]	764	self.graphApp = graphAppearance(self, 'Modify graph appearance',
	765	legend=False)
[959eb01]	766	self.graphApp.setDefaults(self.grid_on, self.legend_on,
	767	self.xaxis_label, self.yaxis_label,
	768	self.xaxis_unit, self.yaxis_unit,
	769	self.xaxis_font, self.yaxis_font,
	770	find_key(self.get_loc_label(), self.legendLoc),
	771	self.xcolor, self.ycolor,
	772	self.is_xtick, self.is_ytick)
	773	self.graphApp.Bind(wx.EVT_CLOSE, self.on_graphApp_close)
	774
	775	def on_graphApp_close(self, e):
	776	"""
	777	Gets values from graph appearance dialog and sends them off
	778	to modify the plot
	779	"""
	780	self.onGridOnOff(self.graphApp.get_togglegrid())
	781	self.xaxis_label = self.graphApp.get_xlab()
	782	self.yaxis_label = self.graphApp.get_ylab()
	783	self.xaxis_unit = self.graphApp.get_xunit()
	784	self.yaxis_unit = self.graphApp.get_yunit()
	785	self.xaxis_font = self.graphApp.get_xfont()
	786	self.yaxis_font = self.graphApp.get_yfont()
	787	self.is_xtick = self.graphApp.get_xtick_check()
	788	self.is_ytick = self.graphApp.get_ytick_check()
	789	if self.is_xtick:
	790	self.xaxis_tick = self.xaxis_font
	791	if self.is_ytick:
	792	self.yaxis_tick = self.yaxis_font
	793
	794	self.xaxis(self.xaxis_label, self.xaxis_unit,
	795	self.graphApp.get_xfont(), self.graphApp.get_xcolor(),
	796	self.xaxis_tick)
	797	self.yaxis(self.yaxis_label, self.yaxis_unit,
	798	self.graphApp.get_yfont(), self.graphApp.get_ycolor(),
	799	self.yaxis_tick)
	800
	801	self.graphApp.Destroy()

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

SasView

source: sasview/src/sas/sasgui/guiframe/local_perspectives/plotting/Plotter2D.py @ b796c72

Download in other formats: