fitpage1D.py @ a69e233

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 a69e233 was 72637e8, checked in by Jae Cho <jhjcho@…>, 16 years ago
Disabled fitting range modifying 2d model plot
Property mode set to `100644`
File size: 29.4 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	name=None
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	self.tcChi = wx.StaticText(self, -1, str(0), style=wx.ALIGN_LEFT)
140	self.tcChi.Hide()
141	self.text1_1 = wx.StaticText(self, -1, 'Chi2/dof', style=wx.ALIGN_LEFT)
142	self.text1_1.Hide()
143	#filling sizer2
144	ix = 0
145	iy = 1
146	self.sizer4.Add(wx.StaticText(self,-1,'Model'),(iy,ix),(1,1)\
147	, wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
148	ix += 1
149	self.sizer4.Add(self.modelbox,(iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
150	ix += 1
151	self.sizer4.Add(self.text1_1,(iy,ix),(1,1),\
152	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
153	ix += 1
154	self.sizer4.Add(self.tcChi,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
155	#----------sizer6-------------------------------------------------
156	self.disable_disp = wx.RadioButton(self, -1, 'No', (10, 10), style=wx.RB_GROUP)
157	self.enable_disp = wx.RadioButton(self, -1, 'Yes', (10, 30))
158	self.Bind(wx.EVT_RADIOBUTTON, self.Set_DipersParam, id=self.disable_disp.GetId())
159	self.Bind(wx.EVT_RADIOBUTTON, self.Set_DipersParam, id=self.enable_disp.GetId())
160	ix= 0
161	iy=1
162	self.sizer6.Add(wx.StaticText(self,-1,'Polydispersity: '),(iy,ix),(1,1)\
163	, wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
164	ix += 1
165	self.sizer6.Add(self.enable_disp ,(iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
166	ix += 1
167	self.sizer6.Add(self.disable_disp ,(iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
168	ix =0
169	iy+=1
170	self.sizer6.Add((20,20),(iy,ix),(1,1),wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
171
172
173	#---------sizer 9 draw----------------------------------------
174
175
176	id = wx.NewId()
177	self.btFit =wx.Button(self,id,'Fit')
178	self.btFit.Bind(wx.EVT_BUTTON, self.onFit,id=id)
179	self.btFit.SetToolTipString("Perform fit.")
180	## Q range
181	#print "self.data fitpage1D" , self.data,hasattr(self.data,"data")
182	# Reversed to the codes; Need to think carefully about consistency in q between 2D plot and fitting
183	if not hasattr(self.data,"data"):
184	self.qmin_x= numpy.min(self.data.x)
185	self.qmax_x= numpy.max(self.data.x)
186	self.num_points= len(self.data.x)
187	else:
188	# Reversed to the codes; Need to think carefully about consistency in q between 2D plot and fitting
189	radius1= math.sqrt(math.pow(self.data.xmin, 2)+ math.pow(self.data.ymin, 2))
190	radius2= math.sqrt(math.pow(self.data.xmax, 2)+ math.pow(self.data.ymin, 2))
191	radius3= math.sqrt(math.pow(self.data.xmin, 2)+ math.pow(self.data.ymax, 2))
192	radius4= math.sqrt(math.pow(self.data.xmax, 2)+ math.pow(self.data.ymax, 2))
193	#self.qmin_x = 0
194	#self.qmax_x = max(radius1, radius2, radius3, radius4)
195	self.qmin_x= 0#self.data.xmin
196	self.qmax_x= math.sqrt(math.pow(max(math.fabs(self.data.xmax),math.fabs(self.data.xmin)),2)
197	+math.pow(max(math.fabs(self.data.ymax),math.fabs(self.data.ymin)),2))#self.data.xmax
198	print "data2D range",self.qmax_x
199
200	self.num_points= 100
201
202
203
204	self.qmin = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
205	self.qmin.SetValue(str(format_number(self.qmin_x)))
206	self.qmin.SetToolTipString("Minimun value of x in linear scale.")
207	self.qmin.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
208	self.qmin.Bind(wx.EVT_TEXT_ENTER, self._onparamEnter)
209	self.qmin.Enable()
210
211	self.qmax = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
212	self.qmax.SetValue(str(format_number(self.qmax_x)))
213	self.qmax.SetToolTipString("Maximum value of x in linear scale.")
214	self.qmax.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
215	self.qmax.Bind(wx.EVT_TEXT_ENTER, self._onparamEnter)
216	self.qmax.Enable()
217
218	self.npts = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
219	self.npts.SetValue(format_number(self.num_points))
220	self.npts.SetToolTipString("Number of point to plot.")
221	self.npts.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
222	self.npts.Bind(wx.EVT_TEXT_ENTER, self._onparamEnter)
223	self.npts.Disable()
224	self.npts.Hide()
225	ix = 0
226	iy = 1
227	self.sizer9.Add(wx.StaticText(self, -1, 'Fitting Range'),(iy, ix),(1,1),\
228	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
229	ix += 1
230	self.sizer9.Add(wx.StaticText(self, -1, 'Min'),(iy, ix),(1,1),\
231	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
232	ix += 1
233	self.sizer9.Add(wx.StaticText(self, -1, 'Max'),(iy, ix),(1,1),\
234	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
235	#ix += 1
236	#self.sizer9.Add(wx.StaticText(self, -1, 'Npts'),(iy, ix),(1,1),\
237	# wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
238	ix = 0
239	iy += 1
240	self.sizer9.Add(wx.StaticText(self, -1, 'Q range'),(iy, ix),(1,1),\
241	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
242	ix += 1
243	self.sizer9.Add(self.qmin,(iy, ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
244	ix += 1
245	self.sizer9.Add(self.qmax,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
246	ix += 1
247	self.sizer9.Add(self.npts,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
248
249	ix += 1
250	self.sizer9.Add(self.btFit,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
251	ix =0
252	iy+=1
253	self.sizer9.Add((20,20),(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
254	#----------sizer 10 draw------------------------------------------------------
255	id = wx.NewId()
256	self.btClose =wx.Button(self,id,'Close')
257	self.btClose.Bind(wx.EVT_BUTTON, self.onClose,id=id)
258	self.btClose.SetToolTipString("Close page.")
259
260	ix= 3
261	iy= 1
262	self.sizer10.Add((20,20),(iy,ix),(1,1),wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
263	ix +=1
264	self.sizer10.Add( self.btClose,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
265	ix =0
266	iy+=1
267	self.sizer10.Add((20,20),(iy,ix),(1,1),wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
268
269	# contains link between model ,all its parameters, and panel organization
270	self.parameters=[]
271	self.fixed_param=[]
272	self.fittable_param=[]
273	#list of dispersion paramaters
274	self.disp_list=[]
275	#contains link between a model and selected parameters to fit
276	self.param_toFit=[]
277	# model on which the fit would be performed
278	self.model=None
279	#dictionary of model name and model class
280	self.model_list_box={}
281
282	if self.model == None:
283	self.qmin.Disable()
284	self.qmax.Disable()
285	else:
286	self.qmin.Enable()
287	self.qmax.Enable()
288
289
290	self.vbox.Layout()
291	self.vbox.Fit(self)
292	self.SetSizer(self.vbox)
293	self.SetScrollbars(20,20,55,40)
294
295	self.Centre()
296	self.Layout()
297	self.GrandParent.GetSizer().Layout()
298
299	def compute_chisqr2D(self):
300	""" @param fn: function that return model value
301	@return residuals
302	"""
303	flag=self.checkFitRange()
304	res=[]
305	if flag== True:
306	try:
307	self.qmin_x = float(self.qmin.GetValue())
308	self.qmax_x = float(self.qmax.GetValue())
309
310	for i in range(len(self.data.x_bins)):
311	if self.data.x_bins[i]>= self.qmin_x and self.data.x_bins[i]<= self.qmax_x:
312	for j in range(len(self.data.y_bins)):
313	if self.data.y_bins[j]>= self.qmin_x and self.data.y_bins[j]<= self.qmax_x:
314	res.append( (self.data.data[j][i]- self.model.runXY(\
315	[self.data.x_bins[i],self.data.y_bins[j]]))\
316	/self.data.err_data[j][i] )
317	sum=0
318
319	for item in res:
320	if numpy.isfinite(item):
321	sum +=item
322	#print "chisqr : sum 2D", xmin, xmax, ymin, ymax,sum
323	#print res
324	self.tcChi.SetLabel(format_number(math.fabs(sum)))
325	except:
326	wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
327	"Chisqr cannot be compute: %s"% sys.exc_value))
328
329
330	def compute_chisqr(self):
331	""" @param fn: function that return model value
332	@return residuals
333	"""
334
335	flag=self.checkFitRange()
336	print "flag", flag
337	if flag== True:
338	try:
339	if hasattr(self.data,"data"):
340	self.compute_chisqr2D()
341	return
342	else:
343	self.qmin_x = float(self.qmin.GetValue())
344	self.qmax_x = float(self.qmax.GetValue())
345	#print "self.qmin_x, self.qmax_x",self.qmin_x,self.qmax_x
346	x,y,dy = [numpy.asarray(v) for v in (self.data.x,self.data.y,self.data.dy)]
347	if self.qmin_x==None and self.qmax_x==None:
348	fx =numpy.asarray([self.model.run(v) for v in x])
349	res=(y - fx)/dy
350	else:
351	idx = (x>= self.qmin_x) & (x <=self.qmax_x)
352	fx = numpy.asarray([self.model.run(item)for item in x[idx ]])
353	res= (y[idx] - fx)/dy[idx]
354
355	sum=0
356	for item in res:
357	if numpy.isfinite(item):
358	sum +=item
359	self.tcChi.SetLabel(format_number(math.fabs(sum)))
360	except:
361	wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
362	"Chisqr cannot be compute: %s"% sys.exc_value))
363
364	def _on_select_model(self,event):
365	"""
366	react when a model is selected from page's combo box
367	post an event to its owner to draw an appropriate theory
368	"""
369	self.btFit.SetFocus()
370	self.disable_disp.SetValue(True)
371	self.sizer8.Clear(True)
372	self.sizer7.Clear(True)
373	self.vbox.Layout()
374	self.SetScrollbars(20,20,55,40)
375	self.Layout()
376	self.parent.GetSizer().Layout()
377
378	for item in self.model_list_box.itervalues():
379	name = item.__name__
380	if hasattr(item, "name"):
381	name = item.name
382	#print "fitpage: _on_select_model model name",name ,event.GetString()
383	if name == None:
384	self.qmin.Disable()
385	self.qmax.Disable()
386	else:
387	self.qmin.Enable()
388	self.qmax.Enable()
389
390	if name ==event.GetString():
391	try:
392	self.model=item()
393	evt = ModelEventbox(model=self.model,name=name)
394	wx.PostEvent(self.event_owner, evt)
395	self.text1_1.Show()
396	self.compute_chisqr()
397	self.tcChi.Show()
398	except:
399	raise #ValueError,"model.name is not equal to model class name"
400	break
401	def onFit(self,event):
402	""" signal for fitting"""
403
404	flag=self.checkFitRange()
405	self.set_manager(self.manager)
406
407	self.qmin_x=float(self.qmin.GetValue())
408	self.qmax_x =float( self.qmax.GetValue())
409	if len(self.param_toFit) >0 and flag==True:
410	self.manager.schedule_for_fit( value=1,fitproblem =None)
411	if hasattr(self.data, "data"):
412	self.manager._on_single_fit(qmin=self.qmin_x,qmax=self.qmax_x,
413	ymin=self.data.ymin, ymax=self.data.ymax,
414	xmin=self.data.xmin,xmax=self.data.xmax)
415	#self.manager._on_single_fit(qmin=self.qmin_x,qmax=self.qmax_x,
416	#ymin=self.data.ymin, ymax=self.data.ymax)
417	else:
418	self.manager._on_single_fit(qmin=self.qmin_x,qmax=self.qmax_x)
419	else:
420	wx.PostEvent(self.parent.parent, StatusEvent(status=\
421	"Select at least on parameter to fit "))
422
423
424	def _onTextEnter(self,event):
425	"""
426	set a flag to determine if the fitting range entered by the user is valid
427	"""
428
429	try:
430	flag=self.checkFitRange()
431	if flag==True and self.model!=None:
432	#print"fit page",self.xmin.GetValue(),self.xmax.GetValue()
433	self.manager.redraw_model(float(self.xmin.GetValue())\
434	,float(self.xmax.GetValue()))
435	except:
436
437	wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
438	"Drawing Error:wrong value entered %s"% sys.exc_value))
439
440
441
442	def get_param_list(self):
443	"""
444	@return self.param_toFit: list containing references to TextCtrl
445	checked.Theses TextCtrl will allow reference to parameters to fit.
446	@raise: if return an empty list of parameter fit will nnote work
447	properly so raise ValueError,"missing parameter to fit"
448	"""
449	if self.param_toFit !=[]:
450	return self.param_toFit
451	else:
452	raise ValueError,"missing parameter to fit"
453
454
455
456	def old_set_panel(self,model):
457	"""
458	Build the panel from the model content
459	@param model: the model selected in combo box for fitting purpose
460	"""
461	self.sizer5.Clear(True)
462	self.parameters = []
463	self.param_toFit=[]
464	self.model = model
465	keys = self.model.getParamList()
466	#print "fitpage1D : dispersion list",self.model.getDispParamList()
467	keys.sort()
468	ik=0
469	im=1
470
471	iy = 1
472	ix = 0
473	self.cb1 = wx.CheckBox(self, -1,"Select all", (10, 10))
474	wx.EVT_CHECKBOX(self, self.cb1.GetId(), self.select_all_param)
475	self.sizer5.Add(self.cb1,(iy, ix),(1,1),\
476	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
477	ix +=1
478	self.text2_2 = wx.StaticText(self, -1, 'Values')
479	self.sizer5.Add(self.text2_2,(iy, ix),(1,1),\
480	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
481	ix +=2
482	self.text2_3 = wx.StaticText(self, -1, 'Errors')
483	self.sizer5.Add(self.text2_3,(iy, ix),(1,1),\
484	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
485	self.text2_3.Hide()
486	ix +=1
487	self.text2_4 = wx.StaticText(self, -1, 'Units')
488	self.sizer5.Add(self.text2_4,(iy, ix),(1,1),\
489	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
490	self.text2_4.Hide()
491	disp_list=self.model.getDispParamList()
492	for item in keys:
493	if not item in disp_list:
494	iy += 1
495	ix = 0
496
497	cb = wx.CheckBox(self, -1, item, (10, 10))
498	cb.SetValue(False)
499	self.sizer5.Add( cb,( iy, ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
500	wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
501
502	ix += 1
503	value= self.model.getParam(item)
504	ctl1 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
505	ctl1.SetValue(str (format_number(value)))
506	ctl1.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
507	ctl1.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
508	self.sizer5.Add(ctl1, (iy,ix),(1,1), wx.EXPAND)
509
510	ix += 1
511	text2=wx.StaticText(self, -1, '+/-')
512	self.sizer5.Add(text2,(iy, ix),(1,1),\
513	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
514	text2.Hide()
515	ix += 1
516	ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
517	self.sizer5.Add(ctl2, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
518	ctl2.Hide()
519	ix +=1
520	# Units
521	try:
522	units = wx.StaticText(self, -1, self.model.details[item][0], style=wx.ALIGN_LEFT)
523	except:
524	units = wx.StaticText(self, -1, "", style=wx.ALIGN_LEFT)
525	self.sizer5.Add(units, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
526	"""
527	else:
528	if not item in fixed:
529	ip = 0
530	iq += 1
531	cb = wx.CheckBox(self, -1, item, (10, 10))
532	cb.SetValue(False)
533	self.sizer5.Add( cb,( iq, ip),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
534	wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
535
536	ip += 1
537	value= self.model.getParam(item)
538	ctl1 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
539	ctl1.SetValue(str (format_number(value)))
540	ctl1.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
541	ctl1.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
542	self.sizer5.Add(ctl1, (iq,ip),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
543	else:
544	ik = 0
545	text = wx.StaticText(self, -1, item, style=wx.ALIGN_LEFT)
546	self.sizer6.Add(text,( im, ik),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
547
548	ik += 1
549	value= self.model.getParam(item)
550	Tctl = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
551	Tctl.SetValue(str (format_number(value)))
552	Tctl.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
553	Tctl.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
554	self.sizer6.Add(Tctl, (im,ik),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
555	self.fixed_param.append([item, Tctl])
556	im += 1
557	#save data
558	"""
559	self.parameters.append([cb,ctl1,text2,ctl2])
560
561	iy+=1
562	self.sizer5.Add((20,20),(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
563
564	#Display units text on panel
565	for item in keys:
566	if self.model.details[item][0]!='':
567	self.text2_4.Show()
568	break
569	else:
570	self.text2_4.Hide()
571
572	#(Disable or )Enable fit button
573	"""
574	if not (len(self.param_toFit ) >0):
575	self.xmin.Enable()
576	self.xmax.Enable()
577	else:
578	self.xmin.Enable()
579	self.xmax.Enable()
580	"""
581
582	self.compute_chisqr()
583	self.vbox.Layout()
584	self.SetScrollbars(20,20,55,40)
585	self.Layout()
586	self.GrandParent.GetSizer().Layout()
587
588
589
590	def _onparamEnter(self,event):
591	"""
592	when enter value on panel redraw model according to changed
593	"""
594	self.set_model()
595	self.compute_chisqr()
596
597	def set_model(self):
598	if len(self.parameters) !=0 and self.model !=None:
599	# Flag to register when a parameter has changed.
600	for item in self.parameters:
601	try:
602	self.text2_3.Hide()
603	item[2].Hide()
604	item[3].Clear()
605	item[3].Hide()
606	except:
607	#enter dispersion value
608	pass
609	self.set_model_parameter()
610
611
612	def select_all_param(self,event):
613	"""
614	set to true or false all checkBox given the main checkbox value cb1
615	"""
616	self.select_all_param_helper()
617
618
619	def select_param(self,event):
620	"""
621	Select TextCtrl checked for fitting purpose and stores them
622	in self.param_toFit=[] list
623	"""
624	self.param_toFit=[]
625	for item in self.parameters:
626	if item[0].GetValue()==True:
627	list= [item[0],item[1],item[2],item[3]]
628	if not (list in self.param_toFit):
629	self.param_toFit.append(list )
630	else:
631	if item in self.param_toFit:
632	self.param_toFit.remove(item)
633
634	for item in self.fittable_param:
635	if item[0].GetValue()==True:
636	list= [item[0],item[1],item[2],item[3]]
637	if not (list in self.param_toFit):
638	self.param_toFit.append(list )
639	else:
640	if item in self.param_toFit:
641	self.param_toFit.remove(item)
642
643
644	if len(self.parameters)+len(self.fittable_param) ==len(self.param_toFit):
645	self.cb1.SetValue(True)
646	else:
647	self.cb1.SetValue(False)
648	"""
649	#qmax, qmin Input enabled all the time
650	self.qmin.Enable()
651	self.qmax.Enable()
652	"""
653
654	def onsetValues(self,chisqr, out,cov):
655	"""
656	Build the panel from the fit result
657	@param chisqr:Value of the goodness of fit metric
658	@param out:list of parameter with the best value found during fitting
659	@param cov:Covariance matrix
660
661	"""
662	#print "fitting : onsetvalues out",out
663	self.tcChi.SetLabel(format_number(chisqr))
664	params = {}
665	is_modified = False
666	has_error = False
667	if out.__class__==numpy.float64:
668	self.param_toFit[0][1].SetValue(format_number(out))
669	self.param_toFit[0][1].Refresh()
670	if cov !=None :
671	self.text2_3.Show()
672	self.param_toFit[0][2].Show()
673	self.param_toFit[0][3].Clear()
674	self.param_toFit[0][3].SetValue(format_number(cov[0]))
675	self.param_toFit[0][3].Show()
676	#out is a list : set parameters and errors in TextCtrl
677	else:
678	i=0
679	#print "fitpage: list param model",list
680	#for item in self.param_toFit:
681	# print "fitpage: list display",item[0].GetLabelText()
682	for item in self.param_toFit:
683	if( out != None ) and len(out)<=len(self.param_toFit)and i < len(out):
684	#item[1].SetValue(format_number(out[i]))
685	item[1].SetValue(format_number(self.model.getParam(item[0].GetLabelText())))
686	item[1].Refresh()
687	if (cov !=None)and len(cov)<=len(self.param_toFit)and i < len(cov):
688	self.text2_3.Show()
689	item[2].Show()
690	item[3].Clear()
691	item[3].SetValue(format_number(cov[i]))
692	item[3].Show()
693	i+=1
694
695	self.vbox.Layout()
696	self.SetScrollbars(20,20,55,40)
697	self.Layout()
698	self.GrandParent.GetSizer().Layout()
699
700
701	def onSmear(self, event):
702	if event.GetString()=="None":
703	self.manager.set_smearer(None)
704
705
706	if event.GetString()=="Gaussian Resolution":
707	from DataLoader.qsmearing import smear_selection
708	smear =smear_selection( self.data )
709	self.manager.set_smearer(smear)
710	# print "on smearing"
711

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

SasView

source: sasview/sansview/perspectives/fitting/fitpage1D.py @ a69e233

Download in other formats: