fitpage.py @ 0e2aa40

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 0e2aa40 was 77e23a2, checked in by Gervaise Alina <gervyh@…>, 16 years ago
add fit with parameter range when selecting park only
Property mode set to `100644`
File size: 35.6 KB

Rev	Line
[c77d859]	1	import sys
	2	import wx
	3	import wx.lib.newevent
	4	import numpy
	5	import copy
	6	import math
	7	from sans.models.dispersion_models import ArrayDispersion, GaussianDispersion
	8
	9	from sans.guicomm.events import StatusEvent
	10	from sans.guiframe.utils import format_number
[77e23a2]	11
[c77d859]	12	(ModelEventbox, EVT_MODEL_BOX) = wx.lib.newevent.NewEvent()
[77e23a2]	13	(FitterTypeEvent, EVT_FITTER_TYPE) = wx.lib.newevent.NewEvent()
[c77d859]	14	_BOX_WIDTH = 80
	15
	16	import basepage
	17	from basepage import BasicPage
	18
	19
	20	class FitPage(BasicPage):
	21	"""
	22	FitPanel class contains fields allowing to display results when
	23	fitting a model and one data
	24	@note: For Fit to be performed the user should check at least one parameter
	25	on fit Panel window.
	26
	27	"""
	28	def __init__(self,parent, page_info, name=""):
	29	BasicPage.__init__(self, parent, page_info,name)
	30	"""
	31	Initialization of the Panel
	32	"""
[6f023e8]	33	self.name= self.data.name
[dd5949d]	34	page_info.page_name= self.data.name
[59a7f2d]	35
[c77d859]	36	## fit page does not content npts txtcrtl
	37	self.npts=None
	38	## if no dispersity parameters is avaible
	39	self.text_disp_1=None
	40
	41	self._fill_datainfo_sizer()
	42	self._fill_model_sizer( self.sizer1)
	43	self._fill_range_sizer()
[59a7f2d]	44	self.model = self.formfactorbox.GetClientData(0)()
	45	self.page_info.model= self.model
	46	self._on_select_model(event=None)
[77e23a2]	47	self.Bind(EVT_FITTER_TYPE,self._on_engine_change)
	48
	49
	50	def _on_engine_change(self, event):
	51	"""
	52	get an event containing the current name of the fit engine type
	53	@param event: FitterTypeEvent containing the name of the current engine
	54	"""
	55	if len(self.parameters)==0:
	56	return
	57	for item in self.parameters:
	58	if event.type =="scipy":
	59	item[5].SetValue(format_number(""))
	60	item[5].Hide()
	61	item[6].SetValue(format_number(""))
	62	item[6].Hide()
	63	self.text2_min.Hide()
	64	self.text2_max.Hide()
	65	else:
	66	item[5].Show(True)
	67	item[6].Show(True)
	68	self.text2_min.Show(True)
	69	self.text2_max.Show(True)
	70
	71	self.sizer3.Layout()
	72	self.SetScrollbars(20,20,200,100)
	73
[c77d859]	74
	75	def _on_display_description(self, event):
	76	"""
	77	Show or Hide description
	78	@param event: wx.EVT_RADIOBUTTON
	79	"""
	80	self._on_display_description_helper()
	81
	82
	83
	84	def _on_display_description_helper(self):
	85	"""
	86	Show or Hide description
	87	@param event: wx.EVT_RADIOBUTTON
	88	"""
	89	## save state of radiobox
	90	self.page_info. save_radiobox_state( self.description_hide )
	91	self.page_info. save_radiobox_state( self.description_show )
	92	## Show description
	93	if not self.description_show.GetValue():
	94	self.sizer_description.Clear(True)
	95
	96	else:
	97	model=self.page_info.model
	98	description=""
	99	if model!=None:
	100	description = self.page_info.model.description
	101	self.description = wx.StaticText( self,-1,str(description) )
	102	self.sizer_description.Add( self.description, 1, wx.EXPAND \| wx.ALL, 10 )
	103
	104	self.Layout()
	105
	106
	107	def _fill_range_sizer(self):
	108	"""
	109	Fill the sizer containing the plotting range
	110	add access to npts
	111	"""
[1ae3fe1]	112	sizer_fit = wx.GridSizer(1, 1,0, 0)
[c77d859]	113
	114	self.btFit = wx.Button(self,wx.NewId(),'Fit')
	115	self.btFit.Bind(wx.EVT_BUTTON, self._onFit,id= self.btFit.GetId())
	116	self.btFit.SetToolTipString("Perform fit.")
	117
[1ae3fe1]	118	#self.btStopFit = wx.Button(self,wx.NewId(),'Stop Fit')
[c77d859]	119	#self.btStopFit.Bind(wx.EVT_BUTTON, self.onStopFit,id= self.btStopFit.GetId())
[1ae3fe1]	120	#self.btStopFit.SetToolTipString("Stop Current fit job.")
[c77d859]	121	#self.btStopFit.Hide()
	122
	123	sizer_fit.Add((5,5),1, wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 5)
[1ae3fe1]	124	sizer_fit.Add(self.btFit,0, wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 5)
	125	#sizer_fit.Add(self.btStopFit,1, wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 5)
[c77d859]	126
	127	sizer_smearer = wx.BoxSizer(wx.HORIZONTAL)
	128	#Filling the sizer containing instruments smearing info.
	129	self.disable_smearer = wx.RadioButton(self, -1, 'No', style=wx.RB_GROUP)
	130	self.enable_smearer = wx.RadioButton(self, -1, 'Yes')
	131	self.Bind(wx.EVT_RADIOBUTTON, self.onSmear, id=self.disable_smearer.GetId())
	132	self.Bind(wx.EVT_RADIOBUTTON, self.onSmear, id=self.enable_smearer.GetId())
	133
	134	sizer_smearer.Add(wx.StaticText(self,-1,'Instrument Smearing? '))
	135	sizer_smearer.Add((10, 10))
	136	sizer_smearer.Add( self.enable_smearer )
	137	sizer_smearer.Add((10,10))
	138	sizer_smearer.Add( self.disable_smearer )
	139
	140	self._set_range_sizer( title="Fitted Q Range",
	141	object1=sizer_smearer, object= sizer_fit)
	142
	143
	144	def _fill_datainfo_sizer(self):
	145	"""
	146	fill sizer 0 with data info
	147	"""
	148	self.sizer0.Clear(True)
	149	## no loaded data , don't fill the sizer
	150	if self.data== None:
	151	self.sizer0.Layout()
	152	return
	153
	154	box_description= wx.StaticBox(self, -1, 'Loaded Data')
	155	boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
	156	#----------------------------------------------------------
	157	sizer_data = wx.GridSizer(3, 3,5, 5)
	158	#Filling the sizer containing data related fields
	159	DataSource =wx.StaticText(self, -1,str(self.data.name))
	160
	161	sizer_data.Add(wx.StaticText(self, -1, 'Data Source Name : '))
	162	sizer_data.Add(DataSource )
	163	sizer_data.Add( (5,5) )
	164
	165	#---------sizer 2 draw--------------------------------
	166	#set maximum range for x in linear scale
	167	if not hasattr(self.data,"data"): #Display only for 1D data fit
	168	# Minimum value of data
	169	data_min = str(format_number(numpy.min(self.data.x)))
	170	# Maximum value of data
	171	data_max = str(format_number(numpy.max(self.data.x)))
	172	text4_3 = wx.StaticText(self, -1, 'Maximum Q Range(Linear)',
	173	style=wx.ALIGN_LEFT)
	174	sizer_data.Add( text4_3 )
	175	sizer_data.Add(wx.StaticText(self, -1, "Min : %s"%data_min))
	176	sizer_data.Add(wx.StaticText(self, -1, "Max : %s"%data_max))
	177
	178	else:
	179	radius_min= 0
	180	x= numpy.max(self.data.xmin, self.data.xmax)
	181	y= numpy.max(self.data.ymin, self.data.ymax)
	182	radius_max = math.sqrt(xx + yy)
	183	# Minimum value of data
	184	data_min = str(format_number(radius_min))
	185	# Maximum value of data
	186	data_max = str(format_number(radius_max))
	187	text4_3 = wx.StaticText(self, -1, 'Maximum Q Range',
	188	style=wx.ALIGN_LEFT)
	189	sizer_data.Add( text4_3 )
	190	sizer_data.Add(wx.StaticText(self, -1, "Min : %s"%data_min))
	191	sizer_data.Add(wx.StaticText(self, -1, "Max : %s"%data_max))
	192
	193	boxsizer1.Add(sizer_data)
	194	#------------------------------------------------------------
	195	self.sizer0.Add(boxsizer1,0, wx.EXPAND \| wx.ALL, 10)
	196	self.sizer0.Layout()
	197
	198	self.qmin_x= data_min
	199	self.qmax_x= data_max
	200
[376916c]	201
[c77d859]	202	def _fill_model_sizer(self, sizer):
	203	"""
	204	fill sizer containing model info
	205	"""
	206	sizer_tcChi = wx.GridSizer(2, 2,5, 5)
[6355f4d]	207	self.tcChi = wx.StaticText(self, -1, "-", style=wx.ALIGN_LEFT)
	208
[c77d859]	209
	210	self.text1_1 = wx.StaticText(self, -1, 'Chi2/dof', style=wx.ALIGN_LEFT)
[6355f4d]	211
[c77d859]	212
	213	self.btChi = wx.Button(self,wx.NewId(),'View Chi')
	214	self.btChi.Bind(wx.EVT_BUTTON, self._onFit,id= self.btChi.GetId())
	215	self.btChi.SetToolTipString("View data - model.")
	216
	217	sizer_tcChi.Add(self.text1_1,1, wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 5)
	218	sizer_tcChi.Add(self.btChi,1, wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 5)
	219	sizer_tcChi.Add(self.tcChi,1, wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 5)
	220	sizer_tcChi.Add((5,5),1, wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 5)
	221
	222	## class base method to add view 2d button
	223	self._set_model_sizer(sizer=sizer, title="Model",object= sizer_tcChi )
	224
	225
	226	def _set_sizer_gaussian(self):
	227	"""
	228	draw sizer with gaussian dispersity parameters
	229	"""
	230	self.fittable_param=[]
	231	self.fixed_param=[]
[376916c]	232
[c77d859]	233	self.sizer4_4.Clear(True)
	234
	235	if self.model==None:
	236	##no model is selected
	237	return
	238	if not self.enable_disp.GetValue():
	239	## the user didn't select dispersity display
	240	return
[376916c]	241
	242	self._reset_dispersity()
	243	# Create the dispersion objects
	244	for item in self.model.dispersion.keys():
	245	disp_model = GaussianDispersion()
	246	self._disp_obj_dict[item] = disp_model
	247	self.model.set_dispersion(item, disp_model)
	248
[c77d859]	249	ix=0
	250	iy=1
	251	disp = wx.StaticText(self, -1, 'Names')
	252	self.sizer4_4.Add(disp,( iy, ix),(1,1),
	253	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	254	ix += 1
	255	values = wx.StaticText(self, -1, 'Values')
	256	self.sizer4_4.Add(values,( iy, ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	257	ix +=2
	258	self.text_disp_1 = wx.StaticText(self, -1, 'Errors')
	259	self.sizer4_4.Add( self.text_disp_1,(iy, ix),(1,1),\
	260	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	261	self.text_disp_1.Hide()
	262	ix += 1
	263	npts = wx.StaticText(self, -1, 'Npts')
	264	self.sizer4_4.Add(npts,( iy, ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	265	ix += 1
	266	nsigmas = wx.StaticText(self, -1, 'Nsigmas')
	267	self.sizer4_4.Add(nsigmas,( iy, ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	268
	269	for item in self.model.dispersion.keys():
[376916c]	270	self.disp_cb_dict[item]= None
[c77d859]	271	name1=item+".width"
	272	name2=item+".npts"
	273	name3=item+".nsigmas"
	274	iy += 1
	275	for p in self.model.dispersion[item].keys():
	276
	277	if p=="width":
	278	ix = 0
	279	cb = wx.CheckBox(self, -1, name1, (10, 10))
	280	wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
	281
	282	self.sizer4_4.Add( cb,( iy, ix),(1,1),
	283	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	284	ix = 1
	285	value= self.model.getParam(name1)
	286	ctl1 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20),
	287	style=wx.TE_PROCESS_ENTER)
	288
	289	ctl1.SetValue(str (format_number(value)))
[1328e03]	290	ctl1.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
[77e23a2]	291	ctl1.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
[c77d859]	292	ctl1.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
	293	self.sizer4_4.Add(ctl1, (iy,ix),(1,1),wx.EXPAND)
	294
	295	## text to show error sign
	296	ix = 2
	297	text2=wx.StaticText(self, -1, '+/-')
	298	self.sizer4_4.Add(text2,(iy, ix),(1,1),
	299	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	300	text2.Hide()
	301	## txtcrtl to add error from fit
	302	ix = 3
	303	ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
	304	self.sizer4_4.Add(ctl2, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	305	ctl2.Hide()
	306
	307	self.fittable_param.append([cb,name1,ctl1,text2,
	308	ctl2, None, None,None])
	309	elif p=="npts":
	310	ix = 4
	311	value= self.model.getParam(name2)
	312	Tctl = wx.TextCtrl(self, -1, size=(_BOX_WIDTH/2,20),
	313	style=wx.TE_PROCESS_ENTER)
	314
	315	Tctl.SetValue(str (format_number(value)))
[1328e03]	316	Tctl.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
[77e23a2]	317	Tctl.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
[c77d859]	318	Tctl.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
	319	self.sizer4_4.Add(Tctl, (iy,ix),(1,1),
	320	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	321
	322	self.fixed_param.append([None,name2, Tctl,None,None,
	323	None, None,None])
	324
	325	elif p=="nsigmas":
	326	ix = 5
	327	value= self.model.getParam(name3)
	328	Tctl = wx.TextCtrl(self, -1, size=(_BOX_WIDTH/2,20),
	329	style=wx.TE_PROCESS_ENTER)
	330	Tctl.SetValue(str (format_number(value)))
[1328e03]	331	Tctl.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
[77e23a2]	332	Tctl.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
[c77d859]	333	Tctl.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
	334	self.sizer4_4.Add(Tctl, (iy,ix),(1,1),
	335	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	336	ix +=1
	337	self.sizer4_4.Add((20,20), (iy,ix),(1,1),
	338	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	339
	340	self.fixed_param.append([None,name3, Tctl
	341	,None,None, None, None,None])
	342
	343	wx.PostEvent(self.parent, StatusEvent(status=\
	344	" Selected Distribution: Gaussian"))
	345	ix =0
	346	iy +=1
	347	self.sizer4_4.Add((20,20),(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	348	self.sizer4_4.Layout()
	349	self.sizer4.Layout()
	350	self.SetScrollbars(20,20,200,100)
[0a518e4c]	351
[c77d859]	352
	353	def _onFit(self, event):
	354	"""
	355	Allow to fit
	356	"""
[1ae3fe1]	357	#self.btFit.SetLabel("Stop")
[c77d859]	358	from sans.guiframe.utils import check_value
	359	flag = check_value( self.qmin, self.qmax)
	360
	361	if not flag:
	362	msg= "Fitting range invalid"
	363	wx.PostEvent(self.parent.parent, StatusEvent(status= msg ))
	364	return
	365
	366	if len(self.param_toFit) <= 0:
	367	msg= "Select at least one parameter to fit"
	368	wx.PostEvent(self.parent.parent, StatusEvent(status= msg ))
	369	return
	370
	371	self.qmin_x=float(self.qmin.GetValue())
	372	self.qmax_x =float( self.qmax.GetValue())
[6f023e8]	373	self.manager._reset_schedule_problem( value=0)
[ca7a626]	374	self.manager.schedule_for_fit( value=1,page=self,fitproblem =None)
	375	self.manager.set_fit_range(page= self,qmin= self.qmin_x, qmax= self.qmax_x)
[c77d859]	376	#single fit
[ca7a626]	377	self.manager.onFit()
[c77d859]	378
	379	self.sizer5.Layout()
	380	self.SetScrollbars(20,20,55,40)
	381
	382
	383	def _on_select_model(self, event):
	384	"""
	385	call back for model selection
	386	"""
[59a7f2d]	387	self._on_select_model_helper()
[c77d859]	388	self.set_model_param_sizer(self.model)
	389	self._set_sizer_gaussian()
	390
	391	evt = ModelEventbox(model=self.model)
	392	wx.PostEvent(self.event_owner, evt)
	393	try:
	394	self.compute_chisqr()
	395	self.text1_1.Show()
	396	self.tcChi.Show()
	397	except:
	398	## error occured on chisqr computation
	399	pass
[59a7f2d]	400
[dd5949d]	401	def _onparamEnter(self,event):
	402	"""
	403	when enter value on panel redraw model according to changed
	404	"""
	405	self._onparamEnter_helper()
	406	self.compute_chisqr()
	407
[2140e68]	408	def get_range(self):
	409	"""
	410	return the fitting range
	411	"""
[c9a4377]	412	return float(self.qmin_x) , float(self.qmax_x)
[c77d859]	413
	414	def get_param_list(self):
	415	"""
	416	@return self.param_toFit: list containing references to TextCtrl
	417	checked.Theses TextCtrl will allow reference to parameters to fit.
	418	@raise: if return an empty list of parameter fit will nnote work
	419	properly so raise ValueError,"missing parameter to fit"
	420	"""
	421	if self.param_toFit !=[]:
	422	return self.param_toFit
	423	else:
	424	raise ValueError,"missing parameter to fit"
	425
	426	def onsetValues(self,chisqr, out,cov):
	427	"""
	428	Build the panel from the fit result
	429	@param chisqr:Value of the goodness of fit metric
	430	@param out:list of parameter with the best value found during fitting
	431	@param cov:Covariance matrix
	432
	433	"""
	434	self.tcChi.SetLabel(format_number(chisqr))
	435	params = {}
	436	is_modified = False
	437	has_error = False
[0a518e4c]	438	self.text2_3.Hide()
	439	if self.text_disp_1 !=None:
	440	self.text_disp_1.Hide()
[c77d859]	441	#set the panel when fit result are float not list
	442	if out.__class__==numpy.float64:
	443	self.param_toFit[0][2].SetValue(format_number(out))
	444	self.param_toFit[0][2].Refresh()
[0a518e4c]	445
	446	self.param_toFit[0][4].Clear()
	447	self.param_toFit[0][4].Hide()
[c77d859]	448	if cov !=None :
	449	self.text2_3.Show(True)
[0a518e4c]	450	if self.text_disp_1 !=None:
[c77d859]	451	self.text_disp_1.Show(True)
	452
	453	self.param_toFit[0][3].Show(True)
	454	self.param_toFit[0][4].Clear()
	455	self.param_toFit[0][4].SetValue(format_number(cov[0]))
	456	self.param_toFit[0][4].Show(True)
	457	else:
	458	i=0
	459	j=0
	460	#Set the panel when fit result are list
	461	for item in self.param_toFit:
[0a518e4c]	462	## reset error value to initial state
	463	item[4].Clear()
	464	item[4].Hide()
	465	item[4].Refresh()
[c77d859]	466	if( out != None ) and len(out)<=len(self.param_toFit)and i < len(out):
	467	item[2].SetValue(format_number(self.model.getParam(item[1])))
	468	item[2].Refresh()
	469	if(cov !=None)and len(cov)<=len(self.param_toFit)and i < len(cov):
	470	self.text2_3.Show(True)
	471	if self.text_disp_1!=None:
	472	self.text_disp_1.Show(True)
	473	item[3].Show(True)
	474	item[4].Clear()
	475	for j in range(len(out)):
	476	if out[j]==self.model.getParam(item[1]):
	477	break
	478	item[4].SetValue(format_number(cov[j]))
	479	item[4].Refresh()
	480	item[4].Show(True)
	481	i+=1
	482
	483	self.sizer3.Layout()
	484	self.sizer4.Layout()
	485	self.SetScrollbars(20,20,200,100)
	486
	487
	488
	489	def onSmear(self, event):
	490	"""
	491	Create a smear object that will change the way residuals
	492	are compute when fitting
	493	"""
	494	smear =None
	495	msg=""
	496	if self.enable_smearer.GetValue():
	497	from DataLoader.qsmearing import smear_selection
[2140e68]	498	smear = smear_selection( self.data )
[c77d859]	499	if hasattr(self.data,"dxl"):
	500	msg= ": Resolution smearing parameters"
	501	if hasattr(self.data,"dxw"):
	502	msg= ": Slit smearing parameters"
	503	if smear ==None:
	504	wx.PostEvent(self.manager.parent, StatusEvent(status=\
	505	"Data contains no smearing information"))
	506	else:
	507	wx.PostEvent(self.manager.parent, StatusEvent(status=\
	508	"Data contains smearing information %s"%msg))
[2140e68]	509	self.manager.set_smearer(smear, qmin= self.qmin_x, qmax= self.qmax_x)
[c77d859]	510
[dd5949d]	511
[c77d859]	512	def compute_chisqr2D(self):
	513	"""
	514	compute chi square given a model and data 2D and set the value
	515	to the tcChi txtcrl
	516	"""
	517	from sans.guiframe.utils import check_value
	518	flag = check_value( self.qmin, self.qmax)
	519
	520	res=[]
	521	if flag== True:
	522	try:
	523	self.qmin_x = float(self.qmin.GetValue())
	524	self.qmax_x = float(self.qmax.GetValue())
	525	for i in range(len(self.data.x_bins)):
	526	for j in range(len(self.data.y_bins)):
	527	#Check the range containing data between self.qmin_x and self.qmax_x
	528	if math.pow(self.data.x_bins[i],2)+math.pow(self.data.y_bins[j],2)>=math.pow(self.qmin_x,2):
	529	if math.pow(self.data.x_bins[i],2)+math.pow(self.data.y_bins[j],2)<=math.pow(self.qmax_x,2):
	530	chisqrji=(self.data.data[j][i]- self.model.runXY(\
[9619367]	531	[self.data.x_bins[i],self.data.y_bins[j]]))\
[c77d859]	532	/self.data.err_data[j][i]
	533	#Vector containing residuals
	534	res.append( math.pow(chisqrji,2) )
	535	# compute sum of residual
	536	sum=0
	537	for item in res:
	538	if numpy.isfinite(item):
	539	sum +=item
	540	self.tcChi.SetLabel(format_number(math.fabs(sum/ len(res))))
	541	except:
	542	wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
	543	"Chisqr cannot be compute: %s"% sys.exc_value))
	544	return
	545
	546
	547	def compute_chisqr(self):
	548	"""
	549	compute chi square given a model and data 1D and set the value
	550	to the tcChi txtcrl
	551	"""
	552	from sans.guiframe.utils import check_value
	553	flag = check_value( self.qmin, self.qmax)
	554
	555	if flag== True:
	556	try:
	557	if hasattr(self.data,"data"):
	558	self.compute_chisqr2D()
	559	return
	560	else:
	561	self.qmin_x = float(self.qmin.GetValue())
	562	self.qmax_x = float(self.qmax.GetValue())
	563	# return residuals within self.qmin_x and self.qmax_x
	564	x,y,dy = [numpy.asarray(v) for v in (self.data.x,self.data.y,self.data.dy)]
	565	if self.qmin_x==None and self.qmax_x==None:
	566	fx =numpy.asarray([self.model.run(v) for v in x])
	567	temp=(y - fx)/dy
	568	res= temp*temp
	569	else:
	570	idx = (x>= self.qmin_x) & (x <=self.qmax_x)
	571	fx = numpy.asarray([self.model.run(item)for item in x[idx ]])
	572	temp=(y[idx] - fx)/dy[idx]
	573	res= temp*temp
	574	#sum of residuals
	575	sum=0
	576	for item in res:
	577	if numpy.isfinite(item):
	578	sum +=item
	579	self.tcChi.SetLabel(format_number(math.fabs(sum/ len(res))))
	580	except:
	581	wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
	582	"Chisqr cannot be compute: %s"% sys.exc_value))
	583	return
	584
	585
	586	def select_all_param(self,event):
	587	"""
	588	set to true or false all checkBox given the main checkbox value cb1
	589	"""
	590	self.param_toFit=[]
	591	if self.parameters !=[]:
	592	if self.cb1.GetValue()==True:
	593	for item in self.parameters:
	594	item[0].SetValue(True)
	595	self.param_toFit.append(item )
	596	if len(self.fittable_param)>0:
	597	for item in self.fittable_param:
	598	item[0].SetValue(True)
	599	self.param_toFit.append(item )
	600	else:
	601	for item in self.parameters:
	602	item[0].SetValue(False)
	603	for item in self.fittable_param:
	604	item[0].SetValue(False)
	605	self.param_toFit=[]
	606
	607
	608
	609	def select_param(self,event):
	610	"""
	611	Select TextCtrl checked for fitting purpose and stores them
	612	in self.param_toFit=[] list
	613	"""
	614	self.param_toFit=[]
	615	for item in self.parameters:
	616	#Select parameters to fit for list of primary parameters
	617	if item[0].GetValue()==True:
	618	if not (item in self.param_toFit):
	619	self.param_toFit.append(item )
	620	else:
	621	#remove parameters from the fitting list
	622	if item in self.param_toFit:
	623	self.param_toFit.remove(item)
	624	#Select parameters to fit for list of fittable parameters with dispersion
	625	for item in self.fittable_param:
	626	if item[0].GetValue()==True:
	627	if not (item in self.param_toFit):
	628	self.param_toFit.append(item)
	629	else:
	630	#remove parameters from the fitting list
	631	if item in self.param_toFit:
	632	self.param_toFit.remove(item)
	633	#Set the value of checkbox that selected every checkbox or not
	634	if len(self.parameters)+len(self.fittable_param) ==len(self.param_toFit):
	635	self.cb1.SetValue(True)
	636	else:
	637	self.cb1.SetValue(False)
	638
	639
	640	def set_model_description(self,description,sizer):
	641	"""
	642	fill a sizer with description
	643	@param description: of type string
	644	@param sizer: wx.BoxSizer()
	645	"""
	646
	647	sizer.Clear(True)
	648	box_description= wx.StaticBox(self, -1, 'Model Description')
	649	boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
	650
	651	self.description = wx.StaticText( self,-1,str(description) )
	652	sizer_selection=wx.BoxSizer(wx.HORIZONTAL)
	653
	654
	655	self.description_show = wx.RadioButton(self, -1, 'Show', style=wx.RB_GROUP)
	656	self.description_hide = wx.RadioButton(self, -1, 'Hide')
	657
	658	self.Bind( wx.EVT_RADIOBUTTON, self._on_display_description,
	659	id=self.description_hide.GetId() )
	660
	661	self.Bind( wx.EVT_RADIOBUTTON, self._on_display_description,
	662	id=self.description_show.GetId() )
	663
	664	self.model_description = wx.Button(self, -1, "More Details")
	665
	666
	667	self.page_info.save_radiobox_state( self.description_hide )
	668	self.page_info.save_radiobox_state( self.description_show )
	669
	670	sizer_selection.Add( self.description_show )
	671	sizer_selection.Add( (20,20))
	672	sizer_selection.Add( self.description_hide )
	673	sizer_selection.Add( (20,20))
	674	sizer_selection.Add( self.model_description )
	675
	676
	677	self.sizer_description=wx.BoxSizer(wx.HORIZONTAL)
	678	self.sizer_description.Add( self.description, 1, wx.EXPAND \| wx.ALL, 10 )
	679
	680	boxsizer1.Add( sizer_selection)
	681	boxsizer1.Add( (20,20))
	682	boxsizer1.Add( self.sizer_description)
	683
	684
	685	sizer.Add(boxsizer1,0, wx.EXPAND \| wx.ALL, 10)
	686	sizer.Layout()
	687
	688
	689
	690
[77e23a2]	691	def set_model_param_sizer(self, model):
[c77d859]	692	"""
	693	Build the panel from the model content
	694	@param model: the model selected in combo box for fitting purpose
	695	"""
	696	self.sizer3.Clear(True)
	697	self.parameters = []
	698	self.param_toFit=[]
	699	self.fittable_param=[]
	700	self.fixed_param=[]
	701
	702	if model ==None:
	703	self.sizer3.Layout()
	704	self.SetScrollbars(20,20,200,100)
	705	return
	706	box_description= wx.StaticBox(self, -1,str("Model Parameters"))
	707	boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
	708	sizer = wx.GridBagSizer(5,5)
	709	## save the current model
	710	self.model = model
	711
	712	keys = self.model.getParamList()
	713	#list of dispersion paramaters
	714	self.disp_list=self.model.getDispParamList()
	715
	716	keys.sort()
	717
	718	iy = 1
	719	ix = 0
	720	self.cb1 = wx.CheckBox(self, -1,"Select all", (10, 10))
	721	wx.EVT_CHECKBOX(self, self.cb1.GetId(), self.select_all_param)
	722	self.cb1.SetValue(False)
	723
	724	sizer.Add(self.cb1,(iy, ix),(1,1),\
	725	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	726	ix +=1
	727	self.text2_2 = wx.StaticText(self, -1, 'Values')
	728	sizer.Add(self.text2_2,(iy, ix),(1,1),\
	729	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	730	ix +=2
	731	self.text2_3 = wx.StaticText(self, -1, 'Errors')
	732	sizer.Add(self.text2_3,(iy, ix),(1,1),\
	733	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	734	self.text2_3.Hide()
	735	ix +=1
	736	self.text2_min = wx.StaticText(self, -1, 'Min')
	737	sizer.Add(self.text2_min,(iy, ix),(1,1),\
	738	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	739	self.text2_min.Hide()
	740	ix +=1
	741	self.text2_max = wx.StaticText(self, -1, 'Max')
	742	sizer.Add(self.text2_max,(iy, ix),(1,1),\
	743	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	744	self.text2_max.Hide()
	745	ix += 1
	746	self.text2_4 = wx.StaticText(self, -1, 'Units')
	747	sizer.Add(self.text2_4,(iy, ix),(1,1),\
	748	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	749	self.text2_4.Hide()
[77e23a2]	750
[c77d859]	751
	752	for item in keys:
	753	if not item in self.disp_list:
	754	iy += 1
	755	ix = 0
	756	## add parameters name with checkbox for selecting to fit
	757	cb = wx.CheckBox(self, -1, item )
	758	cb.SetValue(False)
	759	wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
	760	sizer.Add( cb,( iy, ix),(1,1),
	761	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	762	## add parameter value
	763	ix += 1
	764	value= self.model.getParam(item)
	765	ctl1 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20),
	766	style=wx.TE_PROCESS_ENTER)
	767
[77e23a2]	768	ctl1.SetValue(format_number(value))
	769	ctl1.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
[c77d859]	770	ctl1.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
	771	ctl1.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
	772	sizer.Add(ctl1, (iy,ix),(1,1), wx.EXPAND)
	773	## text to show error sign
	774	ix += 1
	775	text2=wx.StaticText(self, -1, '+/-')
	776	sizer.Add(text2,(iy, ix),(1,1),\
	777	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	778	text2.Hide()
	779	## txtcrtl to add error from fit
	780	ix += 1
	781	ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
	782	sizer.Add(ctl2, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	783	ctl2.Hide()
	784
[77e23a2]	785	param_min, param_max= self.model.details[item][1:]
[c77d859]	786	ix += 1
	787	ctl3 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER)
[77e23a2]	788	ctl3.SetValue(format_number(param_min))
	789	ctl3.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
[c77d859]	790	ctl3.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
	791	ctl3.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
	792	sizer.Add(ctl3, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	793	ctl3.Hide()
[77e23a2]	794
[c77d859]	795	ix += 1
	796	ctl4 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER)
[77e23a2]	797	ctl4.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
[c77d859]	798	ctl4.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
	799	ctl4.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
	800	sizer.Add(ctl4, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[77e23a2]	801	ctl4.SetValue(format_number(param_max))
[c77d859]	802	ctl4.Hide()
[77e23a2]	803
[c77d859]	804	ix +=1
	805	# Units
	806	try:
	807	units = wx.StaticText(self, -1, self.model.details[item][0], style=wx.ALIGN_LEFT)
	808	except:
	809	units = wx.StaticText(self, -1, "", style=wx.ALIGN_LEFT)
	810	sizer.Add(units, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	811
	812	##[cb state, name, value, "+/-", error of fit, min, max , units]
	813	self.parameters.append([cb,item, ctl1,
	814	text2,ctl2, ctl3, ctl4,None])
	815
	816	iy+=1
	817	sizer.Add((20,20),(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	818
	819	#Display units text on panel
	820	for item in keys:
	821	if self.model.details[item][0]!='':
	822	self.text2_4.Show()
	823	break
	824	else:
	825	self.text2_4.Hide()
	826
	827	boxsizer1.Add(sizer)
	828
	829	self.sizer3.Add(boxsizer1,0, wx.EXPAND \| wx.ALL, 10)
	830	self.sizer3.Layout()
	831	self.SetScrollbars(20,20,200,100)
	832
	833
	834
	835
	836	class HelpWindow(wx.Frame):
	837	def __init__(self, parent, id, title):
	838	wx.Frame.__init__(self, parent, id, title, size=(570, 400))
	839
	840	from sans.models.CylinderModel import CylinderModel
	841	model = CylinderModel()
	842	#from sans.models.LineModel import LineModel
	843	#model = LineModel()
	844	from danse.common.plottools.plottables import Data1D
	845	data= Data1D(x=[1,2], y=[3,4], dy=[0.1, 0,1])
	846
	847	from pageInfo import PageInfo
	848	myinfo = PageInfo(self, model, data=data )
	849
	850	## add data
	851
	852	from models import ModelList
	853	mylist= ModelList()
	854
	855	from sans.models.SphereModel import SphereModel
	856	from sans.models.SquareWellStructure import SquareWellStructure
	857	from sans.models.DebyeModel import DebyeModel
	858	from sans.models.LineModel import LineModel
	859	name= "shapes"
	860	list1= [SphereModel]
	861	mylist.set_list( name, list1)
	862
	863	name= "Shape-independent"
	864	list1= [DebyeModel]
	865	mylist.set_list( name, list1)
	866
	867	name= "Structure Factors"
	868	list1= [SquareWellStructure]
	869	mylist.set_list( name, list1)
	870
	871	name= "Added models"
	872	list1= [LineModel]
	873	mylist.set_list( name, list1)
	874
	875	myinfo.model_list_box = mylist.get_list()
	876
	877	self.page = FitPage(self, myinfo)
	878
	879
	880
	881	self.Centre()
	882	self.Show(True)
	883
	884
	885
	886	if __name__=="__main__":
	887	app = wx.App()
	888	HelpWindow(None, -1, 'HelpWindow')
	889	app.MainLoop()
	890

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

SasView

source: sasview/sansview/perspectives/fitting/fitpage.py @ 0e2aa40

Download in other formats: