last_boxSlicer.py @ 3e852ef5

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 3e852ef5 was 6d920cd, checked in by Gervaise Alina <gervyh@…>, 16 years ago
print statement removed
Property mode set to `100644`
File size: 20.5 KB

Rev	Line
[6d920cd]	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	#from config import printEVT
	11	from BaseInteractor import _BaseInteractor
	12	from copy import deepcopy
	13	import math
	14
	15	from sans.guicomm.events import NewPlotEvent, StatusEvent,SlicerParameterEvent,EVT_SLICER_PARS
	16	import SlicerParameters
	17	import wx
	18
	19
	20	class BoxInteractor(_BaseInteractor):
	21	"""
	22	Select an annulus through a 2D plot
	23	"""
	24	def __init__(self,base,axes,color='black', zorder=3,
	25	x_min=0.0025, x_max=0.0025, y_min=0.0025, y_max=0.0025):
	26
	27	_BaseInteractor.__init__(self, base, axes, color=color)
	28	self.markers = []
	29	self.axes = axes
	30	self.qmax = self.base.data2D.xmax
	31	self.connect = self.base.connect
	32	self.x= 0.5*min(math.fabs(self.base.data2D.xmax),math.fabs( self.base.data2D.xmin))
	33	self.y= 0.5*min(math.fabs(self.base.data2D.xmax),math.fabs( self.base.data2D.xmin))
	34
	35	self.theta2= math.pi/3
	36	## Number of points on the plot
	37	self.nbins = 30
	38	self.count=0
	39	self.error=0
	40	self.averager=None
	41	self.left_line = VerticalLine(self, self.base.subplot,color='blue',
	42	zorder=zorder,
	43	ymin= -self.y ,
	44	ymax= self.y ,
	45	xmin= -self.x,
	46	xmax= -self.x)
	47	self.left_line.qmax = self.qmax
	48
	49	self.right_line= VerticalLine(self, self.base.subplot,color='black',
	50	zorder=zorder,
	51	ymin= -self.y ,
	52	ymax= self.y,
	53	xmin= self.x,
	54	xmax= self.x)
	55	self.right_line.qmax = self.qmax
	56
	57	self.top_line= HorizontalLine(self, self.base.subplot,color='green',
	58	zorder=zorder,
	59	xmin= -self.x,
	60	xmax= self.x,
	61	ymin= self.y,
	62	ymax= self.y)
	63	self.top_line.qmax= self.qmax
	64
	65	self.bottom_line= HorizontalLine(self, self.base.subplot,color='gray',
	66	zorder=zorder,
	67	xmin= -self.x,
	68	xmax= self.x,
	69	ymin= -self.y,
	70	ymax= -self.y)
	71	self.bottom_line.qmax= self.qmax
	72
	73	self.update()
	74	self._post_data()
	75
	76	# Bind to slice parameter events
	77	self.base.Bind(EVT_SLICER_PARS, self._onEVT_SLICER_PARS)
	78
	79
	80	def _onEVT_SLICER_PARS(self, event):
	81	wx.PostEvent(self.base, StatusEvent(status="BoxSlicer._onEVT_SLICER_PARS"))
	82	event.Skip()
	83	if event.type == self.__class__.__name__:
	84	self.set_params(event.params)
	85	self.base.update()
	86
	87	def update_and_post(self):
	88	self.update()
	89	self._post_data()
	90
	91	def save_data(self, path, image, x, y):
	92	output = open(path, 'w')
	93
	94	data_x, data_y = self.get_data(image, x, y)
	95
	96	output.write("<phi> <average>\n")
	97	for i in range(len(data_x)):
	98	output.write("%g %g\n" % (data_x[i], data_y[i]))
	99	output.close()
	100
	101	def set_layer(self, n):
	102	self.layernum = n
	103	self.update()
	104
	105	def clear(self):
	106	self.averager=None
	107	self.clear_markers()
	108	self.left_line.clear()
	109	self.right_line.clear()
	110	self.top_line.clear()
	111	self.bottom_line.clear()
	112	#self.base.connect.disconnect()
	113	self.base.Unbind(EVT_SLICER_PARS)
	114
	115	def update(self):
	116	"""
	117	Respond to changes in the model by recalculating the profiles and
	118	resetting the widgets.
	119	"""
	120
	121	if self.top_line.has_move:
	122	#print"top has moved"
	123	self.top_line.update()
	124	self.bottom_line.update(ymin= -self.top_line.y1,
	125	ymax= -self.top_line.y2)
	126	self.left_line.update(ymin= -self.top_line.y1,
	127	ymax= -self.top_line.y2)
	128	self.right_line.update(ymin= -self.top_line.y1,
	129	ymax= -self.top_line.y2)
	130
	131	if self.bottom_line.has_move:
	132	#print "bottom has move"
	133	self.bottom_line.update()
	134	self.top_line.update(ymin= -self.bottom_line.y1,
	135	ymax= -self.bottom_line.y2)
	136	self.left_line.update(ymin= self.bottom_line.y1,
	137	ymax= self.top_line.y1)
	138	self.right_line.update(ymin= self.bottom_line.y1,
	139	ymax=self.top_line.y1)
	140
	141	if self.left_line.has_move:
	142
	143	self.left_line.update()
	144	self.right_line.update(xmin = - self.left_line.x1,
	145	xmax = - self.left_line.x1)
	146	self.bottom_line.update(xmin= self.left_line.x1,
	147	xmax= self.right_line.x1)
	148	self.top_line.update(xmin= self.left_line.x1,
	149	xmax= self.right_line.x1)
	150
	151	if self.right_line.has_move:
	152
	153	self.right_line.update()
	154	self.left_line.update(xmin = -self.right_line.x1,
	155	xmax = -self.right_line.x1)
	156
	157	self.bottom_line.update(xmin= self.left_line.x1,
	158	xmax= self.right_line.x1)
	159
	160	self.top_line.update(xmin= self.left_line.x1,
	161	xmax= self.right_line.x1)
	162
	163
	164	def save(self, ev):
	165	"""
	166	Remember the roughness for this layer and the next so that we
	167	can restore on Esc.
	168	"""
	169	self.base.freeze_axes()
	170	self.left_line.save(ev)
	171	self.right_line.save(ev)
	172	self.top_line.save(ev)
	173	self.bottom_line.save(ev)
	174	def _post_data(self):
	175	pass
	176
	177
	178	def post_data(self,new_slab=None , nbins=None):
	179	""" post data averaging in Q"""
	180	x_min= min(self.left_line.x1, self.right_line.x1)
	181	x_max= max(self.left_line.x1, self.right_line.x1)
	182
	183	y_min= min(self.top_line.y1, self.bottom_line.y1)
	184	y_max= max(self.top_line.y1, self.bottom_line.y1)
	185
	186	if nbins !=None:
	187	self.nbins
	188	if self.averager==None:
	189	if new_slab ==None:
	190	raise ValueError,"post data:cannot average , averager is empty"
	191	self.averager= new_slab
	192	bin_width= (x_max + math.fabs(x_min))/self.nbins
	193
	194	box = self.averager( x_min=x_min, x_max=x_max, y_min=y_min, y_max=y_max,
	195	bin_width=bin_width)
	196
	197	boxavg = box(self.base.data2D)
	198
	199	from sans.guiframe.dataFitting import Data1D
	200	if hasattr(boxavg,"dxl"):
	201	dxl= boxavg.dxl
	202	else:
	203	dxl= None
	204	if hasattr(boxavg,"dxw"):
	205	dxw=boxavg.dxw
	206	else:
	207	dxw= None
	208
	209	new_plot = Data1D(x=boxavg.x,y=boxavg.y,dy=boxavg.dy,dxl=dxl,dxw=dxw)
	210	new_plot.name = str(self.averager.__name__) +"("+ self.base.data2D.name+")"
	211
	212
	213
	214	new_plot.source=self.base.data2D.source
	215	new_plot.interactive = True
	216	#print "loader output.detector",output.source
	217	new_plot.detector =self.base.data2D.detector
	218	# If the data file does not tell us what the axes are, just assume...
	219	new_plot.xaxis("\\rm{Q}", 'rad')
	220	new_plot.yaxis("\\rm{Intensity} ","cm^{-1}")
	221	new_plot.group_id = str(self.averager.__name__)+self.base.data2D.name
	222	new_plot.id = str(self.averager.__name__)
	223	wx.PostEvent(self.base.parent, NewPlotEvent(plot=new_plot,
	224	title=str(self.averager.__name__) ))
	225
	226
	227	def _post_data(self):
	228	# Compute data
	229	data = self.base.data2D
	230	from DataLoader.manipulations import Boxavg
	231	radius = math.sqrt(math.pow(self.qmax,2)+math.pow(self.qmax,2))
	232	self.x= math.fabs(self.right_line.x1)
	233	self.y= math.fabs(self.top_line.y1 )
	234
	235	box = Boxavg (x_min=-self.x, x_max=self.x, y_min=-self.y, y_max=self.y)
	236
	237	self.count, self.error= box(self.base.data2D)
	238
	239	#print "post data"
	240
	241
	242	def moveend(self, ev):
	243	self.base.thaw_axes()
	244
	245	# Post paramters
	246	event = SlicerParameterEvent()
	247	event.type = self.__class__.__name__
	248	event.params = self.get_params()
	249	wx.PostEvent(self.base.parent, event)
	250
	251	self._post_data()
	252
	253	def restore(self):
	254	"""
	255	Restore the roughness for this layer.
	256	"""
	257	self.left_line.restore()
	258	self.right_line.restore()
	259	self.top_line.restore()
	260	self.bottom_line.restore()
	261
	262	def move(self, x, y, ev):
	263	"""
	264	Process move to a new position, making sure that the move is allowed.
	265	"""
	266	pass
	267
	268	def set_cursor(self, x, y):
	269	pass
	270
	271	def get_params(self):
	272	params = {}
	273	params["x_max"]= math.fabs(self.right_line.x1)
	274	params["y_max"]= math.fabs(self.top_line.y1)
	275	params["nbins"]= self.nbins
	276	params["errors"] = self.error
	277	params["count"]= self.count
	278	return params
	279
	280	def set_params(self, params):
	281
	282	self.x = float(math.fabs(params["x_max"]))
	283	self.y = float(math.fabs(params["y_max"] ))
	284	self.nbins=params["nbins"]
	285	self.left_line.update(xmin= -1*self.x,
	286	xmax = -1*self.x,
	287	ymin= -self.y,
	288	ymax= self.y,
	289	)
	290	self.right_line.update(xmin= self.x,
	291	xmax = self.x,
	292	ymin= -self.y,
	293	ymax= self.y,
	294	)
	295	self.top_line.update(xmin= -1*self.x,
	296	xmax= self.x,
	297	ymin= self.y,
	298	ymax= self.y)
	299	self.bottom_line.update(xmin= -1*self.x,
	300	xmax= self.x,
	301	ymin= -1*self.y,
	302	ymax= -1*self.y)
	303
	304	self.post_data( nbins=None)
	305	def freeze_axes(self):
	306	self.base.freeze_axes()
	307
	308	def thaw_axes(self):
	309	self.base.thaw_axes()
	310
	311	def draw(self):
	312	self.base.draw()
	313
	314	class HorizontalLine(_BaseInteractor):
	315	"""
	316	Select an annulus through a 2D plot
	317	"""
	318	def __init__(self,base,axes,color='black', zorder=5,mline=None,ymin=None, ymax=None, y=0.5,
	319	xmin=0.0,xmax=0.5):
	320
	321	_BaseInteractor.__init__(self, base, axes, color=color)
	322	self.markers = []
	323	self.axes = axes
	324	self.x1= xmax
	325	self.save_x1= xmax
	326
	327	self.x2= xmin
	328	self.save_x2= xmin
	329
	330	self.y1= ymax
	331	self.save_y1= ymax
	332
	333	self.y2= ymin
	334	self.save_y2= ymin
	335	self.mline= mline
	336	try:
	337	# Inner circle marker
	338	self.inner_marker = self.axes.plot([0],[self.y1], linestyle='',
	339	marker='s', markersize=10,
	340	color=self.color, alpha=0.6,
	341	pickradius=5, label="pick",
	342	zorder=zorder, # Prefer this to other lines
	343	visible=True)[0]
	344	except:
	345	self.inner_marker = self.axes.plot([0],[self.y1], linestyle='',
	346	marker='s', markersize=10,
	347	color=self.color, alpha=0.6,
	348	label="pick",
	349	visible=True)[0]
	350	message = "\nTHIS PROTOTYPE NEEDS THE LATEST VERSION OF MATPLOTLIB\n"
	351	message += "Get the SVN version that is at least as recent as June 1, 2007"
	352
	353	self.line = self.axes.plot([self.x1,-self.x1],
	354	[self.y1,self.y2],
	355	linestyle='-', marker='',
	356	color=self.color,
	357	visible=True)[0]
	358
	359	self.npts = 20
	360	self.has_move=False
	361	self.connect_markers([self.line, self.inner_marker])
	362	self.update()
	363
	364	def set_layer(self, n):
	365	self.layernum = n
	366	self.update()
	367
	368	def clear(self):
	369	self.clear_markers()
	370	try:
	371
	372	self.line.remove()
	373	except:
	374	# Old version of matplotlib
	375	for item in range(len(self.axes.lines)):
	376	del self.axes.lines[0]
	377
	378	def get_radius(self):
	379
	380	return 0
	381
	382	def update(self,xmin=None, xmax=None,ymin=None,ymax=None, mline=None,translation=False):
	383	"""
	384	Draw the new roughness on the graph.
	385	"""
	386	if xmin !=None:
	387	self.x1 = xmin
	388	if ymin !=None:
	389	self.y1 = ymin
	390	self.inner_marker.set(xdata=[0],ydata=[self.y1])
	391	self.line.set(xdata=[self.x1,-self.x1],
	392	ydata=[self.y1,self.y1])
	393
	394
	395
	396	def save(self, ev):
	397	"""
	398	Remember the roughness for this layer and the next so that we
	399	can restore on Esc.
	400	"""
	401	self.save_x1= self.x1
	402	self.save_x2= self.x2
	403
	404	self.save_y1= self.y1
	405	self.save_y2= self.y2
	406
	407
	408	self.base.freeze_axes()
	409
	410	def moveend(self, ev):
	411
	412	self.has_move=False
	413	self.base.moveend(ev)
	414
	415	def restore(self):
	416	"""
	417	Restore the roughness for this layer.
	418	"""
	419	self.x1 = self.save_x1
	420	self.x2 = self.save_x2
	421	self.y1 = self.save_y1
	422	self.y2 = self.save_y2
	423
	424
	425	def move(self, x, y, ev):
	426	"""
	427	Process move to a new position, making sure that the move is allowed.
	428	"""
	429	#print "horizontal move x y "
	430	self.y1= y
	431	self.has_move=True
	432	self.base.base.update()
	433
	434	def set_cursor(self, x, y):
	435	self.move(x, y, None)
	436	self.update()
	437
	438
	439	def get_params(self):
	440	params = {}
	441	params["radius"] = self.x1
	442	#params["theta"] = self.xmax
	443	return params
	444
	445	def set_params(self, params):
	446
	447	x = params["radius"]
	448	self.set_cursor(x, self._inner_mouse_y)
	449
	450
	451
	452
	453	class VerticalLine(_BaseInteractor):
	454	"""
	455	Select an annulus through a 2D plot
	456	"""
	457	def __init__(self,base,axes,color='black', zorder=5, mline=None, ymin=0.0,
	458	ymax=0.5,xmin=-0.5,xmax=0.5
	459	):
	460
	461	_BaseInteractor.__init__(self, base, axes, color=color)
	462	self.markers = []
	463	self.axes = axes
	464
	465	self.x1= xmax
	466	self.x2= xmin
	467	self.y1= ymax
	468	self.y2= ymin
	469	try:
	470	# Inner circle marker
	471	self.inner_marker = self.axes.plot([self.x1],[0], linestyle='',
	472	marker='s', markersize=10,
	473	color=self.color, alpha=0.6,
	474	pickradius=5, label="pick",
	475	zorder=zorder, # Prefer this to other lines
	476	visible=True)[0]
	477	except:
	478	self.inner_marker = self.axes.plot([self.x1],[0], linestyle='',
	479	marker='s', markersize=10,
	480	color=self.color, alpha=0.6,
	481	label="pick",
	482	visible=True)[0]
	483	message = "\nTHIS PROTOTYPE NEEDS THE LATEST VERSION OF MATPLOTLIB\n"
	484	message += "Get the SVN version that is at least as recent as June 1, 2007"
	485
	486	self.line = self.axes.plot([self.x1,self.x2],[self.y1,self.y2],
	487	linestyle='-', marker='',
	488	color=self.color,
	489	visible=True)[0]
	490
	491	self.has_move=False
	492	self.connect_markers([self.line, self.inner_marker])
	493	self.update()
	494
	495	def set_layer(self, n):
	496	self.layernum = n
	497	self.update()
	498
	499	def clear(self):
	500	self.clear_markers()
	501	try:
	502
	503	self.line.remove()
	504	except:
	505	# Old version of matplotlib
	506	for item in range(len(self.axes.lines)):
	507	del self.axes.lines[0]
	508
	509
	510	def get_radius(self):
	511	return 0
	512
	513	def update(self,xmin=None,xmax=None,ymin=None, ymax=None, opline=None,translation=False):
	514	"""
	515	Draw the new roughness on the graph.
	516	"""
	517
	518
	519	if xmin!=None:
	520	self.x1=xmin
	521	if ymin!=None:
	522	self.y1=ymin
	523	self.inner_marker.set(xdata=[self.x1],ydata=[0])
	524	self.line.set(xdata=[self.x1,self.x1],
	525	ydata=[self.y1,-self.y1])
	526
	527
	528
	529	def save(self, ev):
	530	"""
	531	Remember the roughness for this layer and the next so that we
	532	can restore on Esc.
	533	"""
	534	self.save_x1= self.x1
	535	self.save_x2= self.x2
	536	self.save_y1= self.y1
	537	self.save_y2= self.y2
	538
	539	self.base.freeze_axes()
	540
	541	def moveend(self, ev):
	542
	543	self.has_move=False
	544	self.base.moveend(ev)
	545
	546	def restore(self):
	547	"""
	548	Restore the roughness for this layer.
	549	"""
	550	self.x1 = self.save_x1
	551	self.x2 = self.save_x2
	552	self.y1 = self.save_y1
	553	self.y2= self.save_y2
	554
	555
	556	def move(self, x, y, ev):
	557	"""
	558	Process move to a new position, making sure that the move is allowed.
	559	"""
	560	self.has_move=True
	561
	562	# compute the b intercept of the vertical line
	563	self.x1= x
	564
	565
	566	self.base.base.update()
	567
	568
	569	def set_cursor(self, x, y):
	570	self.move(x, y, None)
	571	self.update()
	572
	573
	574	def get_params(self):
	575	params = {}
	576	params["x"] = self.xmin
	577	params["ymin"] = self.ymin
	578	params["ymax"] = self.ymax
	579	return params
	580
	581	def set_params(self, params):
	582	"""
	583	Draw a vertical line given some value of params
	584	@param params: a dictionary containing value for x, ymin , ymax to draw
	585	a vertical line
	586	"""
	587	x = params["x"]
	588	ymin = params["ymin"]
	589	ymax = params["ymax"]
	590	#self.set_cursor(x, self._inner_mouse_y)
	591	self.update(self,x =x,ymin =ymin, ymax =ymax)
	592
	593
	594	class BoxInteractorX(BoxInteractor):
	595	def __init__(self,base,axes,color='black', zorder=3):
	596	BoxInteractor.__init__(self, base, axes, color=color)
	597	self.base=base
	598	self._post_data()
	599	def _post_data(self):
	600	from DataLoader.manipulations import SlabX
	601	self.post_data(SlabX )
	602
	603
	604	class BoxInteractorY(BoxInteractor):
	605	def __init__(self,base,axes,color='black', zorder=3):
	606	BoxInteractor.__init__(self, base, axes, color=color)
	607	self.base=base
	608	self._post_data()
	609	def _post_data(self):
	610	from DataLoader.manipulations import SlabY
	611	self.post_data(SlabY )
	612
	613

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

SasView

source: sasview/guiframe/local_perspectives/plotting/last_boxSlicer.py @ 3e852ef5

Download in other formats: