fitDialog.py @ 05da1f89

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

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

SasView

source: sasview/guitools/fitDialog.py @ 05da1f89

Download in other formats: