fitDialog.py @ 784e2fa

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 784e2fa was 3755d77, checked in by Mathieu Doucet <doucetm@…>, 16 years ago
fixed bug with transforms
Property mode set to `100644`
File size: 19.7 KB

Line
1	#!/usr/bin/python
2
3	# fitDialog.py
4
5	import wx
6	from PlotPanel import PlotPanel
7	from plottables import Theory1D
8	import math,pylab,fittings
9	import transform
10
11	def format_number(value, high=False):
12	"""
13	Return a float in a standardized, human-readable formatted string
14	"""
15	try:
16	value = float(value)
17	except:
18	print "returning 0"
19	return "0"
20
21	if high:
22	return "%-6.4g" % value
23	else:
24	return "%-5.3g" % value
25
26
27	class LinearFit(wx.Dialog):
28	def __init__(self, parent, plottable, push_data,transform, id, title):
29	wx.Dialog.__init__(self, parent, id, title, size=(400, 350))
30
31	"""
32	Dialog window pops- up when select Linear fit on Context menu
33	Displays fitting parameters
34	"""
35	self.parent = parent
36	self.transform = transform
37
38	# Registered owner for close event
39	self._registered_close = None
40
41	#dialog panel self call function to plot the fitting function
42	self.push_data = push_data
43	#dialog self plottable
44	self.plottable = plottable
45
46	# Receive transformations of x and y
47	self.xLabel,self.yLabel,self.Avalue,self.Bvalue,\
48	self.ErrAvalue,self.ErrBvalue,self.Chivalue= self.transform()
49
50	#Dialog interface
51	vbox = wx.BoxSizer(wx.VERTICAL)
52	sizer = wx.GridBagSizer(5,5)
53
54	_BOX_WIDTH = 100
55
56	self.tcA = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
57	self.tcA.SetToolTipString("Fit value for the slope parameter.")
58	self.tcErrA = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
59	self.tcErrA.SetToolTipString("Error on the slope parameter.")
60	self.tcB = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
61	self.tcA.SetToolTipString("Fit value for the constant parameter.")
62	self.tcErrB = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
63	self.tcErrB.SetToolTipString("Error on the constant parameter.")
64	self.tcChi = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
65	self.tcChi.SetToolTipString("Chi^2 over degrees of freedom.")
66	self.xminFit = wx.TextCtrl(self,-1,size=(_BOX_WIDTH,20))
67	self.xminFit.SetToolTipString("Enter the minimum value on the x-axis to be included in the fit.")
68	self.xmaxFit = wx.TextCtrl(self,-1,size=(_BOX_WIDTH,20))
69	self.xmaxFit.SetToolTipString("Enter the maximum value on the x-axis to be included in the fit.")
70	self.initXmin = wx.TextCtrl(self,-1,size=(_BOX_WIDTH,20))
71	self.initXmin.SetToolTipString("Minimum value on the x-axis for the plotted data.")
72	self.initXmax = wx.TextCtrl(self,-1,size=(_BOX_WIDTH,20))
73	self.initXmax.SetToolTipString("Maximum value on the x-axis for the plotted data.")
74
75	# Make the info box not editable
76	#_BACKGROUND_COLOR = '#ffdf85'
77	_BACKGROUND_COLOR = self.GetBackgroundColour()
78	self.initXmin.SetEditable(False)
79	self.initXmin.SetBackgroundColour(_BACKGROUND_COLOR)
80	self.initXmax.SetEditable(False)
81	self.initXmax.SetBackgroundColour(_BACKGROUND_COLOR)
82
83
84	# Buttons on the bottom
85	self.static_line_1 = wx.StaticLine(self, -1)
86	self.btFit =wx.Button(self,-1,'Fit')
87	self.btFit.Bind(wx.EVT_BUTTON, self._onFit)
88	self.btFit.SetToolTipString("Perform fit.")
89	self.btClose =wx.Button(self, wx.ID_CANCEL,'Close')
90	self.btClose.Bind(wx.EVT_BUTTON, self._on_close)
91
92	# Intro
93	explanation = "Perform fit for y(x) = Ax + B"
94
95	vbox.Add(sizer)
96
97	ix = 0
98	iy = 1
99	sizer.Add(wx.StaticText(self, -1, explanation),(iy, ix),\
100	(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
101	iy += 2
102	sizer.Add(wx.StaticText(self, -1, 'Parameter A'),(iy, ix),\
103	(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
104	ix += 1
105	sizer.Add(self.tcA,(iy, ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
106	ix += 1
107	sizer.Add(wx.StaticText(self, -1, '+/-'),(iy, ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
108	ix += 1
109	sizer.Add(self.tcErrA, (iy, ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
110	iy += 1
111	ix = 0
112	sizer.Add(wx.StaticText(self, -1, 'Parameter B'),(iy, ix),(1,1),\
113	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
114	ix += 1
115	sizer.Add(self.tcB, (iy, ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
116	ix += 1
117	sizer.Add(wx.StaticText(self, -1, '+/-'),(iy, ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
118	ix += 1
119	sizer.Add(self.tcErrB, (iy, ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
120	iy += 1
121	ix = 0
122	sizer.Add(wx.StaticText(self, -1, 'Chi2/dof'),(iy, ix),(1,1),\
123	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
124	ix += 1
125	sizer.Add(self.tcChi, (iy, ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
126	iy += 2
127	ix = 1
128	sizer.Add(wx.StaticText(self, -1, 'Min'),(iy, ix),(1,1),\
129	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
130	ix += 2
131	sizer.Add(wx.StaticText(self, -1, 'Max'),(iy, ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
132	iy += 1
133	ix = 0
134	sizer.Add(wx.StaticText(self, -1, 'Maximum range (linear scale)'),(iy, ix),(1,1),\
135	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
136	ix +=1
137	sizer.Add(self.initXmin, (iy, ix),(1,1),\
138	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
139	ix += 2
140	sizer.Add(self.initXmax, (iy, ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
141
142	iy += 1
143	ix = 0
144	sizer.Add(wx.StaticText(self, -1, 'Fit range of '+self.xLabel),(iy, ix),(1,1),\
145	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
146	ix += 1
147	sizer.Add(self.xminFit, (iy, ix),(1,1),\
148	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
149	ix += 2
150	sizer.Add(self.xmaxFit, (iy, ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
151	iy += 1
152	ix = 1
153
154	vbox.Add(self.static_line_1, 0, wx.EXPAND, 0)
155
156	sizer_button = wx.BoxSizer(wx.HORIZONTAL)
157	sizer_button.Add((20, 20), 1, wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
158	sizer_button.Add(self.btFit, 0, wx.LEFT\|wx.RIGHT\|wx.ADJUST_MINSIZE, 10)
159	sizer_button.Add(self.btClose, 0, wx.LEFT\|wx.RIGHT\|wx.ADJUST_MINSIZE, 10)
160	vbox.Add(sizer_button, 0, wx.EXPAND\|wx.BOTTOM\|wx.TOP, 10)
161
162
163
164	sizer.Add(self.btFit, (iy, ix),(1,1), wx.LEFT\|wx.ADJUST_MINSIZE, 0)
165
166
167	#panel.SetSizer(sizer)
168	self.SetSizer(vbox)
169	self.Centre()
170
171	# Receives the type of model for the fitting
172	from LineModel import LineModel
173	self.model = LineModel()
174
175	#Display the fittings values
176	self.default_A = self.model.getParam('A')
177	self.default_B = self.model.getParam('B')
178	self.cstA = fittings.Parameter(self.model, 'A', self.default_A)
179	self.cstB = fittings.Parameter(self.model, 'B', self.default_B)
180
181	# Set default value of parameter in fit dialog
182
183	if self.Avalue==None:
184	self.tcA.SetLabel(format_number(self.default_A))
185	else :
186	self.tcA.SetLabel(format_number(self.Avalue))
187	if self.Bvalue==None:
188	self.tcB.SetLabel(format_number(self.default_B))
189	else:
190	self.tcB.SetLabel(format_number(self.Bvalue))
191	if self.ErrAvalue==None:
192	self.tcErrA.SetLabel(format_number(0.0))
193	else:
194	self.tcErrA.SetLabel(format_number(self.ErrAvalue))
195	if self.ErrBvalue==None:
196	self.tcErrB.SetLabel(format_number(0.0))
197	else:
198	self.tcErrB.SetLabel(format_number(self.ErrBvalue))
199	if self.Chivalue==None:
200	self.tcChi.SetLabel(format_number(0.0))
201	else:
202	self.tcChi.SetLabel(format_number(self.Chivalue))
203	if self.plottable.x !=[]:
204	#store the values of View in self.x,self.y,self.dx,self.dy
205	self.x,self.y,self.dx,self.dy= self.plottable.returnValuesOfView()
206
207	#self.mini =min(self.x)
208	#self.maxi =max(self.x)
209
210	try:
211	self.mini = self.floatForwardTransform(min(self.x))
212	except:
213	self.mini = "Invalid"
214	try:
215	self.maxi = self.floatForwardTransform(max(self.x))
216	except:
217	self.maxi = "Invalid"
218
219	self.initXmin.SetValue(format_number(min(self.plottable.x)))
220	self.initXmax.SetValue(format_number(max(self.plottable.x)))
221
222	self.xminFit.SetLabel(format_number(self.mini))
223	self.xmaxFit.SetLabel(format_number(self.maxi))
224
225
226	def register_close(self, owner):
227	"""
228	Method to register the close event to a parent
229	window that needs notification when the dialog
230	is closed
231	@param owner: parent window
232	"""
233	self._registered_close = owner
234
235	def _on_close(self, event):
236	"""
237	Close event.
238	Notify registered owner if available.
239	"""
240	event.Skip()
241	if self._registered_close is not None:
242	self._registered_close()
243
244	def _onFit(self ,event):
245	"""
246	Performs the fit. Receive an event when clicking on the button Fit.Computes chisqr ,
247	A and B parameters of the best linear fit y=Ax +B
248	Push a plottable to
249	"""
250	tempx=[]
251	tempy=[]
252	tempdy = []
253
254
255
256
257	# Check if View contains a x array .we online fit when x exits
258	# makes transformation for y as a line to fit
259	if self.x != []:
260
261
262	if(self.checkFitValues(self.xminFit) == True):
263	#Check if the field of Fit Dialog contain values and use the x max and min of the user
264	xminView,xmaxView = self._checkVal(self.xminFit.GetValue(),self.xmaxFit.GetValue())
265	xmin = self.floatInvTransform(xminView)
266	xmax = self.floatInvTransform(xmaxView)
267
268
269	# Store the transformed values of view x, y,dy in variables before the fit
270	if self.yLabel.lower() == "log10(y)":
271	if (self.xLabel.lower() == "log10(x)"):
272	for i in range(len(self.x)):
273	if self.x[i]>= math.log10(xmin):
274	tempy.append(math.log10(self.y[i]))
275	tempdy.append(transform.errToLogX(self.y[i],0,self.dy[i],0))
276	else:
277	for i in range(len(self.y)):
278	tempy.append(math.log10(self.y[i]))
279	tempdy.append(transform.errToLogX(self.y[i],0,self.dy[i],0))
280	else:
281	tempy = self.y
282	tempdy = self.dy
283
284	if (self.xLabel.lower() == "log10(x)"):
285	for x_i in self.x:
286	if x_i >= math.log10(xmin):
287	tempx.append(math.log10(x_i))
288	else:
289	tempx = self.x
290
291	#Find the fitting parameters
292	# Always use the same defaults, so that fit history doesn't play a role!
293	self.cstA = fittings.Parameter(self.model, 'A', self.default_A)
294	self.cstB = fittings.Parameter(self.model, 'B', self.default_B)
295
296	if (self.xLabel.lower() == "log10(x)"):
297	chisqr, out, cov = fittings.sansfit(self.model, [self.cstA, self.cstB],
298	tempx, tempy,tempdy,math.log10(xmin),math.log10(xmax))
299	else:
300	chisqr, out, cov = fittings.sansfit(self.model,
301	[self.cstA, self.cstB],tempx, tempy,tempdy,xminView,xmaxView)
302
303	# Use chi2/dof
304	if len(tempx)>0:
305	chisqr = chisqr/len(tempx)
306
307	#Check that cov and out are iterable before displaying them
308	if cov ==None:
309	errA =0.0
310	errB =0.0
311	else:
312	errA= math.sqrt(cov[0][0])
313	errB= math.sqrt(cov[1][1])
314	if out==None:
315	cstA=0.0
316	cstB=0.0
317	else:
318	cstA=out[0]
319	cstB=out[1]
320	# Reset model with the right values of A and B
321	self.model.setParam('A', float(cstA))
322	self.model.setParam('B', float(cstB))
323
324	tempx = []
325	tempy = []
326	y_model = 0.0
327	# load tempy with the minimum transformation
328
329	if self.xLabel == "log10(x)":
330	y_model = self.model.run(math.log10(xmin))
331	tempx.append(xmin)
332	else:
333	y_model = self.model.run(xminView)
334	tempx.append(xminView)
335
336	if self.yLabel == "log10(y)":
337	tempy.append(math.pow(10,y_model))
338	print "tempy",tempy
339	else:
340	tempy.append(y_model)
341
342	# load tempy with the maximum transformation
343	if self.xLabel == "log10(x)":
344	y_model = self.model.run(math.log10(xmax))
345	tempx.append(xmax)
346	else:
347	y_model = self.model.run(xmaxView)
348	tempx.append(xmaxView)
349
350	if self.yLabel == "log10(y)":
351	tempy.append(math.pow(10,y_model))
352	else:
353	tempy.append(y_model)
354	#Set the fit parameter display when FitDialog is opened again
355	self.Avalue=cstB
356	self.Bvalue=cstA
357	self.ErrAvalue=errA
358	self.ErrBvalue=errB
359	self.Chivalue=chisqr
360	self.push_data(tempx,tempy,xminView,xmaxView,xmin,xmax,self._ongetValues())
361
362	# Display the fitting value on the Fit Dialog
363	self._onsetValues(cstB, cstA, errA,errB,chisqr)
364
365
366
367	def _onsetValues(self,cstA,cstB,errA,errB,Chi):
368	"""
369	Display the value on fit Dialog
370	"""
371	self.tcA.SetValue(format_number(cstA))
372	self.tcB.SetValue(format_number(cstB))
373	self.tcErrA.SetValue(format_number(errA))
374	self.tcErrB.SetValue(format_number(errB))
375	self.tcChi.SetValue(format_number(Chi))
376
377	def _ongetValues(self):
378	"""
379	Display the value on fit Dialog
380	"""
381	return self.Avalue, self.Bvalue,self.ErrAvalue,self.ErrBvalue,self.Chivalue
382
383
384	def _checkVal(self,usermin, usermax):
385	"""
386	Ensure that fields parameter contains a min and a max value
387	within x min and x max range
388	"""
389	if float(usermin) < float(usermax):
390	if float(usermin) >= float(self.mini) and float(usermin) < float(self.maxi):
391	self.xminFit.SetValue(format_number(float(usermin)))
392	else:
393	self.xminFit.SetValue(format_number(float(self.mini)))
394
395	if float(usermax) > float(self.mini) and float(usermax) <= float(self.maxi):
396	self.xmaxFit.SetLabel(format_number(float(usermax)))
397	else:
398	self.xmaxFit.SetLabel(format_number(float(self.maxi)))
399
400	mini =float(self.xminFit.GetValue())
401	maxi =float(self.xmaxFit.GetValue())
402
403	return mini, maxi
404
405	def floatForwardTransform(self,x):
406	"""
407	transform a float.
408	"""
409	#TODO: refactor this with proper object-oriented design
410	# This code stinks.
411
412	if ( self.xLabel=="x" ):
413	return transform.toX(x)
414
415	if ( self.xLabel=="x^(2)" ):
416	return transform.toX2(x)
417
418	if (self.xLabel=="log10(x)" ):
419	return math.log10(x)
420
421	def floatTransform(self,x):
422	"""
423	transform a float.It is use to determine the x.View min and x.View max for values
424	not in x
425	"""
426	#TODO: refactor this with proper object-oriented design
427	# This code stinks.
428
429	if ( self.xLabel=="x" ):
430	return transform.toX(x)
431
432	if ( self.xLabel=="x^(2)" ):
433	return transform.toX2(x)
434
435	if (self.xLabel=="log10(x)" ):
436	if x >0:
437	return x
438	else:
439	raise ValueError,"cannot compute log of a negative number"
440
441	def floatInvTransform(self,x):
442	"""
443	transform a float.It is use to determine the x.View min and x.View max for values
444	not in x
445	"""
446	#TODO: refactor this. This is just a hack to make the
447	# functionality work without rewritting the whole code
448	# with good design (which really should be done...).
449
450	if ( self.xLabel=="x^(2)" ):
451	return math.sqrt(x)
452
453	elif (self.xLabel=="log10(x)" ):
454	return math.pow(10, x)
455
456	return x
457
458	def checkFitValues(self,item):
459	"""
460	Check the validity of input values
461	"""
462	flag = True
463	value = item.GetValue()
464	# Check for possible values entered
465	if (self.xLabel=="log10(x)"):
466	if (float(value) > 0):
467	item.SetBackgroundColour(wx.WHITE)
468	item.Refresh()
469	else:
470	flag = False
471	item.SetBackgroundColour("pink")
472	item.Refresh()
473
474	return flag
475
476	def setFitRange(self,xmin,xmax,xminTrans,xmaxTrans):
477	"""
478	Set fit parameters
479	"""
480	self.xminFit.SetValue(format_number(xmin))
481	self.xmaxFit.SetValue(format_number(xmax))
482
483	def set_fit_region(self, xmin, xmax):
484	"""
485	Set the fit region
486	@param xmin: minimum x-value to be included in fit
487	@param xmax: maximum x-value to be included in fit
488	"""
489	# Check values
490	try:
491	float(xmin)
492	float(xmax)
493	except:
494	raise ValueError, "LinearFit.set_fit_region: fit range must be floats"
495	self.xminFit.SetValue(format_number(xmin))
496	self.xmaxFit.SetValue(format_number(xmax))
497
498	class MyApp(wx.App):
499	def OnInit(self):
500	wx.InitAllImageHandlers()
501	plot = Theory1D([],[])
502	dialog = LinearFit(None, plot, self.onFitDisplay,self.returnTrans, -1, 'Linear Fit')
503	if dialog.ShowModal() == wx.ID_OK:
504	pass
505	dialog.Destroy()
506
507	return 1
508
509	def onFitDisplay(self, tempx,tempy,xminView,xmaxView,xmin,xmax,func):
510	pass
511
512	def returnTrans(self):
513	return '','',0,0,0,0,0
514
515	# end of class MyApp
516
517	if __name__ == "__main__":
518	app = MyApp(0)
519	app.MainLoop()

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

SasView

source: sasview/guitools/fitDialog.py @ 784e2fa

Download in other formats: