fitpage1D.py @ 0aac36f

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 0aac36f was 0aac36f, checked in by Gervaise Alina <gervyh@…>, 15 years ago
scroll_bar problem fixed—> add self.SetScrollbars?(20,20,55,40) on set_panel method
Property mode set to `100644`
File size: 26.5 KB

Rev	Line
[d89f09b]	1	import sys
	2	import wx
	3	import wx.lib
[442895f]	4	import numpy,math
[d89f09b]	5	import copy
[08b9c6c8]	6	import sans.models.dispersion_models
[d89f09b]	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 FitPage1D(wx.ScrolledWindow):
[d89f09b]	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
[9d31a8b]	41	def __init__(self, parent,data, args, *kwargs):
[f39511b]	42	wx.ScrolledWindow.__init__(self, parent, args, *kwargs)
[d89f09b]	43	"""
	44	Initialization of the Panel
	45	"""
[f39511b]	46	#self.scroll = wx.ScrolledWindow(self)
[b5847bd8]	47
[d89f09b]	48	self.manager = None
	49	self.parent = parent
[b5847bd8]	50	self.event_owner = None
[d89f09b]	51	#panel interface
	52	self.vbox = wx.BoxSizer(wx.VERTICAL)
[08b9c6c8]	53	self.sizer6 = wx.GridBagSizer(5,5)
[5cab7d3]	54	self.sizer5 = wx.GridBagSizer(5,5)
[d89f09b]	55	self.sizer4 = wx.GridBagSizer(5,5)
	56	self.sizer3 = wx.GridBagSizer(5,5)
	57	self.sizer2 = wx.GridBagSizer(5,5)
	58	self.sizer1 = wx.GridBagSizer(5,5)
[04edd0d]	59
[1f62278]	60
[f39511b]	61	self.DataSource =wx.StaticText(self, -1,str(data.name))
[08b9c6c8]	62	self.smearer_box = wx.ComboBox(self, -1)
	63	wx.EVT_COMBOBOX( self.smearer_box,-1, self.onSmear )
	64	self.smeares= sans.models.dispersion_models.models
	65	i=0
	66	self.smearer_box.SetValue(str(None))
	67	self.smearer_box.Insert(str(None),i)
	68	for k,v in self.smeares.iteritems():
	69	self.smearer_box.Insert(str(v),i)
	70	i+=1
[f39511b]	71	self.modelbox = wx.ComboBox(self, -1)
[d89f09b]	72	id = wx.NewId()
[f39511b]	73	self.btFit =wx.Button(self,id,'Fit')
[d89f09b]	74	self.btFit.Bind(wx.EVT_BUTTON, self.onFit,id=id)
	75	self.btFit.SetToolTipString("Perform fit.")
[f39511b]	76	self.static_line_1 = wx.StaticLine(self, -1)
[5cab7d3]	77
[d89f09b]	78	self.vbox.Add(self.sizer3)
	79	self.vbox.Add(self.sizer2)
[5cab7d3]	80	self.vbox.Add(self.static_line_1, 0, wx.EXPAND, 0)
	81	self.vbox.Add(self.sizer5)
[08b9c6c8]	82	self.vbox.Add(self.sizer6)
[d89f09b]	83	self.vbox.Add(self.sizer4)
	84	self.vbox.Add(self.sizer1)
	85
[442895f]	86	id = wx.NewId()
[f39511b]	87	self.btClose =wx.Button(self,id,'Close')
[442895f]	88	self.btClose.Bind(wx.EVT_BUTTON, self.onClose,id=id)
	89	self.btClose.SetToolTipString("Close page.")
[d89f09b]	90	ix = 0
	91	iy = 1
[f39511b]	92	self.sizer3.Add(wx.StaticText(self, -1, 'Data Source Name : '),(iy,ix),\
[d89f09b]	93	(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	94	ix += 1
	95	self.sizer3.Add(self.DataSource,(iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	96	ix += 1
	97	self.sizer3.Add((20,20),(iy,ix),(1,1),wx.RIGHT\|wx.EXPAND\|wx.ADJUST_MINSIZE,0)
	98	ix = 0
	99	iy += 1
[08b9c6c8]	100	self.sizer3.Add(wx.StaticText(self,-1,'Averaging (Smearer Type)'),(iy,ix),(1,1)\
	101	, wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	102	ix += 1
	103	self.sizer3.Add(self.smearer_box,(iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	104
	105	ix = 0
	106	iy += 1
[f39511b]	107	self.sizer3.Add(wx.StaticText(self,-1,'Model'),(iy,ix),(1,1)\
[d89f09b]	108	, wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	109	ix += 1
	110	self.sizer3.Add(self.modelbox,(iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[0550752]	111
[9d31a8b]	112	ix = 0
[5cab7d3]	113	iy = 1
[9d31a8b]	114	#set maximum range for x in linear scale
[f39511b]	115	self.text4_3 = wx.StaticText(self, -1, 'Maximum Data Range(Linear)', style=wx.ALIGN_LEFT)
[9d31a8b]	116	self.sizer4.Add(self.text4_3,(iy,ix),(1,1),\
	117	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	118	ix += 1
[f39511b]	119	self.sizer4.Add(wx.StaticText(self, -1, 'Min'),(iy, ix),(1,1),\
[d89f09b]	120	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	121	ix += 2
[f39511b]	122	self.sizer4.Add(wx.StaticText(self, -1, 'Max'),(iy, ix),(1,1),\
[d89f09b]	123	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	124	ix = 0
	125	iy += 1
[f39511b]	126	self.sizer4.Add(wx.StaticText(self, -1, 'x range'),(iy, ix),(1,1),\
[9d31a8b]	127	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
[d89f09b]	128	ix += 1
[f39511b]	129	self.xmin = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
[9d31a8b]	130	self.xmin.SetValue(format_number(numpy.min(data.x)))
[d89f09b]	131	self.xmin.SetToolTipString("Minimun value of x in linear scale.")
	132	self.xmin.Bind(wx.EVT_KILL_FOCUS, self._onTextEnter)
[55fd102]	133	self.xmin.Bind(wx.EVT_TEXT_ENTER, self._onTextEnter)
[44bbf6a]	134	self.xmin.Disable()
[0550752]	135	self.sizer4.Add(self.xmin,(iy, ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	136
	137
[d89f09b]	138	ix += 2
[f39511b]	139	self.xmax = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
[9d31a8b]	140	self.xmax.SetValue(format_number(numpy.max(data.x)))
[d89f09b]	141	self.xmax.SetToolTipString("Maximum value of x in linear scale.")
	142	self.xmax.Bind(wx.EVT_KILL_FOCUS, self._onTextEnter)
[55fd102]	143	self.xmax.Bind(wx.EVT_TEXT_ENTER, self._onTextEnter)
[44bbf6a]	144	self.xmax.Disable()
[0550752]	145	self.sizer4.Add(self.xmax,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	146	ix =0
[d171299]	147	iy+=1
[0550752]	148	self.sizer4.Add((20,20),(iy,ix),(1,1),wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
[d89f09b]	149	#Set chisqr result into TextCtrl
	150	ix = 0
	151	iy = 1
[08b9c6c8]	152
[f39511b]	153	self.text1_1 = wx.StaticText(self, -1, 'Chi2/dof', style=wx.ALIGN_LEFT)
[d171299]	154	#self.sizer1.Add(self.text1_1,1)
[d89f09b]	155	self.sizer1.Add(self.text1_1,(iy,ix),(1,1),\
	156	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	157	ix += 1
[f39511b]	158	self.tcChi = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
[d89f09b]	159	self.tcChi.SetToolTipString("Chi^2 over degrees of freedom.")
[d171299]	160	#self.sizer1.Add(self.tcChi, 1, wx.R \| wx.BOTTOM , 5)
[d89f09b]	161	self.sizer1.Add(self.tcChi,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	162	ix +=2
[d171299]	163	#self.sizer1.Add(self.btFit, 1, wx.LEFT \| wx.BOTTOM , 5)
[d89f09b]	164	self.sizer1.Add(self.btFit,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[0550752]	165	ix+= 2
[442895f]	166	self.sizer1.Add( self.btClose,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[d171299]	167	#self.sizer1.Add( self.btClose,1, wx.LEFT \| wx.BOTTOM , 5)
[47d10a2]	168	self.tcChi.Disable()
[0550752]	169	ix= 0
[2dbb681]	170	iy+=1
	171	self.sizer1.Add((20,20),(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[d171299]	172	#self.sizer1.Add((20,20), 0)
[d89f09b]	173	# contains link between model ,all its parameters, and panel organization
	174	self.parameters=[]
[08b9c6c8]	175	self.fixed_param=[]
[d89f09b]	176	#contains link between a model and selected parameters to fit
	177	self.param_toFit=[]
	178	# model on which the fit would be performed
	179	self.model=None
	180	#dictionary of model name and model class
	181	self.model_list_box={}
[5cab7d3]	182	self.data = data
[d15c0202]	183	## Q range
	184	self.qmin= 0.001
	185	self.qmax= 0.1
[f39511b]	186	"""
[b5847bd8]	187	bs = wx.BoxSizer(wx.VERTICAL)
	188	bs.Add(self.scroll, 1, wx.EXPAND)
	189	self.SetSizer(bs)
	190	self.scroll.SetSizer(self.vbox)
	191	self.scroll.SetScrollbars(20,20,55,40)
[f39511b]	192	#width,height = self.GetSize()
	193	"""
	194	self.vbox.Layout()
	195	self.vbox.Fit(self)
	196	self.SetSizer(self.vbox)
	197	self.SetScrollbars(20,20,55,40)
	198
[d89f09b]	199	self.Centre()
[b5847bd8]	200	self.Layout()
	201	self.GrandParent.GetSizer().Layout()
[f39511b]	202
[b5847bd8]	203
	204
	205
	206
[d89f09b]	207
	208
[9d31a8b]	209
[d89f09b]	210	def set_owner(self,owner):
	211	"""
	212	set owner of fitpage
	213	@param owner: the class responsible of plotting
	214	"""
[d171299]	215	self.event_owner = owner
[d89f09b]	216
	217
	218	def set_manager(self, manager):
	219	"""
	220	set panel manager
	221	@param manager: instance of plugin fitting
	222	"""
	223	self.manager = manager
	224
	225
[442895f]	226	def onClose(self,event):
[00561739]	227	""" close the page associated with this panel"""
[442895f]	228	self.GrandParent.onClose()
	229
	230
	231	def compute_chisqr(self):
	232	""" @param fn: function that return model value
	233	@return residuals
	234	"""
[693ab78]	235
[442895f]	236	flag=self.checkFitRange()
	237	if flag== True:
[948add7]	238	try:
	239	qmin = float(self.xmin.GetValue())
	240	qmax = float(self.xmax.GetValue())
[bcd6d51]	241	x,y,dy = [numpy.asarray(v) for v in (self.data.x,self.data.y,self.data.dy)]
	242	if qmin==None and qmax==None:
	243	fx =numpy.asarray([self.model.run(v) for v in x])
	244	res=(y - fx)/dy
	245	else:
	246	idx = (x>= qmin) & (x <=qmax)
	247	fx = numpy.asarray([self.model.run(item)for item in x[idx ]])
	248	res= (y[idx] - fx)/dy[idx]
	249
	250
[948add7]	251	sum=0
	252	for item in res:
	253	if numpy.isfinite(item):
	254	sum +=item
	255	self.tcChi.SetValue(format_number(math.fabs(sum)))
	256	except:
	257	wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
	258	"Chisqr cannot be compute: %s"% sys.exc_value))
[442895f]	259
	260
[d89f09b]	261	def onFit(self,event):
	262	""" signal for fitting"""
	263
	264	flag=self.checkFitRange()
	265	self.set_manager(self.manager)
	266
	267	qmin=float(self.xmin.GetValue())
	268	qmax =float( self.xmax.GetValue())
	269	if len(self.param_toFit) >0 and flag==True:
[948add7]	270	self.manager.schedule_for_fit( value=1,fitproblem =None)
[d89f09b]	271	self.manager._on_single_fit(qmin=qmin,qmax=qmax)
	272	else:
	273	wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
	274	"Select at least on parameter to fit "))
	275	def populate_box(self, dict):
	276	"""
	277	Populate each combox box of each page
	278	@param page: the page to populate
	279	"""
	280	id=0
	281	self.model_list_box=dict
[49b7efa]	282	list_name=[]
	283	for item in self.model_list_box.itervalues():
[00561739]	284	name = item.__name__
[d89f09b]	285	if hasattr(item, "name"):
	286	name = item.name
[49b7efa]	287	list_name.append(name)
	288	list_name.sort()
	289	for name in list_name:
[d89f09b]	290	self.modelbox.Insert(name,int(id))
	291	id+=1
[49b7efa]	292	wx.EVT_COMBOBOX(self.modelbox,-1, self._on_select_model)
[d89f09b]	293	return 0
	294
	295
	296	def _on_select_model(self,event):
	297	"""
	298	react when a model is selected from page's combo box
	299	post an event to its owner to draw an appropriate theory
	300	"""
[44bbf6a]	301	self.btFit.SetFocus()
[d89f09b]	302	for item in self.model_list_box.itervalues():
[00561739]	303	name = item.__name__
	304	if hasattr(item, "name"):
	305	name = item.name
	306	#print "fitpage: _on_select_model model name",name ,event.GetString()
	307	if name ==event.GetString():
	308	try:
[f39511b]	309	self.model=item()
	310	evt = ModelEventbox(model=self.model,name=name)
[d89f09b]	311	wx.PostEvent(self.event_owner, evt)
[f39511b]	312	#self.model= item()
	313	#self.set_panel(self.model)
[00561739]	314	except:
	315	raise #ValueError,"model.name is not equal to model class name"
	316	break
[d89f09b]	317
	318	def _onTextEnter(self,event):
	319	"""
	320	set a flag to determine if the fitting range entered by the user is valid
	321	"""
[442895f]	322
[d89f09b]	323	try:
	324	flag=self.checkFitRange()
	325	if flag==True and self.model!=None:
[f39511b]	326	#print"fit page",self.xmin.GetValue(),self.xmax.GetValue()
[6f73a08]	327	self.manager.redraw_model(float(self.xmin.GetValue())\
[d89f09b]	328	,float(self.xmax.GetValue()))
	329	except:
[6f73a08]	330
[d89f09b]	331	wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
	332	"Drawing Error:wrong value entered %s"% sys.exc_value))
	333
	334	def checkFitRange(self):
	335	"""
	336	Check the validity of fitting range
	337	@note: xmin should always be less than xmax or else each control box
	338	background is colored in pink.
	339	"""
	340
	341	flag = True
	342	valueMin = self.xmin.GetValue()
	343	valueMax = self.xmax.GetValue()
	344	# Check for possible values entered
[2dbb681]	345	#print "fitpage: checkfitrange:",valueMin,valueMax
[d89f09b]	346	try:
	347	if (float(valueMax)> float(valueMin)):
	348	self.xmax.SetBackgroundColour(wx.WHITE)
	349	self.xmin.SetBackgroundColour(wx.WHITE)
	350	else:
	351	flag = False
	352	self.xmin.SetBackgroundColour("pink")
	353	self.xmax.SetBackgroundColour("pink")
	354	except:
	355	flag = False
	356	self.xmin.SetBackgroundColour("pink")
	357	self.xmax.SetBackgroundColour("pink")
	358
	359	self.xmin.Refresh()
	360	self.xmax.Refresh()
	361	return flag
	362
[9d31a8b]	363
[d89f09b]	364	def get_model_box(self):
	365	""" return reference to combox box self.model"""
	366	return self.modelbox
	367
	368
	369	def get_param_list(self):
	370	"""
	371	@return self.param_toFit: list containing references to TextCtrl
	372	checked.Theses TextCtrl will allow reference to parameters to fit.
	373	@raise: if return an empty list of parameter fit will nnote work
	374	properly so raise ValueError,"missing parameter to fit"
	375	"""
	376	if self.param_toFit !=[]:
	377	return self.param_toFit
	378	else:
	379	raise ValueError,"missing parameter to fit"
	380
	381
	382	def set_panel(self,model):
	383	"""
	384	Build the panel from the model content
	385	@param model: the model selected in combo box for fitting purpose
	386	"""
	387	self.sizer2.Clear(True)
[5cab7d3]	388	self.sizer5.Clear(True)
[08b9c6c8]	389	self.sizer6.Clear(True)
[d89f09b]	390	self.parameters = []
	391	self.param_toFit=[]
	392	self.model = model
	393	keys = self.model.getParamList()
[44bbf6a]	394	#print "fitpage1D : dispersion list",self.model.getDispParamList()
[d89f09b]	395	keys.sort()
[5cab7d3]	396	disp_list=self.model.getDispParamList()
[08b9c6c8]	397	fixed=self.model.fixed
[04edd0d]	398	ip=0
	399	iq=1
[08b9c6c8]	400
	401	ik=0
	402	im=1
[04edd0d]	403	if len(disp_list)>0:
[f39511b]	404	disp = wx.StaticText(self, -1, 'Dispersion')
[04edd0d]	405	self.sizer5.Add(disp,( iq, ip),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	406	ip += 1
[f39511b]	407	values = wx.StaticText(self, -1, 'Values')
[04edd0d]	408	self.sizer5.Add(values,( iq, ip),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	409
[5cab7d3]	410	disp_list.sort()
[d89f09b]	411	iy = 1
	412	ix = 0
[f39511b]	413	self.cb1 = wx.CheckBox(self, -1,'Parameters', (10, 10))
	414	wx.EVT_CHECKBOX(self, self.cb1.GetId(), self.select_all_param)
[d89f09b]	415	self.sizer2.Add(self.cb1,(iy, ix),(1,1),\
	416	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	417	ix +=1
[f39511b]	418	self.text2_2 = wx.StaticText(self, -1, 'Values')
[d89f09b]	419	self.sizer2.Add(self.text2_2,(iy, ix),(1,1),\
	420	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	421	ix +=2
[f39511b]	422	self.text2_3 = wx.StaticText(self, -1, 'Errors')
[d89f09b]	423	self.sizer2.Add(self.text2_3,(iy, ix),(1,1),\
	424	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	425	self.text2_3.Hide()
	426	ix +=1
[f39511b]	427	self.text2_4 = wx.StaticText(self, -1, 'Units')
[d89f09b]	428	self.sizer2.Add(self.text2_4,(iy, ix),(1,1),\
	429	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	430	self.text2_4.Hide()
[5cab7d3]	431
[d89f09b]	432	for item in keys:
[5cab7d3]	433	if not item in disp_list:
	434	iy += 1
	435	ix = 0
	436
[f39511b]	437	cb = wx.CheckBox(self, -1, item, (10, 10))
[5cab7d3]	438	cb.SetValue(False)
	439	self.sizer2.Add( cb,( iy, ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
[f39511b]	440	wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
[08b9c6c8]	441
[5cab7d3]	442	ix += 1
	443	value= self.model.getParam(item)
[f39511b]	444	ctl1 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
[5cab7d3]	445	ctl1.SetValue(str (format_number(value)))
	446	ctl1.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
	447	ctl1.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
	448	self.sizer2.Add(ctl1, (iy,ix),(1,1), wx.EXPAND)
[08b9c6c8]	449
[5cab7d3]	450	ix += 1
[f39511b]	451	text2=wx.StaticText(self, -1, '+/-')
[5cab7d3]	452	self.sizer2.Add(text2,(iy, ix),(1,1),\
	453	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	454	text2.Hide()
	455	ix += 1
[f39511b]	456	ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
[5cab7d3]	457	self.sizer2.Add(ctl2, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	458	ctl2.Hide()
	459	ix +=1
	460	# Units
	461	try:
[f39511b]	462	units = wx.StaticText(self, -1, self.model.details[item][0], style=wx.ALIGN_LEFT)
[5cab7d3]	463	except:
[f39511b]	464	units = wx.StaticText(self, -1, "", style=wx.ALIGN_LEFT)
[5cab7d3]	465	self.sizer2.Add(units, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[04edd0d]	466	else:
[08b9c6c8]	467	if not item in fixed:
	468	ip = 0
	469	iq += 1
	470	cb = wx.CheckBox(self, -1, item, (10, 10))
	471	cb.SetValue(False)
	472	self.sizer5.Add( cb,( iq, ip),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	473	wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
	474
	475	ip += 1
	476	value= self.model.getParam(item)
	477	ctl1 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
	478	ctl1.SetValue(str (format_number(value)))
	479	ctl1.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
	480	ctl1.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
	481	self.sizer5.Add(ctl1, (iq,ip),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	482	else:
	483	ik = 0
	484	text = wx.StaticText(self, -1, item, style=wx.ALIGN_LEFT)
	485	self.sizer6.Add(text,( im, ik),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	486
	487	ik += 1
	488	value= self.model.getParam(item)
	489	Tctl = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
	490	Tctl.SetValue(str (format_number(value)))
	491	Tctl.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
	492	Tctl.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
	493	self.sizer6.Add(Tctl, (im,ik),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	494	self.fixed_param.append([item, Tctl])
	495	im += 1
[04edd0d]	496	#save data
	497	self.parameters.append([cb,ctl1,text2,ctl2])
	498
[5cab7d3]	499	iy+=1
	500	self.sizer2.Add((20,20),(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
[04edd0d]	501
[d89f09b]	502	#Display units text on panel
	503	for item in keys:
	504	if self.model.details[item][0]!='':
	505	self.text2_4.Show()
	506	break
	507	else:
	508	self.text2_4.Hide()
	509	#Disable or enable fit button
[9d31a8b]	510
	511	if not (len(self.param_toFit ) >0):
[d89f09b]	512	self.xmin.Disable()
	513	self.xmax.Disable()
[9d31a8b]	514	else:
	515	self.xmin.Enable()
	516	self.xmax.Enable()
	517
[442895f]	518	self.compute_chisqr()
[d89f09b]	519	self.vbox.Layout()
[0aac36f]	520	self.SetScrollbars(20,20,55,40)
[dc317d1]	521	self.Layout()
[d89f09b]	522	self.GrandParent.GetSizer().Layout()
	523
[bcd6d51]	524
[d89f09b]	525
	526	def _onparamEnter(self,event):
	527	"""
	528	when enter value on panel redraw model according to changed
	529	"""
[1b07935d]	530	self.set_model_parameter()
[442895f]	531	self.compute_chisqr()
[0550752]	532
[442895f]	533
[1b07935d]	534	def set_model_parameter(self):
[442895f]	535	"""
	536	this method redraws the model according to parameters values changes
	537	and the reset model according to paramaters changes
	538	"""
[db709e4]	539	print "went here",len(self.parameters) ,self.model
[d89f09b]	540	if len(self.parameters) !=0 and self.model !=None:
[d15c0202]	541	# Flag to register when a parameter has changed.
	542	is_modified = False
[d89f09b]	543	for item in self.parameters:
	544	try:
[0550752]	545	item[2].Hide()
	546	item[3].Clear()
	547	item[3].Hide()
[2dbb681]	548	name=str(item[0].GetLabelText())
	549	value= float(item[1].GetValue())
[d15c0202]	550	# If the value of the parameter has changed,
	551	# update the model and set the is_modified flag
	552	if value != self.model.getParam(name):
[db709e4]	553	self.model.setParam(name,value)
[d15c0202]	554	is_modified = True
[08b9c6c8]	555	except:
[db709e4]	556	wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
[d15c0202]	557	"Drawing Error:wrong value entered : %s"% sys.exc_value))
	558	# Here we should check whether the boundaries have been modified.
	559	# If qmin and qmax have been modified, update qmin and qmax and
	560	# set the is_modified flag to True
	561	if float(self.xmin.GetValue()) != self.qmin:
	562	self.qmin = float(self.xmin.GetValue())
	563	is_modified = True
	564	if float(self.xmax.GetValue()) != self.qmax:
	565	self.qmax = float(self.xmax.GetValue())
	566	is_modified = True
[db709e4]	567
[d15c0202]	568	if is_modified:
	569	self.manager.redraw_model(
	570	qmin=self.qmin, qmax=self.qmax,
	571	)
	572	#self.manager.draw_model(self,model,description=None,
	573	# enable1D=True,qmin=None,qmax=None, qstep=None)
[db709e4]	574
[d89f09b]	575	def select_all_param(self,event):
	576	"""
	577	set to true or false all checkBox given the main checkbox value cb1
	578	"""
	579	self.param_toFit=[]
	580	if self.parameters !=[]:
	581	if self.cb1.GetValue()==True:
	582	for item in self.parameters:
	583	item[0].SetValue(True)
	584	list= [item[0],item[1],item[2],item[3]]
	585	self.param_toFit.append(list )
[9d31a8b]	586
	587	if not (len(self.param_toFit ) >0):
	588	self.xmin.Disable()
	589	self.xmax.Disable()
	590	else:
	591	self.xmin.Enable()
	592	self.xmax.Enable()
[d89f09b]	593	else:
	594	for item in self.parameters:
	595	item[0].SetValue(False)
	596	self.param_toFit=[]
	597
	598	self.xmin.Disable()
	599	self.xmax.Disable()
	600
	601
	602	def select_param(self,event):
	603	"""
	604	Select TextCtrl checked for fitting purpose and stores them
	605	in self.param_toFit=[] list
	606	"""
	607	self.param_toFit=[]
	608	for item in self.parameters:
	609	if item[0].GetValue()==True:
	610	list= [item[0],item[1],item[2],item[3]]
	611	self.param_toFit.append(list )
	612	else:
	613	if item in self.param_toFit:
	614	self.param_toFit.remove(item)
	615	if len(self.parameters)==len(self.param_toFit):
	616	self.cb1.SetValue(True)
	617	else:
	618	self.cb1.SetValue(False)
[08b9c6c8]	619
[9d31a8b]	620	if not (len(self.param_toFit ) >0):
[d89f09b]	621	self.xmin.Disable()
	622	self.xmax.Disable()
[9d31a8b]	623	else:
	624	self.xmin.Enable()
	625	self.xmax.Enable()
	626
[d89f09b]	627
	628
	629
	630	def onsetValues(self,chisqr, out,cov):
	631	"""
	632	Build the panel from the fit result
	633	@param chisqr:Value of the goodness of fit metric
	634	@param out:list of parameter with the best value found during fitting
	635	@param cov:Covariance matrix
	636
	637	"""
	638	#print "fitting : onsetvalues out",out
	639	self.tcChi.Clear()
	640	self.tcChi.SetValue(format_number(chisqr))
	641	params = {}
	642	is_modified = False
	643	has_error = False
	644	if out.__class__==numpy.float64:
	645	self.param_toFit[0][1].SetValue(format_number(out))
	646	self.param_toFit[0][1].Refresh()
	647	if cov !=None :
	648	self.text2_3.Show()
	649	self.param_toFit[0][2].Show()
	650	self.param_toFit[0][3].Clear()
	651	self.param_toFit[0][3].SetValue(format_number(cov[0]))
	652	self.param_toFit[0][3].Show()
	653	#out is a list : set parameters and errors in TextCtrl
	654	else:
	655	i=0
	656	#print "fitpage: list param model",list
	657	#for item in self.param_toFit:
	658	# print "fitpage: list display",item[0].GetLabelText()
	659	for item in self.param_toFit:
	660	if( out != None ) and len(out)<=len(self.param_toFit)and i < len(out):
[442895f]	661	#item[1].SetValue(format_number(out[i]))
	662	item[1].SetValue(format_number(self.model.getParam(item[0].GetLabelText())))
[d89f09b]	663	item[1].Refresh()
	664	if (cov !=None)and len(cov)<=len(self.param_toFit)and i < len(cov):
	665	self.text2_3.Show()
	666	item[2].Show()
	667	item[3].Clear()
	668	item[3].SetValue(format_number(cov[i]))
	669	item[3].Show()
	670	i+=1
	671
	672	self.vbox.Layout()
[0aac36f]	673	self.SetScrollbars(20,20,55,40)
	674	self.Layout()
[d89f09b]	675	self.GrandParent.GetSizer().Layout()
[a92d51b]	676
	677
[55e13ab]	678	def onSmear(self, event):
[08b9c6c8]	679	if event.GetString()=="None":
	680	self.manager.set_smearer(None)
	681
	682
	683	if event.GetString()=="GaussianModel":
[55e13ab]	684	from DataLoader.qsmearing import smear_selection
	685	smear =smear_selection( self.data )
[08b9c6c8]	686	self.manager.set_smearer(smear)
	687	print "on smearing"
	688

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

SasView

source: sasview/sansview/perspectives/fitting/fitpage1D.py @ 0aac36f

Download in other formats: