fitpage.py @ d734a3b1

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 d734a3b1 was cb190a19, checked in by Jae Cho <jhjcho@…>, 14 years ago
cleaned up chisqr left-over
Property mode set to `100644`
File size: 99.4 KB

Rev	Line
[d1e0473]	1
	2
[c77d859]	3	import sys
	4	import wx
	5	import wx.lib.newevent
	6	import numpy
	7	import copy
	8	import math
[7975f2b]	9	import time
[c77d859]	10	from sans.models.dispersion_models import ArrayDispersion, GaussianDispersion
[7609f1a]	11	from DataLoader.data_info import Data1D
[c77d859]	12	from sans.guicomm.events import StatusEvent
[85b3971]	13	from sans.guiframe.utils import format_number,check_float
[51a71a3]	14	#from sans.guiframe.local_perspectives.plotting.Masking import ModelPanel2D as panel2D
[7609f1a]	15
[b787e68c]	16	## event to post model to fit to fitting plugins
[c77d859]	17	(ModelEventbox, EVT_MODEL_BOX) = wx.lib.newevent.NewEvent()
[b787e68c]	18
	19	## event to know the selected fit engine
[77e23a2]	20	(FitterTypeEvent, EVT_FITTER_TYPE) = wx.lib.newevent.NewEvent()
[34a0c17]	21	(FitStopEvent, EVT_FIT_STOP) = wx.lib.newevent.NewEvent()
[f72333f]	22	(Chi2UpdateEvent, EVT_CHI2_UPDATE) = wx.lib.newevent.NewEvent()
[eda428b]	23	_BOX_WIDTH = 76
[9237df4]	24	_DATA_BOX_WIDTH = 300
[7609f1a]	25	SMEAR_SIZE_L = 0.005
[4fbc93e]	26	SMEAR_SIZE_H = 0.006
[7609f1a]	27
[c77d859]	28	import basepage
	29	from basepage import BasicPage
[b787e68c]	30	from basepage import PageInfoEvent
[997131a]	31	from DataLoader.qsmearing import smear_selection
[c77d859]	32
	33	class FitPage(BasicPage):
	34	"""
	35	FitPanel class contains fields allowing to display results when
	36	fitting a model and one data
	37	@note: For Fit to be performed the user should check at least one parameter
	38	on fit Panel window.
	39
	40	"""
[7975f2b]	41
[cfc0913]	42	def __init__(self,parent, page_info):
	43	BasicPage.__init__(self, parent, page_info)
[7975f2b]	44
[c77d859]	45	"""
	46	Initialization of the Panel
	47	"""
[51a71a3]	48	## total number of point: float
[c77d859]	49	self.npts=None
[2b63df0]	50	## thread for compute Chisqr
	51	self.calc_Chisqr=None
[dcf29d7]	52	## default fitengine type
[707436d]	53	self.engine_type = None
[7609f1a]	54	## smear default
	55	self.smearer = None
	56	self.current_smearer = None
[f72333f]	57	## 2D smear accuracy default
	58	self.smear2d_accuracy = 'Low'
	59	## slit smear:
[7609f1a]	60	self.dxl = None
	61	self.dxw = None
[f72333f]	62	## pinhole smear
[7609f1a]	63	self.dx_min = None
	64	self.dx_max = None
[edd166b]	65
[f72333f]	66	## model data (theory) calculated on drawing.
	67	self.theory_data = None
	68
[b787e68c]	69	## draw sizer
[c77d859]	70	self._fill_datainfo_sizer()
[7609f1a]	71
	72	# get smear info from data
	73	self._get_smear_info()
	74
[c77d859]	75	self._fill_model_sizer( self.sizer1)
[7609f1a]	76
	77	self._get_defult_custom_smear()
	78
[cfc0913]	79	self._fill_range_sizer()
[7609f1a]	80
[ffa69b6]	81	if self.data is None:
	82	self.formfactorbox.Disable()
	83	self.structurebox.Disable()
	84	else:
[7609f1a]	85	self.smearer = smear_selection( self.data )
[847091f]	86	if self.smearer ==None:
	87	self.enable_smearer.Disable()
[f72333f]	88	#if self.data.__class__.__name__ =="Data2D":
	89	#if self.model != None:
	90	#self.smear_description_2d.Show(True)
[2012eae]	91
[edd166b]	92	self.disp_cb_dict = {}
[cfc0913]	93	## to update the panel according to the fit engine type selected
[77e23a2]	94	self.Bind(EVT_FITTER_TYPE,self._on_engine_change)
[34a0c17]	95	self.Bind(EVT_FIT_STOP,self._on_fit_complete)
[f72333f]	96	self.Bind(EVT_CHI2_UPDATE, self.on_complete_chisqr)
[c69b6d5]	97
[34a0c17]	98	def _on_fit_complete(self, event):
	99	"""
	100	When fit is complete ,reset the fit button label.
	101	"""
	102	#self.btFit.SetLabel("Fit")
	103	#self.btFit.Unbind(event=wx.EVT_BUTTON, id=self.btFit.GetId())
	104	#self.btFit.Bind(event=wx.EVT_BUTTON, handler=self._onFit,id=self.btFit.GetId())
	105	pass
	106
[f72333f]	107	def _is_2D(self):
	108	"""
	109	Check if data_name is Data2D
	110	@return: True or False
	111	"""
[34a0c17]	112
[f72333f]	113	if self.data.__class__.__name__ =="Data2D":
	114	return True
	115	return False
	116
[77e23a2]	117	def _on_engine_change(self, event):
	118	"""
	119	get an event containing the current name of the fit engine type
	120	@param event: FitterTypeEvent containing the name of the current engine
	121	"""
[dcf29d7]	122	self.engine_type = event.type
[77e23a2]	123	if len(self.parameters)==0:
[edd166b]	124	self.Layout()
[77e23a2]	125	return
[ad6dd4c]	126	if event.type =="park":
	127	self.btFit.SetLabel("Fit")
[6d91073]	128
[77e23a2]	129	for item in self.parameters:
[920a6e5]	130	if event.type =="scipy" :
[60132ef]	131	item[5].SetValue("")
[77e23a2]	132	item[5].Hide()
[60132ef]	133	item[6].SetValue("")
[77e23a2]	134	item[6].Hide()
	135	self.text2_min.Hide()
	136	self.text2_max.Hide()
[7975f2b]	137
[77e23a2]	138	else:
	139	item[5].Show(True)
	140	item[6].Show(True)
	141	self.text2_min.Show(True)
	142	self.text2_max.Show(True)
[920a6e5]	143	for item in self.fittable_param:
	144	if item[5]!=None and item[6]!=None and not item in self.orientation_params_disp:
	145	if event.type =="scipy" and not item in self.orientation_params:
[c5cd3b9]	146	item[5].SetValue("")
	147	item[5].Hide()
	148	item[6].SetValue("")
	149	item[6].Hide()
	150	self.text2_min.Hide()
	151	self.text2_max.Hide()
[920a6e5]	152	self.text_disp_min.Hide()
	153	self.text_disp_max.Hide()
[c5cd3b9]	154	else:
	155	item[5].Show(True)
	156	item[6].Show(True)
	157	self.text2_min.Show(True)
	158	self.text2_max.Show(True)
[920a6e5]	159	self.text_disp_min.Show(True)
	160	self.text_disp_max.Show(True)
[c5cd3b9]	161
[920a6e5]	162	for item in self.orientation_params:
[c5cd3b9]	163	if item[5]!=None and item[6]!=None:
	164	if event.type =="scipy" or self.data.__class__.__name__ !="Data2D":
	165	item[5].SetValue("")
	166	item[5].Hide()
	167	item[6].SetValue("")
	168	item[6].Hide()
	169	else:
	170	item[5].Show(True)
	171	item[6].Show(True)
[920a6e5]	172
	173	for item in self.orientation_params_disp:
	174	if item[5]!=None and item[6]!=None:
	175	if event.type =="scipy" or self.data.__class__.__name__ !="Data2D":
	176	item[5].SetValue("")
	177	item[5].Hide()
	178	item[6].SetValue("")
	179	item[6].Hide()
	180	else:
	181	item[5].Show(True)
	182	item[6].Show(True)
[6d91073]	183	self.Layout()
[7975f2b]	184	self.Refresh()
	185
[c77d859]	186	def _fill_range_sizer(self):
	187	"""
	188	Fill the sizer containing the plotting range
	189	add access to npts
	190	"""
[51a71a3]	191	is_2Ddata = False
	192
	193	# Check if data is 2D
[6318298]	194	if self.data.__class__.__name__ == 'Data2D':
[51a71a3]	195	is_2Ddata = True
	196
[ff8f99b]	197	title = "Fitting"
[7609f1a]	198	#smear messages & titles
	199	smear_message_none = "No smearing is selected..."
	200	smear_message_dqdata = "The dQ data is being used for smearing..."
[f72333f]	201	smear_message_2d = "Higher accuracy is very time-expensive. Use it with care..."
[7609f1a]	202	smear_message_new_ssmear = "Please enter only the value of interest to customize smearing..."
	203	smear_message_new_psmear = "Please enter both; the dQ will be generated by interpolation..."
[f72333f]	204	smear_message_2d_x_title = "<dQx>[1/A]:"
	205	smear_message_2d_y_title = "<dQy>[1/A]:"
	206	smear_message_pinhole_min_title = "dQ_low[1/A]:"
	207	smear_message_pinhole_max_title = "dQ_high[1/A]:"
[7609f1a]	208	smear_message_slit_height_title = "Slit height[1/A]:"
	209	smear_message_slit_width_title = "Slit width[1/A]:"
	210
	211	self._get_smear_info()
[c77d859]	212
[51a71a3]	213	#Sizers
	214	box_description_range = wx.StaticBox(self, -1,str(title))
	215	boxsizer_range = wx.StaticBoxSizer(box_description_range, wx.VERTICAL)
[7609f1a]	216	self.sizer_set_smearer = wx.BoxSizer(wx.VERTICAL)
[c77d859]	217	sizer_smearer = wx.BoxSizer(wx.HORIZONTAL)
[7609f1a]	218	self.sizer_new_smear= wx.BoxSizer(wx.HORIZONTAL)
	219	self.sizer_set_masking = wx.BoxSizer(wx.HORIZONTAL)
	220	sizer_chi2 = wx.BoxSizer(wx.VERTICAL)
	221	smear_set_box= wx.StaticBox(self, -1,'Set Instrumental Smearing')
	222	sizer_smearer_box = wx.StaticBoxSizer(smear_set_box, wx.HORIZONTAL)
	223	sizer_smearer_box.SetMinSize((_DATA_BOX_WIDTH,85))
[2012eae]	224	sizer_fit = wx.GridSizer(2, 4,2,6)
	225
[f72333f]	226	# combobox for smear2d accuracy selection
	227	self.smear_accuracy = wx.ComboBox(self, -1,size=(50,-1),style=wx.CB_READONLY)
	228	self._set_accuracy_list()
	229	self.smear_accuracy.SetValue(self.smear2d_accuracy)
	230	self.smear_accuracy.SetSelection(0)
	231	self.smear_accuracy.SetToolTipString("'Higher' uses more Gaussian points for smearing computation.")
	232
	233
	234
	235	wx.EVT_COMBOBOX(self.smear_accuracy,-1, self._on_select_accuracy)
	236
[2012eae]	237	#Fit button
	238	self.btFit = wx.Button(self,wx.NewId(),'Fit', size=(88,25))
	239	self.btFit.Bind(wx.EVT_BUTTON, self._onFit,id= self.btFit.GetId())
	240	self.btFit.SetToolTipString("Start fitting.")
[7609f1a]	241
[f72333f]	242	#textcntrl for custom resolution
	243	self.smear_pinhole_max = self.ModelTextCtrl(self, -1,size=(_BOX_WIDTH-25,20),style=wx.TE_PROCESS_ENTER,
[7609f1a]	244	text_enter_callback = self.onPinholeSmear)
[f72333f]	245	self.smear_pinhole_min = self.ModelTextCtrl(self, -1,size=(_BOX_WIDTH-25,20),style=wx.TE_PROCESS_ENTER,
[7609f1a]	246	text_enter_callback = self.onPinholeSmear)
[f72333f]	247	self.smear_slit_height= self.ModelTextCtrl(self, -1,size=(_BOX_WIDTH-25,20),style=wx.TE_PROCESS_ENTER,
[7609f1a]	248	text_enter_callback = self.onSlitSmear)
[f72333f]	249	self.smear_slit_width = self.ModelTextCtrl(self, -1,size=(_BOX_WIDTH-25,20),style=wx.TE_PROCESS_ENTER,
[7609f1a]	250	text_enter_callback = self.onSlitSmear)
[f72333f]	251
	252	## smear
	253	self.smear_data_left= BGTextCtrl(self, -1, size=(_BOX_WIDTH-25,20), style=0)
[7609f1a]	254	self.smear_data_left.SetValue(str(self.dq_l))
[f72333f]	255	self.smear_data_right = BGTextCtrl(self, -1, size=(_BOX_WIDTH-25,20), style=0)
[7609f1a]	256	self.smear_data_right.SetValue(str(self.dq_r))
	257
[f72333f]	258	#set default values for smear
[7609f1a]	259	self.smear_pinhole_max.SetValue(str(self.dx_max))
	260	self.smear_pinhole_min.SetValue(str(self.dx_min))
	261	self.smear_slit_height.SetValue(str(self.dxl))
	262	self.smear_slit_width.SetValue(str(self.dxw))
	263
[c77d859]	264	#Filling the sizer containing instruments smearing info.
[7609f1a]	265	self.disable_smearer = wx.RadioButton(self, -1, 'None', style=wx.RB_GROUP)
	266	self.enable_smearer = wx.RadioButton(self, -1, 'Use dQ Data')
	267	#self.enable_smearer.SetToolTipString("Click to use the loaded dQ data for smearing.")
	268	self.pinhole_smearer = wx.RadioButton(self, -1, 'Custom Pinhole Smear')
	269	#self.pinhole_smearer.SetToolTipString("Click to input custom resolution for pinhole smearing.")
	270	self.slit_smearer = wx.RadioButton(self, -1, 'Custom Slit Smear')
	271	#self.slit_smearer.SetToolTipString("Click to input custom resolution for slit smearing.")
[c77d859]	272	self.Bind(wx.EVT_RADIOBUTTON, self.onSmear, id=self.disable_smearer.GetId())
	273	self.Bind(wx.EVT_RADIOBUTTON, self.onSmear, id=self.enable_smearer.GetId())
[7609f1a]	274	self.Bind(wx.EVT_RADIOBUTTON, self.onPinholeSmear, id=self.pinhole_smearer.GetId())
	275	self.Bind(wx.EVT_RADIOBUTTON, self.onSlitSmear, id=self.slit_smearer.GetId())
[ff8f99b]	276	self.disable_smearer.SetValue(True)
[c77d859]	277
[7609f1a]	278	# add 4 types of smearing to the sizer
	279	sizer_smearer.Add( self.disable_smearer,0, wx.LEFT, 10)
[c77d859]	280	sizer_smearer.Add((10,10))
[7609f1a]	281	sizer_smearer.Add( self.enable_smearer)
	282	sizer_smearer.Add((10,10))
	283	sizer_smearer.Add( self.pinhole_smearer )
	284	sizer_smearer.Add((10,10))
	285	sizer_smearer.Add( self.slit_smearer )
	286	sizer_smearer.Add((10,10))
	287
[51a71a3]	288	# StaticText for chi2, N(for fitting), Npts
	289	self.tcChi = BGTextCtrl(self, -1, "-", size=(75,20), style=0)
	290	self.tcChi.SetToolTipString("Chi2/Npts")
	291	self.Npts_fit = BGTextCtrl(self, -1, "-", size=(75,20), style=0)
	292	self.Npts_fit.SetToolTipString(" Npts : number of points selected for fitting")
	293	self.Npts_total = BGTextCtrl(self, -1, "-", size=(75,20), style=0)
	294	self.Npts_total.SetToolTipString(" Total Npts : total number of data points")
	295	box_description_1= wx.StaticText(self, -1,' Chi2/Npts')
	296	box_description_2= wx.StaticText(self, -1,' Npts')
	297	box_description_3= wx.StaticText(self, -1,' Total Npts')
	298	box_description_4= wx.StaticText(self, -1,' ')
[7609f1a]	299
[2012eae]	300
[51a71a3]	301	sizer_fit.Add(box_description_1,0,0)
	302	sizer_fit.Add(box_description_2,0,0)
	303	sizer_fit.Add(box_description_3,0,0)
	304	sizer_fit.Add(box_description_4,0,0)
	305	sizer_fit.Add(self.tcChi,0,0)
	306	sizer_fit.Add(self.Npts_fit ,0,0)
	307	sizer_fit.Add(self.Npts_total,0,0)
	308	sizer_fit.Add(self.btFit,0,0)
	309
	310	# StaticText for smear
	311	#self.tcChi = wx.StaticText(self, -1, "-", style=wx.ALIGN_LEFT)
[7609f1a]	312	self.smear_description_none = wx.StaticText(self, -1, smear_message_none , style=wx.ALIGN_LEFT)
	313	self.smear_description_dqdata = wx.StaticText(self, -1, smear_message_dqdata , style=wx.ALIGN_LEFT)
	314	self.smear_description_type = wx.StaticText(self, -1, "Type:" , style=wx.ALIGN_LEFT)
[f72333f]	315	self.smear_description_accuracy_type = wx.StaticText(self, -1, "Accuracy:" , style=wx.ALIGN_LEFT)
	316	self.smear_description_smear_type = BGTextCtrl(self, -1, size=(57,20), style=0)
[7609f1a]	317	self.smear_description_smear_type.SetValue(str(self.dq_l))
	318	self.SetBackgroundColour(self.GetParent().GetBackgroundColour())
	319	self.smear_description_2d = wx.StaticText(self, -1, smear_message_2d , style=wx.ALIGN_LEFT)
	320	self.smear_message_new_s = wx.StaticText(self, -1, smear_message_new_ssmear , style=wx.ALIGN_LEFT)
	321	self.smear_message_new_p = wx.StaticText(self, -1, smear_message_new_psmear , style=wx.ALIGN_LEFT)
[f72333f]	322	self.smear_description_2d_x = wx.StaticText(self, -1, smear_message_2d_x_title , style=wx.ALIGN_LEFT)
	323	self.smear_description_2d_y = wx.StaticText(self, -1, smear_message_2d_y_title , style=wx.ALIGN_LEFT)
[7609f1a]	324	self.smear_description_pin_min = wx.StaticText(self, -1, smear_message_pinhole_min_title , style=wx.ALIGN_LEFT)
	325	self.smear_description_pin_max = wx.StaticText(self, -1, smear_message_pinhole_max_title , style=wx.ALIGN_LEFT)
	326	self.smear_description_slit_height = wx.StaticText(self, -1, smear_message_slit_height_title , style=wx.ALIGN_LEFT)
	327	self.smear_description_slit_width = wx.StaticText(self, -1, smear_message_slit_width_title , style=wx.ALIGN_LEFT)
	328
	329	#arrange sizers
[51a71a3]	330	#boxsizer1.Add( self.tcChi )
[7609f1a]	331	self.sizer_set_smearer.Add(sizer_smearer )
	332	self.sizer_set_smearer.Add((10,10))
	333	self.sizer_set_smearer.Add( self.smear_description_none,0, wx.CENTER, 10 )
	334	self.sizer_set_smearer.Add( self.smear_description_dqdata,0, wx.CENTER, 10 )
	335	self.sizer_set_smearer.Add( self.smear_description_2d,0, wx.CENTER, 10 )
	336	self.sizer_new_smear.Add( self.smear_description_type,0, wx.CENTER, 10 )
[f72333f]	337	self.sizer_new_smear.Add( self.smear_description_accuracy_type,0, wx.CENTER, 10 )
	338	self.sizer_new_smear.Add( self.smear_accuracy )
[7609f1a]	339	self.sizer_new_smear.Add( self.smear_description_smear_type,0, wx.CENTER, 10 )
	340	self.sizer_new_smear.Add((15,-1))
[f72333f]	341	self.sizer_new_smear.Add( self.smear_description_2d_x,0, wx.CENTER, 10 )
[7609f1a]	342	self.sizer_new_smear.Add( self.smear_description_pin_min,0, wx.CENTER, 10 )
	343	self.sizer_new_smear.Add( self.smear_description_slit_height,0, wx.CENTER, 10 )
[f72333f]	344
[7609f1a]	345	self.sizer_new_smear.Add( self.smear_pinhole_min,0, wx.CENTER, 10 )
	346	self.sizer_new_smear.Add( self.smear_slit_height,0, wx.CENTER, 10 )
	347	self.sizer_new_smear.Add( self.smear_data_left,0, wx.CENTER, 10 )
	348	self.sizer_new_smear.Add((20,-1))
[f72333f]	349	self.sizer_new_smear.Add( self.smear_description_2d_y,0, wx.CENTER, 10 )
[7609f1a]	350	self.sizer_new_smear.Add( self.smear_description_pin_max,0, wx.CENTER, 10 )
	351	self.sizer_new_smear.Add( self.smear_description_slit_width,0, wx.CENTER, 10 )
[f72333f]	352
[7609f1a]	353	self.sizer_new_smear.Add( self.smear_pinhole_max,0, wx.CENTER, 10 )
	354	self.sizer_new_smear.Add( self.smear_slit_width,0, wx.CENTER, 10 )
	355	self.sizer_new_smear.Add( self.smear_data_right,0, wx.CENTER, 10 )
	356
	357	self.sizer_set_smearer.Add( self.smear_message_new_s,0, wx.CENTER, 10)
	358	self.sizer_set_smearer.Add( self.smear_message_new_p,0, wx.CENTER, 10)
	359	self.sizer_set_smearer.Add((5,2))
	360	self.sizer_set_smearer.Add( self.sizer_new_smear,0, wx.CENTER, 10 )
	361
[51a71a3]	362	# add all to chi2 sizer
	363	sizer_smearer_box.Add(self.sizer_set_smearer)
[7609f1a]	364	sizer_chi2.Add(sizer_smearer_box)
	365	sizer_chi2.Add((-1,5))
[51a71a3]	366
[7609f1a]	367	# hide all smear messages and textctrl
	368	self._hide_all_smear_info()
	369
	370	# get smear_selection
	371	self.current_smearer= smear_selection( self.data )
[83ad478]	372
[7609f1a]	373	# Show only the relevant smear messages, etc
	374	if self.current_smearer == None:
[51a71a3]	375	if not is_2Ddata:
[7609f1a]	376	self.smear_description_none.Show(True)
	377	self.enable_smearer.Disable()
	378	else:
[f72333f]	379	self.smear_description_none.Show(True)
	380	#self.smear_description_2d.Show(True)
	381	#self.pinhole_smearer.Disable()
[7609f1a]	382	self.slit_smearer.Disable()
[f72333f]	383	#self.enable_smearer.Disable()
[7609f1a]	384	else: self._show_smear_sizer()
[51a71a3]	385	boxsizer_range.Add(self.sizer_set_masking)
[7609f1a]	386
	387	#Set sizer for Fitting section
	388	self._set_range_sizer( title=title,box_sizer=boxsizer_range, object1=sizer_chi2, object= sizer_fit)
	389
[c77d859]	390	def _fill_datainfo_sizer(self):
	391	"""
	392	fill sizer 0 with data info
	393	"""
	394	## no loaded data , don't fill the sizer
[9237df4]	395	if self.data is None:
	396	data_min = ""
	397	data_max = ""
	398	data_name = ""
	399	else:
	400	data_name = self.data.name
	401	#set maximum range for x in linear scale
	402	if not hasattr(self.data,"data"): #Display only for 1D data fit
	403	# Minimum value of data
	404	data_min = min(self.data.x)
	405	# Maximum value of data
	406	data_max = max(self.data.x)
	407	else:
	408	## Minimum value of data
	409	data_min = 0
	410	x = max(math.fabs(self.data.xmin), math.fabs(self.data.xmax))
	411	y = max(math.fabs(self.data.ymin), math.fabs(self.data.ymax))
	412	## Maximum value of data
	413	data_max = math.sqrt(xx + yy)
[c77d859]	414
[fa58441]	415	box_description= wx.StaticBox(self, -1, 'Data')
[c77d859]	416	boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
	417	#----------------------------------------------------------
[9237df4]	418	sizer_data = wx.BoxSizer(wx.HORIZONTAL)
[c77d859]	419	#Filling the sizer containing data related fields
[9237df4]	420	#self.dataSource = wx.StaticText(self, -1,str(self.data.name))
[7609f1a]	421	self.dataSource = BGTextCtrl(self, -1)
[9237df4]	422	self.dataSource.SetValue(str(data_name))
[7609f1a]	423	#self.dataSource.SetEditable(False)
[9237df4]	424	self.dataSource.SetMinSize((_DATA_BOX_WIDTH, -1))
[fa58441]	425	sizer_data.Add(wx.StaticText(self, -1, 'Source Name : '))
[9237df4]	426	sizer_data.Add(self.dataSource )
[2a5bba7]	427	sizer_data.Add( (0,5) )
[c77d859]	428
	429	#---------sizer 2 draw--------------------------------
[9237df4]	430	text4_3 = wx.StaticText(self, -1, 'Total Q Range (1/A)',
	431	style=wx.ALIGN_LEFT)
	432	sizer_range = wx.BoxSizer(wx.HORIZONTAL)
	433	sizer_range.Add( text4_3 ,0, wx.RIGHT, 10)
[7609f1a]	434	self.minimum_q = BGTextCtrl(self, -1, size=(_BOX_WIDTH,20))
[9237df4]	435	self.minimum_q.SetValue(str(data_min))
[7609f1a]	436	#self.minimum_q.SetEditable(False)
	437	self.maximum_q = BGTextCtrl(self, -1,size=(_BOX_WIDTH,20))
[9237df4]	438	self.maximum_q.SetValue(str(data_max))
[7609f1a]	439	#self.maximum_q.SetEditable(False)
[9237df4]	440	sizer_range.Add(wx.StaticText(self, -1, "Min: "),0, wx.LEFT, 10)
	441	sizer_range.Add(self.minimum_q,0, wx.LEFT, 10)
	442	sizer_range.Add(wx.StaticText(self, -1, "Max: "),0, wx.LEFT, 10)
	443	sizer_range.Add(self.maximum_q,0, wx.LEFT, 10)
	444
[813334e]	445	## set q range to plot
[9237df4]	446	self.qmin_x = data_min
	447	self.qmax_x = data_max
[813334e]	448
[9237df4]	449	boxsizer1.Add(sizer_data,0, wx.ALL, 10)
	450	boxsizer1.Add(sizer_range, 0 , wx.LEFT, 10)
[c77d859]	451	#------------------------------------------------------------
	452	self.sizer0.Add(boxsizer1,0, wx.EXPAND \| wx.ALL, 10)
	453	self.sizer0.Layout()
[813334e]	454
[c77d859]	455	def _fill_model_sizer(self, sizer):
	456	"""
	457	fill sizer containing model info
	458	"""
[7ad6ff5]	459	##Add model function Details button in fitpanel.
[ff8f99b]	460	##The following 3 lines are for Mac. Let JHC know before modifying...
	461	title = "Model"
	462	box_description= wx.StaticBox(self, -1,str(title))
	463	boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
	464
[7ad6ff5]	465	id = wx.NewId()
[e3a76c8]	466	self.model_help =wx.Button(self,id,'Details', size=(80,23))
[7ad6ff5]	467	self.model_help.Bind(wx.EVT_BUTTON, self.on_model_help_clicked,id=id)
	468	self.model_help.SetToolTipString("Model Function Help")
[7b0fd65]	469
[7ad6ff5]	470	## class base method to add view 2d button
[ff8f99b]	471	self._set_model_sizer(sizer=sizer, box_sizer=boxsizer1, title="Model",object=self.model_help )
[7ad6ff5]	472
[c77d859]	473
[30d103a]	474	def _set_sizer_dispersion(self, dispersity):
[c77d859]	475	"""
	476	draw sizer with gaussian dispersity parameters
	477	"""
	478	self.fittable_param=[]
	479	self.fixed_param=[]
[60132ef]	480	self.orientation_params_disp=[]
[920a6e5]	481
[c77d859]	482	self.sizer4_4.Clear(True)
	483	if self.model==None:
	484	##no model is selected
	485	return
	486	if not self.enable_disp.GetValue():
	487	## the user didn't select dispersity display
	488	return
[b421b1a]	489
[376916c]	490	self._reset_dispersity()
[b421b1a]	491
[376916c]	492	# Create the dispersion objects
	493	for item in self.model.dispersion.keys():
[30d103a]	494	#disp_model = GaussianDispersion()
	495	disp_model = dispersity()
[376916c]	496	self._disp_obj_dict[item] = disp_model
	497	self.model.set_dispersion(item, disp_model)
[c477b31]	498	self.state._disp_obj_dict[item]= disp_model
[376916c]	499
[b421b1a]	500
[c77d859]	501	ix=0
	502	iy=1
[52efcc5]	503	disp = wx.StaticText(self, -1, ' ')
[c77d859]	504	self.sizer4_4.Add(disp,( iy, ix),(1,1),
	505	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	506	ix += 1
[52efcc5]	507	values = wx.StaticText(self, -1, 'Sigma (STD)')
[1a94c36]	508	values.SetToolTipString("Polydispersity multiplied by the value of the original parameter.")
[c77d859]	509	self.sizer4_4.Add(values,( iy, ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	510	ix +=2
[920a6e5]	511	self.text_disp_1 = wx.StaticText(self, -1, '')
[c77d859]	512	self.sizer4_4.Add( self.text_disp_1,(iy, ix),(1,1),\
	513	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	514	self.text_disp_1.Hide()
[920a6e5]	515
	516
	517	ix +=1
	518	self.text_disp_min = wx.StaticText(self, -1, 'Min')
	519	self.sizer4_4.Add(self.text_disp_min,(iy, ix),(1,1),\
	520	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	521	self.text_disp_min.Hide()
	522	ix +=1
	523	self.text_disp_max = wx.StaticText(self, -1, 'Max')
	524	self.sizer4_4.Add(self.text_disp_max,(iy, ix),(1,1),\
	525	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	526	self.text_disp_max.Hide()
	527
	528
[c77d859]	529	ix += 1
	530	npts = wx.StaticText(self, -1, 'Npts')
[1a94c36]	531	npts.SetToolTipString("Number of points for weighting.")
[c77d859]	532	self.sizer4_4.Add(npts,( iy, ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	533	ix += 1
	534	nsigmas = wx.StaticText(self, -1, 'Nsigmas')
[1a94c36]	535	nsigmas.SetToolTipString("Number of sigmas between which the range of the distribution function will be used for weighting. The value '3' covers 99.5% for Gaussian distribution function.")
[c77d859]	536	self.sizer4_4.Add(nsigmas,( iy, ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[920a6e5]	537
	538	if self.engine_type=="park":
	539	self.text_disp_max.Show(True)
	540	self.text_disp_min.Show(True)
[6dc9ad8]	541
[c77d859]	542	for item in self.model.dispersion.keys():
[780d095]	543	if not item in self.model.orientation_params:
[edd166b]	544	if not self.disp_cb_dict.has_key(item):
	545	self.disp_cb_dict[item]= None
[780d095]	546	name1=item+".width"
	547	name2=item+".npts"
	548	name3=item+".nsigmas"
[c99a6c5]	549	if not self.model.details.has_key(name1):
[b421b1a]	550	self.model.details [name1] = ["",None,None]
[c99a6c5]	551
[780d095]	552	iy += 1
	553	for p in self.model.dispersion[item].keys():
	554
	555	if p=="width":
	556	ix = 0
	557	cb = wx.CheckBox(self, -1, name1, (10, 10))
[b9405cd]	558	cb.SetToolTipString("Check for fitting")
[780d095]	559	wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
	560	self.sizer4_4.Add( cb,( iy, ix),(1,1),
	561	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	562	ix = 1
	563	value= self.model.getParam(name1)
[7975f2b]	564	ctl1 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH,20),
[c77d859]	565	style=wx.TE_PROCESS_ENTER)
[1a94c36]	566	ctl1.SetToolTipString("Polydispersity multiplied by the value of '%s'."%item)
[780d095]	567	ctl1.SetValue(str (format_number(value)))
	568	self.sizer4_4.Add(ctl1, (iy,ix),(1,1),wx.EXPAND)
	569	## text to show error sign
	570	ix = 2
	571	text2=wx.StaticText(self, -1, '+/-')
	572	self.sizer4_4.Add(text2,(iy, ix),(1,1),
	573	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	574	text2.Hide()
[b421b1a]	575
[780d095]	576	ix = 3
[c13b8cc]	577	ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=0)
[eacf1d66]	578
[780d095]	579	self.sizer4_4.Add(ctl2, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[eacf1d66]	580
[b421b1a]	581	ctl2.Hide()
[eacf1d66]	582
[920a6e5]	583	ix = 4
[7975f2b]	584	ctl3 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
	585	text_enter_callback = self._onparamRangeEnter)
[b9405cd]	586
[920a6e5]	587	self.sizer4_4.Add(ctl3, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	588	ctl3.Hide()
	589
	590	ix = 5
[7975f2b]	591	ctl4 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
	592	text_enter_callback = self._onparamRangeEnter)
[b9405cd]	593
[920a6e5]	594	self.sizer4_4.Add(ctl4, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[eacf1d66]	595
[920a6e5]	596	ctl4.Hide()
	597
	598	if self.engine_type=="park":
	599	ctl3.Show(True)
	600	ctl4.Show(True)
	601
[780d095]	602	self.fittable_param.append([cb,name1,ctl1,text2,
[c99a6c5]	603	ctl2, ctl3, ctl4,None])
[920a6e5]	604
[780d095]	605	elif p=="npts":
[920a6e5]	606	ix = 6
[780d095]	607	value= self.model.getParam(name2)
[7975f2b]	608	Tctl = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20),
[780d095]	609	style=wx.TE_PROCESS_ENTER)
[1a94c36]	610
[780d095]	611	Tctl.SetValue(str (format_number(value)))
	612	self.sizer4_4.Add(Tctl, (iy,ix),(1,1),
	613	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	614	self.fixed_param.append([None,name2, Tctl,None,None,
	615	None, None,None])
	616	elif p=="nsigmas":
[920a6e5]	617	ix = 7
[780d095]	618	value= self.model.getParam(name3)
[7975f2b]	619	Tct2 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20),
[780d095]	620	style=wx.TE_PROCESS_ENTER)
[1a94c36]	621
[c985bef]	622	Tct2.SetValue(str (format_number(value)))
	623	self.sizer4_4.Add(Tct2, (iy,ix),(1,1),
[780d095]	624	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	625	ix +=1
	626	self.sizer4_4.Add((20,20), (iy,ix),(1,1),
	627	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	628
[c985bef]	629	self.fixed_param.append([None,name3, Tct2
[920a6e5]	630	,None,None,None, None,None])
	631
[780d095]	632	ix =0
	633	iy +=1
	634	self.sizer4_4.Add((20,20),(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	635	for item in self.model.dispersion.keys():
	636	if item in self.model.orientation_params:
[edd166b]	637	if not self.disp_cb_dict.has_key(item):
	638	self.disp_cb_dict[item]= None
[780d095]	639	name1=item+".width"
	640	name2=item+".npts"
	641	name3=item+".nsigmas"
[c99a6c5]	642	if not self.model.details.has_key(name1):
[b421b1a]	643	self.model.details [name1] = ["",None,None]
	644
	645
[780d095]	646	iy += 1
	647	for p in self.model.dispersion[item].keys():
	648
	649	if p=="width":
	650	ix = 0
	651	cb = wx.CheckBox(self, -1, name1, (10, 10))
[b9405cd]	652	cb.SetToolTipString("Check for fitting")
[780d095]	653	wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
	654	self.sizer4_4.Add( cb,( iy, ix),(1,1),
	655	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	656	if self.data.__class__.__name__ =="Data2D":
[6dc9ad8]	657	cb.Show(True)
[c99a6c5]	658	elif cb.IsShown():
[6dc9ad8]	659	cb.Hide()
[780d095]	660	ix = 1
	661	value= self.model.getParam(name1)
[7975f2b]	662	ctl1 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH,20),
[c77d859]	663	style=wx.TE_PROCESS_ENTER)
[1a94c36]	664	ctl1.SetToolTipString("Polydispersity multiplied by the value of '%s'."%item)
[780d095]	665	ctl1.SetValue(str (format_number(value)))
	666	if self.data.__class__.__name__ =="Data2D":
[6dc9ad8]	667	ctl1.Show(True)
[c99a6c5]	668	elif ctl1.IsShown():
[6dc9ad8]	669	ctl1.Hide()
[780d095]	670	self.sizer4_4.Add(ctl1, (iy,ix),(1,1),wx.EXPAND)
	671	## text to show error sign
	672	ix = 2
	673	text2=wx.StaticText(self, -1, '+/-')
	674	self.sizer4_4.Add(text2,(iy, ix),(1,1),
	675	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	676	text2.Hide()
[b421b1a]	677
[780d095]	678	ix = 3
[c13b8cc]	679	ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=0)
[b421b1a]	680
[780d095]	681	self.sizer4_4.Add(ctl2, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[b421b1a]	682	ctl2.Hide()
[920a6e5]	683
	684	ix = 4
[7975f2b]	685	ctl3 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
	686	text_enter_callback = self._onparamRangeEnter)
[eacf1d66]	687
[920a6e5]	688	self.sizer4_4.Add(ctl3, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[c99a6c5]	689
[920a6e5]	690	ctl3.Hide()
	691
	692	ix = 5
[7975f2b]	693	ctl4 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
	694	text_enter_callback = self._onparamRangeEnter)
[920a6e5]	695	self.sizer4_4.Add(ctl4, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	696	ctl4.Hide()
[c99a6c5]	697	#if self.data.__class__.__name__ =="Data2D":
	698	#ctl4.Enable(True)
	699	#elif ctl4.Shown():
	700	#ctl4.Hide()
[920a6e5]	701
	702	if self.engine_type=="park" and self.data.__class__.__name__ =="Data2D":
	703	ctl3.Show(True)
[c99a6c5]	704	ctl4.Show(True)
[920a6e5]	705
	706
	707
	708
[780d095]	709	self.fittable_param.append([cb,name1,ctl1,text2,
[920a6e5]	710	ctl2, ctl3, ctl4,None])
[60132ef]	711	self.orientation_params_disp.append([cb,name1,ctl1,text2,
[920a6e5]	712	ctl2, ctl3, ctl4,None])
[780d095]	713	elif p=="npts":
[920a6e5]	714	ix = 6
[780d095]	715	value= self.model.getParam(name2)
[7975f2b]	716	Tctl = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20),
[780d095]	717	style=wx.TE_PROCESS_ENTER)
[1a94c36]	718
[780d095]	719	Tctl.SetValue(str (format_number(value)))
	720	if self.data.__class__.__name__ =="Data2D":
[6dc9ad8]	721	Tctl.Show(True)
[780d095]	722	else:
[6dc9ad8]	723	Tctl.Hide()
[780d095]	724	self.sizer4_4.Add(Tctl, (iy,ix),(1,1),
	725	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	726	self.fixed_param.append([None,name2, Tctl,None,None,
	727	None, None,None])
[60132ef]	728	self.orientation_params_disp.append([None,name2, Tctl,None,None,
[780d095]	729	None, None,None])
	730	elif p=="nsigmas":
[920a6e5]	731	ix = 7
[780d095]	732	value= self.model.getParam(name3)
[7975f2b]	733	Tct2 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20),
[780d095]	734	style=wx.TE_PROCESS_ENTER)
[1a94c36]	735
[c985bef]	736	Tct2.SetValue(str (format_number(value)))
[780d095]	737	if self.data.__class__.__name__ =="Data2D":
[c985bef]	738	Tct2.Show(True)
[780d095]	739	else:
[c985bef]	740	Tct2.Hide()
	741	self.sizer4_4.Add(Tct2, (iy,ix),(1,1),
[780d095]	742	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	743	ix +=1
[7975f2b]	744
[c985bef]	745	self.fixed_param.append([None,name3, Tct2
[780d095]	746	,None,None, None, None,None])
[920a6e5]	747
[c985bef]	748	self.orientation_params_disp.append([None,name3, Tct2
[1c1436d]	749	,None,None, None, None,None])
[b324aa9]	750
	751
[b421b1a]	752	self.state.disp_cb_dict = copy.deepcopy(self.disp_cb_dict)
	753
[71f0373]	754	self.state.model = self.model.clone()
[240b9966]	755	## save state into
[4043c96]	756	self.state.cb1 = self.cb1.GetValue()
[920a6e5]	757	self._copy_parameters_state(self.parameters, self.state.parameters)
[240b9966]	758	self._copy_parameters_state(self.orientation_params_disp,
	759	self.state.orientation_params_disp)
	760	self._copy_parameters_state(self.fittable_param, self.state.fittable_param)
	761	self._copy_parameters_state(self.fixed_param, self.state.fixed_param)
[d40a9cd]	762
	763
[c77d859]	764	wx.PostEvent(self.parent, StatusEvent(status=\
	765	" Selected Distribution: Gaussian"))
[b324aa9]	766	#Fill the list of fittable parameters
	767	self.select_all_param(event=None)
	768
[71f0373]	769	self.Layout()
[7975f2b]	770
	771
[c77d859]	772	def _onFit(self, event):
	773	"""
	774	Allow to fit
	775	"""
[c13b8cc]	776	#make sure all parameter values are updated.
[ffa69b6]	777	if self.check_invalid_panel():
	778	return
[4470b10]	779	if self.model ==None:
	780	msg="Please select a Model first..."
	781	wx.MessageBox(msg, 'Info')
	782	wx.PostEvent(self.manager.parent, StatusEvent(status=\
	783	"Fit: %s"%msg))
	784	return
	785
[85b3971]	786	flag = self._update_paramv_on_fit()
[acd0bda3]	787
[c77d859]	788	if not flag:
[7609f1a]	789	msg= "Fitting range or parameters are invalid"
[c77d859]	790	wx.PostEvent(self.parent.parent, StatusEvent(status= msg ))
	791	return
	792
	793	if len(self.param_toFit) <= 0:
	794	msg= "Select at least one parameter to fit"
	795	wx.PostEvent(self.parent.parent, StatusEvent(status= msg ))
	796	return
[7975f2b]	797
	798
	799	self.select_param(event =None)
[acd0bda3]	800
[c99a6c5]	801	#Clear errors if exist from previous fitting
[edd166b]	802	#self._clear_Err_on_Fit()
[d0ce2c30]	803
[c13b8cc]	804	# Remove or do not allow fitting on the Q=0 point, especially when y(q=0)=None at x[0].
[1c1436d]	805	self.qmin_x = float(self.qmin.GetValue())
[c13b8cc]	806	self.qmax_x = float( self.qmax.GetValue())
[6f023e8]	807	self.manager._reset_schedule_problem( value=0)
[ca7a626]	808	self.manager.schedule_for_fit( value=1,page=self,fitproblem =None)
	809	self.manager.set_fit_range(page= self,qmin= self.qmin_x, qmax= self.qmax_x)
[ad6dd4c]	810
[c77d859]	811	#single fit
[ca7a626]	812	self.manager.onFit()
[ad6dd4c]	813	## allow stopping the fit
[34a0c17]	814	#if self.engine_type=="scipy":
	815	# self.btFit.SetLabel("Stop")
	816	# self.btFit.Unbind(event=wx.EVT_BUTTON, id= self.btFit.GetId())
	817	# self.btFit.Bind(event= wx.EVT_BUTTON, handler=self._StopFit, id=self.btFit.GetId())
	818	#else:
	819	# self.btFit.SetLabel("Fit")
	820	# self.btFit.Bind(event= wx.EVT_BUTTON, handler=self._onFit, id=self.btFit.GetId())
[7975f2b]	821
[5f9a6b1]	822
[c77d859]	823
[ad6dd4c]	824	def _StopFit(self, event):
	825	"""
	826	Stop fit
	827	"""
	828	self.btFit.SetLabel("Fit")
	829	if self.engine_type=="scipy":
	830	self.manager.stop_fit()
[34a0c17]	831	self.btFit.Unbind(event=wx.EVT_BUTTON, id=self.btFit.GetId())
	832	self.btFit.Bind(event=wx.EVT_BUTTON, handler=self._onFit,id=self.btFit.GetId())
[7975f2b]	833
[ad6dd4c]	834
[c77d859]	835	def _on_select_model(self, event):
	836	"""
	837	call back for model selection
[6747217]	838	"""
[59a7f2d]	839	self._on_select_model_helper()
[70c57ebf]	840	self.set_model_param_sizer(self.model)
[b421b1a]	841
[3b605bb]	842	self.enable_disp.SetValue(False)
	843	self.disable_disp.SetValue(True)
[4043c96]	844	try:
	845	self.set_dispers_sizer()
	846	except:
	847	pass
[6747217]	848	self.btFit.SetFocus()
	849	self.state.enable_disp = self.enable_disp.GetValue()
	850	self.state.disable_disp = self.disable_disp.GetValue()
	851	self.state.pinhole_smearer = self.pinhole_smearer.GetValue()
	852	self.state.slit_smearer = self.slit_smearer.GetValue()
	853
	854	self.state.structurecombobox = self.structurebox.GetCurrentSelection()
	855	self.state.formfactorcombobox = self.formfactorbox.GetCurrentSelection()
	856
[9237df4]	857	if self.model != None:
[70c57ebf]	858	try:
	859	temp_smear= None
	860	if self.enable_smearer.GetValue():
	861	temp_smear= self.smearer
[f72333f]	862	#self.compute_chisqr(temp_smear)
[70c57ebf]	863	except:
	864	## error occured on chisqr computation
	865	pass
[fcd8887d]	866	if self.data is not None and self.data.__class__.__name__ !="Data2D":
[9237df4]	867	## set smearing value whether or not the data contain the smearing info
[6747217]	868	evt = ModelEventbox(model=self.model,
	869	smearer=temp_smear,
	870	qmin= float(self.qmin_x),
[2a2af47]	871	qmax= float(self.qmax_x))
[6747217]	872	else:
	873	evt = ModelEventbox(model=self.model)
[c99a6c5]	874	wx.PostEvent(self.event_owner, evt)
[4043c96]	875
[3595309d]	876	if event !=None:
[3a37fe0]	877	#self._undo.Enable(True)
[3595309d]	878	## post state to fit panel
	879	event = PageInfoEvent(page = self)
	880	wx.PostEvent(self.parent, event)
[330573d]	881
[e2f7b92]	882
[dd5949d]	883	def _onparamEnter(self,event):
	884	"""
	885	when enter value on panel redraw model according to changed
	886	"""
[4470b10]	887	if self.model ==None:
	888	msg="Please select a Model first..."
	889	wx.MessageBox(msg, 'Info')
	890	#wx.PostEvent(self.manager.parent, StatusEvent(status=\
	891	# "Parameters: %s"%msg))
	892	return
	893
[51a71a3]	894	#default flag
[7975f2b]	895	flag = False
[51a71a3]	896	self.fitrange = True
	897	#get event object
[69bee6d]	898	tcrtl= event.GetEventObject()
[4fbc93e]	899	wx.PostEvent(self.manager.parent, StatusEvent(status=" \
	900	updating ... ",type="update"))
[edd166b]	901	#Clear msg if previously shown.
	902	msg= ""
	903	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
[c99a6c5]	904
[7975f2b]	905	if check_float(tcrtl):
[2012eae]	906	flag = self._onparamEnter_helper()
	907	self.set_npts2fit()
[51a71a3]	908	if self.fitrange:
	909	temp_smearer = None
	910	if not self.disable_smearer.GetValue():
	911	temp_smearer= self.current_smearer
	912	## set smearing value whether or not the data contain the smearing info
	913	if self.slit_smearer.GetValue():
	914	flag1 = self.update_slit_smear()
	915	flag = flag or flag1
	916	elif self.pinhole_smearer.GetValue():
	917	flag1 = self.update_pinhole_smear()
	918	flag = flag or flag1
	919	elif self.data.__class__.__name__ !="Data2D":
	920	self.manager.set_smearer(smearer=temp_smearer, qmin= float(self.qmin_x),
	921	qmax= float(self.qmax_x))
	922	if flag:
[f72333f]	923	#self.compute_chisqr(smearer= temp_smearer)
[51a71a3]	924
	925	## new state posted
	926	if self.state_change:
	927	#self._undo.Enable(True)
	928	event = PageInfoEvent(page = self)
	929	wx.PostEvent(self.parent, event)
	930	self.state_change= False
	931	else:
	932	return # invalid fit range: do nothing here: msg already displayed in validate
[69bee6d]	933	else:
[7975f2b]	934	self.save_current_state()
[69bee6d]	935	msg= "Cannot Plot :Must enter a number!!! "
	936	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
[7609f1a]	937
	938	self.save_current_state()
	939	return
	940
[e2f7b92]	941	def _onparamRangeEnter(self, event):
[920a6e5]	942	"""
	943	Check validity of value enter in the parameters range field
	944	"""
[ffa69b6]	945	if self.check_invalid_panel():
	946	return
[920a6e5]	947	tcrtl= event.GetEventObject()
[edd166b]	948	#Clear msg if previously shown.
	949	msg= ""
	950	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
	951	# Flag to register when a parameter has changed.
	952	is_modified = False
[920a6e5]	953	if tcrtl.GetValue().lstrip().rstrip()!="":
	954	try:
	955	value = float(tcrtl.GetValue())
	956	tcrtl.SetBackgroundColour(wx.WHITE)
[edd166b]	957	self._check_value_enter(self.fittable_param ,is_modified)
	958	self._check_value_enter(self.parameters ,is_modified)
[920a6e5]	959	except:
	960	tcrtl.SetBackgroundColour("pink")
[edd166b]	961	msg= "Model Error:wrong value entered : %s"% sys.exc_value
	962	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
	963	return
[920a6e5]	964	else:
	965	tcrtl.SetBackgroundColour(wx.WHITE)
[e2f7b92]	966
[edd166b]	967	#self._undo.Enable(True)
	968	self.save_current_state()
	969	event = PageInfoEvent(page = self)
	970	wx.PostEvent(self.parent, event)
	971	self.state_change= False
[c69b6d5]	972
[12eac73]	973
	974	def _onQrangeEnter(self, event):
	975	"""
	976	Check validity of value enter in the Q range field
	977	"""
[ffa69b6]	978	if self.check_invalid_panel():
	979	return
[12eac73]	980	tcrtl= event.GetEventObject()
	981	#Clear msg if previously shown.
	982	msg= ""
	983	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
	984	# Flag to register when a parameter has changed.
	985	is_modified = False
	986	if tcrtl.GetValue().lstrip().rstrip()!="":
	987	try:
	988	value = float(tcrtl.GetValue())
	989	tcrtl.SetBackgroundColour(wx.WHITE)
	990
	991	# If qmin and qmax have been modified, update qmin and qmax
[85b3971]	992	if self._validate_qrange( self.qmin, self.qmax):
[12eac73]	993	tempmin = float(self.qmin.GetValue())
	994	if tempmin != self.qmin_x:
	995	self.qmin_x = tempmin
	996	tempmax = float(self.qmax.GetValue())
	997	if tempmax != self.qmax_x:
	998	self.qmax_x = tempmax
	999	else:
	1000	tcrtl.SetBackgroundColour("pink")
	1001	msg= "Model Error:wrong value entered : %s"% sys.exc_value
	1002	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
	1003	return
	1004
	1005	except:
	1006	tcrtl.SetBackgroundColour("pink")
	1007	msg= "Model Error:wrong value entered : %s"% sys.exc_value
	1008	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
	1009	return
	1010	#Check if # of points for theory model are valid(>0).
[51a71a3]	1011	# check for 2d
	1012	if self.data.__class__.__name__ =="Data2D":
	1013	# set mask
	1014	radius= numpy.sqrt( self.data.qx_dataself.data.qx_data + self.data.qy_dataself.data.qy_data )
	1015	index_data = ((self.qmin <= radius)&(radius<= self.qmax))
	1016	index_data = (index_data)&(self.data.mask)
	1017	index_data = (index_data)&(numpy.isfinite(self.data.data))
	1018	if len(index_data[index_data]) < 10:
	1019	msg= "Cannot Plot :No or too little npts in that data range!!! "
[12eac73]	1020	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
[51a71a3]	1021	return
	1022	else:
	1023	self.data.mask = index_data
	1024	self.Npts_fit.Setvalue(str(len(self.data.mask)))
	1025	else:
	1026	index_data = ((self.qmin <= self.data.x)&(self.data.x <= self.qmax))
	1027	self.Npts_fit.SetValue(str(len(self.data.x[index_data])))
[12eac73]	1028
	1029	else:
	1030	tcrtl.SetBackgroundColour("pink")
	1031	msg= "Model Error:wrong value entered!!!"
	1032	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
	1033
[f72333f]	1034	self._draw_model()
	1035	##update chi2
	1036	#self.compute_chisqr(smearer= temp_smearer)
[12eac73]	1037	#self._undo.Enable(True)
	1038	self.save_current_state()
	1039	event = PageInfoEvent(page = self)
	1040	wx.PostEvent(self.parent, event)
	1041	self.state_change= False
[f72333f]	1042
[51a71a3]	1043	return
[edd166b]	1044	def _clear_Err_on_Fit(self):
	1045	"""
	1046	hide the error text control shown
	1047	after fitting
	1048	"""
	1049
	1050	if hasattr(self,"text2_3"):
	1051	self.text2_3.Hide()
	1052
	1053	if len(self.parameters)>0:
	1054	for item in self.parameters:
	1055	#Skip t ifhe angle parameters if 1D data
	1056	if self.data.__class__.__name__ !="Data2D":
	1057	if item in self.orientation_params:
	1058	continue
	1059	if item in self.param_toFit:
	1060	continue
	1061	## hide statictext +/-
	1062	if item[3]!=None and item[3].IsShown():
	1063	item[3].Hide()
	1064	## hide textcrtl for error after fit
	1065	if item[4]!=None and item[4].IsShown():
	1066	item[4].Hide()
[7975f2b]	1067
[edd166b]	1068	if len(self.fittable_param)>0:
	1069	for item in self.fittable_param:
	1070	#Skip t ifhe angle parameters if 1D data
	1071	if self.data.__class__.__name__ !="Data2D":
	1072	if item in self.orientation_params:
	1073	continue
	1074	if item in self.param_toFit:
	1075	continue
	1076	## hide statictext +/-
	1077	if item[3]!=None and item[3].IsShown():
	1078	item[3].Hide()
	1079	## hide textcrtl for error after fit
	1080	if item[4]!=None and item[4].IsShown():
	1081	item[4].Hide()
	1082	return
[920a6e5]	1083
[7609f1a]	1084	def _get_defult_custom_smear(self):
	1085	"""
	1086	Get the defult values for custum smearing.
	1087	"""
	1088
	1089	# get the default values
	1090	if self.dxl == None: self.dxl = 0.0
	1091	if self.dxw == None: self.dxw = ""
	1092	if self.dx_min == None: self.dx_min = SMEAR_SIZE_L
	1093	if self.dx_max == None: self.dx_max = SMEAR_SIZE_H
	1094
	1095	def _get_smear_info(self):
	1096	"""
	1097	Get the smear info from data.
	1098	@return: self.smear_type, self.dq_l and self.dq_r,
	1099	respectively the type of the smear, dq_min and dq_max for pinhole smear data
	1100	while dxl and dxw for slit smear
	1101	"""
	1102
	1103	# default
	1104	self.smear_type = None
	1105	self.dq_l = None
	1106	self.dq_r = None
[f72333f]	1107	data = self.data
	1108	if self.data is None:
[7609f1a]	1109	return
[f72333f]	1110	elif self.data.__class__.__name__ == 'Data2D':
	1111	if data.dqx_data == None or data.dqy_data ==None:
	1112	return
[fd50e2f]	1113	elif self.smearer != None and (data.dqx_data.any()!=0) and (data.dqx_data.any()!=0):
[f72333f]	1114	self.smear_type = "Pinhole2d"
	1115	self.dq_l = format_number(numpy.average(data.dqx_data))
	1116	self.dq_r = format_number(numpy.average(data.dqy_data))
	1117	return
	1118	else:
	1119	return
[7609f1a]	1120
[f72333f]	1121
[7609f1a]	1122
	1123	# check if it is pinhole smear and get min max if it is.
	1124	if data.dx != None and all(data.dx !=0):
	1125	self.smear_type = "Pinhole"
[f72333f]	1126	self.dq_l = data.dx[0]
	1127	self.dq_r = data.dx[-1]
[7609f1a]	1128
	1129	# check if it is slit smear and get min max if it is.
	1130	elif data.dxl != None or data.dxw != None:
	1131	self.smear_type = "Slit"
	1132	if data.dxl != None and all(data.dxl !=0):
	1133	self.dq_l = data.dxl[0]
	1134	if data.dxw != None and all(data.dxw !=0):
	1135	self.dq_r = data.dxw[0]
	1136
	1137	#return self.smear_type,self.dq_l,self.dq_r
	1138
	1139	def _show_smear_sizer(self):
	1140	"""
	1141	Show only the sizers depending on smear selection
	1142	"""
[f72333f]	1143	# smear disabled
[7609f1a]	1144	if self.disable_smearer.GetValue():
	1145	self.smear_description_none.Show(True)
[f72333f]	1146	# 2Dsmear
	1147	elif self._is_2D():
	1148	self.smear_description_accuracy_type.Show(True)
	1149	self.smear_accuracy.Show(True)
	1150	self.smear_description_accuracy_type.Show(True)
	1151	self.smear_description_2d.Show(True)
	1152	self.smear_description_2d_x.Show(True)
	1153	self.smear_description_2d_y.Show(True)
	1154	if self.pinhole_smearer.GetValue():
	1155	self.smear_pinhole_min.Show(True)
	1156	self.smear_pinhole_max.Show(True)
	1157	# smear from data
[7609f1a]	1158	elif self.enable_smearer.GetValue():
[f72333f]	1159
[7609f1a]	1160	self.smear_description_dqdata.Show(True)
	1161	if self.smear_type != None:
	1162	self.smear_description_smear_type.Show(True)
	1163	if self.smear_type == 'Slit':
	1164	self.smear_description_slit_height.Show(True)
[f72333f]	1165	self.smear_description_slit_width.Show(True)
[7609f1a]	1166	elif self.smear_type == 'Pinhole':
	1167	self.smear_description_pin_min.Show(True)
	1168	self.smear_description_pin_max.Show(True)
[f72333f]	1169	self.smear_description_smear_type.Show(True)
	1170	self.smear_description_type.Show(True)
[7609f1a]	1171	self.smear_data_left.Show(True)
	1172	self.smear_data_right.Show(True)
[f72333f]	1173	# custom pinhole smear
[7609f1a]	1174	elif self.pinhole_smearer.GetValue():
[f72333f]	1175	if self.smear_type == 'Pinhole':
	1176	self.smear_message_new_p.Show(True)
	1177	self.smear_description_pin_min.Show(True)
	1178	self.smear_description_pin_max.Show(True)
	1179
[7609f1a]	1180	self.smear_pinhole_min.Show(True)
	1181	self.smear_pinhole_max.Show(True)
[f72333f]	1182	# custom slit smear
[7609f1a]	1183	elif self.slit_smearer.GetValue():
	1184	self.smear_message_new_s.Show(True)
	1185	self.smear_description_slit_height.Show(True)
	1186	self.smear_slit_height.Show(True)
	1187	self.smear_description_slit_width.Show(True)
	1188	self.smear_slit_width.Show(True)
	1189
	1190	def _hide_all_smear_info(self):
	1191	"""
	1192	Hide all smearing messages in the set_smearer sizer
	1193	"""
	1194	self.smear_description_none.Hide()
	1195	self.smear_description_dqdata.Hide()
	1196	self.smear_description_type.Hide()
	1197	self.smear_description_smear_type.Hide()
[f72333f]	1198	self.smear_description_accuracy_type.Hide()
	1199	self.smear_description_2d_x.Hide()
	1200	self.smear_description_2d_y.Hide()
[7609f1a]	1201	self.smear_description_2d.Hide()
[f72333f]	1202
	1203	self.smear_accuracy.Hide()
[7609f1a]	1204	self.smear_data_left.Hide()
	1205	self.smear_data_right.Hide()
	1206	self.smear_description_pin_min.Hide()
	1207	self.smear_pinhole_min.Hide()
	1208	self.smear_description_pin_max.Hide()
	1209	self.smear_pinhole_max.Hide()
	1210	self.smear_description_slit_height.Hide()
	1211	self.smear_slit_height.Hide()
	1212	self.smear_description_slit_width.Hide()
	1213	self.smear_slit_width.Hide()
	1214	self.smear_message_new_p.Hide()
	1215	self.smear_message_new_s.Hide()
	1216
[f72333f]	1217	def _set_accuracy_list(self):
	1218	"""
	1219	Set the list of an accuracy in 2D custum smear: Xhigh, High, Med, or Low
	1220	"""
	1221	# list of accuracy choices
	1222	list = ['Low','Med','High','Xhigh']
	1223	for idx in range(len(list)):
	1224	self.smear_accuracy.Append(list[idx],idx)
	1225
	1226	def _on_select_accuracy(self,event):
	1227	"""
	1228	Select an accuracy in 2D custom smear: Xhigh, High, Med, or Low
	1229	"""
	1230	#event.Skip()
	1231
[4fbc93e]	1232	# Check if the accuracy is same as before
[4470b10]	1233
[4fbc93e]	1234	#self.smear2d_accuracy = event.GetEventObject().GetValue()
	1235	self.smear2d_accuracy = self.smear_accuracy.GetValue()
[f72333f]	1236	if self.pinhole_smearer.GetValue():
	1237	self.onPinholeSmear(event=None)
[4fbc93e]	1238	else:
	1239	self.onSmear(event=None)
	1240	if self.current_smearer != None:
[f72333f]	1241	self.current_smearer.set_accuracy(accuracy = self.smear2d_accuracy)
[4fbc93e]	1242
[f72333f]	1243	event.Skip()
[4fbc93e]	1244
[51a71a3]	1245	def _onMask(self, event):
	1246	"""
	1247	Build a panel to allow to edit Mask
	1248	"""
	1249
	1250	from sans.guiframe.local_perspectives.plotting.masking import MaskPanel as MaskDialog
	1251
	1252	self.panel = MaskDialog(self, data=self.data,id =-1 )
	1253	self.panel.Bind(wx.EVT_CLOSE, self._draw_masked_model)
	1254	self.panel.ShowModal()
	1255	#wx.PostEvent(self.parent, event)
	1256
	1257	def _draw_masked_model(self,event):
[f72333f]	1258	"""
	1259	Draw model image w/mask
	1260	"""
[51a71a3]	1261	event.Skip()
	1262
	1263	is_valid_qrange = self._update_paramv_on_fit()
[247cb58]	1264
[51a71a3]	1265	if is_valid_qrange:
[247cb58]	1266	# try re draw the model plot if it exists
[51a71a3]	1267	self._draw_model()
	1268	self.panel.Destroy() # frame
	1269	self.set_npts2fit()
[247cb58]	1270	elif self.model == None:
	1271	self.panel.Destroy()
	1272	self.set_npts2fit()
	1273	msg= "No model is found on updating MASK in the model plot... "
	1274	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
[51a71a3]	1275	else:
[247cb58]	1276	msg = ' Please consider your Q range, too.'
[51a71a3]	1277	self.panel.ShowMessage(msg)
[882a912]	1278
[9237df4]	1279	def set_data(self, data):
[882a912]	1280	"""
	1281	reset the current data
	1282	"""
[9237df4]	1283	self.data = data
[51a71a3]	1284
	1285
[9237df4]	1286	if self.data is None:
	1287	data_min = ""
	1288	data_max = ""
	1289	data_name = ""
[ffa69b6]	1290	self.formfactorbox.Disable()
	1291	self.structurebox.Disable()
[9237df4]	1292	else:
[00e8df8]	1293	self.smearer = smear_selection( self.data )
[d493f66]	1294	self.disable_smearer.SetValue(True)
[00e8df8]	1295	if self.smearer == None:
	1296	self.enable_smearer.Disable()
	1297	else:
	1298	self.enable_smearer.Enable()
[51a71a3]	1299
[7609f1a]	1300	# more disables for 2D
	1301	if self.data.__class__.__name__ =="Data2D":
	1302	self.slit_smearer.Disable()
[51a71a3]	1303	self.default_mask = copy.deepcopy(self.data.mask)
[00e8df8]	1304
[d493f66]	1305
[ffa69b6]	1306	self.formfactorbox.Enable()
	1307	self.structurebox.Enable()
[9237df4]	1308	data_name = self.data.name
	1309	#set maximum range for x in linear scale
	1310	if not hasattr(self.data,"data"): #Display only for 1D data fit
	1311	# Minimum value of data
	1312	data_min = min(self.data.x)
	1313	# Maximum value of data
	1314	data_max = max(self.data.x)
[51a71a3]	1315	#number of total data points
	1316	self.Npts_total.SetValue(str(len(self.data.x)))
	1317	#default:number of data points selected to fit
	1318	self.Npts_fit.SetValue(str(len(self.data.x)))
[19403da]	1319	self.btEditMask.Disable()
	1320	self.EditMask_title.Disable()
[9237df4]	1321	else:
	1322	## Minimum value of data
	1323	data_min = 0
	1324	x = max(math.fabs(self.data.xmin), math.fabs(self.data.xmax))
	1325	y = max(math.fabs(self.data.ymin), math.fabs(self.data.ymax))
	1326	## Maximum value of data
	1327	data_max = math.sqrt(xx + yy)
[51a71a3]	1328	#number of total data points
	1329	self.Npts_total.SetValue(str(len(self.data.data)))
	1330	#default:number of data points selected to fit
	1331	self.Npts_fit.SetValue(str(len(self.data.data)))
[19403da]	1332	self.btEditMask.Enable()
	1333	self.EditMask_title.Enable()
	1334
[51a71a3]	1335
[9237df4]	1336	self.dataSource.SetValue(data_name)
	1337	self.qmin_x = data_min
	1338	self.qmax_x = data_max
[ffa69b6]	1339	self.minimum_q.SetValue(str(data_min))
	1340	self.maximum_q.SetValue(str(data_max))
	1341	self.qmin.SetValue(str(data_min))
	1342	self.qmax.SetValue(str(data_max))
	1343	self.qmin.SetBackgroundColour("white")
	1344	self.qmax.SetBackgroundColour("white")
	1345	self.state.data = data
	1346	self.state.qmin = self.qmin_x
	1347	self.state.qmax = self.qmax_x
	1348
[7609f1a]	1349
[240b9966]	1350	def reset_page(self, state,first=False):
[a074145]	1351	"""
	1352	reset the state
	1353	"""
	1354	self.reset_page_helper(state)
[b421b1a]	1355	import sans.guiframe.gui_manager
[f20767b]	1356	evt = ModelEventbox(model=state.model)
[edd166b]	1357	wx.PostEvent(self.event_owner, evt)
	1358
	1359	if self.engine_type != None:
	1360	self.manager._on_change_engine(engine=self.engine_type)
[7975f2b]	1361
[edd166b]	1362	self.select_param(event = None)
	1363	#Save state_fit
[c985bef]	1364	self.save_current_state_fit()
[7975f2b]	1365	self._lay_out()
[c985bef]	1366	self.Refresh()
[edd166b]	1367
[2140e68]	1368	def get_range(self):
	1369	"""
	1370	return the fitting range
	1371	"""
[c9a4377]	1372	return float(self.qmin_x) , float(self.qmax_x)
[7975f2b]	1373
	1374	def get_npts2fit(self):
	1375	"""
	1376	return numbers of data points within qrange
	1377	Note: This is for Park where chi2 is not normalized by Npts of fit
	1378	"""
	1379	npts2fit = 0
	1380	qmin,qmax = self.get_range()
[51a71a3]	1381	if self.data.__class__.__name__ =="Data2D":
	1382	radius= numpy.sqrt( self.data.qx_dataself.data.qx_data + self.data.qy_dataself.data.qy_data )
	1383	index_data = (self.qmin_x <= radius)&(radius<= self.qmax_x)
	1384	index_data= (index_data)&(self.data.mask)
	1385	index_data = (index_data)&(numpy.isfinite(self.data.data))
	1386	npts2fit = len(self.data.data[index_data])
[7975f2b]	1387	else:
	1388	for qx in self.data.x:
	1389	if qx >= qmin and qx <= qmax:
	1390	npts2fit += 1
	1391	return npts2fit
	1392
[51a71a3]	1393	def set_npts2fit(self):
	1394	"""
	1395	setValue Npts for fitting
	1396	"""
	1397	self.Npts_fit.SetValue(str(self.get_npts2fit()))
	1398
	1399
[c69b6d5]	1400	def get_chi2(self):
	1401	"""
	1402	return the current chi2
	1403	"""
[2012eae]	1404	return self.tcChi.GetValue()
[c77d859]	1405
	1406	def get_param_list(self):
	1407	"""
	1408	@return self.param_toFit: list containing references to TextCtrl
	1409	checked.Theses TextCtrl will allow reference to parameters to fit.
	1410	@raise: if return an empty list of parameter fit will nnote work
	1411	properly so raise ValueError,"missing parameter to fit"
	1412	"""
	1413	if self.param_toFit !=[]:
	1414	return self.param_toFit
	1415	else:
	1416	raise ValueError,"missing parameter to fit"
	1417
[6d91073]	1418	def onsetValues(self,chisqr,p_name, out,cov):
[c77d859]	1419	"""
	1420	Build the panel from the fit result
	1421	@param chisqr:Value of the goodness of fit metric
[6d91073]	1422	@p_name: the name of parameters
[c77d859]	1423	@param out:list of parameter with the best value found during fitting
	1424	@param cov:Covariance matrix
	1425
	1426	"""
[edd166b]	1427	if out == None or not numpy.isfinite(chisqr):
	1428	raise ValueError,"Fit error occured..."
	1429
	1430	is_modified = False
	1431	has_error = False
[7975f2b]	1432
	1433	#Hide textctrl boxes of errors.
[edd166b]	1434	self._clear_Err_on_Fit()
[7975f2b]	1435
	1436	#Check if chi2 is finite
	1437	if chisqr != None or numpy.isfinite(chisqr):
[e575db9]	1438	#format chi2
[7609f1a]	1439	if self.engine_type == "park":
[e575db9]	1440	npt_fit = float(self.get_npts2fit())
[7609f1a]	1441	if npt_fit > 0:
	1442	chisqr =chisqr/npt_fit
[7975f2b]	1443	chi2 = format_number(chisqr)
[2012eae]	1444	self.tcChi.SetValue(chi2)
[7975f2b]	1445	self.tcChi.Refresh()
	1446	else:
[2012eae]	1447	self.tcChi.SetValue("-")
[c13b8cc]	1448
[edd166b]	1449	#Hide error title
	1450	if self.text2_3.IsShown():
	1451	self.text2_3.Hide()
	1452
[e473e4f5]	1453	try:
	1454	n = self.disp_box.GetCurrentSelection()
	1455	dispersity= self.disp_box.GetClientData(n)
[7975f2b]	1456	if dispersity !=None and self.enable_disp.GetValue():
	1457	name= dispersity.__name__
	1458	if name == "GaussianDispersion":
	1459	if hasattr(self,"text_disp_1" ):
	1460	if self.text_disp_1 !=None:
	1461	self.text_disp_1.Hide()
[e473e4f5]	1462	except:
[7975f2b]	1463	dispersty = None
[e473e4f5]	1464	pass
[c77d859]	1465	#set the panel when fit result are float not list
[c99a6c5]	1466	if out.__class__== numpy.float64:
[c77d859]	1467	self.param_toFit[0][2].SetValue(format_number(out))
[0a518e4c]	1468
[c69b6d5]	1469	if self.param_toFit[0][4].IsShown:
	1470	self.param_toFit[0][4].Hide()
[c77d859]	1471	if cov !=None :
	1472	self.text2_3.Show(True)
[e473e4f5]	1473	try:
[7975f2b]	1474	if dispersity !=None:
	1475	name= dispersity.__name__
	1476	if name == "GaussianDispersion" and self.enable_disp.GetValue():
	1477	if hasattr(self,"text_disp_1" ):
	1478	if self.text_disp_1 !=None:
	1479	self.text_disp_1.Show(True)
[e473e4f5]	1480	except:
	1481	pass
[c99a6c5]	1482
[ad6dd4c]	1483	if cov[0]==None or not numpy.isfinite(cov[0]):
[c69b6d5]	1484	if self.param_toFit[0][3].IsShown:
	1485	self.param_toFit[0][3].Hide()
[7975f2b]	1486	else:
	1487	self.param_toFit[0][3].Show(True)
[69bee6d]	1488	self.param_toFit[0][4].Show(True)
[c69b6d5]	1489	self.param_toFit[0][4].SetValue(format_number(cov[0]))
[ffb838f]	1490	has_error = True
[c77d859]	1491	else:
[edd166b]	1492
[c69b6d5]	1493	i = 0
[c77d859]	1494	#Set the panel when fit result are list
[7975f2b]	1495	for item in self.param_toFit:
[edd166b]	1496	if len(item)>5 and item != None:
	1497	## reset error value to initial state
[7975f2b]	1498	item[3].Hide()
	1499	item[4].Hide()
	1500
[edd166b]	1501	for ind in range(len(out)):
	1502
	1503	if item[1] == p_name[ind]:
	1504	break
[7975f2b]	1505	if len(out)<=len(self.param_toFit) and out[ind] !=None:
	1506	val_out = format_number(out[ind])
	1507	item[2].SetValue(val_out)
[c99a6c5]	1508
[7975f2b]	1509	if(cov !=None):
[edd166b]	1510
	1511	try:
[7975f2b]	1512	if dispersity !=None:
	1513	name= dispersity.__name__
	1514	if name == "GaussianDispersion" and self.enable_disp.GetValue():
	1515	if hasattr(self,"text_disp_1" ):
	1516	if self.text_disp_1!=None:
	1517	if not self.text_disp_1.IsShown():
	1518	self.text_disp_1.Show(True)
[edd166b]	1519	except:
	1520	pass
	1521
[7975f2b]	1522	if cov[ind]!=None :
	1523	if numpy.isfinite(float(cov[ind])):
	1524	val_err = format_number(cov[ind])
	1525	item[3].Show(True)
	1526	item[4].Show(True)
	1527	item[4].SetValue(val_err)
	1528
	1529	has_error = True
[e575db9]	1530	i += 1
[c99a6c5]	1531	#Show error title when any errors displayed
[b421b1a]	1532	if has_error:
[c99a6c5]	1533	if not self.text2_3.IsShown():
[edd166b]	1534	self.text2_3.Show(True)
[7975f2b]	1535
	1536	## save current state
	1537	self.save_current_state()
	1538	#plot model
	1539	self._draw_model()
	1540	self._lay_out()
	1541	#PostStatusEvent
[9237df4]	1542	msg = "Fit completed! "
[7975f2b]	1543	wx.PostEvent(self.manager.parent, StatusEvent(status=msg))
[c69b6d5]	1544
[7609f1a]	1545	def onPinholeSmear(self, event):
	1546	"""
	1547	Create a custom pinhole smear object that will change the way residuals
	1548	are compute when fitting
[4fbc93e]	1549	@ accuracy: given by strings'High','Med', 'Low' FOR 2d, None for 1D
[7609f1a]	1550	"""
	1551
	1552	if self.check_invalid_panel():
	1553	return
	1554	if self.model ==None:
[4470b10]	1555	self.disable_smearer.SetValue(True)
	1556	msg="Please select a Model first..."
	1557	wx.MessageBox(msg, 'Info')
[7609f1a]	1558	wx.PostEvent(self.manager.parent, StatusEvent(status=\
	1559	"Smear: %s"%msg))
	1560	return
[4470b10]	1561
[2d409fa]	1562	# Need update param values
[f72333f]	1563	self._update_paramv_on_fit()
	1564
[7609f1a]	1565	# msg default
	1566	msg = None
	1567	if event != None:
	1568	tcrtl= event.GetEventObject()
	1569	# event case of radio button
	1570	if tcrtl.GetValue()== True:
	1571	self.dx_min = 0.0
	1572	self.dx_max = 0.0
	1573	is_new_pinhole = True
	1574	else:
	1575	is_new_pinhole = self._is_changed_pinhole()
[d493f66]	1576	else:
[1da28b8]	1577	is_new_pinhole = True
[7609f1a]	1578	# if any value is changed
	1579	if is_new_pinhole:
	1580	msg = self._set_pinhole_smear()
	1581	# hide all silt sizer
	1582	self._hide_all_smear_info()
[f72333f]	1583
[7609f1a]	1584	##Calculate chi2
[f72333f]	1585	temp_smearer = self.current_smearer
	1586	#self.compute_chisqr(smearer= temp_smearer)
	1587
[7609f1a]	1588	# show relevant slit sizers
	1589	self._show_smear_sizer()
[f72333f]	1590
[7609f1a]	1591	self.sizer_set_smearer.Layout()
	1592	self.Layout()
	1593
	1594	if event != None:
	1595	event.Skip()
	1596	#self._undo.Enable(True)
	1597	self.save_current_state()
	1598	event = PageInfoEvent(page = self)
	1599	wx.PostEvent(self.parent, event)
	1600
	1601	def _is_changed_pinhole(self):
	1602	"""
	1603	check if any of pinhole smear is changed
	1604	return: True or False
	1605	"""
	1606	# get the values
	1607	pin_min = self.smear_pinhole_min.GetValue()
	1608	pin_max = self.smear_pinhole_max.GetValue()
	1609
	1610	# Check changes in slit width
	1611	try:
	1612	dx_min = float(pin_min)
	1613	except:
	1614	return True
	1615	if self.dx_min != dx_min:
	1616	return True
	1617
	1618	# Check changes in slit heigth
	1619	try:
	1620	dx_max = float(pin_max)
	1621	except:
	1622	return True
	1623	if self.dx_max != dx_max:
	1624	return True
	1625	return False
	1626
	1627	def _set_pinhole_smear(self):
	1628	"""
	1629	Set custom pinhole smear
	1630	return: msg
	1631	"""
	1632	# copy data
	1633	data = copy.deepcopy(self.data)
[f72333f]	1634	if self._is_2D():
	1635	self.smear_type = 'Pinhole2d'
	1636	len_data = len(data.data)
	1637	data.dqx_data = numpy.zeros(len_data)
	1638	data.dqy_data = numpy.zeros(len_data)
	1639	else:
	1640	self.smear_type = 'Pinhole'
	1641	len_data = len(data.x)
	1642	data.dx = numpy.zeros(len_data)
	1643	data.dxl = None
	1644	data.dxw = None
[7609f1a]	1645	msg = None
	1646
	1647	get_pin_min = self.smear_pinhole_min
	1648	get_pin_max = self.smear_pinhole_max
	1649
[f72333f]	1650	if not check_float(get_pin_min):
	1651	get_pin_min.SetBackgroundColour("pink")
	1652	msg= "Model Error:wrong value entered!!!"
	1653	elif not check_float(get_pin_max ):
	1654	get_pin_max.SetBackgroundColour("pink")
	1655	msg= "Model Error:wrong value entered!!!"
	1656	else:
[7609f1a]	1657	if len_data < 2: len_data = 2
	1658	self.dx_min = float(get_pin_min.GetValue())
	1659	self.dx_max = float(get_pin_max.GetValue())
[f72333f]	1660	if self.dx_min < 0:
	1661	get_pin_min.SetBackgroundColour("pink")
	1662	msg= "Model Error:This value can not be negative!!!"
	1663	elif self.dx_max <0:
	1664	get_pin_max.SetBackgroundColour("pink")
	1665	msg= "Model Error:This value can not be negative!!!"
	1666	elif self.dx_min != None and self.dx_max != None:
	1667	if self._is_2D():
	1668	data.dqx_data[data.dqx_data==0] = self.dx_min
	1669	data.dqy_data[data.dqy_data==0] = self.dx_max
	1670	elif self.dx_min == self.dx_max:
[7609f1a]	1671	data.dx[data.dx==0] = self.dx_min
	1672	else:
[f72333f]	1673	step = (self.dx_max - self.dx_min)/(len_data-1)
[7609f1a]	1674	data.dx = numpy.arange(self.dx_min,self.dx_max+step/1.1,step)
	1675	elif self.dx_min != None:
[f72333f]	1676	if self._is_2D(): data.dqx_data[data.dqx_data==0] = self.dx_min
	1677	else: data.dx[data.dx==0] = self.dx_min
[7609f1a]	1678	elif self.dx_max != None:
[f72333f]	1679	if self._is_2D(): data.dqy_data[data.dqy_data==0] = self.dx_max
	1680	else: data.dx[data.dx==0] = self.dx_max
[7609f1a]	1681	self.current_smearer = smear_selection(data)
[4fbc93e]	1682	# 2D need to set accuracy
	1683	if self._is_2D():
	1684	self.current_smearer.set_accuracy(accuracy = self.smear2d_accuracy)
[7609f1a]	1685
[f72333f]	1686	if msg != None:
[7609f1a]	1687	wx.PostEvent(self.manager.parent, StatusEvent(status = msg ))
[f72333f]	1688	else:
	1689	get_pin_min.SetBackgroundColour("white")
	1690	get_pin_max.SetBackgroundColour("white")
[7609f1a]	1691	## set smearing value whether or not the data contain the smearing info
	1692	self.manager.set_smearer(smearer=self.current_smearer, qmin= float(self.qmin_x),qmax= float(self.qmax_x))
	1693	return msg
	1694
	1695	def update_pinhole_smear(self):
	1696	"""
	1697	called by kill_focus on pinhole TextCntrl
	1698	to update the changes
	1699	return: msg: False when wrong value was entered
	1700	"""
	1701	# msg default
	1702	msg = None
	1703	# check if any value is changed
	1704	if self._is_changed_pinhole():
	1705	msg = self._set_pinhole_smear()
	1706	#self._undo.Enable(True)
	1707	self.save_current_state()
	1708
	1709	if msg != None:
	1710	return False
	1711	else:
	1712	return True
	1713
	1714	def onSlitSmear(self, event):
	1715	"""
	1716	Create a custom slit smear object that will change the way residuals
	1717	are compute when fitting
	1718	"""
[4470b10]	1719
[7609f1a]	1720	if self.check_invalid_panel():
	1721	return
[4470b10]	1722
[7609f1a]	1723	if self.model ==None:
[4470b10]	1724	self.disable_smearer.SetValue(True)
	1725
	1726	msg="Please select a Model first..."
	1727	wx.MessageBox(msg, 'Info')
[7609f1a]	1728	wx.PostEvent(self.manager.parent, StatusEvent(status=\
	1729	"Smear: %s"%msg))
	1730	return
[4470b10]	1731
[2d409fa]	1732	# Need update param values
	1733	self._update_paramv_on_fit()
[f72333f]	1734
[7609f1a]	1735	# msg default
	1736	msg = None
	1737	# for event given
	1738	if event != None:
	1739	tcrtl= event.GetEventObject()
	1740	# event case of radio button
	1741	if tcrtl.GetValue():
	1742	self.dxl = 0.0
	1743	self.dxw = 0.0
	1744	is_new_slit = True
	1745	else:
	1746	is_new_slit = self._is_changed_slit()
[d493f66]	1747	else:
[1da28b8]	1748	is_new_slit = True
[4470b10]	1749
[7609f1a]	1750	# if any value is changed
	1751	if is_new_slit:
	1752	msg = self._set_slit_smear()
[4470b10]	1753
[7609f1a]	1754	# hide all silt sizer
	1755	self._hide_all_smear_info()
	1756	##Calculate chi2
[f72333f]	1757	#self.compute_chisqr(smearer= self.current_smearer)
[7609f1a]	1758	# show relevant slit sizers
	1759	self._show_smear_sizer()
	1760	self.sizer_set_smearer.Layout()
	1761	self.Layout()
[4470b10]	1762
[7609f1a]	1763	if event != None:
	1764	event.Skip()
	1765	#self._undo.Enable(True)
	1766	self.save_current_state()
	1767	event = PageInfoEvent(page = self)
	1768	wx.PostEvent(self.parent, event)
	1769	if msg != None:
	1770	wx.PostEvent(self.manager.parent, StatusEvent(status = msg))
	1771
[4470b10]	1772
[7609f1a]	1773	def _is_changed_slit(self):
	1774	"""
	1775	check if any of slit lengths is changed
	1776	return: True or False
	1777	"""
[4470b10]	1778
[7609f1a]	1779	# get the values
	1780	width = self.smear_slit_width.GetValue()
	1781	height = self.smear_slit_height.GetValue()
	1782
	1783	# check and change the box bg color if it was pink but it should be white now
	1784	# because this is the case that _set_slit_smear() will not handle
	1785	if height.lstrip().rstrip()=="":
	1786	self.smear_slit_height.SetBackgroundColour(wx.WHITE)
	1787	if width.lstrip().rstrip()=="":
	1788	self.smear_slit_width.SetBackgroundColour(wx.WHITE)
	1789
	1790	# Check changes in slit width
	1791	if width == "":
	1792	dxw = 0.0
	1793	else:
	1794	try:
	1795	dxw = float(width)
	1796	except:
	1797	return True
	1798	if self.dxw != dxw:
	1799	return True
	1800
	1801	# Check changes in slit heigth
	1802	if height == "":
	1803	dxl = 0.0
	1804	else:
	1805	try:
	1806	dxl = float(height)
	1807	except:
	1808	return True
	1809	if self.dxl != dxl:
	1810	return True
[4470b10]	1811
[7609f1a]	1812	return False
	1813
	1814	def _set_slit_smear(self):
	1815	"""
	1816	Set custom slit smear
	1817	return: msg
	1818	"""
	1819	temp_smearer = None
	1820	# make sure once more if it is smearer
	1821	data = copy.deepcopy(self.data)
	1822	data_len = len(data.x)
	1823	data.dx = None
	1824	data.dxl = None
	1825	data.dxw = None
	1826	msg = None
	1827
	1828	try:
	1829	self.dxl = float(self.smear_slit_height.GetValue())
	1830	data.dxl = self.dxl* numpy.ones(data_len)
	1831	self.smear_slit_height.SetBackgroundColour(wx.WHITE)
	1832	except:
	1833	data.dxl = numpy.zeros(data_len)
	1834	if self.smear_slit_height.GetValue().lstrip().rstrip()!="":
	1835	self.smear_slit_height.SetBackgroundColour("pink")
	1836	msg = "Wrong value entered... "
	1837	else:
	1838	self.smear_slit_height.SetBackgroundColour(wx.WHITE)
	1839	try:
	1840	self.dxw = float(self.smear_slit_width.GetValue())
	1841	self.smear_slit_width.SetBackgroundColour(wx.WHITE)
	1842	data.dxw = self.dxw* numpy.ones(data_len)
	1843	except:
	1844	data.dxw = numpy.zeros(data_len)
	1845	if self.smear_slit_width.GetValue().lstrip().rstrip()!="":
	1846	self.smear_slit_width.SetBackgroundColour("pink")
	1847	msg = "Wrong Fit value entered... "
	1848	else:
	1849	self.smear_slit_width.SetBackgroundColour(wx.WHITE)
[4470b10]	1850
[7609f1a]	1851	self.current_smearer = smear_selection(data)
	1852	#temp_smearer = self.current_smearer
	1853	## set smearing value whether or not the data contain the smearing info
	1854	self.manager.set_smearer(smearer=self.current_smearer, qmin= float(self.qmin_x), qmax= float(self.qmax_x))
	1855
	1856	return msg
	1857
	1858
	1859	def update_slit_smear(self):
	1860	"""
	1861	called by kill_focus on pinhole TextCntrl
	1862	to update the changes
	1863	return: msg: False when wrong value was entered
	1864	"""
	1865	# msg default
	1866	msg = None
	1867	# check if any value is changed
	1868	if self._is_changed_slit():
	1869	msg = self._set_slit_smear()
	1870	#self._undo.Enable(True)
	1871	self.save_current_state()
	1872
	1873	if msg != None:
	1874	return False
	1875	else:
	1876	return True
	1877
	1878
[c77d859]	1879	def onSmear(self, event):
	1880	"""
	1881	Create a smear object that will change the way residuals
	1882	are compute when fitting
	1883	"""
[f72333f]	1884	if event != None:
	1885	event.Skip()
[ffa69b6]	1886	if self.check_invalid_panel():
	1887	return
[2a2af47]	1888	if self.model ==None:
[4470b10]	1889	self.disable_smearer.SetValue(True)
	1890	msg="Please select a Model first..."
	1891	wx.MessageBox(msg, 'Info')
[2a2af47]	1892	wx.PostEvent(self.manager.parent, StatusEvent(status=\
	1893	"Smear: %s"%msg))
	1894	return
[f72333f]	1895
[2d409fa]	1896	# Need update param values
	1897	self._update_paramv_on_fit()
[f72333f]	1898
[3370922]	1899	temp_smearer = None
[7609f1a]	1900	self._get_smear_info()
	1901
	1902	#renew smear sizer
	1903	if self.smear_type != None:
	1904	self.smear_description_smear_type.SetValue(str(self.smear_type))
	1905	self.smear_data_left.SetValue(str(self.dq_l))
	1906	self.smear_data_right.SetValue(str(self.dq_r))
[f72333f]	1907
[7609f1a]	1908	self._hide_all_smear_info()
	1909
	1910	data = copy.deepcopy(self.data)
	1911	# make sure once more if it is smearer
	1912	self.current_smearer = smear_selection(data)
[f72333f]	1913
[c77d859]	1914	if self.enable_smearer.GetValue():
	1915	if hasattr(self.data,"dxl"):
[7609f1a]	1916
[c77d859]	1917	msg= ": Resolution smearing parameters"
	1918	if hasattr(self.data,"dxw"):
	1919	msg= ": Slit smearing parameters"
[997131a]	1920	if self.smearer ==None:
[c77d859]	1921	wx.PostEvent(self.manager.parent, StatusEvent(status=\
	1922	"Data contains no smearing information"))
	1923	else:
	1924	wx.PostEvent(self.manager.parent, StatusEvent(status=\
[f72333f]	1925	"Data contains smearing information"))
	1926
	1927	#self.smear_description_dqdata.Show(True)
[7609f1a]	1928	self.smear_data_left.Show(True)
	1929	self.smear_data_right.Show(True)
[f72333f]	1930	temp_smearer= self.current_smearer
[7609f1a]	1931	elif self.disable_smearer.GetValue():
	1932	self.smear_description_none.Show(True)
	1933
	1934	self._show_smear_sizer()
[f72333f]	1935
[7609f1a]	1936	self.sizer_set_smearer.Layout()
	1937	self.Layout()
[fb8daaaf]	1938	## set smearing value whether or not the data contain the smearing info
[3370922]	1939	self.manager.set_smearer(smearer=temp_smearer, qmin= float(self.qmin_x),
[2b63df0]	1940	qmax= float(self.qmax_x))
[f72333f]	1941
[875f1a2]	1942	##Calculate chi2
[f72333f]	1943	#self.compute_chisqr(smearer= temp_smearer)
[5582a9f1]	1944
	1945	self.state.enable_smearer= self.enable_smearer.GetValue()
	1946	self.state.disable_smearer=self.disable_smearer.GetValue()
[7609f1a]	1947	self.state.pinhole_smearer = self.pinhole_smearer.GetValue()
	1948	self.state.slit_smearer = self.slit_smearer.GetValue()
[f72333f]	1949
	1950	def on_complete_chisqr(self, event):
	1951	"""
	1952	print result chisqr
	1953	@event: activated by fitting/ complete after draw
	1954	"""
	1955	try:
	1956	if event ==None:
	1957	output= "-"
	1958	else:
	1959	output = event.output
	1960	self.tcChi.SetValue(str(format_number(output)))
	1961
	1962	self.state.tcChi =self.tcChi
	1963	except:
	1964	pass
[c77d859]	1965
	1966
	1967	def select_all_param(self,event):
	1968	"""
	1969	set to true or false all checkBox given the main checkbox value cb1
[d1e0473]	1970	"""
	1971
[c77d859]	1972	self.param_toFit=[]
	1973	if self.parameters !=[]:
[998b6b8]	1974	if self.cb1.GetValue():
[c77d859]	1975	for item in self.parameters:
[3fef0a8]	1976	## for data2D select all to fit
	1977	if self.data.__class__.__name__=="Data2D":
[c77d859]	1978	item[0].SetValue(True)
	1979	self.param_toFit.append(item )
[3fef0a8]	1980	else:
	1981	## for 1D all parameters except orientation
	1982	if not item in self.orientation_params:
	1983	item[0].SetValue(True)
	1984	self.param_toFit.append(item )
[b324aa9]	1985	#if len(self.fittable_param)>0:
	1986	for item in self.fittable_param:
	1987	if self.data.__class__.__name__=="Data2D":
	1988	item[0].SetValue(True)
	1989	self.param_toFit.append(item )
	1990	else:
	1991	## for 1D all parameters except orientation
	1992	if not item in self.orientation_params_disp:
[3fef0a8]	1993	item[0].SetValue(True)
	1994	self.param_toFit.append(item )
[c77d859]	1995	else:
	1996	for item in self.parameters:
	1997	item[0].SetValue(False)
	1998	for item in self.fittable_param:
	1999	item[0].SetValue(False)
	2000	self.param_toFit=[]
[330573d]	2001
[52cac46]	2002	self.save_current_state_fit()
[b324aa9]	2003
[3595309d]	2004	if event !=None:
[3a37fe0]	2005	#self._undo.Enable(True)
[3595309d]	2006	## post state to fit panel
	2007	event = PageInfoEvent(page = self)
	2008	wx.PostEvent(self.parent, event)
	2009
[c77d859]	2010
	2011
	2012	def select_param(self,event):
	2013	"""
	2014	Select TextCtrl checked for fitting purpose and stores them
	2015	in self.param_toFit=[] list
	2016	"""
	2017	self.param_toFit=[]
	2018	for item in self.parameters:
[edd166b]	2019	#Skip t ifhe angle parameters if 1D data
	2020	if self.data.__class__.__name__ !="Data2D":
	2021	if item in self.orientation_params:
	2022	continue
[c77d859]	2023	#Select parameters to fit for list of primary parameters
[998b6b8]	2024	if item[0].GetValue():
[c77d859]	2025	if not (item in self.param_toFit):
	2026	self.param_toFit.append(item )
	2027	else:
	2028	#remove parameters from the fitting list
	2029	if item in self.param_toFit:
	2030	self.param_toFit.remove(item)
[edd166b]	2031
[c77d859]	2032	#Select parameters to fit for list of fittable parameters with dispersion
	2033	for item in self.fittable_param:
[edd166b]	2034	#Skip t ifhe angle parameters if 1D data
	2035	if self.data.__class__.__name__ !="Data2D":
	2036	if item in self.orientation_params:
	2037	continue
[998b6b8]	2038	if item[0].GetValue():
[c77d859]	2039	if not (item in self.param_toFit):
	2040	self.param_toFit.append(item)
	2041	else:
	2042	#remove parameters from the fitting list
	2043	if item in self.param_toFit:
[edd166b]	2044	self.param_toFit.remove(item)
	2045
	2046	#Calculate num. of angle parameters
	2047	if self.data.__class__.__name__ =="Data2D":
	2048	len_orient_para = 0
	2049	else:
	2050	len_orient_para = len(self.orientation_params) #assume even len
	2051	#Total num. of angle parameters
	2052	if len(self.fittable_param) > 0:
	2053	len_orient_para *= 2
[c77d859]	2054	#Set the value of checkbox that selected every checkbox or not
[edd166b]	2055	if len(self.parameters)+len(self.fittable_param)-len_orient_para ==len(self.param_toFit):
[c77d859]	2056	self.cb1.SetValue(True)
	2057	else:
	2058	self.cb1.SetValue(False)
[52cac46]	2059	self.save_current_state_fit()
[3595309d]	2060	if event !=None:
[3a37fe0]	2061	#self._undo.Enable(True)
[3595309d]	2062	## post state to fit panel
	2063	event = PageInfoEvent(page = self)
	2064	wx.PostEvent(self.parent, event)
	2065
[c77d859]	2066
	2067
[77e23a2]	2068	def set_model_param_sizer(self, model):
[c77d859]	2069	"""
	2070	Build the panel from the model content
	2071	@param model: the model selected in combo box for fitting purpose
	2072	"""
	2073	self.sizer3.Clear(True)
	2074	self.parameters = []
	2075	self.param_toFit=[]
	2076	self.fittable_param=[]
	2077	self.fixed_param=[]
[780d095]	2078	self.orientation_params=[]
[60132ef]	2079	self.orientation_params_disp=[]
[c77d859]	2080
	2081	if model ==None:
	2082	self.sizer3.Layout()
[08ba57d]	2083	self.SetScrollbars(20,20,25,65)
[c77d859]	2084	return
[707436d]	2085	## the panel is drawn using the current value of the fit engine
	2086	if self.engine_type==None and self.manager !=None:
	2087	self.engine_type= self.manager._return_engine_type()
[c99a6c5]	2088
[707436d]	2089
[c77d859]	2090	box_description= wx.StaticBox(self, -1,str("Model Parameters"))
	2091	boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
	2092	sizer = wx.GridBagSizer(5,5)
	2093	## save the current model
	2094	self.model = model
	2095
	2096	keys = self.model.getParamList()
	2097	#list of dispersion paramaters
	2098	self.disp_list=self.model.getDispParamList()
[b421b1a]	2099
[c77d859]	2100	keys.sort()
	2101
[6dc9ad8]	2102	iy = 0
[c77d859]	2103	ix = 0
[2f6c260]	2104	select_text = "Select All"
[b324aa9]	2105	self.cb1 = wx.CheckBox(self, -1,str(select_text), (10, 10))
[c77d859]	2106	wx.EVT_CHECKBOX(self, self.cb1.GetId(), self.select_all_param)
[2f6c260]	2107	self.cb1.SetToolTipString("To check/uncheck all the boxes below.")
	2108	#self.cb1.SetValue(True)
[c77d859]	2109
	2110	sizer.Add(self.cb1,(iy, ix),(1,1),\
[eda428b]	2111	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 5)
[c77d859]	2112	ix +=1
	2113	self.text2_2 = wx.StaticText(self, -1, 'Values')
	2114	sizer.Add(self.text2_2,(iy, ix),(1,1),\
	2115	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	2116	ix +=2
	2117	self.text2_3 = wx.StaticText(self, -1, 'Errors')
	2118	sizer.Add(self.text2_3,(iy, ix),(1,1),\
	2119	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	2120	self.text2_3.Hide()
	2121	ix +=1
	2122	self.text2_min = wx.StaticText(self, -1, 'Min')
	2123	sizer.Add(self.text2_min,(iy, ix),(1,1),\
	2124	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	2125	self.text2_min.Hide()
	2126	ix +=1
	2127	self.text2_max = wx.StaticText(self, -1, 'Max')
	2128	sizer.Add(self.text2_max,(iy, ix),(1,1),\
	2129	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	2130	self.text2_max.Hide()
[6fdfc8f]	2131	ix += 1
	2132	self.text2_4 = wx.StaticText(self, -1, '[Units]')
	2133	sizer.Add(self.text2_4,(iy, ix),(1,1),\
	2134	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	2135	self.text2_4.Hide()
[dcf29d7]	2136	if self.engine_type=="park":
	2137	self.text2_max.Show(True)
	2138	self.text2_min.Show(True)
[b421b1a]	2139
[c77d859]	2140	for item in keys:
[780d095]	2141	if not item in self.disp_list and not item in self.model.orientation_params:
[b421b1a]	2142
	2143	##prepare a spot to store errors
	2144	if not self.model.details.has_key(item):
	2145	self.model.details [item] = ["",None,None]
	2146
[c77d859]	2147	iy += 1
	2148	ix = 0
	2149	## add parameters name with checkbox for selecting to fit
[2f6c260]	2150	cb = wx.CheckBox(self, -1, item )
[479eced]	2151	cb.SetToolTipString(" Check for fitting.")
[2f6c260]	2152	#cb.SetValue(True)
[c77d859]	2153	wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
[2f6c260]	2154
[c77d859]	2155	sizer.Add( cb,( iy, ix),(1,1),
[eda428b]	2156	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 5)
[d1e0473]	2157
[c77d859]	2158	## add parameter value
	2159	ix += 1
	2160	value= self.model.getParam(item)
[7975f2b]	2161	ctl1 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH,20),
[c77d859]	2162	style=wx.TE_PROCESS_ENTER)
[1a94c36]	2163	ctl1.SetToolTipString("Hit 'Enter' after typing.")
[77e23a2]	2164	ctl1.SetValue(format_number(value))
[c77d859]	2165	sizer.Add(ctl1, (iy,ix),(1,1), wx.EXPAND)
	2166	## text to show error sign
	2167	ix += 1
	2168	text2=wx.StaticText(self, -1, '+/-')
	2169	sizer.Add(text2,(iy, ix),(1,1),\
	2170	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	2171	text2.Hide()
	2172	ix += 1
[c13b8cc]	2173	ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=0)
[c77d859]	2174	sizer.Add(ctl2, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[b421b1a]	2175	ctl2.Hide()
[c77d859]	2176
	2177	ix += 1
[7975f2b]	2178	ctl3 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
	2179	text_enter_callback = self._onparamRangeEnter)
[eacf1d66]	2180
[c77d859]	2181	sizer.Add(ctl3, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	2182	ctl3.Hide()
[77e23a2]	2183
[c77d859]	2184	ix += 1
[7975f2b]	2185	ctl4 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
	2186	text_enter_callback = self._onparamRangeEnter)
[c77d859]	2187	sizer.Add(ctl4, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[eacf1d66]	2188
[c77d859]	2189	ctl4.Hide()
[edd166b]	2190
[dcf29d7]	2191	if self.engine_type=="park":
	2192	ctl3.Show(True)
	2193	ctl4.Show(True)
[6fdfc8f]	2194	ix +=1
	2195	# Units
[7975f2b]	2196	if self.model.details.has_key(item):
[6fdfc8f]	2197	units = wx.StaticText(self, -1, self.model.details[item][0], style=wx.ALIGN_LEFT)
[7975f2b]	2198	else:
[6fdfc8f]	2199	units = wx.StaticText(self, -1, "", style=wx.ALIGN_LEFT)
	2200	sizer.Add(units, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[dcf29d7]	2201
[c77d859]	2202	##[cb state, name, value, "+/-", error of fit, min, max , units]
	2203	self.parameters.append([cb,item, ctl1,
[240b9966]	2204	text2,ctl2, ctl3, ctl4,units])
[2d5f7a1]	2205
[c77d859]	2206	iy+=1
[2a5bba7]	2207	sizer.Add((10,10),(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
[5af5183]	2208
[f20767b]	2209	# type can be either Guassian or Array
[5b53a5c]	2210	if len(self.model.dispersion.values())>0:
	2211	type= self.model.dispersion.values()[0]["type"]
	2212	else:
	2213	type = "Gaussian"
[6dc9ad8]	2214
	2215	iy += 1
	2216	ix = 0
	2217	#Add tile for orientational angle
	2218	for item in keys:
	2219	if item in self.model.orientation_params:
	2220	orient_angle = wx.StaticText(self, -1, '[For 2D only]:')
	2221	sizer.Add(orient_angle,(iy, ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	2222	if not self.data.__class__.__name__ =="Data2D":
	2223	orient_angle.Hide()
	2224	else:
	2225	orient_angle.Show(True)
	2226	break
[5b53a5c]	2227
[f20767b]	2228	#For Gaussian only
	2229	if type.lower() != "array":
	2230	for item in self.model.orientation_params:
	2231	if not item in self.disp_list:
[c13b8cc]	2232	##prepare a spot to store min max
	2233	if not self.model.details.has_key(item):
	2234	self.model.details [item] = ["",None,None]
	2235
[f20767b]	2236	iy += 1
	2237	ix = 0
	2238	## add parameters name with checkbox for selecting to fit
	2239	cb = wx.CheckBox(self, -1, item )
	2240	cb.SetValue(False)
[b9405cd]	2241	cb.SetToolTipString("Check for fitting")
[f20767b]	2242	wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
	2243	if self.data.__class__.__name__ =="Data2D":
[6dc9ad8]	2244	cb.Show(True)
[f20767b]	2245	else:
[6dc9ad8]	2246	cb.Hide()
[f20767b]	2247	sizer.Add( cb,( iy, ix),(1,1),
	2248	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 5)
	2249
	2250	## add parameter value
	2251	ix += 1
	2252	value= self.model.getParam(item)
[7975f2b]	2253	ctl1 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH,20),
[f20767b]	2254	style=wx.TE_PROCESS_ENTER)
[1a94c36]	2255	ctl1.SetToolTipString("Hit 'Enter' after typing.")
[f20767b]	2256	ctl1.SetValue(format_number(value))
	2257	if self.data.__class__.__name__ =="Data2D":
[c5cd3b9]	2258	ctl1.Show(True)
[f20767b]	2259	else:
[c5cd3b9]	2260	ctl1.Hide()
[f20767b]	2261	sizer.Add(ctl1, (iy,ix),(1,1), wx.EXPAND)
	2262	## text to show error sign
	2263	ix += 1
	2264	text2=wx.StaticText(self, -1, '+/-')
	2265	sizer.Add(text2,(iy, ix),(1,1),\
	2266	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	2267	text2.Hide()
	2268	ix += 1
[c13b8cc]	2269	ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=0)
[f20767b]	2270	sizer.Add(ctl2, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	2271	ctl2.Hide()
[e2f7b92]	2272
[c99a6c5]	2273
[f20767b]	2274	ix += 1
[7975f2b]	2275	ctl3 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
	2276	text_enter_callback = self._onparamRangeEnter)
[b9405cd]	2277
[f20767b]	2278	sizer.Add(ctl3, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	2279	ctl3.Hide()
[b324aa9]	2280
[f20767b]	2281	ix += 1
[7975f2b]	2282	ctl4 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
	2283	text_enter_callback = self._onparamRangeEnter)
[f20767b]	2284	sizer.Add(ctl4, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[eacf1d66]	2285
[f20767b]	2286	ctl4.Hide()
[edd166b]	2287
	2288	if self.engine_type =="park" and self.data.__class__.__name__ =="Data2D":
[f20767b]	2289	ctl3.Show(True)
	2290	ctl4.Show(True)
[c5cd3b9]	2291
[f20767b]	2292	ix +=1
	2293	# Units
[7975f2b]	2294	if self.model.details.has_key(item):
[f20767b]	2295	units = wx.StaticText(self, -1, self.model.details[item][0], style=wx.ALIGN_LEFT)
[7975f2b]	2296	else:
[f20767b]	2297	units = wx.StaticText(self, -1, "", style=wx.ALIGN_LEFT)
	2298	if self.data.__class__.__name__ =="Data2D":
[6dc9ad8]	2299	units.Show(True)
[c99a6c5]	2300
[f20767b]	2301	else:
[6dc9ad8]	2302	units.Hide()
[c99a6c5]	2303
[f20767b]	2304	sizer.Add(units, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[7975f2b]	2305
[f20767b]	2306	##[cb state, name, value, "+/-", error of fit, min, max , units]
	2307	self.parameters.append([cb,item, ctl1,
	2308	text2,ctl2, ctl3, ctl4,units])
	2309	self.orientation_params.append([cb,item, ctl1,
	2310	text2,ctl2, ctl3, ctl4,units])
[780d095]	2311
	2312	iy+=1
[c77d859]	2313
	2314	#Display units text on panel
	2315	for item in keys:
[7975f2b]	2316	if self.model.details.has_key(item):
[c77d859]	2317	self.text2_4.Show()
[b324aa9]	2318	#Fill the list of fittable parameters
	2319	self.select_all_param(event=None)
[7975f2b]	2320
[52cac46]	2321	self.save_current_state_fit()
[240b9966]	2322	boxsizer1.Add(sizer)
[c77d859]	2323	self.sizer3.Add(boxsizer1,0, wx.EXPAND \| wx.ALL, 10)
	2324	self.sizer3.Layout()
[330573d]	2325	self.Layout()
	2326	self.Refresh()
[08ba57d]	2327	self.SetScrollbars(20,20,25,65)
[52cac46]	2328
[7609f1a]	2329	class BGTextCtrl(wx.TextCtrl):
	2330	"""
	2331	Text control used to display outputs.
	2332	No editing allowed. The background is
	2333	grayed out. User can't select text.
	2334	"""
	2335	def __init__(self, args, *kwds):
	2336	wx.TextCtrl.__init__(self, args, *kwds)
	2337	self.SetEditable(False)
	2338	self.SetBackgroundColour(self.GetParent().GetBackgroundColour())
	2339
	2340	# Bind to mouse event to avoid text highlighting
	2341	# The event will be skipped once the call-back
	2342	# is called.
	2343	self.Bind(wx.EVT_MOUSE_EVENTS, self._click)
	2344
	2345	def _click(self, event):
	2346	"""
	2347	Prevent further handling of the mouse event
	2348	by not calling Skip().
	2349	"""
	2350	pass
[c77d859]	2351
	2352	class HelpWindow(wx.Frame):
	2353	def __init__(self, parent, id, title):
	2354	wx.Frame.__init__(self, parent, id, title, size=(570, 400))
	2355
	2356	from sans.models.CylinderModel import CylinderModel
	2357	model = CylinderModel()
[2d5f7a1]	2358
[c77d859]	2359	from danse.common.plottools.plottables import Data1D
	2360	data= Data1D(x=[1,2], y=[3,4], dy=[0.1, 0,1])
	2361
[cfc0913]	2362	from fitpanel import PageInfo
[c77d859]	2363	myinfo = PageInfo(self, model, data=data )
	2364
	2365	## add data
	2366
	2367	from models import ModelList
	2368	mylist= ModelList()
	2369
	2370	from sans.models.SphereModel import SphereModel
	2371	from sans.models.SquareWellStructure import SquareWellStructure
	2372	from sans.models.DebyeModel import DebyeModel
	2373	from sans.models.LineModel import LineModel
	2374	name= "shapes"
	2375	list1= [SphereModel]
	2376	mylist.set_list( name, list1)
	2377
	2378	name= "Shape-independent"
	2379	list1= [DebyeModel]
	2380	mylist.set_list( name, list1)
	2381
	2382	name= "Structure Factors"
	2383	list1= [SquareWellStructure]
	2384	mylist.set_list( name, list1)
	2385
	2386	name= "Added models"
	2387	list1= [LineModel]
	2388	mylist.set_list( name, list1)
	2389
	2390	myinfo.model_list_box = mylist.get_list()
	2391
	2392	self.page = FitPage(self, myinfo)
	2393
	2394	self.Centre()
	2395	self.Show(True)
[eacf1d66]	2396
[c77d859]	2397	if __name__=="__main__":
	2398	app = wx.App()
	2399	HelpWindow(None, -1, 'HelpWindow')
	2400	app.MainLoop()
	2401

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

SasView

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

Download in other formats: