fitpage.py @ 298b762

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 298b762 was 298b762, checked in by Gervaise Alina <gervyh@…>, 15 years ago
parameters range cahnge
Property mode set to `100644`
File size: 45.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,check_float
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("")
65	item[5].Hide()
66	item[6].SetValue("")
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((60,-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( (0,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	data_min = str((numpy.min(self.data.x)))
149	# Maximum value of data
150	# data_max = str(format_number(numpy.max(self.data.x)))
151	data_max = str((numpy.max(self.data.x)))
152	text4_3 = wx.StaticText(self, -1, 'Total Q Range (1/A)',
153	style=wx.ALIGN_LEFT)
154	sizer_data.Add( text4_3 )
155	sizer_data.Add(wx.StaticText(self, -1, "Min : %s"%data_min))
156
157	sizer_data.Add(wx.StaticText(self, -1, "Max : %s"%data_max))
158
159	else:
160	radius_min= 0
161	x= numpy.max(self.data.xmin, self.data.xmax)
162	y= numpy.max(self.data.ymin, self.data.ymax)
163	radius_max = math.sqrt(xx + yy)
164
165	#For qmin and qmax, do not use format_number.(If do, qmin and max could be different from what is in the data.)
166	# Minimum value of data
167	#data_min = str(format_number(radius_min))
168	data_min = str((radius_min))
169	# Maximum value of data
170	#data_max = str(format_number(radius_max))
171	data_max = str((radius_max))
172	text4_3 = wx.StaticText(self, -1, 'Total Q Range (1/A)',
173	style=wx.ALIGN_LEFT)
174	sizer_data.Add( text4_3 )
175	sizer_data.Add(wx.StaticText(self, -1, "Min : %s"%data_min))
176	sizer_data.Add(wx.StaticText(self, -1, "Max : %s"%data_max))
177
178	boxsizer1.Add(sizer_data)
179	#------------------------------------------------------------
180	self.sizer0.Add(boxsizer1,0, wx.EXPAND \| wx.ALL, 10)
181	self.sizer0.Layout()
182
183	self.qmin_x= data_min
184	self.qmax_x= data_max
185
186
187	def _fill_model_sizer(self, sizer):
188	"""
189	fill sizer containing model info
190	"""
191
192	## class base method to add view 2d button
193	self._set_model_sizer(sizer=sizer, title="Model",object=None )
194
195
196	def _set_sizer_gaussian(self):
197	"""
198	draw sizer with gaussian dispersity parameters
199	"""
200	self.fittable_param=[]
201	self.fixed_param=[]
202	self.orientation_params_disp=[]
203
204	self.sizer4_4.Clear(True)
205
206	if self.model==None:
207	##no model is selected
208	return
209	if not self.enable_disp.GetValue():
210	## the user didn't select dispersity display
211	return
212
213	self._reset_dispersity()
214	# Create the dispersion objects
215	for item in self.model.dispersion.keys():
216	disp_model = GaussianDispersion()
217	self._disp_obj_dict[item] = disp_model
218	self.model.set_dispersion(item, disp_model)
219
220	ix=0
221	iy=1
222	disp = wx.StaticText(self, -1, 'Names')
223	self.sizer4_4.Add(disp,( iy, ix),(1,1),
224	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
225	ix += 1
226	values = wx.StaticText(self, -1, 'Values')
227	self.sizer4_4.Add(values,( iy, ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
228	ix +=2
229	self.text_disp_1 = wx.StaticText(self, -1, 'Errors')
230	self.sizer4_4.Add( self.text_disp_1,(iy, ix),(1,1),\
231	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
232	self.text_disp_1.Hide()
233	ix += 1
234	npts = wx.StaticText(self, -1, 'Npts')
235	self.sizer4_4.Add(npts,( iy, ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
236	ix += 1
237	nsigmas = wx.StaticText(self, -1, 'Nsigmas')
238	self.sizer4_4.Add(nsigmas,( iy, ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
239
240	for item in self.model.dispersion.keys():
241	if not item in self.model.orientation_params:
242	self.disp_cb_dict[item]= None
243	name1=item+".width"
244	name2=item+".npts"
245	name3=item+".nsigmas"
246	iy += 1
247	for p in self.model.dispersion[item].keys():
248
249	if p=="width":
250	ix = 0
251	cb = wx.CheckBox(self, -1, name1, (10, 10))
252	wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
253	self.sizer4_4.Add( cb,( iy, ix),(1,1),
254	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
255	ix = 1
256	value= self.model.getParam(name1)
257	ctl1 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20),
258	style=wx.TE_PROCESS_ENTER)
259	ctl1.SetValue(str (format_number(value)))
260	ctl1.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
261	ctl1.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
262	ctl1.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
263	self.sizer4_4.Add(ctl1, (iy,ix),(1,1),wx.EXPAND)
264	## text to show error sign
265	ix = 2
266	text2=wx.StaticText(self, -1, '+/-')
267	self.sizer4_4.Add(text2,(iy, ix),(1,1),
268	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
269	text2.Hide()
270	## txtcrtl to add error from fit
271	ix = 3
272	ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
273	self.sizer4_4.Add(ctl2, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
274	ctl2.Hide()
275	self.fittable_param.append([cb,name1,ctl1,text2,
276	ctl2, None, None,None])
277	elif p=="npts":
278	ix = 4
279	value= self.model.getParam(name2)
280	Tctl = wx.TextCtrl(self, -1, size=(_BOX_WIDTH/2,20),
281	style=wx.TE_PROCESS_ENTER)
282
283	Tctl.SetValue(str (format_number(value)))
284	Tctl.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
285	Tctl.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
286	Tctl.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
287	self.sizer4_4.Add(Tctl, (iy,ix),(1,1),
288	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
289	self.fixed_param.append([None,name2, Tctl,None,None,
290	None, None,None])
291	elif p=="nsigmas":
292	ix = 5
293	value= self.model.getParam(name3)
294	Tctl = wx.TextCtrl(self, -1, size=(_BOX_WIDTH/2,20),
295	style=wx.TE_PROCESS_ENTER)
296	Tctl.SetValue(str (format_number(value)))
297	Tctl.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
298	Tctl.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
299	Tctl.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
300	self.sizer4_4.Add(Tctl, (iy,ix),(1,1),
301	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
302	ix +=1
303	self.sizer4_4.Add((20,20), (iy,ix),(1,1),
304	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
305
306	self.fixed_param.append([None,name3, Tctl
307	,None,None, None, None,None])
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	for item in self.model.dispersion.keys():
312	if item in self.model.orientation_params:
313	self.disp_cb_dict[item]= None
314	name1=item+".width"
315	name2=item+".npts"
316	name3=item+".nsigmas"
317	iy += 1
318	for p in self.model.dispersion[item].keys():
319
320	if p=="width":
321	ix = 0
322	cb = wx.CheckBox(self, -1, name1, (10, 10))
323	wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
324	self.sizer4_4.Add( cb,( iy, ix),(1,1),
325	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
326	if self.data.__class__.__name__ =="Data2D":
327	cb.Enable()
328	else:
329	cb.Disable()
330	ix = 1
331	value= self.model.getParam(name1)
332	ctl1 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20),
333	style=wx.TE_PROCESS_ENTER)
334	ctl1.SetValue(str (format_number(value)))
335	if self.data.__class__.__name__ =="Data2D":
336	ctl1.Enable()
337	else:
338	ctl1.Disable()
339	ctl1.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
340	ctl1.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
341	ctl1.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
342	self.sizer4_4.Add(ctl1, (iy,ix),(1,1),wx.EXPAND)
343	## text to show error sign
344	ix = 2
345	text2=wx.StaticText(self, -1, '+/-')
346	self.sizer4_4.Add(text2,(iy, ix),(1,1),
347	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
348	text2.Hide()
349	## txtcrtl to add error from fit
350	ix = 3
351	ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
352	self.sizer4_4.Add(ctl2, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
353	ctl2.Hide()
354	if self.data.__class__.__name__ =="Data2D":
355	ctl2.Enable()
356	else:
357	ctl2.Disable()
358	self.fittable_param.append([cb,name1,ctl1,text2,
359	ctl2, None, None,None])
360	self.orientation_params_disp.append([cb,name1,ctl1,text2,
361	ctl2, None, None,None])
362	elif p=="npts":
363	ix = 4
364	value= self.model.getParam(name2)
365	Tctl = wx.TextCtrl(self, -1, size=(_BOX_WIDTH/2,20),
366	style=wx.TE_PROCESS_ENTER)
367
368	Tctl.SetValue(str (format_number(value)))
369	if self.data.__class__.__name__ =="Data2D":
370	Tctl.Enable()
371	else:
372	Tctl.Disable()
373	Tctl.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
374	Tctl.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
375	Tctl.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
376	self.sizer4_4.Add(Tctl, (iy,ix),(1,1),
377	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
378	self.fixed_param.append([None,name2, Tctl,None,None,
379	None, None,None])
380	self.orientation_params_disp.append([None,name2, Tctl,None,None,
381	None, None,None])
382	elif p=="nsigmas":
383	ix = 5
384	value= self.model.getParam(name3)
385	Tctl = wx.TextCtrl(self, -1, size=(_BOX_WIDTH/2,20),
386	style=wx.TE_PROCESS_ENTER)
387	Tctl.SetValue(str (format_number(value)))
388	if self.data.__class__.__name__ =="Data2D":
389	Tctl.Enable()
390	else:
391	Tctl.Disable()
392	Tctl.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
393	Tctl.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
394	Tctl.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
395	self.sizer4_4.Add(Tctl, (iy,ix),(1,1),
396	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
397	ix +=1
398	self.sizer4_4.Add((20,20), (iy,ix),(1,1),
399	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
400	self.fixed_param.append([None,name3, Tctl
401	,None,None, None, None,None])
402	self.orientation_params_disp.append([None,name3, Tctl
403	,None,None, None, None,None])
404
405	wx.PostEvent(self.parent, StatusEvent(status=\
406	" Selected Distribution: Gaussian"))
407	ix =0
408	iy +=1
409	self.sizer4_4.Add((20,20),(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
410	self.sizer4_4.Layout()
411	self.sizer4.Layout()
412	self.SetScrollbars(20,20,200,100)
413
414
415	def _onFit(self, event):
416	"""
417	Allow to fit
418	"""
419	#self.btFit.SetLabel("Stop")
420	from sans.guiframe.utils import check_value
421	flag = check_value( self.qmin, self.qmax)
422
423	if not flag:
424	msg= "Fitting range invalid"
425	wx.PostEvent(self.parent.parent, StatusEvent(status= msg ))
426	return
427
428	if len(self.param_toFit) <= 0:
429	msg= "Select at least one parameter to fit"
430	wx.PostEvent(self.parent.parent, StatusEvent(status= msg ))
431	return
432
433	self.qmin_x=float(self.qmin.GetValue())
434	self.qmax_x =float( self.qmax.GetValue())
435	self.manager._reset_schedule_problem( value=0)
436	self.manager.schedule_for_fit( value=1,page=self,fitproblem =None)
437	self.manager.set_fit_range(page= self,qmin= self.qmin_x, qmax= self.qmax_x)
438	#single fit
439	self.manager.onFit()
440
441	self.sizer5.Layout()
442	self.SetScrollbars(20,20,55,40)
443
444
445	def _on_select_model(self, event):
446	"""
447	call back for model selection
448	"""
449	self._on_select_model_helper()
450	self.set_model_param_sizer(self.model)
451	try:
452	self.compute_chisqr()
453	except:
454	## error occured on chisqr computation
455	pass
456	self.enable_disp.SetValue(False)
457	self.disable_disp.SetValue(True)
458	self._set_dipers_Param(event=None)
459
460	evt = ModelEventbox(model=self.model)
461	wx.PostEvent(self.event_owner, evt)
462
463
464	def _onparamRangeEnter(self, event):
465	"""
466	Check validity of value enter in the parameters range field
467	"""
468	tcrtl= event.GetEventObject()
469	if tcrtl.GetValue().lstrip().rstrip()!="":
470	try:
471	value = float(tcrtl.GetValue())
472	tcrtl.SetBackgroundColour(wx.WHITE)
473	tcrtl.Refresh()
474	except:
475	tcrtl.SetBackgroundColour("pink")
476	tcrtl.Refresh()
477	else:
478	tcrtl.SetBackgroundColour(wx.WHITE)
479	tcrtl.Refresh()
480	self._onparamEnter_helper()
481
482	def _onparamEnter(self,event):
483	"""
484	when enter value on panel redraw model according to changed
485	"""
486	tcrtl= event.GetEventObject()
487	if check_float(tcrtl):
488	self._onparamEnter_helper()
489	self.compute_chisqr()
490	else:
491	msg= "Cannot Plot :Must enter a number!!! "
492	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
493	return
494
495	def reset_page(self, state):
496	"""
497	reset the state
498	"""
499	self.reset_page_helper(state)
500	evt = ModelEventbox(model=self.model)
501	wx.PostEvent(self.event_owner, evt)
502
503
504	def get_range(self):
505	"""
506	return the fitting range
507	"""
508	return float(self.qmin_x) , float(self.qmax_x)
509
510	def get_param_list(self):
511	"""
512	@return self.param_toFit: list containing references to TextCtrl
513	checked.Theses TextCtrl will allow reference to parameters to fit.
514	@raise: if return an empty list of parameter fit will nnote work
515	properly so raise ValueError,"missing parameter to fit"
516	"""
517	if self.param_toFit !=[]:
518	return self.param_toFit
519	else:
520	raise ValueError,"missing parameter to fit"
521
522	def onsetValues(self,chisqr, out,cov):
523	"""
524	Build the panel from the fit result
525	@param chisqr:Value of the goodness of fit metric
526	@param out:list of parameter with the best value found during fitting
527	@param cov:Covariance matrix
528
529	"""
530	self.tcChi.SetLabel(format_number(chisqr))
531	params = {}
532	is_modified = False
533	has_error = False
534	self.text2_3.Hide()
535	if self.text_disp_1 !=None:
536	self.text_disp_1.Hide()
537	#set the panel when fit result are float not list
538	if out.__class__==numpy.float64:
539	self.param_toFit[0][2].SetValue(format_number(out))
540	self.param_toFit[0][2].Refresh()
541
542	self.param_toFit[0][4].Clear()
543	self.param_toFit[0][4].Hide()
544	if cov !=None :
545	self.text2_3.Show(True)
546	if self.text_disp_1 !=None:
547	self.text_disp_1.Show(True)
548	if cov[0]==None:
549	self.param_toFit[0][3].Hide()
550	self.param_toFit[0][4].Clear()
551	self.param_toFit[0][4].Hide()
552	self.param_toFit[0][4].Refresh()
553	else:
554	self.param_toFit[0][3].Show(True)
555	self.param_toFit[0][4].Clear()
556	self.param_toFit[0][4].SetValue(format_number(cov[0]))
557	self.param_toFit[0][4].Show(True)
558	self.param_toFit[0][4].Refresh()
559	else:
560	i=0
561	j=0
562	#Set the panel when fit result are list
563	for item in self.param_toFit:
564	## reset error value to initial state
565	item[4].Clear()
566	item[4].Hide()
567	item[4].Refresh()
568	if( out != None ) and len(out)<=len(self.param_toFit)and i < len(out):
569	item[2].SetValue(format_number(self.model.getParam(item[1])))
570	item[2].Refresh()
571	if(cov !=None)and len(cov)<=len(self.param_toFit)and i < len(cov):
572	self.text2_3.Show(True)
573	if self.text_disp_1!=None:
574	self.text_disp_1.Show(True)
575	item[3].Show(True)
576	item[4].Clear()
577	for j in range(len(out)):
578	if out[j]==self.model.getParam(item[1]):
579	break
580	## unable to compare cov[j]==numpy.nan so switch to None
581	if cov[j]==None:
582	item[3].Hide()
583	item[4].Refresh()
584	item[4].Clear()
585	item[4].Hide()
586	else:
587	item[4].SetValue(format_number(cov[j]))
588	item[4].Refresh()
589	item[4].Show(True)
590	i+=1
591
592	self.sizer3.Layout()
593	self.sizer4.Layout()
594	self.SetScrollbars(20,20,200,100)
595
596
597
598	def onSmear(self, event):
599	"""
600	Create a smear object that will change the way residuals
601	are compute when fitting
602	"""
603	smear =None
604	msg=""
605	if self.enable_smearer.GetValue():
606	from DataLoader.qsmearing import smear_selection
607	smear = smear_selection( self.data )
608	if hasattr(self.data,"dxl"):
609	msg= ": Resolution smearing parameters"
610	if hasattr(self.data,"dxw"):
611	msg= ": Slit smearing parameters"
612	if smear ==None:
613	wx.PostEvent(self.manager.parent, StatusEvent(status=\
614	"Data contains no smearing information"))
615	else:
616	wx.PostEvent(self.manager.parent, StatusEvent(status=\
617	"Data contains smearing information %s"%msg))
618	## set smearing value whether or not the data contain the smearing info
619	self.manager.set_smearer(smear, qmin= float(self.qmin_x),
620	qmax= float(self.qmax_x))
621	##Calculate chi2
622	self.compute_chisqr()
623	## save the state enable smearing
624	self.save_current_state()
625
626
627	def compute_chisqr2D(self):
628	"""
629	compute chi square given a model and data 2D and set the value
630	to the tcChi txtcrl
631	"""
632	from sans.guiframe.utils import check_value
633	flag = check_value( self.qmin, self.qmax)
634	err_image = self.data.err_data
635	if err_image==[] or err_image==None:
636	err_image= numpy.zeros(len(self.data.x_bins),len(self.data.y_bins))
637
638	err_image[err_image==0]=1
639	res=[]
640	if flag== True:
641	try:
642	self.qmin_x = float(self.qmin.GetValue())
643	self.qmax_x = float(self.qmax.GetValue())
644	for i in range(len(self.data.x_bins)):
645	for j in range(len(self.data.y_bins)):
646	#Check the range containing data between self.qmin_x and self.qmax_x
647	value = math.pow(self.data.x_bins[i],2)+ math.pow(self.data.y_bins[j],2)
648	if value >= math.pow(self.qmin_x,2) and value <= math.pow(self.qmax_x,2):
649
650	temp = [self.data.x_bins[i],self.data.y_bins[j]]
651	error= err_image[j][i]
652	chisqrji = (self.data.data[j][i]- self.model.runXY(temp ))/error
653	#Vector containing residuals
654	res.append( math.pow(chisqrji,2) )
655
656	# compute sum of residual
657	sum=0
658	for item in res:
659	if numpy.isfinite(item):
660	sum +=item
661	self.tcChi.SetLabel(format_number(math.fabs(sum/ len(res))))
662	except:
663	wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
664	"Chisqr cannot be compute: %s"% sys.exc_value))
665	return
666
667
668	def compute_chisqr(self):
669	"""
670	compute chi square given a model and data 1D and set the value
671	to the tcChi txtcrl
672	"""
673	from sans.guiframe.utils import check_value
674	flag = check_value( self.qmin, self.qmax)
675
676	if flag== True:
677	try:
678	if hasattr(self.data,"data"):
679	self.compute_chisqr2D()
680	return
681	else:
682	self.qmin_x = float(self.qmin.GetValue())
683	self.qmax_x = float(self.qmax.GetValue())
684	# return residuals within self.qmin_x and self.qmax_x
685	x,y = [numpy.asarray(v) for v in (self.data.x,self.data.y)]
686
687	if self.data.dy==None:
688	dy= numpy.zeros(len(y))
689	else:
690	dy= numpy.asarray(self.data.dy)
691	dy[dy==0]=1
692	if self.qmin_x==None and self.qmax_x==None:
693	fx =numpy.asarray([self.model.run(v) for v in x])
694	temp=(y - fx)/dy
695	res= temp*temp
696	else:
697	idx = (x>= self.qmin_x) & (x <=self.qmax_x)
698	fx = numpy.asarray([self.model.run(item)for item in x[idx ]])
699	temp=(y[idx] - fx)/dy[idx]
700	res= temp*temp
701	#sum of residuals
702	sum=0
703	for item in res:
704	if numpy.isfinite(item):
705	sum +=item
706	self.tcChi.SetLabel(format_number(math.fabs(sum/ len(res))))
707	except:
708	wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
709	"Chisqr cannot be compute: %s"% sys.exc_value))
710	return
711
712
713	def select_all_param(self,event):
714	"""
715	set to true or false all checkBox given the main checkbox value cb1
716	"""
717	self.param_toFit=[]
718
719	if self.parameters !=[]:
720	if self.cb1.GetValue():
721	for item in self.parameters:
722	item[0].SetValue(True)
723	self.param_toFit.append(item )
724	if len(self.fittable_param)>0:
725	for item in self.fittable_param:
726	item[0].SetValue(True)
727	self.param_toFit.append(item )
728	else:
729	for item in self.parameters:
730	item[0].SetValue(False)
731	for item in self.fittable_param:
732	item[0].SetValue(False)
733	self.param_toFit=[]
734
735	self.save_current_state()
736
737
738
739	def select_param(self,event):
740	"""
741	Select TextCtrl checked for fitting purpose and stores them
742	in self.param_toFit=[] list
743	"""
744	self.param_toFit=[]
745	for item in self.parameters:
746	#Select parameters to fit for list of primary parameters
747	if item[0].GetValue():
748	if not (item in self.param_toFit):
749	self.param_toFit.append(item )
750	else:
751	#remove parameters from the fitting list
752	if item in self.param_toFit:
753	self.param_toFit.remove(item)
754	#Select parameters to fit for list of fittable parameters with dispersion
755	for item in self.fittable_param:
756	if item[0].GetValue():
757	if not (item in self.param_toFit):
758	self.param_toFit.append(item)
759	else:
760	#remove parameters from the fitting list
761	if item in self.param_toFit:
762	self.param_toFit.remove(item)
763	#Set the value of checkbox that selected every checkbox or not
764	if len(self.parameters)+len(self.fittable_param) ==len(self.param_toFit):
765	self.cb1.SetValue(True)
766	else:
767	self.cb1.SetValue(False)
768	## save current state of the page
769	self.save_current_state()
770
771
772
773	def set_model_param_sizer(self, model):
774	"""
775	Build the panel from the model content
776	@param model: the model selected in combo box for fitting purpose
777	"""
778	self.sizer3.Clear(True)
779	self.parameters = []
780	self.param_toFit=[]
781	self.fittable_param=[]
782	self.fixed_param=[]
783	self.orientation_params=[]
784	self.orientation_params_disp=[]
785
786	if model ==None:
787	self.sizer3.Layout()
788	self.SetScrollbars(20,20,200,100)
789	return
790	## the panel is drawn using the current value of the fit engine
791	if self.engine_type==None and self.manager !=None:
792	self.engine_type= self.manager._return_engine_type()
793
794	box_description= wx.StaticBox(self, -1,str("Model Parameters"))
795	boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
796	sizer = wx.GridBagSizer(5,5)
797	## save the current model
798	self.model = model
799
800	keys = self.model.getParamList()
801	#list of dispersion paramaters
802	self.disp_list=self.model.getDispParamList()
803
804	keys.sort()
805
806	iy = 1
807	ix = 0
808	self.cb1 = wx.CheckBox(self, -1,"Select all", (10, 10))
809	wx.EVT_CHECKBOX(self, self.cb1.GetId(), self.select_all_param)
810	self.cb1.SetValue(False)
811
812	sizer.Add(self.cb1,(iy, ix),(1,1),\
813	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
814	ix +=1
815	self.text2_2 = wx.StaticText(self, -1, 'Values')
816	sizer.Add(self.text2_2,(iy, ix),(1,1),\
817	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
818	ix +=2
819	self.text2_3 = wx.StaticText(self, -1, 'Errors')
820	sizer.Add(self.text2_3,(iy, ix),(1,1),\
821	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
822	self.text2_3.Hide()
823	ix +=1
824	self.text2_min = wx.StaticText(self, -1, 'Min')
825	sizer.Add(self.text2_min,(iy, ix),(1,1),\
826	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
827	self.text2_min.Hide()
828	ix +=1
829	self.text2_max = wx.StaticText(self, -1, 'Max')
830	sizer.Add(self.text2_max,(iy, ix),(1,1),\
831	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
832	self.text2_max.Hide()
833	ix += 1
834	self.text2_4 = wx.StaticText(self, -1, '[Units]')
835	sizer.Add(self.text2_4,(iy, ix),(1,1),\
836	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
837	self.text2_4.Hide()
838	if self.engine_type=="park":
839	self.text2_max.Show(True)
840	self.text2_min.Show(True)
841
842	for item in keys:
843	if not item in self.disp_list and not item in self.model.orientation_params:
844	iy += 1
845	ix = 0
846	## add parameters name with checkbox for selecting to fit
847	cb = wx.CheckBox(self, -1, item )
848	cb.SetValue(False)
849	wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
850	sizer.Add( cb,( iy, ix),(1,1),
851	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
852	## add parameter value
853	ix += 1
854	value= self.model.getParam(item)
855	ctl1 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20),
856	style=wx.TE_PROCESS_ENTER)
857
858	ctl1.SetValue(format_number(value))
859	ctl1.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
860	ctl1.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
861	ctl1.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
862	sizer.Add(ctl1, (iy,ix),(1,1), wx.EXPAND)
863	## text to show error sign
864	ix += 1
865	text2=wx.StaticText(self, -1, '+/-')
866	sizer.Add(text2,(iy, ix),(1,1),\
867	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
868	text2.Hide()
869	## txtcrtl to add error from fit
870	ix += 1
871	ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
872	sizer.Add(ctl2, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
873	ctl2.Hide()
874
875	param_min, param_max= self.model.details[item][1:]
876	ix += 1
877	ctl3 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER)
878	if param_min ==None:
879	ctl3.SetValue("")
880	else:
881	ctl3.SetValue(str(param_min))
882	ctl3.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
883	ctl3.Bind(wx.EVT_KILL_FOCUS, self._onparamRangeEnter)
884	ctl3.Bind(wx.EVT_TEXT_ENTER,self._onparamRangeEnter)
885	sizer.Add(ctl3, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
886	ctl3.Hide()
887
888	ix += 1
889	ctl4 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER)
890	ctl4.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
891	ctl4.Bind(wx.EVT_KILL_FOCUS, self._onparamRangeEnter)
892	ctl4.Bind(wx.EVT_TEXT_ENTER,self._onparamRangeEnter)
893	sizer.Add(ctl4, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
894	if param_max==None:
895	ctl4.SetValue("")
896	else:
897	ctl4.SetValue(str(param_max))
898	ctl4.Hide()
899
900	if self.engine_type=="park":
901	ctl3.Show(True)
902	ctl4.Show(True)
903
904	ix +=1
905	# Units
906	try:
907	units = wx.StaticText(self, -1, self.model.details[item][0], style=wx.ALIGN_LEFT)
908	except:
909	units = wx.StaticText(self, -1, "", style=wx.ALIGN_LEFT)
910	sizer.Add(units, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
911
912	##[cb state, name, value, "+/-", error of fit, min, max , units]
913	self.parameters.append([cb,item, ctl1,
914	text2,ctl2, ctl3, ctl4,None])
915
916	iy+=1
917	sizer.Add((10,10),(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
918	for item in self.model.orientation_params:
919	if not item in self.disp_list :
920	iy += 1
921	ix = 0
922	## add parameters name with checkbox for selecting to fit
923	cb = wx.CheckBox(self, -1, item )
924	cb.SetValue(False)
925	wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
926	if self.data.__class__.__name__ =="Data2D":
927	cb.Enable()
928	else:
929	cb.Disable()
930	sizer.Add( cb,( iy, ix),(1,1),
931	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
932	## add parameter value
933	ix += 1
934	value= self.model.getParam(item)
935	ctl1 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20),
936	style=wx.TE_PROCESS_ENTER)
937
938	ctl1.SetValue(format_number(value))
939	if self.data.__class__.__name__ =="Data2D":
940	ctl1.Enable()
941	else:
942	ctl1.Disable()
943	ctl1.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
944	ctl1.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
945	ctl1.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
946	sizer.Add(ctl1, (iy,ix),(1,1), wx.EXPAND)
947	## text to show error sign
948	ix += 1
949	text2=wx.StaticText(self, -1, '+/-')
950	sizer.Add(text2,(iy, ix),(1,1),\
951	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
952	text2.Hide()
953	## txtcrtl to add error from fit
954	ix += 1
955	ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
956	sizer.Add(ctl2, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
957	ctl2.Hide()
958	if self.data.__class__.__name__ =="Data2D":
959	ctl1.Enable()
960	else:
961	ctl1.Disable()
962	param_min, param_max= self.model.details[item][1:]
963	ix += 1
964	ctl3 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER)
965	if param_min ==None:
966	ctl3.SetValue("")
967	else:
968	ctl3.SetValue(str(param_min))
969	ctl3.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
970	ctl3.Bind(wx.EVT_KILL_FOCUS, self._onparamRangeEnter)
971	ctl3.Bind(wx.EVT_TEXT_ENTER,self._onparamRangeEnter)
972	sizer.Add(ctl3, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
973	ctl3.Hide()
974	if self.data.__class__.__name__ =="Data2D":
975	ctl3.Enable()
976	else:
977	ctl3.Disable()
978	ix += 1
979	ctl4 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER)
980	ctl4.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
981	ctl4.Bind(wx.EVT_KILL_FOCUS, self._onparamRangeEnter)
982	ctl4.Bind(wx.EVT_TEXT_ENTER,self._onparamRangeEnter)
983	sizer.Add(ctl4, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
984	ctl4.SetValue(str(param_max))
985	ctl4.Hide()
986	if self.data.__class__.__name__ =="Data2D":
987	ctl4.Enable()
988	else:
989	ctl4.Disable()
990	if self.engine_type=="park":
991	ctl3.Show(True)
992	ctl4.Show(True)
993
994	ix +=1
995	# Units
996	try:
997	units = wx.StaticText(self, -1, self.model.details[item][0], style=wx.ALIGN_LEFT)
998	except:
999	units = wx.StaticText(self, -1, "", style=wx.ALIGN_LEFT)
1000	sizer.Add(units, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
1001
1002	##[cb state, name, value, "+/-", error of fit, min, max , units]
1003	self.parameters.append([cb,item, ctl1,
1004	text2,ctl2, ctl3, ctl4,None])
1005	self.orientation_params.append([cb,item, ctl1,
1006	text2,ctl2, ctl3, ctl4,None])
1007
1008	iy+=1
1009	sizer.Add((10,10),(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
1010
1011	#Display units text on panel
1012	for item in keys:
1013	if self.model.details[item][0]!='':
1014	self.text2_4.Show()
1015	break
1016	else:
1017	self.text2_4.Hide()
1018
1019	boxsizer1.Add(sizer)
1020
1021	self.sizer3.Add(boxsizer1,0, wx.EXPAND \| wx.ALL, 10)
1022	self.sizer3.Layout()
1023	self.SetScrollbars(20,20,200,100)
1024
1025
1026
1027
1028	class HelpWindow(wx.Frame):
1029	def __init__(self, parent, id, title):
1030	wx.Frame.__init__(self, parent, id, title, size=(570, 400))
1031
1032	from sans.models.CylinderModel import CylinderModel
1033	model = CylinderModel()
1034
1035	from danse.common.plottools.plottables import Data1D
1036	data= Data1D(x=[1,2], y=[3,4], dy=[0.1, 0,1])
1037
1038	from fitpanel import PageInfo
1039	myinfo = PageInfo(self, model, data=data )
1040
1041	## add data
1042
1043	from models import ModelList
1044	mylist= ModelList()
1045
1046	from sans.models.SphereModel import SphereModel
1047	from sans.models.SquareWellStructure import SquareWellStructure
1048	from sans.models.DebyeModel import DebyeModel
1049	from sans.models.LineModel import LineModel
1050	name= "shapes"
1051	list1= [SphereModel]
1052	mylist.set_list( name, list1)
1053
1054	name= "Shape-independent"
1055	list1= [DebyeModel]
1056	mylist.set_list( name, list1)
1057
1058	name= "Structure Factors"
1059	list1= [SquareWellStructure]
1060	mylist.set_list( name, list1)
1061
1062	name= "Added models"
1063	list1= [LineModel]
1064	mylist.set_list( name, list1)
1065
1066	myinfo.model_list_box = mylist.get_list()
1067
1068	self.page = FitPage(self, myinfo)
1069
1070
1071
1072	self.Centre()
1073	self.Show(True)
1074
1075
1076
1077	if __name__=="__main__":
1078	app = wx.App()
1079	HelpWindow(None, -1, 'HelpWindow')
1080	app.MainLoop()
1081

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

SasView

source: sasview/sansview/perspectives/fitting/fitpage.py @ 298b762

Download in other formats: