modelpage.py @ 86c1832

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 86c1832 was 86c1832, checked in by Gervaise Alina <gervyh@…>, 16 years ago
working on model plot
Property mode set to `100644`
File size: 14.3 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
83	qmin= 0.001
84	qmax= 1.0
85	ix += 1
86	self.xmin = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
87	self.xmin.SetValue(format_number(qmin))
88	self.xmin.SetToolTipString("Minimun value of x in linear scale.")
89	self.xmin.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
90	self.xmin.Bind(wx.EVT_TEXT_ENTER, self._onparamEnter)
91	self.sizer4.Add(self.xmin,(iy, ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
92
93
94	ix += 2
95	self.xmax = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
96	self.xmax.SetValue(format_number(qmax))
97	self.xmax.SetToolTipString("Maximum value of x in linear scale.")
98	self.xmax.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
99	self.xmax.Bind(wx.EVT_TEXT_ENTER, self._onparamEnter)
100
101	self.sizer4.Add(self.xmax,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
102	ix = 0
103	iy += 1
104	self.sizer4.Add((20,20),(iy, ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
105	ix +=3
106	self.sizer4.Add( self.btClose,(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
107	ix = 0
108	iy = 1
109	self.sizer3.Add(wx.StaticText(self,-1,'Model'),(iy,ix),(1,1)\
110	, wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
111	ix += 1
112	self.sizer3.Add(self.modelbox,(iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
113	#ix = 0
114	#iy += 1
115	ix += 1
116	id = wx.NewId()
117	self.model_view =wx.Button(self,id,'View 2D')
118	self.model_view.Bind(wx.EVT_BUTTON, self.onModel2D,id=id)
119	self.model_view.SetToolTipString("View model in 2D")
120	self.sizer3.Add(self.model_view,(iy,ix),(1,1),\
121	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
122	# contains link between model ,all its parameters, and panel organization
123	self.parameters=[]
124	#contains link between a model and selected parameters to fit
125	self.param_toFit=[]
126	# model on which the fit would be performed
127	self.model=model
128	try:
129	#print"init modelpage",model.name
130	self.set_panel(model)
131	except:
132	raise
133	# preview selected model name
134	self.prevmodel_name=name
135	self.modelbox.SetValue(self.prevmodel_name)
136	# flag to check if the user has selected a new model in the combox box
137	self.model_hasChanged=False
138	#dictionary of model name and model class
139	self.model_list_box={}
140
141	#enable model 2D draw
142	self.enable2D= False
143	# Data1D to make a deep comparison between 2 Data1D for checking data
144	#change
145	self.vbox.Layout()
146	self.vbox.Fit(self)
147
148	self.SetSizer(self.vbox)
149	self.SetScrollbars(20,20,55,40)
150	self.Centre()
151
152
153	def onClose(self,event):
154	""" close the page associated with this panel"""
155	self.GrandParent.onClose()
156
157	def set_owner(self,owner):
158	"""
159	set owner of fitpage
160	@param owner: the class responsible of plotting
161	"""
162	self.event_owner=owner
163
164
165	def set_manager(self, manager):
166	"""
167	set panel manager
168	@param manager: instance of plugin fitting
169	"""
170	self.manager = manager
171
172	def onModel2D(self, event):
173	"""
174	call manager to plot model in 2D
175	"""
176	self.enable2D=True
177	print "modelpage name", self.prevmodel_name
178	self.manager.draw_model(model=self.model,name=self.prevmodel_name,
179	description=None,
180	enable2D=self.enable2D,
181	qmin=float(self.xmin.GetValue()),
182	qmax=float(self.xmax.GetValue()))
183	def populate_box(self, dict):
184	"""
185	Populate each combox box of each page
186	@param page: the page to populate
187	"""
188	id=0
189	self.model_list_box=dict
190	list_name=[]
191	for item in self.model_list_box.itervalues():
192	name = item.__name__
193	if hasattr(item, "name"):
194	name = item.name
195	list_name.append(name)
196	list_name.sort()
197	for name in list_name:
198	self.modelbox.Insert(name,int(id))
199	id+=1
200	wx.EVT_COMBOBOX(self.modelbox,-1, self._on_select_model)
201	return 0
202
203
204
205	def _on_select_model(self,event):
206	"""
207	react when a model is selected from page's combo box
208	post an event to its owner to draw an appropriate theory
209	"""
210	#print "modelpage: self.model_list_box ",self.model_list_box
211	for item in self.model_list_box.itervalues():
212	name = item.__name__
213	if hasattr(item, "name"):
214	name = item.name
215	if name ==event.GetString():
216	model=item()
217	#print "fitpage: _on_select_model model name",name ,event.GetString()
218	self.model= model
219	self.set_panel(model)
220	print "name in model page", name,event.GetString()
221	self.name= name
222	self.manager.draw_model(model, name)
223
224
225
226	def get_model_box(self):
227	""" return reference to combox box self.model"""
228	return self.modelbox
229
230
231	def get_param_list(self):
232	"""
233	@return self.param_toFit: list containing references to TextCtrl
234	checked.Theses TextCtrl will allow reference to parameters to fit.
235	@raise: if return an empty list of parameter fit will nnote work
236	properly so raise ValueError,"missing parameter to fit"
237	"""
238	if self.param_toFit !=[]:
239	return self.param_toFit
240	else:
241	raise ValueError,"missing parameter to fit"
242
243
244	def set_panel(self,model):
245	"""
246	Build the panel from the model content
247	@param model: the model selected in combo box for fitting purpose
248	"""
249
250	self.sizer2.Clear(True)
251	self.sizer1.Clear(True)
252	self.sizer5.Clear(True)
253	self.parameters = []
254	self.param_toFit=[]
255	self.model = model
256	keys = self.model.getParamList()
257	keys.sort()
258	description=None
259	if hasattr(self.model,'description'):
260	description =model.description
261	disp_list=self.model.getDispParamList()
262	ip=0
263	iq=1
264	if len(disp_list)>0:
265	disp = wx.StaticText(self, -1, 'Dispersion')
266	self.sizer5.Add(disp,( iq, ip),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
267	ip += 1
268	values = wx.StaticText(self, -1, 'Values')
269	self.sizer5.Add(values,( iq, ip),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
270
271	disp_list.sort()
272	#print "went here",self.model.name,model.description
273	iy = 1
274	ix = 0
275	self.cb0 = wx.StaticText(self, -1,'Model Description:')
276	self.sizer1.Add(self.cb0,(iy, ix),(1,1),\
277	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
278	iy += 1
279
280	self.cb01 = wx.StaticText(self, -1,str(description),style=wx.ALIGN_LEFT)
281	self.cb01.Wrap(400)
282	#self.cb01 = wx.StaticText(self, -1,str(description),(45, 25),style=wx.ALIGN_LEFT)
283
284	self.sizer1.Add(self.cb01,(iy, ix),(1,1),\
285	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
286	ix = 0
287	iy = 1
288	self.cb1 = wx.StaticText(self, -1,'Parameters')
289	self.sizer2.Add(self.cb1,(iy, ix),(1,1),\
290	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
291	ix +=1
292	self.text2_2 = wx.StaticText(self, -1, 'Values')
293	self.sizer2.Add(self.text2_2,(iy, ix),(1,1),\
294	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
295	ix +=1
296	self.text2_4 = wx.StaticText(self, -1, 'Units')
297	self.sizer2.Add(self.text2_4,(iy, ix),(1,1),\
298	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
299	self.text2_4.Hide()
300	for item in keys:
301	if not item in disp_list:
302	iy += 1
303	ix = 0
304	cb=wx.StaticText(self, -1, item)
305	self.sizer2.Add( cb,( iy, ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
306	ix += 1
307	value= self.model.getParam(item)
308	ctl1 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
309	ctl1.SetValue(str (format_number(value)))
310	ctl1.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
311	ctl1.Bind(wx.EVT_TEXT_ENTER, self._onparamEnter)
312	self.sizer2.Add(ctl1, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
313	ix +=1
314
315	# Units
316	try:
317	units = wx.StaticText(self, -1, self.model.details[item][0], style=wx.ALIGN_LEFT)
318	except:
319	units = wx.StaticText(self, -1, "", style=wx.ALIGN_LEFT)
320	self.sizer2.Add(units, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
321	else:
322	ip = 0
323	iq += 1
324	cb = wx.CheckBox(self, -1, item, (10, 10))
325	cb.SetValue(False)
326	self.sizer5.Add( cb,( iq, ip),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
327	wx.EVT_CHECKBOX(self, cb.GetId(), self._on_select_model)
328
329	ip += 1
330	value= self.model.getParam(item)
331	ctl1 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
332	ctl1.SetValue(str (format_number(value)))
333	ctl1.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
334	ctl1.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
335	self.sizer5.Add(ctl1, (iq,ip),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
336	#save data
337	self.parameters.append([cb,ctl1])
338	iy+=1
339	self.sizer2.Add((20,20),(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
340
341	#Display units text on panel
342	for item in keys:
343	if self.model.details[item][0]!='':
344	self.text2_4.Show()
345	break
346	else:
347	self.text2_4.Hide()
348	self.vbox.Layout()
349	self.GrandParent.GetSizer().Layout()
350
351
352	def _onparamEnter(self,event):
353	"""
354	when enter value on panel redraw model according to changed
355	"""
356	self.set_model_parameter()
357
358	def set_model_parameter(self):
359	if len(self.parameters) !=0 and self.model !=None:
360	for item in self.parameters:
361	try:
362	name=str(item[0].GetLabelText())
363	value= float(item[1].GetValue())
364	self.model.setParam(name,value)
365	except:
366	wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
367	"Model Drawing Error:wrong value entered : %s"% sys.exc_value))
368
369	print self.enable2D
370	self.manager.draw_model(self.model,qmin=float(self.xmin.GetValue()),
371	qmax=float(self.xmax.GetValue()),
372	enable2D=self.enable2D)
373	#self.manager.draw_model(self,model,description=None,
374	# enable1D=True,qmin=None,qmax=None, qstep=None)
375
376
377
378
379

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

SasView

source: sasview/sansview/perspectives/fitting/modelpage.py @ 86c1832

Download in other formats: