fitpage.py @ a4340d1

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 a4340d1 was 9e11cf5, checked in by gonzalezm, 10 years ago
Bug fix for ticket 320 - NaN errors are now displayed in the error boxes instead of making the box disappear
Property mode set to `100644`
File size: 134.7 KB

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

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

SasView

source: sasview/src/sas/perspectives/fitting/fitpage.py @ a4340d1

Download in other formats: