fitpage.py @ 2b63df0

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

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

SasView

source: sasview/sansview/perspectives/fitting/fitpage.py @ 2b63df0

Download in other formats: