modelpage.py @ 92c2345

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 92c2345 was db709e4, checked in by Gervaise Alina <gervyh@…>, 16 years ago
model draw —> parameter changes fixed to redraw model
Property mode set to `100644`
File size: 15.5 KB

Rev	Line
[1b07935d]	1	import sys
	2	import wx
	3	import wx.lib
	4	import numpy
	5	import copy
	6
	7	from sans.guicomm.events import StatusEvent
	8	(ModelEventbox, EVT_MODEL_BOX) = wx.lib.newevent.NewEvent()
	9	_BOX_WIDTH = 80
	10
	11	def format_number(value, high=False):
	12	"""
	13	Return a float in a standardized, human-readable formatted string
	14	"""
	15	try:
	16	value = float(value)
	17	except:
	18	print "returning 0"
	19	return "0"
	20
	21	if high:
	22	return "%-6.4g" % value
	23	else:
	24	return "%-5.3g" % value
	25
	26
[f39511b]	27	class ModelPage(wx.ScrolledWindow):
[1b07935d]	28	"""
	29	FitPanel class contains fields allowing to display results when
	30	fitting a model and one data
	31	@note: For Fit to be performed the user should check at least one parameter
	32	on fit Panel window.
	33
	34	"""
	35	## Internal name for the AUI manager
	36	window_name = "Fit page"
	37	## Title to appear on top of the window
	38	window_caption = "Fit Page"
	39
	40
[86c1832]	41	def __init__(self, parent,model,name, args, *kwargs):
[f39511b]	42	wx.ScrolledWindow.__init__(self, parent, args, *kwargs)
[1b07935d]	43	"""
	44	Initialization of the Panel
	45	"""
[f39511b]	46	#self.scroll = wx.ScrolledWindow(self)
[1b07935d]	47	self.manager = None
	48	self.parent = parent
	49	self.event_owner=None
	50	#panel interface
	51	self.vbox = wx.BoxSizer(wx.VERTICAL)
	52	self.sizer3 = wx.GridBagSizer(5,5)
[49b7efa]	53	self.sizer1 = wx.GridBagSizer(5,5)
[1b07935d]	54	self.sizer2 = wx.GridBagSizer(5,5)
[00561739]	55	self.sizer4 = wx.GridBagSizer(5,5)
[04edd0d]	56	self.sizer5 = wx.GridBagSizer(5,5)
	57	self.static_line_1 = wx.StaticLine(self, -1)
[1b07935d]	58	self.modelbox = wx.ComboBox(self, -1)
	59	id = wx.NewId()
	60	self.vbox.Add(self.sizer3)
[49b7efa]	61	self.vbox.Add(self.sizer1)
[1b07935d]	62	self.vbox.Add(self.sizer2)
[04edd0d]	63	self.vbox.Add(self.static_line_1, 0, wx.EXPAND, 0)
	64	self.vbox.Add(self.sizer5)
[00561739]	65	self.vbox.Add(self.sizer4)
	66
	67	id = wx.NewId()
	68	self.btClose =wx.Button(self,id,'Close')
	69	self.btClose.Bind(wx.EVT_BUTTON, self.onClose,id=id)
	70	self.btClose.SetToolTipString("Close page.")
[86c1832]	71	ix = 1
[dc317d1]	72	iy = 1
	73	self.sizer4.Add(wx.StaticText(self, -1, 'Min'),(iy, ix),(1,1),\
	74	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	75	ix += 2
	76	self.sizer4.Add(wx.StaticText(self, -1, 'Max'),(iy, ix),(1,1),\
	77	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	78	ix = 0
	79	iy += 1
	80	self.sizer4.Add(wx.StaticText(self, -1, 'x range'),(iy, ix),(1,1),\
	81	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
[24ea33c]	82	## Q range
	83	self.qmin= 0.001
	84	self.qmax= 0.1
	85
[dc317d1]	86	ix += 1
	87	self.xmin = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
[24ea33c]	88	self.xmin.SetValue(format_number(self.qmin))
[dc317d1]	89	self.xmin.SetToolTipString("Minimun value of x in linear scale.")
	90	self.xmin.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
	91	self.xmin.Bind(wx.EVT_TEXT_ENTER, self._onparamEnter)
	92	self.sizer4.Add(self.xmin,(iy, ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
[00561739]	93
[dc317d1]	94
	95	ix += 2
	96	self.xmax = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
[24ea33c]	97	self.xmax.SetValue(format_number(self.qmax))
[dc317d1]	98	self.xmax.SetToolTipString("Maximum value of x in linear scale.")
	99	self.xmax.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
	100	self.xmax.Bind(wx.EVT_TEXT_ENTER, self._onparamEnter)
	101
	102	self.sizer4.Add(self.xmax,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	103	ix = 0
	104	iy += 1
	105	self.sizer4.Add((20,20),(iy, ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	106	ix +=3
[00561739]	107	self.sizer4.Add( self.btClose,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[1b07935d]	108	ix = 0
	109	iy = 1
	110	self.sizer3.Add(wx.StaticText(self,-1,'Model'),(iy,ix),(1,1)\
	111	, wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	112	ix += 1
	113	self.sizer3.Add(self.modelbox,(iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[86c1832]	114	#ix = 0
	115	#iy += 1
	116	ix += 1
	117	id = wx.NewId()
	118	self.model_view =wx.Button(self,id,'View 2D')
	119	self.model_view.Bind(wx.EVT_BUTTON, self.onModel2D,id=id)
	120	self.model_view.SetToolTipString("View model in 2D")
[f39511b]	121	self.sizer3.Add(self.model_view,(iy,ix),(1,1),\
	122	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
[1b07935d]	123	# contains link between model ,all its parameters, and panel organization
	124	self.parameters=[]
	125	#contains link between a model and selected parameters to fit
	126	self.param_toFit=[]
	127	# model on which the fit would be performed
	128	self.model=model
[2dbb681]	129	try:
[f39511b]	130	#print"init modelpage",model.name
[2dbb681]	131	self.set_panel(model)
	132	except:
	133	raise
[1b07935d]	134	# preview selected model name
[86c1832]	135	self.prevmodel_name=name
[20be946]	136	print "model view prev_model",name
[1b07935d]	137	self.modelbox.SetValue(self.prevmodel_name)
	138	# flag to check if the user has selected a new model in the combox box
	139	self.model_hasChanged=False
	140	#dictionary of model name and model class
	141	self.model_list_box={}
[86c1832]	142
	143	#enable model 2D draw
	144	self.enable2D= False
[1b07935d]	145	# Data1D to make a deep comparison between 2 Data1D for checking data
	146	#change
	147	self.vbox.Layout()
	148	self.vbox.Fit(self)
[f39511b]	149
[1b07935d]	150	self.SetSizer(self.vbox)
[f39511b]	151	self.SetScrollbars(20,20,55,40)
[1b07935d]	152	self.Centre()
[2dbb681]	153
[1b07935d]	154
[00561739]	155	def onClose(self,event):
	156	""" close the page associated with this panel"""
	157	self.GrandParent.onClose()
	158
[1b07935d]	159	def set_owner(self,owner):
	160	"""
	161	set owner of fitpage
	162	@param owner: the class responsible of plotting
	163	"""
	164	self.event_owner=owner
	165
	166
	167	def set_manager(self, manager):
	168	"""
	169	set panel manager
	170	@param manager: instance of plugin fitting
	171	"""
	172	self.manager = manager
[dc317d1]	173
[f39511b]	174	def onModel2D(self, event):
[86c1832]	175	"""
	176	call manager to plot model in 2D
	177	"""
[24ea33c]	178	# If the 2D display is not currently enabled, plot the model in 2D
	179	# and set the enable2D flag.
	180	if self.enable2D==False:
	181	self.enable2D=True
[20be946]	182	self.manager.draw_model(model=self.model,
[d74d751]	183	name=self.model.name,
[86c1832]	184	description=None,
	185	enable2D=self.enable2D,
[24ea33c]	186	qmin=float(self.qmin),
	187	qmax=float(self.qmax))
	188
[1b07935d]	189	def populate_box(self, dict):
	190	"""
	191	Populate each combox box of each page
	192	@param page: the page to populate
	193	"""
	194	id=0
	195	self.model_list_box=dict
[2dbb681]	196	list_name=[]
	197	for item in self.model_list_box.itervalues():
	198	name = item.__name__
[86c1832]	199	if hasattr(item, "name"):
	200	name = item.name
[2dbb681]	201	list_name.append(name)
	202	list_name.sort()
	203	for name in list_name:
[1b07935d]	204	self.modelbox.Insert(name,int(id))
	205	id+=1
[2dbb681]	206	wx.EVT_COMBOBOX(self.modelbox,-1, self._on_select_model)
[1b07935d]	207	return 0
[86c1832]	208
	209
[060b857]	210
[1b07935d]	211	def _on_select_model(self,event):
	212	"""
	213	react when a model is selected from page's combo box
	214	post an event to its owner to draw an appropriate theory
	215	"""
[00561739]	216	#print "modelpage: self.model_list_box ",self.model_list_box
[1b07935d]	217	for item in self.model_list_box.itervalues():
[2dbb681]	218	name = item.__name__
[86c1832]	219	if hasattr(item, "name"):
	220	name = item.name
[2dbb681]	221	if name ==event.GetString():
[86c1832]	222	model=item()
[f39511b]	223	#print "fitpage: _on_select_model model name",name ,event.GetString()
[b2c3225]	224	self.model= model
[2dbb681]	225	self.set_panel(model)
[86c1832]	226	print "name in model page", name,event.GetString()
	227	self.name= name
	228	self.manager.draw_model(model, name)
[f39511b]	229
[d74d751]	230	self.model_view.SetFocus()
[1b07935d]	231
	232	def get_model_box(self):
	233	""" return reference to combox box self.model"""
	234	return self.modelbox
	235
	236
	237	def get_param_list(self):
	238	"""
	239	@return self.param_toFit: list containing references to TextCtrl
	240	checked.Theses TextCtrl will allow reference to parameters to fit.
	241	@raise: if return an empty list of parameter fit will nnote work
	242	properly so raise ValueError,"missing parameter to fit"
	243	"""
	244	if self.param_toFit !=[]:
	245	return self.param_toFit
	246	else:
	247	raise ValueError,"missing parameter to fit"
	248
	249
[2dbb681]	250	def set_panel(self,model):
[1b07935d]	251	"""
	252	Build the panel from the model content
	253	@param model: the model selected in combo box for fitting purpose
	254	"""
[2dbb681]	255
[1b07935d]	256	self.sizer2.Clear(True)
[49b7efa]	257	self.sizer1.Clear(True)
[04edd0d]	258	self.sizer5.Clear(True)
[1b07935d]	259	self.parameters = []
	260	self.param_toFit=[]
	261	self.model = model
	262	keys = self.model.getParamList()
	263	keys.sort()
[5cab7d3]	264	description=None
	265	if hasattr(self.model,'description'):
	266	description =model.description
[04edd0d]	267	disp_list=self.model.getDispParamList()
	268	ip=0
	269	iq=1
	270	if len(disp_list)>0:
	271	disp = wx.StaticText(self, -1, 'Dispersion')
	272	self.sizer5.Add(disp,( iq, ip),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	273	ip += 1
	274	values = wx.StaticText(self, -1, 'Values')
	275	self.sizer5.Add(values,( iq, ip),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	276
	277	disp_list.sort()
[f39511b]	278	#print "went here",self.model.name,model.description
[1b07935d]	279	iy = 1
	280	ix = 0
[49b7efa]	281	self.cb0 = wx.StaticText(self, -1,'Model Description:')
	282	self.sizer1.Add(self.cb0,(iy, ix),(1,1),\
	283	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
[d171299]	284	iy += 1
	285
	286	self.cb01 = wx.StaticText(self, -1,str(description),style=wx.ALIGN_LEFT)
	287	self.cb01.Wrap(400)
	288	#self.cb01 = wx.StaticText(self, -1,str(description),(45, 25),style=wx.ALIGN_LEFT)
	289
[49b7efa]	290	self.sizer1.Add(self.cb01,(iy, ix),(1,1),\
	291	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	292	ix = 0
	293	iy = 1
[1b07935d]	294	self.cb1 = wx.StaticText(self, -1,'Parameters')
	295	self.sizer2.Add(self.cb1,(iy, ix),(1,1),\
	296	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	297	ix +=1
	298	self.text2_2 = wx.StaticText(self, -1, 'Values')
	299	self.sizer2.Add(self.text2_2,(iy, ix),(1,1),\
	300	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	301	ix +=1
	302	self.text2_4 = wx.StaticText(self, -1, 'Units')
	303	self.sizer2.Add(self.text2_4,(iy, ix),(1,1),\
	304	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	305	self.text2_4.Hide()
	306	for item in keys:
[04edd0d]	307	if not item in disp_list:
	308	iy += 1
	309	ix = 0
	310	cb=wx.StaticText(self, -1, item)
	311	self.sizer2.Add( cb,( iy, ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	312	ix += 1
	313	value= self.model.getParam(item)
	314	ctl1 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
	315	ctl1.SetValue(str (format_number(value)))
	316	ctl1.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
	317	ctl1.Bind(wx.EVT_TEXT_ENTER, self._onparamEnter)
	318	self.sizer2.Add(ctl1, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	319	ix +=1
	320
	321	# Units
	322	try:
	323	units = wx.StaticText(self, -1, self.model.details[item][0], style=wx.ALIGN_LEFT)
	324	except:
	325	units = wx.StaticText(self, -1, "", style=wx.ALIGN_LEFT)
	326	self.sizer2.Add(units, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	327	else:
	328	ip = 0
	329	iq += 1
	330	cb = wx.CheckBox(self, -1, item, (10, 10))
	331	cb.SetValue(False)
	332	self.sizer5.Add( cb,( iq, ip),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	333	wx.EVT_CHECKBOX(self, cb.GetId(), self._on_select_model)
	334
	335	ip += 1
	336	value= self.model.getParam(item)
	337	ctl1 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
	338	ctl1.SetValue(str (format_number(value)))
	339	ctl1.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
	340	ctl1.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
	341	self.sizer5.Add(ctl1, (iq,ip),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[1b07935d]	342	#save data
	343	self.parameters.append([cb,ctl1])
[04edd0d]	344	iy+=1
	345	self.sizer2.Add((20,20),(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	346
[1b07935d]	347	#Display units text on panel
	348	for item in keys:
	349	if self.model.details[item][0]!='':
	350	self.text2_4.Show()
	351	break
	352	else:
	353	self.text2_4.Hide()
	354	self.vbox.Layout()
	355	self.GrandParent.GetSizer().Layout()
[04edd0d]	356
[1b07935d]	357
	358	def _onparamEnter(self,event):
	359	"""
	360	when enter value on panel redraw model according to changed
	361	"""
	362	self.set_model_parameter()
	363
	364	def set_model_parameter(self):
	365	if len(self.parameters) !=0 and self.model !=None:
[24ea33c]	366	# Flag to register when a parameter has changed.
	367	is_modified = False
	368	if float(self.xmin.GetValue()) != self.qmin:
	369	self.qmin = float(self.xmin.GetValue())
	370	is_modified = True
	371	if float(self.xmax.GetValue()) != self.qmax:
	372	self.qmax = float(self.xmax.GetValue())
	373	is_modified = True
[db709e4]	374	for item in self.parameters:
	375	try:
	376
	377	if item[1].IsModified() or is_modified:
	378	print str(item[0].GetLabelText()),item[1].IsModified()
	379	item[1].SetModified(False)
	380	name=str(item[0].GetLabelText())
	381	value= float(item[1].GetValue())
	382	self.model.setParam(name,value)
	383
	384	is_modified=False
	385	print "self.enable2d",self.enable2D
	386	self.manager.draw_model(self.model, self.model.name,
[24ea33c]	387	qmin=self.qmin, qmax=self.qmax,
	388	enable2D=self.enable2D)
[db709e4]	389
	390
	391	except:
	392	wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
	393	"Model Drawing Error:wrong value entered : %s"% sys.exc_value))
	394
	395	# Here we should check whether the boundaries have been modified.
	396	# If qmin and qmax have been modified, update qmin and qmax and
	397	# set the is_modified flag to True
	398
[2e10b70]	399	# enable1D=True,qmin=None,qmax=None, qstep=None)
	400
[d74d751]	401	self.model_view.SetFocus()
[2e10b70]	402
	403
	404

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

SasView

source: sasview/sansview/perspectives/fitting/modelpage.py @ 92c2345

Download in other formats: