modelpage.py @ a5aaec9

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 a5aaec9 was a5aaec9, checked in by Gervaise Alina <gervyh@…>, 15 years ago
reverse set_model_paramater_ to the version on Mathieu
Property mode set to `100644`
File size: 15.4 KB

Line
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
27	class ModelPage(wx.ScrolledWindow):
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
41	def __init__(self, parent,model,name, args, *kwargs):
42	wx.ScrolledWindow.__init__(self, parent, args, *kwargs)
43	"""
44	Initialization of the Panel
45	"""
46	#self.scroll = wx.ScrolledWindow(self)
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)
53	self.sizer1 = wx.GridBagSizer(5,5)
54	self.sizer2 = wx.GridBagSizer(5,5)
55	self.sizer4 = wx.GridBagSizer(5,5)
56	self.sizer5 = wx.GridBagSizer(5,5)
57	self.static_line_1 = wx.StaticLine(self, -1)
58	self.modelbox = wx.ComboBox(self, -1)
59	id = wx.NewId()
60	self.vbox.Add(self.sizer3)
61	self.vbox.Add(self.sizer1)
62	self.vbox.Add(self.sizer2)
63	self.vbox.Add(self.static_line_1, 0, wx.EXPAND, 0)
64	self.vbox.Add(self.sizer5)
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.")
71	ix = 1
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)
82	## Q range
83	self.qmin= 0.001
84	self.qmax= 0.1
85
86	ix += 1
87	self.xmin = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
88	self.xmin.SetValue(format_number(self.qmin))
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)
93
94
95	ix += 2
96	self.xmax = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
97	self.xmax.SetValue(format_number(self.qmax))
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
107	self.sizer4.Add( self.btClose,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
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)
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")
121	self.sizer3.Add(self.model_view,(iy,ix),(1,1),\
122	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
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
129	try:
130	#print"init modelpage",model.name
131	self.set_panel(model)
132	except:
133	raise
134	# preview selected model name
135	self.prevmodel_name=name
136	print "model view prev_model",name
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={}
142
143	#enable model 2D draw
144	self.enable2D= False
145	# Data1D to make a deep comparison between 2 Data1D for checking data
146	#change
147	self.vbox.Layout()
148	self.vbox.Fit(self)
149
150	self.SetSizer(self.vbox)
151	self.SetScrollbars(20,20,55,40)
152	self.Centre()
153
154
155	def onClose(self,event):
156	""" close the page associated with this panel"""
157	self.GrandParent.onClose()
158
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
173
174	def onModel2D(self, event):
175	"""
176	call manager to plot model in 2D
177	"""
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
182	self.manager.draw_model(model=self.model,
183	name=self.model.name,
184	description=None,
185	enable2D=self.enable2D,
186	qmin=float(self.qmin),
187	qmax=float(self.qmax))
188
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
196	list_name=[]
197	for item in self.model_list_box.itervalues():
198	name = item.__name__
199	if hasattr(item, "name"):
200	name = item.name
201	list_name.append(name)
202	list_name.sort()
203	for name in list_name:
204	self.modelbox.Insert(name,int(id))
205	id+=1
206	wx.EVT_COMBOBOX(self.modelbox,-1, self._on_select_model)
207	return 0
208
209
210
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	"""
216	#print "modelpage: self.model_list_box ",self.model_list_box
217	for item in self.model_list_box.itervalues():
218	name = item.__name__
219	if hasattr(item, "name"):
220	name = item.name
221	if name ==event.GetString():
222	model=item()
223	#print "fitpage: _on_select_model model name",name ,event.GetString()
224	self.model= model
225	self.set_panel(model)
226	print "name in model page", name,event.GetString()
227	self.name= name
228	self.manager.draw_model(model, name)
229
230	self.model_view.SetFocus()
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
250	def set_panel(self,model):
251	"""
252	Build the panel from the model content
253	@param model: the model selected in combo box for fitting purpose
254	"""
255
256	self.sizer2.Clear(True)
257	self.sizer1.Clear(True)
258	self.sizer5.Clear(True)
259	self.parameters = []
260	self.param_toFit=[]
261	self.model = model
262	keys = self.model.getParamList()
263	keys.sort()
264	description=None
265	if hasattr(self.model,'description'):
266	description =model.description
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()
278	#print "went here",self.model.name,model.description
279	iy = 1
280	ix = 0
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)
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
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
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:
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)
342	#save data
343	self.parameters.append([cb,ctl1])
344	iy+=1
345	self.sizer2.Add((20,20),(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
346
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()
356
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:
366	# Flag to register when a parameter has changed.
367	is_modified = False
368	for item in self.parameters:
369	try:
370	name=str(item[0].GetLabelText())
371	value= float(item[1].GetValue())
372	# If the value of the parameter has changed,
373	# update the model and set the is_modified flag
374	if value != self.model.getParam(name):
375	self.model.setParam(name,value)
376	is_modified = True
377	except:
378	wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
379	"Model Drawing Error:wrong value entered : %s"% sys.exc_value))
380
381	# Here we should check whether the boundaries have been modified.
382	# If qmin and qmax have been modified, update qmin and qmax and
383	# set the is_modified flag to True
384	if float(self.xmin.GetValue()) != self.qmin:
385	self.qmin = float(self.xmin.GetValue())
386	is_modified = True
387	if float(self.xmax.GetValue()) != self.qmax:
388	self.qmax = float(self.xmax.GetValue())
389	is_modified = True
390
391	if is_modified:
392	self.manager.draw_model(self.model, self.model.name,
393	qmin=self.qmin, qmax=self.qmax,
394	enable2D=self.enable2D)
395	#self.manager.draw_model(self,model,description=None,
396	# enable1D=True,qmin=None,qmax=None, qstep=None)
397
398
399
400
401

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

SasView

source: sasview/sansview/perspectives/fitting/modelpage.py @ a5aaec9

Download in other formats: