batchfitpage.py @ c4f79f0

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 c4f79f0 was 940aca7, checked in by Mathieu Doucet <doucetm@…>, 13 years ago
Merge 2.1.1 into trunk
Property mode set to `100644`
File size: 30.0 KB

Rev	Line
[cc31608]	1
	2
	3	import sys
	4	import wx
	5	import wx.lib.newevent
	6	import numpy
	7	import copy
	8	import math
	9	import time
	10	from sans.models.dispersion_models import ArrayDispersion, GaussianDispersion
[c21e87b]	11	from sans.dataloader.data_info import Data1D
[cc31608]	12	from sans.guiframe.events import StatusEvent
	13	from sans.guiframe.events import NewPlotEvent
	14	from sans.guiframe.utils import format_number,check_float
	15
	16	(Chi2UpdateEvent, EVT_CHI2_UPDATE) = wx.lib.newevent.NewEvent()
	17	_BOX_WIDTH = 76
	18	_DATA_BOX_WIDTH = 300
	19	SMEAR_SIZE_L = 0.00
	20	SMEAR_SIZE_H = 0.00
	21
	22	import basepage
	23	from basepage import BasicPage
	24	from basepage import PageInfoEvent
	25	from sans.models.qsmearing import smear_selection
	26	from fitpage import FitPage
	27
	28	class BatchFitPage(FitPage):
	29	window_name = "BatchFit"
	30	window_caption = "BatchFit"
	31
	32	def __init__(self,parent, color='rand'):
	33	"""
	34	Initialization of the Panel
	35	"""
	36	FitPage.__init__(self, parent, color=color)
	37	self.window_name = "BatchFit"
	38	self.window_caption = "BatchFit"
[978feaa]	39	self._set_save_flag(False)
[940aca7]	40	self._set_bookmark_flag(False)
[cc31608]	41
	42	def _fill_range_sizer(self):
	43	"""
	44	Fill the sizer containing the plotting range
	45	add access to npts
	46	"""
	47	is_2Ddata = False
	48
	49	# Check if data is 2D
	50	if self.data.__class__.__name__ == "Data2D" or \
	51	self.enable2D:
	52	is_2Ddata = True
	53
	54	title = "Fitting"
	55	#smear messages & titles
	56	smear_message_none = "No smearing is selected..."
	57	smear_message_dqdata = "The dQ data is being used for smearing..."
	58	smear_message_2d = "Higher accuracy is very time-expensive. Use it with care..."
	59	smear_message_new_ssmear = "Please enter only the value of interest to customize smearing..."
	60	smear_message_new_psmear = "Please enter both; the dQ will be generated by interpolation..."
	61	smear_message_2d_x_title = "<dQp>[1/A]:"
	62	smear_message_2d_y_title = "<dQs>[1/A]:"
	63	smear_message_pinhole_min_title = "dQ_low[1/A]:"
	64	smear_message_pinhole_max_title = "dQ_high[1/A]:"
	65	smear_message_slit_height_title = "Slit height[1/A]:"
	66	smear_message_slit_width_title = "Slit width[1/A]:"
	67
	68	self._get_smear_info()
	69
	70	#Sizers
	71	box_description_range = wx.StaticBox(self, -1,str(title))
	72	boxsizer_range = wx.StaticBoxSizer(box_description_range, wx.VERTICAL)
	73	self.sizer_set_smearer = wx.BoxSizer(wx.VERTICAL)
	74	sizer_smearer = wx.BoxSizer(wx.HORIZONTAL)
	75	self.sizer_new_smear= wx.BoxSizer(wx.HORIZONTAL)
	76	self.sizer_set_masking = wx.BoxSizer(wx.HORIZONTAL)
	77	sizer_chi2 = wx.BoxSizer(wx.VERTICAL)
	78	"""
	79	smear_set_box= wx.StaticBox(self, -1,'Set Instrumental Smearing')
	80	sizer_smearer_box = wx.StaticBoxSizer(smear_set_box, wx.HORIZONTAL)
	81	sizer_smearer_box.SetMinSize((_DATA_BOX_WIDTH,85))
	82	"""
	83	sizer_fit = wx.GridSizer(2, 4, 2, 6)
	84	"""
	85	# combobox for smear2d accuracy selection
	86	self.smear_accuracy = wx.ComboBox(self, -1,size=(50,-1),style=wx.CB_READONLY)
	87	self._set_accuracy_list()
	88	self.smear_accuracy.SetValue(self.smear2d_accuracy)
	89	self.smear_accuracy.SetSelection(0)
	90	self.smear_accuracy.SetToolTipString("'Higher' uses more Gaussian points for smearing computation.")
	91
	92	wx.EVT_COMBOBOX(self.smear_accuracy,-1, self._on_select_accuracy)
	93	"""
	94	#Fit button
	95	self.btFit = wx.Button(self,wx.NewId(),'Fit', size=(88,25))
	96	self.default_bt_colour = self.btFit.GetDefaultAttributes()
	97	self.btFit.Bind(wx.EVT_BUTTON, self._onFit,id= self.btFit.GetId())
	98	self.btFit.SetToolTipString("Start fitting.")
	99	"""
	100	#textcntrl for custom resolution
	101	self.smear_pinhole_max = self.ModelTextCtrl(self, -1,size=(_BOX_WIDTH-25,20),style=wx.TE_PROCESS_ENTER,
	102	text_enter_callback = self.onPinholeSmear)
	103	self.smear_pinhole_min = self.ModelTextCtrl(self, -1,size=(_BOX_WIDTH-25,20),style=wx.TE_PROCESS_ENTER,
	104	text_enter_callback = self.onPinholeSmear)
	105	self.smear_slit_height= self.ModelTextCtrl(self, -1,size=(_BOX_WIDTH-25,20),style=wx.TE_PROCESS_ENTER,
	106	text_enter_callback = self.onSlitSmear)
	107	self.smear_slit_width = self.ModelTextCtrl(self, -1,size=(_BOX_WIDTH-25,20),style=wx.TE_PROCESS_ENTER,
	108	text_enter_callback = self.onSlitSmear)
	109
	110	## smear
	111	self.smear_data_left= BGTextCtrl(self, -1, size=(_BOX_WIDTH-25,20), style=0)
	112	self.smear_data_left.SetValue(str(self.dq_l))
	113	self.smear_data_right = BGTextCtrl(self, -1, size=(_BOX_WIDTH-25,20), style=0)
	114	self.smear_data_right.SetValue(str(self.dq_r))
	115
	116	#set default values for smear
	117	self.smear_pinhole_max.SetValue(str(self.dx_max))
	118	self.smear_pinhole_min.SetValue(str(self.dx_min))
	119	self.smear_slit_height.SetValue(str(self.dxl))
	120	self.smear_slit_width.SetValue(str(self.dxw))
	121
	122	#Filling the sizer containing instruments smearing info.
	123	self.disable_smearer = wx.RadioButton(self, -1, 'None', style=wx.RB_GROUP)
	124	self.enable_smearer = wx.RadioButton(self, -1, 'Use dQ Data')
	125	#self.enable_smearer.SetToolTipString("Click to use the loaded dQ data for smearing.")
	126	self.pinhole_smearer = wx.RadioButton(self, -1, 'Custom Pinhole Smear')
	127	#self.pinhole_smearer.SetToolTipString("Click to input custom resolution for pinhole smearing.")
	128	self.slit_smearer = wx.RadioButton(self, -1, 'Custom Slit Smear')
	129	#self.slit_smearer.SetToolTipString("Click to input custom resolution for slit smearing.")
	130	self.Bind(wx.EVT_RADIOBUTTON, self.onSmear, id=self.disable_smearer.GetId())
	131	self.Bind(wx.EVT_RADIOBUTTON, self.onSmear, id=self.enable_smearer.GetId())
	132	self.Bind(wx.EVT_RADIOBUTTON, self.onPinholeSmear, id=self.pinhole_smearer.GetId())
	133	self.Bind(wx.EVT_RADIOBUTTON, self.onSlitSmear, id=self.slit_smearer.GetId())
	134	self.disable_smearer.SetValue(True)
	135
	136	# add 4 types of smearing to the sizer
	137	sizer_smearer.Add( self.disable_smearer,0, wx.LEFT, 10)
	138	sizer_smearer.Add((10,10))
	139	sizer_smearer.Add( self.enable_smearer)
	140	sizer_smearer.Add((10,10))
	141	sizer_smearer.Add( self.pinhole_smearer )
	142	sizer_smearer.Add((10,10))
	143	sizer_smearer.Add( self.slit_smearer )
	144	sizer_smearer.Add((10,10))
	145	"""
[3f36675]	146	"""
[cc31608]	147	# StaticText for chi2, N(for fitting), Npts
	148	self.tcChi = BGTextCtrl(self, -1, "-", size=(75,20), style=0)
	149	self.tcChi.SetToolTipString("Chi2/Npts(Fit)")
	150	self.Npts_fit = BGTextCtrl(self, -1, "-", size=(75,20), style=0)
	151	self.Npts_fit.SetToolTipString(\
	152	" Npts : number of points selected for fitting")
	153	self.Npts_total = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH, 20),
	154	style=wx.TE_PROCESS_ENTER,
	155	text_enter_callback=self._onQrangeEnter)
	156	self.Npts_total.SetValue(format_number(self.npts_x))
	157	self.Npts_total.SetToolTipString(\
	158	" Total Npts : total number of data points")
[3f36675]	159	"""
[cc31608]	160	# Update and Draw button
	161	self.draw_button = wx.Button(self,wx.NewId(),'Compute', size=(88,24))
	162	self.draw_button.Bind(wx.EVT_BUTTON, \
	163	self._onDraw,id= self.draw_button.GetId())
	164	self.draw_button.SetToolTipString("Compute and Draw.")
[3f36675]	165	"""
[cc31608]	166	box_description_1= wx.StaticText(self, -1,' Chi2/Npts')
	167	box_description_2= wx.StaticText(self, -1,'Npts(Fit)')
	168	box_description_3= wx.StaticText(self, -1,'Total Npts')
	169	box_description_3.SetToolTipString( \
	170	" Total Npts : total number of data points")
	171	#box_description_4= wx.StaticText(self, -1,' ')
[3f36675]	172	"""
[cc31608]	173
[3f36675]	174	"""
[cc31608]	175	sizer_fit.Add(box_description_1,0,0)
	176	sizer_fit.Add(box_description_2,0,0)
[3f36675]	177	sizer_fit.Add(box_description_3,0,0)
	178	"""
[cc31608]	179	sizer_fit.Add(self.draw_button,0,0)
[3f36675]	180	"""
[cc31608]	181	sizer_fit.Add(self.tcChi,0,0)
	182	sizer_fit.Add(self.Npts_fit ,0,0)
	183	sizer_fit.Add(self.Npts_total,0,0)
[3f36675]	184	"""
[cc31608]	185	sizer_fit.Add(self.btFit,0,0)
	186	"""
	187	self.smear_description_none = wx.StaticText(self, -1, smear_message_none , style=wx.ALIGN_LEFT)
	188	self.smear_description_dqdata = wx.StaticText(self, -1, smear_message_dqdata , style=wx.ALIGN_LEFT)
	189	self.smear_description_type = wx.StaticText(self, -1, "Type:" , style=wx.ALIGN_LEFT)
	190	self.smear_description_accuracy_type = wx.StaticText(self, -1, "Accuracy:" , style=wx.ALIGN_LEFT)
	191	self.smear_description_smear_type = BGTextCtrl(self, -1, size=(57,20), style=0)
	192	self.smear_description_smear_type.SetValue(str(self.dq_l))
	193	self.SetBackgroundColour(self.GetParent().GetBackgroundColour())
	194	self.smear_description_2d = wx.StaticText(self, -1, smear_message_2d , style=wx.ALIGN_LEFT)
	195	self.smear_message_new_s = wx.StaticText(self, -1, smear_message_new_ssmear , style=wx.ALIGN_LEFT)
	196	self.smear_message_new_p = wx.StaticText(self, -1, smear_message_new_psmear , style=wx.ALIGN_LEFT)
	197	self.smear_description_2d_x = wx.StaticText(self, -1, smear_message_2d_x_title , style=wx.ALIGN_LEFT)
	198	self.smear_description_2d_x.SetToolTipString(" dQp(parallel) in q_r direction.")
	199	self.smear_description_2d_y = wx.StaticText(self, -1, smear_message_2d_y_title , style=wx.ALIGN_LEFT)
	200	self.smear_description_2d_y.SetToolTipString(" dQs(perpendicular) in q_phi direction.")
	201	self.smear_description_pin_min = wx.StaticText(self, -1, smear_message_pinhole_min_title , style=wx.ALIGN_LEFT)
	202	self.smear_description_pin_max = wx.StaticText(self, -1, smear_message_pinhole_max_title , style=wx.ALIGN_LEFT)
	203	self.smear_description_slit_height = wx.StaticText(self, -1, smear_message_slit_height_title , style=wx.ALIGN_LEFT)
	204	self.smear_description_slit_width = wx.StaticText(self, -1, smear_message_slit_width_title , style=wx.ALIGN_LEFT)
	205	"""
	206	#arrange sizers
	207	#boxsizer1.Add( self.tcChi )
	208	"""
	209	self.sizer_set_smearer.Add(sizer_smearer )
	210	self.sizer_set_smearer.Add((10,10))
	211	self.sizer_set_smearer.Add( self.smear_description_none,0, wx.CENTER, 10 )
	212	self.sizer_set_smearer.Add( self.smear_description_dqdata,0, wx.CENTER, 10 )
	213	self.sizer_set_smearer.Add( self.smear_description_2d,0, wx.CENTER, 10 )
	214	self.sizer_new_smear.Add( self.smear_description_type,0, wx.CENTER, 10 )
	215	self.sizer_new_smear.Add( self.smear_description_accuracy_type,0, wx.CENTER, 10 )
	216	self.sizer_new_smear.Add( self.smear_accuracy )
	217	self.sizer_new_smear.Add( self.smear_description_smear_type,0, wx.CENTER, 10 )
	218	self.sizer_new_smear.Add((15,-1))
	219	self.sizer_new_smear.Add( self.smear_description_2d_x,0, wx.CENTER, 10 )
	220	self.sizer_new_smear.Add( self.smear_description_pin_min,0, wx.CENTER, 10 )
	221	self.sizer_new_smear.Add( self.smear_description_slit_height,0, wx.CENTER, 10 )
	222
	223	self.sizer_new_smear.Add( self.smear_pinhole_min,0, wx.CENTER, 10 )
	224	self.sizer_new_smear.Add( self.smear_slit_height,0, wx.CENTER, 10 )
	225	self.sizer_new_smear.Add( self.smear_data_left,0, wx.CENTER, 10 )
	226	self.sizer_new_smear.Add((20,-1))
	227	self.sizer_new_smear.Add( self.smear_description_2d_y,0, wx.CENTER, 10 )
	228	self.sizer_new_smear.Add( self.smear_description_pin_max,0, wx.CENTER, 10 )
	229	self.sizer_new_smear.Add( self.smear_description_slit_width,0, wx.CENTER, 10 )
	230
	231	self.sizer_new_smear.Add( self.smear_pinhole_max,0, wx.CENTER, 10 )
	232	self.sizer_new_smear.Add( self.smear_slit_width,0, wx.CENTER, 10 )
	233	self.sizer_new_smear.Add( self.smear_data_right,0, wx.CENTER, 10 )
	234
	235	self.sizer_set_smearer.Add( self.smear_message_new_s,0, wx.CENTER, 10)
	236	self.sizer_set_smearer.Add( self.smear_message_new_p,0, wx.CENTER, 10)
	237	self.sizer_set_smearer.Add((5,2))
	238	self.sizer_set_smearer.Add( self.sizer_new_smear,0, wx.CENTER, 10 )
	239
	240	# add all to chi2 sizer
	241	sizer_smearer_box.Add(self.sizer_set_smearer)
	242
	243	sizer_chi2.Add(sizer_smearer_box)
	244	"""
	245	sizer_chi2.Add((-1,5))
	246	"""
	247	# hide all smear messages and textctrl
	248	self._hide_all_smear_info()
	249	"""
	250	# get smear_selection
	251	self.current_smearer= smear_selection( self.data, self.model )
	252	"""
	253	# Show only the relevant smear messages, etc
	254	if self.current_smearer == None:
	255	if not is_2Ddata:
	256	self.smear_description_none.Show(True)
	257	self.enable_smearer.Disable()
	258	else:
	259	self.smear_description_none.Show(True)
	260	#self.smear_description_2d.Show(True)
	261	#self.pinhole_smearer.Disable()
	262	self.slit_smearer.Disable()
	263	#self.enable_smearer.Disable()
	264	if self.data == None:
	265	self.slit_smearer.Disable()
	266	self.pinhole_smearer.Disable()
	267	self.enable_smearer.Disable()
	268	else: self._show_smear_sizer()
	269	"""
	270	boxsizer_range.Add(self.sizer_set_masking)
	271	#2D data? default
	272	is_2Ddata = False
	273
	274	#check if it is 2D data
	275	if self.data.__class__.__name__ == "Data2D" or \
	276	self.enable2D:
	277	is_2Ddata = True
	278
	279	self.sizer5.Clear(True)
	280
	281	self.qmin = self.ModelTextCtrl(self, -1,size=(_BOX_WIDTH,20),
	282	style=wx.TE_PROCESS_ENTER,
	283	text_enter_callback = self._onQrangeEnter)
	284	self.qmin.SetValue(str(self.qmin_x))
	285	self.qmin.SetToolTipString("Minimun value of Q in linear scale.")
	286
	287	self.qmax = self.ModelTextCtrl(self, -1,size=(_BOX_WIDTH,20),
	288	style=wx.TE_PROCESS_ENTER,
	289	text_enter_callback=self._onQrangeEnter)
	290	self.qmax.SetValue(str(self.qmax_x))
	291	self.qmax.SetToolTipString("Maximum value of Q in linear scale.")
	292
	293	id = wx.NewId()
	294	self.reset_qrange =wx.Button(self,id,'Reset',size=(77,20))
	295
	296	self.reset_qrange.Bind(wx.EVT_BUTTON, self.on_reset_clicked,id=id)
	297	self.reset_qrange.SetToolTipString("Reset Q range to the default values")
	298
	299	sizer_horizontal=wx.BoxSizer(wx.HORIZONTAL)
	300	sizer= wx.GridSizer(2, 4, 2, 6)
	301
	302	self.btEditMask = wx.Button(self,wx.NewId(),'Editor', size=(88,23))
	303	self.btEditMask.Bind(wx.EVT_BUTTON, self._onMask,id= self.btEditMask.GetId())
	304	self.btEditMask.SetToolTipString("Edit Mask.")
	305	self.EditMask_title = wx.StaticText(self, -1, ' Masking(2D)')
	306
	307	sizer.Add(wx.StaticText(self, -1, 'Q range'))
	308	sizer.Add(wx.StaticText(self, -1, ' Min[1/A]'))
	309	sizer.Add(wx.StaticText(self, -1, ' Max[1/A]'))
	310	sizer.Add(self.EditMask_title)
	311	#sizer.Add(wx.StaticText(self, -1, ''))
	312	sizer.Add(self.reset_qrange)
	313	sizer.Add(self.qmin)
	314	sizer.Add(self.qmax)
	315	#sizer.Add(self.theory_npts_tcrtl)
	316	sizer.Add(self.btEditMask)
	317	boxsizer_range.Add(sizer_chi2)
	318	boxsizer_range.Add((10,10))
	319	boxsizer_range.Add(sizer)
	320
	321	boxsizer_range.Add((10,15))
	322	boxsizer_range.Add(sizer_fit)
	323	if is_2Ddata:
	324	self.btEditMask.Enable()
	325	self.EditMask_title.Enable()
	326	else:
	327	self.btEditMask.Disable()
	328	self.EditMask_title.Disable()
	329	"""
	330	## save state
	331	self.save_current_state()
	332	"""
	333	self.sizer5.Add(boxsizer_range,0, wx.EXPAND \| wx.ALL, 10)
	334	self.sizer5.Layout()
	335
	336	def _on_select_model(self, event=None):
	337	"""
	338	call back for model selection
	339	"""
	340
	341	self.Show(False)
	342	self._on_select_model_helper()
	343	self.set_model_param_sizer(self.model)
	344	if self.model is None:
	345	self._set_bookmark_flag(False)
	346	self._keep.Enable(False)
	347	self._set_save_flag(False)
	348	self.enable_disp.SetValue(False)
	349	self.disable_disp.SetValue(True)
	350	try:
	351	self.set_dispers_sizer()
	352	except:
	353	pass
	354	"""
	355	#self.btFit.SetFocus()
	356	self.state.enable_disp = self.enable_disp.GetValue()
	357	self.state.disable_disp = self.disable_disp.GetValue()
	358	self.state.pinhole_smearer = self.pinhole_smearer.GetValue()
	359	self.state.slit_smearer = self.slit_smearer.GetValue()
	360	"""
	361	self.state.structurecombobox = self.structurebox.GetCurrentSelection()
	362	self.state.formfactorcombobox = self.formfactorbox.GetCurrentSelection()
	363
	364	if self.model != None:
	365	self._set_copy_flag(True)
	366	self._set_paste_flag(True)
	367	if self.data != None:
[940aca7]	368	self._set_bookmark_flag(False)
	369	self._keep.Enable(False)
[cc31608]	370
	371	temp_smear = None
	372	"""
	373	#self._set_save_flag(True)
	374	# Reset smearer, model and data
	375	self._set_smear(self.data)
	376	try:
	377	# update smearer sizer
	378	self.onSmear(None)
	379	temp_smear = None
	380	if self.enable_smearer.GetValue():
	381	# Set the smearer environments
	382	temp_smear = self.smearer
	383	except:
	384	raise
	385	## error occured on chisqr computation
	386	#pass
	387	"""
	388	## event to post model to fit to fitting plugins
	389	(ModelEventbox, EVT_MODEL_BOX) = wx.lib.newevent.NewEvent()
	390
	391	## set smearing value whether or not
	392	# the data contain the smearing info
	393	evt = ModelEventbox(model=self.model,
	394	smearer=temp_smear,
	395	qmin=float(self.qmin_x),
	396	uid=self.uid,
	397	qmax=float(self.qmax_x))
	398
	399	self._manager._on_model_panel(evt=evt)
	400	self.mbox_description.SetLabel("Model [%s]" % str(self.model.name))
	401	self.state.model = self.model.clone()
	402	self.state.model.name = self.model.name
	403
	404
	405	if event != None:
	406	## post state to fit panel
	407	new_event = PageInfoEvent(page = self)
	408	wx.PostEvent(self.parent, new_event)
	409	#update list of plugins if new plugin is available
	410	if self.plugin_rbutton.GetValue():
	411	temp = self.parent.update_model_list()
	412	if temp:
	413	self.model_list_box = temp
	414	current_val = self.formfactorbox.GetValue()
	415	pos = self.formfactorbox.GetSelection()
	416	self._show_combox_helper()
	417	self.formfactorbox.SetSelection(pos)
	418	self.formfactorbox.SetValue(current_val)
	419	self._onDraw(event=None)
	420	else:
	421	self._draw_model()
	422	self.SetupScrolling()
	423	self.Show(True)
	424
	425	def _update_paramv_on_fit(self):
	426	"""
	427	make sure that update param values just before the fitting
	428	"""
	429	#flag for qmin qmax check values
	430	flag = True
	431	self.fitrange = True
	432	is_modified = False
	433
	434	#wx.PostEvent(self._manager.parent, StatusEvent(status=" \
	435	#updating ... ",type="update"))
	436
	437	##So make sure that update param values on_Fit.
	438	#self._undo.Enable(True)
	439	if self.model !=None:
	440	##Check the values
	441	self._check_value_enter( self.fittable_param ,is_modified)
	442	self._check_value_enter( self.fixed_param ,is_modified)
	443	self._check_value_enter( self.parameters ,is_modified)
	444
	445	# If qmin and qmax have been modified, update qmin and qmax and
	446	# Here we should check whether the boundaries have been modified.
	447	# If qmin and qmax have been modified, update qmin and qmax and
	448	# set the is_modified flag to True
	449	self.fitrange = self._validate_qrange(self.qmin, self.qmax)
	450	if self.fitrange:
	451	tempmin = float(self.qmin.GetValue())
	452	if tempmin != self.qmin_x:
	453	self.qmin_x = tempmin
	454	tempmax = float(self.qmax.GetValue())
	455	if tempmax != self.qmax_x:
	456	self.qmax_x = tempmax
	457	if tempmax == tempmin:
	458	flag = False
	459	temp_smearer = None
	460	"""
	461	if not self.disable_smearer.GetValue():
	462	temp_smearer= self.current_smearer
	463	if self.slit_smearer.GetValue():
	464	flag = self.update_slit_smear()
	465	elif self.pinhole_smearer.GetValue():
	466	flag = self.update_pinhole_smear()
	467	else:
	468	self._manager.set_smearer(smearer=temp_smearer,
	469	uid=self.uid,
	470	qmin=float(self.qmin_x),
	471	qmax=float(self.qmax_x),
	472	draw=False)
	473	elif not self._is_2D():
	474	self._manager.set_smearer(smearer=temp_smearer,
	475	qmin=float(self.qmin_x),
	476	uid=self.uid,
	477	qmax= float(self.qmax_x),
	478	draw=False)
	479
	480	if self.data != None:
	481	index_data = ((self.qmin_x <= self.data.x)&\
	482	(self.data.x <= self.qmax_x))
	483	val = str(len(self.data.x[index_data==True]))
	484	self.Npts_fit.SetValue(val)
	485	else:
	486	# No data in the panel
	487	try:
	488	self.npts_x = float(self.Npts_total.GetValue())
	489	except:
	490	flag = False
	491	return flag
	492	flag = True
	493	"""
	494	if self._is_2D():
	495	# only 2D case set mask
	496	flag = self._validate_Npts()
	497	if not flag:
	498	return flag
	499	else: flag = False
	500	else:
	501	flag = False
	502
	503	#For invalid q range, disable the mask editor and fit button, vs.
	504	if not self.fitrange:
	505	#self.btFit.Disable()
	506	if self._is_2D():
	507	self.btEditMask.Disable()
	508	else:
	509	#self.btFit.Enable(True)
	510	if self._is_2D() and self.data != None:
	511	self.btEditMask.Enable(True)
	512
	513	if not flag:
	514	msg = "Cannot Plot or Fit :Must select a "
	515	msg += " model or Fitting range is not valid!!! "
	516	wx.PostEvent(self.parent.parent, StatusEvent(status=msg))
	517
	518	self.save_current_state()
	519
	520	return flag
	521	def save_current_state(self):
	522	"""
	523	Currently no save option implemented for batch page
	524	"""
	525	pass
	526	def save_current_state_fit(self):
	527	"""
	528	Currently no save option implemented for batch page
	529	"""
	530	pass
	531	def set_data(self, data):
	532	"""
	533	reset the current data
	534	"""
	535	id = None
	536	group_id = None
	537	flag = False
	538	if self.data is None and data is not None:
	539	flag = True
	540	if data is not None:
	541	id = data.id
	542	group_id = data.group_id
	543	if self.data is not None:
	544	flag = (data.id != self.data.id)
	545	self.data = data
	546	if self.data is None:
	547	data_min = ""
	548	data_max = ""
	549	data_name = ""
	550	self._set_bookmark_flag(False)
	551	self._keep.Enable(False)
	552	self._set_save_flag(False)
	553	else:
	554	if self.model != None:
[940aca7]	555	self._set_bookmark_flag(False)
	556	self._keep.Enable(False)
	557	self._set_save_flag(False)
[cc31608]	558	self._set_preview_flag(True)
	559	"""
	560	self._set_smear(data)
	561	# more disables for 2D
	562	if self.data.__class__.__name__ == "Data2D" or \
	563	self.enable2D:
	564	self.slit_smearer.Disable()
	565	self.pinhole_smearer.Enable(True)
	566	self.default_mask = copy.deepcopy(self.data.mask)
	567	else:
	568	self.slit_smearer.Enable(True)
	569	self.pinhole_smearer.Enable(True)
	570	"""
	571	self.formfactorbox.Enable()
	572	self.structurebox.Enable()
	573	data_name = self.data.name
	574	#set maximum range for x in linear scale
	575	if not hasattr(self.data,"data"): #Display only for 1D data fit
	576	# Minimum value of data
	577	data_min = min(self.data.x)
	578	# Maximum value of data
	579	data_max = max(self.data.x)
[3f36675]	580	"""
[cc31608]	581	#number of total data points
	582	self.Npts_total.SetValue(str(len(self.data.x)))
	583	#default:number of data points selected to fit
	584	self.Npts_fit.SetValue(str(len(self.data.x)))
[3f36675]	585	"""
[cc31608]	586	self.btEditMask.Disable()
	587	self.EditMask_title.Disable()
	588	else:
	589
	590	## Minimum value of data
	591	data_min = 0
	592	x = max(math.fabs(self.data.xmin), math.fabs(self.data.xmax))
	593	y = max(math.fabs(self.data.ymin), math.fabs(self.data.ymax))
	594	## Maximum value of data
	595	data_max = math.sqrt(xx + yy)
[3f36675]	596	"""
[cc31608]	597	#number of total data points
	598	self.Npts_total.SetValue(str(len(self.data.data)))
	599	#default:number of data points selected to fit
	600	self.Npts_fit.SetValue(str(len(self.data.data)))
[3f36675]	601	"""
[cc31608]	602	self.btEditMask.Enable()
	603	self.EditMask_title.Enable()
[3f36675]	604	"""
[cc31608]	605	self.Npts_total.SetEditable(False)
	606	self.Npts_total.SetBackgroundColour(\
	607	self.GetParent().GetBackgroundColour())
	608
	609	self.Npts_total.Bind(wx.EVT_MOUSE_EVENTS, self._npts_click)
	610	#self.Npts_total.Disable()
[3f36675]	611	"""
[cc31608]	612	self.dataSource.SetValue(data_name)
	613	self.qmin_x = data_min
	614	self.qmax_x = data_max
	615	#self.minimum_q.SetValue(str(data_min))
	616	#self.maximum_q.SetValue(str(data_max))
	617	self.qmin.SetValue(str(data_min))
	618	self.qmax.SetValue(str(data_max))
	619	self.qmin.SetBackgroundColour("white")
	620	self.qmax.SetBackgroundColour("white")
	621	self.state.data = data
	622	self.state.qmin = self.qmin_x
	623	self.state.qmax = self.qmax_x
	624
	625	#update model plot with new data information
	626	if flag:
	627	#set model view button
	628	if self.data.__class__.__name__ == "Data2D":
	629	self.enable2D = True
	630	self.model_view.SetLabel("2D Mode")
	631	else:
	632	self.enable2D = False
	633	self.model_view.SetLabel("1D Mode")
	634
	635	self.model_view.Disable()
	636
	637	#replace data plot on combo box selection
	638	#by removing the previous selected data
[3f36675]	639	#wx.PostEvent(self._manager.parent, NewPlotEvent(action="remove",
	640	# group_id=group_id, id=id))
	641	wx.PostEvent(self._manager.parent,
	642	NewPlotEvent(group_id=group_id,
	643	action="delete"))
[cc31608]	644	#plot the current selected data
	645	wx.PostEvent(self._manager.parent, NewPlotEvent(plot=self.data,
	646	title=str(self.data.title)))
	647	self._manager.store_data(uid=self.uid, data=data,
	648	data_list=self.data_list,
	649	caption=self.window_name)
	650	self._draw_model()
	651
	652
	653
	654	class BGTextCtrl(wx.TextCtrl):
	655	"""
	656	Text control used to display outputs.
	657	No editing allowed. The background is
	658	grayed out. User can't select text.
	659	"""
	660	def __init__(self, args, *kwds):
	661	wx.TextCtrl.__init__(self, args, *kwds)
	662	self.SetEditable(False)
	663	self.SetBackgroundColour(self.GetParent().parent.GetBackgroundColour())
	664
	665	# Bind to mouse event to avoid text highlighting
	666	# The event will be skipped once the call-back
	667	# is called.
	668	self.Bind(wx.EVT_MOUSE_EVENTS, self._click)
	669
	670	def _click(self, event):
	671	"""
	672	Prevent further handling of the mouse event
	673	by not calling Skip().
	674	"""
	675	pass
	676

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

SasView

source: sasview/fittingview/src/sans/perspectives/fitting/batchfitpage.py @ c4f79f0

Download in other formats: