fitpage1D.py @ 26bf293

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 26bf293 was 26bf293, checked in by Gervaise Alina <gervyh@…>, 16 years ago
redesign the model1D and madel page
Property mode set to `100644`
File size: 28.1 KB

Line
1	import sys
2	import wx
3	import wx.lib
4	import numpy,math
5	import copy
6	import sans.models.dispersion_models
7	from sans.guicomm.events import StatusEvent
8	from sans.guiframe.utils import format_number
9	from modelpage import ModelPage
10	from modelpage import format_number
11	(ModelEventbox, EVT_MODEL_BOX) = wx.lib.newevent.NewEvent()
12	_BOX_WIDTH = 80
13
14
15
16	class FitPage1D(ModelPage):
17	"""
18	FitPanel class contains fields allowing to display results when
19	fitting a model and one data
20	@note: For Fit to be performed the user should check at least one parameter
21	on fit Panel window.
22
23	"""
24	## Internal name for the AUI manager
25	window_name = "Fit page"
26	## Title to appear on top of the window
27	window_caption = "Fit Page"
28
29
30	def __init__(self, parent,data, args, *kwargs):
31	wx.ScrolledWindow.__init__(self, parent, args, *kwargs)
32	"""
33	Initialization of the Panel
34	"""
35	self.data = data
36	self.enable2D=False
37	self.manager = None
38	self.parent = parent
39	self.event_owner = None
40	#panel interface
41	self.vbox = wx.BoxSizer(wx.VERTICAL)
42	self.sizer10 = wx.GridBagSizer(5,5)
43	self.sizer9 = wx.GridBagSizer(5,5)
44	self.sizer8 = wx.GridBagSizer(5,5)
45	self.sizer7 = wx.GridBagSizer(5,5)
46	self.sizer6 = wx.GridBagSizer(5,5)
47	self.sizer5 = wx.GridBagSizer(5,5)
48	self.sizer4 = wx.GridBagSizer(5,5)
49	self.sizer3 = wx.GridBagSizer(5,5)
50	self.sizer2 = wx.GridBagSizer(5,5)
51	self.sizer1 = wx.GridBagSizer(5,5)
52	# Add layer
53	#data info layer
54	self.vbox.Add(self.sizer1)
55	#data range
56	self.vbox.Add(self.sizer2)
57	#instrument smearing selection layer
58	self.vbox.Add(self.sizer3)
59	#model selection
60	self.vbox.Add(wx.StaticLine(self, -1), 0, wx.EXPAND, 0)
61	self.vbox.Add(self.sizer4)
62	#model paramaters layer
63	self.vbox.Add(self.sizer5)
64	#polydispersion selected
65	self.vbox.Add(wx.StaticLine(self, -1), 0, wx.EXPAND, 0)
66	self.vbox.Add(self.sizer6)
67	#combox box for type of dispersion
68	self.vbox.Add(self.sizer7)
69	#dispersion parameters layer
70	self.vbox.Add(self.sizer8)
71	#fit info layer
72	self.vbox.Add(wx.StaticLine(self, -1), 0, wx.EXPAND, 0)
73	self.vbox.Add(self.sizer9)
74	#close layer
75	self.vbox.Add(wx.StaticLine(self, -1), 0, wx.EXPAND, 0)
76	self.vbox.Add(self.sizer10)
77
78	#---------sizer 1 draw--------------------------------
79	self.DataSource =wx.StaticText(self, -1,str(data.name))
80	self.smearer_box = wx.ComboBox(self, -1)
81	wx.EVT_COMBOBOX( self.smearer_box,-1, self.onSmear )
82	self.smeares= sans.models.dispersion_models.models
83	i=0
84	self.smearer_box.SetValue(str(None))
85	self.smearer_box.Insert(str(None),i)
86
87	for k,v in self.smeares.iteritems():
88	if str(v)=="GaussianModel":
89	self.smearer_box.Insert("Gaussian Resolution",i)
90	else:
91	self.smearer_box.Insert(str(v),i)
92	i+=1
93
94	# Minimum value of data
95	self.data_min = wx.StaticText(self, -1,str(format_number(numpy.min(data.x))))
96	# Maximum value of data
97	self.data_max = wx.StaticText(self, -1,str(format_number(numpy.max(data.x))))
98	#Filing the sizer containing data related fields
99	ix = 0
100	iy = 1
101	self.sizer1.Add(wx.StaticText(self, -1, 'Data Source Name : '),(iy,ix),\
102	(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
103
104	ix += 1
105	self.sizer1.Add(self.DataSource,(iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
106
107	#---------sizer 2 draw--------------------------------
108	ix = 0
109	iy = 1
110	#set maximum range for x in linear scale
111	self.text4_3 = wx.StaticText(self, -1, 'Maximum Data Range(Linear)', style=wx.ALIGN_LEFT)
112	self.sizer2.Add(self.text4_3,(iy,ix),(1,1),\
113	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
114	ix += 2
115	self.sizer2.Add(wx.StaticText(self, -1, 'Min :'),(iy, ix),(1,1),\
116	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
117	ix += 1
118	self.sizer2.Add(self.data_min,(iy, ix),(1,1),\
119	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
120	ix += 1
121	self.sizer2.Add(wx.StaticText(self, -1, 'Max : '),(iy, ix),(1,1),\
122	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
123	ix += 1
124	self.sizer2.Add(self.data_max,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
125
126	#----sizer 3 draw--------------------------------
127	ix = 0
128	iy = 1
129	self.sizer3.Add(wx.StaticText(self,-1,'Instrument Smearing'),(iy,ix),(1,1)\
130	, wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
131	ix += 1
132	self.sizer3.Add(self.smearer_box,(iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
133	ix =0
134	iy+=1
135	self.sizer3.Add((20,20),(iy,ix),(1,1),wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
136
137	#------------------ sizer 4 draw------------------------
138	self.modelbox = wx.ComboBox(self, -1)
139
140	#filling sizer2
141	ix = 0
142	iy = 1
143	self.sizer4.Add(wx.StaticText(self,-1,'Model'),(iy,ix),(1,1)\
144	, wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
145	ix += 1
146	self.sizer4.Add(self.modelbox,(iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
147
148	#----------sizer6-------------------------------------------------
149	self.disable_disp = wx.RadioButton(self, -1, 'No', (10, 10), style=wx.RB_GROUP)
150	self.enable_disp = wx.RadioButton(self, -1, 'Yes', (10, 30))
151	self.Bind(wx.EVT_RADIOBUTTON, self.Set_DipersParam, id=self.disable_disp.GetId())
152	self.Bind(wx.EVT_RADIOBUTTON, self.Set_DipersParam, id=self.enable_disp.GetId())
153	ix= 0
154	iy=1
155	self.sizer6.Add(wx.StaticText(self,-1,'Polydispersity: '),(iy,ix),(1,1)\
156	, wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
157	ix += 1
158	self.sizer6.Add(self.enable_disp ,(iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
159	ix += 1
160	self.sizer6.Add(self.disable_disp ,(iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
161	ix =0
162	iy+=1
163	self.sizer6.Add((20,20),(iy,ix),(1,1),wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
164
165
166	#---------sizer 9 draw----------------------------------------
167	self.tcChi = wx.StaticText(self, -1, str(0), style=wx.ALIGN_LEFT)
168	self.tcChi.Hide()
169	self.text1_1 = wx.StaticText(self, -1, 'Chi2/dof', style=wx.ALIGN_LEFT)
170	self.text1_1.Hide()
171
172	id = wx.NewId()
173	self.btFit =wx.Button(self,id,'Fit')
174	self.btFit.Bind(wx.EVT_BUTTON, self.onFit,id=id)
175	self.btFit.SetToolTipString("Perform fit.")
176	## Q range
177	print "self.data fitpage1D" , self.data,hasattr(self.data,"data")
178	if not hasattr(self.data,"data"):
179	self.qmin_x= numpy.min(self.data.x)
180	self.qmax_x= numpy.max(self.data.x)
181	else:
182	radius1= math.sqrt(math.pow(self.data.xmin, 2)+ math.pow(self.data.ymin, 2))
183	radius2= math.sqrt(math.pow(self.data.xmax, 2)+ math.pow(self.data.ymin, 2))
184	radius3= math.sqrt(math.pow(self.data.xmin, 2)+ math.pow(self.data.ymax, 2))
185	radius4= math.sqrt(math.pow(self.data.xmax, 2)+ math.pow(self.data.ymax, 2))
186	self.qmin_x =0
187	self.qmax_x= max(radius1, radius2, radius3, radius4)
188	print "data2D range",self.qmax_x
189	self.num_points= 100
190
191
192
193	self.qmin = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
194	self.qmin.SetValue(str(format_number(self.qmin_x)))
195	self.qmin.SetToolTipString("Minimun value of x in linear scale.")
196	self.qmin.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
197	self.qmin.Bind(wx.EVT_TEXT_ENTER, self._onparamEnter)
198	self.qmin.Disable()
199
200	self.qmax = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
201	self.qmax.SetValue(str(format_number(self.qmax_x)))
202	self.qmax.SetToolTipString("Maximum value of x in linear scale.")
203	self.qmax.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
204	self.qmax.Bind(wx.EVT_TEXT_ENTER, self._onparamEnter)
205	self.qmax.Disable()
206
207	self.npts = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
208	self.npts.SetValue(format_number(self.num_points))
209	self.npts.SetToolTipString("Number of point to plot.")
210	self.npts.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
211	self.npts.Bind(wx.EVT_TEXT_ENTER, self._onparamEnter)
212	self.npts.Disable()
213	self.npts.Hide()
214	ix = 0
215	iy = 1
216	self.sizer9.Add(wx.StaticText(self, -1, 'Fitting Range'),(iy, ix),(1,1),\
217	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
218	ix += 1
219	self.sizer9.Add(wx.StaticText(self, -1, 'Min'),(iy, ix),(1,1),\
220	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
221	ix += 1
222	self.sizer9.Add(wx.StaticText(self, -1, 'Max'),(iy, ix),(1,1),\
223	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
224	#ix += 1
225	#self.sizer9.Add(wx.StaticText(self, -1, 'Npts'),(iy, ix),(1,1),\
226	# wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
227	ix = 0
228	iy += 1
229	self.sizer9.Add(wx.StaticText(self, -1, 'x range'),(iy, ix),(1,1),\
230	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
231	ix += 1
232	self.sizer9.Add(self.qmin,(iy, ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
233	ix += 1
234	self.sizer9.Add(self.qmax,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
235	ix += 1
236	self.sizer9.Add(self.npts,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
237	ix = 0
238	iy += 1
239	self.sizer9.Add(self.text1_1,(iy,ix),(1,1),\
240	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
241	ix += 1
242	self.sizer9.Add(self.tcChi,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
243	ix +=2
244	self.sizer9.Add(self.btFit,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
245	ix =0
246	iy+=1
247	self.sizer9.Add((20,20),(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
248	#----------sizer 10 draw------------------------------------------------------
249	id = wx.NewId()
250	self.btClose =wx.Button(self,id,'Close')
251	self.btClose.Bind(wx.EVT_BUTTON, self.onClose,id=id)
252	self.btClose.SetToolTipString("Close page.")
253
254	ix= 3
255	iy= 1
256	self.sizer10.Add((20,20),(iy,ix),(1,1),wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
257	ix +=1
258	self.sizer10.Add( self.btClose,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
259	ix =0
260	iy+=1
261	self.sizer10.Add((20,20),(iy,ix),(1,1),wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
262
263	# contains link between model ,all its parameters, and panel organization
264	self.parameters=[]
265	self.fixed_param=[]
266	#list of dispersion paramaters
267	self.disp_list=None
268	#contains link between a model and selected parameters to fit
269	self.param_toFit=[]
270	# model on which the fit would be performed
271	self.model=None
272	#dictionary of model name and model class
273	self.model_list_box={}
274
275
276
277	self.vbox.Layout()
278	self.vbox.Fit(self)
279	self.SetSizer(self.vbox)
280	self.SetScrollbars(20,20,55,40)
281
282	self.Centre()
283	self.Layout()
284	self.GrandParent.GetSizer().Layout()
285
286	def compute_chisqr2D(self):
287	""" @param fn: function that return model value
288	@return residuals
289	"""
290	flag=self.checkFitRange()
291	res=[]
292	if flag== True:
293	try:
294	self.qmin_x = float(self.qmin.GetValue())
295	self.qmax_x = float(self.qmax.GetValue())
296
297	for i in range(len(self.data.x_bins)):
298	if self.data.x_bins[i]>= self.qmin_x and self.data.x_bins[i]<= self.qmax_x:
299	for j in range(len(self.data.y_bins)):
300	if self.data.y_bins[j]>= self.qmin_x and self.data.y_bins[j]<= self.qmax_x:
301	res.append( (self.data.data[j][i]- self.model.runXY(\
302	[self.data.x_bins[i],self.data.y_bins[j]]))\
303	/self.data.err_data[j][i] )
304	sum=0
305
306	for item in res:
307	if numpy.isfinite(item):
308	sum +=item
309	#print "chisqr : sum 2D", xmin, xmax, ymin, ymax,sum
310	#print res
311	self.tcChi.SetValue(format_number(math.fabs(sum)))
312	except:
313	wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
314	"Chisqr cannot be compute: %s"% sys.exc_value))
315
316
317	def compute_chisqr(self):
318	""" @param fn: function that return model value
319	@return residuals
320	"""
321
322	flag=self.checkFitRange()
323	if flag== True:
324	try:
325	if hasattr(self.data,"data"):
326	self.compute_chisqr2D()
327	return
328	else:
329	self.qmin_x = float(self.qmin.GetValue())
330	self.qmax_x = float(self.qmax.GetValue())
331	x,y,dy = [numpy.asarray(v) for v in (self.data.x,self.data.y,self.data.dy)]
332	if self.qmin_x==None and self.qmax_x==None:
333	fx =numpy.asarray([self.model.run(v) for v in x])
334	res=(y - fx)/dy
335	else:
336	idx = (x>= self.qmin_x) & (x <=self.qmax_x)
337	fx = numpy.asarray([self.model.run(item)for item in x[idx ]])
338	res= (y[idx] - fx)/dy[idx]
339
340
341	sum=0
342	for item in res:
343	if numpy.isfinite(item):
344	sum +=item
345	self.tcChi.SetLabel(format_number(math.fabs(sum)))
346	except:
347	wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
348	"Chisqr cannot be compute: %s"% sys.exc_value))
349
350	def _on_select_model(self,event):
351	"""
352	react when a model is selected from page's combo box
353	post an event to its owner to draw an appropriate theory
354	"""
355	self.btFit.SetFocus()
356	self.disable_disp.SetValue(True)
357	self.sizer8.Clear(True)
358	self.sizer7.Clear(True)
359	self.vbox.Layout()
360	self.SetScrollbars(20,20,55,40)
361	self.Layout()
362	self.parent.GetSizer().Layout()
363	for item in self.model_list_box.itervalues():
364	name = item.__name__
365	if hasattr(item, "name"):
366	name = item.name
367	#print "fitpage: _on_select_model model name",name ,event.GetString()
368	if name ==event.GetString():
369	try:
370	self.model=item()
371	evt = ModelEventbox(model=self.model,name=name)
372	wx.PostEvent(self.event_owner, evt)
373	#self.model= item()
374	#self.set_panel(self.model)
375	except:
376	raise #ValueError,"model.name is not equal to model class name"
377	break
378	def onFit(self,event):
379	""" signal for fitting"""
380
381	flag=self.checkFitRange()
382	self.set_manager(self.manager)
383
384	self.qmin_x=float(self.qmin.GetValue())
385	self.qmax_x =float( self.qmax.GetValue())
386	if len(self.param_toFit) >0 and flag==True:
387	self.manager.schedule_for_fit( value=1,fitproblem =None)
388	self.manager._on_single_fit(qmin=self.qmin_x,qmax=self.qmax_x)
389	else:
390	wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
391	"Select at least on parameter to fit "))
392
393
394	def _onTextEnter(self,event):
395	"""
396	set a flag to determine if the fitting range entered by the user is valid
397	"""
398
399	try:
400	flag=self.checkFitRange()
401	if flag==True and self.model!=None:
402	#print"fit page",self.xmin.GetValue(),self.xmax.GetValue()
403	self.manager.redraw_model(float(self.xmin.GetValue())\
404	,float(self.xmax.GetValue()))
405	except:
406
407	wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
408	"Drawing Error:wrong value entered %s"% sys.exc_value))
409
410
411
412	def get_param_list(self):
413	"""
414	@return self.param_toFit: list containing references to TextCtrl
415	checked.Theses TextCtrl will allow reference to parameters to fit.
416	@raise: if return an empty list of parameter fit will nnote work
417	properly so raise ValueError,"missing parameter to fit"
418	"""
419	if self.param_toFit !=[]:
420	return self.param_toFit
421	else:
422	raise ValueError,"missing parameter to fit"
423
424
425
426	def old_set_panel(self,model):
427	"""
428	Build the panel from the model content
429	@param model: the model selected in combo box for fitting purpose
430	"""
431	self.sizer5.Clear(True)
432	self.parameters = []
433	self.param_toFit=[]
434	self.model = model
435	keys = self.model.getParamList()
436	#print "fitpage1D : dispersion list",self.model.getDispParamList()
437	keys.sort()
438	ik=0
439	im=1
440
441	iy = 1
442	ix = 0
443	self.cb1 = wx.CheckBox(self, -1,"Select all", (10, 10))
444	wx.EVT_CHECKBOX(self, self.cb1.GetId(), self.select_all_param)
445	self.sizer5.Add(self.cb1,(iy, ix),(1,1),\
446	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
447	ix +=1
448	self.text2_2 = wx.StaticText(self, -1, 'Values')
449	self.sizer5.Add(self.text2_2,(iy, ix),(1,1),\
450	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
451	ix +=2
452	self.text2_3 = wx.StaticText(self, -1, 'Errors')
453	self.sizer5.Add(self.text2_3,(iy, ix),(1,1),\
454	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
455	self.text2_3.Hide()
456	ix +=1
457	self.text2_4 = wx.StaticText(self, -1, 'Units')
458	self.sizer5.Add(self.text2_4,(iy, ix),(1,1),\
459	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
460	self.text2_4.Hide()
461	disp_list=self.model.getDispParamList()
462	for item in keys:
463	if not item in disp_list:
464	iy += 1
465	ix = 0
466
467	cb = wx.CheckBox(self, -1, item, (10, 10))
468	cb.SetValue(False)
469	self.sizer5.Add( cb,( iy, ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
470	wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
471
472	ix += 1
473	value= self.model.getParam(item)
474	ctl1 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
475	ctl1.SetValue(str (format_number(value)))
476	ctl1.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
477	ctl1.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
478	self.sizer5.Add(ctl1, (iy,ix),(1,1), wx.EXPAND)
479
480	ix += 1
481	text2=wx.StaticText(self, -1, '+/-')
482	self.sizer5.Add(text2,(iy, ix),(1,1),\
483	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
484	text2.Hide()
485	ix += 1
486	ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
487	self.sizer5.Add(ctl2, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
488	ctl2.Hide()
489	ix +=1
490	# Units
491	try:
492	units = wx.StaticText(self, -1, self.model.details[item][0], style=wx.ALIGN_LEFT)
493	except:
494	units = wx.StaticText(self, -1, "", style=wx.ALIGN_LEFT)
495	self.sizer5.Add(units, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
496	"""
497	else:
498	if not item in fixed:
499	ip = 0
500	iq += 1
501	cb = wx.CheckBox(self, -1, item, (10, 10))
502	cb.SetValue(False)
503	self.sizer5.Add( cb,( iq, ip),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
504	wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
505
506	ip += 1
507	value= self.model.getParam(item)
508	ctl1 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
509	ctl1.SetValue(str (format_number(value)))
510	ctl1.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
511	ctl1.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
512	self.sizer5.Add(ctl1, (iq,ip),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
513	else:
514	ik = 0
515	text = wx.StaticText(self, -1, item, style=wx.ALIGN_LEFT)
516	self.sizer6.Add(text,( im, ik),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
517
518	ik += 1
519	value= self.model.getParam(item)
520	Tctl = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
521	Tctl.SetValue(str (format_number(value)))
522	Tctl.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
523	Tctl.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
524	self.sizer6.Add(Tctl, (im,ik),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
525	self.fixed_param.append([item, Tctl])
526	im += 1
527	#save data
528	"""
529	self.parameters.append([cb,ctl1,text2,ctl2])
530
531	iy+=1
532	self.sizer5.Add((20,20),(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
533
534	#Display units text on panel
535	for item in keys:
536	if self.model.details[item][0]!='':
537	self.text2_4.Show()
538	break
539	else:
540	self.text2_4.Hide()
541	#Disable or enable fit button
542
543	if not (len(self.param_toFit ) >0):
544	self.xmin.Disable()
545	self.xmax.Disable()
546	else:
547	self.xmin.Enable()
548	self.xmax.Enable()
549
550	self.compute_chisqr()
551	self.vbox.Layout()
552	self.SetScrollbars(20,20,55,40)
553	self.Layout()
554	self.GrandParent.GetSizer().Layout()
555
556
557
558	def _onparamEnter(self,event):
559	"""
560	when enter value on panel redraw model according to changed
561	"""
562	self.set_model()
563	self.compute_chisqr()
564
565	def set_model(self):
566	if len(self.parameters) !=0 and self.model !=None:
567	# Flag to register when a parameter has changed.
568	for item in self.parameters:
569	try:
570	self.text2_3.Hide()
571	item[2].Hide()
572	item[3].Clear()
573	item[3].Hide()
574	except:
575	#enter dispersion value
576	pass
577	self.set_model_parameter()
578
579
580	def select_all_param(self,event):
581	"""
582	set to true or false all checkBox given the main checkbox value cb1
583	"""
584	self.param_toFit=[]
585	if self.parameters !=[]:
586	if self.cb1.GetValue()==True:
587	for item in self.parameters:
588	item[0].SetValue(True)
589	list= [item[0],item[1],item[2],item[3]]
590	self.param_toFit.append(list )
591
592	if not (len(self.param_toFit ) >0):
593	self.xmin.Disable()
594	self.xmax.Disable()
595	else:
596	self.xmin.Enable()
597	self.xmax.Enable()
598	else:
599	for item in self.parameters:
600	item[0].SetValue(False)
601	self.param_toFit=[]
602
603	self.xmin.Disable()
604	self.xmax.Disable()
605
606
607	def select_param(self,event):
608	"""
609	Select TextCtrl checked for fitting purpose and stores them
610	in self.param_toFit=[] list
611	"""
612	self.param_toFit=[]
613	for item in self.parameters:
614	if item[0].GetValue()==True:
615	list= [item[0],item[1],item[2],item[3]]
616	if not (list in self.param_toFit):
617	self.param_toFit.append(list )
618	else:
619	if item in self.param_toFit:
620	self.param_toFit.remove(item)
621	if len(self.parameters)==len(self.param_toFit):
622	self.cb1.SetValue(True)
623	else:
624	self.cb1.SetValue(False)
625
626	if not (len(self.param_toFit ) >0):
627	self.qmin.Disable()
628	self.qmax.Disable()
629	else:
630	self.qmin.Enable()
631	self.qmax.Enable()
632
633
634
635
636	def onsetValues(self,chisqr, out,cov):
637	"""
638	Build the panel from the fit result
639	@param chisqr:Value of the goodness of fit metric
640	@param out:list of parameter with the best value found during fitting
641	@param cov:Covariance matrix
642
643	"""
644	#print "fitting : onsetvalues out",out
645	self.tcChi.SetLabel(format_number(chisqr))
646	params = {}
647	is_modified = False
648	has_error = False
649	if out.__class__==numpy.float64:
650	self.param_toFit[0][1].SetValue(format_number(out))
651	self.param_toFit[0][1].Refresh()
652	if cov !=None :
653	self.text2_3.Show()
654	self.param_toFit[0][2].Show()
655	self.param_toFit[0][3].Clear()
656	self.param_toFit[0][3].SetValue(format_number(cov[0]))
657	self.param_toFit[0][3].Show()
658	#out is a list : set parameters and errors in TextCtrl
659	else:
660	i=0
661	#print "fitpage: list param model",list
662	#for item in self.param_toFit:
663	# print "fitpage: list display",item[0].GetLabelText()
664	for item in self.param_toFit:
665	if( out != None ) and len(out)<=len(self.param_toFit)and i < len(out):
666	#item[1].SetValue(format_number(out[i]))
667	item[1].SetValue(format_number(self.model.getParam(item[0].GetLabelText())))
668	item[1].Refresh()
669	if (cov !=None)and len(cov)<=len(self.param_toFit)and i < len(cov):
670	self.text2_3.Show()
671	item[2].Show()
672	item[3].Clear()
673	item[3].SetValue(format_number(cov[i]))
674	item[3].Show()
675	i+=1
676
677	self.vbox.Layout()
678	self.SetScrollbars(20,20,55,40)
679	self.Layout()
680	self.GrandParent.GetSizer().Layout()
681
682
683	def onSmear(self, event):
684	if event.GetString()=="None":
685	self.manager.set_smearer(None)
686
687
688	if event.GetString()=="Gaussian Resolution":
689	from DataLoader.qsmearing import smear_selection
690	smear =smear_selection( self.data )
691	self.manager.set_smearer(smear)
692	# print "on smearing"
693

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

SasView

source: sasview/sansview/perspectives/fitting/fitpage1D.py @ 26bf293

Download in other formats: