fitpage.py @ 4b5fe655

ESS_GUIESS_GUI_DocsESS_GUI_batch_fittingESS_GUI_bumps_abstractionESS_GUI_iss1116ESS_GUI_iss879ESS_GUI_iss959ESS_GUI_openclESS_GUI_orderingESS_GUI_sync_sascalccostrafo411magnetic_scattrelease-4.1.1release-4.1.2release-4.2.2release_4.0.1ticket-1009ticket-1094-headlessticket-1242-2d-resolutionticket-1243ticket-1249ticket885unittest-saveload

Last change on this file since 4b5fe655 was 5265420, checked in by Paul Kienzle <pkienzle@…>, 10 years ago
Fix the resolution reset problem by removing the Fit button disable step. For reasons unknown, self.btFit.Enable(False) somehow triggers self.disable_smeared even though the buttons are of different type and have different ids. This is a probably a bug in wx. Fixes #405.
Property mode set to `100644`
File size: 126.7 KB

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

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

SasView

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

Download in other formats: