Plotter2D.py @ 3426077

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 3426077 was 3426077, checked in by Gervaise Alina <gervyh@…>, 16 years ago
working on boxslicer
Property mode set to `100644`
File size: 14.5 KB

Rev	Line
[1bf33c1]	1	"""
	2	This software was developed by the University of Tennessee as part of the
	3	Distributed Data Analysis of Neutron Scattering Experiments (DANSE)
	4	project funded by the US National Science Foundation.
	5
	6	See the license text in license.txt
	7
	8	copyright 2008, University of Tennessee
	9	"""
	10
	11
	12	import wx
	13	import sys
	14	import danse.common.plottools
	15	from danse.common.plottools.PlotPanel import PlotPanel
	16	from danse.common.plottools.plottables import Graph,Data1D
	17	from sans.guicomm.events import EVT_NEW_PLOT
	18	from sans.guicomm.events import StatusEvent ,NewPlotEvent
	19
	20
	21	from SlicerParameters import SlicerEvent
	22	from binder import BindArtist
	23	(InternalEvent, EVT_INTERNAL) = wx.lib.newevent.NewEvent()
	24	#from SlicerParameters import SlicerEvent
	25	#(InternalEvent, EVT_INTERNAL) = wx.lib.newevent.NewEvent()
	26	from Plotter1D import ModelPanel1D
	27	DEFAULT_QMAX = 0.05
	28
	29	DEFAULT_QSTEP = 0.001
	30	DEFAULT_BEAM = 0.005
[ef0c170]	31	BIN_WIDTH = 1.0
[1bf33c1]	32	import pylab
	33	from Plotter1D import PanelMenu
	34
	35	class ModelPanel2D( ModelPanel1D):
	36	"""
	37	Plot panel for use with the GUI manager
	38	"""
	39
	40	## Internal name for the AUI manager
	41	window_name = "plotpanel"
	42	## Title to appear on top of the window
	43	window_caption = "Plot Panel"
	44	## Flag to tell the GUI manager that this panel is not
	45	# tied to any perspective
	46	ALWAYS_ON = True
	47	## Group ID
	48	group_id = None
	49
	50
	51	def __init__(self, parent, id = -1,data2d=None, color = None,\
	52	dpi = None, style = wx.NO_FULL_REPAINT_ON_RESIZE, **kwargs):
	53	"""
	54	Initialize the panel
	55	"""
	56	ModelPanel1D.__init__(self, parent, id = id, style = style, **kwargs)
	57
	58	## Reference to the parent window
	59	self.parent = parent
	60	## Plottables
	61	self.plots = {}
	62	self.data2D= data2d
	63	self.data = data2d.data
	64	## Unique ID (from gui_manager)
	65	self.uid = None
	66
	67	## Action IDs for internal call-backs
	68	self.action_ids = {}
	69	self.connect = BindArtist(self.subplot.figure)
	70
	71	# Beam stop
	72	self.beamstop_radius = DEFAULT_BEAM
	73	# Slicer
	74	if data2d.xmax==None:
	75	data2d.xmax=DEFAULT_QMAX+self.qstep*0.01
	76
	77	self.qmax = data2d.xmax
	78	if data2d.ymax==None:
	79	data2d.ymax=DEFAULT_QMAX+self.qstep*0.01
	80	self.imax = data2d.ymax
	81	#self.radius = math.sqrt(data2d.)
	82	self.qstep = DEFAULT_QSTEP
[ef0c170]	83	#print "panel2D qmax",self.qmax,
	84
[1bf33c1]	85	self.x = pylab.arange(-1self.qmax, self.qmax+self.qstep0.01, self.qstep)
	86	self.y = pylab.arange(-1self.imax, self.imax+self.qstep0.01, self.qstep)
	87
	88	self.slicer_z = 5
	89	self.slicer = None
	90	self.parent.Bind(EVT_INTERNAL, self._onEVT_INTERNAL)
	91	self.axes_frozen = False
	92
	93
	94	## Graph
	95	self.graph = Graph()
	96	self.graph.xaxis("\\rm{Q}", 'A^{-1}')
	97	self.graph.yaxis("\\rm{Intensity} ","cm^{-1}")
	98	self.graph.render(self)
	99
	100	def _onEVT_1DREPLOT(self, event):
	101	"""
	102	Data is ready to be displayed
	103	@param event: data event
	104	"""
	105	#TODO: Check for existence of plot attribute
[ab8f936]	106
[1bf33c1]	107	# Check whether this is a replot. If we ask for a replot
	108	# and the plottable no longer exists, ignore the event.
	109	if hasattr(event, "update") and event.update==True \
	110	and event.plot.name not in self.plots.keys():
	111	return
[ab8f936]	112
[1bf33c1]	113	if hasattr(event, "reset"):
	114	self._reset()
[ef0c170]	115	#print "model2 d event",event.plot.name, event.plot.id, event.plot.group_id
	116	#print "plottable list ",self.plots.keys()
	117	#print self.plots
[1bf33c1]	118	is_new = True
	119	if event.plot.name in self.plots.keys():
	120	# Check whether the class of plottable changed
	121	if not event.plot.__class__==self.plots[event.plot.name].__class__:
[ab8f936]	122	#overwrite a plottable using the same name
	123	print "is deleting the new plottable"
[1bf33c1]	124	self.graph.delete(self.plots[event.plot.name])
	125	else:
[ab8f936]	126	# plottable is already draw on the panel
[1bf33c1]	127	is_new = False
	128
[ab8f936]	129
	130	if is_new:
	131	# a new plottable overwrites a plotted one using the same id
[ef0c170]	132	#print "went here",self.plots.itervalues()
[ab8f936]	133	for plottable in self.plots.itervalues():
	134	if event.plot.id==plottable.id :
	135	self.graph.delete(plottable)
	136
	137	self.plots[event.plot.name] = event.plot
	138	self.graph.add(self.plots[event.plot.name])
	139	else:
	140	#replot the graph
	141	self.plots[event.plot.name].x = event.plot.x
	142	self.plots[event.plot.name].y = event.plot.y
	143	self.plots[event.plot.name].dy = event.plot.dy
	144	if hasattr(event.plot, 'dx') and hasattr(self.plots[event.plot.name], 'dx'):
	145	self.plots[event.plot.name].dx = event.plot.dx
	146
	147
[1bf33c1]	148	# Check axis labels
	149	#TODO: Should re-factor this
	150	#if event.plot._xunit != self.graph.prop["xunit"]:
	151	self.graph.xaxis(event.plot._xaxis, event.plot._xunit)
[ab8f936]	152
[1bf33c1]	153	#if event.plot._yunit != self.graph.prop["yunit"]:
	154	self.graph.yaxis(event.plot._yaxis, event.plot._yunit)
[ab8f936]	155
[1bf33c1]	156	self.graph.render(self)
	157	self.subplot.figure.canvas.draw_idle()
	158
	159
	160	def onContextMenu(self, event):
	161	"""
	162	2D plot context menu
	163	@param event: wx context event
	164	"""
	165
	166	#slicerpop = wx.Menu()
	167	slicerpop = PanelMenu()
	168	slicerpop.set_plots(self.plots)
	169	slicerpop.set_graph(self.graph)
	170
	171	# Option to save the data displayed
	172
	173	# Various plot options
	174	id = wx.NewId()
	175	slicerpop.Append(id,'&Save image', 'Save image as PNG')
	176	wx.EVT_MENU(self, id, self.onSaveImage)
	177
	178
	179	item_list = self.parent.get_context_menu(self.graph)
	180	if (not item_list==None) and (not len(item_list)==0):
	181	slicerpop.AppendSeparator()
	182	for item in item_list:
	183	try:
	184	id = wx.NewId()
	185	slicerpop.Append(id, item[0], item[1])
	186	wx.EVT_MENU(self, id, item[2])
	187	except:
	188	print sys.exc_value
	189	print RuntimeError, "View1DPanel2D.onContextMenu: bad menu item"
	190
	191	slicerpop.AppendSeparator()
	192
	193	id = wx.NewId()
	194	slicerpop.Append(id, '&Perform circular average')
	195	wx.EVT_MENU(self, id, self.onCircular)
	196
	197	id = wx.NewId()
[ef0c170]	198	slicerpop.Append(id, '&Sector [Q view]')
[1bf33c1]	199	wx.EVT_MENU(self, id, self.onSectorQ)
	200
	201	id = wx.NewId()
[ef0c170]	202	slicerpop.Append(id, '&Annulus [Phi view ]')
[1bf33c1]	203	wx.EVT_MENU(self, id, self.onSectorPhi)
	204
[38224f10]	205	id = wx.NewId()
	206	slicerpop.Append(id, '&Box averaging')
	207	wx.EVT_MENU(self, id, self.onBoxavg)
[1bf33c1]	208
	209	id = wx.NewId()
	210	slicerpop.Append(id, '&Clear slicer')
	211	wx.EVT_MENU(self, id, self.onClearSlicer)
	212
	213	id = wx.NewId()
	214	slicerpop.Append(id, '&Edit Slicer Parameters')
	215	wx.EVT_MENU(self, id, self._onEditSlicer)
	216
	217	slicerpop.AppendSeparator()
	218
	219	id = wx.NewId()
	220	slicerpop.Append(id, '&Save image')
	221	wx.EVT_MENU(self, id, self.onSaveImage)
	222
	223	id = wx.NewId()
	224	slicerpop.Append(id, '&Toggle Linear/Log scale')
	225	wx.EVT_MENU(self, id, self._onToggleScale)
	226
	227	pos = event.GetPosition()
	228	pos = self.ScreenToClient(pos)
	229	self.PopupMenu(slicerpop, pos)
	230
	231	def _setSlicer(self, slicer):
	232	# Clear current slicer
	233	#printEVT("Plotter2D._setSlicer %s" % slicer)
	234
	235	if not self.slicer == None:
	236	self.slicer.clear()
	237
	238	self.slicer_z += 1
	239	self.slicer = slicer(self, self.subplot, zorder=self.slicer_z)
	240	self.subplot.set_ylim(-self.qmax, self.qmax)
	241	self.subplot.set_xlim(-self.qmax, self.qmax)
	242	self.update()
	243	self.slicer.update()
	244
	245
	246	def get_corrected_data(self):
	247	# Protect against empty data set
	248	if self.data == None:
	249	return None
	250	import copy
	251	output = copy.deepcopy(self.data)
	252	return output
	253	def freeze_axes(self):
	254	self.axes_frozen = True
	255
	256	def thaw_axes(self):
	257	self.axes_frozen = False
	258
	259	def onMouseMotion(self,event):
	260	pass
	261	def onWheel(self, event):
	262	pass
	263	def update(self, draw=True):
	264	"""
	265	Respond to changes in the model by recalculating the
	266	profiles and resetting the widgets.
	267	"""
	268	#self.slicer.update()
	269	self.draw()
	270
	271
	272	def _getEmptySlicerEvent(self):
	273	return SlicerEvent(type=None,
	274	params=None,
	275	obj_class=None)
	276	def _onEVT_INTERNAL(self, event):
	277	"""
	278	I don't understand why Unbind followed by a Bind
	279	using a modified self.slicer doesn't work.
	280	For now, I post a clear event followed by
	281	a new slicer event...
	282	"""
	283	self._setSlicer(event.slicer)
	284
	285	def _setSlicer(self, slicer):
	286	# Clear current slicer
	287	#printEVT("Plotter2D._setSlicer %s" % slicer)
	288
	289	if not self.slicer == None:
	290	self.slicer.clear()
	291
	292	self.slicer_z += 1
	293	self.slicer = slicer(self, self.subplot, zorder=self.slicer_z)
	294	self.subplot.set_ylim(-self.qmax, self.qmax)
	295	self.subplot.set_xlim(-self.qmax, self.qmax)
	296	self.update()
	297	self.slicer.update()
	298
	299	# Post slicer event
	300	event = self._getEmptySlicerEvent()
	301	event.type = self.slicer.__class__.__name__
	302	event.obj_class = self.slicer.__class__
	303	event.params = self.slicer.get_params()
	304	wx.PostEvent(self.parent, event)
	305
	306	def onCircular(self, event):
	307	"""
	308	perform circular averaging on Data2D
	309	"""
	310
	311	from DataLoader.manipulations import CircularAverage
[ef0c170]	312	import math
	313	self.radius= math.sqrt( math.pow(self.qmax,2)+math.pow(self.qmax,2))
	314	print "radius?",self.radius
	315	Circle = CircularAverage( r_min=0, r_max=self.radius, bin_width=0.001)
	316	#Circle = CircularAverage( r_min=0, r_max=13705.0, bin_width=BIN_WIDTH )
[1bf33c1]	317	#Circle = CircularAverage( r_min= -1, r_max= 1, bin_width=0.001 )
[ef0c170]	318	#Circle = CircularAverage( r_min= -1*self.radius, r_max= self.radius, bin_width=0.001 )
[1bf33c1]	319	circ = Circle(self.data2D)
	320	from sans.guiframe.dataFitting import Data1D
	321	if hasattr(circ,"dxl"):
	322	dxl= circ.dxl
	323	else:
	324	dxl= None
	325	if hasattr(circ,"dxw"):
	326	dxw= circ.dxw
	327	else:
	328	dxw= None
[ef0c170]	329
[1bf33c1]	330	new_plot = Data1D(x=circ.x,y=circ.y,dy=circ.dy,dxl=dxl,dxw=dxw)
	331	new_plot.name = "Circ avg "+ self.data2D.name
	332	new_plot.source=self.data2D.source
	333	new_plot.info=self.data2D.info
	334	new_plot.interactive = True
	335	#print "loader output.detector",output.source
	336	new_plot.detector =self.data2D.detector
	337
	338	# If the data file does not tell us what the axes are, just assume...
	339	new_plot.xaxis("\\rm{q}","A^{-1}")
	340	new_plot.yaxis("\\rm{Intensity} ","cm^{-1}")
	341	new_plot.group_id = "Circ avg "+ self.data2D.name
	342	self.scale = 'log'
[ef0c170]	343
[1bf33c1]	344	wx.PostEvent(self.parent, NewPlotEvent(plot=new_plot, title=new_plot.name))
[ef0c170]	345
[1bf33c1]	346	def _onEditSlicer(self, event):
	347	if self.slicer !=None:
	348	from SlicerParameters import SlicerParameterPanel
	349	dialog = SlicerParameterPanel(self.parent, -1, "Slicer Parameters")
	350	dialog.set_slicer(self.slicer.__class__.__name__,
	351	self.slicer.get_params())
	352	if dialog.ShowModal() == wx.ID_OK:
	353	dialog.Destroy()
	354
	355	def onSectorQ(self, event):
	356	"""
	357	Perform sector averaging on Q
	358	"""
	359
[ef0c170]	360	from SectorSlicer import SectorInteractor
[1bf33c1]	361	self.onClearSlicer(event)
[ef0c170]	362	wx.PostEvent(self.parent, InternalEvent(slicer= SectorInteractor))
[1bf33c1]	363
	364	def onSectorPhi(self, event):
	365	"""
	366	Perform sector averaging on Phi
	367	"""
[ef0c170]	368	from AnnulusSlicer import AnnulusInteractor
[1bf33c1]	369	self.onClearSlicer(event)
[ef0c170]	370	wx.PostEvent(self.parent, InternalEvent(slicer= AnnulusInteractor))
[1bf33c1]	371
[38224f10]	372	def onBoxavg(self,event):
	373	from boxSlicer import BoxInteractor
	374	self.onClearSlicer(event)
[3426077]	375	wx.PostEvent(self.parent, InternalEvent(slicer= BoxInteractor))
	376	"""
	377	self.onClearSlicer(event)
[b319def8]	378	self.slicer=BoxInteractor
	379	from SlicerParameters import SlicerParameterPanel
[3426077]	380
[b319def8]	381	dialog = SlicerParameterPanel(self.parent, -1, "Slicer Parameters")
	382	dialog.set_slicer(self.slicer.__name__,
	383	self.slicer.get_params())
	384	if dialog.ShowModal() == wx.ID_OK:
	385	dialog.Destroy()
[38224f10]	386	wx.PostEvent(self.parent, InternalEvent(slicer= BoxInteractor))
	387	print "onboxavg",self.slicer
[3426077]	388	"""
[b319def8]	389
[1bf33c1]	390	def onClearSlicer(self, event):
	391	"""
	392	Clear the slicer on the plot
	393	"""
	394	if not self.slicer==None:
	395	self.slicer.clear()
	396	self.subplot.figure.canvas.draw()
	397	self.slicer = None
	398
	399	# Post slicer None event
	400	event = self._getEmptySlicerEvent()
	401	wx.PostEvent(self.parent, event)
	402
	403	def _onEditDetector(self, event):
	404	print "on parameter"
	405
	406
	407	def _onToggleScale(self, event):
	408	"""
	409	toggle pixel scale and replot image
	410	"""
	411	if self.scale == 'log':
	412	self.scale = 'linear'
	413	else:
	414	self.scale = 'log'
	415	self.image(self.data,self.xmin_2D,self.xmax_2D,self.ymin_2D,
	416	self.ymax_2D,self.zmin_2D ,self.zmax_2D )
	417	wx.PostEvent(self.parent, StatusEvent(status="Image is in %s scale"%self.scale))
	418
	419
	420
	421
	422

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

SasView

source: sasview/guiframe/local_perspectives/plotting/Plotter2D.py @ 3426077

Download in other formats: