fitpage.py @ 7b78bb6

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 7b78bb6 was 7b78bb6, checked in by Gervaise Alina <gervyh@…>, 15 years ago
compuite chiqsr for 2 d
Property mode set to `100644`
File size: 33.5 KB

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

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

SasView

source: sasview/sansview/perspectives/fitting/fitpage.py @ 7b78bb6

Download in other formats: