fitpage1D.py @ 169fb5f

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 169fb5f was 169fb5f, checked in by Jae Cho <jhjcho@…>, 15 years ago
fixed bug
Property mode set to `100644`
File size: 26.3 KB

Rev	Line
[d89f09b]	1	import sys
	2	import wx
	3	import wx.lib
[442895f]	4	import numpy,math
[d89f09b]	5	import copy
[ef8b580]	6
[26bf293]	7	from sans.guicomm.events import StatusEvent
	8	from sans.guiframe.utils import format_number
	9	from modelpage import ModelPage
	10	from modelpage import format_number
[d89f09b]	11	(ModelEventbox, EVT_MODEL_BOX) = wx.lib.newevent.NewEvent()
	12	_BOX_WIDTH = 80
	13
	14
[26bf293]	15
[d23544dc]	16	class FitPage1D(ModelPage):
[d89f09b]	17	"""
	18	FitPanel class contains fields allowing to display results when
	19	fitting a model and one data
	20	@note: For Fit to be performed the user should check at least one parameter
	21	on fit Panel window.
	22
	23	"""
	24	## Internal name for the AUI manager
	25	window_name = "Fit page"
	26	## Title to appear on top of the window
	27	window_caption = "Fit Page"
[10c43a5]	28	name=None
[d89f09b]	29
[9d31a8b]	30	def __init__(self, parent,data, args, *kwargs):
[f39511b]	31	wx.ScrolledWindow.__init__(self, parent, args, *kwargs)
[6e659ae8]	32
[d89f09b]	33	"""
	34	Initialization of the Panel
	35	"""
[26bf293]	36	self.data = data
	37	self.enable2D=False
[d89f09b]	38	self.manager = None
	39	self.parent = parent
[b5847bd8]	40	self.event_owner = None
[d89f09b]	41	#panel interface
	42	self.vbox = wx.BoxSizer(wx.VERTICAL)
[51d47b5]	43	#self.sizer10 = wx.GridBagSizer(5,5)
[26bf293]	44	self.sizer9 = wx.GridBagSizer(5,5)
	45	self.sizer8 = wx.GridBagSizer(5,5)
	46	self.sizer7 = wx.GridBagSizer(5,5)
[08b9c6c8]	47	self.sizer6 = wx.GridBagSizer(5,5)
[5cab7d3]	48	self.sizer5 = wx.GridBagSizer(5,5)
[d89f09b]	49	self.sizer4 = wx.GridBagSizer(5,5)
	50	self.sizer3 = wx.GridBagSizer(5,5)
	51	self.sizer2 = wx.GridBagSizer(5,5)
	52	self.sizer1 = wx.GridBagSizer(5,5)
[26bf293]	53	# Add layer
	54	#data info layer
	55	self.vbox.Add(self.sizer1)
	56	#data range
	57	self.vbox.Add(self.sizer2)
	58	#instrument smearing selection layer
	59	self.vbox.Add(self.sizer3)
	60	#model selection
	61	self.vbox.Add(wx.StaticLine(self, -1), 0, wx.EXPAND, 0)
	62	self.vbox.Add(self.sizer4)
	63	#model paramaters layer
	64	self.vbox.Add(self.sizer5)
	65	#polydispersion selected
	66	self.vbox.Add(wx.StaticLine(self, -1), 0, wx.EXPAND, 0)
	67	self.vbox.Add(self.sizer6)
	68	#combox box for type of dispersion
	69	self.vbox.Add(self.sizer7)
	70	#dispersion parameters layer
	71	self.vbox.Add(self.sizer8)
	72	#fit info layer
	73	self.vbox.Add(wx.StaticLine(self, -1), 0, wx.EXPAND, 0)
	74	self.vbox.Add(self.sizer9)
[51d47b5]	75
[04edd0d]	76
[26bf293]	77	#---------sizer 1 draw--------------------------------
[f39511b]	78	self.DataSource =wx.StaticText(self, -1,str(data.name))
[26bf293]	79	#Filing the sizer containing data related fields
	80	ix = 0
	81	iy = 1
	82	self.sizer1.Add(wx.StaticText(self, -1, 'Data Source Name : '),(iy,ix),\
	83	(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
[5cab7d3]	84
[26bf293]	85	ix += 1
	86	self.sizer1.Add(self.DataSource,(iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[d89f09b]	87
[26bf293]	88	#---------sizer 2 draw--------------------------------
	89	#set maximum range for x in linear scale
[8aac6e1]	90	if not hasattr(self.data,"data"): #Display only for 1D data fit
[ef8b580]	91	ix = 0
	92	iy = 1
[2268d10]	93	# Minimum value of data
	94	self.data_min = wx.StaticText(self, -1,str(format_number(numpy.min(data.x))))
	95	# Maximum value of data
	96	self.data_max = wx.StaticText(self, -1,str(format_number(numpy.max(data.x))))
[8aac6e1]	97	self.text4_3 = wx.StaticText(self, -1, 'Maximum Data Range(Linear)', style=wx.ALIGN_LEFT)
	98	self.sizer2.Add(self.text4_3,(iy,ix),(1,1),\
[26bf293]	99	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
[8aac6e1]	100	ix += 2
	101	self.sizer2.Add(wx.StaticText(self, -1, 'Min :'),(iy, ix),(1,1),\
[26bf293]	102	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[8aac6e1]	103	ix += 1
	104	self.sizer2.Add(self.data_min,(iy, ix),(1,1),\
[26bf293]	105	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[8aac6e1]	106	ix += 1
	107	self.sizer2.Add(wx.StaticText(self, -1, 'Max : '),(iy, ix),(1,1),\
[26bf293]	108	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[8aac6e1]	109	ix += 1
	110	self.sizer2.Add(self.data_max,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[26bf293]	111
	112	#----sizer 3 draw--------------------------------
[ef8b580]	113	self.disable_smearer = wx.RadioButton(self, -1, 'No', (10, 10), style=wx.RB_GROUP)
	114	self.enable_smearer = wx.RadioButton(self, -1, 'Yes', (10, 30))
	115	self.Bind(wx.EVT_RADIOBUTTON, self.onSmear, id=self.disable_smearer.GetId())
	116	self.Bind(wx.EVT_RADIOBUTTON, self.onSmear, id=self.enable_smearer.GetId())
[d89f09b]	117	ix = 0
[26bf293]	118	iy = 1
[32d802f]	119	self.sizer3.Add(wx.StaticText(self,-1,'Instrument Smearing'),(iy,ix),(1,1)\
[08b9c6c8]	120	, wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	121	ix += 1
[ef8b580]	122	self.sizer3.Add(self.enable_smearer,(iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	123	ix += 1
	124	self.sizer3.Add(self.disable_smearer,(iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[26bf293]	125	ix =0
	126	iy+=1
	127	self.sizer3.Add((20,20),(iy,ix),(1,1),wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	128
	129	#------------------ sizer 4 draw------------------------
	130	self.modelbox = wx.ComboBox(self, -1)
[94999eb]	131	self.tcChi = wx.StaticText(self, -1, str(0), style=wx.ALIGN_LEFT)
	132	self.tcChi.Hide()
	133	self.text1_1 = wx.StaticText(self, -1, 'Chi2/dof', style=wx.ALIGN_LEFT)
	134	self.text1_1.Hide()
[26bf293]	135	#filling sizer2
[08b9c6c8]	136	ix = 0
[26bf293]	137	iy = 1
	138	self.sizer4.Add(wx.StaticText(self,-1,'Model'),(iy,ix),(1,1)\
[d89f09b]	139	, wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	140	ix += 1
[26bf293]	141	self.sizer4.Add(self.modelbox,(iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[94999eb]	142	ix += 1
	143	self.sizer4.Add(self.text1_1,(iy,ix),(1,1),\
	144	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	145	ix += 1
	146	self.sizer4.Add(self.tcChi,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[26bf293]	147	#----------sizer6-------------------------------------------------
	148	self.disable_disp = wx.RadioButton(self, -1, 'No', (10, 10), style=wx.RB_GROUP)
	149	self.enable_disp = wx.RadioButton(self, -1, 'Yes', (10, 30))
	150	self.Bind(wx.EVT_RADIOBUTTON, self.Set_DipersParam, id=self.disable_disp.GetId())
	151	self.Bind(wx.EVT_RADIOBUTTON, self.Set_DipersParam, id=self.enable_disp.GetId())
	152	ix= 0
	153	iy=1
	154	self.sizer6.Add(wx.StaticText(self,-1,'Polydispersity: '),(iy,ix),(1,1)\
	155	, wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
[9d31a8b]	156	ix += 1
[26bf293]	157	self.sizer6.Add(self.enable_disp ,(iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	158	ix += 1
	159	self.sizer6.Add(self.disable_disp ,(iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	160	ix =0
	161	iy+=1
	162	self.sizer6.Add((20,20),(iy,ix),(1,1),wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	163
	164
	165	#---------sizer 9 draw----------------------------------------
[94999eb]	166
[26bf293]	167
	168	id = wx.NewId()
	169	self.btFit =wx.Button(self,id,'Fit')
	170	self.btFit.Bind(wx.EVT_BUTTON, self.onFit,id=id)
	171	self.btFit.SetToolTipString("Perform fit.")
	172	## Q range
[3e8e627]	173	#print "self.data fitpage1D" , self.data,hasattr(self.data,"data")
[44999f3]	174	# Reversed to the codes; Need to think carefully about consistency in q between 2D plot and fitting
[26bf293]	175	if not hasattr(self.data,"data"):
	176	self.qmin_x= numpy.min(self.data.x)
	177	self.qmax_x= numpy.max(self.data.x)
[3e8e627]	178	self.num_points= len(self.data.x)
[26bf293]	179	else:
[44999f3]	180	# Reversed to the codes; Need to think carefully about consistency in q between 2D plot and fitting
[c4550b0]	181	radius1= math.sqrt(math.pow(self.data.xmin, 2)+ math.pow(self.data.ymin, 2))
	182	radius2= math.sqrt(math.pow(self.data.xmax, 2)+ math.pow(self.data.ymin, 2))
	183	radius3= math.sqrt(math.pow(self.data.xmin, 2)+ math.pow(self.data.ymax, 2))
	184	radius4= math.sqrt(math.pow(self.data.xmax, 2)+ math.pow(self.data.ymax, 2))
[3e8e627]	185	#self.qmin_x = 0
	186	#self.qmax_x = max(radius1, radius2, radius3, radius4)
[4cb07c0]	187	self.qmin_x= 0#self.data.xmin
	188	self.qmax_x= math.sqrt(math.pow(max(math.fabs(self.data.xmax),math.fabs(self.data.xmin)),2)
	189	+math.pow(max(math.fabs(self.data.ymax),math.fabs(self.data.ymin)),2))#self.data.xmax
[1309e919]	190	#print "data2D range",self.qmax_x
[44999f3]	191
[888e62c]	192	self.num_points= 100
[26bf293]	193
	194
	195
	196	self.qmin = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
	197	self.qmin.SetValue(str(format_number(self.qmin_x)))
	198	self.qmin.SetToolTipString("Minimun value of x in linear scale.")
	199	self.qmin.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
	200	self.qmin.Bind(wx.EVT_TEXT_ENTER, self._onparamEnter)
[4cb07c0]	201	self.qmin.Enable()
[26bf293]	202
	203	self.qmax = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
	204	self.qmax.SetValue(str(format_number(self.qmax_x)))
	205	self.qmax.SetToolTipString("Maximum value of x in linear scale.")
	206	self.qmax.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
	207	self.qmax.Bind(wx.EVT_TEXT_ENTER, self._onparamEnter)
[4cb07c0]	208	self.qmax.Enable()
[26bf293]	209
	210	self.npts = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
	211	self.npts.SetValue(format_number(self.num_points))
	212	self.npts.SetToolTipString("Number of point to plot.")
	213	self.npts.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
	214	self.npts.Bind(wx.EVT_TEXT_ENTER, self._onparamEnter)
	215	self.npts.Disable()
	216	self.npts.Hide()
	217	ix = 0
	218	iy = 1
	219	self.sizer9.Add(wx.StaticText(self, -1, 'Fitting Range'),(iy, ix),(1,1),\
[d89f09b]	220	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
[26bf293]	221	ix += 1
	222	self.sizer9.Add(wx.StaticText(self, -1, 'Min'),(iy, ix),(1,1),\
[d89f09b]	223	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[26bf293]	224	ix += 1
	225	self.sizer9.Add(wx.StaticText(self, -1, 'Max'),(iy, ix),(1,1),\
	226	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	227	#ix += 1
	228	#self.sizer9.Add(wx.StaticText(self, -1, 'Npts'),(iy, ix),(1,1),\
	229	# wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[d89f09b]	230	ix = 0
	231	iy += 1
[8aac6e1]	232	self.sizer9.Add(wx.StaticText(self, -1, 'Q range'),(iy, ix),(1,1),\
[9d31a8b]	233	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
[d89f09b]	234	ix += 1
[26bf293]	235	self.sizer9.Add(self.qmin,(iy, ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	236	ix += 1
	237	self.sizer9.Add(self.qmax,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	238	ix += 1
	239	self.sizer9.Add(self.npts,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[94999eb]	240
[d89f09b]	241	ix += 1
[26bf293]	242	self.sizer9.Add(self.btFit,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[f3113c9]	243	"""
[51d47b5]	244	id = wx.NewId()
	245	self.btStopFit =wx.Button(self,id,'Stop')
	246	self.btStopFit.Bind(wx.EVT_BUTTON, self.onStopFit,id=id)
	247	self.btStopFit.SetToolTipString("Stop the current fitting job.")
	248	self.btStopFit.Hide()
	249	ix += 1
	250	self.sizer9.Add(self.btStopFit,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[f3113c9]	251	"""
[26bf293]	252	ix =0
	253	iy+=1
	254	self.sizer9.Add((20,20),(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	255	#----------sizer 10 draw------------------------------------------------------
[51d47b5]	256	"""
[26bf293]	257	id = wx.NewId()
	258	self.btClose =wx.Button(self,id,'Close')
	259	self.btClose.Bind(wx.EVT_BUTTON, self.onClose,id=id)
	260	self.btClose.SetToolTipString("Close page.")
	261
	262	ix= 3
	263	iy= 1
	264	self.sizer10.Add((20,20),(iy,ix),(1,1),wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	265	ix +=1
	266	self.sizer10.Add( self.btClose,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	267	ix =0
[2dbb681]	268	iy+=1
[26bf293]	269	self.sizer10.Add((20,20),(iy,ix),(1,1),wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
[51d47b5]	270	"""
[d89f09b]	271	# contains link between model ,all its parameters, and panel organization
	272	self.parameters=[]
[08b9c6c8]	273	self.fixed_param=[]
[6b44403]	274	self.fittable_param=[]
[26bf293]	275	#list of dispersion paramaters
[7a0fb4e]	276	self.disp_list=[]
[d89f09b]	277	#contains link between a model and selected parameters to fit
	278	self.param_toFit=[]
	279	# model on which the fit would be performed
	280	self.model=None
[169fb5f]	281	self.back_up_model= None
[d89f09b]	282	#dictionary of model name and model class
[72637e8]	283	self.model_list_box={}
	284
	285	if self.model == None:
	286	self.qmin.Disable()
	287	self.qmax.Disable()
	288	else:
	289	self.qmin.Enable()
	290	self.qmax.Enable()
	291
[32673ac]	292
[f39511b]	293	self.vbox.Layout()
	294	self.vbox.Fit(self)
	295	self.SetSizer(self.vbox)
	296	self.SetScrollbars(20,20,55,40)
	297
[d89f09b]	298	self.Centre()
[b5847bd8]	299	self.Layout()
	300	self.GrandParent.GetSizer().Layout()
[26bf293]	301
	302	def compute_chisqr2D(self):
[442895f]	303	""" @param fn: function that return model value
	304	@return residuals
	305	"""
	306	flag=self.checkFitRange()
[26bf293]	307	res=[]
[442895f]	308	if flag== True:
[948add7]	309	try:
[fd25fb0]	310	print "compute",self.data.err_data
[26bf293]	311	self.qmin_x = float(self.qmin.GetValue())
	312	self.qmax_x = float(self.qmax.GetValue())
	313	for i in range(len(self.data.x_bins)):
[1309e919]	314	#if self.data.x_bins[i]>= self.qmin_x and self.data.x_bins[i]<= self.qmax_x:
[26bf293]	315	for j in range(len(self.data.y_bins)):
[1309e919]	316	if math.pow(self.data.x_bins[i],2)+math.pow(self.data.y_bins[j],2)>=math.pow(self.qmin_x,2):
	317	if math.pow(self.data.x_bins[i],2)+math.pow(self.data.y_bins[j],2)<=math.pow(self.qmax_x,2):
	318	#if self.data.y_bins[j]>= self.qmin_x and self.data.y_bins[j]<= self.qmax_x:
	319	chisqrji=(self.data.data[j][i]- self.model.runXY(\
	320	[self.data.y_bins[j],self.data.x_bins[i]]))\
	321	/self.data.err_data[j][i]
	322	res.append( math.pow(chisqrji,2) )
[948add7]	323	sum=0
[26bf293]	324
[948add7]	325	for item in res:
	326	if numpy.isfinite(item):
	327	sum +=item
[26bf293]	328	#print "chisqr : sum 2D", xmin, xmax, ymin, ymax,sum
[1309e919]	329	#print len(res)
	330	self.tcChi.SetLabel(format_number(math.fabs(sum/ len(res))))
[948add7]	331	except:
[fd25fb0]	332	raise
[948add7]	333	wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
	334	"Chisqr cannot be compute: %s"% sys.exc_value))
[26bf293]	335
[51d47b5]	336	def onStopFit(self, event):
	337	self.manager.stop_fit()
	338	self.btStopFit.Hide()
[f3113c9]	339	self.btFit.Show(True)
[51d47b5]	340	self.vbox.Layout()
	341	self.SetScrollbars(20,20,55,40)
	342	self.Layout()
	343	self.parent.GetSizer().Layout()
[f3113c9]	344
[26bf293]	345	def compute_chisqr(self):
	346	""" @param fn: function that return model value
	347	@return residuals
	348	"""
	349
[d89f09b]	350	flag=self.checkFitRange()
[94999eb]	351	print "flag", flag
[26bf293]	352	if flag== True:
	353	try:
	354	if hasattr(self.data,"data"):
	355	self.compute_chisqr2D()
	356	return
	357	else:
	358	self.qmin_x = float(self.qmin.GetValue())
	359	self.qmax_x = float(self.qmax.GetValue())
[3e8e627]	360	#print "self.qmin_x, self.qmax_x",self.qmin_x,self.qmax_x
[26bf293]	361	x,y,dy = [numpy.asarray(v) for v in (self.data.x,self.data.y,self.data.dy)]
	362	if self.qmin_x==None and self.qmax_x==None:
	363	fx =numpy.asarray([self.model.run(v) for v in x])
[8f274b11]	364	temp=(y - fx)/dy
	365	res= temp*temp
[26bf293]	366	else:
	367	idx = (x>= self.qmin_x) & (x <=self.qmax_x)
	368	fx = numpy.asarray([self.model.run(item)for item in x[idx ]])
[8f274b11]	369	temp=(y[idx] - fx)/dy[idx]
	370	res= temp*temp
[26bf293]	371
	372	sum=0
	373	for item in res:
	374	if numpy.isfinite(item):
[8f274b11]	375	sum +=item
[1309e919]	376	self.tcChi.SetLabel(format_number(math.fabs(sum/ len(res))))
[26bf293]	377	except:
	378	wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
	379	"Chisqr cannot be compute: %s"% sys.exc_value))
	380
[d89f09b]	381	def _on_select_model(self,event):
	382	"""
	383	react when a model is selected from page's combo box
	384	post an event to its owner to draw an appropriate theory
	385	"""
[44bbf6a]	386	self.btFit.SetFocus()
[26bf293]	387	self.disable_disp.SetValue(True)
	388	self.sizer8.Clear(True)
	389	self.sizer7.Clear(True)
	390	self.vbox.Layout()
	391	self.SetScrollbars(20,20,55,40)
	392	self.Layout()
	393	self.parent.GetSizer().Layout()
[72637e8]	394
[d89f09b]	395	for item in self.model_list_box.itervalues():
[00561739]	396	name = item.__name__
	397	if hasattr(item, "name"):
	398	name = item.name
	399	#print "fitpage: _on_select_model model name",name ,event.GetString()
[72637e8]	400	if name == None:
	401	self.qmin.Disable()
	402	self.qmax.Disable()
	403	else:
	404	self.qmin.Enable()
	405	self.qmax.Enable()
	406
[00561739]	407	if name ==event.GetString():
	408	try:
[f39511b]	409	self.model=item()
[169fb5f]	410	self.back_up_model= self.model.clone()
[f39511b]	411	evt = ModelEventbox(model=self.model,name=name)
[d89f09b]	412	wx.PostEvent(self.event_owner, evt)
[94999eb]	413	self.text1_1.Show()
[44999f3]	414	self.compute_chisqr()
[94999eb]	415	self.tcChi.Show()
[00561739]	416	except:
	417	raise #ValueError,"model.name is not equal to model class name"
[26bf293]	418	break
	419	def onFit(self,event):
	420	""" signal for fitting"""
	421
	422	flag=self.checkFitRange()
	423	self.set_manager(self.manager)
	424
	425	self.qmin_x=float(self.qmin.GetValue())
	426	self.qmax_x =float( self.qmax.GetValue())
	427	if len(self.param_toFit) >0 and flag==True:
[fe3aba9]	428	#if self.data.name == self.model.__class__.__name__:
	429	#print "when here have the same name "
	430	#wx.PostEvent(self.parent.parent, StatusEvent(status=\
	431	#"Attempt to fit model %s with itself. fit will\
	432	#not be performed"%self.data.name))
	433	#return
[26bf293]	434	self.manager.schedule_for_fit( value=1,fitproblem =None)
[3e8e627]	435	if hasattr(self.data, "data"):
[4cb07c0]	436	self.manager._on_single_fit(qmin=self.qmin_x,qmax=self.qmax_x,
[e191da5]	437	ymin=self.data.ymin, ymax=self.data.ymax,
	438	xmin=self.data.xmin,xmax=self.data.xmax)
[f3113c9]	439	#self.btStopFit.Show()
	440	#self.btFit.Hide()
[3e8e627]	441	else:
	442	self.manager._on_single_fit(qmin=self.qmin_x,qmax=self.qmax_x)
[f3113c9]	443	#self.btStopFit.Show()
	444	#self.btFit.Hide()
[51d47b5]	445	self.vbox.Layout()
	446	self.SetScrollbars(20,20,55,40)
	447	self.Layout()
	448	self.parent.GetSizer().Layout()
[26bf293]	449	else:
[80f3169]	450	wx.PostEvent(self.parent.parent, StatusEvent(status=\
[fe3aba9]	451	"Select at least one parameter to fit "))
[26bf293]	452
	453
[d89f09b]	454	def _onTextEnter(self,event):
	455	"""
	456	set a flag to determine if the fitting range entered by the user is valid
	457	"""
[442895f]	458
[d89f09b]	459	try:
	460	flag=self.checkFitRange()
	461	if flag==True and self.model!=None:
[f39511b]	462	#print"fit page",self.xmin.GetValue(),self.xmax.GetValue()
[6f73a08]	463	self.manager.redraw_model(float(self.xmin.GetValue())\
[d89f09b]	464	,float(self.xmax.GetValue()))
	465	except:
[6f73a08]	466
[d89f09b]	467	wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
	468	"Drawing Error:wrong value entered %s"% sys.exc_value))
	469
[26bf293]	470
[d89f09b]	471
	472	def get_param_list(self):
	473	"""
	474	@return self.param_toFit: list containing references to TextCtrl
	475	checked.Theses TextCtrl will allow reference to parameters to fit.
	476	@raise: if return an empty list of parameter fit will nnote work
	477	properly so raise ValueError,"missing parameter to fit"
	478	"""
	479	if self.param_toFit !=[]:
	480	return self.param_toFit
	481	else:
	482	raise ValueError,"missing parameter to fit"
	483
[26bf293]	484
[d89f09b]	485	def _onparamEnter(self,event):
	486	"""
	487	when enter value on panel redraw model according to changed
	488	"""
[e9b4cc4]	489	self.set_model()
[442895f]	490	self.compute_chisqr()
[0550752]	491
[e9b4cc4]	492	def set_model(self):
[d89f09b]	493	if len(self.parameters) !=0 and self.model !=None:
[d15c0202]	494	# Flag to register when a parameter has changed.
[d89f09b]	495	for item in self.parameters:
	496	try:
[d23544dc]	497	self.text2_3.Hide()
[0550752]	498	item[2].Hide()
	499	item[3].Clear()
	500	item[3].Hide()
[08b9c6c8]	501	except:
[26bf293]	502	#enter dispersion value
	503	pass
[e9b4cc4]	504	self.set_model_parameter()
	505
[26bf293]	506
[d89f09b]	507	def select_all_param(self,event):
	508	"""
	509	set to true or false all checkBox given the main checkbox value cb1
	510	"""
[94999eb]	511	self.select_all_param_helper()
[d89f09b]	512
	513
	514	def select_param(self,event):
	515	"""
	516	Select TextCtrl checked for fitting purpose and stores them
	517	in self.param_toFit=[] list
	518	"""
	519	self.param_toFit=[]
	520	for item in self.parameters:
	521	if item[0].GetValue()==True:
	522	list= [item[0],item[1],item[2],item[3]]
[89032bf]	523	if not (list in self.param_toFit):
	524	self.param_toFit.append(list )
[d89f09b]	525	else:
	526	if item in self.param_toFit:
	527	self.param_toFit.remove(item)
[6b44403]	528
	529	for item in self.fittable_param:
	530	if item[0].GetValue()==True:
	531	list= [item[0],item[1],item[2],item[3]]
	532	if not (list in self.param_toFit):
	533	self.param_toFit.append(list )
	534	else:
	535	if item in self.param_toFit:
	536	self.param_toFit.remove(item)
	537
	538
	539	if len(self.parameters)+len(self.fittable_param) ==len(self.param_toFit):
[d89f09b]	540	self.cb1.SetValue(True)
	541	else:
	542	self.cb1.SetValue(False)
[4cb07c0]	543	"""
[8d93ec4]	544	#qmax, qmin Input enabled all the time
	545	self.qmin.Enable()
[4cb07c0]	546	self.qmax.Enable()
	547	"""
[d89f09b]	548
	549	def onsetValues(self,chisqr, out,cov):
	550	"""
	551	Build the panel from the fit result
	552	@param chisqr:Value of the goodness of fit metric
	553	@param out:list of parameter with the best value found during fitting
	554	@param cov:Covariance matrix
	555
	556	"""
	557	#print "fitting : onsetvalues out",out
[26bf293]	558	self.tcChi.SetLabel(format_number(chisqr))
[d89f09b]	559	params = {}
	560	is_modified = False
	561	has_error = False
	562	if out.__class__==numpy.float64:
	563	self.param_toFit[0][1].SetValue(format_number(out))
	564	self.param_toFit[0][1].Refresh()
	565	if cov !=None :
	566	self.text2_3.Show()
	567	self.param_toFit[0][2].Show()
	568	self.param_toFit[0][3].Clear()
	569	self.param_toFit[0][3].SetValue(format_number(cov[0]))
	570	self.param_toFit[0][3].Show()
	571	#out is a list : set parameters and errors in TextCtrl
	572	else:
	573	i=0
[d63ef2f]	574	j=0
[4cb07c0]	575	#print "fitpage: list param model",list
[d89f09b]	576	#for item in self.param_toFit:
[d63ef2f]	577	#print "fitpage: list display",item[0].GetLabelText()
[d89f09b]	578	for item in self.param_toFit:
	579	if( out != None ) and len(out)<=len(self.param_toFit)and i < len(out):
[442895f]	580	#item[1].SetValue(format_number(out[i]))
	581	item[1].SetValue(format_number(self.model.getParam(item[0].GetLabelText())))
[d63ef2f]	582	item[1].Refresh()
	583	if(cov !=None)and len(cov)<=len(self.param_toFit)and i < len(cov):
[d89f09b]	584	self.text2_3.Show()
	585	item[2].Show()
	586	item[3].Clear()
[d63ef2f]	587	for j in range(len(out)):
	588	if out[j]==self.model.getParam(item[0].GetLabelText()):#.SetValue(format_number(self.model.getParam(item[0].GetLabelText()))):
	589	#print "jjj", j,item[1],item[1].SetValue(format_number(self.model.getParam(item[0].GetLabelText())))
	590	break
	591	item[3].SetValue(format_number(cov[j]))
[d89f09b]	592	item[3].Show()
	593	i+=1
	594
	595	self.vbox.Layout()
[0aac36f]	596	self.SetScrollbars(20,20,55,40)
	597	self.Layout()
[d89f09b]	598	self.GrandParent.GetSizer().Layout()
[a92d51b]	599
	600
[55e13ab]	601	def onSmear(self, event):
[ef8b580]	602	#print "in smearer",self.enable_smearer.GetValue()
	603	smear =None
[cce33b3]	604	msg=""
[ef8b580]	605	if self.enable_smearer.GetValue():
[55e13ab]	606	from DataLoader.qsmearing import smear_selection
	607	smear =smear_selection( self.data )
[cce33b3]	608	if hasattr(self.data,"dxl"):
	609	msg= ": Resolution smearing parameters"
	610	if hasattr(self.data,"dxw"):
	611	msg= ": Slit smearing parameters"
[ef8b580]	612	if smear ==None:
	613	wx.PostEvent(self.manager.parent, StatusEvent(status=\
	614	"Data contains no smearing information"))
	615	else:
	616	wx.PostEvent(self.manager.parent, StatusEvent(status=\
[cce33b3]	617	"Data contains smearing information %s"%msg))
[ef8b580]	618	self.manager.set_smearer(smear)
	619
	620
	621
[08b9c6c8]	622

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

SasView

source: sasview/sansview/perspectives/fitting/fitpage1D.py @ 169fb5f

Download in other formats: