fitpage.py @ d40a9cd

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 d40a9cd was d40a9cd, checked in by Jae Cho <jhjcho@…>, 15 years ago
fixed a bug in polydispersion saving function with checkbox marked
Property mode set to `100644`
File size: 63.4 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	(FitStopEvent, EVT_FIT_STOP) = wx.lib.newevent.NewEvent()
20	_BOX_WIDTH = 76
21
22	import basepage
23	from basepage import BasicPage
24	from basepage import PageInfoEvent
25	from DataLoader.qsmearing import smear_selection
26
27	class FitPage(BasicPage):
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	def __init__(self,parent, page_info):
36	BasicPage.__init__(self, parent, page_info)
37	"""
38	Initialization of the Panel
39	"""
40	## fit page does not content npts txtcrtl
41	self.npts=None
42	## thread for compute Chisqr
43	self.calc_Chisqr=None
44	## default fitengine type
45	self.engine_type = None
46	## draw sizer
47	self._fill_datainfo_sizer()
48
49	self._fill_model_sizer( self.sizer1)
50
51	self._fill_range_sizer()
52	#self._on_select_model(event=None)
53	if self.data !=None:
54	self.smearer = smear_selection( self.data )
55	if self.smearer ==None:
56	self.enable_smearer.Disable()
57	self.disable_smearer.Disable()
58
59	## to update the panel according to the fit engine type selected
60	self.Bind(EVT_FITTER_TYPE,self._on_engine_change)
61	self.Bind(EVT_FIT_STOP,self._on_fit_complete)
62
63	def _on_fit_complete(self, event):
64	"""
65	When fit is complete ,reset the fit button label.
66	"""
67	#self.btFit.SetLabel("Fit")
68	#self.btFit.Unbind(event=wx.EVT_BUTTON, id=self.btFit.GetId())
69	#self.btFit.Bind(event=wx.EVT_BUTTON, handler=self._onFit,id=self.btFit.GetId())
70	pass
71
72
73	def _on_engine_change(self, event):
74	"""
75	get an event containing the current name of the fit engine type
76	@param event: FitterTypeEvent containing the name of the current engine
77	"""
78	self.engine_type = event.type
79
80	if len(self.parameters)==0:
81	return
82	if event.type =="park":
83	self.btFit.SetLabel("Fit")
84
85	for item in self.parameters:
86	if event.type =="scipy" :
87	item[5].SetValue("")
88	item[5].Hide()
89	item[6].SetValue("")
90	item[6].Hide()
91	self.text2_min.Hide()
92	self.text2_max.Hide()
93	else:
94	item[5].Show(True)
95	item[6].Show(True)
96	self.text2_min.Show(True)
97	self.text2_max.Show(True)
98	for item in self.fittable_param:
99	if item[5]!=None and item[6]!=None and not item in self.orientation_params_disp:
100	if event.type =="scipy" and not item in self.orientation_params:
101	item[5].SetValue("")
102	item[5].Hide()
103	item[6].SetValue("")
104	item[6].Hide()
105	self.text2_min.Hide()
106	self.text2_max.Hide()
107	self.text_disp_min.Hide()
108	self.text_disp_max.Hide()
109	else:
110	item[5].Show(True)
111	item[6].Show(True)
112	self.text2_min.Show(True)
113	self.text2_max.Show(True)
114	self.text_disp_min.Show(True)
115	self.text_disp_max.Show(True)
116
117	for item in self.orientation_params:
118	if item[5]!=None and item[6]!=None:
119	if event.type =="scipy" or self.data.__class__.__name__ !="Data2D":
120	item[5].SetValue("")
121	item[5].Hide()
122	item[6].SetValue("")
123	item[6].Hide()
124	else:
125	item[5].Show(True)
126	item[6].Show(True)
127
128	for item in self.orientation_params_disp:
129	if item[5]!=None and item[6]!=None:
130	if event.type =="scipy" or self.data.__class__.__name__ !="Data2D":
131	item[5].SetValue("")
132	item[5].Hide()
133	item[6].SetValue("")
134	item[6].Hide()
135	else:
136	item[5].Show(True)
137	item[6].Show(True)
138
139	self.sizer3.Layout()
140	self.SetScrollbars(20,20,25,65)
141
142
143	def _fill_range_sizer(self):
144	"""
145	Fill the sizer containing the plotting range
146	add access to npts
147	"""
148	sizer_fit = wx.GridSizer(1, 1,0, 0)
149
150	self.btFit = wx.Button(self,wx.NewId(),'Fit')
151	self.btFit.Bind(wx.EVT_BUTTON, self._onFit,id= self.btFit.GetId())
152	self.btFit.SetToolTipString("Perform fit.")
153
154
155	sizer_fit.Add((5,5),1, wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 5)
156	sizer_fit.Add(self.btFit,0, wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 5)
157
158	sizer_smearer = wx.BoxSizer(wx.HORIZONTAL)
159	#Filling the sizer containing instruments smearing info.
160	self.disable_smearer = wx.RadioButton(self, -1, 'No', style=wx.RB_GROUP)
161	self.enable_smearer = wx.RadioButton(self, -1, 'Yes')
162	self.Bind(wx.EVT_RADIOBUTTON, self.onSmear, id=self.disable_smearer.GetId())
163	self.Bind(wx.EVT_RADIOBUTTON, self.onSmear, id=self.enable_smearer.GetId())
164
165	sizer_smearer.Add(wx.StaticText(self,-1,'Instrument Smearing? '))
166	sizer_smearer.Add((10, 10))
167	sizer_smearer.Add( self.enable_smearer )
168	sizer_smearer.Add((10,10))
169	sizer_smearer.Add( self.disable_smearer )
170
171	#Display Chi^2/dof
172	sizer_smearer.Add((68,10))
173	box_description= wx.StaticBox(self, -1,'Chi2/dof')
174	boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
175	boxsizer1.SetMinSize((60,-1))
176	temp_smearer = None
177	if self.enable_smearer.GetValue():
178	temp_smearer= self.smearer
179
180	self.tcChi = wx.StaticText(self, -1, "-", style=wx.ALIGN_LEFT)
181
182	boxsizer1.Add( self.tcChi )
183	sizer_smearer.Add( boxsizer1 )
184
185	#Set sizer for Fitting section
186	self._set_range_sizer( title="Fitting",object1=sizer_smearer, object= sizer_fit)
187
188
189	def _fill_datainfo_sizer(self):
190	"""
191	fill sizer 0 with data info
192	"""
193	self.sizer0.Clear(True)
194	## no loaded data , don't fill the sizer
195	if self.data== None:
196	self.sizer0.Layout()
197	return
198
199	box_description= wx.StaticBox(self, -1, 'Data')
200	boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
201	#----------------------------------------------------------
202	sizer_data = wx.GridSizer(3, 3,5, 5)
203	#Filling the sizer containing data related fields
204	DataSource =wx.StaticText(self, -1,str(self.data.name))
205
206	sizer_data.Add(wx.StaticText(self, -1, 'Source Name : '))
207	sizer_data.Add(DataSource )
208	sizer_data.Add( (0,5) )
209
210	#---------sizer 2 draw--------------------------------
211	#set maximum range for x in linear scale
212	if not hasattr(self.data,"data"): #Display only for 1D data fit
213	# Minimum value of data
214	data_min = min(self.data.x)
215	# Maximum value of data
216	data_max = max(self.data.x)
217	text4_3 = wx.StaticText(self, -1, 'Total Q Range (1/A)',
218	style=wx.ALIGN_LEFT)
219	sizer_data.Add( text4_3 )
220	sizer_data.Add(wx.StaticText(self, -1, "Min : %s"%str(data_min)))
221	sizer_data.Add(wx.StaticText(self, -1, "Max : %s"%str(data_max)))
222
223	else:
224	## Minimum value of data
225	data_min= 0
226	x= max(math.fabs(self.data.xmin), math.fabs(self.data.xmax))
227	y= max(math.fabs(self.data.ymin), math.fabs(self.data.ymax))
228	## Maximum value of data
229	data_max = math.sqrt(xx + yy)
230
231	#For qmin and qmax, do not use format_number
232	#.(If do, qmin and max could be different from what is in the data.)
233	text4_3 = wx.StaticText(self, -1, "Total Q Range (1/A)",
234	style=wx.ALIGN_LEFT)
235	sizer_data.Add( text4_3 )
236	sizer_data.Add(wx.StaticText(self, -1, "Min : %s"%str(data_min)))
237	sizer_data.Add(wx.StaticText(self, -1, "Max : %s"%str(data_max)))
238	## set q range to plot
239	self.qmin_x= data_min
240	self.qmax_x= data_max
241
242	boxsizer1.Add(sizer_data)
243	#------------------------------------------------------------
244	self.sizer0.Add(boxsizer1,0, wx.EXPAND \| wx.ALL, 10)
245	self.sizer0.Layout()
246
247	def _fill_model_sizer(self, sizer):
248	"""
249	fill sizer containing model info
250	"""
251	##Add model function Details button in fitpanel.
252	id = wx.NewId()
253	self.model_help =wx.Button(self,id,'Details')
254	self.model_help.Bind(wx.EVT_BUTTON, self.on_model_help_clicked,id=id)
255	self.model_help.SetToolTipString("Model Function Help")
256
257	## class base method to add view 2d button
258	self._set_model_sizer(sizer=sizer, title="Model",object=self.model_help )
259
260
261	#def _set_sizer_gaussian(self):
262	def _set_sizer_dispersion(self, dispersity):
263	"""
264	draw sizer with gaussian dispersity parameters
265	"""
266	self.fittable_param=[]
267	self.fixed_param=[]
268	self.orientation_params_disp=[]
269
270	self.sizer4_4.Clear(True)
271	if self.model==None:
272	##no model is selected
273	return
274	if not self.enable_disp.GetValue():
275	## the user didn't select dispersity display
276	return
277
278	self._reset_dispersity()
279	# Create the dispersion objects
280	for item in self.model.dispersion.keys():
281	#disp_model = GaussianDispersion()
282	disp_model = dispersity()
283	self._disp_obj_dict[item] = disp_model
284	self.model.set_dispersion(item, disp_model)
285	self.state._disp_obj_dict[item]= disp_model
286
287	ix=0
288	iy=1
289	disp = wx.StaticText(self, -1, 'Names')
290	self.sizer4_4.Add(disp,( iy, ix),(1,1),
291	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
292	ix += 1
293	values = wx.StaticText(self, -1, 'Sigmas (STD)')
294	self.sizer4_4.Add(values,( iy, ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
295	ix +=2
296	self.text_disp_1 = wx.StaticText(self, -1, '')
297	self.sizer4_4.Add( self.text_disp_1,(iy, ix),(1,1),\
298	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
299	self.text_disp_1.Hide()
300
301
302	ix +=1
303	self.text_disp_min = wx.StaticText(self, -1, 'Min')
304	self.sizer4_4.Add(self.text_disp_min,(iy, ix),(1,1),\
305	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
306	self.text_disp_min.Hide()
307	ix +=1
308	self.text_disp_max = wx.StaticText(self, -1, 'Max')
309	self.sizer4_4.Add(self.text_disp_max,(iy, ix),(1,1),\
310	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
311	self.text_disp_max.Hide()
312
313
314
315
316
317
318	ix += 1
319	npts = wx.StaticText(self, -1, 'Npts')
320	self.sizer4_4.Add(npts,( iy, ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
321	ix += 1
322	nsigmas = wx.StaticText(self, -1, 'Nsigmas')
323	self.sizer4_4.Add(nsigmas,( iy, ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
324
325	if self.engine_type=="park":
326	self.text_disp_max.Show(True)
327	self.text_disp_min.Show(True)
328
329	for item in self.model.dispersion.keys():
330	if not item in self.model.orientation_params:
331	self.disp_cb_dict[item]= None
332	name1=item+".width"
333	name2=item+".npts"
334	name3=item+".nsigmas"
335	iy += 1
336	for p in self.model.dispersion[item].keys():
337
338	if p=="width":
339	ix = 0
340	cb = wx.CheckBox(self, -1, name1, (10, 10))
341	wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
342	self.sizer4_4.Add( cb,( iy, ix),(1,1),
343	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
344	ix = 1
345	value= self.model.getParam(name1)
346	ctl1 = BasicPage.ModelTextCtrl(self, -1, size=(_BOX_WIDTH,20),
347	style=wx.TE_PROCESS_ENTER)
348	ctl1.SetValue(str (format_number(value)))
349	self.sizer4_4.Add(ctl1, (iy,ix),(1,1),wx.EXPAND)
350	## text to show error sign
351	ix = 2
352	text2=wx.StaticText(self, -1, '+/-')
353	self.sizer4_4.Add(text2,(iy, ix),(1,1),
354	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
355	text2.Hide()
356	## txtcrtl to add error from fit
357	ix = 3
358	ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
359	self.sizer4_4.Add(ctl2, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
360	ctl2.Hide()
361
362	#Prepare add min and max in model.details.(Maybe Not correct way)
363	if not self.model.details.has_key(name1):
364	self.model.details [name1] = ["",None, None]
365	param_min, param_max= self.model.details[name1][1:]
366	ix = 4
367	ctl3 = BasicPage.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
368	kill_focus_callback = self._on_paramRangeEnter,
369	set_focus_callback = self._on_paramRangeEnter)
370	if param_min ==None:
371	ctl3.SetValue("")
372	else:
373	ctl3.SetValue(str(param_min))
374	self.sizer4_4.Add(ctl3, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
375	ctl3.Hide()
376
377	ix = 5
378	ctl4 = BasicPage.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
379	kill_focus_callback = self._on_paramRangeEnter,
380	set_focus_callback = self._on_paramRangeEnter)
381	self.sizer4_4.Add(ctl4, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
382	if param_max==None:
383	ctl4.SetValue("")
384	else:
385	ctl4.SetValue(str(param_max))
386	ctl4.Hide()
387
388	if self.engine_type=="park":
389	ctl3.Show(True)
390	ctl4.Show(True)
391
392	self.fittable_param.append([cb,name1,ctl1,text2,
393	ctl2, ctl3, ctl4,None])
394	#self.parameters.append([cb,name1, ctl1,text2,ctl2, ctl3, ctl4,None])
395	#self.orientation_params.append([cb,name1, ctl1,text2,ctl2, ctl3, ctl4,None])
396
397
398	elif p=="npts":
399	ix = 6
400	value= self.model.getParam(name2)
401	Tctl = BasicPage.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20),
402	style=wx.TE_PROCESS_ENTER)
403
404	Tctl.SetValue(str (format_number(value)))
405	self.sizer4_4.Add(Tctl, (iy,ix),(1,1),
406	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
407	self.fixed_param.append([None,name2, Tctl,None,None,
408	None, None,None])
409	elif p=="nsigmas":
410	ix = 7
411	value= self.model.getParam(name3)
412	Tctl = BasicPage.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20),
413	style=wx.TE_PROCESS_ENTER)
414	Tctl.SetValue(str (format_number(value)))
415	self.sizer4_4.Add(Tctl, (iy,ix),(1,1),
416	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
417	ix +=1
418	self.sizer4_4.Add((20,20), (iy,ix),(1,1),
419	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
420
421	self.fixed_param.append([None,name3, Tctl
422	,None,None,None, None,None])
423
424	ix =0
425	iy +=1
426	self.sizer4_4.Add((20,20),(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
427	for item in self.model.dispersion.keys():
428	if item in self.model.orientation_params:
429	self.disp_cb_dict[item]= None
430	name1=item+".width"
431	name2=item+".npts"
432	name3=item+".nsigmas"
433	iy += 1
434	for p in self.model.dispersion[item].keys():
435
436	if p=="width":
437	ix = 0
438	cb = wx.CheckBox(self, -1, name1, (10, 10))
439	wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
440	self.sizer4_4.Add( cb,( iy, ix),(1,1),
441	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
442	if self.data.__class__.__name__ =="Data2D":
443	cb.Show(True)
444	else:
445	cb.Hide()
446	ix = 1
447	value= self.model.getParam(name1)
448	ctl1 = BasicPage.ModelTextCtrl(self, -1, size=(_BOX_WIDTH,20),
449	style=wx.TE_PROCESS_ENTER)
450	ctl1.SetValue(str (format_number(value)))
451	if self.data.__class__.__name__ =="Data2D":
452	ctl1.Show(True)
453	else:
454	ctl1.Hide()
455	self.sizer4_4.Add(ctl1, (iy,ix),(1,1),wx.EXPAND)
456	## text to show error sign
457	ix = 2
458	text2=wx.StaticText(self, -1, '+/-')
459	self.sizer4_4.Add(text2,(iy, ix),(1,1),
460	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
461	text2.Hide()
462	## txtcrtl to add error from fit
463	ix = 3
464	ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
465	self.sizer4_4.Add(ctl2, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
466	ctl2.Hide()
467	if self.data.__class__.__name__ =="Data2D":
468	ctl2.Enable()
469	else:
470	ctl2.Disable()
471
472
473	if not self.model.details.has_key(name1):
474	self.model.details [name1] = ["",None, None]
475
476	param_min, param_max= self.model.details[name1][1:]
477	ix = 4
478	ctl3 = BasicPage.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
479	kill_focus_callback = self._onparamRangeEnter,
480	set_focus_callback = self._onparamRangeEnter)
481	if param_min ==None:
482	ctl3.SetValue("")
483	else:
484	ctl3.SetValue(str(param_min))
485	self.sizer4_4.Add(ctl3, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
486	ctl3.Hide()
487	if self.data.__class__.__name__ =="Data2D":
488	ctl3.Enable()
489	else:
490	ctl3.Disable()
491
492	ix = 5
493	ctl4 = BasicPage.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
494	kill_focus_callback = self._onparamRangeEnter,
495	set_focus_callback = self._onparamRangeEnter)
496	self.sizer4_4.Add(ctl4, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
497	if param_max==None:
498	ctl4.SetValue("")
499	else:
500	ctl4.SetValue(str(param_max))
501	ctl4.Hide()
502	if self.data.__class__.__name__ =="Data2D":
503	ctl4.Enable()
504	else:
505	ctl4.Disable()
506
507	if self.engine_type=="park" and self.data.__class__.__name__ =="Data2D":
508	ctl3.Show(True)
509	ctl4.Show(True)
510
511
512
513
514	self.fittable_param.append([cb,name1,ctl1,text2,
515	ctl2, ctl3, ctl4,None])
516	#self.parameters.append([cb,name1, ctl1,text2,ctl2, ctl3, ctl4,None])
517	#self.orientation_params.append([cb,name1,ctl1,text2, ctl2, ctl3, ctl4,None])
518	self.orientation_params_disp.append([cb,name1,ctl1,text2,
519	ctl2, ctl3, ctl4,None])
520	elif p=="npts":
521	ix = 6
522	value= self.model.getParam(name2)
523	Tctl = BasicPage.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20),
524	style=wx.TE_PROCESS_ENTER)
525
526	Tctl.SetValue(str (format_number(value)))
527	if self.data.__class__.__name__ =="Data2D":
528	Tctl.Show(True)
529	else:
530	Tctl.Hide()
531	self.sizer4_4.Add(Tctl, (iy,ix),(1,1),
532	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
533	self.fixed_param.append([None,name2, Tctl,None,None,
534	None, None,None])
535	self.orientation_params_disp.append([None,name2, Tctl,None,None,
536	None, None,None])
537	elif p=="nsigmas":
538	ix = 7
539	value= self.model.getParam(name3)
540	Tctl = BasicPage.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20),
541	style=wx.TE_PROCESS_ENTER)
542	Tctl.SetValue(str (format_number(value)))
543	if self.data.__class__.__name__ =="Data2D":
544	Tctl.Show(True)
545	else:
546	Tctl.Hide()
547	self.sizer4_4.Add(Tctl, (iy,ix),(1,1),
548	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
549	ix +=1
550	#self.sizer4_4.Add((20,20), (iy,ix),(1,1),
551	#wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
552	self.fixed_param.append([None,name3, Tctl
553	,None,None, None, None,None])
554
555	self.orientation_params_disp.append([None,name3, Tctl
556	,None,None, None, None,None])
557
558	self.state.disp_cb_dict = copy.deepcopy(self.disp_cb_dict)
559	self.state.model = self.model.clone()
560	## save state into
561
562	self._copy_parameters_state(self.parameters, self.state.parameters)
563	self._copy_parameters_state(self.orientation_params_disp,
564	self.state.orientation_params_disp)
565	self._copy_parameters_state(self.fittable_param, self.state.fittable_param)
566	self._copy_parameters_state(self.fixed_param, self.state.fixed_param)
567
568
569	wx.PostEvent(self.parent, StatusEvent(status=\
570	" Selected Distribution: Gaussian"))
571	ix =0
572	iy +=1
573	#self.sizer4_4.Add((20,20),(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
574	self.sizer4_4.Layout()
575	self.sizer4.Layout()
576	self.Layout()
577	self.SetScrollbars(20,20,25,65)
578	self.Refresh()
579
580
581	def _onFit(self, event):
582	"""
583	Allow to fit
584	"""
585	from sans.guiframe.utils import check_value
586	flag = check_value( self.qmin, self.qmax)
587
588	if not flag:
589	msg= "Fitting range invalid"
590	wx.PostEvent(self.parent.parent, StatusEvent(status= msg ))
591	return
592
593	if len(self.param_toFit) <= 0:
594	msg= "Select at least one parameter to fit"
595	wx.PostEvent(self.parent.parent, StatusEvent(status= msg ))
596	return
597
598	# Remove or do not allow fitting on the Q=0 point, especially when y(q=0)=None at x[0].
599	#ToDo: Fix this.
600	self.qmin_x = float(self.qmin.GetValue())
601	self.qmax_x =float( self.qmax.GetValue())
602	self.manager._reset_schedule_problem( value=0)
603	self.manager.schedule_for_fit( value=1,page=self,fitproblem =None)
604	self.manager.set_fit_range(page= self,qmin= self.qmin_x, qmax= self.qmax_x)
605
606	#single fit
607	self.manager.onFit()
608	## allow stopping the fit
609	#if self.engine_type=="scipy":
610	# self.btFit.SetLabel("Stop")
611	# self.btFit.Unbind(event=wx.EVT_BUTTON, id= self.btFit.GetId())
612	# self.btFit.Bind(event= wx.EVT_BUTTON, handler=self._StopFit, id=self.btFit.GetId())
613	#else:
614	# self.btFit.SetLabel("Fit")
615	# self.btFit.Bind(event= wx.EVT_BUTTON, handler=self._onFit, id=self.btFit.GetId())
616	self.btFit.SetFocus()
617	self.sizer5.Layout()
618	self.SetScrollbars(20,20,55,40)
619
620	def _StopFit(self, event):
621	"""
622	Stop fit
623	"""
624	self.btFit.SetLabel("Fit")
625	if self.engine_type=="scipy":
626	self.manager.stop_fit()
627	self.btFit.Unbind(event=wx.EVT_BUTTON, id=self.btFit.GetId())
628	self.btFit.Bind(event=wx.EVT_BUTTON, handler=self._onFit,id=self.btFit.GetId())
629
630
631	self.btFit.SetFocus()
632	self.sizer5.Layout()
633	self.SetScrollbars(20,20,55,40)
634
635	def _on_select_model(self, event):
636	"""
637	call back for model selection
638	"""
639	self._on_select_model_helper()
640	self.set_model_param_sizer(self.model)
641
642	self.enable_disp.SetValue(False)
643	self.disable_disp.SetValue(True)
644	self.set_dispers_sizer()
645	if self.model !=None:
646	try:
647	temp_smear= None
648	if self.enable_smearer.GetValue():
649	temp_smear= self.smearer
650	self.compute_chisqr(temp_smear)
651	except:
652	## error occured on chisqr computation
653	pass
654	## set smearing value whether or not the data contain the smearing info
655	self.manager.set_smearer(smearer=temp_smear, qmin= float(self.qmin_x),
656	qmax= float(self.qmax_x))
657	evt = ModelEventbox(model=self.model)
658	wx.PostEvent(self.event_owner, evt)
659	self.btFit.SetFocus()
660	self.state.enable_disp = self.enable_disp.GetValue()
661	self.state.disable_disp = self.disable_disp.GetValue()
662
663	self.state.structurecombobox = self.structurebox.GetCurrentSelection()
664	self.state.formfactorcombobox = self.formfactorbox.GetCurrentSelection()
665
666
667	if event !=None:
668	#self._undo.Enable(True)
669	## post state to fit panel
670	event = PageInfoEvent(page = self)
671	wx.PostEvent(self.parent, event)
672
673
674	def _onparamRangeEnter(self, event):
675	"""
676	Check validity of value enter in the parameters range field
677	"""
678	tcrtl= event.GetEventObject()
679	if tcrtl.GetValue().lstrip().rstrip()!="":
680	try:
681	value = float(tcrtl.GetValue())
682	tcrtl.SetBackgroundColour(wx.WHITE)
683	tcrtl.Refresh()
684	except:
685	tcrtl.SetBackgroundColour("pink")
686	tcrtl.Refresh()
687	return
688	else:
689	tcrtl.SetBackgroundColour(wx.WHITE)
690	tcrtl.Refresh()
691	self._onparamEnter_helper()
692
693	##compute chisqr for range change
694	temp_smearer = None
695	if self.enable_smearer.GetValue():
696	msg=""
697	temp_smearer= self.smearer
698	##Calculate chi2
699
700	self.compute_chisqr(smearer= temp_smearer)
701	## new state posted
702	if self.state_change:
703	#self._undo.Enable(True)
704	event = PageInfoEvent(page = self)
705	wx.PostEvent(self.parent, event)
706	self.state_change= False
707
708
709	def _onparamEnter(self,event):
710	"""
711	when enter value on panel redraw model according to changed
712	"""
713	tcrtl= event.GetEventObject()
714
715	if check_float(tcrtl):
716	self._onparamEnter_helper()
717	temp_smearer = None
718	if self.enable_smearer.GetValue():
719	temp_smearer= self.smearer
720
721	self.compute_chisqr(smearer= temp_smearer)
722	## new state posted
723	if self.state_change:
724	#self._undo.Enable(True)
725	event = PageInfoEvent(page = self)
726	wx.PostEvent(self.parent, event)
727	self.state_change= False
728	else:
729	msg= "Cannot Plot :Must enter a number!!! "
730	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
731	return
732
733	### Same as _onparamRangeEnter but smear and chi^2 calculations are removed
734	### and _onparamEnter_helper is not used, which causing mess sizer display
735	def _on_paramRangeEnter(self, event):
736	"""
737	Check validity of value enter in the parameters range field
738	"""
739	tcrtl= event.GetEventObject()
740	if tcrtl.GetValue().lstrip().rstrip()!="":
741	try:
742	value = float(tcrtl.GetValue())
743	tcrtl.SetBackgroundColour(wx.WHITE)
744	tcrtl.Refresh()
745	except:
746	tcrtl.SetBackgroundColour("pink")
747	tcrtl.Refresh()
748	return
749	else:
750	tcrtl.SetBackgroundColour(wx.WHITE)
751	tcrtl.Refresh()
752	self._onparamEnter_helper()
753
754	## new state posted
755	if self.state_change:
756	#self._undo.Enable(True)
757	event = PageInfoEvent(page = self)
758	wx.PostEvent(self.parent, event)
759	self.state_change= False
760
761
762
763	def set_data(self, data ):
764	"""
765	reset the current data
766	"""
767	if data ==None:
768	return
769	self.data =data
770	self.state.data= data
771	self._fill_datainfo_sizer()
772	self.SetScrollbars(20,20,25,65)
773	self.Layout()
774
775	def reset_page(self, state,first=False):
776	"""
777	reset the state
778	"""
779	#if first:
780	# self._undo.Enable(False)
781
782	self.reset_page_helper(state)
783	evt = ModelEventbox(model=state.model)
784	wx.PostEvent(self.event_owner, evt)
785
786
787	def get_range(self):
788	"""
789	return the fitting range
790	"""
791	return float(self.qmin_x) , float(self.qmax_x)
792
793	def get_param_list(self):
794	"""
795	@return self.param_toFit: list containing references to TextCtrl
796	checked.Theses TextCtrl will allow reference to parameters to fit.
797	@raise: if return an empty list of parameter fit will nnote work
798	properly so raise ValueError,"missing parameter to fit"
799	"""
800	if self.param_toFit !=[]:
801	return self.param_toFit
802	else:
803	raise ValueError,"missing parameter to fit"
804
805	def onsetValues(self,chisqr, out,cov):
806	"""
807	Build the panel from the fit result
808	@param chisqr:Value of the goodness of fit metric
809	@param out:list of parameter with the best value found during fitting
810	@param cov:Covariance matrix
811
812	"""
813	self.tcChi.SetLabel(format_number(chisqr))
814	params = {}
815	is_modified = False
816	has_error = False
817	self.text2_3.Hide()
818	try:
819	n = self.disp_box.GetCurrentSelection()
820	dispersity= self.disp_box.GetClientData(n)
821	name= dispersity.__name__
822	if name == "GaussianDispersion":
823	if hasattr(self,"text_disp_1" ):
824	if self.text_disp_1 !=None:
825	self.text_disp_1.Hide()
826	except:
827	pass
828	#set the panel when fit result are float not list
829	if out.__class__==numpy.float64:
830	self.param_toFit[0][2].SetValue(format_number(out))
831	self.param_toFit[0][2].Refresh()
832
833	self.param_toFit[0][4].Clear()
834	self.param_toFit[0][4].Hide()
835	if cov !=None :
836	self.text2_3.Show(True)
837	try:
838	name= dispersity.__name__
839	if name == "GaussianDispersion":
840	if hasattr(self,"text_disp_1" ):
841	if self.text_disp_1 !=None:
842	self.text_disp_1.Show(True)
843	except:
844	pass
845	if cov[0]==None or not numpy.isfinite(cov[0]):
846	self.param_toFit[0][3].Hide()
847	self.param_toFit[0][4].Clear()
848	self.param_toFit[0][4].Hide()
849	self.param_toFit[0][4].Refresh()
850	else:
851	self.param_toFit[0][3].Show(True)
852	self.param_toFit[0][4].Clear()
853	self.param_toFit[0][4].SetValue(format_number(cov[0]))
854	self.param_toFit[0][4].Show(True)
855	self.param_toFit[0][4].Refresh()
856	else:
857	i=0
858	j=0
859	#Set the panel when fit result are list
860	for item in self.param_toFit:
861	## reset error value to initial state
862	item[4].Clear()
863	item[4].Hide()
864	item[4].Refresh()
865	if( out != None ) and len(out)<=len(self.param_toFit)and i < len(out):
866	item[2].SetValue(format_number(self.model.getParam(item[1])))
867	item[2].Refresh()
868	if(cov !=None)and len(cov)<=len(self.param_toFit)and i < len(cov):
869	self.text2_3.Show(True)
870	try:
871	name= dispersity.__name__
872	if name == "GaussianDispersion":
873	if hasattr(self,"text_disp_1" ):
874	if self.text_disp_1!=None:
875	self.text_disp_1.Show(True)
876	except:
877	pass
878	item[3].Show(True)
879	item[4].Clear()
880	for j in range(len(out)):
881	if out[j]==self.model.getParam(item[1]):
882	break
883	## unable to compare cov[j]==numpy.nan so switch to None
884	if cov[j]==None or not numpy.isfinite(cov[j]):
885	item[3].Hide()
886	item[4].Refresh()
887	item[4].Clear()
888	item[4].Hide()
889	else:
890	item[4].SetValue(format_number(cov[j]))
891	item[4].Refresh()
892	item[4].Show(True)
893	i+=1
894
895	self.sizer3.Layout()
896	self.sizer4.Layout()
897	self.Layout()
898	self.Refresh()
899	self.SetScrollbars(20,20,25,65)
900
901
902	def onSmear(self, event):
903	"""
904	Create a smear object that will change the way residuals
905	are compute when fitting
906	"""
907	if self.model ==None:
908	msg="Need model and data to smear plot"
909	wx.PostEvent(self.manager.parent, StatusEvent(status=\
910	"Smear: %s"%msg))
911	return
912	temp_smearer = None
913	if self.enable_smearer.GetValue():
914	temp_smearer= self.smearer
915	if hasattr(self.data,"dxl"):
916	msg= ": Resolution smearing parameters"
917	if hasattr(self.data,"dxw"):
918	msg= ": Slit smearing parameters"
919	if self.smearer ==None:
920	wx.PostEvent(self.manager.parent, StatusEvent(status=\
921	"Data contains no smearing information"))
922	else:
923	wx.PostEvent(self.manager.parent, StatusEvent(status=\
924	"Data contains smearing information %s"%msg))
925
926	## set smearing value whether or not the data contain the smearing info
927	self.manager.set_smearer(smearer=temp_smearer, qmin= float(self.qmin_x),
928	qmax= float(self.qmax_x))
929	##Calculate chi2
930	self.compute_chisqr(smearer= temp_smearer)
931
932	self.state.enable_smearer= self.enable_smearer.GetValue()
933	self.state.disable_smearer=self.disable_smearer.GetValue()
934
935	def complete_chisqr(self, output, elapsed=None):
936	"""
937	print result chisqr
938	"""
939	try:
940	if output ==None:
941	output= "-"
942	self.tcChi.SetLabel(str(format_number(output)))
943
944	self.sizer5.Layout()
945	self.Layout()
946	self.Refresh
947	## post state to fit panel
948	self.state.tcChi =output
949
950	except:
951	pass
952
953
954	def compute_chisqr1D(self, smearer=None):
955	"""
956	Compute chisqr for 1D
957	"""
958	from sans.guiframe.utils import check_value
959	flag = check_value( self.qmin, self.qmax)
960
961	if not flag:
962	return
963
964	try:
965	self.qmin_x = float(self.qmin.GetValue())
966	self.qmax_x = float(self.qmax.GetValue())
967	##return residuals within self.qmin_x and self.qmax_x
968	from gui_thread import CalcChisqr1D
969	## If a thread is already started, stop it
970	if self.calc_Chisqr!= None and self.calc_Chisqr.isrunning():
971	self.calc_Chisqr.stop()
972
973	self.calc_Chisqr= CalcChisqr1D( data1d= self.data,
974	model= self.model,
975	smearer=smearer,
976	qmin=self.qmin_x,
977	qmax=self.qmax_x,
978	completefn = self.complete_chisqr,
979	updatefn = None)
980
981	self.calc_Chisqr.queue()
982
983	except:
984	raise ValueError," Could not compute Chisqr for %s Model 2D: "%self.model.name
985
986
987
988
989
990	def compute_chisqr2D(self):
991	"""
992	compute chi square given a model and data 2D and set the value
993	to the tcChi txtcrl
994	"""
995	from sans.guiframe.utils import check_value
996	flag = check_value( self.qmin, self.qmax)
997	if not flag:
998	return
999
1000	try:
1001	self.qmin_x = float(self.qmin.GetValue())
1002	self.qmax_x = float(self.qmax.GetValue())
1003
1004	##return residuals within self.qmin_x and self.qmax_x
1005	from gui_thread import CalcChisqr2D
1006	## If a thread is already started, stop it
1007	if self.calc_Chisqr!= None and self.calc_Chisqr.isrunning():
1008	self.calc_Chisqr.stop()
1009
1010	self.calc_Chisqr= CalcChisqr2D( data2d= self.data,
1011	model= self.model,
1012	qmin= self.qmin_x,
1013	qmax = self.qmax_x,
1014	completefn = self.complete_chisqr,
1015	updatefn = None)
1016
1017	self.calc_Chisqr.queue()
1018
1019	except:
1020	raise
1021
1022
1023	def compute_chisqr(self , smearer=None):
1024	"""
1025	compute chi square given a model and data 1D and set the value
1026	to the tcChi txtcrl
1027	"""
1028	from sans.guiframe.utils import check_value
1029	flag = check_value( self.qmin, self.qmax)
1030	if flag== True:
1031	try:
1032	if hasattr(self.data,"data"):
1033	self.compute_chisqr2D()
1034	return
1035	else:
1036	self.compute_chisqr1D(smearer=smearer)
1037	return
1038	except:
1039	wx.PostEvent(self.parent.parent, StatusEvent(status=\
1040	"Chisqr Error: %s"% sys.exc_value))
1041	return
1042
1043
1044	def select_all_param(self,event):
1045	"""
1046	set to true or false all checkBox given the main checkbox value cb1
1047	"""
1048
1049	self.param_toFit=[]
1050	if self.parameters !=[]:
1051	if self.cb1.GetValue():
1052	for item in self.parameters:
1053	## for data2D select all to fit
1054	if self.data.__class__.__name__=="Data2D":
1055	item[0].SetValue(True)
1056	self.param_toFit.append(item )
1057	else:
1058	## for 1D all parameters except orientation
1059	if not item in self.orientation_params:
1060	item[0].SetValue(True)
1061	self.param_toFit.append(item )
1062	if len(self.fittable_param)>0:
1063	for item in self.fittable_param:
1064	if self.data.__class__.__name__=="Data2D":
1065	item[0].SetValue(True)
1066	self.param_toFit.append(item )
1067	else:
1068	## for 1D all parameters except orientation
1069	if not item in self.orientation_params_disp:
1070	item[0].SetValue(True)
1071	self.param_toFit.append(item )
1072	else:
1073	for item in self.parameters:
1074	item[0].SetValue(False)
1075	for item in self.fittable_param:
1076	item[0].SetValue(False)
1077	self.param_toFit=[]
1078	self._copy_parameters_state(self.orientation_params,
1079	self.state.orientation_params)
1080	self._copy_parameters_state(self.orientation_params_disp,
1081	self.state.orientation_params_disp)
1082	self._copy_parameters_state(self.parameters, self.state.parameters)
1083	self._copy_parameters_state(self.fittable_param, self.state.fittable_param)
1084	self._copy_parameters_state(self.fixed_param, self.state.fixed_param)
1085
1086	#self.save_current_state()
1087	if event !=None:
1088	#self._undo.Enable(True)
1089	## post state to fit panel
1090	event = PageInfoEvent(page = self)
1091	wx.PostEvent(self.parent, event)
1092
1093
1094
1095	def select_param(self,event):
1096	"""
1097	Select TextCtrl checked for fitting purpose and stores them
1098	in self.param_toFit=[] list
1099	"""
1100	self.param_toFit=[]
1101	for item in self.parameters:
1102	#Select parameters to fit for list of primary parameters
1103	if item[0].GetValue():
1104	if not (item in self.param_toFit):
1105	self.param_toFit.append(item )
1106	else:
1107	#remove parameters from the fitting list
1108	if item in self.param_toFit:
1109	self.param_toFit.remove(item)
1110	#Select parameters to fit for list of fittable parameters with dispersion
1111	for item in self.fittable_param:
1112	if item[0].GetValue():
1113	if not (item in self.param_toFit):
1114	self.param_toFit.append(item)
1115	else:
1116	#remove parameters from the fitting list
1117	if item in self.param_toFit:
1118	self.param_toFit.remove(item)
1119	#Set the value of checkbox that selected every checkbox or not
1120	if len(self.parameters)+len(self.fittable_param) ==len(self.param_toFit):
1121	self.cb1.SetValue(True)
1122	else:
1123	self.cb1.SetValue(False)
1124	## save current state of the page
1125	self.state.cb1 = self.cb1.GetValue()
1126	self._copy_parameters_state(self.orientation_params,
1127	self.state.orientation_params)
1128	self._copy_parameters_state(self.orientation_params_disp,
1129	self.state.orientation_params_disp)
1130	self._copy_parameters_state(self.parameters, self.state.parameters)
1131	self._copy_parameters_state(self.fittable_param, self.state.fittable_param)
1132	self._copy_parameters_state(self.fixed_param, self.state.fixed_param)
1133
1134	#self.save_current_state()
1135	if event !=None:
1136	#self._undo.Enable(True)
1137	## post state to fit panel
1138	event = PageInfoEvent(page = self)
1139	wx.PostEvent(self.parent, event)
1140
1141
1142
1143	def set_model_param_sizer(self, model):
1144	"""
1145	Build the panel from the model content
1146	@param model: the model selected in combo box for fitting purpose
1147	"""
1148	self.sizer3.Clear(True)
1149	self.parameters = []
1150	self.param_toFit=[]
1151	self.fittable_param=[]
1152	self.fixed_param=[]
1153	self.orientation_params=[]
1154	self.orientation_params_disp=[]
1155
1156	if model ==None:
1157	self.sizer3.Layout()
1158	self.SetScrollbars(20,20,25,65)
1159	return
1160	## the panel is drawn using the current value of the fit engine
1161	if self.engine_type==None and self.manager !=None:
1162	self.engine_type= self.manager._return_engine_type()
1163
1164	box_description= wx.StaticBox(self, -1,str("Model Parameters"))
1165	boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
1166	sizer = wx.GridBagSizer(5,5)
1167	## save the current model
1168	self.model = model
1169
1170	keys = self.model.getParamList()
1171	#list of dispersion paramaters
1172	self.disp_list=self.model.getDispParamList()
1173
1174	keys.sort()
1175
1176	iy = 0
1177	ix = 0
1178	self.cb1 = wx.CheckBox(self, -1,"Select all", (10, 10))
1179	wx.EVT_CHECKBOX(self, self.cb1.GetId(), self.select_all_param)
1180	self.cb1.SetValue(False)
1181
1182	sizer.Add(self.cb1,(iy, ix),(1,1),\
1183	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 5)
1184	ix +=1
1185	self.text2_2 = wx.StaticText(self, -1, 'Values')
1186	sizer.Add(self.text2_2,(iy, ix),(1,1),\
1187	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
1188	ix +=2
1189	self.text2_3 = wx.StaticText(self, -1, 'Errors')
1190	sizer.Add(self.text2_3,(iy, ix),(1,1),\
1191	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
1192	self.text2_3.Hide()
1193	ix +=1
1194	self.text2_min = wx.StaticText(self, -1, 'Min')
1195	sizer.Add(self.text2_min,(iy, ix),(1,1),\
1196	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
1197	self.text2_min.Hide()
1198	ix +=1
1199	self.text2_max = wx.StaticText(self, -1, 'Max')
1200	sizer.Add(self.text2_max,(iy, ix),(1,1),\
1201	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
1202	self.text2_max.Hide()
1203	ix += 1
1204	self.text2_4 = wx.StaticText(self, -1, '[Units]')
1205	sizer.Add(self.text2_4,(iy, ix),(1,1),\
1206	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
1207	self.text2_4.Hide()
1208	if self.engine_type=="park":
1209	self.text2_max.Show(True)
1210	self.text2_min.Show(True)
1211
1212	for item in keys:
1213	if not item in self.disp_list and not item in self.model.orientation_params:
1214
1215	iy += 1
1216	ix = 0
1217	## add parameters name with checkbox for selecting to fit
1218	cb = wx.CheckBox(self, -1, item )
1219	cb.SetValue(False)
1220	wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
1221	sizer.Add( cb,( iy, ix),(1,1),
1222	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 5)
1223
1224	## add parameter value
1225	ix += 1
1226	value= self.model.getParam(item)
1227	ctl1 = BasicPage.ModelTextCtrl(self, -1, size=(_BOX_WIDTH,20),
1228	style=wx.TE_PROCESS_ENTER)
1229
1230	ctl1.SetValue(format_number(value))
1231	sizer.Add(ctl1, (iy,ix),(1,1), wx.EXPAND)
1232	## text to show error sign
1233	ix += 1
1234	text2=wx.StaticText(self, -1, '+/-')
1235	sizer.Add(text2,(iy, ix),(1,1),\
1236	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
1237	text2.Hide()
1238	## txtcrtl to add error from fit
1239	ix += 1
1240	ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
1241	sizer.Add(ctl2, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
1242	ctl2.Hide()
1243
1244	param_min, param_max= self.model.details[item][1:]
1245	ix += 1
1246	ctl3 = BasicPage.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
1247	kill_focus_callback = self._on_paramRangeEnter,
1248	set_focus_callback = self._on_paramRangeEnter)
1249	if param_min ==None:
1250	ctl3.SetValue("")
1251	else:
1252	ctl3.SetValue(str(param_min))
1253	sizer.Add(ctl3, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
1254	ctl3.Hide()
1255
1256	ix += 1
1257	ctl4 = BasicPage.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
1258	kill_focus_callback = self._on_paramRangeEnter,
1259	set_focus_callback = self._on_paramRangeEnter)
1260	sizer.Add(ctl4, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
1261	if param_max==None:
1262	ctl4.SetValue("")
1263	else:
1264	ctl4.SetValue(str(param_max))
1265	ctl4.Hide()
1266
1267	if self.engine_type=="park":
1268	ctl3.Show(True)
1269	ctl4.Show(True)
1270	ix +=1
1271	# Units
1272	try:
1273	units = wx.StaticText(self, -1, self.model.details[item][0], style=wx.ALIGN_LEFT)
1274	except:
1275	units = wx.StaticText(self, -1, "", style=wx.ALIGN_LEFT)
1276	sizer.Add(units, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
1277
1278	##[cb state, name, value, "+/-", error of fit, min, max , units]
1279	self.parameters.append([cb,item, ctl1,
1280	text2,ctl2, ctl3, ctl4,units])
1281
1282	iy+=1
1283	sizer.Add((10,10),(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
1284
1285	# type can be either Guassian or Array
1286	if len(self.model.dispersion.values())>0:
1287	type= self.model.dispersion.values()[0]["type"]
1288	else:
1289	type = "Gaussian"
1290
1291	iy += 1
1292	ix = 0
1293	#Add tile for orientational angle
1294	for item in keys:
1295	if item in self.model.orientation_params:
1296	orient_angle = wx.StaticText(self, -1, '[For 2D only]:')
1297	sizer.Add(orient_angle,(iy, ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
1298	if not self.data.__class__.__name__ =="Data2D":
1299	orient_angle.Hide()
1300	else:
1301	orient_angle.Show(True)
1302	break
1303
1304	#For Gaussian only
1305	if type.lower() != "array":
1306	for item in self.model.orientation_params:
1307	if not item in self.disp_list:
1308	iy += 1
1309	ix = 0
1310	## add parameters name with checkbox for selecting to fit
1311	cb = wx.CheckBox(self, -1, item )
1312	cb.SetValue(False)
1313	wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
1314	if self.data.__class__.__name__ =="Data2D":
1315	cb.Show(True)
1316	#cb.Enable()
1317	else:
1318	cb.Hide()
1319	#cb.Disable()
1320	sizer.Add( cb,( iy, ix),(1,1),
1321	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 5)
1322
1323	## add parameter value
1324	ix += 1
1325	value= self.model.getParam(item)
1326	ctl1 = BasicPage.ModelTextCtrl(self, -1, size=(_BOX_WIDTH,20),
1327	style=wx.TE_PROCESS_ENTER)
1328
1329	ctl1.SetValue(format_number(value))
1330	if self.data.__class__.__name__ =="Data2D":
1331	ctl1.Show(True)
1332	#ctl1.Enable()
1333	else:
1334	ctl1.Hide()
1335	#ctl1.Disable()
1336	sizer.Add(ctl1, (iy,ix),(1,1), wx.EXPAND)
1337	## text to show error sign
1338	ix += 1
1339	text2=wx.StaticText(self, -1, '+/-')
1340	sizer.Add(text2,(iy, ix),(1,1),\
1341	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
1342	text2.Hide()
1343	## txtcrtl to add error from fit
1344	ix += 1
1345	ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
1346	sizer.Add(ctl2, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
1347	ctl2.Hide()
1348	if self.data.__class__.__name__ =="Data2D":
1349	ctl1.Show(True)
1350	#ctl1.Enable()
1351	else:
1352	ctl1.Hide()
1353	#ctl1.Disable()
1354	param_min, param_max= self.model.details[item][1:]
1355	ix += 1
1356	ctl3 = BasicPage.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
1357	kill_focus_callback = self._onparamRangeEnter,
1358	set_focus_callback = self._onparamRangeEnter)
1359	if param_min ==None:
1360	ctl3.SetValue("")
1361	else:
1362	ctl3.SetValue(str(param_min))
1363	sizer.Add(ctl3, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
1364	ctl3.Hide()
1365	if self.data.__class__.__name__ =="Data2D":
1366	ctl3.Show(True)
1367	#ctl3.Enable()
1368	else:
1369	ctl3.Hide()
1370	#ctl3.Disable()
1371	ix += 1
1372	ctl4 = BasicPage.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
1373	kill_focus_callback = self._onparamRangeEnter,
1374	set_focus_callback = self._onparamRangeEnter)
1375	sizer.Add(ctl4, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
1376	if param_max ==None:
1377	ctl4.SetValue("")
1378	else:
1379	ctl4.SetValue(str(param_max))
1380	ctl4.Hide()
1381	if self.data.__class__.__name__ =="Data2D":
1382	ctl4.Show(True)
1383	#ctl4.Enable()
1384	else:
1385	ctl4.Hide()
1386	#ctl4.Disable()
1387	if self.engine_type !="park" or self.data.__class__.__name__ !="Data2D":
1388	ctl3.Hide()
1389	ctl4.Hide()
1390	else:
1391	ctl3.Show(True)
1392	ctl4.Show(True)
1393
1394	ix +=1
1395	# Units
1396	try:
1397	units = wx.StaticText(self, -1, self.model.details[item][0], style=wx.ALIGN_LEFT)
1398	except:
1399	units = wx.StaticText(self, -1, "", style=wx.ALIGN_LEFT)
1400	if self.data.__class__.__name__ =="Data2D":
1401	units.Show(True)
1402	#units.Enable()
1403	else:
1404	units.Hide()
1405	#units.Disable()
1406	sizer.Add(units, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
1407
1408
1409	##[cb state, name, value, "+/-", error of fit, min, max , units]
1410	self.parameters.append([cb,item, ctl1,
1411	text2,ctl2, ctl3, ctl4,units])
1412	self.orientation_params.append([cb,item, ctl1,
1413	text2,ctl2, ctl3, ctl4,units])
1414
1415	iy+=1
1416	#sizer.Add((10,10),(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
1417
1418	#Display units text on panel
1419	for item in keys:
1420	if self.model.details[item][0]!='':
1421	self.text2_4.Show()
1422	break
1423	else:
1424	self.text2_4.Hide()
1425
1426	self._copy_parameters_state(self.orientation_params,
1427	self.state.orientation_params)
1428	self._copy_parameters_state(self.orientation_params_disp,
1429	self.state.orientation_params_disp)
1430	self._copy_parameters_state(self.parameters, self.state.parameters)
1431	self._copy_parameters_state(self.fittable_param, self.state.fittable_param)
1432	self._copy_parameters_state(self.fixed_param, self.state.fixed_param)
1433
1434	boxsizer1.Add(sizer)
1435	self.sizer3.Add(boxsizer1,0, wx.EXPAND \| wx.ALL, 10)
1436	self.sizer3.Layout()
1437	self.Layout()
1438	self.Refresh()
1439	self.SetScrollbars(20,20,25,65)
1440
1441
1442	class HelpWindow(wx.Frame):
1443	def __init__(self, parent, id, title):
1444	wx.Frame.__init__(self, parent, id, title, size=(570, 400))
1445
1446	from sans.models.CylinderModel import CylinderModel
1447	model = CylinderModel()
1448
1449	from danse.common.plottools.plottables import Data1D
1450	data= Data1D(x=[1,2], y=[3,4], dy=[0.1, 0,1])
1451
1452	from fitpanel import PageInfo
1453	myinfo = PageInfo(self, model, data=data )
1454
1455	## add data
1456
1457	from models import ModelList
1458	mylist= ModelList()
1459
1460	from sans.models.SphereModel import SphereModel
1461	from sans.models.SquareWellStructure import SquareWellStructure
1462	from sans.models.DebyeModel import DebyeModel
1463	from sans.models.LineModel import LineModel
1464	name= "shapes"
1465	list1= [SphereModel]
1466	mylist.set_list( name, list1)
1467
1468	name= "Shape-independent"
1469	list1= [DebyeModel]
1470	mylist.set_list( name, list1)
1471
1472	name= "Structure Factors"
1473	list1= [SquareWellStructure]
1474	mylist.set_list( name, list1)
1475
1476	name= "Added models"
1477	list1= [LineModel]
1478	mylist.set_list( name, list1)
1479
1480	myinfo.model_list_box = mylist.get_list()
1481
1482	self.page = FitPage(self, myinfo)
1483
1484
1485
1486	self.Centre()
1487	self.Show(True)
1488
1489
1490
1491	if __name__=="__main__":
1492	app = wx.App()
1493	HelpWindow(None, -1, 'HelpWindow')
1494	app.MainLoop()
1495

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

SasView

source: sasview/sansview/perspectives/fitting/fitpage.py @ d40a9cd

Download in other formats: