fitpage1D.py @ 00d083c

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

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

SasView

source: sasview/sansview/perspectives/fitting/fitpage1D.py @ 00d083c

Download in other formats: