batchfitpage.py @ f21d496

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 f21d496 was 373d4ee, checked in by Paul Kienzle <pkienzle@…>, 10 years ago
code cleanup for fit panel
Property mode set to `100644`
File size: 16.1 KB

Rev	Line
[2f4b430]	1	"""
[fa09d62]	2	Batch panel
	3	"""
	4	import wx
	5	import wx.lib.newevent
	6	import math
[2f4b430]	7	from sas.guiframe.events import StatusEvent
	8	from sas.guiframe.events import NewPlotEvent
[fa09d62]	9
[2f4b430]	10	(Chi2UpdateEvent, EVT_CHI2_UPDATE) = wx.lib.newevent.NewEvent()
[fa09d62]	11	_BOX_WIDTH = 76
	12	_DATA_BOX_WIDTH = 300
	13	SMEAR_SIZE_L = 0.00
	14	SMEAR_SIZE_H = 0.00
	15
	16	from sas.perspectives.fitting.basepage import PageInfoEvent
	17	from sas.models.qsmearing import smear_selection
	18	from sas.perspectives.fitting.fitpage import FitPage
	19	from sas.perspectives.fitting.fitpage import check_data_validity
	20
	21	class BatchFitPage(FitPage):
	22	"""
	23	Batch Page
	24	"""
	25	window_name = "BatchFit"
[2f4b430]	26	window_caption = "BatchFit"
[fa09d62]	27
	28	def __init__(self, parent, color=None):
[2f4b430]	29	"""
[fa09d62]	30	Initialization of the Panel
	31	"""
	32	FitPage.__init__(self, parent, color=color)
[2f4b430]	33
[fa09d62]	34	## draw sizer
[2f4b430]	35
[fa09d62]	36	def _fill_data_sizer(self):
	37	"""
	38	fill sizer 0 with data info
	39	"""
	40	self.data_box_description = wx.StaticBox(self, -1, 'I(q) Data Source')
	41	if check_data_validity(self.data):
	42	dname_color = wx.BLUE
	43	else:
	44	dname_color = wx.RED
	45	self.data_box_description.SetForegroundColour(dname_color)
	46	boxsizer1 = wx.StaticBoxSizer(self.data_box_description, wx.VERTICAL)
	47	#----------------------------------------------------------
	48	sizer_data = wx.BoxSizer(wx.VERTICAL)
	49	text1 = wx.StaticText(self, -1, ' - Choose a file to set initial fit parameters -')
	50	text1.SetForegroundColour(wx.RED)
	51	sizer_data.Add(text1)
	52	text2 = wx.StaticText(self, -1, ' - This panel is not designed to view individual fits. - ')
	53	text2.SetForegroundColour(wx.RED)
	54	sizer_data.Add(text2)
[2f4b430]	55
[fa09d62]	56	combo = wx.BoxSizer(wx.HORIZONTAL)
	57	self.dataSource = wx.ComboBox(self, -1, style=wx.CB_READONLY)
	58	wx.EVT_COMBOBOX(self.dataSource, -1, self.on_select_data)
	59	self.dataSource.SetMinSize((_DATA_BOX_WIDTH, -1))
[2f4b430]	60
[fa09d62]	61	combo.Add(wx.StaticText(self, -1, 'Name : '))
[2f4b430]	62	combo.Add((0, 5))
[fa09d62]	63	combo.Add(self.dataSource)
[2f4b430]	64
[fa09d62]	65	sizer_data.Add(combo, 0, wx.ALL, 10)
	66	boxsizer1.Add(sizer_data, 0, wx.ALL, 0)
	67	self.sizer0.Add(boxsizer1, 0, wx.EXPAND \| wx.ALL, 10)
	68	self.sizer0.Layout()
	69
	70	# COMMENTED OUT TO USE METHODS INHERITED FROM fitpage.py
[2f4b430]	71
[fa09d62]	72	# def _fill_range_sizer(self):
	73	# """
	74	# Fill the sizer containing the plotting range
	75	# add access to npts
	76	# """
	77	# is_2Ddata = False
	78	#
	79	# # Check if data is 2D
	80	# if self.data.__class__.__name__ == "Data2D" or \
	81	# self.enable2D:
	82	# is_2Ddata = True
	83	#
	84	# title = "Fitting"
	85	# self._get_smear_info()
	86	#
	87	# #Sizers
	88	# box_description_range = wx.StaticBox(self, -1, str(title))
	89	# boxsizer_range = wx.StaticBoxSizer(box_description_range, wx.VERTICAL)
	90	# self.sizer_set_smearer = wx.BoxSizer(wx.VERTICAL)
	91	# #sizer_smearer = wx.BoxSizer(wx.HORIZONTAL)
	92	# self.sizer_new_smear = wx.BoxSizer(wx.HORIZONTAL)
	93	# self.sizer_set_masking = wx.BoxSizer(wx.HORIZONTAL)
	94	# sizer_chi2 = wx.BoxSizer(wx.VERTICAL)
	95	#
	96	# sizer_fit = wx.GridSizer(2, 4, 2, 6)
	97	# #Fit button
	98	# self.btFit = wx.Button(self, wx.NewId(), 'Fit', size=(88, 25))
	99	# self.default_bt_colour = self.btFit.GetDefaultAttributes()
	100	# self.btFit.Bind(wx.EVT_BUTTON, self._onFit, id= self.btFit.GetId())
	101	# self.btFit.SetToolTipString("Start fitting.")
	102	#
	103	# # Update and Draw button
	104	# self.draw_button = wx.Button(self, wx.NewId(), 'Compute', size=(88, 24))
	105	# self.draw_button.Bind(wx.EVT_BUTTON, \
	106	# self._onDraw,id=self.draw_button.GetId())
	107	# self.draw_button.SetToolTipString("Compute and Draw.")
	108	# sizer_fit.Add(self.draw_button, 0, 0)
	109	# sizer_fit.Add(self.btFit, 0, 0)
	110	# sizer_chi2.Add((-1, 5))
	111	# # get smear_selection
	112	# self.current_smearer = smear_selection( self.data, self.model )
	113	# boxsizer_range.Add(self.sizer_set_masking)
	114	# #2D data? default
	115	# is_2Ddata = False
	116	#
	117	# #check if it is 2D data
	118	# if self.data.__class__.__name__ == "Data2D" or \
	119	# self.enable2D:
	120	# is_2Ddata = True
	121	#
	122	# self.sizer5.Clear(True)
	123	#
[373d4ee]	124	# self.qmin = ModelTextCtrl(self, -1, size=(_BOX_WIDTH, 20),
[fa09d62]	125	# style=wx.TE_PROCESS_ENTER,
	126	# text_enter_callback = self._onQrangeEnter)
	127	# self.qmin.SetValue(str(self.qmin_x))
	128	# self.qmin.SetToolTipString("Minimun value of Q in linear scale.")
	129	#
[373d4ee]	130	# self.qmax = ModelTextCtrl(self, -1, size=(_BOX_WIDTH, 20),
[fa09d62]	131	# style=wx.TE_PROCESS_ENTER,
	132	# text_enter_callback=self._onQrangeEnter)
	133	# self.qmax.SetValue(str(self.qmax_x))
	134	# self.qmax.SetToolTipString("Maximum value of Q in linear scale.")
	135	#
	136	# id = wx.NewId()
	137	# self.reset_qrange =wx.Button(self, id, 'Reset', size=(77, 20))
	138	#
	139	# self.reset_qrange.Bind(wx.EVT_BUTTON, self.on_reset_clicked, id=id)
	140	# self.reset_qrange.SetToolTipString(\
	141	# "Reset Q range to the default values")
	142	#
	143	# sizer_horizontal = wx.BoxSizer(wx.HORIZONTAL)
	144	# sizer = wx.GridSizer(2, 4, 2, 6)
	145	#
	146	# self.btEditMask = wx.Button(self, wx.NewId(),'Editor', size=(88, 23))
	147	# self.btEditMask.Bind(wx.EVT_BUTTON,
	148	# self._onMask,id=self.btEditMask.GetId())
	149	# self.btEditMask.SetToolTipString("Edit Mask.")
	150	# self.EditMask_title = wx.StaticText(self, -1, ' Masking(2D)')
	151	#
	152	# sizer.Add(wx.StaticText(self, -1, 'Q range'))
	153	# sizer.Add(wx.StaticText(self, -1, ' Min[1/A]'))
	154	# sizer.Add(wx.StaticText(self, -1, ' Max[1/A]'))
	155	# sizer.Add(self.EditMask_title)
	156	#
	157	# sizer.Add(self.reset_qrange)
	158	# sizer.Add(self.qmin)
	159	# sizer.Add(self.qmax)
	160	#
	161	# sizer.Add(self.btEditMask)
	162	# boxsizer_range.Add(sizer_chi2)
	163	# boxsizer_range.Add((10, 10))
	164	# boxsizer_range.Add(sizer)
	165	#
	166	# boxsizer_range.Add((10, 15))
	167	# boxsizer_range.Add(sizer_fit)
	168	# if is_2Ddata:
	169	# self.btEditMask.Enable()
	170	# self.EditMask_title.Enable()
	171	# else:
	172	# self.btEditMask.Disable()
	173	# self.EditMask_title.Disable()
	174	#
	175	# ## save state
	176	# #self.save_current_state()
	177	#
	178	# self.sizer5.Add(boxsizer_range, 0, wx.EXPAND \| wx.ALL, 10)
	179	# self.sizer5.Layout()
	180	#
	181	# def _on_select_model(self, event=None):
	182	# """
	183	# call back for model selection
	184	# """
	185	#
	186	# self.Show(False)
	187	# self._on_select_model_helper()
	188	# self.set_model_param_sizer(self.model)
	189	# if self.model is None:
	190	# self._set_bookmark_flag(False)
	191	# self._keep.Enable(False)
	192	# self._set_save_flag(False)
	193	# self.enable_disp.SetValue(False)
	194	# self.disable_disp.SetValue(True)
	195	# try:
	196	# self.set_dispers_sizer()
	197	# except:
	198	# pass
	199	# self.state.structurecombobox = self.structurebox.GetCurrentSelection()
	200	# self.state.formfactorcombobox = self.formfactorbox.GetCurrentSelection()
	201	#
	202	# if self.model != None:
	203	# self._set_copy_flag(True)
	204	# self._set_paste_flag(True)
	205	# if self.data != None:
	206	# self._set_bookmark_flag(False)
	207	# self._keep.Enable(False)
	208	#
	209	# temp_smear = None
	210	# ## event to post model to fit to fitting plugins
	211	# (ModelEventbox, _) = wx.lib.newevent.NewEvent()
	212	#
	213	# ## set smearing value whether or not
	214	# # the data contain the smearing info
	215	# evt = ModelEventbox(model=self.model,
	216	# smearer=temp_smear,
	217	# qmin=float(self.qmin_x),
	218	# uid=self.uid,
	219	# qmax=float(self.qmax_x))
	220	#
	221	# self._manager._on_model_panel(evt=evt)
	222	# self.mbox_description.SetLabel("Model [%s]" % str(self.model.name))
	223	# self.state.model = self.model.clone()
	224	# self.state.model.name = self.model.name
	225	#
	226	#
	227	# if event != None:
	228	# ## post state to fit panel
	229	# new_event = PageInfoEvent(page = self)
	230	# wx.PostEvent(self.parent, new_event)
	231	# #update list of plugins if new plugin is available
	232	# if self.plugin_rbutton.GetValue():
	233	# temp = self.parent.update_model_list()
	234	# if temp:
	235	# self.model_list_box = temp
	236	# current_val = self.formfactorbox.GetValue()
	237	# pos = self.formfactorbox.GetSelection()
	238	# self._show_combox_helper()
	239	# self.formfactorbox.SetSelection(pos)
	240	# self.formfactorbox.SetValue(current_val)
	241	# self._onDraw(event=None)
	242	# else:
	243	# self._draw_model()
	244	# self.SetupScrolling()
	245	# self.Show(True)
	246	#
	247	# def _update_paramv_on_fit(self):
	248	# """
	249	# make sure that update param values just before the fitting
	250	# """
	251	# #flag for qmin qmax check values
	252	# flag = True
	253	# self.fitrange = True
	254	# is_modified = False
	255	#
	256	# if self.model != None:
	257	# ##Check the values
	258	# self._check_value_enter( self.fittable_param, is_modified)
	259	# self._check_value_enter( self.fixed_param, is_modified)
	260	# self._check_value_enter( self.parameters, is_modified)
	261	#
	262	# # If qmin and qmax have been modified, update qmin and qmax and
	263	# # Here we should check whether the boundaries have been modified.
	264	# # If qmin and qmax have been modified, update qmin and qmax and
	265	# # set the is_modified flag to True
	266	# self.fitrange = self._validate_qrange(self.qmin, self.qmax)
	267	# if self.fitrange:
	268	# tempmin = float(self.qmin.GetValue())
	269	# if tempmin != self.qmin_x:
	270	# self.qmin_x = tempmin
	271	# tempmax = float(self.qmax.GetValue())
	272	# if tempmax != self.qmax_x:
	273	# self.qmax_x = tempmax
	274	# if tempmax == tempmin:
	275	# flag = False
	276	# #temp_smearer = None
	277	# if self._is_2D():
	278	# # only 2D case set mask
	279	# flag = self._validate_Npts()
	280	# if not flag:
	281	# return flag
	282	# else: flag = False
	283	# else:
	284	# flag = False
	285	#
	286	# #For invalid q range, disable the mask editor and fit button, vs.
	287	# if not self.fitrange:
	288	# #self.btFit.Disable()
	289	# if self._is_2D():
	290	# self.btEditMask.Disable()
	291	# else:
	292	# #self.btFit.Enable(True)
	293	# if self._is_2D() and self.data != None:
	294	# self.btEditMask.Enable(True)
	295	#
	296	# if not flag:
	297	# msg = "Cannot Plot or Fit :Must select a "
	298	# msg += " model or Fitting range is not valid!!! "
	299	# wx.PostEvent(self.parent.parent, StatusEvent(status=msg))
	300	#
	301	# self.save_current_state()
	302	#
	303	# return flag
	304	# def save_current_state(self):
	305	# """
	306	# Currently no save option implemented for batch page
	307	# """
	308	# pass
	309	# def save_current_state_fit(self):
	310	# """
	311	# Currently no save option implemented for batch page
	312	# """
	313	# pass
	314	# def set_data(self, data):
	315	# """
	316	# reset the current data
	317	# """
	318	# #id = None
	319	# group_id = None
	320	# flag = False
	321	# if self.data is None and data is not None:
	322	# flag = True
	323	# if data is not None:
	324	# #id = data.id
	325	# group_id = data.group_id
	326	# if self.data is not None:
	327	# flag = (data.id != self.data.id)
	328	# self.data = data
	329	# if self.data is None:
	330	# data_min = ""
	331	# data_max = ""
	332	# data_name = ""
	333	# self._set_bookmark_flag(False)
	334	# self._keep.Enable(False)
	335	# self._set_save_flag(False)
	336	# else:
	337	# if self.model != None:
	338	# self._set_bookmark_flag(False)
	339	# self._keep.Enable(False)
	340	# self._set_save_flag(False)
	341	# self._set_preview_flag(True)
	342	#
	343	# self.formfactorbox.Enable()
	344	# self.structurebox.Enable()
	345	# data_name = self.data.name
	346	# #set maximum range for x in linear scale
	347	# if not hasattr(self.data,"data"): #Display only for 1D data fit
	348	# # Minimum value of data
	349	# data_min = min(self.data.x)
	350	# # Maximum value of data
	351	# data_max = max(self.data.x)
	352	# self.btEditMask.Disable()
	353	# self.EditMask_title.Disable()
	354	# else:
	355	#
	356	# ## Minimum value of data
	357	# data_min = 0
	358	# x = max(math.fabs(self.data.xmin), math.fabs(self.data.xmax))
	359	# y = max(math.fabs(self.data.ymin), math.fabs(self.data.ymax))
	360	# ## Maximum value of data
	361	# data_max = math.sqrt(xx + yy)
	362	# self.btEditMask.Enable()
	363	# self.EditMask_title.Enable()
	364	#
	365	# self.dataSource.SetValue(data_name)
	366	# self.qmin_x = data_min
	367	# self.qmax_x = data_max
	368	# #self.minimum_q.SetValue(str(data_min))
	369	# #self.maximum_q.SetValue(str(data_max))
	370	# self.qmin.SetValue(str(data_min))
	371	# self.qmax.SetValue(str(data_max))
	372	# self.qmin.SetBackgroundColour("white")
	373	# self.qmax.SetBackgroundColour("white")
	374	# self.state.data = data
	375	# self.state.qmin = self.qmin_x
	376	# self.state.qmax = self.qmax_x
	377	#
	378	# #update model plot with new data information
	379	# if flag:
	380	# #set model view button
	381	# if self.data.__class__.__name__ == "Data2D":
	382	# self.enable2D = True
	383	# self.model_view.SetLabel("2D Mode")
	384	# else:
	385	# self.enable2D = False
	386	# self.model_view.SetLabel("1D Mode")
	387	#
	388	# self.model_view.Disable()
	389	#
	390	# wx.PostEvent(self._manager.parent,
	391	# NewPlotEvent(group_id=group_id,
	392	# action="delete"))
	393	# #plot the current selected data
	394	# wx.PostEvent(self._manager.parent, NewPlotEvent(plot=self.data,
	395	# title=str(self.data.title)))
	396	# self._manager.store_data(uid=self.uid, data=data,
	397	# data_list=self.data_list,
	398	# caption=self.window_name)
	399	# self._draw_model()
	400
[2f4b430]	401
	402
[fa09d62]	403	class BGTextCtrl(wx.TextCtrl):
	404	"""
	405	Text control used to display outputs.
[2f4b430]	406	No editing allowed. The background is
[fa09d62]	407	grayed out. User can't select text.
	408	"""
	409	def __init__(self, args, *kwds):
	410	wx.TextCtrl.__init__(self, args, *kwds)
	411	self.SetEditable(False)
	412	self.SetBackgroundColour(self.GetParent().parent.GetBackgroundColour())
[2f4b430]	413
[fa09d62]	414	# Bind to mouse event to avoid text highlighting
	415	# The event will be skipped once the call-back
	416	# is called.
	417	self.Bind(wx.EVT_MOUSE_EVENTS, self._click)
[2f4b430]	418
[fa09d62]	419	def _click(self, event):
	420	"""
	421	Prevent further handling of the mouse event
	422	by not calling Skip().
[2f4b430]	423	"""
[fa09d62]	424	pass
[2f4b430]	425

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

SasView

source: sasview/src/sas/perspectives/fitting/batchfitpage.py @ f21d496

Download in other formats: