fitpage.py @ 2e83ff3

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 2e83ff3 was 55fd102, checked in by Gervaise Alina <gervyh@…>, 16 years ago
modified constraint button
Property mode set to `100644`
File size: 23.1 KB

Line
1	import sys
2	import wx
3	import wx.lib
4	import numpy
5	import copy
6
7	from sans.guicomm.events import StatusEvent
8	(ModelEventbox, EVT_MODEL_BOX) = wx.lib.newevent.NewEvent()
9	_BOX_WIDTH = 80
10
11	def format_number(value, high=False):
12	"""
13	Return a float in a standardized, human-readable formatted string
14	"""
15	try:
16	value = float(value)
17	except:
18	print "returning 0"
19	return "0"
20
21	if high:
22	return "%-6.4g" % value
23	else:
24	return "%-5.3g" % value
25
26
27	class FitPage(wx.Panel):
28	"""
29	FitPanel class contains fields allowing to display results when
30	fitting a model and one data
31	@note: For Fit to be performed the user should check at least one parameter
32	on fit Panel window.
33
34	"""
35	## Internal name for the AUI manager
36	window_name = "Fit page"
37	## Title to appear on top of the window
38	window_caption = "Fit Page"
39
40
41	def __init__(self, parent, args, *kwargs):
42	wx.Panel.__init__(self, parent, args, *kwargs)
43	"""
44	Initialization of the Panel
45	"""
46	self.manager = None
47	self.parent = parent
48	self.event_owner=None
49	#panel interface
50	self.vbox = wx.BoxSizer(wx.VERTICAL)
51	self.sizer4 = wx.GridBagSizer(5,5)
52	self.sizer3 = wx.GridBagSizer(5,5)
53	self.sizer2 = wx.GridBagSizer(5,5)
54	self.sizer1 = wx.GridBagSizer(5,5)
55	self.DataSource = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
56	self.DataSource.SetToolTipString("name of data to fit")
57	self.modelbox = wx.ComboBox(self, -1)
58	id = wx.NewId()
59	self.btFit =wx.Button(self,id,'Fit')
60	self.btFit.Bind(wx.EVT_BUTTON, self.onFit,id=id)
61	self.btFit.SetToolTipString("Perform fit.")
62	self.vbox.Add(self.sizer3)
63	self.vbox.Add(self.sizer2)
64	self.vbox.Add(self.sizer4)
65	self.vbox.Add(self.sizer1)
66
67	ix = 0
68	iy = 1
69	self.sizer3.Add(wx.StaticText(self, -1, 'Data Source'),(iy,ix),\
70	(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
71	ix += 1
72	self.sizer3.Add(self.DataSource,(iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
73	ix += 1
74	self.sizer3.Add((20,20),(iy,ix),(1,1),wx.RIGHT\|wx.EXPAND\|wx.ADJUST_MINSIZE,0)
75	ix = 0
76	iy += 1
77	self.sizer3.Add(wx.StaticText(self,-1,'Model'),(iy,ix),(1,1)\
78	, wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
79	ix += 1
80	self.sizer3.Add(self.modelbox,(iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
81	ix = 1
82	iy = 1
83	self.text4_1 = wx.StaticText(self, -1, 'Min')
84	self.sizer4.Add(self.text4_1,(iy, ix),(1,1),\
85	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
86	self.text4_1.Hide()
87	ix += 2
88	self.text4_2 = wx.StaticText(self, -1, 'Max')
89	self.sizer4.Add(self.text4_2,(iy, ix),(1,1),\
90	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
91	self.text4_2.Hide()
92	ix = 0
93	iy += 1
94	#set maximum range for x in linear scale
95	self.text4_3 = wx.StaticText(self, -1, 'Maximum Data\n Range (Linear)', style=wx.ALIGN_LEFT)
96	self.sizer4.Add(self.text4_3,(iy,ix),(1,1),\
97	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
98	self.text4_3.Hide()
99	ix += 1
100	self.xmin = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
101	self.xmin.SetToolTipString("Minimun value of x in linear scale.")
102	self.sizer4.Add(self.xmin,(iy, ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
103	self.xmin.Bind(wx.EVT_KILL_FOCUS, self._onTextEnter)
104	self.xmin.Bind(wx.EVT_TEXT_ENTER, self._onTextEnter)
105	self.xmin.Hide()
106	ix += 2
107	self.xmax = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
108	self.xmax.SetToolTipString("Maximum value of x in linear scale.")
109	self.sizer4.Add(self.xmax,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
110	self.xmax.Bind(wx.EVT_KILL_FOCUS, self._onTextEnter)
111	self.xmax.Bind(wx.EVT_TEXT_ENTER, self._onTextEnter)
112	self.xmax.Hide()
113	#Set chisqr result into TextCtrl
114	ix = 0
115	iy = 1
116	self.text1_1 = wx.StaticText(self, -1, 'Chi2/dof', style=wx.ALIGN_LEFT)
117	self.sizer1.Add(self.text1_1,(iy,ix),(1,1),\
118	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
119	ix += 1
120	self.tcChi = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
121	self.tcChi.SetToolTipString("Chi^2 over degrees of freedom.")
122	self.sizer1.Add(self.tcChi,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
123	ix +=2
124	self.sizer1.Add(self.btFit,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
125
126	# contains link between model ,all its parameters, and panel organization
127	self.parameters=[]
128	#contains link between a model and selected parameters to fit
129	self.param_toFit=[]
130	# model on which the fit would be performed
131	self.model=None
132	# preview selected model name
133	self.prevmodel_name=None
134	# flag to check if the user has selected a new model in the combox box
135	self.model_hasChanged=False
136	#flag to check if the user has selected a new data
137	self.data_hasChanged=False
138	#dictionary of model name and model class
139	self.model_list_box={}
140	# Data1D to make a deep comparison between 2 Data1D for checking data
141	#change
142	from sans.guitools.plottables import Data1D
143	self.prevData = Data1D(x=[], y=[],dx=None, dy=None)
144	self.prevData.name= None
145
146	self.data=None
147	self.vbox.Layout()
148	self.vbox.Fit(self)
149	self.SetSizer(self.vbox)
150	self.Centre()
151
152
153	def set_owner(self,owner):
154	"""
155	set owner of fitpage
156	@param owner: the class responsible of plotting
157	"""
158	self.event_owner=owner
159
160
161	def set_manager(self, manager):
162	"""
163	set panel manager
164	@param manager: instance of plugin fitting
165	"""
166	self.manager = manager
167
168
169	def _DataNameEnter(self):
170	"""
171	reset the panel when a new data is selected
172	"""
173	if len(self.parameters )>0:
174	for item in self.parameters:
175	item[0].SetValue(False)
176	wx.EVT_CHECKBOX(self, item[0].GetId(), self.select_param)
177	item[2].Hide()
178	item[3].Clear()
179	item[3].Hide()
180	self.xmax.Disable()
181	self.xmin.Disable()
182	self.text1_1.Hide()
183	self.tcChi.Clear()
184	self.vbox.Layout()
185	self.GrandParent.GetSizer().Layout()
186
187
188	def copy_data(self,dataset):
189	"""
190	copy a select data in the panel to compared with the previous selected data
191	@param dataset: the data selected
192	@note : everything a data is selected, the panel check is it is different
193	from the previous selected data
194	"""
195
196	try:
197	if dataset !=None:
198	self.prevData.x= copy.deepcopy(dataset.x)
199	self.prevData.y= copy.deepcopy(dataset.y)
200	self.prevData.name=dataset.name
201	if dataset.dy==None:
202	self.prevData.dy= copy.deepcopy(dataset.dy)
203	except:
204	raise ValueError, " copy_data: cannot copy data"
205
206
207	def onFit(self,event):
208	""" signal for fitting"""
209
210	flag=self.checkFitRange()
211	self.set_manager(self.manager)
212
213	qmin=float(self.xmin.GetValue())
214	qmax =float( self.xmax.GetValue())
215	if len(self.param_toFit) >0 and flag==True:
216	#self.manager._on_single_fit(model=self.model,qmin=qmin,qmax=qmax)
217	self.manager._on_single_fit(qmin=qmin,qmax=qmax)
218	else:
219	wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
220	"Select at least on parameter to fit "))
221	def populate_box(self, dict):
222	"""
223	Populate each combox box of each page
224	@param page: the page to populate
225	"""
226	id=0
227	self.model_list_box=dict
228	list_name=[]
229	for item in self.model_list_box.itervalues():
230	if hasattr(item, "name"):
231	name = item.name
232	else:
233	name = item.__name__
234	list_name.append(name)
235	list_name.sort()
236	for name in list_name:
237	self.modelbox.Insert(name,int(id))
238	id+=1
239	wx.EVT_COMBOBOX(self.modelbox,-1, self._on_select_model)
240	return 0
241
242
243	def _on_select_model(self,event):
244	"""
245	react when a model is selected from page's combo box
246	post an event to its owner to draw an appropriate theory
247	"""
248	for item in self.model_list_box.itervalues():
249	model=item()
250	if hasattr(model, "name"):
251	name = model.name
252	else:
253	name = model.__class__.__name__
254	try:
255	if name ==event.GetString():
256	evt = ModelEventbox(model=model,name=name)
257	wx.PostEvent(self.event_owner, evt)
258	except:
259	raise #ValueError,"model.name is not equal to model class name"
260
261
262	def _onTextEnter(self,event):
263	"""
264	set a flag to determine if the fitting range entered by the user is valid
265	"""
266	try:
267	flag=self.checkFitRange()
268	if flag==True and self.model!=None:
269	self.manager.redraw_model(float(self.xmin.GetValue())\
270	,float(self.xmax.GetValue()))
271	except:
272	wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
273	"Drawing Error:wrong value entered %s"% sys.exc_value))
274
275	def checkFitRange(self):
276	"""
277	Check the validity of fitting range
278	@note: xmin should always be less than xmax or else each control box
279	background is colored in pink.
280	"""
281
282	flag = True
283	valueMin = self.xmin.GetValue()
284	valueMax = self.xmax.GetValue()
285	# Check for possible values entered
286
287	try:
288	if (float(valueMax)> float(valueMin)):
289	self.xmax.SetBackgroundColour(wx.WHITE)
290	self.xmin.SetBackgroundColour(wx.WHITE)
291	else:
292	flag = False
293	self.xmin.SetBackgroundColour("pink")
294	self.xmax.SetBackgroundColour("pink")
295	except:
296	flag = False
297	self.xmin.SetBackgroundColour("pink")
298	self.xmax.SetBackgroundColour("pink")
299
300	self.xmin.Refresh()
301	self.xmax.Refresh()
302	return flag
303
304	def check_data_change(self,dataset):
305	"""
306	check if data selected has changed and set flag self.data_hasChanged
307	has true is changed or false is unchanged
308	@param dataset: data selected
309	"""
310	self.data_hasChanged=False
311	if self.prevData.__class__ !=dataset.__class__:
312	self.data_hasChanged=True
313
314	elif self.prevData.name != dataset.name:
315	self.data_hasChanged=True
316
317	elif len(self.prevData.x) != len(self.prevData.y):
318	self.data_hasChanged=True
319
320	else:
321	for i in range(len(dataset.x)):
322	if self.prevData.x[i] !=dataset.x[i]:
323	self.data_hasChanged=True
324	break
325
326	if self.prevData.y[i] !=dataset.y[i]:
327	self.data_hasChanged=True
328	break
329
330	if dataset.dy !=None:
331	if self.prevData.dy[i] !=dataset.dy[i]:
332	self.data_hasChanged=True
333	break
334
335
336	def set_data_name(self,dataset):
337	"""
338	set data's name.if data has changed reset the panel and xmin and xmax
339	@param name: data 's name
340	@attention: still haven't find a better way to display Q name and unit
341	for xmin and xmax range sizer
342	"""
343	self.check_data_change(dataset)
344	if self.data_hasChanged==True:
345	self._DataNameEnter()
346	self.data=dataset
347	self.copy_data(dataset)
348	#Displaying Data information
349	self.DataSource.SetValue(str(dataset.name))
350	self._xaxis,self._xunit=dataset.get_xaxis()
351	self.text4_3.SetLabel(self._xaxis+"["+self._xunit+"]")
352	self.text4_1.Show()
353	self.text4_2.Show()
354	self.text4_3.Show()
355
356	self.xmin.SetValue(format_number(min(dataset.x)))
357	self.xmin.Show()
358
359	self.xmax.SetValue(format_number(max(dataset.x)))
360	self.xmax.Show()
361
362	if ((len(self.param_toFit ) >0) and self.DataSource.GetValue()and \
363	self.modelbox.GetValue() and (self.model_hasChanged ==False)):
364	self.xmin.Enable()
365	self.xmax.Enable()
366	else:
367	self.xmin.Disable()
368	self.xmax.Disable()
369
370	self.vbox.Layout()
371	self.GrandParent.GetSizer().Layout()
372
373
374	def set_model_name(self,name):
375	"""
376	set model name. set also self.model_hasChanged to true is the model
377	type has changed or false if it didn't
378	@param name: model 's name
379	"""
380	self.model_hasChanged=False
381	if (name != self.prevmodel_name):
382	self.model_hasChanged=True
383	self.tcChi.Clear()
384	#self.modelbox.SetValue(str(name))
385	self.prevmodel_name=self.modelbox.GetValue()
386
387
388	def get_model_box(self):
389	""" return reference to combox box self.model"""
390	return self.modelbox
391
392
393	def get_param_list(self):
394	"""
395	@return self.param_toFit: list containing references to TextCtrl
396	checked.Theses TextCtrl will allow reference to parameters to fit.
397	@raise: if return an empty list of parameter fit will nnote work
398	properly so raise ValueError,"missing parameter to fit"
399	"""
400	if self.param_toFit !=[]:
401	return self.param_toFit
402	else:
403	raise ValueError,"missing parameter to fit"
404
405
406	def set_panel(self,model):
407	"""
408	Build the panel from the model content
409	@param model: the model selected in combo box for fitting purpose
410	"""
411
412	self.sizer2.Clear(True)
413	self.parameters = []
414	self.param_toFit=[]
415	self.model = model
416	keys = self.model.getParamList()
417	keys.sort()
418	iy = 1
419	ix = 0
420	self.cb1 = wx.CheckBox(self, -1,'Parameters', (10, 10))
421	wx.EVT_CHECKBOX(self, self.cb1.GetId(), self.select_all_param)
422	self.sizer2.Add(self.cb1,(iy, ix),(1,1),\
423	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
424	ix +=1
425	self.text2_2 = wx.StaticText(self, -1, 'Values')
426	self.sizer2.Add(self.text2_2,(iy, ix),(1,1),\
427	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
428	ix +=2
429	self.text2_3 = wx.StaticText(self, -1, 'Errors')
430	self.sizer2.Add(self.text2_3,(iy, ix),(1,1),\
431	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
432	self.text2_3.Hide()
433	ix +=1
434	self.text2_4 = wx.StaticText(self, -1, 'Units')
435	self.sizer2.Add(self.text2_4,(iy, ix),(1,1),\
436	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
437	self.text2_4.Hide()
438	for item in keys:
439	iy += 1
440	ix = 0
441
442	cb = wx.CheckBox(self, -1, item, (10, 10))
443	cb.SetValue(False)
444	self.sizer2.Add( cb,( iy, ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
445	wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
446
447	ix += 1
448	value= self.model.getParam(item)
449	ctl1 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
450	ctl1.SetValue(str (format_number(value)))
451	ctl1.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
452	ctl1.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
453	self.sizer2.Add(ctl1, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
454	ix += 1
455	text2=wx.StaticText(self, -1, '+/-')
456	self.sizer2.Add(text2,(iy, ix),(1,1),\
457	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
458	text2.Hide()
459	ix += 1
460	ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
461	self.sizer2.Add(ctl2, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
462	ctl2.Hide()
463	ix +=1
464	# Units
465	try:
466	units = wx.StaticText(self, -1, self.model.details[item][0], style=wx.ALIGN_LEFT)
467	except:
468	units = wx.StaticText(self, -1, "", style=wx.ALIGN_LEFT)
469
470	self.sizer2.Add(units, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
471	#save data
472	self.parameters.append([cb,ctl1,text2,ctl2])
473	#Display units text on panel
474	for item in keys:
475	if self.model.details[item][0]!='':
476	self.text2_4.Show()
477	break
478	else:
479	self.text2_4.Hide()
480	#Disable or enable fit button
481	if (self.modelbox.GetValue() and self.DataSource.GetValue()):
482	if not (len(self.param_toFit ) >0):
483	self.xmin.Disable()
484	self.xmax.Disable()
485	else:
486	self.xmin.Enable()
487	self.xmax.Enable()
488	else:
489	self.xmin.Disable()
490	self.xmax.Disable()
491
492	self.vbox.Layout()
493	self.GrandParent.GetSizer().Layout()
494
495
496
497	def _onparamEnter(self,event):
498	"""
499	when enter value on panel redraw model according to changed
500	"""
501	self.set_model_parameter()
502
503	def set_model_parameter(self):
504	if len(self.parameters) !=0 and self.model !=None:
505	for item in self.parameters:
506	try:
507	name=str(item[0].GetLabelText())
508	value= float(item[1].GetValue())
509
510	self.model.setParam(name,value)
511	self.manager.redraw_model(float(self.xmin.GetValue())\
512	,float(self.xmax.GetValue()))
513	except:
514	wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
515	"Drawing Error:wrong value entered : %s"% sys.exc_value))
516	def select_all_param(self,event):
517	"""
518	set to true or false all checkBox given the main checkbox value cb1
519	"""
520	self.param_toFit=[]
521	if self.parameters !=[]:
522	if self.cb1.GetValue()==True:
523	for item in self.parameters:
524
525	item[0].SetValue(True)
526	list= [item[0],item[1],item[2],item[3]]
527	self.param_toFit.append(list )
528	if (self.modelbox.GetValue() and self.DataSource.GetValue()):
529	if not (len(self.param_toFit ) >0):
530	self.xmin.Disable()
531	self.xmax.Disable()
532	else:
533	self.xmin.Enable()
534	self.xmax.Enable()
535	else:
536	for item in self.parameters:
537	item[0].SetValue(False)
538	self.param_toFit=[]
539
540	self.xmin.Disable()
541	self.xmax.Disable()
542
543
544	def select_param(self,event):
545	"""
546	Select TextCtrl checked for fitting purpose and stores them
547	in self.param_toFit=[] list
548	"""
549	self.param_toFit=[]
550	for item in self.parameters:
551	if item[0].GetValue()==True:
552	list= [item[0],item[1],item[2],item[3]]
553	self.param_toFit.append(list )
554	else:
555	if item in self.param_toFit:
556	self.param_toFit.remove(item)
557	if len(self.parameters)==len(self.param_toFit):
558	self.cb1.SetValue(True)
559	else:
560	self.cb1.SetValue(False)
561	if (self.modelbox.GetValue() and self.DataSource.GetValue()):
562	if not (len(self.param_toFit ) >0):
563	self.xmin.Disable()
564	self.xmax.Disable()
565	else:
566	self.xmin.Enable()
567	self.xmax.Enable()
568	else:
569	self.xmin.Disable()
570	self.xmax.Disable()
571
572
573
574	def onsetValues(self,chisqr, out,cov):
575	"""
576	Build the panel from the fit result
577	@param chisqr:Value of the goodness of fit metric
578	@param out:list of parameter with the best value found during fitting
579	@param cov:Covariance matrix
580
581	"""
582	#print "fitting : onsetvalues out",out
583	self.tcChi.Clear()
584	self.tcChi.SetValue(format_number(chisqr))
585	params = {}
586	is_modified = False
587	has_error = False
588	if out.__class__==numpy.float64:
589	self.param_toFit[0][1].SetValue(format_number(out))
590	self.param_toFit[0][1].Refresh()
591	if cov !=None :
592	self.text2_3.Show()
593	self.param_toFit[0][2].Show()
594	self.param_toFit[0][3].Clear()
595	self.param_toFit[0][3].SetValue(format_number(cov[0]))
596	self.param_toFit[0][3].Show()
597	#out is a list : set parameters and errors in TextCtrl
598	else:
599	i=0
600	#print "fitpage: list param model",list
601	#for item in self.param_toFit:
602	# print "fitpage: list display",item[0].GetLabelText()
603	for item in self.param_toFit:
604	if( out != None ) and len(out)<=len(self.param_toFit)and i < len(out):
605	item[1].SetValue(format_number(out[i]))
606	item[1].Refresh()
607	if (cov !=None)and len(cov)<=len(self.param_toFit)and i < len(cov):
608	self.text2_3.Show()
609	item[2].Show()
610	item[3].Clear()
611	item[3].SetValue(format_number(cov[i]))
612	item[3].Show()
613	i+=1
614
615	self.vbox.Layout()
616	self.GrandParent.GetSizer().Layout()
617

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

SasView

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

Download in other formats: