fitDialog.py @ 54b9ad2

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 54b9ad2 was b9a5f0e, checked in by krzywon, 10 years ago
90% complete with the conversion.
Property mode set to `100644`
File size: 28.3 KB

Line
1	import wx
2	from plottables import Theory1D
3	import math
4	import numpy
5	import fittings
6	import transform
7	import sys
8
9	#Linear fit panel size
10	if sys.platform.count("win32") > 0:
11	FONT_VARIANT = 0
12	PNL_WIDTH = 450
13	PNL_HEIGHT = 500
14	else:
15	FONT_VARIANT = 1
16	PNL_WIDTH = 500
17	PNL_HEIGHT = 500
18	RG_ON = True
19
20	def format_number(value, high=False):
21	"""
22	Return a float in a standardized, human-readable formatted string
23	"""
24	try:
25	value = float(value)
26	except:
27	output = "NaN"
28	return output.lstrip().rstrip()
29
30	if high:
31	output = "%-6.4g" % value
32
33	else:
34	output = "%-5.3g" % value
35	return output.lstrip().rstrip()
36
37
38	class LinearFit(wx.Dialog):
39	def __init__(self, parent, plottable, push_data, transform, title):
40	"""
41	Dialog window pops- up when select Linear fit on Context menu
42	Displays fitting parameters
43	"""
44	wx.Dialog.__init__(self, parent, title=title,
45	size=(PNL_WIDTH, 350))
46	self.parent = parent
47	self.transform = transform
48	#Font
49	self.SetWindowVariant(variant=FONT_VARIANT)
50	# Registered owner for close event
51	self._registered_close = None
52
53	#dialog panel self call function to plot the fitting function
54	self.push_data = push_data
55	#dialog self plottable
56	self.plottable = plottable
57	self.rg_on = False
58	# Receive transformations of x and y
59	self.xLabel, self.yLabel, self.Avalue, self.Bvalue,\
60	self.ErrAvalue, self.ErrBvalue, self.Chivalue = self.transform()
61
62	#Dialog interface
63	vbox = wx.BoxSizer(wx.VERTICAL)
64	sizer = wx.GridBagSizer(5, 5)
65	_BOX_WIDTH = 100
66
67	self.tcA = wx.TextCtrl(self, -1, size=(_BOX_WIDTH, 20))
68	self.tcA.SetToolTipString("Fit value for the slope parameter.")
69	self.tcErrA = wx.TextCtrl(self, -1, size=(_BOX_WIDTH, 20))
70	self.tcErrA.SetToolTipString("Error on the slope parameter.")
71	self.tcB = wx.TextCtrl(self, -1, size=(_BOX_WIDTH, 20))
72	self.tcA.SetToolTipString("Fit value for the constant parameter.")
73	self.tcErrB = wx.TextCtrl(self, -1, size=(_BOX_WIDTH, 20))
74	self.tcErrB.SetToolTipString("Error on the constant parameter.")
75	self.tcChi = wx.TextCtrl(self, -1, size=(_BOX_WIDTH, 20))
76	self.tcChi.SetToolTipString("Chi^2 over degrees of freedom.")
77	self.xminFit = wx.TextCtrl(self, -1, size=(_BOX_WIDTH, 20))
78	msg = "Enter the minimum value on "
79	msg += "the x-axis to be included in the fit."
80	self.xminFit.SetToolTipString(msg)
81	self.xmaxFit = wx.TextCtrl(self, -1, size=(_BOX_WIDTH, 20))
82	msg = "Enter the maximum value on "
83	msg += " the x-axis to be included in the fit."
84	self.xmaxFit.SetToolTipString(msg)
85	self.initXmin = wx.TextCtrl(self, -1, size=(_BOX_WIDTH, 20))
86	msg = "Minimum value on the x-axis for the plotted data."
87	self.initXmin.SetToolTipString(msg)
88	self.initXmax = wx.TextCtrl(self, -1, size=(_BOX_WIDTH, 20))
89	msg = "Maximum value on the x-axis for the plotted data."
90	self.initXmax.SetToolTipString(msg)
91
92	# Make the info box not editable
93	#_BACKGROUND_COLOR = '#ffdf85'
94	_BACKGROUND_COLOR = self.GetBackgroundColour()
95	self.initXmin.SetEditable(False)
96	self.initXmin.SetBackgroundColour(_BACKGROUND_COLOR)
97	self.initXmax.SetEditable(False)
98	self.initXmax.SetBackgroundColour(_BACKGROUND_COLOR)
99
100	# Buttons on the bottom
101	self.bg_on = False
102	self.static_line_1 = wx.StaticLine(self, -1)
103	self.btFit = wx.Button(self, -1, 'Fit')
104	self.btFit.Bind(wx.EVT_BUTTON, self._onFit)
105	self.btFit.SetToolTipString("Perform fit.")
106	self.btClose =wx.Button(self, wx.ID_CANCEL, 'Close')
107	self.btClose.Bind(wx.EVT_BUTTON, self._on_close)
108	if RG_ON:
109	if (self.yLabel == "ln(y)" or self.yLabel == "ln(y*x)") and \
110	(self.xLabel == "x^(2)"):
111	self.rg_on = True
112	if (self.xLabel == "x^(4)") and (self.yLabel == "y*x^(4)"):
113	self.bg_on = True
114	# Intro
115	explanation = "Perform fit for y(x) = ax + b"
116	if self.bg_on:
117	param_a = 'Background (= Parameter a)'
118	else:
119	param_a = 'Parameter a'
120	vbox.Add(sizer)
121	ix = 0
122	iy = 1
123	sizer.Add(wx.StaticText(self, -1, explanation), (iy, ix),
124	(1, 1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
125	iy += 2
126	sizer.Add(wx.StaticText(self, -1, param_a), (iy, ix),
127	(1, 1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
128	ix += 1
129	sizer.Add(self.tcA,(iy, ix),(1, 1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
130	ix += 1
131	sizer.Add(wx.StaticText(self, -1, '+/-'),
132	(iy, ix), (1, 1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
133	ix += 1
134	sizer.Add(self.tcErrA, (iy, ix), (1, 1),
135	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
136	iy += 1
137	ix = 0
138	sizer.Add(wx.StaticText(self, -1, 'Parameter b'), (iy, ix),(1, 1),
139	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
140	ix += 1
141	sizer.Add(self.tcB, (iy, ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
142	ix += 1
143	sizer.Add(wx.StaticText(self, -1, '+/-'), (iy, ix),
144	(1, 1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
145	ix += 1
146	sizer.Add(self.tcErrB, (iy, ix), (1, 1),
147	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
148	iy += 1
149	ix = 0
150	sizer.Add(wx.StaticText(self, -1, 'Chi2/dof'), (iy, ix), (1, 1),
151	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
152	ix += 1
153	sizer.Add(self.tcChi, (iy, ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
154
155
156	#sizer.Add(wx.StaticLine(self, -1), 0, wx.EXPAND, 0)
157	iy += 2
158	ix = 1
159	sizer.Add(wx.StaticText(self, -1, 'Min'), (iy, ix), (1, 1),
160	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
161	ix += 2
162	sizer.Add(wx.StaticText(self, -1, 'Max'), (iy, ix),
163	(1, 1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
164
165	iy += 1
166	ix = 0
167	sizer.Add(wx.StaticText(self, -1, 'Maximum range (linear scale)'),
168	(iy, ix),(1,1),
169	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
170	ix += 1
171	sizer.Add(self.initXmin, (iy, ix), (1,1),
172	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
173	ix += 2
174	sizer.Add(self.initXmax, (iy, ix), (1,1),
175	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
176
177	iy += 1
178	ix = 0
179	sizer.Add(wx.StaticText(self, -1, 'Fit range of ' + self.xLabel),
180	(iy, ix), (1, 1),
181	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
182	ix += 1
183	sizer.Add(self.xminFit, (iy, ix), (1, 1),
184	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
185	ix += 2
186	sizer.Add(self.xmaxFit, (iy, ix), (1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
187	if self.rg_on:
188	self.SetSize((PNL_WIDTH, PNL_HEIGHT))
189	I0_stxt = wx.StaticText(self, -1, 'I(q=0)')
190	self.I0_tctr = wx.TextCtrl(self, -1, '')
191	self.I0_tctr.SetEditable(False)
192	self.I0_tctr.SetBackgroundColour(_BACKGROUND_COLOR)
193	self.I0err_tctr = wx.TextCtrl(self, -1, '')
194	self.I0err_tctr.SetEditable(False)
195	self.I0err_tctr.SetBackgroundColour(_BACKGROUND_COLOR)
196	Rg_stxt = wx.StaticText(self, -1, 'Rg [A]')
197	Rg_stxt.Show(self.yLabel == "ln(y)" )
198	self.Rg_tctr = wx.TextCtrl(self, -1, '')
199	self.Rg_tctr.SetEditable(False)
200	self.Rg_tctr.SetBackgroundColour(_BACKGROUND_COLOR)
201	self.Rg_tctr.Show(self.yLabel == "ln(y)" )
202	self.Rgerr_tctr = wx.TextCtrl(self, -1, '')
203	self.Rgerr_tctr.SetEditable(False)
204	self.Rgerr_tctr.SetBackgroundColour(_BACKGROUND_COLOR)
205	self.Rgerr_tctr.Show(self.yLabel == "ln(y)" )
206	self.Rgerr_pm = wx.StaticText(self, -1, '+/-')
207	self.Rgerr_pm.Show(self.yLabel == "ln(y)" )
208	Diameter_stxt = wx.StaticText(self, -1, 'Rod Diameter [A]')
209	Diameter_stxt.Show(self.yLabel == "ln(y*x)")
210	self.Diameter_tctr = wx.TextCtrl(self, -1, '')
211	self.Diameter_tctr.SetEditable(False)
212	self.Diameter_tctr.SetBackgroundColour(_BACKGROUND_COLOR)
213	self.Diameter_tctr.Show(self.yLabel == "ln(y*x)")
214	self.Diameter_pm = wx.StaticText(self, -1, '+/-')
215	self.Diameter_pm.Show(self.yLabel == "ln(y*x)")
216	self.Diametererr_tctr = wx.TextCtrl(self, -1, '')
217	self.Diametererr_tctr.SetEditable(False)
218	self.Diametererr_tctr.SetBackgroundColour(_BACKGROUND_COLOR)
219	self.Diametererr_tctr.Show(self.yLabel == "ln(y*x)")
220	RgQmin_stxt = wx.StaticText(self, -1, 'Rg*Qmin')
221	self.RgQmin_tctr = wx.TextCtrl(self, -1, '')
222	self.RgQmin_tctr.SetEditable(False)
223	self.RgQmin_tctr.SetBackgroundColour(_BACKGROUND_COLOR)
224	RgQmax_stxt = wx.StaticText(self, -1, 'Rg*Qmax')
225	self.RgQmax_tctr = wx.TextCtrl(self, -1, '')
226	self.RgQmax_tctr.SetEditable(False)
227	self.RgQmax_tctr.SetBackgroundColour(_BACKGROUND_COLOR)
228
229	iy += 2
230	ix = 0
231	sizer.Add(I0_stxt, (iy, ix), (1,1),
232	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
233	ix += 1
234	sizer.Add(self.I0_tctr, (iy, ix), (1,1),
235	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
236	ix += 1
237	sizer.Add(wx.StaticText(self, -1, '+/-'), (iy, ix),
238	(1, 1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
239	ix += 1
240	sizer.Add(self.I0err_tctr, (iy, ix), (1,1),
241	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
242
243	iy += 1
244	ix = 0
245	sizer.Add(Rg_stxt, (iy, ix),(1,1),
246	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
247	ix += 1
248	sizer.Add(self.Rg_tctr, (iy, ix), (1,1),
249	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
250
251	ix += 1
252	sizer.Add(self.Rgerr_pm, (iy, ix),
253	(1, 1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
254	ix += 1
255	sizer.Add(self.Rgerr_tctr, (iy, ix), (1,1),
256	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
257	iy += 1
258	ix = 0
259	sizer.Add(Diameter_stxt, (iy, ix),(1,1),
260	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
261	ix += 1
262	sizer.Add(self.Diameter_tctr, (iy, ix), (1,1),
263	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
264
265	ix += 1
266	sizer.Add(self.Diameter_pm, (iy, ix),
267	(1, 1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
268	ix += 1
269	sizer.Add(self.Diametererr_tctr, (iy, ix), (1,1),
270	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
271	iy += 1
272	ix = 0
273	sizer.Add(RgQmin_stxt, (iy, ix),(1,1),
274	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
275	ix += 1
276	sizer.Add(self.RgQmin_tctr, (iy, ix), (1,1),
277	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
278	iy += 1
279	ix = 0
280	sizer.Add(RgQmax_stxt, (iy, ix),(1,1),
281	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
282	ix += 1
283	sizer.Add(self.RgQmax_tctr, (iy, ix), (1,1),
284	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
285
286	iy += 1
287	ix = 1
288
289	vbox.Add(self.static_line_1, 0, wx.EXPAND, 0)
290	sizer_button = wx.BoxSizer(wx.HORIZONTAL)
291	sizer_button.Add((20, 20), 1, wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
292	sizer_button.Add(self.btFit, 0, wx.LEFT\|wx.RIGHT\|wx.ADJUST_MINSIZE, 10)
293	sizer_button.Add(self.btClose, 0,
294	wx.LEFT\|wx.RIGHT\|wx.ADJUST_MINSIZE, 10)
295	vbox.Add(sizer_button, 0, wx.EXPAND\|wx.BOTTOM\|wx.TOP, 10)
296
297	sizer.Add(self.btFit, (iy, ix), (1,1), wx.LEFT\|wx.ADJUST_MINSIZE, 0)
298	#panel.SetSizer(sizer)
299	self.SetSizer(vbox)
300	self.Centre()
301	# Receives the type of model for the fitting
302	from LineModel import LineModel
303	self.model = LineModel()
304	#Display the fittings values
305	self.default_A = self.model.getParam('A')
306	self.default_B = self.model.getParam('B')
307	self.cstA = fittings.Parameter(self.model, 'A', self.default_A)
308	self.cstB = fittings.Parameter(self.model, 'B', self.default_B)
309
310	# Set default value of parameter in fit dialog
311	if self.Avalue == None:
312	self.tcA.SetValue(format_number(self.default_A))
313	else:
314	self.tcA.SetLabel(format_number(self.Avalue))
315	if self.Bvalue == None:
316	self.tcB.SetValue(format_number(self.default_B))
317	else:
318	self.tcB.SetLabel(format_number(self.Bvalue))
319	if self.ErrAvalue == None:
320	self.tcErrA.SetLabel(format_number(0.0))
321	else:
322	self.tcErrA.SetLabel(format_number(self.ErrAvalue))
323	if self.ErrBvalue == None:
324	self.tcErrB.SetLabel(format_number(0.0))
325	else:
326	self.tcErrB.SetLabel(format_number(self.ErrBvalue))
327	if self.Chivalue == None:
328	self.tcChi.SetLabel(format_number(0.0))
329	else:
330	self.tcChi.SetLabel(format_number(self.Chivalue))
331	if self.plottable.x != []:
332	#store the values of View in self.x,self.y,self.dx,self.dy
333	self.x, self.y, self.dx, \
334	self.dy = self.plottable.returnValuesOfView()
335	try:
336	self.mini = self.floatForwardTransform(min(self.x))
337	except:
338	self.mini = "Invalid"
339	try:
340	self.maxi = self.floatForwardTransform(max(self.x))
341	except:
342	self.maxi = "Invalid"
343
344	self.initXmin.SetValue(format_number(min(self.plottable.x)))
345	self.initXmax.SetValue(format_number(max(self.plottable.x)))
346	self.mini = min(self.x)
347	self.maxi = max(self.x)
348	self.xminFit.SetValue(format_number(self.mini))
349	self.xmaxFit.SetValue(format_number(self.maxi))
350
351	def register_close(self, owner):
352	"""
353	Method to register the close event to a parent
354	window that needs notification when the dialog
355	is closed
356
357	:param owner: parent window
358
359	"""
360	self._registered_close = owner
361
362	def _on_close(self, event):
363	"""
364	Close event.
365	Notify registered owner if available.
366	"""
367	event.Skip()
368	if self._registered_close is not None:
369	self._registered_close()
370
371	def _onFit(self, event):
372	"""
373	Performs the fit. Receive an event when clicking on
374	the button Fit.Computes chisqr ,
375	A and B parameters of the best linear fit y=Ax +B
376	Push a plottable to
377	"""
378	tempx = []
379	tempy = []
380	tempdy = []
381
382	# Check if View contains a x array .we online fit when x exits
383	# makes transformation for y as a line to fit
384	if self.x != []:
385	if(self.checkFitValues(self.xminFit) == True):
386	#Check if the field of Fit Dialog contain values
387	# and use the x max and min of the user
388	#xminView,xmaxView = self._checkVal(self.xminFit.GetValue(),
389	#self.xmaxFit.GetValue())
390	if not self._checkVal(self.xminFit, self.xmaxFit):
391	return
392	xminView = float(self.xminFit.GetValue())
393	xmaxView = float(self.xmaxFit.GetValue())
394	#xmin = self.floatInvTransform(xminView)
395	#xmax = self.floatInvTransform(xmaxView)
396	xmin = xminView
397	xmax = xmaxView
398	# Store the transformed values of view x, y,dy
399	# in variables before the fit
400	if self.yLabel.lower() == "log10(y)":
401	if (self.xLabel.lower() == "log10(x)"):
402	for i in range(len(self.x)):
403	if self.x[i] >= math.log10(xmin):
404	tempy.append(math.log10(self.y[i]))
405	tempdy.append(transform.errToLogX(self.y[i],
406	0, self.dy[i], 0))
407	else:
408	for i in range(len(self.y)):
409	tempy.append(math.log10(self.y[i]))
410	tempdy.append(transform.errToLogX(self.y[i],
411	0, self.dy[i], 0))
412	else:
413	tempy = self.y
414	tempdy = self.dy
415
416	if (self.xLabel.lower() == "log10(x)"):
417	for x_i in self.x:
418	if x_i >= math.log10(xmin):
419	tempx.append(math.log10(x_i))
420	else:
421	tempx = self.x
422
423	#Find the fitting parameters
424	# Always use the same defaults, so that fit history
425	#doesn't play a role!
426	self.cstA = fittings.Parameter(self.model, 'A', self.default_A)
427	self.cstB = fittings.Parameter(self.model, 'B', self.default_B)
428
429	if (self.xLabel.lower() == "log10(x)"):
430	tempdy = numpy.asarray(tempdy)
431	tempdy[tempdy == 0] = 1
432	chisqr, out, cov = fittings.sasfit(self.model,
433	[self.cstA, self.cstB],
434	tempx, tempy,
435	tempdy,
436	math.log10(xmin),
437	math.log10(xmax))
438	else:
439	tempdy = numpy.asarray(tempdy)
440	tempdy[tempdy == 0] = 1
441	chisqr, out, cov = fittings.sasfit(self.model,
442	[self.cstA, self.cstB],
443	tempx, tempy, tempdy,
444	xminView, xmaxView)
445	# Use chi2/dof
446	if len(tempx) > 0:
447	chisqr = chisqr/len(tempx)
448
449	#Check that cov and out are iterable before displaying them
450	if cov == None:
451	errA = 0.0
452	errB = 0.0
453	else:
454	errA = math.sqrt(cov[0][0])
455	errB = math.sqrt(cov[1][1])
456	if out == None:
457	cstA = 0.0
458	cstB = 0.0
459	else:
460	cstA = out[0]
461	cstB = out[1]
462	# Reset model with the right values of A and B
463	self.model.setParam('A', float(cstA))
464	self.model.setParam('B', float(cstB))
465
466	tempx = []
467	tempy = []
468	y_model = 0.0
469	# load tempy with the minimum transformation
470
471	if self.xLabel == "log10(x)":
472	y_model = self.model.run(math.log10(xmin))
473	tempx.append(xmin)
474	else:
475	y_model = self.model.run(xminView)
476	tempx.append(xminView)
477
478	if self.yLabel == "log10(y)":
479	tempy.append(math.pow(10, y_model))
480	else:
481	tempy.append(y_model)
482
483	# load tempy with the maximum transformation
484	if self.xLabel == "log10(x)":
485	y_model = self.model.run(math.log10(xmax))
486	tempx.append(xmax)
487	else:
488	y_model = self.model.run(xmaxView)
489	tempx.append(xmaxView)
490
491	if self.yLabel == "log10(y)":
492	tempy.append(math.pow(10, y_model))
493	else:
494	tempy.append(y_model)
495	#Set the fit parameter display when FitDialog is opened again
496	self.Avalue = cstB
497	self.Bvalue = cstA
498	self.ErrAvalue = errA
499	self.ErrBvalue = errB
500	self.Chivalue = chisqr
501	self.push_data(tempx, tempy, xminView, xmaxView,
502	xmin, xmax, self._ongetValues())
503
504	# Display the fitting value on the Fit Dialog
505	self._onsetValues(cstB, cstA, errA, errB, chisqr)
506
507	def _onsetValues(self, cstA, cstB, errA, errB, Chi):
508	"""
509	Display the value on fit Dialog
510	"""
511	rg = None
512	self.tcA.SetValue(format_number(cstA))
513	self.tcB.SetValue(format_number(cstB))
514	self.tcErrA.SetValue(format_number(errA))
515	self.tcErrB.SetValue(format_number(errB))
516	self.tcChi.SetValue(format_number(Chi))
517	if self.rg_on:
518	if self.Rg_tctr.IsShown():
519	rg = numpy.sqrt(-3 * float(cstA))
520	value = format_number(rg)
521	self.Rg_tctr.SetValue(value)
522	if self.I0_tctr.IsShown():
523	val = numpy.exp(cstB)
524	self.I0_tctr.SetValue(format_number(val))
525	if self.Rgerr_tctr.IsShown():
526	if rg != None and rg != 0:
527	value = format_number(3 * float(cstA) / (2 * rg))
528	else:
529	value =''
530	self.Rgerr_tctr.SetValue(value)
531	if self.I0err_tctr.IsShown():
532	val = numpy.abs(numpy.exp(cstB) - numpy.exp(cstB + errB))
533	self.I0err_tctr.SetValue(format_number(val))
534	if self.Diameter_tctr.IsShown():
535	rg = 4 * numpy.sqrt(-float(cstA))
536	value = format_number(rg)
537	self.Diameter_tctr.SetValue(value)
538	if self.Diametererr_tctr.IsShown():
539	if rg != None and rg != 0:
540	value = format_number(8 * float(cstA) / rg)
541	else:
542	value =''
543	self.Diametererr_tctr.SetValue(value)
544	if self.RgQmin_tctr.IsShown():
545	value = format_number(rg * self.mini)
546	self.RgQmin_tctr.SetValue(value)
547	if self.RgQmax_tctr.IsShown():
548	value = format_number(rg * self.maxi)
549	self.RgQmax_tctr.SetValue(value)
550
551	def _ongetValues(self):
552	"""
553	Display the value on fit Dialog
554	"""
555	return self.Avalue, self.Bvalue, self.ErrAvalue, \
556	self.ErrBvalue, self.Chivalue
557
558	def _checkVal(self, usermin, usermax):
559	"""
560	Ensure that fields parameter contains a min and a max value
561	within x min and x max range
562	"""
563	self.mini = float(self.xminFit.GetValue())
564	self.maxi = float(self.xmaxFit.GetValue())
565	flag = True
566	try:
567	mini = float(usermin.GetValue())
568	maxi = float(usermax.GetValue())
569	if mini < maxi:
570	usermin.SetBackgroundColour(wx.WHITE)
571	usermin.Refresh()
572	else:
573	flag = False
574	usermin.SetBackgroundColour("pink")
575	usermin.Refresh()
576	except:
577	# Check for possible values entered
578	flag = False
579	usermin.SetBackgroundColour("pink")
580	usermin.Refresh()
581
582	return flag
583
584	def floatForwardTransform(self, x):
585	"""
586	transform a float.
587	"""
588	#TODO: refactor this with proper object-oriented design
589	# This code stinks.
590	if(self.xLabel == "x"):
591	return transform.toX(x)
592	if(self.xLabel == "x^(2)"):
593	return transform.toX2(x)
594	if(self.xLabel == "ln(x)"):
595	return transform.toLogX(x)
596	if(self.xLabel == "log10(x)"):
597	return math.log10(x)
598
599	def floatTransform(self, x):
600	"""
601	transform a float.It is use to determine the x.
602	View min and x.View max for values
603	not in x
604	"""
605	#TODO: refactor this with proper object-oriented design
606	# This code stinks.
607	if(self.xLabel == "x"):
608	return transform.toX(x)
609	if(self.xLabel == "x^(2)"):
610	return transform.toX2(x)
611	if(self.xLabel == "ln(x)"):
612	return transform.toLogX(x)
613	if(self.xLabel == "log10(x)"):
614	if x > 0:
615	return x
616	else:
617	raise ValueError, "cannot compute log of a negative number"
618
619	def floatInvTransform(self, x):
620	"""
621	transform a float.It is use to determine the x.View min and x.View
622	max for values not in x
623
624	"""
625	#TODO: refactor this. This is just a hack to make the
626	# functionality work without rewritting the whole code
627	# with good design (which really should be done...).
628	if(self.xLabel == "x^(2)"):
629	return math.sqrt(x)
630
631	elif(self.xLabel == "log10(x)"):
632	return math.pow(10, x)
633
634	elif(self.xLabel == "ln(x)"):
635	return math.exp(x)
636	return x
637
638	def checkFitValues(self, item):
639	"""
640	Check the validity of input values
641	"""
642	flag = True
643	value = item.GetValue()
644	# Check for possible values entered
645	if (self.xLabel == "log10(x)"): #or self.xLabel=="ln(x)"):
646	if(float(value) > 0):
647	item.SetBackgroundColour(wx.WHITE)
648	item.Refresh()
649	else:
650	flag = False
651	item.SetBackgroundColour("pink")
652	item.Refresh()
653	return flag
654
655	def setFitRange(self, xmin, xmax, xminTrans, xmaxTrans):
656	"""
657	Set fit parameters
658	"""
659	self.xminFit.SetValue(format_number(xmin))
660	self.xmaxFit.SetValue(format_number(xmax))
661
662	def set_fit_region(self, xmin, xmax):
663	"""
664	Set the fit region
665	:param xmin: minimum x-value to be included in fit
666	:param xmax: maximum x-value to be included in fit
667	"""
668	# Check values
669	try:
670	float(xmin)
671	float(xmax)
672	except:
673	msg = "LinearFit.set_fit_region: fit range must be floats"
674	raise ValueError, msg
675	self.xminFit.SetValue(format_number(xmin))
676	self.xmaxFit.SetValue(format_number(xmax))
677
678
679	class MyApp(wx.App):
680	"""
681	"""
682	def OnInit(self):
683	"""
684	"""
685	wx.InitAllImageHandlers()
686	plot = Theory1D([], [])
687	dialog = LinearFit(parent=None, plottable=plot,
688	push_data=self.onFitDisplay,
689	transform=self.returnTrans,
690	title='Linear Fit')
691	if dialog.ShowModal() == wx.ID_OK:
692	pass
693	dialog.Destroy()
694	return 1
695
696	def onFitDisplay(self, tempx, tempy, xminView, xmaxView, xmin, xmax, func):
697	"""
698	"""
699	pass
700
701	def returnTrans(self):
702	"""
703	"""
704	return '', '', 0, 0, 0, 0, 0

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

SasView

source: sasview/src/sas/plottools/fitDialog.py @ 54b9ad2

Download in other formats: