fitpage1D.py @ 72637e8

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 72637e8 was 72637e8, checked in by Jae Cho <jhjcho@…>, 16 years ago
Disabled fitting range modifying 2d model plot
Property mode set to `100644`
File size: 29.4 KB

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

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

SasView

source: sasview/sansview/perspectives/fitting/fitpage1D.py @ 72637e8

Download in other formats: