fitpage.py @ e8ec2fc

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

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

SasView

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

Download in other formats: