fitpage.py @ fb8daaaf

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

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

SasView

source: sasview/sansview/perspectives/fitting/fitpage.py @ fb8daaaf

Download in other formats: