fitpage.py @ 813334e

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

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

SasView

source: sasview/sansview/perspectives/fitting/fitpage.py @ 813334e

Download in other formats: