ring.py @ 0bd2cd8

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 0bd2cd8 was 0d9dae8, checked in by Gervaise Alina <gervyh@…>, 16 years ago

add a new module containing common classes that I used often:
sans.guiframe.utils

remove event slicer and add it in sans.guicoom.events,
modified slicer for those events

Property mode set to 100644

File size: 10.8 KB

Rev	Line
[2d107b8]	1	#TODO: the line slicer should listen to all 2DREFRESH events, get the data and slice it
	2	# before pushing a new 1D data update.
	3
	4	#
	5	#TODO: NEED MAJOR REFACTOR
	6	#
	7
	8
	9	# Debug printout
	10	import math
	11	import wx
[0d9dae8]	12	from copy import deepcopy
	13
	14	from BaseInteractor import _BaseInteractor
	15	from sans.guicomm.events import NewPlotEvent, StatusEvent,SlicerParameterEvent,EVT_SLICER_PARS
[2d107b8]	16
	17	class AnnulusInteractor(_BaseInteractor):
	18	"""
	19	Select an annulus through a 2D plot
	20	"""
	21	def __init__(self,base,axes,color='black', zorder=3):
	22
	23	_BaseInteractor.__init__(self, base, axes, color=color)
	24	self.markers = []
	25	self.axes = axes
[ffd23b5]	26	self.qmax = self.base.data2D.xmax
[2d107b8]	27	self.connect = self.base.connect
	28
	29	## Number of points on the plot
	30	self.nbins = 20
	31
	32	# Inner circle
	33	self.inner_circle = RingInteractor(self, self.base.subplot, zorder=zorder, r=self.qmax/2.0)
[ffd23b5]	34	self.inner_circle.qmax = self.qmax
[2d107b8]	35
	36	#self.outer_circle.set_cursor(self.base.qmax/1.8, 0)
	37
	38
	39	self.update()
	40	self._post_data()
	41
	42	# Bind to slice parameter events
[0d9dae8]	43	self.base.parent.Bind(EVT_SLICER_PARS, self._onEVT_SLICER_PARS)
[2d107b8]	44
	45
	46	def _onEVT_SLICER_PARS(self, event):
	47	printEVT("AnnulusSlicer._onEVT_SLICER_PARS")
	48	event.Skip()
	49	if event.type == self.__class__.__name__:
	50	self.set_params(event.params)
	51	self.base.update()
	52
	53	def update_and_post(self):
	54	self.update()
	55	self._post_data()
	56
	57	def save_data(self, path, image, x, y):
	58	output = open(path, 'w')
	59
	60	data_x, data_y = self.get_data(image, x, y)
	61
	62	output.write("<phi> <average>\n")
	63	for i in range(len(data_x)):
	64	output.write("%g %g\n" % (data_x[i], data_y[i]))
	65	output.close()
	66
	67	def set_layer(self, n):
	68	self.layernum = n
	69	self.update()
	70
	71	def clear(self):
	72	self.clear_markers()
	73
	74	self.inner_circle.clear()
	75	#self.base.connect.disconnect()
[0d9dae8]	76	self.base.parent.Unbind(EVT_SLICER_PARS)
[2d107b8]	77
	78	def update(self):
	79	"""
	80	Respond to changes in the model by recalculating the profiles and
	81	resetting the widgets.
	82	"""
	83	# Update locations
	84	self.inner_circle.update()
	85
	86
	87
	88	def get_data(self, image, x, y):
	89	"""
	90	Return a 1D vector corresponding to the slice
	91	@param image: data matrix
	92	@param x: x matrix
	93	@param y: y matrix
	94	"""
	95	# If we have no data, just return
	96	if image == None:
	97	return
	98
	99	nbins = self.nbins
	100
	101	data_x = nbins*[0]
	102	data_y = nbins*[0]
	103	counts = nbins*[0]
	104	length = len(image)
	105	print "length x , y , image", len(x), len(y), length
	106
	107	for i_x in range(length):
	108	for i_y in range(length):
	109
	110	q = math.sqrt(x[i_x]x[i_x] + y[i_y]y[i_y])
	111	if (q>self.inner_circle._inner_mouse_x \
	112	and q<self.outer_circle._inner_mouse_x) \
	113	or (q<self.inner_circle._inner_mouse_x \
	114	and q>self.outer_circle._inner_mouse_x):
	115
	116	i_bin = int(math.ceil(nbins(math.atan2(y[i_y], x[i_x])+math.pi)/(2.0math.pi)) - 1)
	117
	118
	119	#data_y[i_bin] += math.exp(image[i_x][i_y])
	120	data_y[i_bin] += image[i_y][i_x]
	121	counts[i_bin] += 1.0
	122
	123	for i in range(nbins):
	124	data_x[i] = (1.0i+0.5)2.0*math.pi/nbins
	125	if counts[i]>0:
	126	data_y[i] = data_y[i]/counts[i]
	127
	128	return data_x, data_y
	129
	130	def save(self, ev):
	131	"""
	132	Remember the roughness for this layer and the next so that we
	133	can restore on Esc.
	134	"""
	135	self.base.freeze_axes()
	136	self.inner_circle.save(ev)
	137
	138
	139	def _post_data(self):
	140	# Compute data
	141	data = self.base.get_corrected_data()
	142	# If we have no data, just return
	143	if data == None:
	144	return
	145
	146	data_x, data_y = self.get_data(data, self.base.x, self.base.y)
	147
	148	name = "Ring"
	149	if hasattr(self.base, "name"):
	150	name += " %s" % self.base.name
	151
	152	wx.PostEvent(self.base.parent, AddPlotEvent(name=name,
	153	x = data_x,
	154	y = data_y,
	155	qmin = self.inner_circle._inner_mouse_x,
	156	qmax = self.outer_circle._inner_mouse_x,
	157	yscale = 'log',
	158	variable = 'ANGLE',
	159	ylabel = "\\rm{Intensity} ",
	160	yunits = "cm^{-1}",
	161	xlabel = "\\rm{\phi}",
	162	xunits = "rad",
	163	parent = self.base.__class__.__name__))
	164
	165
	166	def moveend(self, ev):
	167	self.base.thaw_axes()
	168
	169	# Post paramters
[0d9dae8]	170	event = SlicerParameterEvent()
[2d107b8]	171	event.type = self.__class__.__name__
	172	event.params = self.get_params()
	173	wx.PostEvent(self.base.parent, event)
	174
	175	self._post_data()
	176
	177	def restore(self):
	178	"""
	179	Restore the roughness for this layer.
	180	"""
	181	self.inner_circle.restore()
	182
	183
	184	def move(self, x, y, ev):
	185	"""
	186	Process move to a new position, making sure that the move is allowed.
	187	"""
	188	pass
	189
	190	def set_cursor(self, x, y):
	191	pass
	192
	193	def get_params(self):
	194	params = {}
	195	params["inner_radius"] = self.inner_circle._inner_mouse_x
	196
	197	params["nbins"] = self.nbins
	198	return params
	199
	200	def set_params(self, params):
	201
	202	inner = params["inner_radius"]
	203
	204	self.nbins = int(params["nbins"])
	205	self.inner_circle.set_cursor(inner, self.inner_circle._inner_mouse_y)
	206
	207	self._post_data()
	208
	209	def freeze_axes(self):
	210	self.base.freeze_axes()
	211
	212	def thaw_axes(self):
	213	self.base.thaw_axes()
	214
	215	def draw(self):
	216	self.base.draw()
	217
	218
	219	class RingInteractor(_BaseInteractor):
	220	"""
	221	Select an annulus through a 2D plot
	222	"""
	223	def __init__(self,base,axes,color='black', zorder=5, r=1.0):
	224
	225	_BaseInteractor.__init__(self, base, axes, color=color)
	226	self.markers = []
	227	self.axes = axes
	228	self._inner_mouse_x = r
	229	self._inner_mouse_y = 0
	230	self._inner_save_x = r
	231	self._inner_save_y = 0
	232	self.scale = 10.0
	233
	234	try:
	235	# Inner circle marker
	236	self.inner_marker = self.axes.plot([self._inner_mouse_x],[0], linestyle='',
	237	marker='s', markersize=10,
	238	color=self.color, alpha=0.6,
	239	pickradius=5, label="pick",
	240	zorder=zorder, # Prefer this to other lines
	241	visible=True)[0]
	242	except:
	243	self.inner_marker = self.axes.plot([self._inner_mouse_x],[0], linestyle='',
	244	marker='s', markersize=10,
	245	color=self.color, alpha=0.6,
	246	label="pick",
	247	visible=True)[0]
	248	message = "\nTHIS PROTOTYPE NEEDS THE LATEST VERSION OF MATPLOTLIB\n"
	249	message += "Get the SVN version that is at least as recent as June 1, 2007"
	250
	251	#raise "Version error", message
	252
	253	# Inner circle
	254
	255
	256	[self.inner_circle] = self.axes.plot([],[],
	257	linestyle='-', marker='',
	258	color=self.color)
	259	self.npts = 20
	260
	261	self.connect_markers([self.inner_marker])
	262	self.update()
	263
	264	def set_layer(self, n):
	265	self.layernum = n
	266	self.update()
	267
	268	def clear(self):
	269	self.clear_markers()
	270	try:
	271	self.inner_marker.remove()
	272	self.inner_circle.remove()
	273	except:
	274	# Old version of matplotlib
	275	for item in range(len(self.axes.lines)):
	276	del self.axes.lines[0]
	277
	278
	279
	280	def get_radius(self):
	281	return self._inner_mouse_x
	282
	283	def update(self):
	284	"""
	285	Draw the new roughness on the graph.
	286	"""
	287	# Plot inner circle
	288	x = []
	289	y = []
	290	for i in range(self.npts):
	291	phi = 2.0math.pi/(self.npts-1)i
	292
	293	xval = 1.0self._inner_mouse_xmath.cos(phi)
	294	yval = 1.0self._inner_mouse_xmath.sin(phi)
	295
	296	x.append(xval)
	297	y.append(yval)
	298
	299	self.inner_marker.set(xdata=[self._inner_mouse_x],ydata=[0])
	300	self.inner_circle.set_data(x, y)
	301
	302	def save(self, ev):
	303	"""
	304	Remember the roughness for this layer and the next so that we
	305	can restore on Esc.
	306	"""
	307	self._inner_save_x = self._inner_mouse_x
	308	self._inner_save_y = self._inner_mouse_y
	309	self.base.freeze_axes()
	310
	311	def moveend(self, ev):
	312	self.base.moveend(ev)
	313
	314	def restore(self):
	315	"""
	316	Restore the roughness for this layer.
	317	"""
	318	self._inner_mouse_x = self._inner_save_x
	319	self._inner_mouse_y = self._inner_save_y
	320
	321	def move(self, x, y, ev):
	322	"""
	323	Process move to a new position, making sure that the move is allowed.
	324	"""
	325	self._inner_mouse_x = x
	326	self._inner_mouse_y = y
	327	self.base.base.update()
	328
	329	def set_cursor(self, x, y):
	330	self.move(x, y, None)
	331	self.update()
	332
	333
	334	def get_params(self):
	335	params = {}
	336	params["radius"] = self._inner_mouse_x
	337	return params
	338
	339	def set_params(self, params):
	340
	341	x = params["radius"]
	342	self.set_cursor(x, self._inner_mouse_y)
	343
	344

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

SasView

source: sasview/guiframe/local_perspectives/plotting/ring.py @ 0bd2cd8

Download in other formats: