fitpage1D.py @ e9e914f

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 e9e914f was f3113c9, checked in by Gervaise Alina <gervyh@…>, 16 years ago
small bug fixed
Property mode set to `100644`
File size: 26.2 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
	281	#dictionary of model name and model class
[72637e8]	282	self.model_list_box={}
	283
	284	if self.model == None:
	285	self.qmin.Disable()
	286	self.qmax.Disable()
	287	else:
	288	self.qmin.Enable()
	289	self.qmax.Enable()
	290
[32673ac]	291
[f39511b]	292	self.vbox.Layout()
	293	self.vbox.Fit(self)
	294	self.SetSizer(self.vbox)
	295	self.SetScrollbars(20,20,55,40)
	296
[d89f09b]	297	self.Centre()
[b5847bd8]	298	self.Layout()
	299	self.GrandParent.GetSizer().Layout()
[26bf293]	300
	301	def compute_chisqr2D(self):
[442895f]	302	""" @param fn: function that return model value
	303	@return residuals
	304	"""
	305	flag=self.checkFitRange()
[26bf293]	306	res=[]
[442895f]	307	if flag== True:
[948add7]	308	try:
[fd25fb0]	309	print "compute",self.data.err_data
[26bf293]	310	self.qmin_x = float(self.qmin.GetValue())
	311	self.qmax_x = float(self.qmax.GetValue())
	312	for i in range(len(self.data.x_bins)):
[1309e919]	313	#if self.data.x_bins[i]>= self.qmin_x and self.data.x_bins[i]<= self.qmax_x:
[26bf293]	314	for j in range(len(self.data.y_bins)):
[1309e919]	315	if math.pow(self.data.x_bins[i],2)+math.pow(self.data.y_bins[j],2)>=math.pow(self.qmin_x,2):
	316	if math.pow(self.data.x_bins[i],2)+math.pow(self.data.y_bins[j],2)<=math.pow(self.qmax_x,2):
	317	#if self.data.y_bins[j]>= self.qmin_x and self.data.y_bins[j]<= self.qmax_x:
	318	chisqrji=(self.data.data[j][i]- self.model.runXY(\
	319	[self.data.y_bins[j],self.data.x_bins[i]]))\
	320	/self.data.err_data[j][i]
	321	res.append( math.pow(chisqrji,2) )
[948add7]	322	sum=0
[26bf293]	323
[948add7]	324	for item in res:
	325	if numpy.isfinite(item):
	326	sum +=item
[26bf293]	327	#print "chisqr : sum 2D", xmin, xmax, ymin, ymax,sum
[1309e919]	328	#print len(res)
	329	self.tcChi.SetLabel(format_number(math.fabs(sum/ len(res))))
[948add7]	330	except:
[fd25fb0]	331	raise
[948add7]	332	wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
	333	"Chisqr cannot be compute: %s"% sys.exc_value))
[26bf293]	334
[51d47b5]	335	def onStopFit(self, event):
	336	self.manager.stop_fit()
	337	self.btStopFit.Hide()
[f3113c9]	338	self.btFit.Show(True)
[51d47b5]	339	self.vbox.Layout()
	340	self.SetScrollbars(20,20,55,40)
	341	self.Layout()
	342	self.parent.GetSizer().Layout()
[f3113c9]	343
[26bf293]	344	def compute_chisqr(self):
	345	""" @param fn: function that return model value
	346	@return residuals
	347	"""
	348
[d89f09b]	349	flag=self.checkFitRange()
[94999eb]	350	print "flag", flag
[26bf293]	351	if flag== True:
	352	try:
	353	if hasattr(self.data,"data"):
	354	self.compute_chisqr2D()
	355	return
	356	else:
	357	self.qmin_x = float(self.qmin.GetValue())
	358	self.qmax_x = float(self.qmax.GetValue())
[3e8e627]	359	#print "self.qmin_x, self.qmax_x",self.qmin_x,self.qmax_x
[26bf293]	360	x,y,dy = [numpy.asarray(v) for v in (self.data.x,self.data.y,self.data.dy)]
	361	if self.qmin_x==None and self.qmax_x==None:
	362	fx =numpy.asarray([self.model.run(v) for v in x])
[8f274b11]	363	temp=(y - fx)/dy
	364	res= temp*temp
[26bf293]	365	else:
	366	idx = (x>= self.qmin_x) & (x <=self.qmax_x)
	367	fx = numpy.asarray([self.model.run(item)for item in x[idx ]])
[8f274b11]	368	temp=(y[idx] - fx)/dy[idx]
	369	res= temp*temp
[26bf293]	370
	371	sum=0
	372	for item in res:
	373	if numpy.isfinite(item):
[8f274b11]	374	sum +=item
[1309e919]	375	self.tcChi.SetLabel(format_number(math.fabs(sum/ len(res))))
[26bf293]	376	except:
	377	wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
	378	"Chisqr cannot be compute: %s"% sys.exc_value))
	379
[d89f09b]	380	def _on_select_model(self,event):
	381	"""
	382	react when a model is selected from page's combo box
	383	post an event to its owner to draw an appropriate theory
	384	"""
[44bbf6a]	385	self.btFit.SetFocus()
[26bf293]	386	self.disable_disp.SetValue(True)
	387	self.sizer8.Clear(True)
	388	self.sizer7.Clear(True)
	389	self.vbox.Layout()
	390	self.SetScrollbars(20,20,55,40)
	391	self.Layout()
	392	self.parent.GetSizer().Layout()
[72637e8]	393
[d89f09b]	394	for item in self.model_list_box.itervalues():
[00561739]	395	name = item.__name__
	396	if hasattr(item, "name"):
	397	name = item.name
	398	#print "fitpage: _on_select_model model name",name ,event.GetString()
[72637e8]	399	if name == None:
	400	self.qmin.Disable()
	401	self.qmax.Disable()
	402	else:
	403	self.qmin.Enable()
	404	self.qmax.Enable()
	405
[00561739]	406	if name ==event.GetString():
	407	try:
[f39511b]	408	self.model=item()
	409	evt = ModelEventbox(model=self.model,name=name)
[d89f09b]	410	wx.PostEvent(self.event_owner, evt)
[94999eb]	411	self.text1_1.Show()
[44999f3]	412	self.compute_chisqr()
[94999eb]	413	self.tcChi.Show()
[00561739]	414	except:
	415	raise #ValueError,"model.name is not equal to model class name"
[26bf293]	416	break
	417	def onFit(self,event):
	418	""" signal for fitting"""
	419
	420	flag=self.checkFitRange()
	421	self.set_manager(self.manager)
	422
	423	self.qmin_x=float(self.qmin.GetValue())
	424	self.qmax_x =float( self.qmax.GetValue())
	425	if len(self.param_toFit) >0 and flag==True:
[fe3aba9]	426	#if self.data.name == self.model.__class__.__name__:
	427	#print "when here have the same name "
	428	#wx.PostEvent(self.parent.parent, StatusEvent(status=\
	429	#"Attempt to fit model %s with itself. fit will\
	430	#not be performed"%self.data.name))
	431	#return
[26bf293]	432	self.manager.schedule_for_fit( value=1,fitproblem =None)
[3e8e627]	433	if hasattr(self.data, "data"):
[4cb07c0]	434	self.manager._on_single_fit(qmin=self.qmin_x,qmax=self.qmax_x,
[e191da5]	435	ymin=self.data.ymin, ymax=self.data.ymax,
	436	xmin=self.data.xmin,xmax=self.data.xmax)
[f3113c9]	437	#self.btStopFit.Show()
	438	#self.btFit.Hide()
[3e8e627]	439	else:
	440	self.manager._on_single_fit(qmin=self.qmin_x,qmax=self.qmax_x)
[f3113c9]	441	#self.btStopFit.Show()
	442	#self.btFit.Hide()
[51d47b5]	443	self.vbox.Layout()
	444	self.SetScrollbars(20,20,55,40)
	445	self.Layout()
	446	self.parent.GetSizer().Layout()
[26bf293]	447	else:
[80f3169]	448	wx.PostEvent(self.parent.parent, StatusEvent(status=\
[fe3aba9]	449	"Select at least one parameter to fit "))
[26bf293]	450
	451
[d89f09b]	452	def _onTextEnter(self,event):
	453	"""
	454	set a flag to determine if the fitting range entered by the user is valid
	455	"""
[442895f]	456
[d89f09b]	457	try:
	458	flag=self.checkFitRange()
	459	if flag==True and self.model!=None:
[f39511b]	460	#print"fit page",self.xmin.GetValue(),self.xmax.GetValue()
[6f73a08]	461	self.manager.redraw_model(float(self.xmin.GetValue())\
[d89f09b]	462	,float(self.xmax.GetValue()))
	463	except:
[6f73a08]	464
[d89f09b]	465	wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
	466	"Drawing Error:wrong value entered %s"% sys.exc_value))
	467
[26bf293]	468
[d89f09b]	469
	470	def get_param_list(self):
	471	"""
	472	@return self.param_toFit: list containing references to TextCtrl
	473	checked.Theses TextCtrl will allow reference to parameters to fit.
	474	@raise: if return an empty list of parameter fit will nnote work
	475	properly so raise ValueError,"missing parameter to fit"
	476	"""
	477	if self.param_toFit !=[]:
	478	return self.param_toFit
	479	else:
	480	raise ValueError,"missing parameter to fit"
	481
[26bf293]	482
[d89f09b]	483	def _onparamEnter(self,event):
	484	"""
	485	when enter value on panel redraw model according to changed
	486	"""
[e9b4cc4]	487	self.set_model()
[442895f]	488	self.compute_chisqr()
[0550752]	489
[e9b4cc4]	490	def set_model(self):
[d89f09b]	491	if len(self.parameters) !=0 and self.model !=None:
[d15c0202]	492	# Flag to register when a parameter has changed.
[d89f09b]	493	for item in self.parameters:
	494	try:
[d23544dc]	495	self.text2_3.Hide()
[0550752]	496	item[2].Hide()
	497	item[3].Clear()
	498	item[3].Hide()
[08b9c6c8]	499	except:
[26bf293]	500	#enter dispersion value
	501	pass
[e9b4cc4]	502	self.set_model_parameter()
	503
[26bf293]	504
[d89f09b]	505	def select_all_param(self,event):
	506	"""
	507	set to true or false all checkBox given the main checkbox value cb1
	508	"""
[94999eb]	509	self.select_all_param_helper()
[d89f09b]	510
	511
	512	def select_param(self,event):
	513	"""
	514	Select TextCtrl checked for fitting purpose and stores them
	515	in self.param_toFit=[] list
	516	"""
	517	self.param_toFit=[]
	518	for item in self.parameters:
	519	if item[0].GetValue()==True:
	520	list= [item[0],item[1],item[2],item[3]]
[89032bf]	521	if not (list in self.param_toFit):
	522	self.param_toFit.append(list )
[d89f09b]	523	else:
	524	if item in self.param_toFit:
	525	self.param_toFit.remove(item)
[6b44403]	526
	527	for item in self.fittable_param:
	528	if item[0].GetValue()==True:
	529	list= [item[0],item[1],item[2],item[3]]
	530	if not (list in self.param_toFit):
	531	self.param_toFit.append(list )
	532	else:
	533	if item in self.param_toFit:
	534	self.param_toFit.remove(item)
	535
	536
	537	if len(self.parameters)+len(self.fittable_param) ==len(self.param_toFit):
[d89f09b]	538	self.cb1.SetValue(True)
	539	else:
	540	self.cb1.SetValue(False)
[4cb07c0]	541	"""
[8d93ec4]	542	#qmax, qmin Input enabled all the time
	543	self.qmin.Enable()
[4cb07c0]	544	self.qmax.Enable()
	545	"""
[d89f09b]	546
	547	def onsetValues(self,chisqr, out,cov):
	548	"""
	549	Build the panel from the fit result
	550	@param chisqr:Value of the goodness of fit metric
	551	@param out:list of parameter with the best value found during fitting
	552	@param cov:Covariance matrix
	553
	554	"""
	555	#print "fitting : onsetvalues out",out
[26bf293]	556	self.tcChi.SetLabel(format_number(chisqr))
[d89f09b]	557	params = {}
	558	is_modified = False
	559	has_error = False
	560	if out.__class__==numpy.float64:
	561	self.param_toFit[0][1].SetValue(format_number(out))
	562	self.param_toFit[0][1].Refresh()
	563	if cov !=None :
	564	self.text2_3.Show()
	565	self.param_toFit[0][2].Show()
	566	self.param_toFit[0][3].Clear()
	567	self.param_toFit[0][3].SetValue(format_number(cov[0]))
	568	self.param_toFit[0][3].Show()
	569	#out is a list : set parameters and errors in TextCtrl
	570	else:
	571	i=0
[d63ef2f]	572	j=0
[4cb07c0]	573	#print "fitpage: list param model",list
[d89f09b]	574	#for item in self.param_toFit:
[d63ef2f]	575	#print "fitpage: list display",item[0].GetLabelText()
[d89f09b]	576	for item in self.param_toFit:
	577	if( out != None ) and len(out)<=len(self.param_toFit)and i < len(out):
[442895f]	578	#item[1].SetValue(format_number(out[i]))
	579	item[1].SetValue(format_number(self.model.getParam(item[0].GetLabelText())))
[d63ef2f]	580	item[1].Refresh()
	581	if(cov !=None)and len(cov)<=len(self.param_toFit)and i < len(cov):
[d89f09b]	582	self.text2_3.Show()
	583	item[2].Show()
	584	item[3].Clear()
[d63ef2f]	585	for j in range(len(out)):
	586	if out[j]==self.model.getParam(item[0].GetLabelText()):#.SetValue(format_number(self.model.getParam(item[0].GetLabelText()))):
	587	#print "jjj", j,item[1],item[1].SetValue(format_number(self.model.getParam(item[0].GetLabelText())))
	588	break
	589	item[3].SetValue(format_number(cov[j]))
[d89f09b]	590	item[3].Show()
	591	i+=1
	592
	593	self.vbox.Layout()
[0aac36f]	594	self.SetScrollbars(20,20,55,40)
	595	self.Layout()
[d89f09b]	596	self.GrandParent.GetSizer().Layout()
[a92d51b]	597
	598
[55e13ab]	599	def onSmear(self, event):
[ef8b580]	600	#print "in smearer",self.enable_smearer.GetValue()
	601	smear =None
[cce33b3]	602	msg=""
[ef8b580]	603	if self.enable_smearer.GetValue():
[55e13ab]	604	from DataLoader.qsmearing import smear_selection
	605	smear =smear_selection( self.data )
[cce33b3]	606	if hasattr(self.data,"dxl"):
	607	msg= ": Resolution smearing parameters"
	608	if hasattr(self.data,"dxw"):
	609	msg= ": Slit smearing parameters"
[ef8b580]	610	if smear ==None:
	611	wx.PostEvent(self.manager.parent, StatusEvent(status=\
	612	"Data contains no smearing information"))
	613	else:
	614	wx.PostEvent(self.manager.parent, StatusEvent(status=\
[cce33b3]	615	"Data contains smearing information %s"%msg))
[ef8b580]	616	self.manager.set_smearer(smear)
	617
	618
	619
[08b9c6c8]	620

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

SasView

source: sasview/sansview/perspectives/fitting/fitpage1D.py @ e9e914f

Download in other formats: