fitpage.py @ cbbb6a4

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.2ticket-1009ticket-1094-headlessticket-1242-2d-resolutionticket-1243ticket-1249ticket885unittest-saveload

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

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

SasView

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

Download in other formats: