modelpage.py @ 9d091ed

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

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

SasView

source: sasview/sansview/perspectives/fitting/modelpage.py @ 9d091ed

Download in other formats: