fitpage.py @ 2872035

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 2872035 was 2872035, checked in by Gervaise Alina <gervyh@…>, 15 years ago
setvalue after fit.fix display of statictext
Property mode set to `100644`
File size: 49.9 KB

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

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

SasView

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

Download in other formats: