simfitpage.py @ d1241f5

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 d1241f5 was 6bcdad1, checked in by Gervaise Alina <gervyh@…>, 16 years ago
close page bug fixed
Property mode set to `100644`
File size: 11.1 KB

Rev	Line
[d89f09b]	1
	2	import sys,re,string, wx
	3	from sans.guicomm.events import StatusEvent
	4
[3aae6b6]	5	class SimultaneousFitPage(wx.ScrolledWindow):
[d89f09b]	6	"""
	7	Simultaneous fitting panel
	8	All that needs to be defined are the
	9	two data members window_name and window_caption
	10	"""
	11	## Internal name for the AUI manager
	12	window_name = "simultaneous Fit page"
	13	## Title to appear on top of the window
	14	window_caption = "Simultaneous Fit Page"
	15
	16
	17	def __init__(self, parent, args, *kwargs):
[3aae6b6]	18	wx.ScrolledWindow.__init__(self, parent, args, *kwargs)
[d89f09b]	19	"""
	20	Simultaneous page display
	21	"""
	22	self.parent = parent
	23	self.sizer3 = wx.GridBagSizer(5,5)
	24	self.sizer1 = wx.GridBagSizer(5,5)
[55fd102]	25	#self.sizer2 = wx.GridBagSizer(5,5)
	26	self.sizer2 = wx.BoxSizer(wx.HORIZONTAL)
[d89f09b]	27	self.vbox = wx.BoxSizer(wx.VERTICAL)
	28	self.vbox.Add(self.sizer3)
	29	self.vbox.Add(self.sizer1)
	30	self.vbox.Add(self.sizer2)
	31	id = wx.NewId()
	32	self.btFit =wx.Button(self,id,'Constraint Fit')
	33	self.btFit.Bind(wx.EVT_BUTTON, self.onFit,id=id)
	34	self.btFit.SetToolTipString("Perform fit.")
	35	ix = 0
	36	iy = 1
	37	self.cb1 = wx.CheckBox(self, -1,'Models', (10, 10))
	38	self.sizer3.Add(self.cb1,(iy, ix),(1,1),\
	39	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	40	wx.EVT_CHECKBOX(self, self.cb1.GetId(), self.select_all_model_name)
[442895f]	41
[d89f09b]	42	text=wx.StaticText(self, -1, 'Constraint')
[55fd102]	43	self.sizer2.Add(text,0, wx.LEFT\|wx.RIGHT\|wx.ADJUST_MINSIZE, 10)
[442895f]	44
[d89f09b]	45	self.ctl2 = wx.TextCtrl(self, -1, style=wx.TE_MULTILINE)
[948add7]	46
[55fd102]	47	self.sizer2.Add(self.ctl2, 0, wx.LEFT\|wx.RIGHT\|wx.ADJUST_MINSIZE, 10)
[442895f]	48
[55fd102]	49	self.sizer2.Add(self.btFit, 0, wx.LEFT\|wx.RIGHT\|wx.ADJUST_MINSIZE, 10)
[d89f09b]	50	self.params=[]
	51	self.model_list=[]
	52	self.model_toFit=[]
[1b07935d]	53	self.page_finder={}
[442895f]	54
[d89f09b]	55	self.vbox.Layout()
	56	self.vbox.Fit(self)
	57	self.SetSizer(self.vbox)
[3aae6b6]	58	self.SetScrollbars(20,20,55,40)
[d89f09b]	59	self.Centre()
	60
	61
	62	def onFit(self,event):
[1b07935d]	63	""" signal for fitting"""
[8e81af0]	64	for page in self.page_finder.iterkeys():
	65	page.set_model_parameter()
	66
[948add7]	67	if len(self.model_toFit) ==1 :
	68	self.manager._on_single_fit()
[8e81af0]	69	# print "simfitpage: when here"
[948add7]	70	elif len(self.model_toFit) > 1 :
[8e81af0]	71	self._onTextEnter()
	72	self.set_model()
[1b07935d]	73	self.manager._on_simul_fit()
	74	else:
	75	wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
[d89f09b]	76	"Select at least on model to fit "))
[00561739]	77
	78
[d89f09b]	79	def set_manager(self, manager):
	80	"""
	81	set panel manager
	82	@param manager: instance of plugin fitting
	83	"""
	84	self.manager = manager
	85
	86
	87	def select_all_model_name(self,event):
	88	"""
	89	check all models names
	90	"""
	91	self.model_toFit=[]
	92	if self.cb1.GetValue()==True:
	93	for item in self.model_list:
	94	item[0].SetValue(True)
[948add7]	95	#item[1].schedule_tofit(1)
	96	self.manager.schedule_for_fit( value=1,fitproblem =item[1])
[d89f09b]	97	self.model_toFit.append(item)
	98	else:
[8e81af0]	99	#print"simfit: deselected all"
[89fef2c]	100
[d89f09b]	101	for item in self.model_list:
	102	item[0].SetValue(False)
[89fef2c]	103	self.manager.schedule_for_fit( value=0,fitproblem =item[1])
[948add7]	104
[d89f09b]	105	self.model_toFit=[]
	106
	107
	108	def add_model(self,page_finder):
	109	"""
	110	Receive a dictionary containing information to display model name
	111	@param page_finder: the dictionary containing models information
	112	"""
[00561739]	113
[d89f09b]	114	self.model_list=[]
	115	self.model_toFit=[]
	116	self.sizer1.Clear(True)
[1b07935d]	117	self.page_finder=page_finder
[948add7]	118	self.cb1.SetValue(False)
[d89f09b]	119	ix = 0
	120	iy = 1
	121	list=[]
	122	for page, value in page_finder.iteritems():
	123	try:
[442895f]	124	list = value.get_model()
	125	model = list[0]
	126	modelname = list[1]
[d89f09b]	127	cb = wx.CheckBox(self, -1, modelname, (10, 10))
	128	cb.SetValue(False)
	129	self.sizer1.Add( cb,( iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	130	ix = 0
	131	iy += 1
	132	wx.EVT_CHECKBOX(self, cb.GetId(), self.select_model_name)
	133	self.model_list.append([cb,value,page,modelname])
	134	except:
	135	#wx.PostEvent(self.parent.GrandParent, StatusEvent(status="Simultaneous fit: %s doesn't have a model selected yet %s" % \
	136	#(value.get_data().group_id,sys.exc_value)))
	137	pass
[442895f]	138	iy +=1
	139	self.sizer1.Add((20,20),( iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
[d89f09b]	140	self.sizer1.Layout()
	141	self.vbox.Layout()
	142
	143	def remove_model(self,delpage):
	144	"""
[948add7]	145	Remove a checkbox and the name related to a model selected on page delpage
[d89f09b]	146	@param delpage: the page removed
	147	"""
[6bcdad1]	148	print "self.model_list",self.model_list
	149
	150	self.model_list=[]
	151	self.model_toFit=[]
[d89f09b]	152	self.sizer1.Clear(True)
[6bcdad1]	153
	154	self.cb1.SetValue(False)
[d89f09b]	155	ix = 0
	156	iy = 1
[6bcdad1]	157	list=[]
	158	for page, value in self.page_finder.iteritems():
[d89f09b]	159	try:
[6bcdad1]	160	if page!= delpage:
	161	list = value.get_model()
	162	model = list[0]
	163	modelname = list[1]
	164	cb = wx.CheckBox(self, -1, modelname, (10, 10))
[d89f09b]	165	cb.SetValue(False)
	166	self.sizer1.Add( cb,( iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	167	ix = 0
	168	iy += 1
	169	wx.EVT_CHECKBOX(self, cb.GetId(), self.select_model_name)
[6bcdad1]	170	self.model_list.append([cb,value,page,modelname])
[d89f09b]	171	except:
[6bcdad1]	172	#wx.PostEvent(self.parent.GrandParent, StatusEvent(status="Simultaneous fit: %s doesn't have a model selected yet %s" % \
	173	#(value.get_data().group_id,sys.exc_value)))
	174	pass
	175	iy +=1
	176	self.sizer1.Add((20,20),( iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
[d89f09b]	177	self.sizer1.Layout()
	178	self.vbox.Layout()
	179
	180	def select_model_name(self,event):
	181	"""
	182	Save interformation related to checkbox and their states
	183	"""
	184	self.model_toFit=[]
	185	for item in self.model_list:
	186	if item[0].GetValue()==True:
[3b19ac9]	187	item[1].schedule_tofit('True')
[d89f09b]	188	self.model_toFit.append(item)
[948add7]	189	self.manager.schedule_for_fit( value=1,fitproblem =item[1])
[d89f09b]	190	else:
[8e81af0]	191	#print"simfit: deselected one"
[948add7]	192	self.manager.schedule_for_fit( value=0,fitproblem =item[1])
[d89f09b]	193	if item in self.model_toFit:
	194	self.model_toFit.remove(item)
	195	self.cb1.SetValue(False)
[8e81af0]	196
[d89f09b]	197	if len(self.model_list)==len(self.model_toFit):
	198	self.cb1.SetValue(True)
	199	else:
	200	self.cb1.SetValue(False)
[948add7]	201
	202
[d89f09b]	203
	204	def set_model(self):
	205	"""
	206	set_model take values in self.params which are the values enters by the user
	207	and try to assign them into the model concerned in self.manager. page.finder
	208	"""
	209	if len(self.params) >0:
	210	for item in self.model_toFit:
	211	#print"simfitpage: self.model_toFit",item[1]
	212	list=item[1].get_model()
	213	#print "simfitpage: list fitpanel2",list,list[0]
	214	model=list[0]
[1b07935d]	215	param_list=model.getParamList()
[8e81af0]	216	print "simfitpage: on set_model self.params ",self.params
[d89f09b]	217	if self.params !=[]:
	218	for element in self.params:
[1b07935d]	219	if model.name == str(element[0]):
[d89f09b]	220	for item in param_list:
	221	if item==str(element[1]):
[8e81af0]	222	print "simfitpage: on set_model page 1",param_list
	223	print "simfitpage: model name",element[0], model.name
	224	print "simfitpage: param name ,param value",element[1],element[2]
[1b07935d]	225	self.manager.set_page_finder(model.name,element[1],\
[d89f09b]	226	str(element[2]))
	227	#print "simfitpage:on set_model page 2",model.params['A'],self.params[2]
[1b07935d]	228
[00561739]	229	def _onTextEnter(self):
[d89f09b]	230	"""
	231	get values from the constrainst textcrtl ,parses them into model name
	232	parameter name and parameters values.
	233	store them in a list self.params .when when params is not empty set_model uses it
	234	to reset the appropriate model and its appropriates parameters
	235	"""
	236	value= self.ctl2.GetValue()
[8e81af0]	237	if value:
	238	self.params=[]
	239	#print "simfitpage: value",value
	240	try:
	241	expression='[\s,;]'
	242	if re.search(expression,value) !=None:
	243	word=re.split(expression,value)
	244	#print "simfitpage: when herre",word
	245	for item in word:
[d89f09b]	246	self.params.append(self.parser_helper(item))
[8e81af0]	247	else:
	248	self.params.append(self.parser_helper(value))
	249	except:
	250	raise
	251	wx.PostEvent(self.parent.GrandParent, StatusEvent(status="Constraint Error: %s" % sys.exc_value))
[d89f09b]	252	#print "simfitpage: self.params",self.params
	253
	254
	255
	256	def parser_helper(self,value):
	257	"""
	258	@return param:a list containing the name of a model ,its parameters
	259	value and name extracted from the constrainst controlbox
	260	"""
	261	#print "simfitpage: value",value
	262	if string.find(value, "=") !=-1:
	263	model_param= re.split("=",value)
	264	param_name=model_param[0]
	265	param_value=model_param[1]
	266	#print"simfitpage: ", param_name
	267	#print "simfitpage: ",param_value
	268	if string.find(param_name,".")!=-1:
	269	param_names= re.split("\.",param_name)
	270	model_name=param_names[0]
	271	param_name=param_names[1]
[00561739]	272	##### just added
[8e81af0]	273	print "simfitpage: param name",model_name,param_name
	274
	275	param=[str(model_name),param_name,str(param_value)]
[d89f09b]	276	#print "simfitpage: param",param
	277	return param
	278	else:
	279	raise ValueError,"cannot evaluate this expression"
	280	return
	281	else:
	282	raise ValueError,"Missing '=' in expression"
	283
	284
	285

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

SasView

source: sasview/sansview/perspectives/fitting/simfitpage.py @ d1241f5

Download in other formats: