modelpage.py @ a074145

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 a074145 was a074145, checked in by Gervaise Alina <gervyh@…>, 15 years ago
removing unsused code in fitpage
Property mode set to `100644`
File size: 17.4 KB

Line
1
2	import wx
3	import wx.lib.newevent
4	import numpy
5	import copy
6	import math
7	from sans.models.dispersion_models import ArrayDispersion, GaussianDispersion
8
9	from sans.guicomm.events import StatusEvent
10	from sans.guiframe.utils import format_number
11	(ModelEventbox, EVT_MODEL_BOX) = wx.lib.newevent.NewEvent()
12	_BOX_WIDTH = 80
13
14	import basepage
15	from basepage import BasicPage
16
17
18	class ModelPage(BasicPage):
19	"""
20	FitPanel class contains fields allowing to display results when
21	fitting a model and one data
22	@note: For Fit to be performed the user should check at least one parameter
23	on fit Panel window.
24
25	"""
26	def __init__(self,parent, page_info):
27	BasicPage.__init__(self, parent, page_info)
28	"""
29	Initialization of the Panel
30	"""
31	self._fill_model_sizer( self.sizer1)
32	self._fill_range_sizer()
33
34	description=""
35	if self.model!=None:
36	description = self.model.description
37
38	self.select_model(self.model, self.model.name)
39	self.set_model_description(description,self.sizer2)
40
41
42
43
44	def _on_display_description(self, event):
45	"""
46	Show or Hide description
47	@param event: wx.EVT_RADIOBUTTON
48	"""
49	self._on_display_description_helper()
50
51
52
53	def _on_display_description_helper(self):
54	"""
55	Show or Hide description
56	@param event: wx.EVT_RADIOBUTTON
57	"""
58
59	## Show description
60	if not self.description_show.GetValue():
61	self.sizer_description.Clear(True)
62
63	else:
64	description=""
65	if self.model!=None:
66	description = self.model.description
67
68	self.description = wx.StaticText( self,-1,str(description) )
69	self.sizer_description.Add( self.description, 1, wx.EXPAND \| wx.ALL, 10 )
70
71	self.Layout()
72
73
74	def _fill_range_sizer(self):
75	"""
76	Fill the sizer containing the plotting range
77	add access to npts
78	"""
79	sizer_npts= wx.GridSizer(1, 1,5, 5)
80
81	self.npts = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
82	self.npts.SetValue(format_number(self.num_points))
83	self.npts.SetToolTipString("Number of point to plot.")
84	self.npts.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
85	self.npts.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
86	self.npts.Bind(wx.EVT_TEXT_ENTER, self._onparamEnter)
87
88	sizer_npts.Add(wx.StaticText(self, -1, 'Npts'),1, wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 5)
89	sizer_npts.Add(self.npts,1, wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 5)
90	self._set_range_sizer( title="Plotted Q Range", object= sizer_npts)
91
92
93	def _on_select_model(self, event):
94	"""
95	call back for model selection
96	"""
97	self._on_select_model_helper()
98	self.select_model(self.model, self.model.name)
99
100
101	def _fill_model_sizer(self, sizer):
102	"""
103	fill sizer containing model info
104	"""
105	id = wx.NewId()
106	self.model_view =wx.Button(self,id,'View 2D')
107	self.model_view.Bind(wx.EVT_BUTTON, self._onModel2D,id=id)
108	self.model_view.SetToolTipString("View model in 2D")
109
110	## class base method to add view 2d button
111	self._set_model_sizer(sizer=sizer, title="Model",object= self.model_view )
112
113
114	def _set_sizer_gaussian(self):
115	"""
116	draw sizer with gaussian dispersity parameters
117	"""
118	self.fittable_param=[]
119	self.fixed_param=[]
120
121	self.sizer4_4.Clear(True)
122	if self.model==None:
123	##no model is selected
124	return
125	if not self.enable_disp.GetValue():
126	## the user didn't select dispersity display
127	return
128	self._reset_dispersity()
129	# Create the dispersion objects
130	for item in self.model.dispersion.keys():
131	disp_model = GaussianDispersion()
132	self._disp_obj_dict[item] = disp_model
133	self.model.set_dispersion(item, disp_model)
134
135
136	ix=0
137	iy=1
138	disp = wx.StaticText(self, -1, 'Names')
139	self.sizer4_4.Add(disp,( iy, ix),(1,1),
140	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
141	ix += 1
142	values = wx.StaticText(self, -1, 'Values')
143	self.sizer4_4.Add(values,( iy, ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
144
145	ix += 1
146	npts = wx.StaticText(self, -1, 'Npts')
147	self.sizer4_4.Add(npts,( iy, ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
148	ix += 1
149	nsigmas = wx.StaticText(self, -1, 'Nsigmas')
150	self.sizer4_4.Add(nsigmas,( iy, ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
151
152
153	for item in self.model.dispersion.keys():
154
155	self.disp_cb_dict[item]= None
156	name1=item+".width"
157	name2=item+".npts"
158	name3=item+".nsigmas"
159	iy += 1
160	for p in self.model.dispersion[item].keys():
161	if p=="width":
162	ix = 0
163	name = wx.StaticText(self, -1, name1)
164	self.sizer4_4.Add( name,( iy, ix),(1,1),
165	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
166	ix = 1
167	value= self.model.getParam(name1)
168	ctl1 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20),
169	style=wx.TE_PROCESS_ENTER)
170
171	ctl1.SetValue(str (format_number(value)))
172	ctl1.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
173	ctl1.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
174	ctl1.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
175	self.sizer4_4.Add(ctl1, (iy,ix),(1,1), wx.EXPAND)
176
177	self.fittable_param.append([None,name1,ctl1,None,
178	None, None, None,None])
179
180	elif p=="npts":
181	ix =2
182	value= self.model.getParam(name2)
183	Tctl1 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH/2,20),
184	style=wx.TE_PROCESS_ENTER)
185
186	Tctl1.SetValue(str (format_number(value)))
187	Tctl1.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
188	Tctl1.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
189	Tctl1.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
190	self.sizer4_4.Add(Tctl1, (iy,ix),(1,1),
191	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
192
193	self.fixed_param.append([None,name2, Tctl1,None,None,
194	None, None,None])
195
196	elif p=="nsigmas":
197
198	ix =3
199	value= self.model.getParam(name3)
200	Tctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH/2,20),
201	style=wx.TE_PROCESS_ENTER)
202	Tctl2.SetValue(str (format_number(value)))
203	Tctl2.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
204	Tctl2.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
205	Tctl2.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
206	self.sizer4_4.Add(Tctl2, (iy,ix),(1,1),
207	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
208	ix +=1
209	self.sizer4_4.Add((20,20), (iy,ix),(1,1),
210	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
211
212	self.fixed_param.append([None,name3, Tctl2,
213	None,None, None, None,None])
214
215	msg = " Selected Distribution: Gaussian"
216	wx.PostEvent(self.parent.parent, StatusEvent( status= msg ))
217	ix =0
218	iy +=1
219	self.sizer4_4.Add((20,20),(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
220	self.sizer4_4.Layout()
221	self.sizer4.Layout()
222	self.SetScrollbars(20,20,200,100)
223
224
225	def reset_page(self, state):
226	"""
227	reset the state
228	"""
229	self.reset_page_helper(state)
230
231
232	def _onModel2D(self, event):
233	"""
234	call manager to plot model in 2D
235	"""
236	# If the 2D display is not currently enabled, plot the model in 2D
237	# and set the enable2D flag.
238	if self.fitrange:
239	self.enable2D = True
240
241	if self.enable2D:
242	self._draw_model()
243	self.model_view.Disable()
244
245	def select_model(self, model, name):
246	"""
247	Select a new model
248	@param model: model object
249	"""
250	self.model = model
251	self.set_model_param_sizer(self.model)
252	self._draw_model()
253	## keep the sizer view consistent with the model menu selecting
254	self._set_model_sizer_selection( self.model )
255	self.enable_disp.SetValue(False)
256	self.disable_disp.SetValue(True)
257	self._set_dipers_Param(event=None)
258	self.model_view.SetFocus()
259
260
261	def set_model_description(self,description,sizer):
262	"""
263	fill a sizer with description
264	@param description: of type string
265	@param sizer: wx.BoxSizer()
266	"""
267
268	sizer.Clear(True)
269	box_description= wx.StaticBox(self, -1, 'Model Description')
270	boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
271
272	sizer_selection=wx.BoxSizer(wx.HORIZONTAL)
273
274	self.description_show = wx.RadioButton(self, -1, 'Show', style=wx.RB_GROUP)
275	self.description_hide = wx.RadioButton(self, -1, 'Hide')
276
277	if description=="":
278	self.description_hide.SetValue(True)
279	description=" Description unavailable. Click for details"
280
281	self.description = wx.StaticText( self,-1,str(description) )
282
283	self.Bind( wx.EVT_RADIOBUTTON, self._on_display_description,
284	id=self.description_hide.GetId() )
285
286	self.Bind( wx.EVT_RADIOBUTTON, self._on_display_description,
287	id=self.description_show.GetId() )
288
289	self.model_description = wx.Button(self,-1, label="More Details")
290	self.model_description.Bind(wx.EVT_BUTTON,self.on_button_clicked)
291	self.model_description.SetToolTipString("Click Model Functions in HelpWindow...")
292
293	sizer_selection.Add( self.description_show )
294	sizer_selection.Add( (20,20))
295	sizer_selection.Add( self.description_hide )
296	sizer_selection.Add( (20,20))
297	sizer_selection.Add( self.model_description )
298
299
300	self.sizer_description=wx.BoxSizer(wx.HORIZONTAL)
301	self.sizer_description.Add( self.description, 1, wx.EXPAND \| wx.ALL, 10 )
302
303	boxsizer1.Add( sizer_selection)
304	boxsizer1.Add( (20,20))
305	boxsizer1.Add( self.sizer_description)
306
307
308	sizer.Add(boxsizer1,0, wx.EXPAND \| wx.ALL, 10)
309	sizer.Layout()
310
311	def on_button_clicked(self,event):
312	"""
313	#On 'More details' button
314	"""
315
316	from helpPanel import HelpWindow
317	frame = HelpWindow(None, -1, 'HelpWindow')
318	frame.Show(True)
319
320	def set_range(self, qmin_x, qmax_x, npts):
321	"""
322	Set the range for the plotted models
323	@param qmin: minimum Q
324	@param qmax: maximum Q
325	@param npts: number of Q bins
326	"""
327	# Set the data members
328	self.qmin_x = qmin_x
329	self.qmax_x = qmax_x
330	self.num_points = npts
331
332	# Set the controls
333	self.qmin.SetValue(format_number(self.qmin_x))
334	self.qmax.SetValue(format_number(self.qmax_x))
335	self.npts.SetValue(format_number(self.num_points))
336
337
338	def set_model_param_sizer(self, model):
339	"""
340	Build the panel from the model content
341	@param model: the model selected in combo box for fitting purpose
342	"""
343	self.sizer3.Clear(True)
344	self.parameters = []
345	self.param_toFit=[]
346	self.fixed_param=[]
347
348	if model ==None:
349	##no model avaiable to draw sizer
350	return
351	box_description= wx.StaticBox(self, -1,str("Model Parameters"))
352	boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
353	sizer = wx.GridBagSizer(5,5)
354
355	self.model = model
356	self.set_model_description(self.model.description,self.sizer2)
357
358	keys = self.model.getParamList()
359	#list of dispersion paramaters
360	self.disp_list=self.model.getDispParamList()
361
362	keys.sort()
363
364	iy = 1
365	ix = 0
366	self.text1_2 = wx.StaticText(self, -1, 'Names')
367	sizer.Add(self.text1_2,(iy, ix),(1,1),\
368	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
369	ix +=1
370	self.text2_2 = wx.StaticText(self, -1, 'Values')
371	sizer.Add(self.text2_2,(iy, ix),(1,1),\
372	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
373	ix +=1
374	self.text2_4 = wx.StaticText(self, -1, 'Units')
375	sizer.Add(self.text2_4,(iy, ix),(1,1),\
376	wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
377	self.text2_4.Hide()
378
379
380	for item in keys:
381	if not item in self.disp_list:
382	iy += 1
383	ix = 0
384
385	name = wx.StaticText(self, -1,item)
386	sizer.Add( name,( iy, ix),(1,1),
387	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
388
389	ix += 1
390	value= self.model.getParam(item)
391	ctl1 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20),
392	style=wx.TE_PROCESS_ENTER)
393
394	ctl1.SetValue(str (format_number(value)))
395	ctl1.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
396	ctl1.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
397	ctl1.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
398
399	sizer.Add(ctl1, (iy,ix),(1,1), wx.EXPAND)
400
401	ix +=1
402	# Units
403	try:
404	units = wx.StaticText(self, -1, self.model.details[item][0], style=wx.ALIGN_LEFT)
405	except:
406	units = wx.StaticText(self, -1, "", style=wx.ALIGN_LEFT)
407	sizer.Add(units, (iy,ix),(1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
408
409	##[cb state, name, value, "+/-", error of fit, min, max , units]
410	self.parameters.append([None,item, ctl1,
411	None,None, None, None,None])
412
413	iy+=1
414	sizer.Add((20,20),(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
415
416	#Display units text on panel
417	for item in keys:
418	if self.model.details[item][0]!='':
419	self.text2_4.Show()
420	break
421	else:
422	self.text2_4.Hide()
423
424	boxsizer1.Add(sizer)
425
426	self.sizer3.Add(boxsizer1,0, wx.EXPAND \| wx.ALL, 10)
427	self.sizer3.Layout()
428	self.SetScrollbars(20,20,200,100)
429
430
431
432
433	class HelpWindow(wx.Frame):
434	def __init__(self, parent, id, title):
435	wx.Frame.__init__(self, parent, id, title, size=(570, 400))
436
437	from sans.models.CylinderModel import CylinderModel
438	model = CylinderModel()
439	#from sans.models.LineModel import LineModel
440	#model = LineModel()
441	from fitpanel import PageInfo
442	myinfo = PageInfo(self,model)
443	from models import ModelList
444	mylist= ModelList()
445
446	from sans.models.SphereModel import SphereModel
447	from sans.models.SquareWellStructure import SquareWellStructure
448	from sans.models.DebyeModel import DebyeModel
449	from sans.models.LineModel import LineModel
450	name= "shapes"
451	list1= [SphereModel]
452	mylist.set_list( name, list1)
453
454	name= "Shape-independent"
455	list1= [DebyeModel]
456	mylist.set_list( name, list1)
457
458	name= "Structure Factors"
459	list1= [SquareWellStructure]
460	mylist.set_list( name, list1)
461
462	name= "Added models"
463	list1= [LineModel]
464	mylist.set_list( name, list1)
465
466	myinfo.model_list_box = mylist.get_list()
467
468	self.page = ModelPage(self, myinfo)
469
470
471
472	self.Centre()
473	self.Show(True)
474
475
476
477	if __name__=="__main__":
478	app = wx.App()
479	HelpWindow(None, -1, 'HelpWindow')
480	app.MainLoop()
481

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

SasView

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

Download in other formats: