modelpage.py @ 08b9c6c8

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

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

SasView

source: sasview/sansview/perspectives/fitting/modelpage.py @ 08b9c6c8

Download in other formats: