Plotter2D.py @ 7619f50

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

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

SasView

source: sasview/sansguiframe/src/sans/guiframe/local_perspectives/plotting/Plotter2D.py @ 7619f50

Download in other formats: