fitDialog.py @ be2775f

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.2.2ticket-1009ticket-1094-headlessticket-1242-2d-resolutionticket-1243ticket-1249ticket885unittest-saveload

Last change on this file since be2775f was d7bb526, checked in by Piotr Rozyczko <piotr.rozyczko@…>, 9 years ago
Refactored plottools into sasgui
Property mode set to `100644`
File size: 27.4 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	iy += 2
155	ix = 1
156	sizer.Add(wx.StaticText(self, -1, 'Min'), (iy, ix), (1, 1),
157	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
158	ix += 2
159	sizer.Add(wx.StaticText(self, -1, 'Max'), (iy, ix),
160	(1, 1), wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
161
162	iy += 1
163	ix = 0
164	sizer.Add(wx.StaticText(self, -1, 'Maximum range (linear scale)'),
165	(iy, ix), (1, 1),
166	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 15)
167	ix += 1
168	sizer.Add(self.initXmin, (iy, ix), (1, 1),
169	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
170	ix += 2
171	sizer.Add(self.initXmax, (iy, ix), (1, 1),
172	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
173
174	iy += 1
175	ix = 0
176	sizer.Add(wx.StaticText(self, -1, 'Fit range of ' + self.xLabel),
177	(iy, ix), (1, 1),
178	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 15)
179	ix += 1
180	sizer.Add(self.xminFit, (iy, ix), (1, 1),
181	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
182	ix += 2
183	sizer.Add(self.xmaxFit, (iy, ix), (1, 1), wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
184	if self.rg_on:
185	self.SetSize((PNL_WIDTH, PNL_HEIGHT))
186	I0_stxt = wx.StaticText(self, -1, 'I(q=0)')
187	self.I0_tctr = wx.TextCtrl(self, -1, '')
188	self.I0_tctr.SetEditable(False)
189	self.I0_tctr.SetBackgroundColour(_BACKGROUND_COLOR)
190	self.I0err_tctr = wx.TextCtrl(self, -1, '')
191	self.I0err_tctr.SetEditable(False)
192	self.I0err_tctr.SetBackgroundColour(_BACKGROUND_COLOR)
193	Rg_stxt = wx.StaticText(self, -1, 'Rg [A]')
194	Rg_stxt.Show(self.yLabel == "ln(y)")
195	self.Rg_tctr = wx.TextCtrl(self, -1, '')
196	self.Rg_tctr.SetEditable(False)
197	self.Rg_tctr.SetBackgroundColour(_BACKGROUND_COLOR)
198	self.Rg_tctr.Show(self.yLabel == "ln(y)")
199	self.Rgerr_tctr = wx.TextCtrl(self, -1, '')
200	self.Rgerr_tctr.SetEditable(False)
201	self.Rgerr_tctr.SetBackgroundColour(_BACKGROUND_COLOR)
202	self.Rgerr_tctr.Show(self.yLabel == "ln(y)")
203	self.Rgerr_pm = wx.StaticText(self, -1, '+/-')
204	self.Rgerr_pm.Show(self.yLabel == "ln(y)")
205	Diameter_stxt = wx.StaticText(self, -1, 'Rod Diameter [A]')
206	Diameter_stxt.Show(self.yLabel == "ln(y*x)")
207	self.Diameter_tctr = wx.TextCtrl(self, -1, '')
208	self.Diameter_tctr.SetEditable(False)
209	self.Diameter_tctr.SetBackgroundColour(_BACKGROUND_COLOR)
210	self.Diameter_tctr.Show(self.yLabel == "ln(y*x)")
211	self.Diameter_pm = wx.StaticText(self, -1, '+/-')
212	self.Diameter_pm.Show(self.yLabel == "ln(y*x)")
213	self.Diametererr_tctr = wx.TextCtrl(self, -1, '')
214	self.Diametererr_tctr.SetEditable(False)
215	self.Diametererr_tctr.SetBackgroundColour(_BACKGROUND_COLOR)
216	self.Diametererr_tctr.Show(self.yLabel == "ln(y*x)")
217	RgQmin_stxt = wx.StaticText(self, -1, 'Rg*Qmin')
218	self.RgQmin_tctr = wx.TextCtrl(self, -1, '')
219	self.RgQmin_tctr.SetEditable(False)
220	self.RgQmin_tctr.SetBackgroundColour(_BACKGROUND_COLOR)
221	RgQmax_stxt = wx.StaticText(self, -1, 'Rg*Qmax')
222	self.RgQmax_tctr = wx.TextCtrl(self, -1, '')
223	self.RgQmax_tctr.SetEditable(False)
224	self.RgQmax_tctr.SetBackgroundColour(_BACKGROUND_COLOR)
225
226	iy += 2
227	ix = 0
228	sizer.Add(I0_stxt, (iy, ix), (1, 1),
229	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 15)
230	ix += 1
231	sizer.Add(self.I0_tctr, (iy, ix), (1, 1),
232	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
233	ix += 1
234	sizer.Add(wx.StaticText(self, -1, '+/-'), (iy, ix),
235	(1, 1), wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
236	ix += 1
237	sizer.Add(self.I0err_tctr, (iy, ix), (1, 1),
238	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
239
240	iy += 1
241	ix = 0
242	sizer.Add(Rg_stxt, (iy, ix), (1, 1),
243	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 15)
244	ix += 1
245	sizer.Add(self.Rg_tctr, (iy, ix), (1, 1),
246	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
247
248	ix += 1
249	sizer.Add(self.Rgerr_pm, (iy, ix),
250	(1, 1), wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
251	ix += 1
252	sizer.Add(self.Rgerr_tctr, (iy, ix), (1, 1),
253	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
254	iy += 1
255	ix = 0
256	sizer.Add(Diameter_stxt, (iy, ix), (1, 1),
257	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 15)
258	ix += 1
259	sizer.Add(self.Diameter_tctr, (iy, ix), (1, 1),
260	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
261
262	ix += 1
263	sizer.Add(self.Diameter_pm, (iy, ix),
264	(1, 1), wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
265	ix += 1
266	sizer.Add(self.Diametererr_tctr, (iy, ix), (1, 1),
267	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
268	iy += 1
269	ix = 0
270	sizer.Add(RgQmin_stxt, (iy, ix), (1, 1),
271	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 15)
272	ix += 1
273	sizer.Add(self.RgQmin_tctr, (iy, ix), (1, 1),
274	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
275	iy += 1
276	ix = 0
277	sizer.Add(RgQmax_stxt, (iy, ix), (1, 1),
278	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 15)
279	ix += 1
280	sizer.Add(self.RgQmax_tctr, (iy, ix), (1, 1),
281	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
282
283	iy += 1
284	ix = 1
285
286	vbox.Add(self.static_line_1, 0, wx.EXPAND, 0)
287	sizer_button = wx.BoxSizer(wx.HORIZONTAL)
288	sizer_button.Add((20, 20), 1, wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
289	sizer_button.Add(self.btFit, 0, wx.LEFT \| wx.RIGHT \| wx.ADJUST_MINSIZE, 10)
290	sizer_button.Add(self.btClose, 0,
291	wx.LEFT \| wx.RIGHT \| wx.ADJUST_MINSIZE, 10)
292	vbox.Add(sizer_button, 0, wx.EXPAND \| wx.BOTTOM \| wx.TOP, 10)
293
294	sizer.Add(self.btFit, (iy, ix), (1, 1), wx.LEFT \| wx.ADJUST_MINSIZE, 0)
295	# panel.SetSizer(sizer)
296	self.SetSizer(vbox)
297	self.Centre()
298	# Receives the type of model for the fitting
299	from LineModel import LineModel
300	self.model = LineModel()
301	# Display the fittings values
302	self.default_A = self.model.getParam('A')
303	self.default_B = self.model.getParam('B')
304	self.cstA = fittings.Parameter(self.model, 'A', self.default_A)
305	self.cstB = fittings.Parameter(self.model, 'B', self.default_B)
306
307	# Set default value of parameter in fit dialog
308	if self.Avalue == None:
309	self.tcA.SetValue(format_number(self.default_A))
310	else:
311	self.tcA.SetLabel(format_number(self.Avalue))
312	if self.Bvalue == None:
313	self.tcB.SetValue(format_number(self.default_B))
314	else:
315	self.tcB.SetLabel(format_number(self.Bvalue))
316	if self.ErrAvalue == None:
317	self.tcErrA.SetLabel(format_number(0.0))
318	else:
319	self.tcErrA.SetLabel(format_number(self.ErrAvalue))
320	if self.ErrBvalue == None:
321	self.tcErrB.SetLabel(format_number(0.0))
322	else:
323	self.tcErrB.SetLabel(format_number(self.ErrBvalue))
324	if self.Chivalue == None:
325	self.tcChi.SetLabel(format_number(0.0))
326	else:
327	self.tcChi.SetLabel(format_number(self.Chivalue))
328	if self.plottable.x != []:
329	# store the values of View in self.x,self.y,self.dx,self.dy
330	self.x, self.y, self.dx, \
331	self.dy = self.plottable.returnValuesOfView()
332	try:
333	self.mini = self.floatForwardTransform(min(self.x))
334	except:
335	self.mini = "Invalid"
336	try:
337	self.maxi = self.floatForwardTransform(max(self.x))
338	except:
339	self.maxi = "Invalid"
340
341	self.initXmin.SetValue(format_number(min(self.plottable.x)))
342	self.initXmax.SetValue(format_number(max(self.plottable.x)))
343	self.mini = min(self.x)
344	self.maxi = max(self.x)
345	self.xminFit.SetValue(format_number(self.mini))
346	self.xmaxFit.SetValue(format_number(self.maxi))
347
348	def register_close(self, owner):
349	"""
350	Method to register the close event to a parent
351	window that needs notification when the dialog
352	is closed
353
354	:param owner: parent window
355
356	"""
357	self._registered_close = owner
358
359	def _on_close(self, event):
360	"""
361	Close event.
362	Notify registered owner if available.
363	"""
364	event.Skip()
365	if self._registered_close is not None:
366	self._registered_close()
367
368	def _onFit(self, event):
369	"""
370	Performs the fit. Receive an event when clicking on
371	the button Fit.Computes chisqr ,
372	A and B parameters of the best linear fit y=Ax +B
373	Push a plottable to
374	"""
375	tempx = []
376	tempy = []
377	tempdy = []
378
379	# Check if View contains a x array .we online fit when x exits
380	# makes transformation for y as a line to fit
381	if self.x != []:
382	if self.checkFitValues(self.xminFit) == True:
383	# Check if the field of Fit Dialog contain values
384	# and use the x max and min of the user
385	if not self._checkVal(self.xminFit, self.xmaxFit):
386	return
387	xminView = float(self.xminFit.GetValue())
388	xmaxView = float(self.xmaxFit.GetValue())
389	xmin = xminView
390	xmax = xmaxView
391	# Store the transformed values of view x, y,dy
392	# in variables before the fit
393	if self.yLabel.lower() == "log10(y)":
394	if self.xLabel.lower() == "log10(x)":
395	for i in range(len(self.x)):
396	if self.x[i] >= math.log10(xmin):
397	tempy.append(math.log10(self.y[i]))
398	tempdy.append(transform.errToLogX(self.y[i], 0, self.dy[i], 0))
399	else:
400	for i in range(len(self.y)):
401	tempy.append(math.log10(self.y[i]))
402	tempdy.append(transform.errToLogX(self.y[i], 0, self.dy[i], 0))
403	else:
404	tempy = self.y
405	tempdy = self.dy
406
407	if self.xLabel.lower() == "log10(x)":
408	for x_i in self.x:
409	if x_i >= math.log10(xmin):
410	tempx.append(math.log10(x_i))
411	else:
412	tempx = self.x
413
414	# Find the fitting parameters
415	# Always use the same defaults, so that fit history
416	# doesn't play a role!
417	self.cstA = fittings.Parameter(self.model, 'A', self.default_A)
418	self.cstB = fittings.Parameter(self.model, 'B', self.default_B)
419
420	if self.xLabel.lower() == "log10(x)":
421	tempdy = numpy.asarray(tempdy)
422	tempdy[tempdy == 0] = 1
423	chisqr, out, cov = fittings.sasfit(self.model,
424	[self.cstA, self.cstB],
425	tempx, tempy,
426	tempdy,
427	math.log10(xmin),
428	math.log10(xmax))
429	else:
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, tempdy,
435	xminView, xmaxView)
436	# Use chi2/dof
437	if len(tempx) > 0:
438	chisqr = chisqr / len(tempx)
439
440	# Check that cov and out are iterable before displaying them
441	if cov == None:
442	errA = 0.0
443	errB = 0.0
444	else:
445	errA = math.sqrt(cov[0][0])
446	errB = math.sqrt(cov[1][1])
447	if out == None:
448	cstA = 0.0
449	cstB = 0.0
450	else:
451	cstA = out[0]
452	cstB = out[1]
453	# Reset model with the right values of A and B
454	self.model.setParam('A', float(cstA))
455	self.model.setParam('B', float(cstB))
456
457	tempx = []
458	tempy = []
459	y_model = 0.0
460	# load tempy with the minimum transformation
461
462	if self.xLabel == "log10(x)":
463	y_model = self.model.run(math.log10(xmin))
464	tempx.append(xmin)
465	else:
466	y_model = self.model.run(xminView)
467	tempx.append(xminView)
468
469	if self.yLabel == "log10(y)":
470	tempy.append(math.pow(10, y_model))
471	else:
472	tempy.append(y_model)
473
474	# load tempy with the maximum transformation
475	if self.xLabel == "log10(x)":
476	y_model = self.model.run(math.log10(xmax))
477	tempx.append(xmax)
478	else:
479	y_model = self.model.run(xmaxView)
480	tempx.append(xmaxView)
481
482	if self.yLabel == "log10(y)":
483	tempy.append(math.pow(10, y_model))
484	else:
485	tempy.append(y_model)
486	# Set the fit parameter display when FitDialog is opened again
487	self.Avalue = cstB
488	self.Bvalue = cstA
489	self.ErrAvalue = errA
490	self.ErrBvalue = errB
491	self.Chivalue = chisqr
492	self.push_data(tempx, tempy, xminView, xmaxView,
493	xmin, xmax, self._ongetValues())
494
495	# Display the fitting value on the Fit Dialog
496	self._onsetValues(cstB, cstA, errA, errB, chisqr)
497
498	def _onsetValues(self, cstA, cstB, errA, errB, Chi):
499	"""
500	Display the value on fit Dialog
501	"""
502	rg = None
503	self.tcA.SetValue(format_number(cstA))
504	self.tcB.SetValue(format_number(cstB))
505	self.tcErrA.SetValue(format_number(errA))
506	self.tcErrB.SetValue(format_number(errB))
507	self.tcChi.SetValue(format_number(Chi))
508	if self.rg_on:
509	if self.Rg_tctr.IsShown():
510	rg = numpy.sqrt(-3 * float(cstA))
511	value = format_number(rg)
512	self.Rg_tctr.SetValue(value)
513	if self.I0_tctr.IsShown():
514	val = numpy.exp(cstB)
515	self.I0_tctr.SetValue(format_number(val))
516	if self.Rgerr_tctr.IsShown():
517	if rg != None and rg != 0:
518	value = format_number(3 * float(cstA) / (2 * rg))
519	else:
520	value = ''
521	self.Rgerr_tctr.SetValue(value)
522	if self.I0err_tctr.IsShown():
523	val = numpy.abs(numpy.exp(cstB) - numpy.exp(cstB + errB))
524	self.I0err_tctr.SetValue(format_number(val))
525	if self.Diameter_tctr.IsShown():
526	rg = 4 * numpy.sqrt(-float(cstA))
527	value = format_number(rg)
528	self.Diameter_tctr.SetValue(value)
529	if self.Diametererr_tctr.IsShown():
530	if rg != None and rg != 0:
531	value = format_number(8 * float(cstA) / rg)
532	else:
533	value = ''
534	self.Diametererr_tctr.SetValue(value)
535	if self.RgQmin_tctr.IsShown():
536	value = format_number(rg * self.mini)
537	self.RgQmin_tctr.SetValue(value)
538	if self.RgQmax_tctr.IsShown():
539	value = format_number(rg * self.maxi)
540	self.RgQmax_tctr.SetValue(value)
541
542	def _ongetValues(self):
543	"""
544	Display the value on fit Dialog
545	"""
546	return self.Avalue, self.Bvalue, self.ErrAvalue, \
547	self.ErrBvalue, self.Chivalue
548
549	def _checkVal(self, usermin, usermax):
550	"""
551	Ensure that fields parameter contains a min and a max value
552	within x min and x max range
553	"""
554	self.mini = float(self.xminFit.GetValue())
555	self.maxi = float(self.xmaxFit.GetValue())
556	flag = True
557	try:
558	mini = float(usermin.GetValue())
559	maxi = float(usermax.GetValue())
560	if mini < maxi:
561	usermin.SetBackgroundColour(wx.WHITE)
562	usermin.Refresh()
563	else:
564	flag = False
565	usermin.SetBackgroundColour("pink")
566	usermin.Refresh()
567	except:
568	# Check for possible values entered
569	flag = False
570	usermin.SetBackgroundColour("pink")
571	usermin.Refresh()
572
573	return flag
574
575	def floatForwardTransform(self, x):
576	"""
577	transform a float.
578	"""
579	# TODO: refactor this with proper object-oriented design
580	# This code stinks.
581	if self.xLabel == "x":
582	return transform.toX(x)
583	if self.xLabel == "x^(2)":
584	return transform.toX2(x)
585	if self.xLabel == "ln(x)":
586	return transform.toLogX(x)
587	if self.xLabel == "log10(x)":
588	return math.log10(x)
589
590	def floatTransform(self, x):
591	"""
592	transform a float.It is use to determine the x.
593	View min and x.View max for values
594	not in x
595	"""
596	# TODO: refactor this with proper object-oriented design
597	# This code stinks.
598	if self.xLabel == "x":
599	return transform.toX(x)
600	if self.xLabel == "x^(2)":
601	return transform.toX2(x)
602	if self.xLabel == "ln(x)":
603	return transform.toLogX(x)
604	if self.xLabel == "log10(x)":
605	if x > 0:
606	return x
607	else:
608	raise ValueError, "cannot compute log of a negative number"
609
610	def floatInvTransform(self, x):
611	"""
612	transform a float.It is use to determine the x.View min and x.View
613	max for values not in x
614
615	"""
616	# TODO: refactor this. This is just a hack to make the
617	# functionality work without rewritting the whole code
618	# with good design (which really should be done...).
619	if self.xLabel == "x^(2)":
620	return math.sqrt(x)
621	elif self.xLabel == "log10(x)":
622	return math.pow(10, x)
623	elif self.xLabel == "ln(x)":
624	return math.exp(x)
625	return x
626
627	def checkFitValues(self, item):
628	"""
629	Check the validity of input values
630	"""
631	flag = True
632	value = item.GetValue()
633	# Check for possible values entered
634	if self.xLabel == "log10(x)":
635	if float(value) > 0:
636	item.SetBackgroundColour(wx.WHITE)
637	item.Refresh()
638	else:
639	flag = False
640	item.SetBackgroundColour("pink")
641	item.Refresh()
642	return flag
643
644	def setFitRange(self, xmin, xmax, xminTrans, xmaxTrans):
645	"""
646	Set fit parameters
647	"""
648	self.xminFit.SetValue(format_number(xmin))
649	self.xmaxFit.SetValue(format_number(xmax))
650
651	def set_fit_region(self, xmin, xmax):
652	"""
653	Set the fit region
654	:param xmin: minimum x-value to be included in fit
655	:param xmax: maximum x-value to be included in fit
656	"""
657	# Check values
658	try:
659	float(xmin)
660	float(xmax)
661	except:
662	msg = "LinearFit.set_fit_region: fit range must be floats"
663	raise ValueError, msg
664	self.xminFit.SetValue(format_number(xmin))
665	self.xmaxFit.SetValue(format_number(xmax))
666
667
668	class MyApp(wx.App):
669	"""
670	Test application
671	"""
672	def OnInit(self):
673	"""
674	Test application initialization
675	"""
676	wx.InitAllImageHandlers()
677	plot = Theory1D([], [])
678	dialog = LinearFit(parent=None, plottable=plot,
679	push_data=self.onFitDisplay,
680	transform=self.returnTrans,
681	title='Linear Fit')
682	if dialog.ShowModal() == wx.ID_OK:
683	pass
684	dialog.Destroy()
685	return 1
686
687	def onFitDisplay(self, tempx, tempy, xminView, xmaxView, xmin, xmax, func):
688	"""
689	Test application dummy method
690	"""
691	pass
692
693	def returnTrans(self):
694	"""
695	Test application dummy method
696	"""
697	return '', '', 0, 0, 0, 0, 0

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

SasView

source: sasview/src/sas/sasgui/plottools/fitDialog.py @ be2775f

Download in other formats: