fitpage.py @ 70468a5

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 70468a5 was 70468a5, checked in by Jae Cho <jhjcho@…>, 13 years ago
faster gui-updating on selecting a model
Property mode set to `100644`
File size: 121.7 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	import time
10	from sans.models.dispersion_models import ArrayDispersion, GaussianDispersion
11	from DataLoader.data_info import Data1D
12	from sans.guiframe.events import StatusEvent
13	from sans.guiframe.utils import format_number,check_float
14
15	(Chi2UpdateEvent, EVT_CHI2_UPDATE) = wx.lib.newevent.NewEvent()
16	_BOX_WIDTH = 76
17	_DATA_BOX_WIDTH = 300
18	SMEAR_SIZE_L = 0.00
19	SMEAR_SIZE_H = 0.00
20
21	import basepage
22	from basepage import BasicPage
23	from basepage import PageInfoEvent
24	from sans.models.qsmearing import smear_selection
25
26
27	class FitPage(BasicPage):
28	"""
29	FitPanel class contains fields allowing to display results when
30	fitting a model and one data
31
32	:note: For Fit to be performed the user should check at least one parameter
33	on fit Panel window.
34	"""
35
36	def __init__(self,parent, color='rand'):
37	"""
38	Initialization of the Panel
39	"""
40	BasicPage.__init__(self, parent, color=color)
41
42	## draw sizer
43	self._fill_datainfo_sizer()
44	self.is_2D = None
45	self.fit_started = False
46	# get smear info from data
47	self._get_smear_info()
48	self._fill_model_sizer( self.sizer1)
49	self._get_defult_custom_smear()
50	self._fill_range_sizer()
51	self._set_smear(self.data)
52	self.Bind(EVT_CHI2_UPDATE, self.on_complete_chisqr)
53	# bind key event
54	self.Bind(wx.EVT_RIGHT_DOWN, self.on_right_down)
55	self._set_bookmark_flag(False)
56	self._set_save_flag(False)
57	self._set_preview_flag(False)
58	self._set_copy_flag(False)
59	self._set_paste_flag(False)
60	self.btFit.SetFocus()
61
62
63	def _on_fit_complete(self):
64	"""
65	When fit is complete ,reset the fit button label.
66	"""
67	self.btFit.SetLabel("Fit")
68	self.bind_fit_button()
69
70	def _is_2D(self):
71	"""
72	Check if data_name is Data2D
73
74	:return: True or False
75
76	"""
77	if self.data.__class__.__name__ == "Data2D" or \
78	self.enable2D:
79	return True
80	return False
81
82	def _on_engine_change(self, name):
83	"""
84	get the current name of the fit engine type
85	and update the panel accordingly
86	"""
87
88	self.engine_type = str(name)
89	self.state.engine_type = self.engine_type
90	if not self.is_mac:
91	if len(self.parameters) == 0:
92	self.Layout()
93	return
94	self.Layout()
95	self.Refresh()
96
97
98
99	def _fill_range_sizer(self):
100	"""
101	Fill the sizer containing the plotting range
102	add access to npts
103	"""
104	is_2Ddata = False
105
106	# Check if data is 2D
107	if self.data.__class__.__name__ == "Data2D" or \
108	self.enable2D:
109	is_2Ddata = True
110
111	title = "Fitting"
112	#smear messages & titles
113	smear_message_none = "No smearing is selected..."
114	smear_message_dqdata = "The dQ data is being used for smearing..."
115	smear_message_2d = "Higher accuracy is very time-expensive. Use it with care..."
116	smear_message_new_ssmear = "Please enter only the value of interest to customize smearing..."
117	smear_message_new_psmear = "Please enter both; the dQ will be generated by interpolation..."
118	smear_message_2d_x_title = "<dQp>[1/A]:"
119	smear_message_2d_y_title = "<dQs>[1/A]:"
120	smear_message_pinhole_min_title = "dQ_low[1/A]:"
121	smear_message_pinhole_max_title = "dQ_high[1/A]:"
122	smear_message_slit_height_title = "Slit height[1/A]:"
123	smear_message_slit_width_title = "Slit width[1/A]:"
124
125	self._get_smear_info()
126
127	#Sizers
128	box_description_range = wx.StaticBox(self, -1,str(title))
129	boxsizer_range = wx.StaticBoxSizer(box_description_range, wx.VERTICAL)
130	self.sizer_set_smearer = wx.BoxSizer(wx.VERTICAL)
131	sizer_smearer = wx.BoxSizer(wx.HORIZONTAL)
132	self.sizer_new_smear= wx.BoxSizer(wx.HORIZONTAL)
133	self.sizer_set_masking = wx.BoxSizer(wx.HORIZONTAL)
134	sizer_chi2 = wx.BoxSizer(wx.VERTICAL)
135	smear_set_box= wx.StaticBox(self, -1,'Set Instrumental Smearing')
136	sizer_smearer_box = wx.StaticBoxSizer(smear_set_box, wx.HORIZONTAL)
137	sizer_smearer_box.SetMinSize((_DATA_BOX_WIDTH,85))
138	sizer_fit = wx.GridSizer(2, 4, 2, 6)
139
140	# combobox for smear2d accuracy selection
141	self.smear_accuracy = wx.ComboBox(self, -1,size=(50,-1),style=wx.CB_READONLY)
142	self._set_accuracy_list()
143	self.smear_accuracy.SetValue(self.smear2d_accuracy)
144	self.smear_accuracy.SetSelection(0)
145	self.smear_accuracy.SetToolTipString("'Higher' uses more Gaussian points for smearing computation.")
146
147	wx.EVT_COMBOBOX(self.smear_accuracy,-1, self._on_select_accuracy)
148
149	#Fit button
150	self.btFit = wx.Button(self,wx.NewId(),'Fit', size=(88,25))
151	self.default_bt_colour = self.btFit.GetDefaultAttributes()
152	self.btFit.Bind(wx.EVT_BUTTON, self._onFit,id= self.btFit.GetId())
153	self.btFit.SetToolTipString("Start fitting.")
154
155	#textcntrl for custom resolution
156	self.smear_pinhole_max = self.ModelTextCtrl(self, -1,size=(_BOX_WIDTH-25,20),style=wx.TE_PROCESS_ENTER,
157	text_enter_callback = self.onPinholeSmear)
158	self.smear_pinhole_min = self.ModelTextCtrl(self, -1,size=(_BOX_WIDTH-25,20),style=wx.TE_PROCESS_ENTER,
159	text_enter_callback = self.onPinholeSmear)
160	self.smear_slit_height= self.ModelTextCtrl(self, -1,size=(_BOX_WIDTH-25,20),style=wx.TE_PROCESS_ENTER,
161	text_enter_callback = self.onSlitSmear)
162	self.smear_slit_width = self.ModelTextCtrl(self, -1,size=(_BOX_WIDTH-25,20),style=wx.TE_PROCESS_ENTER,
163	text_enter_callback = self.onSlitSmear)
164
165	## smear
166	self.smear_data_left= BGTextCtrl(self, -1, size=(_BOX_WIDTH-25,20), style=0)
167	self.smear_data_left.SetValue(str(self.dq_l))
168	self.smear_data_right = BGTextCtrl(self, -1, size=(_BOX_WIDTH-25,20), style=0)
169	self.smear_data_right.SetValue(str(self.dq_r))
170
171	#set default values for smear
172	self.smear_pinhole_max.SetValue(str(self.dx_max))
173	self.smear_pinhole_min.SetValue(str(self.dx_min))
174	self.smear_slit_height.SetValue(str(self.dxl))
175	self.smear_slit_width.SetValue(str(self.dxw))
176
177	#Filling the sizer containing instruments smearing info.
178	self.disable_smearer = wx.RadioButton(self, -1, 'None', style=wx.RB_GROUP)
179	self.enable_smearer = wx.RadioButton(self, -1, 'Use dQ Data')
180	#self.enable_smearer.SetToolTipString("Click to use the loaded dQ data for smearing.")
181	self.pinhole_smearer = wx.RadioButton(self, -1, 'Custom Pinhole Smear')
182	#self.pinhole_smearer.SetToolTipString("Click to input custom resolution for pinhole smearing.")
183	self.slit_smearer = wx.RadioButton(self, -1, 'Custom Slit Smear')
184	#self.slit_smearer.SetToolTipString("Click to input custom resolution for slit smearing.")
185	self.Bind(wx.EVT_RADIOBUTTON, self.onSmear, id=self.disable_smearer.GetId())
186	self.Bind(wx.EVT_RADIOBUTTON, self.onSmear, id=self.enable_smearer.GetId())
187	self.Bind(wx.EVT_RADIOBUTTON, self.onPinholeSmear, id=self.pinhole_smearer.GetId())
188	self.Bind(wx.EVT_RADIOBUTTON, self.onSlitSmear, id=self.slit_smearer.GetId())
189	self.disable_smearer.SetValue(True)
190
191	# add 4 types of smearing to the sizer
192	sizer_smearer.Add( self.disable_smearer,0, wx.LEFT, 10)
193	sizer_smearer.Add((10,10))
194	sizer_smearer.Add( self.enable_smearer)
195	sizer_smearer.Add((10,10))
196	sizer_smearer.Add( self.pinhole_smearer )
197	sizer_smearer.Add((10,10))
198	sizer_smearer.Add( self.slit_smearer )
199	sizer_smearer.Add((10,10))
200
201	# StaticText for chi2, N(for fitting), Npts
202	self.tcChi = BGTextCtrl(self, -1, "-", size=(75,20), style=0)
203	self.tcChi.SetToolTipString("Chi2/Npts(Fit)")
204	self.Npts_fit = BGTextCtrl(self, -1, "-", size=(75,20), style=0)
205	self.Npts_fit.SetToolTipString(\
206	" Npts : number of points selected for fitting")
207	self.Npts_total = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH, 20),
208	style=wx.TE_PROCESS_ENTER,
209	text_enter_callback=self._onQrangeEnter)
210	self.Npts_total.SetValue(format_number(self.npts_x))
211	self.Npts_total.SetToolTipString(\
212	" Total Npts : total number of data points")
213
214	# Update and Draw button
215	self.draw_button = wx.Button(self,wx.NewId(),'Compute', size=(88,24))
216	self.draw_button.Bind(wx.EVT_BUTTON, \
217	self._onDraw,id= self.draw_button.GetId())
218	self.draw_button.SetToolTipString("Compute and Draw.")
219
220	box_description_1= wx.StaticText(self, -1,' Chi2/Npts')
221	box_description_2= wx.StaticText(self, -1,'Npts(Fit)')
222	box_description_3= wx.StaticText(self, -1,'Total Npts')
223	box_description_3.SetToolTipString( \
224	" Total Npts : total number of data points")
225	#box_description_4= wx.StaticText(self, -1,' ')
226
227
228	sizer_fit.Add(box_description_1,0,0)
229	sizer_fit.Add(box_description_2,0,0)
230	sizer_fit.Add(box_description_3,0,0)
231	sizer_fit.Add(self.draw_button,0,0)
232	sizer_fit.Add(self.tcChi,0,0)
233	sizer_fit.Add(self.Npts_fit ,0,0)
234	sizer_fit.Add(self.Npts_total,0,0)
235	sizer_fit.Add(self.btFit,0,0)
236
237	# StaticText for smear
238	#self.tcChi = wx.StaticText(self, -1, "-", style=wx.ALIGN_LEFT)
239	self.smear_description_none = wx.StaticText(self, -1, smear_message_none , style=wx.ALIGN_LEFT)
240	self.smear_description_dqdata = wx.StaticText(self, -1, smear_message_dqdata , style=wx.ALIGN_LEFT)
241	self.smear_description_type = wx.StaticText(self, -1, "Type:" , style=wx.ALIGN_LEFT)
242	self.smear_description_accuracy_type = wx.StaticText(self, -1, "Accuracy:" , style=wx.ALIGN_LEFT)
243	self.smear_description_smear_type = BGTextCtrl(self, -1, size=(57,20), style=0)
244	self.smear_description_smear_type.SetValue(str(self.dq_l))
245	self.SetBackgroundColour(self.GetParent().GetBackgroundColour())
246	self.smear_description_2d = wx.StaticText(self, -1, smear_message_2d , style=wx.ALIGN_LEFT)
247	self.smear_message_new_s = wx.StaticText(self, -1, smear_message_new_ssmear , style=wx.ALIGN_LEFT)
248	self.smear_message_new_p = wx.StaticText(self, -1, smear_message_new_psmear , style=wx.ALIGN_LEFT)
249	self.smear_description_2d_x = wx.StaticText(self, -1, smear_message_2d_x_title , style=wx.ALIGN_LEFT)
250	self.smear_description_2d_x.SetToolTipString(" dQp(parallel) in q_r direction.")
251	self.smear_description_2d_y = wx.StaticText(self, -1, smear_message_2d_y_title , style=wx.ALIGN_LEFT)
252	self.smear_description_2d_y.SetToolTipString(" dQs(perpendicular) in q_phi direction.")
253	self.smear_description_pin_min = wx.StaticText(self, -1, smear_message_pinhole_min_title , style=wx.ALIGN_LEFT)
254	self.smear_description_pin_max = wx.StaticText(self, -1, smear_message_pinhole_max_title , style=wx.ALIGN_LEFT)
255	self.smear_description_slit_height = wx.StaticText(self, -1, smear_message_slit_height_title , style=wx.ALIGN_LEFT)
256	self.smear_description_slit_width = wx.StaticText(self, -1, smear_message_slit_width_title , style=wx.ALIGN_LEFT)
257
258	#arrange sizers
259	#boxsizer1.Add( self.tcChi )
260	self.sizer_set_smearer.Add(sizer_smearer )
261	self.sizer_set_smearer.Add((10,10))
262	self.sizer_set_smearer.Add( self.smear_description_none,0, wx.CENTER, 10 )
263	self.sizer_set_smearer.Add( self.smear_description_dqdata,0, wx.CENTER, 10 )
264	self.sizer_set_smearer.Add( self.smear_description_2d,0, wx.CENTER, 10 )
265	self.sizer_new_smear.Add( self.smear_description_type,0, wx.CENTER, 10 )
266	self.sizer_new_smear.Add( self.smear_description_accuracy_type,0, wx.CENTER, 10 )
267	self.sizer_new_smear.Add( self.smear_accuracy )
268	self.sizer_new_smear.Add( self.smear_description_smear_type,0, wx.CENTER, 10 )
269	self.sizer_new_smear.Add((15,-1))
270	self.sizer_new_smear.Add( self.smear_description_2d_x,0, wx.CENTER, 10 )
271	self.sizer_new_smear.Add( self.smear_description_pin_min,0, wx.CENTER, 10 )
272	self.sizer_new_smear.Add( self.smear_description_slit_height,0, wx.CENTER, 10 )
273
274	self.sizer_new_smear.Add( self.smear_pinhole_min,0, wx.CENTER, 10 )
275	self.sizer_new_smear.Add( self.smear_slit_height,0, wx.CENTER, 10 )
276	self.sizer_new_smear.Add( self.smear_data_left,0, wx.CENTER, 10 )
277	self.sizer_new_smear.Add((20,-1))
278	self.sizer_new_smear.Add( self.smear_description_2d_y,0, wx.CENTER, 10 )
279	self.sizer_new_smear.Add( self.smear_description_pin_max,0, wx.CENTER, 10 )
280	self.sizer_new_smear.Add( self.smear_description_slit_width,0, wx.CENTER, 10 )
281
282	self.sizer_new_smear.Add( self.smear_pinhole_max,0, wx.CENTER, 10 )
283	self.sizer_new_smear.Add( self.smear_slit_width,0, wx.CENTER, 10 )
284	self.sizer_new_smear.Add( self.smear_data_right,0, wx.CENTER, 10 )
285
286	self.sizer_set_smearer.Add( self.smear_message_new_s,0, wx.CENTER, 10)
287	self.sizer_set_smearer.Add( self.smear_message_new_p,0, wx.CENTER, 10)
288	self.sizer_set_smearer.Add((5,2))
289	self.sizer_set_smearer.Add( self.sizer_new_smear,0, wx.CENTER, 10 )
290
291	# add all to chi2 sizer
292	sizer_smearer_box.Add(self.sizer_set_smearer)
293	sizer_chi2.Add(sizer_smearer_box)
294	sizer_chi2.Add((-1,5))
295
296	# hide all smear messages and textctrl
297	self._hide_all_smear_info()
298
299	# get smear_selection
300	self.current_smearer= smear_selection( self.data, self.model )
301
302	# Show only the relevant smear messages, etc
303	if self.current_smearer == None:
304	if not is_2Ddata:
305	self.smear_description_none.Show(True)
306	self.enable_smearer.Disable()
307	else:
308	self.smear_description_none.Show(True)
309	#self.smear_description_2d.Show(True)
310	#self.pinhole_smearer.Disable()
311	self.slit_smearer.Disable()
312	#self.enable_smearer.Disable()
313	if self.data == None:
314	self.slit_smearer.Disable()
315	self.pinhole_smearer.Disable()
316	self.enable_smearer.Disable()
317	else: self._show_smear_sizer()
318	boxsizer_range.Add(self.sizer_set_masking)
319	#2D data? default
320	is_2Ddata = False
321
322	#check if it is 2D data
323	if self.data.__class__.__name__ == "Data2D" or \
324	self.enable2D:
325	is_2Ddata = True
326
327	self.sizer5.Clear(True)
328
329	self.qmin = self.ModelTextCtrl(self, -1,size=(_BOX_WIDTH,20),
330	style=wx.TE_PROCESS_ENTER,
331	text_enter_callback = self._onQrangeEnter)
332	self.qmin.SetValue(str(self.qmin_x))
333	self.qmin.SetToolTipString("Minimun value of Q in linear scale.")
334
335	self.qmax = self.ModelTextCtrl(self, -1,size=(_BOX_WIDTH,20),
336	style=wx.TE_PROCESS_ENTER,
337	text_enter_callback=self._onQrangeEnter)
338	self.qmax.SetValue(str(self.qmax_x))
339	self.qmax.SetToolTipString("Maximum value of Q in linear scale.")
340	"""
341	self.theory_npts_tcrtl = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH, 20),
342	style=wx.TE_PROCESS_ENTER,
343	text_enter_callback=self._onQrangeEnter)
344	self.theory_npts_tcrtl.SetValue(format_number(self.npts_x))
345	self.theory_npts_tcrtl.SetToolTipString("Number of point to plot.")
346	"""
347	id = wx.NewId()
348	self.reset_qrange =wx.Button(self,id,'Reset',size=(77,20))
349
350	self.reset_qrange.Bind(wx.EVT_BUTTON, self.on_reset_clicked,id=id)
351	self.reset_qrange.SetToolTipString("Reset Q range to the default values")
352
353	sizer_horizontal=wx.BoxSizer(wx.HORIZONTAL)
354	sizer= wx.GridSizer(2, 4, 2, 6)
355
356	self.btEditMask = wx.Button(self,wx.NewId(),'Editor', size=(88,23))
357	self.btEditMask.Bind(wx.EVT_BUTTON, self._onMask,id= self.btEditMask.GetId())
358	self.btEditMask.SetToolTipString("Edit Mask.")
359	self.EditMask_title = wx.StaticText(self, -1, ' Masking(2D)')
360
361	sizer.Add(wx.StaticText(self, -1, 'Q range'))
362	sizer.Add(wx.StaticText(self, -1, ' Min[1/A]'))
363	sizer.Add(wx.StaticText(self, -1, ' Max[1/A]'))
364	sizer.Add(self.EditMask_title)
365	#sizer.Add(wx.StaticText(self, -1, ''))
366	sizer.Add(self.reset_qrange)
367	sizer.Add(self.qmin)
368	sizer.Add(self.qmax)
369	#sizer.Add(self.theory_npts_tcrtl)
370	sizer.Add(self.btEditMask)
371	boxsizer_range.Add(sizer_chi2)
372	boxsizer_range.Add((10,10))
373	boxsizer_range.Add(sizer)
374
375	boxsizer_range.Add((10,15))
376	boxsizer_range.Add(sizer_fit)
377	if is_2Ddata:
378	self.btEditMask.Enable()
379	self.EditMask_title.Enable()
380	else:
381	self.btEditMask.Disable()
382	self.EditMask_title.Disable()
383	## save state
384	self.save_current_state()
385	self.sizer5.Add(boxsizer_range,0, wx.EXPAND \| wx.ALL, 10)
386	self.sizer5.Layout()
387
388	def _fill_datainfo_sizer(self):
389	"""
390	fill sizer 0 with data info
391	"""
392	## no loaded data , don't fill the sizer
393	if self.data is None:
394	data_min = ""
395	data_max = ""
396	data_name = " None"
397	else:
398	data_name = self.data.name
399	#set maximum range for x in linear scale
400	if not hasattr(self.data,"data"): #Display only for 1D data fit
401	# Minimum value of data
402	data_min = min(self.data.x)
403	# Maximum value of data
404	data_max = max(self.data.x)
405	else:
406	## Minimum value of data
407	data_min = 0
408	x = max(math.fabs(self.data.xmin), math.fabs(self.data.xmax))
409	y = max(math.fabs(self.data.ymin), math.fabs(self.data.ymax))
410	## Maximum value of data
411	data_max = math.sqrt(xx + yy)
412	## set q range to plot
413	self.qmin_x = data_min
414	self.qmax_x = data_max
415	box_description= wx.StaticBox(self, -1, 'I(q) Data Source')
416	boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
417	#----------------------------------------------------------
418	sizer_data = wx.BoxSizer(wx.HORIZONTAL)
419	#Filling the sizer containing data related fields
420	#self.dataSource = wx.StaticText(self, -1,str(self.data.name))
421	self.dataSource = BGTextCtrl(self, -1)
422	self.dataSource.SetValue(str(data_name))
423	#self.dataSource.SetEditable(False)
424	self.dataSource.SetMinSize((_DATA_BOX_WIDTH, -1))
425	sizer_data.Add(wx.StaticText(self, -1, 'Name : '))
426	sizer_data.Add(self.dataSource )
427	sizer_data.Add( (0,5) )
428	"""
429	#---------sizer 2 draw--------------------------------
430	text4_3 = wx.StaticText(self, -1, 'Total Q Range (1/A)',
431	style=wx.ALIGN_LEFT)
432	sizer_range = wx.BoxSizer(wx.HORIZONTAL)
433	sizer_range.Add( text4_3 ,0, wx.RIGHT, 10)
434	self.minimum_q = BGTextCtrl(self, -1, size=(_BOX_WIDTH,20))
435	self.minimum_q.SetValue(str(data_min))
436	self.maximum_q = BGTextCtrl(self, -1,size=(_BOX_WIDTH,20))
437	self.maximum_q.SetValue(str(data_max))
438	sizer_range.Add(wx.StaticText(self, -1, "Min: "),0, wx.LEFT, 10)
439	sizer_range.Add(self.minimum_q,0, wx.LEFT, 10)
440	sizer_range.Add(wx.StaticText(self, -1, "Max: "),0, wx.LEFT, 10)
441	sizer_range.Add(self.maximum_q,0, wx.LEFT, 10)
442	"""
443	boxsizer1.Add(sizer_data,0, wx.ALL, 10)
444	#boxsizer1.Add(sizer_range, 0 , wx.LEFT, 10)
445	#------------------------------------------------------------
446
447	self.sizer0.Add(boxsizer1,0, wx.EXPAND \| wx.ALL, 10)
448	self.sizer0.Layout()
449
450	def _fill_model_sizer(self, sizer):
451	"""
452	fill sizer containing model info
453	"""
454	##Add model function Details button in fitpanel.
455	##The following 3 lines are for Mac. Let JHC know before modifying...
456	title = "Model"
457	self.formfactorbox = None
458	self.multifactorbox = None
459	self.mbox_description= wx.StaticBox(self, -1,str(title))
460	boxsizer1 = wx.StaticBoxSizer(self.mbox_description, wx.VERTICAL)
461
462	id = wx.NewId()
463	self.model_help =wx.Button(self,id,'Details', size=(80,23))
464	self.model_help.Bind(wx.EVT_BUTTON, self.on_model_help_clicked,id=id)
465	self.model_help.SetToolTipString("Model Function Help")
466	id = wx.NewId()
467	self.model_view = wx.Button(self, id,"1D Mode", size=(80, 23))
468	self.model_view.Bind(wx.EVT_BUTTON, self._onModel2D, id=id)
469	hint = "toggle view of model from 1D to 2D or 2D to 1D"
470	self.model_view.SetToolTipString(hint)
471
472	self.shape_rbutton = wx.RadioButton(self, -1, 'Shapes', style=wx.RB_GROUP)
473	self.shape_indep_rbutton = wx.RadioButton(self, -1, "Shape-Independent")
474	self.struct_rbutton = wx.RadioButton(self, -1, "Structure Factor ")
475	self.plugin_rbutton = wx.RadioButton(self, -1, "Customized Models")
476
477	self.Bind(wx.EVT_RADIOBUTTON, self._show_combox,
478	id= self.shape_rbutton.GetId())
479	self.Bind(wx.EVT_RADIOBUTTON, self._show_combox,
480	id= self.shape_indep_rbutton.GetId())
481	self.Bind(wx.EVT_RADIOBUTTON, self._show_combox,
482	id= self.struct_rbutton.GetId())
483	self.Bind(wx.EVT_RADIOBUTTON, self._show_combox,
484	id= self.plugin_rbutton.GetId())
485	#MAC needs SetValue
486	self.shape_rbutton.SetValue(True)
487
488	sizer_radiobutton = wx.GridSizer(2, 3, 5, 5)
489	sizer_radiobutton.Add(self.shape_rbutton)
490	sizer_radiobutton.Add(self.shape_indep_rbutton)
491	#sizer_radiobutton.Add((5, 5))
492	sizer_radiobutton.Add(self.model_view,1, wx.LEFT, 20)
493	sizer_radiobutton.Add(self.plugin_rbutton)
494	sizer_radiobutton.Add(self.struct_rbutton)
495	#sizer_radiobutton.Add((5, 5))
496	sizer_radiobutton.Add(self.model_help,1, wx.LEFT, 20)
497
498	sizer_selection = wx.BoxSizer(wx.HORIZONTAL)
499	mutifactor_selection = wx.BoxSizer(wx.HORIZONTAL)
500
501	self.text1 = wx.StaticText(self,-1,"" )
502	self.text2 = wx.StaticText(self,-1,"P(Q)*S(Q)" )
503	self.mutifactor_text = wx.StaticText( self,-1,"No. of Shells: ")
504	self.mutifactor_text1 = wx.StaticText( self,-1,"" )
505	self.show_sld_button = wx.Button( self,-1,"Show SLD Profile" )
506	self.show_sld_button.Bind(wx.EVT_BUTTON,self._on_show_sld)
507
508	self.formfactorbox = wx.ComboBox(self, -1,style=wx.CB_READONLY)
509	if self.model!= None:
510	self.formfactorbox.SetValue(self.model.name)
511	self.structurebox = wx.ComboBox(self, -1, style=wx.CB_READONLY)
512	self.multifactorbox = wx.ComboBox(self, -1, style=wx.CB_READONLY)
513	self.initialize_combox()
514	wx.EVT_COMBOBOX(self.formfactorbox, -1, self._on_select_model)
515
516	wx.EVT_COMBOBOX(self.structurebox, -1, self._on_select_model)
517	wx.EVT_COMBOBOX(self.multifactorbox, -1, self._on_select_model)
518	## check model type to show sizer
519	if self.model !=None:
520	self._set_model_sizer_selection(self.model)
521
522	sizer_selection.Add(self.text1)
523	sizer_selection.Add((5, 5))
524	sizer_selection.Add(self.formfactorbox)
525	sizer_selection.Add((5, 5))
526	sizer_selection.Add(self.text2)
527	sizer_selection.Add((5, 5))
528	sizer_selection.Add(self.structurebox)
529
530	mutifactor_selection.Add((10,5))
531	mutifactor_selection.Add(self.mutifactor_text)
532	mutifactor_selection.Add(self.multifactorbox)
533	mutifactor_selection.Add((5, 5))
534	mutifactor_selection.Add(self.mutifactor_text1)
535	mutifactor_selection.Add((10, 5))
536	mutifactor_selection.Add(self.show_sld_button)
537
538
539	boxsizer1.Add(sizer_radiobutton)
540	boxsizer1.Add((10, 10))
541	boxsizer1.Add(sizer_selection)
542	boxsizer1.Add((10, 10))
543	boxsizer1.Add(mutifactor_selection)
544
545	self._set_multfactor_combobox()
546	self.multifactorbox.SetSelection(1)
547	self.show_sld_button.Hide()
548	sizer.Add(boxsizer1,0, wx.EXPAND \| wx.ALL, 10)
549	sizer.Layout()
550
551	def _set_sizer_dispersion(self):
552	"""
553	draw sizer with gaussian dispersity parameters
554	"""
555	self.fittable_param=[]
556	self.fixed_param=[]
557	self.orientation_params_disp=[]
558
559	self.sizer4_4.Clear(True)
560	if self.model==None:
561	##no model is selected
562	return
563	if not self.enable_disp.GetValue():
564	## the user didn't select dispersity display
565	return
566
567	self._reset_dispersity()
568
569	## fill a sizer with the combobox to select dispersion type
570	#sizer_select_dispers = wx.BoxSizer(wx.HORIZONTAL)
571	model_disp = wx.StaticText(self, -1, 'Function')
572
573	import sans.models.dispersion_models
574	self.polydisp= sans.models.dispersion_models.models
575
576	ix = 0
577	iy = 0
578	disp = wx.StaticText(self, -1, ' ')
579	self.sizer4_4.Add(disp,( iy, ix),(1,1),
580	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
581	ix += 1
582	values = wx.StaticText(self, -1, 'PD[ratio]')
583	polytext = "Polydispersity (= STD/mean); "
584	polytext += "the standard deviation over the mean value."
585	values.SetToolTipString(polytext)
586
587	self.sizer4_4.Add(values,( iy, ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE,
588	0)
589	ix +=2
590	if self.is_mac:
591	err_text = 'Error'
592	else:
593	err_text = ''
594	self.text_disp_1 = wx.StaticText(self, -1, err_text)
595	self.sizer4_4.Add( self.text_disp_1,(iy, ix),(1,1),\
596	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
597
598
599	ix +=1
600	self.text_disp_min = wx.StaticText(self, -1, 'Min')
601	self.sizer4_4.Add(self.text_disp_min,(iy, ix),(1,1),\
602	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
603
604	ix +=1
605	self.text_disp_max = wx.StaticText(self, -1, 'Max')
606	self.sizer4_4.Add(self.text_disp_max,(iy, ix),(1,1),\
607	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
608
609	ix += 1
610	npts = wx.StaticText(self, -1, 'Npts')
611	npts.SetToolTipString("Number of sampling points for the numerical\n\
612	integration over the distribution function.")
613	self.sizer4_4.Add(npts,( iy, ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
614	ix += 1
615	nsigmas = wx.StaticText(self, -1, 'Nsigs')
616	nsigmas.SetToolTipString(" Number of sigmas between which the range\n\
617	of the distribution function will be used for weighting. \n\
618	The value '3' covers 99.5% for Gaussian distribution \n\
619	function. Note: Not recommended to change this value.")
620	self.sizer4_4.Add(nsigmas,( iy, ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE,
621	0)
622	ix +=1
623	self.sizer4_4.Add(model_disp,(iy, ix),(1,1),\
624	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
625
626	if self.engine_type=="park":
627	self.text_disp_max.Show(True)
628	self.text_disp_min.Show(True)
629
630	for item in self.model.dispersion.keys():
631	if not item in self.model.orientation_params:
632	if not self.disp_cb_dict.has_key(item):
633	self.disp_cb_dict[item]= None
634	name0="Distribution of " + item
635	name1=item+".width"
636	name2=item+".npts"
637	name3=item+".nsigmas"
638	if not self.model.details.has_key(name1):
639	self.model.details [name1] = ["",None,None]
640
641	iy += 1
642	for p in self.model.dispersion[item].keys():
643
644	if p=="width":
645	ix = 0
646	cb = wx.CheckBox(self, -1, name0, (10, 10))
647	cb.SetToolTipString("Check mark to fit")
648	wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
649	self.sizer4_4.Add( cb,( iy, ix),(1,1),
650	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE,
651	5)
652	ix = 1
653	value= self.model.getParam(name1)
654	ctl1 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/1.3,20)
655	,style=wx.TE_PROCESS_ENTER)
656	ctl1.SetLabel('PD[ratio]')
657	poly_text = "Polydispersity (STD/mean) of %s\n" % item
658	poly_text += "STD: the standard deviation"
659	poly_text += " from the mean value."
660	ctl1.SetToolTipString(poly_text)
661	ctl1.SetValue(str (format_number(value, True)))
662	self.sizer4_4.Add(ctl1, (iy,ix),(1,1),wx.EXPAND)
663	## text to show error sign
664	ix = 2
665	text2=wx.StaticText(self, -1, '+/-')
666	self.sizer4_4.Add(text2,(iy, ix),(1,1),
667	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
668	if not self.is_mac:
669	text2.Hide()
670
671	ix = 3
672	ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH/1.3,20),
673	style=0)
674
675	self.sizer4_4.Add(ctl2, (iy,ix),(1,1),
676	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
677	if not self.is_mac:
678	ctl2.Hide()
679
680	ix = 4
681	ctl3 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,
682	20), style=wx.TE_PROCESS_ENTER,
683	text_enter_callback = self._onparamRangeEnter)
684
685	self.sizer4_4.Add(ctl3, (iy,ix),(1,1),
686	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
687
688
689	ix = 5
690	ctl4 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,
691	20), style=wx.TE_PROCESS_ENTER,
692	text_enter_callback = self._onparamRangeEnter)
693
694	self.sizer4_4.Add(ctl4, (iy,ix),(1,1),
695	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
696
697
698	if self.engine_type=="park":
699	ctl3.Show(True)
700	ctl4.Show(True)
701
702	elif p=="npts":
703	ix = 6
704	value= self.model.getParam(name2)
705	Tctl = self.ModelTextCtrl(self, -1,
706	size=(_BOX_WIDTH/2.2,20),
707	style=wx.TE_PROCESS_ENTER)
708
709	Tctl.SetValue(str (format_number(value)))
710	self.sizer4_4.Add(Tctl, (iy,ix),(1,1),
711	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
712	self.fixed_param.append([None,name2, Tctl,None,None,
713	None, None,None])
714	elif p=="nsigmas":
715	ix = 7
716	value= self.model.getParam(name3)
717	Tct2 = self.ModelTextCtrl(self, -1,
718	size=(_BOX_WIDTH/2.2,20),
719	style=wx.TE_PROCESS_ENTER)
720
721	Tct2.SetValue(str (format_number(value)))
722	self.sizer4_4.Add(Tct2, (iy,ix),(1,1),
723	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
724	#ix +=1
725	#self.sizer4_4.Add((20,20), (iy,ix),(1,1),
726	# wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
727
728	self.fixed_param.append([None, name3, Tct2,
729	None, None, None,
730	None, None])
731
732
733	ix = 8
734	disp_box = wx.ComboBox(self, -1,size=(65,-1),
735	style=wx.CB_READONLY, name = '%s'% name1)
736	for key, value in self.polydisp.iteritems():
737	name_disp = str(key)
738	disp_box.Append(name_disp,value)
739	disp_box.SetStringSelection("gaussian")
740	wx.EVT_COMBOBOX(disp_box,-1, self._on_disp_func)
741	self.sizer4_4.Add(disp_box,(iy,ix),(1,1), wx.EXPAND)
742	self.fittable_param.append([cb,name1,ctl1,text2,
743	ctl2, ctl3, ctl4, disp_box])
744
745	ix =0
746	iy +=1
747	self.sizer4_4.Add((20,20),(iy,ix),(1,1),
748	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
749	first_orient = True
750	for item in self.model.dispersion.keys():
751	if item in self.model.orientation_params:
752	if not self.disp_cb_dict.has_key(item):
753	self.disp_cb_dict[item]= None
754	name0="Distribution of " + item
755	name1=item+".width"
756	name2=item+".npts"
757	name3=item+".nsigmas"
758
759	if not self.model.details.has_key(name1):
760	self.model.details [name1] = ["",None,None]
761
762
763	iy += 1
764	for p in self.model.dispersion[item].keys():
765
766	if p=="width":
767	ix = 0
768	cb = wx.CheckBox(self, -1, name0, (10, 10))
769	cb.SetToolTipString("Check mark to fit")
770	wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
771	self.sizer4_4.Add( cb,( iy, ix),(1,1),
772	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE,
773	5)
774	if self.data.__class__.__name__ == "Data2D" or \
775	self.enable2D:
776	cb.Show(True)
777	elif cb.IsShown():
778	cb.Hide()
779	ix = 1
780	value= self.model.getParam(name1)
781	ctl1 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/1.3,
782	20),style=wx.TE_PROCESS_ENTER)
783	poly_tip = "Absolute Sigma for %s." % item
784	ctl1.SetToolTipString(poly_tip)
785	ctl1.SetValue(str (format_number(value, True)))
786	if self.data.__class__.__name__ == "Data2D" or \
787	self.enable2D:
788	if first_orient:
789	values.SetLabel('PD[ratio], Sig[deg]')
790	poly_text = "PD(polydispersity for lengths):\n"
791	poly_text += "It should be a value between"
792	poly_text += "0 and 1\n"
793	poly_text += "Sigma for angles: \n"
794	poly_text += "It is the STD (ratio*mean)"
795	poly_text += " of the distribution.\n "
796
797	values.SetToolTipString(poly_text)
798	first_orient = False
799	ctl1.Show(True)
800	elif ctl1.IsShown():
801	ctl1.Hide()
802
803	self.sizer4_4.Add(ctl1, (iy,ix),(1,1),wx.EXPAND)
804	## text to show error sign
805	ix = 2
806	text2=wx.StaticText(self, -1, '+/-')
807	self.sizer4_4.Add(text2,(iy, ix),(1,1),
808	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
809
810	text2.Hide()
811
812	ix = 3
813	ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH/1.3,20),
814	style=0)
815
816	self.sizer4_4.Add(ctl2, (iy,ix),(1,1),
817	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
818
819	ctl2.Hide()
820	if self.data.__class__.__name__ == "Data2D" or \
821	self.enable2D:
822	if self.is_mac:
823	text2.Show(True)
824	ctl2.Show(True)
825
826	ix = 4
827	ctl3 = self.ModelTextCtrl(self, -1,
828	size=(_BOX_WIDTH/2,20),
829	style=wx.TE_PROCESS_ENTER,
830	text_enter_callback = self._onparamRangeEnter)
831
832	self.sizer4_4.Add(ctl3, (iy,ix),(1,1),
833	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
834
835	ctl3.Hide()
836
837	ix = 5
838	ctl4 = self.ModelTextCtrl(self, -1,
839	size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
840	text_enter_callback = self._onparamRangeEnter)
841	self.sizer4_4.Add(ctl4, (iy,ix),(1,1),
842	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
843	ctl4.Hide()
844
845	if self.data.__class__.__name__ == "Data2D" or \
846	self.enable2D:
847	ctl3.Show(True)
848	ctl4.Show(True)
849
850	elif p=="npts":
851	ix = 6
852	value= self.model.getParam(name2)
853	Tctl = self.ModelTextCtrl(self, -1,
854	size=(_BOX_WIDTH/2.2, 20),
855	style=wx.TE_PROCESS_ENTER)
856
857	Tctl.SetValue(str (format_number(value)))
858	if self.data.__class__.__name__ == "Data2D" or \
859	self.enable2D:
860	Tctl.Show(True)
861	else:
862	Tctl.Hide()
863	self.sizer4_4.Add(Tctl, (iy,ix),(1,1),
864	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
865	self.fixed_param.append([None,name2, Tctl,None,None,
866	None, None,None])
867	self.orientation_params_disp.append([None,name2,
868	Tctl,None,None,
869	None, None,None])
870	elif p=="nsigmas":
871	ix = 7
872	value= self.model.getParam(name3)
873	Tct2 = self.ModelTextCtrl(self, -1,
874	size=(_BOX_WIDTH/2.2,20),
875	style=wx.TE_PROCESS_ENTER)
876
877	Tct2.SetValue(str (format_number(value)))
878	if self.data.__class__.__name__ == "Data2D" or \
879	self.enable2D:
880	Tct2.Show(True)
881	else:
882	Tct2.Hide()
883	self.sizer4_4.Add(Tct2, (iy,ix),(1,1),
884	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
885
886
887	self.fixed_param.append([None,name3, Tct2
888	,None,None, None, None,None])
889
890	self.orientation_params_disp.append([None,name3,
891	Tct2 ,None,None, None, None,None])
892
893
894	ix = 8
895	disp_box = wx.ComboBox(self, -1,size=(65,-1),
896	style=wx.CB_READONLY, name = '%s'% name1)
897	for key, value in self.polydisp.iteritems():
898	name_disp = str(key)
899	disp_box.Append(name_disp,value)
900	disp_box.SetStringSelection("gaussian")
901	wx.EVT_COMBOBOX(disp_box,-1, self._on_disp_func)
902	self.sizer4_4.Add(disp_box,(iy,ix),(1,1), wx.EXPAND)
903	self.fittable_param.append([cb,name1,ctl1,text2,
904	ctl2, ctl3, ctl4, disp_box])
905	self.orientation_params_disp.append([cb,name1,ctl1,
906	text2, ctl2, ctl3, ctl4, disp_box])
907
908	if self.data.__class__.__name__ == "Data2D" or \
909	self.enable2D:
910	disp_box.Show(True)
911	else:
912	disp_box.Hide()
913
914
915	self.state.disp_cb_dict = copy.deepcopy(self.disp_cb_dict)
916
917	self.state.model = self.model.clone()
918	## save state into
919	self.state.cb1 = self.cb1.GetValue()
920	self._copy_parameters_state(self.parameters, self.state.parameters)
921	self._copy_parameters_state(self.orientation_params_disp,
922	self.state.orientation_params_disp)
923	self._copy_parameters_state(self.fittable_param,
924	self.state.fittable_param)
925	self._copy_parameters_state(self.fixed_param, self.state.fixed_param)
926
927
928	wx.PostEvent(self.parent, StatusEvent(status=\
929	" Selected Distribution: Gaussian"))
930	#Fill the list of fittable parameters
931	self.select_all_param(event=None)
932
933	self.Layout()
934
935
936	def _onDraw(self, event):
937	"""
938	Update and Draw the model
939	"""
940	if self.model ==None:
941	msg="Please select a Model first..."
942	wx.MessageBox(msg, 'Info')
943
944	return
945	flag = self._update_paramv_on_fit()
946	self._onparamEnter_helper()
947	if not flag:
948	msg= "The parameters are invalid"
949	wx.PostEvent(self.parent.parent, StatusEvent(status= msg ))
950	return
951
952
953	def _onFit(self, event):
954	"""
955	Allow to fit
956	"""
957	if event != None:
958	event.Skip()
959	if len(self.parent._manager.fit_thread_list)>0 and\
960	self.parent._manager._fit_engine != "park":
961	msg = "The FitEnging will be set to 'ParkMC'\n"
962	msg += " to fit with more than one data set..."
963	wx.MessageBox(msg, 'Info')
964	#wx.PostEvent(self._manager.parent, StatusEvent(status=\
965	# "Fitting: %s"%msg))
966
967	if self.data is None:
968	msg = "Please get Data first..."
969	wx.MessageBox(msg, 'Info')
970	wx.PostEvent(self._manager.parent, StatusEvent(status=\
971	"Fit: %s" % msg))
972	return
973	if self.model is None:
974	msg = "Please select a Model first..."
975	wx.MessageBox(msg, 'Info')
976	wx.PostEvent(self._manager.parent, StatusEvent(status=\
977	"Fit: %s"%msg, type="stop"))
978	return
979
980	if len(self.param_toFit) <= 0:
981	msg= "Select at least one parameter to fit"
982	wx.MessageBox(msg, 'Info')
983	wx.PostEvent(self._manager.parent, StatusEvent(status= msg,
984	type="stop" ))
985	return
986
987	flag = self._update_paramv_on_fit()
988
989	if not flag:
990	msg= "Fitting range or parameters are invalid"
991	wx.PostEvent(self.parent.parent, StatusEvent(status= msg,
992	type="stop"))
993	return
994
995	self.select_param(event =None)
996
997	#Clear errors if exist from previous fitting
998	#self._clear_Err_on_Fit()
999
1000	# Remove or do not allow fitting on the Q=0 point, especially
1001	# when y(q=0)=None at x[0].
1002	self.qmin_x = float(self.qmin.GetValue())
1003	self.qmax_x = float( self.qmax.GetValue())
1004	self._manager._reset_schedule_problem(value=0, uid=self.uid)
1005	self._manager.schedule_for_fit(uid=self.uid,value=1, fitproblem =None)
1006	self._manager.set_fit_range(uid=self.uid,qmin= self.qmin_x,
1007	qmax= self.qmax_x)
1008
1009	#single fit
1010	self._manager.onFit(uid=self.uid)
1011	self.btFit.SetLabel("Stop")
1012	self.bind_fit_button()
1013
1014
1015	def bind_fit_button(self):
1016	"""
1017	bind the fit button to either fit handler or stop fit handler
1018	"""
1019	self.btFit.Unbind(event=wx.EVT_BUTTON, id= self.btFit.GetId())
1020	if self.btFit.GetLabel().lower() == "stop":
1021	self.fit_started = True
1022	self.btFit.SetForegroundColour('red')
1023	self.btFit.Bind(event=wx.EVT_BUTTON, handler=self._StopFit,
1024	id=self.btFit.GetId())
1025	elif self.btFit.GetLabel().lower() == "fit":
1026	self.fit_started = False
1027	self.btFit.SetDefault()
1028	self.btFit.SetForegroundColour('black')
1029	#self.btFit.SetBackgroundColour(self.default_bt_colour)
1030	self.btFit.Bind(event=wx.EVT_BUTTON, handler=self._onFit,
1031	id=self.btFit.GetId())
1032	else:
1033	msg = "FitPage: fit button has unknown label"
1034	raise ValuerError, msg
1035	self._manager._reset_schedule_problem(value=0)
1036
1037	def is_fitting(self):
1038	if self.fit_started:
1039	self._StopFit(event=None)
1040
1041	def _StopFit(self, event=None):
1042	"""
1043	Stop fit
1044	"""
1045	#time.sleep(0.1)
1046	if event != None:
1047	event.Skip()
1048	#if self.engine_type=="scipy":
1049	self._manager.stop_fit(self.uid)
1050	self._manager._reset_schedule_problem(value=0)
1051	self._on_fit_complete()
1052
1053	def _on_select_model(self, event=None):
1054	"""
1055	call back for model selection
1056	"""
1057
1058	self.Show(False)
1059	self._on_select_model_helper()
1060	self.set_model_param_sizer(self.model)
1061	if self.model is None:
1062	self._set_bookmark_flag(False)
1063	self._keep.Enable(False)
1064	self._set_save_flag(False)
1065	self.enable_disp.SetValue(False)
1066	self.disable_disp.SetValue(True)
1067	try:
1068	self.set_dispers_sizer()
1069	except:
1070	pass
1071	#self.btFit.SetFocus()
1072	self.state.enable_disp = self.enable_disp.GetValue()
1073	self.state.disable_disp = self.disable_disp.GetValue()
1074	self.state.pinhole_smearer = self.pinhole_smearer.GetValue()
1075	self.state.slit_smearer = self.slit_smearer.GetValue()
1076
1077	self.state.structurecombobox = self.structurebox.GetCurrentSelection()
1078	self.state.formfactorcombobox = self.formfactorbox.GetCurrentSelection()
1079
1080	if self.model != None:
1081	self._set_copy_flag(True)
1082	self._set_paste_flag(True)
1083	if self.data != None:
1084	self._set_bookmark_flag(True)
1085	self._keep.Enable(True)
1086	#self._set_save_flag(True)
1087	# Reset smearer, model and data
1088	self._set_smear(self.data)
1089	try:
1090	# update smearer sizer
1091	self.onSmear(None)
1092	temp_smear = None
1093	if self.enable_smearer.GetValue():
1094	# Set the smearer environments
1095	temp_smear = self.smearer
1096	except:
1097	raise
1098	## error occured on chisqr computation
1099	#pass
1100	## event to post model to fit to fitting plugins
1101	(ModelEventbox, EVT_MODEL_BOX) = wx.lib.newevent.NewEvent()
1102
1103	## set smearing value whether or not
1104	# the data contain the smearing info
1105	evt = ModelEventbox(model=self.model,
1106	smearer=temp_smear,
1107	qmin=float(self.qmin_x),
1108	uid=self.uid,
1109	qmax=float(self.qmax_x))
1110
1111	self._manager._on_model_panel(evt=evt)
1112	self.mbox_description.SetLabel("Model [%s]" % str(self.model.name))
1113	self.state.model = self.model.clone()
1114	self.state.model.name = self.model.name
1115
1116
1117	if event != None:
1118	## post state to fit panel
1119	new_event = PageInfoEvent(page = self)
1120	wx.PostEvent(self.parent, new_event)
1121	#update list of plugins if new plugin is available
1122	if self.plugin_rbutton.GetValue():
1123	temp = self.parent.update_model_list()
1124	if temp:
1125	self.model_list_box = temp
1126	current_val = self.formfactorbox.GetValue()
1127	pos = self.formfactorbox.GetSelection()
1128	self._show_combox_helper()
1129	self.formfactorbox.SetSelection(pos)
1130	self.formfactorbox.SetValue(current_val)
1131	self._onDraw(event=None)
1132	else:
1133	self._draw_model()
1134	self.SetupScrolling()
1135	self.Show(True)
1136
1137	def _onparamEnter(self,event):
1138	"""
1139	when enter value on panel redraw model according to changed
1140	"""
1141	if self.model ==None:
1142	msg="Please select a Model first..."
1143	wx.MessageBox(msg, 'Info')
1144	#wx.PostEvent(self._manager.parent, StatusEvent(status=\
1145	# "Parameters: %s"%msg))
1146	return
1147
1148	#default flag
1149	flag = False
1150	self.fitrange = True
1151	#get event object
1152	tcrtl = event.GetEventObject()
1153
1154	#wx.PostEvent(self._manager.parent, StatusEvent(status=" \
1155	# updating ... ",type="update"))
1156	#Clear msg if previously shown.
1157	msg= ""
1158	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
1159
1160	if check_float(tcrtl):
1161	flag = self._onparamEnter_helper()
1162	self.set_npts2fit()
1163	if self.fitrange:
1164	temp_smearer = None
1165	if not self.disable_smearer.GetValue():
1166	temp_smearer= self.current_smearer
1167	## set smearing value whether or not
1168	# the data contain the smearing info
1169	if self.slit_smearer.GetValue():
1170	flag1 = self.update_slit_smear()
1171	flag = flag or flag1
1172	elif self.pinhole_smearer.GetValue():
1173	flag1 = self.update_pinhole_smear()
1174	flag = flag or flag1
1175	elif self.data.__class__.__name__ != "Data2D" and \
1176	not self.enable2D:
1177	self._manager.set_smearer(smearer=temp_smearer,
1178	enable2D=self.enable2D,
1179	uid=self.uid,
1180	qmin= float(self.qmin_x),
1181	qmax= float(self.qmax_x),
1182	draw=True)
1183	if flag:
1184	#self.compute_chisqr(smearer= temp_smearer)
1185
1186	## new state posted
1187	if self.state_change:
1188	#self._undo.Enable(True)
1189	event = PageInfoEvent(page = self)
1190	wx.PostEvent(self.parent, event)
1191	self.state_change= False
1192	else:
1193	# invalid fit range: do nothing here:
1194	# msg already displayed in validate
1195	return
1196	else:
1197	self.save_current_state()
1198	msg= "Cannot Plot :Must enter a number!!! "
1199	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
1200
1201	self.save_current_state()
1202	return
1203
1204	def _onparamRangeEnter(self, event):
1205	"""
1206	Check validity of value enter in the parameters range field
1207	"""
1208	#if self.check_invalid_panel():
1209	# return
1210	tcrtl= event.GetEventObject()
1211	#Clear msg if previously shown.
1212	msg= ""
1213	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
1214	# Flag to register when a parameter has changed.
1215	is_modified = False
1216	if tcrtl.GetValue().lstrip().rstrip()!="":
1217	try:
1218	value = float(tcrtl.GetValue())
1219	tcrtl.SetBackgroundColour(wx.WHITE)
1220	self._check_value_enter(self.fittable_param ,is_modified)
1221	self._check_value_enter(self.parameters ,is_modified)
1222	except:
1223	tcrtl.SetBackgroundColour("pink")
1224	msg= "Model Error:wrong value entered : %s"% sys.exc_value
1225	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
1226	return
1227	else:
1228	tcrtl.SetBackgroundColour(wx.WHITE)
1229
1230	#self._undo.Enable(True)
1231	self.save_current_state()
1232	event = PageInfoEvent(page = self)
1233	wx.PostEvent(self.parent, event)
1234	self.state_change= False
1235
1236
1237	def _onQrangeEnter(self, event):
1238	"""
1239	Check validity of value enter in the Q range field
1240	"""
1241	#if self.check_invalid_panel():
1242	# return
1243	tcrtl= event.GetEventObject()
1244	#Clear msg if previously shown.
1245	msg= ""
1246	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
1247	# Flag to register when a parameter has changed.
1248	is_modified = False
1249	if tcrtl.GetValue().lstrip().rstrip()!="":
1250	try:
1251	value = float(tcrtl.GetValue())
1252	tcrtl.SetBackgroundColour(wx.WHITE)
1253
1254	# If qmin and qmax have been modified, update qmin and qmax
1255	if self._validate_qrange( self.qmin, self.qmax):
1256	tempmin = float(self.qmin.GetValue())
1257	if tempmin != self.qmin_x:
1258	self.qmin_x = tempmin
1259	tempmax = float(self.qmax.GetValue())
1260	if tempmax != self.qmax_x:
1261	self.qmax_x = tempmax
1262	else:
1263	tcrtl.SetBackgroundColour("pink")
1264	msg= "Model Error:wrong value entered : %s"% sys.exc_value
1265	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
1266	return
1267
1268	except:
1269	tcrtl.SetBackgroundColour("pink")
1270	msg= "Model Error:wrong value entered : %s"% sys.exc_value
1271	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
1272	return
1273	#Check if # of points for theory model are valid(>0).
1274	# check for 2d
1275	if self.data is not None:
1276	if self.data.__class__.__name__ == "Data2D" or \
1277	self.enable2D:
1278	# set mask
1279	radius= numpy.sqrt( self.data.qx_data*self.data.qx_data +
1280	self.data.qy_data*self.data.qy_data )
1281	index_data = ((self.qmin_x <= radius)& \
1282	(radius<= self.qmax_x))
1283	index_data = (index_data)&(self.data.mask)
1284	index_data = (index_data)&(numpy.isfinite(self.data.data))
1285	if len(index_data[index_data]) < 10:
1286	msg = "Cannot Plot :No or too little npts in"
1287	msg += " that data range!!! "
1288	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
1289	return
1290	else:
1291	self.data.mask = index_data
1292	#self.Npts_fit.SetValue(str(len(self.data.mask)))
1293	self.set_npts2fit()
1294	else:
1295	index_data = ((self.qmin_x <= self.data.x)& \
1296	(self.data.x <= self.qmax_x))
1297	self.Npts_fit.SetValue(str(len(self.data.x[index_data])))
1298	else:
1299	self.npts_x = self.Npts_total.GetValue()
1300	self._save_plotting_range()
1301
1302	else:
1303	tcrtl.SetBackgroundColour("pink")
1304	msg= "Model Error:wrong value entered!!!"
1305	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
1306
1307	self._draw_model()
1308	##update chi2
1309	#self.compute_chisqr(smearer= temp_smearer)
1310	#self._undo.Enable(True)
1311	self.save_current_state()
1312	event = PageInfoEvent(page = self)
1313	wx.PostEvent(self.parent, event)
1314	self.state_change= False
1315	return
1316
1317	def _clear_Err_on_Fit(self):
1318	"""
1319	hide the error text control shown
1320	after fitting
1321	"""
1322	if self.is_mac:
1323	return
1324	if hasattr(self,"text2_3"):
1325	self.text2_3.Hide()
1326
1327	if len(self.parameters)>0:
1328	for item in self.parameters:
1329	#Skip t ifhe angle parameters if 1D data
1330	if self.data.__class__.__name__ != "Data2D" and \
1331	not self.enable2D:
1332	if item in self.orientation_params:
1333	continue
1334	if item in self.param_toFit:
1335	continue
1336	## hide statictext +/-
1337	if len(item) < 4 :
1338	continue
1339	if item[3]!=None and item[3].IsShown():
1340	item[3].Hide()
1341	## hide textcrtl for error after fit
1342	if item[4]!=None and item[4].IsShown():
1343	item[4].Hide()
1344
1345	if len(self.fittable_param)>0:
1346	for item in self.fittable_param:
1347	#Skip t ifhe angle parameters if 1D data
1348	if self.data.__class__.__name__ != "Data2D" and \
1349	not self.enable2D:
1350	if item in self.orientation_params:
1351	continue
1352	if item in self.param_toFit:
1353	continue
1354	if len(item) < 4 :
1355	continue
1356	## hide statictext +/-
1357	if item[3]!=None and item[3].IsShown():
1358	item[3].Hide()
1359	## hide textcrtl for error after fit
1360	if item[4]!=None and item[4].IsShown():
1361	item[4].Hide()
1362	return
1363
1364	def _get_defult_custom_smear(self):
1365	"""
1366	Get the defult values for custum smearing.
1367	"""
1368	# get the default values
1369	if self.dxl == None: self.dxl = 0.0
1370	if self.dxw == None: self.dxw = ""
1371	if self.dx_min == None: self.dx_min = SMEAR_SIZE_L
1372	if self.dx_max == None: self.dx_max = SMEAR_SIZE_H
1373
1374	def _get_smear_info(self):
1375	"""
1376	Get the smear info from data.
1377
1378	:return: self.smear_type, self.dq_l and self.dq_r,
1379	respectively the type of the smear, dq_min and
1380	dq_max for pinhole smear data
1381	while dxl and dxw for slit smear
1382
1383	"""
1384
1385	# default
1386	self.smear_type = None
1387	self.dq_l = None
1388	self.dq_r = None
1389	data = self.data
1390	if self.data is None:
1391	return
1392	elif self.data.__class__.__name__ == "Data2D" or \
1393	self.enable2D:
1394	if data.dqx_data == None or data.dqy_data ==None:
1395	return
1396	elif self.smearer != None and (data.dqx_data.any()!=0) and \
1397	(data.dqx_data.any()!=0):
1398	self.smear_type = "Pinhole2d"
1399	self.dq_l = format_number(numpy.average(data.dqx_data))
1400	self.dq_r = format_number(numpy.average(data.dqy_data))
1401	return
1402	else:
1403	return
1404	# check if it is pinhole smear and get min max if it is.
1405	if data.dx != None and all(data.dx !=0):
1406	self.smear_type = "Pinhole"
1407	self.dq_l = data.dx[0]
1408	self.dq_r = data.dx[-1]
1409
1410	# check if it is slit smear and get min max if it is.
1411	elif data.dxl != None or data.dxw != None:
1412	self.smear_type = "Slit"
1413	if data.dxl != None and all(data.dxl !=0):
1414	self.dq_l = data.dxl[0]
1415	if data.dxw != None and all(data.dxw !=0):
1416	self.dq_r = data.dxw[0]
1417	#return self.smear_type,self.dq_l,self.dq_r
1418
1419	def _show_smear_sizer(self):
1420	"""
1421	Show only the sizers depending on smear selection
1422	"""
1423	# smear disabled
1424	if self.disable_smearer.GetValue():
1425	self.smear_description_none.Show(True)
1426	# 2Dsmear
1427	elif self._is_2D():
1428	self.smear_description_accuracy_type.Show(True)
1429	self.smear_accuracy.Show(True)
1430	self.smear_description_accuracy_type.Show(True)
1431	self.smear_description_2d.Show(True)
1432	self.smear_description_2d_x.Show(True)
1433	self.smear_description_2d_y.Show(True)
1434	if self.pinhole_smearer.GetValue():
1435	self.smear_pinhole_min.Show(True)
1436	self.smear_pinhole_max.Show(True)
1437	# smear from data
1438	elif self.enable_smearer.GetValue():
1439
1440	self.smear_description_dqdata.Show(True)
1441	if self.smear_type != None:
1442	self.smear_description_smear_type.Show(True)
1443	if self.smear_type == 'Slit':
1444	self.smear_description_slit_height.Show(True)
1445	self.smear_description_slit_width.Show(True)
1446	elif self.smear_type == 'Pinhole':
1447	self.smear_description_pin_min.Show(True)
1448	self.smear_description_pin_max.Show(True)
1449	self.smear_description_smear_type.Show(True)
1450	self.smear_description_type.Show(True)
1451	self.smear_data_left.Show(True)
1452	self.smear_data_right.Show(True)
1453	# custom pinhole smear
1454	elif self.pinhole_smearer.GetValue():
1455	if self.smear_type == 'Pinhole':
1456	self.smear_message_new_p.Show(True)
1457	self.smear_description_pin_min.Show(True)
1458	self.smear_description_pin_max.Show(True)
1459
1460	self.smear_pinhole_min.Show(True)
1461	self.smear_pinhole_max.Show(True)
1462	# custom slit smear
1463	elif self.slit_smearer.GetValue():
1464	self.smear_message_new_s.Show(True)
1465	self.smear_description_slit_height.Show(True)
1466	self.smear_slit_height.Show(True)
1467	self.smear_description_slit_width.Show(True)
1468	self.smear_slit_width.Show(True)
1469
1470	def _hide_all_smear_info(self):
1471	"""
1472	Hide all smearing messages in the set_smearer sizer
1473	"""
1474	self.smear_description_none.Hide()
1475	self.smear_description_dqdata.Hide()
1476	self.smear_description_type.Hide()
1477	self.smear_description_smear_type.Hide()
1478	self.smear_description_accuracy_type.Hide()
1479	self.smear_description_2d_x.Hide()
1480	self.smear_description_2d_y.Hide()
1481	self.smear_description_2d.Hide()
1482
1483	self.smear_accuracy.Hide()
1484	self.smear_data_left.Hide()
1485	self.smear_data_right.Hide()
1486	self.smear_description_pin_min.Hide()
1487	self.smear_pinhole_min.Hide()
1488	self.smear_description_pin_max.Hide()
1489	self.smear_pinhole_max.Hide()
1490	self.smear_description_slit_height.Hide()
1491	self.smear_slit_height.Hide()
1492	self.smear_description_slit_width.Hide()
1493	self.smear_slit_width.Hide()
1494	self.smear_message_new_p.Hide()
1495	self.smear_message_new_s.Hide()
1496
1497	def _set_accuracy_list(self):
1498	"""
1499	Set the list of an accuracy in 2D custum smear: Xhigh, High, Med, or Low
1500	"""
1501	# list of accuracy choices
1502	list = ['Low','Med','High','Xhigh']
1503	for idx in range(len(list)):
1504	self.smear_accuracy.Append(list[idx],idx)
1505
1506	def _set_fun_box_list(self,fun_box):
1507	"""
1508	Set the list of func for multifunctional models
1509	"""
1510	# Check if it is multi_functional model
1511	if self.model.__class__ not in self.model_list_box["Multi-Functions"] \
1512	and not self.temp_multi_functional:
1513	return None
1514	# Get the func name list
1515	list = self.model.fun_list
1516	if len(list) == 0:
1517	return None
1518	# build function (combo)box
1519	ind = 0
1520	while(ind < len(list)):
1521	for key, val in list.iteritems():
1522	if (val == ind):
1523	fun_box.Append(key,val)
1524	break
1525	ind += 1
1526
1527	def _on_select_accuracy(self,event):
1528	"""
1529	Select an accuracy in 2D custom smear: Xhigh, High, Med, or Low
1530	"""
1531	#event.Skip()
1532	# Check if the accuracy is same as before
1533	#self.smear2d_accuracy = event.GetEventObject().GetValue()
1534	self.smear2d_accuracy = self.smear_accuracy.GetValue()
1535	if self.pinhole_smearer.GetValue():
1536	self.onPinholeSmear(event=None)
1537	else:
1538	self.onSmear(event=None)
1539	if self.current_smearer != None:
1540	self.current_smearer.set_accuracy(accuracy =
1541	self.smear2d_accuracy)
1542	event.Skip()
1543
1544	def _on_fun_box(self,event):
1545	"""
1546	Select an func: Erf,Rparabola,LParabola
1547	"""
1548	fun_val = None
1549	fun_box = event.GetEventObject()
1550	name = fun_box.Name
1551	value = fun_box.GetValue()
1552	if self.model.fun_list.has_key(value):
1553	fun_val = self.model.fun_list[value]
1554
1555	self.model.setParam(name,fun_val)
1556	# save state
1557	self._copy_parameters_state(self.str_parameters,
1558	self.state.str_parameters)
1559	# update params
1560	self._update_paramv_on_fit()
1561	# draw
1562	self._draw_model()
1563	self.Refresh()
1564	# get ready for new event
1565	event.Skip()
1566
1567	def _onMask(self, event):
1568	"""
1569	Build a panel to allow to edit Mask
1570	"""
1571
1572	from sans.guiframe.local_perspectives.plotting.masking \
1573	import MaskPanel as MaskDialog
1574
1575	self.panel = MaskDialog(base=self, data=self.data, id=wx.NewId())
1576	#self.panel.Bind(wx.EVT_CLOSE, self._draw_masked_model)
1577	self.panel.ShowModal()
1578	#wx.PostEvent(self.parent, event)
1579
1580	def _draw_masked_model(self, event):
1581	"""
1582	Draw model image w/mask
1583	"""
1584	#event.Skip()
1585	is_valid_qrange = self._update_paramv_on_fit()
1586
1587	if is_valid_qrange and self.model != None:
1588	#self.panel.Show(0)
1589	#self.panel.Destroy() # frame
1590	self.panel.MakeModal(False)
1591	event.Skip()
1592	# try re draw the model plot if it exists
1593	self._draw_model()
1594	self.set_npts2fit()
1595	elif self.model == None:
1596	#self.panel.Show(0)
1597	#self.panel.Destroy() # frame
1598	self.panel.MakeModal(False)
1599	event.Skip()
1600	self.set_npts2fit()
1601	msg= "No model is found on updating MASK in the model plot... "
1602	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
1603	else:
1604	event.Skip()
1605	msg = ' Please consider your Q range, too.'
1606	self.panel.ShowMessage(msg)
1607
1608
1609	def _set_smear(self, data):
1610	"""
1611	"""
1612	if data is None:
1613	return
1614	self.smearer = smear_selection(data, self.model)
1615	self.disable_smearer.SetValue(True)
1616	if self.smearer == None:
1617	self.enable_smearer.Disable()
1618	else:
1619	self.enable_smearer.Enable()
1620
1621	def _mac_sleep(self, sec=0.2):
1622	"""
1623	Give sleep to MAC
1624	"""
1625	if self.is_mac:
1626	time.sleep(sec)
1627
1628	def get_view_mode(self):
1629	"""
1630	return True if the panel allow 2D or False if 1D
1631	"""
1632	return self.enable2D
1633
1634	def set_data(self, data):
1635	"""
1636	reset the current data
1637	"""
1638	flag = False
1639	if self.data is None and data is not None:
1640	self.window_name = str(data.name)
1641	## Title to appear on top of the window
1642	self.window_caption = str(data.name)
1643	flag = True
1644	self.data = data
1645	if self.data is None:
1646	data_min = ""
1647	data_max = ""
1648	data_name = ""
1649	self._set_bookmark_flag(False)
1650	self._keep.Enable(False)
1651	self._set_save_flag(False)
1652	else:
1653	if self.model != None:
1654	self._set_bookmark_flag(True)
1655	self._keep.Enable(True)
1656	self._set_save_flag(True)
1657	self._set_preview_flag(True)
1658
1659	self._set_smear(data)
1660	# more disables for 2D
1661	if self.data.__class__.__name__ == "Data2D" or \
1662	self.enable2D:
1663	self.slit_smearer.Disable()
1664	self.pinhole_smearer.Enable(True)
1665	self.default_mask = copy.deepcopy(self.data.mask)
1666	else:
1667	self.slit_smearer.Enable(True)
1668	self.pinhole_smearer.Enable(True)
1669
1670	self.formfactorbox.Enable()
1671	self.structurebox.Enable()
1672	data_name = self.data.name
1673	#set maximum range for x in linear scale
1674	if not hasattr(self.data,"data"): #Display only for 1D data fit
1675	# Minimum value of data
1676	data_min = min(self.data.x)
1677	# Maximum value of data
1678	data_max = max(self.data.x)
1679	#number of total data points
1680	self.Npts_total.SetValue(str(len(self.data.x)))
1681	#default:number of data points selected to fit
1682	self.Npts_fit.SetValue(str(len(self.data.x)))
1683	self.btEditMask.Disable()
1684	self.EditMask_title.Disable()
1685	else:
1686
1687	## Minimum value of data
1688	data_min = 0
1689	x = max(math.fabs(self.data.xmin), math.fabs(self.data.xmax))
1690	y = max(math.fabs(self.data.ymin), math.fabs(self.data.ymax))
1691	## Maximum value of data
1692	data_max = math.sqrt(xx + yy)
1693	#number of total data points
1694	self.Npts_total.SetValue(str(len(self.data.data)))
1695	#default:number of data points selected to fit
1696	self.Npts_fit.SetValue(str(len(self.data.data)))
1697	self.btEditMask.Enable()
1698	self.EditMask_title.Enable()
1699	self.Npts_total.SetEditable(False)
1700	self.Npts_total.SetBackgroundColour(\
1701	self.GetParent().GetBackgroundColour())
1702
1703	self.Npts_total.Bind(wx.EVT_MOUSE_EVENTS, self._npts_click)
1704	#self.Npts_total.Disable()
1705	self.dataSource.SetValue(data_name)
1706	self.qmin_x = data_min
1707	self.qmax_x = data_max
1708	#self.minimum_q.SetValue(str(data_min))
1709	#self.maximum_q.SetValue(str(data_max))
1710	self.qmin.SetValue(str(data_min))
1711	self.qmax.SetValue(str(data_max))
1712	self.qmin.SetBackgroundColour("white")
1713	self.qmax.SetBackgroundColour("white")
1714	self.state.data = data
1715	self.state.qmin = self.qmin_x
1716	self.state.qmax = self.qmax_x
1717
1718	#update model plot with new data information
1719	if flag:
1720	#set model view button
1721	if self.data.__class__.__name__ == "Data2D":
1722	self.enable2D = True
1723	self.model_view.SetLabel("2D Mode")
1724	else:
1725	self.enable2D = False
1726	self.model_view.SetLabel("1D Mode")
1727
1728	self.model_view.Disable()
1729	self._draw_model()
1730
1731	def _npts_click(self, event):
1732	"""
1733	Prevent further handling of the mouse event on Npts_total
1734	by not calling Skip().
1735	"""
1736	pass
1737
1738	def reset_page(self, state,first=False):
1739	"""
1740	reset the state
1741	"""
1742	self.reset_page_helper(state)
1743	#import sans.guiframe.gui_manager
1744	#evt = ModelEventbox(model=state.model)
1745	#wx.PostEvent(self.event_owner, evt)
1746
1747	if self.engine_type != None:
1748	self._manager._on_change_engine(engine=self.engine_type)
1749
1750	self.select_param(event = None)
1751	#Save state_fit
1752	self.save_current_state_fit()
1753	self._lay_out()
1754	self.Refresh()
1755
1756	def get_range(self):
1757	"""
1758	return the fitting range
1759	"""
1760	return float(self.qmin_x) , float(self.qmax_x)
1761
1762	def get_npts2fit(self):
1763	"""
1764	return numbers of data points within qrange
1765
1766	:Note: This is for Park where chi2 is not normalized by Npts of fit
1767
1768	"""
1769	if self.data is None:
1770	return
1771	npts2fit = 0
1772	qmin,qmax = self.get_range()
1773	if self.data.__class__.__name__ == "Data2D" or \
1774	self.enable2D:
1775	radius= numpy.sqrt( self.data.qx_data*self.data.qx_data +
1776	self.data.qy_data*self.data.qy_data )
1777	index_data = (self.qmin_x <= radius)&(radius<= self.qmax_x)
1778	index_data= (index_data)&(self.data.mask)
1779	index_data = (index_data)&(numpy.isfinite(self.data.data))
1780	npts2fit = len(self.data.data[index_data])
1781	else:
1782	for qx in self.data.x:
1783	if qx >= qmin and qx <= qmax:
1784	npts2fit += 1
1785	return npts2fit
1786
1787	def set_npts2fit(self):
1788	"""
1789	setValue Npts for fitting
1790	"""
1791	self.Npts_fit.SetValue(str(self.get_npts2fit()))
1792
1793	def get_chi2(self):
1794	"""
1795	return the current chi2
1796	"""
1797	return self.tcChi.GetValue()
1798
1799	def get_param_list(self):
1800	"""
1801	:return self.param_toFit: list containing references to TextCtrl
1802	checked.Theses TextCtrl will allow reference to parameters to fit.
1803
1804	:raise: if return an empty list of parameter fit will nnote work
1805	properly so raise ValueError,"missing parameter to fit"
1806	"""
1807	if self.param_toFit !=[]:
1808	return self.param_toFit
1809	else:
1810	msg = "missing parameters to fit"
1811	wx.MessageBox(msg, 'warning')
1812	return False
1813	#raise ValueError,"missing parameters to fit"
1814
1815	def onsetValues(self, chisqr, p_name, out, cov):
1816	"""
1817	Build the panel from the fit result
1818
1819	:param chisqr: Value of the goodness of fit metric
1820	:param p_name: the name of parameters
1821	:param out: list of parameter with the best value found during fitting
1822	:param cov: Covariance matrix
1823
1824	"""
1825	if out == None or not numpy.isfinite(chisqr):
1826	raise ValueError,"Fit error occured..."
1827
1828	is_modified = False
1829	has_error = False
1830	dispersity = ''
1831
1832	#Hide textctrl boxes of errors.
1833	self._clear_Err_on_Fit()
1834
1835	#Check if chi2 is finite
1836	if chisqr != None and numpy.isfinite(chisqr):
1837	#format chi2
1838	if self.engine_type == "park":
1839	npt_fit = float(self.get_npts2fit())
1840	if npt_fit > 0:
1841	chisqr =chisqr/npt_fit
1842	chi2 = format_number(chisqr, True)
1843	self.tcChi.SetValue(chi2)
1844	self.tcChi.Refresh()
1845	else:
1846	self.tcChi.SetValue("-")
1847
1848	#Hide error title
1849	if self.text2_3.IsShown() and not self.is_mac:
1850	self.text2_3.Hide()
1851
1852	try:
1853	if self.enable_disp.GetValue():
1854	if hasattr(self,"text_disp_1" ):
1855	if self.text_disp_1 != None and not self.is_mac:
1856	self.text_disp_1.Hide()
1857	except:
1858	dispersity = None
1859	pass
1860	#set the panel when fit result are float not list
1861	if out.__class__== numpy.float64:
1862	self.param_toFit[0][2].SetValue(format_number(out, True))
1863
1864	if self.param_toFit[0][4].IsShown and not self.is_mac:
1865	self.param_toFit[0][4].Hide()
1866	if cov !=None :
1867	self.text2_3.Show(True)
1868	try:
1869	if self.enable_disp.GetValue():
1870	if hasattr(self,"text_disp_1" ):
1871	if self.text_disp_1 !=None:
1872	self.text_disp_1.Show(True)
1873	except:
1874	pass
1875
1876	if cov[0]==None or not numpy.isfinite(cov[0]):
1877	if self.param_toFit[0][3].IsShown and not self.is_mac:
1878	self.param_toFit[0][3].Hide()
1879	else:
1880	self.param_toFit[0][3].Show(True)
1881	self.param_toFit[0][4].Show(True)
1882	self.param_toFit[0][4].SetValue(format_number(cov[0], True))
1883	has_error = True
1884	else:
1885
1886	i = 0
1887	#Set the panel when fit result are list
1888	for item in self.param_toFit:
1889	if len(item)>5 and item != None:
1890	## reset error value to initial state
1891	if not self.is_mac:
1892	item[3].Hide()
1893	item[4].Hide()
1894
1895	for ind in range(len(out)):
1896
1897	if item[1] == p_name[ind]:
1898	break
1899	if len(out)<=len(self.param_toFit) and out[ind] !=None:
1900	val_out = format_number(out[ind], True)
1901	item[2].SetValue(val_out)
1902
1903
1904	if(cov !=None):
1905
1906	try:
1907	if dispersity !=None:
1908	if self.enable_disp.GetValue():
1909	if hasattr(self,"text_disp_1" ):
1910	if self.text_disp_1!=None:
1911	if not self.text_disp_1.IsShown()\
1912	and not self.is_mac:
1913	self.text_disp_1.Show(True)
1914	except:
1915	pass
1916
1917	if cov[ind]!=None :
1918	if numpy.isfinite(float(cov[ind])):
1919	val_err = format_number(cov[ind], True)
1920	if not self.is_mac:
1921	item[3].Show(True)
1922	item[4].Show(True)
1923	item[4].SetValue(val_err)
1924
1925	has_error = True
1926	i += 1
1927	#Show error title when any errors displayed
1928	if has_error:
1929	if not self.text2_3.IsShown():
1930	self.text2_3.Show(True)
1931	## save current state
1932	self.save_current_state()
1933
1934	#self._lay_out()
1935	if not self.is_mac:
1936	self.Layout()
1937	self.Refresh()
1938	self._mac_sleep(0.1)
1939	#plot model ( when drawing, do not update chisqr value again)
1940	self._draw_model(update_chisqr=False)
1941	#PostStatusEvent
1942	#msg = "Fit completed!dddd "
1943	#wx.PostEvent(self._manager.parent, StatusEvent(status=msg))
1944
1945	def onPinholeSmear(self, event):
1946	"""
1947	Create a custom pinhole smear object that will change the way residuals
1948	are compute when fitting
1949
1950	:Note: accuracy is given by strings'High','Med', 'Low' FOR 2d,
1951	None for 1D
1952
1953	"""
1954
1955	#if self.check_invalid_panel():
1956	# return
1957	if self.model ==None:
1958	self.disable_smearer.SetValue(True)
1959	msg="Please select a Model first..."
1960	wx.MessageBox(msg, 'Info')
1961	wx.PostEvent(self._manager.parent, StatusEvent(status=\
1962	"Smear: %s"%msg))
1963	return
1964
1965	# Need update param values
1966	self._update_paramv_on_fit()
1967
1968	# msg default
1969	msg = None
1970	if event != None:
1971	tcrtl= event.GetEventObject()
1972	# event case of radio button
1973	if tcrtl.GetValue()== True:
1974	self.dx_min = 0.0
1975	self.dx_max = 0.0
1976	is_new_pinhole = True
1977	else:
1978	is_new_pinhole = self._is_changed_pinhole()
1979	else:
1980	is_new_pinhole = True
1981	# if any value is changed
1982	if is_new_pinhole:
1983	msg = self._set_pinhole_smear()
1984	# hide all silt sizer
1985	self._hide_all_smear_info()
1986
1987	##Calculate chi2
1988	temp_smearer = self.current_smearer
1989	#self.compute_chisqr(smearer= temp_smearer)
1990
1991	# show relevant slit sizers
1992	self._show_smear_sizer()
1993
1994	self.sizer_set_smearer.Layout()
1995	self.Layout()
1996
1997	if event != None:
1998	event.Skip()
1999	#self._undo.Enable(True)
2000	self.save_current_state()
2001	event = PageInfoEvent(page = self)
2002	wx.PostEvent(self.parent, event)
2003
2004	def _is_changed_pinhole(self):
2005	"""
2006	check if any of pinhole smear is changed
2007
2008	:return: True or False
2009
2010	"""
2011	# get the values
2012	pin_min = self.smear_pinhole_min.GetValue()
2013	pin_max = self.smear_pinhole_max.GetValue()
2014
2015	# Check changes in slit width
2016	try:
2017	dx_min = float(pin_min)
2018	except:
2019	return True
2020	if self.dx_min != dx_min:
2021	return True
2022
2023	# Check changes in slit heigth
2024	try:
2025	dx_max = float(pin_max)
2026	except:
2027	return True
2028	if self.dx_max != dx_max:
2029	return True
2030	return False
2031
2032	def _set_pinhole_smear(self):
2033	"""
2034	Set custom pinhole smear
2035
2036	:return: msg
2037
2038	"""
2039	# copy data
2040	data = copy.deepcopy(self.data)
2041	if self._is_2D():
2042	self.smear_type = 'Pinhole2d'
2043	len_data = len(data.data)
2044	data.dqx_data = numpy.zeros(len_data)
2045	data.dqy_data = numpy.zeros(len_data)
2046	else:
2047	self.smear_type = 'Pinhole'
2048	len_data = len(data.x)
2049	data.dx = numpy.zeros(len_data)
2050	data.dxl = None
2051	data.dxw = None
2052	msg = None
2053
2054	get_pin_min = self.smear_pinhole_min
2055	get_pin_max = self.smear_pinhole_max
2056
2057	if not check_float(get_pin_min):
2058	get_pin_min.SetBackgroundColour("pink")
2059	msg= "Model Error:wrong value entered!!!"
2060	elif not check_float(get_pin_max ):
2061	get_pin_max.SetBackgroundColour("pink")
2062	msg= "Model Error:wrong value entered!!!"
2063	else:
2064	if len_data < 2: len_data = 2
2065	self.dx_min = float(get_pin_min.GetValue())
2066	self.dx_max = float(get_pin_max.GetValue())
2067	if self.dx_min < 0:
2068	get_pin_min.SetBackgroundColour("pink")
2069	msg= "Model Error:This value can not be negative!!!"
2070	elif self.dx_max <0:
2071	get_pin_max.SetBackgroundColour("pink")
2072	msg= "Model Error:This value can not be negative!!!"
2073	elif self.dx_min != None and self.dx_max != None:
2074	if self._is_2D():
2075	data.dqx_data[data.dqx_data==0] = self.dx_min
2076	data.dqy_data[data.dqy_data==0] = self.dx_max
2077	elif self.dx_min == self.dx_max:
2078	data.dx[data.dx==0] = self.dx_min
2079	else:
2080	step = (self.dx_max - self.dx_min)/(len_data-1)
2081	data.dx = numpy.arange(self.dx_min,self.dx_max+step/1.1,
2082	step)
2083	elif self.dx_min != None:
2084	if self._is_2D(): data.dqx_data[data.dqx_data==0] = self.dx_min
2085	else: data.dx[data.dx==0] = self.dx_min
2086	elif self.dx_max != None:
2087	if self._is_2D(): data.dqy_data[data.dqy_data==0] = self.dx_max
2088	else: data.dx[data.dx==0] = self.dx_max
2089	self.current_smearer = smear_selection(data, self.model)
2090	# 2D need to set accuracy
2091	if self._is_2D():
2092	self.current_smearer.set_accuracy(accuracy = \
2093	self.smear2d_accuracy)
2094
2095	if msg != None:
2096	wx.PostEvent(self._manager.parent, StatusEvent(status = msg ))
2097	else:
2098	get_pin_min.SetBackgroundColour("white")
2099	get_pin_max.SetBackgroundColour("white")
2100	## set smearing value whether or not the data contain the smearing info
2101	self._manager.set_smearer(smearer=self.current_smearer,
2102	enable2D=self.enable2D,
2103	qmin=float(self.qmin_x),
2104	qmax= float(self.qmax_x),
2105	uid=self.uid)
2106	return msg
2107
2108	def update_pinhole_smear(self):
2109	"""
2110	called by kill_focus on pinhole TextCntrl
2111	to update the changes
2112
2113	:return: False when wrong value was entered
2114
2115	"""
2116	# msg default
2117	msg = None
2118	# check if any value is changed
2119	if self._is_changed_pinhole():
2120	msg = self._set_pinhole_smear()
2121	#self._undo.Enable(True)
2122	self.save_current_state()
2123
2124	if msg != None:
2125	return False
2126	else:
2127	return True
2128
2129	def onSlitSmear(self, event):
2130	"""
2131	Create a custom slit smear object that will change the way residuals
2132	are compute when fitting
2133	"""
2134
2135	#if self.check_invalid_panel():
2136	# return
2137
2138	if self.model ==None:
2139	self.disable_smearer.SetValue(True)
2140
2141	msg="Please select a Model first..."
2142	wx.MessageBox(msg, 'Info')
2143	wx.PostEvent(self._manager.parent, StatusEvent(status=\
2144	"Smear: %s"%msg))
2145	return
2146
2147	# Need update param values
2148	self._update_paramv_on_fit()
2149
2150	# msg default
2151	msg = None
2152	# for event given
2153	if event != None:
2154	tcrtl= event.GetEventObject()
2155	# event case of radio button
2156	if tcrtl.GetValue():
2157	self.dxl = 0.0
2158	self.dxw = 0.0
2159	is_new_slit = True
2160	else:
2161	is_new_slit = self._is_changed_slit()
2162	else:
2163	is_new_slit = True
2164
2165	# if any value is changed
2166	if is_new_slit:
2167	msg = self._set_slit_smear()
2168
2169	# hide all silt sizer
2170	self._hide_all_smear_info()
2171	##Calculate chi2
2172	#self.compute_chisqr(smearer= self.current_smearer)
2173	# show relevant slit sizers
2174	self._show_smear_sizer()
2175	self.sizer_set_smearer.Layout()
2176	self.Layout()
2177
2178	if event != None:
2179	event.Skip()
2180	#self._undo.Enable(True)
2181	self.save_current_state()
2182	event = PageInfoEvent(page = self)
2183	wx.PostEvent(self.parent, event)
2184	if msg != None:
2185	wx.PostEvent(self._manager.parent, StatusEvent(status = msg))
2186
2187	def _is_changed_slit(self):
2188	"""
2189	check if any of slit lengths is changed
2190
2191	:return: True or False
2192
2193	"""
2194	# get the values
2195	width = self.smear_slit_width.GetValue()
2196	height = self.smear_slit_height.GetValue()
2197
2198	# check and change the box bg color if it was pink
2199	# but it should be white now
2200	# because this is the case that _set_slit_smear() will not handle
2201	if height.lstrip().rstrip()=="":
2202	self.smear_slit_height.SetBackgroundColour(wx.WHITE)
2203	if width.lstrip().rstrip()=="":
2204	self.smear_slit_width.SetBackgroundColour(wx.WHITE)
2205
2206	# Check changes in slit width
2207	if width == "":
2208	dxw = 0.0
2209	else:
2210	try:
2211	dxw = float(width)
2212	except:
2213	return True
2214	if self.dxw != dxw:
2215	return True
2216
2217	# Check changes in slit heigth
2218	if height == "":
2219	dxl = 0.0
2220	else:
2221	try:
2222	dxl = float(height)
2223	except:
2224	return True
2225	if self.dxl != dxl:
2226	return True
2227
2228	return False
2229
2230	def _set_slit_smear(self):
2231	"""
2232	Set custom slit smear
2233
2234	:return: message to inform the user about the validity
2235	of the values entered for slit smear
2236	"""
2237	if self.data.__class__.__name__ == "Data2D" or \
2238	self.enable2D:
2239	return
2240	temp_smearer = None
2241	# make sure once more if it is smearer
2242	data = copy.deepcopy(self.data)
2243	data_len = len(data.x)
2244	data.dx = None
2245	data.dxl = None
2246	data.dxw = None
2247	msg = None
2248
2249	try:
2250	self.dxl = float(self.smear_slit_height.GetValue())
2251	data.dxl = self.dxl* numpy.ones(data_len)
2252	self.smear_slit_height.SetBackgroundColour(wx.WHITE)
2253	except:
2254	data.dxl = numpy.zeros(data_len)
2255	if self.smear_slit_height.GetValue().lstrip().rstrip()!="":
2256	self.smear_slit_height.SetBackgroundColour("pink")
2257	msg = "Wrong value entered... "
2258	else:
2259	self.smear_slit_height.SetBackgroundColour(wx.WHITE)
2260	try:
2261	self.dxw = float(self.smear_slit_width.GetValue())
2262	self.smear_slit_width.SetBackgroundColour(wx.WHITE)
2263	data.dxw = self.dxw* numpy.ones(data_len)
2264	except:
2265	data.dxw = numpy.zeros(data_len)
2266	if self.smear_slit_width.GetValue().lstrip().rstrip()!="":
2267	self.smear_slit_width.SetBackgroundColour("pink")
2268	msg = "Wrong Fit value entered... "
2269	else:
2270	self.smear_slit_width.SetBackgroundColour(wx.WHITE)
2271
2272	self.current_smearer = smear_selection(data, self.model)
2273	#temp_smearer = self.current_smearer
2274	## set smearing value whether or not the data contain the smearing info
2275	self._manager.set_smearer(smearer=self.current_smearer,
2276	enable2D=self.enable2D,
2277	qmin=float(self.qmin_x),
2278	qmax= float(self.qmax_x),
2279	uid=self.uid)
2280	return msg
2281
2282	def update_slit_smear(self):
2283	"""
2284	called by kill_focus on pinhole TextCntrl
2285	to update the changes
2286
2287	:return: False when wrong value was entered
2288
2289	"""
2290	# msg default
2291	msg = None
2292	# check if any value is changed
2293	if self._is_changed_slit():
2294	msg = self._set_slit_smear()
2295	#self._undo.Enable(True)
2296	self.save_current_state()
2297
2298	if msg != None:
2299	return False
2300	else:
2301	return True
2302
2303	def onSmear(self, event):
2304	"""
2305	Create a smear object that will change the way residuals
2306	are compute when fitting
2307	"""
2308	if event != None:
2309	event.Skip()
2310	if self.data is None:
2311	return
2312
2313	if self.model == None:
2314	self.disable_smearer.SetValue(True)
2315	msg="Please select a Model first..."
2316	wx.MessageBox(msg, 'Info')
2317	wx.PostEvent(self._manager.parent, StatusEvent(status=\
2318	"Smear: %s"%msg))
2319	return
2320
2321	# Need update param values
2322	self._update_paramv_on_fit()
2323
2324	temp_smearer = None
2325	self._get_smear_info()
2326
2327	#renew smear sizer
2328	if self.smear_type != None:
2329	self.smear_description_smear_type.SetValue(str(self.smear_type))
2330	self.smear_data_left.SetValue(str(self.dq_l))
2331	self.smear_data_right.SetValue(str(self.dq_r))
2332
2333	self._hide_all_smear_info()
2334
2335	data = copy.deepcopy(self.data)
2336	# make sure once more if it is smearer
2337	self.current_smearer = smear_selection(data, self.model)
2338
2339	if self.enable_smearer.GetValue():
2340	if hasattr(self.data,"dxl"):
2341
2342	msg= ": Resolution smearing parameters"
2343	if hasattr(self.data,"dxw"):
2344	msg= ": Slit smearing parameters"
2345	if self.smearer ==None:
2346	wx.PostEvent(self._manager.parent, StatusEvent(status=\
2347	"Data contains no smearing information"))
2348	else:
2349	wx.PostEvent(self._manager.parent, StatusEvent(status=\
2350	"Data contains smearing information"))
2351
2352	#self.smear_description_dqdata.Show(True)
2353	self.smear_data_left.Show(True)
2354	self.smear_data_right.Show(True)
2355	temp_smearer= self.current_smearer
2356	elif self.disable_smearer.GetValue():
2357	self.smear_description_none.Show(True)
2358
2359	self._show_smear_sizer()
2360
2361	self.sizer_set_smearer.Layout()
2362	self.Layout()
2363	## set smearing value whether or not the data contain the smearing info
2364	self._manager.set_smearer(uid=self.uid, smearer=temp_smearer,
2365	enable2D=self.enable2D,
2366	qmin= float(self.qmin_x),
2367	qmax= float(self.qmax_x), draw=True)
2368
2369	self.state.enable_smearer= self.enable_smearer.GetValue()
2370	self.state.disable_smearer=self.disable_smearer.GetValue()
2371	self.state.pinhole_smearer = self.pinhole_smearer.GetValue()
2372	self.state.slit_smearer = self.slit_smearer.GetValue()
2373
2374	def on_complete_chisqr(self, event):
2375	"""
2376	print result chisqr
2377
2378	:event: activated by fitting/ complete after draw
2379
2380	"""
2381	try:
2382	if event == None:
2383	output= "-"
2384	elif not numpy.isfinite(event.output):
2385	output= "-"
2386	else:
2387	output = event.output
2388	self.tcChi.SetValue(str(format_number(output, True)))
2389
2390	self.state.tcChi = self.tcChi.GetValue()
2391	except:
2392	pass
2393
2394
2395	def select_all_param(self,event):
2396	"""
2397	set to true or false all checkBox given the main checkbox value cb1
2398	"""
2399	self.param_toFit=[]
2400	if self.parameters !=[]:
2401	if self.cb1.GetValue():
2402	for item in self.parameters:
2403	## for data2D select all to fit
2404	if self.data.__class__.__name__== "Data2D" or \
2405	self.enable2D:
2406	item[0].SetValue(True)
2407	self.param_toFit.append(item )
2408	else:
2409	## for 1D all parameters except orientation
2410	if not item in self.orientation_params:
2411	item[0].SetValue(True)
2412	self.param_toFit.append(item )
2413	#if len(self.fittable_param)>0:
2414	for item in self.fittable_param:
2415	if self.data.__class__.__name__== "Data2D" or \
2416	self.enable2D:
2417	item[0].SetValue(True)
2418	self.param_toFit.append(item )
2419	try:
2420	if len(self.values[item[1]]) > 0:
2421	item[0].SetValue(False)
2422	except:
2423	pass
2424
2425	else:
2426	## for 1D all parameters except orientation
2427	if not item in self.orientation_params_disp:
2428	item[0].SetValue(True)
2429	self.param_toFit.append(item )
2430	try:
2431	if len(self.values[item[1]]) > 0:
2432	item[0].SetValue(False)
2433	except:
2434	pass
2435
2436	else:
2437	for item in self.parameters:
2438	item[0].SetValue(False)
2439	for item in self.fittable_param:
2440	item[0].SetValue(False)
2441	self.param_toFit=[]
2442
2443	self.save_current_state_fit()
2444
2445	if event !=None:
2446	#self._undo.Enable(True)
2447	## post state to fit panel
2448	event = PageInfoEvent(page = self)
2449	wx.PostEvent(self.parent, event)
2450
2451	def select_param(self,event):
2452	"""
2453	Select TextCtrl checked for fitting purpose and stores them
2454	in self.param_toFit=[] list
2455	"""
2456	self.param_toFit=[]
2457	for item in self.parameters:
2458	#Skip t ifhe angle parameters if 1D data
2459	if self.data.__class__.__name__ != "Data2D" and\
2460	not self.enable2D:
2461	if item in self.orientation_params:
2462	continue
2463	#Select parameters to fit for list of primary parameters
2464	if item[0].GetValue():
2465	if not (item in self.param_toFit):
2466	self.param_toFit.append(item )
2467	else:
2468	#remove parameters from the fitting list
2469	if item in self.param_toFit:
2470	self.param_toFit.remove(item)
2471
2472	#Select parameters to fit for list of fittable parameters
2473	# with dispersion
2474	for item in self.fittable_param:
2475	#Skip t ifhe angle parameters if 1D data
2476	if self.data.__class__.__name__ != "Data2D" and\
2477	not self.enable2D:
2478	if item in self.orientation_params:
2479	continue
2480	if item[0].GetValue():
2481	if not (item in self.param_toFit):
2482	self.param_toFit.append(item)
2483	else:
2484	#remove parameters from the fitting list
2485	if item in self.param_toFit:
2486	self.param_toFit.remove(item)
2487
2488	#Calculate num. of angle parameters
2489	if self.data.__class__.__name__ == "Data2D" or \
2490	self.enable2D:
2491	len_orient_para = 0
2492	else:
2493	len_orient_para = len(self.orientation_params) #assume even len
2494	#Total num. of angle parameters
2495	if len(self.fittable_param) > 0:
2496	len_orient_para *= 2
2497	#Set the value of checkbox that selected every checkbox or not
2498	if len(self.parameters)+len(self.fittable_param)-len_orient_para ==\
2499	len(self.param_toFit):
2500	self.cb1.SetValue(True)
2501	else:
2502	self.cb1.SetValue(False)
2503	self.save_current_state_fit()
2504	if event !=None:
2505	#self._undo.Enable(True)
2506	## post state to fit panel
2507	event = PageInfoEvent(page = self)
2508	wx.PostEvent(self.parent, event)
2509
2510	def set_model_param_sizer(self, model):
2511	"""
2512	Build the panel from the model content
2513
2514	:param model: the model selected in combo box for fitting purpose
2515
2516	"""
2517	self.sizer3.Clear(True)
2518	self.parameters = []
2519	self.str_parameters = []
2520	self.param_toFit=[]
2521	self.fittable_param=[]
2522	self.fixed_param=[]
2523	self.orientation_params=[]
2524	self.orientation_params_disp=[]
2525
2526	if model ==None:
2527	self.sizer3.Layout()
2528	self.SetupScrolling()
2529	return
2530	## the panel is drawn using the current value of the fit engine
2531	if self.engine_type==None and self._manager !=None:
2532	self.engine_type= self._manager._return_engine_type()
2533
2534	box_description= wx.StaticBox(self, -1,str("Model Parameters"))
2535	boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
2536	sizer = wx.GridBagSizer(5,5)
2537	## save the current model
2538	self.model = model
2539
2540	keys = self.model.getParamList()
2541	#list of dispersion paramaters
2542	self.disp_list=self.model.getDispParamList()
2543
2544	def custom_compare(a,b):
2545	"""
2546	Custom compare to order, first by alphabets then second by number.
2547	"""
2548	# number at the last digit
2549	a_last = a[len(a)-1]
2550	b_last = b[len(b)-1]
2551	# default
2552	num_a = None
2553	num_b = None
2554	# split the names
2555	a2 = a.lower().split('_')
2556	b2 = b.lower().split('_')
2557	# check length of a2, b2
2558	len_a2 = len(a2)
2559	len_b2 = len(b2)
2560	# check if it contains a int number(<10)
2561	try:
2562	num_a = int(a_last)
2563	except: pass
2564	try:
2565	num_b = int(b_last)
2566	except: pass
2567	# Put 'scale' near the top; happens
2568	# when numbered param name exists
2569	if a == 'scale':
2570	return -1
2571	# both have a number
2572	if num_a != None and num_b != None:
2573	if num_a > num_b: return -1
2574	# same number
2575	elif num_a == num_b:
2576	# different last names
2577	if a2[len_a2-1] != b2[len_b2-1] and num_a != 0:
2578	return -cmp(a2[len_a2-1], b2[len_b2-1])
2579	else:
2580	return cmp(a, b)
2581	else: return 1
2582	# one of them has a number
2583	elif num_a != None: return 1
2584	elif num_b != None: return -1
2585	# no numbers
2586	else: return cmp(a.lower(), b.lower())
2587
2588
2589	keys.sort(custom_compare)
2590
2591	iy = 0
2592	ix = 0
2593	select_text = "Select All"
2594	self.cb1 = wx.CheckBox(self, -1,str(select_text), (10, 10))
2595	wx.EVT_CHECKBOX(self, self.cb1.GetId(), self.select_all_param)
2596	self.cb1.SetToolTipString("To check/uncheck all the boxes below.")
2597	self.cb1.SetValue(True)
2598
2599	sizer.Add(self.cb1,(iy, ix),(1,1),\
2600	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 5)
2601	ix +=1
2602	self.text2_2 = wx.StaticText(self, -1, 'Value')
2603	sizer.Add(self.text2_2,(iy, ix),(1,1),\
2604	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
2605	ix +=2
2606	self.text2_3 = wx.StaticText(self, -1, 'Error')
2607	sizer.Add(self.text2_3,(iy, ix),(1,1),\
2608	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
2609	if not self.is_mac:
2610	self.text2_3.Hide()
2611	ix +=1
2612	self.text2_min = wx.StaticText(self, -1, 'Min')
2613	sizer.Add(self.text2_min,(iy, ix),(1,1),\
2614	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
2615	#self.text2_min.Hide()
2616	ix +=1
2617	self.text2_max = wx.StaticText(self, -1, 'Max')
2618	sizer.Add(self.text2_max,(iy, ix),(1,1),\
2619	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
2620	#self.text2_max.Hide()
2621	ix += 1
2622	self.text2_4 = wx.StaticText(self, -1, '[Units]')
2623	sizer.Add(self.text2_4,(iy, ix),(1,1),\
2624	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
2625	self.text2_4.Hide()
2626
2627	for item in keys:
2628	if not item in self.disp_list and not item in \
2629	self.model.orientation_params:
2630
2631	##prepare a spot to store errors
2632	if not self.model.details.has_key(item):
2633	self.model.details [item] = ["",None,None]
2634
2635	iy += 1
2636	ix = 0
2637	if (self.model.__class__ in \
2638	self.model_list_box["Multi-Functions"] or \
2639	self.temp_multi_functional)\
2640	and (item in self.model.non_fittable):
2641	non_fittable_name = wx.StaticText(self, -1, item )
2642	sizer.Add(non_fittable_name,(iy, ix),(1,1),\
2643	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 21)
2644	## add parameter value
2645	ix += 1
2646	value= self.model.getParam(item)
2647	if len(self.model.fun_list) > 0:
2648	num = item.split('_')[1][5:7]
2649	fun_box = wx.ComboBox(self, -1,size=(100,-1),
2650	style=wx.CB_READONLY, name = '%s'% item)
2651	self._set_fun_box_list(fun_box)
2652	fun_box.SetSelection(0)
2653	#self.fun_box.SetToolTipString("A function
2654	# describing the interface")
2655	wx.EVT_COMBOBOX(fun_box,-1, self._on_fun_box)
2656	else:
2657	fun_box = self.ModelTextCtrl(self, -1,
2658	size=(_BOX_WIDTH,20),
2659	style=wx.TE_PROCESS_ENTER, name ='%s'% item)
2660	fun_box.SetToolTipString("Hit 'Enter' after typing.")
2661	fun_box.SetValue(format_number(value, True))
2662	sizer.Add(fun_box, (iy,ix),(1,1), wx.EXPAND)
2663	self.str_parameters.append([None,item, fun_box,
2664	None, None, None,
2665	None, None])
2666
2667
2668	else:
2669	## add parameters name with checkbox for selecting to fit
2670	cb = wx.CheckBox(self, -1, item )
2671	cb.SetToolTipString(" Check mark to fit.")
2672	#cb.SetValue(True)
2673	wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
2674
2675	sizer.Add( cb,( iy, ix),(1,1),
2676	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 5)
2677
2678	## add parameter value
2679	ix += 1
2680	value= self.model.getParam(item)
2681	ctl1 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH,20),
2682	style=wx.TE_PROCESS_ENTER)
2683	ctl1.SetToolTipString("Hit 'Enter' after typing.")
2684	ctl1.SetValue(format_number(value, True))
2685	sizer.Add(ctl1, (iy,ix),(1,1), wx.EXPAND)
2686	## text to show error sign
2687	ix += 1
2688	text2=wx.StaticText(self, -1, '+/-')
2689	sizer.Add(text2,(iy, ix),(1,1),\
2690	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
2691	if not self.is_mac:
2692	text2.Hide()
2693	ix += 1
2694	ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH/1.2,20), style=0)
2695	sizer.Add(ctl2, (iy,ix),(1,1),
2696	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
2697	if not self.is_mac:
2698	ctl2.Hide()
2699
2700	ix += 1
2701	ctl3 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/1.9,20),
2702	style=wx.TE_PROCESS_ENTER,
2703	text_enter_callback = self._onparamRangeEnter)
2704
2705	sizer.Add(ctl3, (iy,ix),(1,1),
2706	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
2707
2708	ix += 1
2709	ctl4 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/1.9,20),
2710	style=wx.TE_PROCESS_ENTER,
2711	text_enter_callback = self._onparamRangeEnter)
2712	sizer.Add(ctl4, (iy,ix),(1,1),
2713	wx.EXPAND\|wx.FIXED_MINSIZE, 0)
2714
2715	ix +=1
2716	# Units
2717	if self.model.details.has_key(item):
2718	units = wx.StaticText(self, -1,
2719	self.model.details[item][0], style=wx.ALIGN_LEFT)
2720	else:
2721	units = wx.StaticText(self, -1, "", style=wx.ALIGN_LEFT)
2722	sizer.Add(units, (iy,ix),(1,1),
2723	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
2724
2725	##[cb state, name, value, "+/-", error of fit, min, max , units]
2726	self.parameters.append([cb,item, ctl1,
2727	text2,ctl2, ctl3, ctl4,units])
2728
2729	iy+=1
2730	sizer.Add((10,10),(iy,ix),(1,1),
2731	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
2732
2733	# type can be either Guassian or Array
2734	if len(self.model.dispersion.values())>0:
2735	type= self.model.dispersion.values()[0]["type"]
2736	else:
2737	type = "Gaussian"
2738
2739	iy += 1
2740	ix = 0
2741	#Add tile for orientational angle
2742	for item in keys:
2743	if item in self.model.orientation_params:
2744	orient_angle = wx.StaticText(self, -1, '[For 2D only]:')
2745	sizer.Add(orient_angle,(iy, ix),(1,1),
2746	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
2747	if not self.data.__class__.__name__ == "Data2D" and \
2748	not self.enable2D:
2749	orient_angle.Hide()
2750	else:
2751	orient_angle.Show(True)
2752	break
2753
2754	#For Gaussian only
2755	if type.lower() != "array":
2756	for item in self.model.orientation_params:
2757	if not item in self.disp_list:
2758	##prepare a spot to store min max
2759	if not self.model.details.has_key(item):
2760	self.model.details [item] = ["",None,None]
2761
2762	iy += 1
2763	ix = 0
2764	## add parameters name with checkbox for selecting to fit
2765	cb = wx.CheckBox(self, -1, item )
2766	cb.SetValue(False)
2767	cb.SetToolTipString("Check mark to fit")
2768	wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
2769	if self.data.__class__.__name__ == "Data2D" or \
2770	self.enable2D:
2771	cb.Show(True)
2772	else:
2773	cb.Hide()
2774	sizer.Add( cb,( iy, ix),(1,1),
2775	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 5)
2776
2777	## add parameter value
2778	ix += 1
2779	value= self.model.getParam(item)
2780	ctl1 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH,20),
2781	style=wx.TE_PROCESS_ENTER)
2782	ctl1.SetToolTipString("Hit 'Enter' after typing.")
2783	ctl1.SetValue(format_number(value, True))
2784	if self.data.__class__.__name__ == "Data2D" or \
2785	self.enable2D:
2786	ctl1.Show(True)
2787	else:
2788	ctl1.Hide()
2789	sizer.Add(ctl1, (iy,ix),(1,1), wx.EXPAND)
2790	## text to show error sign
2791	ix += 1
2792	text2=wx.StaticText(self, -1, '+/-')
2793	sizer.Add(text2,(iy, ix),(1,1),\
2794	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
2795
2796	text2.Hide()
2797	ix += 1
2798	ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH/1.2,20), style=0)
2799	sizer.Add(ctl2, (iy,ix),(1,1),
2800	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
2801
2802	ctl2.Hide()
2803
2804
2805	ix += 1
2806	ctl3 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/1.8,20),
2807	style=wx.TE_PROCESS_ENTER,
2808	text_enter_callback = self._onparamRangeEnter)
2809
2810	sizer.Add(ctl3, (iy,ix),(1,1),
2811	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
2812	ctl3.Hide()
2813
2814	ix += 1
2815	ctl4 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/1.8,20),
2816	style=wx.TE_PROCESS_ENTER,
2817	text_enter_callback = self._onparamRangeEnter)
2818	sizer.Add(ctl4, (iy,ix),(1,1),
2819	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
2820
2821	ctl4.Hide()
2822
2823	if self.data.__class__.__name__ == "Data2D" or \
2824	self.enable2D:
2825	if self.is_mac:
2826	text2.Show(True)
2827	ctl2.Show(True)
2828	ctl3.Show(True)
2829	ctl4.Show(True)
2830
2831	ix +=1
2832	# Units
2833	if self.model.details.has_key(item):
2834	units = wx.StaticText(self, -1,
2835	self.model.details[item][0],
2836	style=wx.ALIGN_LEFT)
2837	else:
2838	units = wx.StaticText(self, -1, "", style=wx.ALIGN_LEFT)
2839	if self.data.__class__.__name__ == "Data2D" or \
2840	self.enable2D:
2841	units.Show(True)
2842
2843	else:
2844	units.Hide()
2845
2846	sizer.Add(units, (iy,ix),(1,1),
2847	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
2848
2849	##[cb state, name, value, "+/-", error of fit,min,max,units]
2850	self.parameters.append([cb,item, ctl1,
2851	text2,ctl2, ctl3, ctl4,units])
2852	self.orientation_params.append([cb,item, ctl1,
2853	text2,ctl2, ctl3, ctl4,units])
2854
2855	iy+=1
2856
2857	#Display units text on panel
2858	for item in keys:
2859	if self.model.details.has_key(item):
2860	self.text2_4.Show()
2861	#Fill the list of fittable parameters
2862	self.select_all_param(event=None)
2863
2864	self.save_current_state_fit()
2865	boxsizer1.Add(sizer)
2866	self.sizer3.Add(boxsizer1,0, wx.EXPAND \| wx.ALL, 10)
2867	self.sizer3.Layout()
2868	self.Layout()
2869	#self.Refresh()
2870
2871	def on_right_down(self, event):
2872	"""
2873	Get key stroke event
2874	"""
2875	if self.data == None:
2876	return
2877	# Figuring out key combo: Cmd for copy, Alt for paste
2878	if event.AltDown() and event.ShiftDown():
2879	self._manager.show_ftol_dialog()
2880	flag = True
2881	elif event.AltDown() or event.ShiftDown():
2882	flag = False
2883	else:
2884	return
2885	# make event free
2886	event.Skip()
2887	# messages depending on the flag
2888	if not flag:
2889	msg = " Could not open ftol dialog;"
2890	msg += " Check if the Scipy fit engine is selected in the menubar."
2891	infor = 'warning'
2892	# inform msg to wx
2893	wx.PostEvent( self.parent.parent,
2894	StatusEvent(status= msg, info=infor))
2895
2896
2897	def _onModel2D(self, event):
2898	"""
2899	toggle view of model from 1D to 2D or 2D from 1D
2900	"""
2901	if self.model_view.GetLabelText() == "1D Mode":
2902	self.model_view.SetLabel("2D Mode")
2903	self.enable2D = True
2904
2905	else:
2906	self.model_view.SetLabel("1D Mode")
2907	self.enable2D = False
2908
2909	self.set_model_param_sizer(self.model)
2910	self._set_sizer_dispersion()
2911	self._draw_model()
2912
2913
2914	self.state.enable2D = copy.deepcopy(self.enable2D)
2915
2916	class BGTextCtrl(wx.TextCtrl):
2917	"""
2918	Text control used to display outputs.
2919	No editing allowed. The background is
2920	grayed out. User can't select text.
2921	"""
2922	def __init__(self, args, *kwds):
2923	wx.TextCtrl.__init__(self, args, *kwds)
2924	self.SetEditable(False)
2925	self.SetBackgroundColour(self.GetParent().parent.GetBackgroundColour())
2926
2927	# Bind to mouse event to avoid text highlighting
2928	# The event will be skipped once the call-back
2929	# is called.
2930	self.Bind(wx.EVT_MOUSE_EVENTS, self._click)
2931
2932	def _click(self, event):
2933	"""
2934	Prevent further handling of the mouse event
2935	by not calling Skip().
2936	"""
2937	pass
2938

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

SasView

source: sasview/sansview/perspectives/fitting/fitpage.py @ 70468a5

Download in other formats: