modelpage.py @ 68dada4

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 68dada4 was 68dada4, checked in by Gervaise Alina <gervyh@…>, 15 years ago
refactoring panels
Property mode set to `100644`
File size: 17.8 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 += 1
76	self.sizer4.Add(wx.StaticText(self, -1, 'Max'),(iy, ix),(1,1),\
77	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
78	ix += 1
79	self.sizer4.Add(wx.StaticText(self, -1, 'Npts'),(iy, ix),(1,1),\
80	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
81	ix = 0
82	iy += 1
83	self.sizer4.Add(wx.StaticText(self, -1, 'x range'),(iy, ix),(1,1),\
84	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
85	## Q range
86	self.qmin= 0.001
87	self.qmax= 0.1
88	self.num_points= 100
89
90	ix += 1
91	self.xmin = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
92	self.xmin.SetValue(format_number(self.qmin))
93	self.xmin.SetToolTipString("Minimun value of x in linear scale.")
94	self.xmin.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
95	self.xmin.Bind(wx.EVT_TEXT_ENTER, self._onparamEnter)
96	self.sizer4.Add(self.xmin,(iy, ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
97
98
99	ix += 1
100	self.xmax = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
101	self.xmax.SetValue(format_number(self.qmax))
102	self.xmax.SetToolTipString("Maximum value of x in linear scale.")
103	self.xmax.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
104	self.xmax.Bind(wx.EVT_TEXT_ENTER, self._onparamEnter)
105	self.sizer4.Add(self.xmax,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
106	ix += 1
107	self.npts = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
108	self.npts.SetValue(format_number(self.num_points))
109	self.npts.SetToolTipString("Number of point to plot.")
110	self.npts.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
111	self.npts.Bind(wx.EVT_TEXT_ENTER, self._onparamEnter)
112
113	self.sizer4.Add(self.npts,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
114	ix = 0
115	iy += 1
116	self.sizer4.Add((20,20),(iy, ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
117	ix +=3
118	self.sizer4.Add( self.btClose,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
119	ix = 0
120	iy = 1
121	self.sizer3.Add(wx.StaticText(self,-1,'Model'),(iy,ix),(1,1)\
122	, wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
123	ix += 1
124	self.sizer3.Add(self.modelbox,(iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
125	#ix = 0
126	#iy += 1
127	ix += 1
128	id = wx.NewId()
129	self.model_view =wx.Button(self,id,'View 2D')
130	self.model_view.Bind(wx.EVT_BUTTON, self.onModel2D,id=id)
131	self.model_view.SetToolTipString("View model in 2D")
132	self.sizer3.Add(self.model_view,(iy,ix),(1,1),\
133	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
134	# contains link between model ,all its parameters, and panel organization
135	self.parameters=[]
136	#contains link between a model and selected parameters to fit
137	self.param_toFit=[]
138	# model on which the fit would be performed
139	self.model=model
140	try:
141	#print"init modelpage",model.name
142	self.set_panel(model)
143	except:
144	raise
145	# preview selected model name
146	self.prevmodel_name=name
147	#print "model view prev_model",name
148	self.modelbox.SetValue(self.prevmodel_name)
149	# flag to check if the user has selected a new model in the combox box
150	self.model_hasChanged=False
151	#dictionary of model name and model class
152	self.model_list_box={}
153
154	#enable model 2D draw
155	self.enable2D= False
156	# Data1D to make a deep comparison between 2 Data1D for checking data
157	#change
158	self.vbox.Layout()
159	self.vbox.Fit(self)
160
161	self.SetSizer(self.vbox)
162	self.SetScrollbars(20,20,55,40)
163
164	self.Centre()
165
166	def set_range(self, qmin, qmax, npts):
167	"""
168	Set the range for the plotted models
169	@param qmin: minimum Q
170	@param qmax: maximum Q
171	@param npts: number of Q bins
172	"""
173	# Set the data members
174	self.qmin = qmin
175	self.qmax = qmax
176	self.num_points = npts
177
178	# Set the controls
179	self.xmin.SetValue(format_number(self.qmin))
180	self.xmax.SetValue(format_number(self.qmax))
181	self.npts.SetValue(format_number(self.num_points))
182
183	def onClose(self,event):
184	""" close the page associated with this panel"""
185	self.GrandParent.onClose()
186
187	def set_owner(self,owner):
188	"""
189	set owner of fitpage
190	@param owner: the class responsible of plotting
191	"""
192	self.event_owner=owner
193
194
195	def set_manager(self, manager):
196	"""
197	set panel manager
198	@param manager: instance of plugin fitting
199	"""
200	self.manager = manager
201
202	def onModel2D(self, event):
203	"""
204	call manager to plot model in 2D
205	"""
206	# If the 2D display is not currently enabled, plot the model in 2D
207	# and set the enable2D flag.
208	if self.enable2D==False:
209	self.enable2D=True
210	self._draw_model()
211
212	else:
213	print "enable is true:",self.enable2D
214	#self.manager.parent. show_panel(147)
215	self.manager.show_panel2D( id=None )
216	#self.manager.menu1.Append(event_id, new_panel.window_caption,
217	# "Show %s plot panel" % new_panel.window_caption)
218
219	def populate_box(self, dict):
220	"""
221	Populate each combox box of each page
222	@param page: the page to populate
223	"""
224	id=0
225	self.model_list_box=dict
226	list_name=[]
227	for item in self.model_list_box.itervalues():
228	name = item.__name__
229	if hasattr(item, "name"):
230	name = item.name
231	list_name.append(name)
232	list_name.sort()
233	for name in list_name:
234	self.modelbox.Insert(name,int(id))
235	id+=1
236	wx.EVT_COMBOBOX(self.modelbox,-1, self._on_select_model)
237	return 0
238
239	def select_model(self, model, name):
240	"""
241	Select a new model
242	@param model: model object
243	"""
244	self.model= model
245	print "select_model", model.__class__
246	self.set_panel(model)
247	self._draw_model(name)
248
249	# Select the model from the combo box
250	items = self.modelbox.GetItems()
251	for i in range(len(items)):
252	print "model name",items[i],model.name, model.__class__.__name__
253	#if items[i]==model.__class__.__name__:
254	if items[i]==name:
255	self.modelbox.SetSelection(i)
256
257	def _on_select_model(self,event):
258	"""
259	react when a model is selected from page's combo box
260	post an event to its owner to draw an appropriate theory
261	"""
262	self.model_view.SetFocus()
263	for item in self.model_list_box.itervalues():
264	name = item.__name__
265	if hasattr(item, "name"):
266	name = item.name
267	if name ==event.GetString():
268	model=item()
269	self.model= model
270	self.set_panel(model)
271	self.name= name
272	#self.manager.draw_model(model, name)
273	self.enable2D=False
274	self._draw_model(name)
275
276
277	def get_model_box(self):
278	""" return reference to combox box self.model"""
279	return self.modelbox
280
281
282	def get_param_list(self):
283	"""
284	@return self.param_toFit: list containing references to TextCtrl
285	checked.Theses TextCtrl will allow reference to parameters to fit.
286	@raise: if return an empty list of parameter fit will nnote work
287	properly so raise ValueError,"missing parameter to fit"
288	"""
289	if self.param_toFit !=[]:
290	return self.param_toFit
291	else:
292	raise ValueError,"missing parameter to fit"
293
294
295	def set_panel(self,model):
296	"""
297	Build the panel from the model content
298	@param model: the model selected in combo box for fitting purpose
299	"""
300
301	self.sizer2.Clear(True)
302	self.sizer1.Clear(True)
303	self.sizer5.Clear(True)
304	self.parameters = []
305	self.param_toFit=[]
306	self.model = model
307	keys = self.model.getParamList()
308	keys.sort()
309	description=None
310	if hasattr(self.model,'description'):
311	description =model.description
312	disp_list=self.model.getDispParamList()
313	ip=0
314	iq=1
315	if len(disp_list)>0:
316	disp = wx.StaticText(self, -1, 'Dispersion')
317	self.sizer5.Add(disp,( iq, ip),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
318	ip += 1
319	values = wx.StaticText(self, -1, 'Values')
320	self.sizer5.Add(values,( iq, ip),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
321
322	disp_list.sort()
323	#print "went here",self.model.name,model.description
324	iy = 1
325	ix = 0
326	self.cb0 = wx.StaticText(self, -1,'Model Description:')
327	self.sizer1.Add(self.cb0,(iy, ix),(1,1),\
328	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
329	iy += 1
330
331	self.cb01 = wx.StaticText(self, -1,str(description),style=wx.ALIGN_LEFT)
332	self.cb01.Wrap(400)
333	#self.cb01 = wx.StaticText(self, -1,str(description),(45, 25),style=wx.ALIGN_LEFT)
334
335	self.sizer1.Add(self.cb01,(iy, ix),(1,1),\
336	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
337	ix = 0
338	iy = 1
339	self.cb1 = wx.StaticText(self, -1,'Parameters')
340	self.sizer2.Add(self.cb1,(iy, ix),(1,1),\
341	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
342	ix +=1
343	self.text2_2 = wx.StaticText(self, -1, 'Values')
344	self.sizer2.Add(self.text2_2,(iy, ix),(1,1),\
345	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
346	ix +=1
347	self.text2_4 = wx.StaticText(self, -1, 'Units')
348	self.sizer2.Add(self.text2_4,(iy, ix),(1,1),\
349	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
350	self.text2_4.Hide()
351	for item in keys:
352	if not item in disp_list:
353	iy += 1
354	ix = 0
355	cb=wx.StaticText(self, -1, item)
356	self.sizer2.Add( cb,( iy, ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
357	ix += 1
358	value= self.model.getParam(item)
359	ctl1 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
360	ctl1.SetValue(str (format_number(value)))
361	ctl1.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
362	ctl1.Bind(wx.EVT_TEXT_ENTER, self._onparamEnter)
363	self.sizer2.Add(ctl1, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
364	ix +=1
365
366	# Units
367	try:
368	units = wx.StaticText(self, -1, self.model.details[item][0], style=wx.ALIGN_LEFT)
369	except:
370	units = wx.StaticText(self, -1, "", style=wx.ALIGN_LEFT)
371	self.sizer2.Add(units, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
372	else:
373	ip = 0
374	iq += 1
375	cb=wx.StaticText(self, -1, item)
376	self.sizer5.Add( cb,( iq, ip),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
377	wx.EVT_CHECKBOX(self, cb.GetId(), self._on_select_model)
378
379	ip += 1
380	value= self.model.getParam(item)
381	ctl1 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
382	ctl1.SetValue(str (format_number(value)))
383	ctl1.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
384	ctl1.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
385	self.sizer5.Add(ctl1, (iq,ip),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
386	#save data
387	self.parameters.append([cb,ctl1])
388	iy+=1
389	self.sizer2.Add((20,20),(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
390
391	#Display units text on panel
392	for item in keys:
393	if self.model.details[item][0]!='':
394	self.text2_4.Show()
395	break
396	else:
397	self.text2_4.Hide()
398	self.vbox.Layout()
399	self.SetScrollbars(20,20,55,40)
400
401	self.GrandParent.GetSizer().Layout()
402
403
404	def _onparamEnter(self,event):
405	"""
406	when enter value on panel redraw model according to changed
407	"""
408	self.set_model_parameter()
409
410	def set_model_parameter(self):
411	if len(self.parameters) !=0 and self.model !=None:
412	# Flag to register when a parameter has changed.
413	is_modified = False
414	for item in self.parameters:
415	try:
416	name=str(item[0].GetLabelText())
417	value= float(item[1].GetValue())
418	# If the value of the parameter has changed,
419	# update the model and set the is_modified flag
420	if value != self.model.getParam(name):
421	self.model.setParam(name,value)
422	is_modified = True
423	except:
424	wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
425	"Model Drawing Error:wrong value entered : %s"% sys.exc_value))
426
427	# Here we should check whether the boundaries have been modified.
428	# If qmin and qmax have been modified, update qmin and qmax and
429	# set the is_modified flag to True
430	if float(self.xmin.GetValue()) != self.qmin:
431	self.qmin = float(self.xmin.GetValue())
432	is_modified = True
433	if float(self.xmax.GetValue()) != self.qmax:
434	self.qmax = float(self.xmax.GetValue())
435	is_modified = True
436
437	if float(self.npts.GetValue()) != self.num_points:
438	self.num_points = float(self.npts.GetValue())
439	is_modified = True
440
441	if is_modified:
442	self._draw_model()
443
444	def _draw_model(self, name=None):
445	"""
446	Method to draw or refresh a plotted model.
447	The method will use the data member from the model page
448	to build a call to the fitting perspective manager.
449
450	[Note to coder: This way future changes will be done in only one place.]
451	"""
452	if name==None:
453	name= self.model.name
454	self.manager.draw_model(self.model, name,
455	qmin=self.qmin, qmax=self.qmax,
456	qstep= self.num_points,
457	enable2D=self.enable2D)
458	"""
459	self.manager.draw_model(self.model, self.model.name,
460	qmin=self.qmin, qmax=self.qmax,
461	qstep= self.num_points,
462	enable2D=self.enable2D)
463	"""
464
465
466

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

SasView

source: sasview/sansview/perspectives/fitting/modelpage.py @ 68dada4

Download in other formats: