fitDialog.py @ e46cbc5

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

Rev	Line
[a9d5684]	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
[b9a5f0e]	432	chisqr, out, cov = fittings.sasfit(self.model,
[a9d5684]	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
[b9a5f0e]	441	chisqr, out, cov = fittings.sasfit(self.model,
[a9d5684]	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 @ e46cbc5

Download in other formats: