fitpage.py @ 7c845cb

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

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

SasView

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

Download in other formats: