fitpage.py @ dcf29d7

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

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

SasView

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

Download in other formats: