basepage.py @ 920a6e5

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 920a6e5 was 920a6e5, checked in by Jae Cho <jhjcho@…>, 15 years ago
fixed park fit w/ polydispersion
Property mode set to `100644`
File size: 69.1 KB

Line
1
2	import sys, os
3	import wx
4	import numpy
5	import time
6	import copy
7	from sans.guiframe.utils import format_number,check_float
8	from sans.guicomm.events import StatusEvent
9	import pagestate
10	from pagestate import PageState
11	(PageInfoEvent, EVT_PAGE_INFO) = wx.lib.newevent.NewEvent()
12	(PreviousStateEvent, EVT_PREVIOUS_STATE) = wx.lib.newevent.NewEvent()
13	(NextStateEvent, EVT_NEXT_STATE) = wx.lib.newevent.NewEvent()
14	_BOX_WIDTH = 76
15
16	class BasicPage(wx.ScrolledWindow):
17	"""
18	This class provide general structure of fitpanel page
19	"""
20	## Internal name for the AUI manager
21	window_name = "Basic Page"
22	## Title to appear on top of the window
23	window_caption = "Basic page "
24
25	def __init__(self,parent, page_info):
26	wx.ScrolledWindow.__init__(self, parent,
27	style= wx.FULL_REPAINT_ON_RESIZE )
28	##window_name
29	self.window_name = page_info.window_name
30	##window_caption
31	self.window_caption = page_info.window_caption
32	## parent of the page
33	self.parent = parent
34	## manager is the fitting plugin
35	self.manager= page_info.manager
36	## owner of the page (fitting plugin)
37	self.event_owner= page_info.event_owner
38	## current model
39	self.model = page_info.model
40	## data
41	self.data = page_info.data
42	## dictionary containing list of models
43	self.model_list_box = page_info.model_list_box
44	## Data member to store the dispersion object created
45	self._disp_obj_dict = {}
46	## selected parameters to apply dispersion
47	self.disp_cb_dict ={}
48	## smearer object
49	self.smearer = None
50
51	##list of model parameters. each item must have same length
52	## each item related to a given parameters
53	##[cb state, name, value, "+/-", error of fit, min, max , units]
54	self.parameters=[]
55	## list of parameters to fit , must be like self.parameters
56	self.param_toFit=[]
57	## list of looking like parameters but with non fittable parameters info
58	self.fixed_param=[]
59	## list of looking like parameters but with fittable parameters info
60	self.fittable_param=[]
61	##list of dispersion parameters
62	self.disp_list=[]
63	self.disp_name=""
64	## list of orientation parameters
65	self.orientation_params=[]
66	self.orientation_params_disp=[]
67	if self.model !=None:
68	self.disp_list= self.model.getDispParamList()
69
70	##enable model 2D draw
71	self.enable2D= False
72	## check that the fit range is correct to plot the model again
73	self.fitrange= True
74	## Q range
75	self.qmin_x= 0.001
76	self.qmax_x= 0.13
77	self.num_points= 50
78
79	## Create memento to save the current state
80	self.state= PageState(parent= self.parent,model=self.model, data=self.data)
81	## flag to determine if state has change
82	self.state_change= False
83	## save customized array
84	self.values=[]
85	self.weights=[]
86	## retrieve saved state
87	self.number_saved_state= 0
88	## dictionary of saved state
89	self.saved_states={}
90
91	## Create context menu for page
92	self.popUpMenu = wx.Menu()
93	id = wx.NewId()
94	self._undo = wx.MenuItem(self.popUpMenu,id, "Undo","cancel the previous action")
95	self.popUpMenu.AppendItem(self._undo)
96	self._undo.Enable(False)
97	wx.EVT_MENU(self, id, self.onUndo)
98
99	id = wx.NewId()
100	self._redo = wx.MenuItem(self.popUpMenu,id,"Redo"," Restore the previous action")
101	self.popUpMenu.AppendItem(self._redo)
102	self._redo.Enable(False)
103	wx.EVT_MENU(self, id, self.onRedo)
104	self.popUpMenu.AppendSeparator()
105
106	id = wx.NewId()
107	self._keep = wx.MenuItem(self.popUpMenu,id,"Keep"," Keep the panel status to recall it later")
108	self.popUpMenu.AppendItem(self._keep)
109	self._keep.Enable(True)
110	wx.EVT_MENU(self, id, self.onSave)
111	self.popUpMenu.AppendSeparator()
112
113	## Default locations
114	self._default_save_location = os.getcwd()
115	## save initial state on context menu
116	#self.onSave(event=None)
117	self.Bind(wx.EVT_CONTEXT_MENU, self.onContextMenu)
118
119	## create the basic structure of the panel with empty sizer
120	self.define_page_structure()
121	## drawing Initial dispersion parameters sizer
122	self.set_dispers_sizer()
123	self._fill_save_sizer()
124	## layout
125	self.set_layout()
126
127	class ModelTextCtrl(wx.TextCtrl):
128	"""
129	Text control for model and fit parameters.
130	Binds the appropriate events for user interactions.
131	Default callback methods can be overwritten on initialization
132
133	@param kill_focus_callback: callback method for EVT_KILL_FOCUS event
134	@param set_focus_callback: callback method for EVT_SET_FOCUS event
135	@param mouse_up_callback: callback method for EVT_LEFT_UP event
136	@param text_enter_callback: callback method for EVT_TEXT_ENTER event
137	"""
138	def __init__(self, parent, id=-1, value=wx.EmptyString, pos=wx.DefaultPosition,
139	size=wx.DefaultSize, style=0, validator=wx.DefaultValidator, name=wx.TextCtrlNameStr,
140	kill_focus_callback = None, set_focus_callback = None,
141	mouse_up_callback = None, text_enter_callback = None):
142
143	wx.TextCtrl.__init__(self, parent, id, value, pos, size, style, validator, name)
144
145	# Bind appropriate events
146	self.Bind(wx.EVT_SET_FOCUS, parent.onSetFocus \
147	if set_focus_callback is None else set_focus_callback)
148	self.Bind(wx.EVT_KILL_FOCUS, parent._onparamEnter \
149	if kill_focus_callback is None else kill_focus_callback)
150	self.Bind(wx.EVT_TEXT_ENTER, parent._onparamEnter \
151	if text_enter_callback is None else text_enter_callback)
152	self.Bind(wx.EVT_LEFT_UP, parent._highlight_text \
153	if mouse_up_callback is None else mouse_up_callback)
154
155	def onContextMenu(self, event):
156	"""
157	Retrieve the state selected state
158	"""
159	# Skipping the save state functionality for release 0.9.0
160	#return
161
162	pos = event.GetPosition()
163	pos = self.ScreenToClient(pos)
164
165	self.PopupMenu(self.popUpMenu, pos)
166
167
168	def onUndo(self, event):
169	"""
170	Cancel the previous action
171	"""
172	#print "enable undo"
173	event = PreviousStateEvent(page = self)
174	wx.PostEvent(self.parent, event)
175
176
177	def onRedo(self, event):
178	"""
179	Restore the previous action cancelled
180	"""
181	print "enable redo"
182	event = NextStateEvent(page= self)
183	wx.PostEvent(self.parent, event)
184
185
186	def define_page_structure(self):
187	"""
188	Create empty sizer for a panel
189	"""
190	self.vbox = wx.BoxSizer(wx.VERTICAL)
191	self.sizer0 = wx.BoxSizer(wx.VERTICAL)
192	self.sizer1 = wx.BoxSizer(wx.VERTICAL)
193	self.sizer2 = wx.BoxSizer(wx.VERTICAL)
194	self.sizer3 = wx.BoxSizer(wx.VERTICAL)
195	self.sizer4 = wx.BoxSizer(wx.VERTICAL)
196	self.sizer5 = wx.BoxSizer(wx.VERTICAL)
197	self.sizer6 = wx.BoxSizer(wx.VERTICAL)
198
199	self.sizer0.SetMinSize((375,-1))
200	self.sizer1.SetMinSize((375,-1))
201	self.sizer2.SetMinSize((375,-1))
202	self.sizer3.SetMinSize((375,-1))
203	self.sizer4.SetMinSize((375,-1))
204	self.sizer5.SetMinSize((375,-1))
205	self.sizer6.SetMinSize((375,-1))
206
207	self.vbox.Add(self.sizer0)
208	self.vbox.Add(self.sizer1)
209	self.vbox.Add(self.sizer2)
210	self.vbox.Add(self.sizer3)
211	self.vbox.Add(self.sizer4)
212	self.vbox.Add(self.sizer5)
213	self.vbox.Add(self.sizer6)
214
215
216	def set_layout(self):
217	"""
218	layout
219	"""
220	self.vbox.Layout()
221	self.vbox.Fit(self)
222	self.SetSizer(self.vbox)
223
224	self.set_scroll()
225	self.Centre()
226
227
228	def set_scroll(self):
229	self.SetScrollbars(20,20,25,65)
230	self.Layout()
231	self.SetAutoLayout(True)
232
233
234	def set_owner(self,owner):
235	"""
236	set owner of fitpage
237	@param owner: the class responsible of plotting
238	"""
239	self.event_owner = owner
240	self.state.event_owner = owner
241
242	def set_manager(self, manager):
243	"""
244	set panel manager
245	@param manager: instance of plugin fitting
246	"""
247	self.manager = manager
248	self.state.manager = manager
249
250	def populate_box(self, dict):
251	"""
252	Store list of model
253	@param dict: dictionary containing list of models
254	"""
255	self.model_list_box = dict
256	self.state.model_list_box = self.model_list_box
257
258
259
260	def set_dispers_sizer(self):
261	"""
262	fill sizer containing dispersity info
263	"""
264	self.sizer4.Clear(True)
265	name="Polydispersity and Orientational Distribution"
266	box_description= wx.StaticBox(self, -1,name)
267	boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
268	#----------------------------------------------------
269	self.disable_disp = wx.RadioButton(self, -1, 'Off', (10, 10), style=wx.RB_GROUP)
270	self.enable_disp = wx.RadioButton(self, -1, 'On', (10, 30))
271
272
273	self.Bind(wx.EVT_RADIOBUTTON, self._set_dipers_Param, id=self.disable_disp.GetId())
274	self.Bind(wx.EVT_RADIOBUTTON, self._set_dipers_Param, id=self.enable_disp.GetId())
275
276	sizer_dispersion = wx.BoxSizer(wx.HORIZONTAL)
277	sizer_dispersion.Add((20,20))
278	name=""#Polydispersity and \nOrientational Distribution "
279	sizer_dispersion.Add(wx.StaticText(self,-1,name))
280	sizer_dispersion.Add(self.enable_disp )
281	sizer_dispersion.Add((20,20))
282	sizer_dispersion.Add(self.disable_disp )
283	sizer_dispersion.Add((10,10))
284
285	## fill a sizer with the combobox to select dispersion type
286	sizer_select_dispers = wx.BoxSizer(wx.HORIZONTAL)
287	self.model_disp = wx.StaticText(self, -1, 'Distribution Function ')
288
289	import sans.models.dispersion_models
290	self.polydisp= sans.models.dispersion_models.models
291	self.disp_box = wx.ComboBox(self, -1)
292
293	for key, value in self.polydisp.iteritems():
294	name = str(key)
295	self.disp_box.Append(name,value)
296	self.disp_box.SetStringSelection("gaussian")
297	wx.EVT_COMBOBOX(self.disp_box,-1, self._on_select_Disp)
298
299	sizer_select_dispers.Add((10,10))
300	sizer_select_dispers.Add(self.model_disp)
301	sizer_select_dispers.Add(self.disp_box,0,
302	wx.TOP\|wx.BOTTOM\|wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE,border=5)
303
304	self.model_disp.Hide()
305	self.disp_box.Hide()
306
307	boxsizer1.Add( sizer_dispersion,0,
308	wx.TOP\|wx.BOTTOM\|wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE,border=5)
309	#boxsizer1.Add( (10,10) )
310	boxsizer1.Add( sizer_select_dispers )
311	self.sizer4_4 = wx.GridBagSizer(5,5)
312	boxsizer1.Add( self.sizer4_4 )
313	#-----------------------------------------------------
314	self.sizer4.Add(boxsizer1,0, wx.EXPAND \| wx.ALL, 10)
315	self.sizer4_4.Layout()
316	self.sizer4.Layout()
317	self.Layout()
318	self.SetScrollbars(20,20,25,65)
319	self.Refresh()
320	## saving the state of enable dispersity button
321	self.state.enable_disp= self.enable_disp.GetValue()
322	self.state.disable_disp= self.disable_disp.GetValue()
323
324
325	def select_disp_angle(self, event):
326	"""
327	Event for when a user select a parameter to average over.
328	@param event: radiobutton event
329	"""
330	self.values=[]
331	self.weights=[]
332	# Go through the list of dispersion check boxes to identify which one has changed
333
334	for p in self.disp_cb_dict:
335	self.state.disp_cb_dict[p]= self.disp_cb_dict[p].GetValue()
336	# Catch which one of the box was just checked or unchecked.
337	if event.GetEventObject() == self.disp_cb_dict[p]:
338	if self.disp_cb_dict[p].GetValue() == True:
339
340	##Temp. FIX for V1.0 regarding changing checkbox to radiobutton.
341	##This (self._reset_dispersity) should be removed when the array dispersion is fixed.
342	self._reset_dispersity()
343
344	# The user wants this parameter to be averaged.
345	# Pop up the file selection dialog.
346	path = self._selectDlg()
347
348	# If nothing was selected, just return
349	if path is None:
350	self.disp_cb_dict[p].SetValue(False)
351	return
352	try:
353	self._default_save_location = os.path.dirname(path)
354	except:
355	pass
356	try:
357	self.values,self.weights = self.read_file(path)
358	except:
359	msg="Could not read input file"
360	wx.PostEvent(self.parent.parent, StatusEvent(status= msg))
361	return
362
363	# If any of the two arrays is empty, notify the user that we won't
364	# proceed
365	if self.values is None or self.weights is None or \
366	self.values ==[] or self.weights ==[]:
367	wx.PostEvent(self.parent.parent, StatusEvent(status=\
368	"The loaded %s distrubtion is corrupted or empty" % p))
369	return
370
371	# Tell the user that we are about to apply the distribution
372	wx.PostEvent(self.parent.parent, StatusEvent(status=\
373	"Applying loaded %s distribution: %s" % (p, path)))
374
375	# Create the dispersion objects
376	from sans.models.dispersion_models import ArrayDispersion
377	disp_model = ArrayDispersion()
378	disp_model.set_weights(self.values, self.weights)
379
380	# Store the object to make it persist outside the scope of this method
381	#TODO: refactor model to clean this up?
382	self._disp_obj_dict[p] = disp_model
383	self.state._disp_obj_dict [p]= disp_model
384	self.state.values=[]
385	self.state.weights=[]
386	self.state.values = copy.deepcopy(self.values)
387	self.state.weights = copy.deepcopy(self.weights)
388	# Set the new model as the dispersion object for the selected parameter
389	self.model.set_dispersion(p, disp_model)
390	# Store a reference to the weights in the model object so that
391	# it's not lost when we use the model within another thread.
392	#TODO: total hack - fix this
393	self.state.model= self.model.clone()
394	#if not hasattr(self.model, "_persistency_dict"):
395	self.model._persistency_dict = {}
396	self.model._persistency_dict[p] = [self.values, self.weights]
397	self.state.model._persistency_dict[p] = [self.values, self.weights]
398	else:
399	self._reset_dispersity()
400
401	## Redraw the model ???
402	self._draw_model()
403
404	## post state to fit panel
405	event = PageInfoEvent(page = self)
406	wx.PostEvent(self.parent, event)
407
408
409	def onResetModel(self, event):
410	"""
411	Reset model state
412	"""
413	## post help message for the selected model
414	msg = self.popUpMenu.GetHelpString(event.GetId())
415	msg +=" reloaded"
416	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
417
418	name= self.popUpMenu.GetLabel(event.GetId())
419	self._on_select_model_helper()
420
421	if name in self.saved_states.keys():
422	previous_state = self.saved_states[name]
423	## reset state of checkbox,textcrtl and regular parameters value
424	self.reset_page(previous_state)
425
426	def onSave(self, event):
427	"""
428	save history of the data and model
429	"""
430	if self.model==None:
431	return
432	if hasattr(self,"enable_disp"):
433	self.state.enable_disp = copy.deepcopy(self.enable_disp.GetValue())
434	if hasattr(self, "disp_box"):
435	self.state.disp_box = copy.deepcopy(self.disp_box.GetSelection())
436
437	self.state.model = self.model.clone()
438	self.state.model.name= self.model.name
439	#if not hasattr(self.model, "_persistency_dict"):
440	#self.model._persistency_dict = {}
441	#self.state.model._persistency_dict= copy.deepcopy(self.model._persistency_dict)
442
443	new_state = self.state.clone()
444	new_state.model.name = self.state.model.name
445
446	new_state.enable2D = copy.deepcopy(self.enable2D)
447	##Add model state on context menu
448	self.number_saved_state += 1
449	name= self.model.name+"[%g]"%self.number_saved_state
450	self.saved_states[name]= new_state
451
452	## Add item in the context menu
453
454	year, month, day,hour,minute,second,tda,ty,tm_isdst= time.localtime()
455	my_time= str(hour)+" : "+str(minute)+" : "+str(second)+" "
456	date= str( month)+"\|"+str(day)+"\|"+str(year)
457	msg= "Model saved at %s on %s"%(my_time, date)
458	## post help message for the selected model
459	msg +=" Saved! right click on this page to retrieve this model"
460	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
461
462	id = wx.NewId()
463	self.popUpMenu.Append(id,name,str(msg))
464	wx.EVT_MENU(self, id, self.onResetModel)
465
466
467	def onSetFocus(self, evt):
468	"""
469	highlight the current textcrtl and hide the error text control shown
470	after fitting
471	"""
472
473	if hasattr(self,"text2_3"):
474	self.text2_3.Hide()
475	if len(self.parameters)>0:
476	for item in self.parameters:
477	## hide statictext +/-
478	if item[3]!=None:
479	item[3].Hide()
480	## hide textcrtl for error after fit
481	if item[4]!=None:
482	item[4].Clear()
483	item[4].Hide()
484	if len(self.fittable_param)>0:
485	for item in self.fittable_param:
486	## hide statictext +/-
487	if item[3]!=None:
488	item[3].Hide()
489	## hide textcrtl for error after fit
490	if item[4]!=None:
491	item[4].Clear()
492	item[4].Hide()
493	self.Layout()
494	return
495
496	def _highlight_text(self, event):
497	"""
498	Highlight text of a TextCtrl only of no text has be selected
499	@param event: mouse event
500	"""
501	control = event.GetEventObject()
502	# Check that we have a TextCtrl
503	if issubclass(control.__class__, wx.TextCtrl):
504	# Check whether text has been selected,
505	# if not, select the whole string
506
507	(start, end) = control.GetSelection()
508	if start==end:
509	control.SetSelection(-1,-1)
510
511	# Make sure the mouse event is available to other listeners
512	event.Skip()
513
514
515	def read_file(self, path):
516	"""
517	Read two columns file
518	@param path: the path to the file to read
519	"""
520	try:
521	if path==None:
522	wx.PostEvent(self.parent.parent, StatusEvent(status=\
523	" Selected Distribution was not loaded: %s"%path))
524	return None, None
525	input_f = open(path, 'r')
526	buff = input_f.read()
527	lines = buff.split('\n')
528
529	angles = []
530	weights=[]
531	for line in lines:
532	toks = line.split()
533	try:
534	angle = float(toks[0])
535	weight = float(toks[1])
536	except:
537	# Skip non-data lines
538	pass
539	angles.append(angle)
540	weights.append(weight)
541	return numpy.array(angles), numpy.array(weights)
542	except:
543	raise
544
545
546	def createMemento(self):
547	"""
548	return the current state of the page
549	"""
550	return self.state.clone()
551
552
553	def save_current_state(self):
554	"""
555	Store current state
556	"""
557	## save model option
558	if self.model!= None:
559	self.disp_list= self.model.getDispParamList()
560	self.state.disp_list= copy.deepcopy(self.disp_list)
561	self.state.model = self.model.clone()
562
563	self.state.enable2D = copy.deepcopy(self.enable2D)
564	self.state.values= copy.deepcopy(self.values)
565	self.state.weights = copy.deepcopy( self.weights)
566	## save data
567	self.state.data= copy.deepcopy(self.data)
568	try:
569	n = self.disp_box.GetCurrentSelection()
570	dispersity= self.disp_box.GetClientData(n)
571	name= dispersity.__name__
572	self.disp_name = name
573	if name == "GaussianDispersion" :
574	if hasattr(self,"cb1"):
575	self.state.cb1= self.cb1.GetValue()
576	except:
577	pass
578
579	if hasattr(self,"enable_disp"):
580	self.state.enable_disp= self.enable_disp.GetValue()
581	self.state.disable_disp = self.disable_disp.GetValue()
582
583	self.state.smearer = copy.deepcopy(self.smearer)
584	if hasattr(self,"enable_smearer"):
585	self.state.enable_smearer = copy.deepcopy(self.enable_smearer.GetValue())
586	self.state.disable_smearer = copy.deepcopy(self.disable_smearer.GetValue())
587
588	if hasattr(self,"disp_box"):
589	self.state.disp_box = self.disp_box.GetCurrentSelection()
590
591	if len(self.disp_cb_dict)>0:
592	for k , v in self.disp_cb_dict.iteritems():
593
594	if v ==None :
595	self.state.disp_cb_dict[k]= v
596	else:
597	try:
598	self.state.disp_cb_dict[k]=v.GetValue()
599	except:
600	self.state.disp_cb_dict[k]= None
601
602	if len(self._disp_obj_dict)>0:
603	for k , v in self._disp_obj_dict.iteritems():
604
605	self.state._disp_obj_dict[k]= v
606
607
608	self.state.values = copy.deepcopy(self.values)
609	self.state.weights = copy.deepcopy(self.weights)
610	## save plotting range
611	self._save_plotting_range()
612
613	self.state.orientation_params =[]
614	self.state.orientation_params_disp =[]
615	self.state.parameters =[]
616	self.state.fittable_param =[]
617	self.state.fixed_param =[]
618
619
620	## save checkbutton state and txtcrtl values
621	self._copy_parameters_state(self.orientation_params,
622	self.state.orientation_params)
623	self._copy_parameters_state(self.orientation_params_disp,
624	self.state.orientation_params_disp)
625
626	self._copy_parameters_state(self.parameters, self.state.parameters)
627	self._copy_parameters_state(self.fittable_param, self.state.fittable_param)
628	self._copy_parameters_state(self.fixed_param, self.state.fixed_param)
629
630
631	def reset_page_helper(self, state):
632	"""
633	Use page_state and change the state of existing page
634	@precondition: the page is already drawn or created
635	@postcondition: the state of the underlying data change as well as the
636	state of the graphic interface
637	"""
638	if state ==None:
639	self._undo.Enable(False)
640	return
641
642	self.model= state.model
643	self.data = state.data
644	self.smearer= state.smearer
645	self.enable2D= state.enable2D
646
647	#???
648	self.disp_cb_dict = state.disp_cb_dict
649	self.disp_list =state.disp_list
650
651
652	## set the state of the radio box
653	self.shape_rbutton.SetValue(state.shape_rbutton )
654	self.shape_indep_rbutton.SetValue(state.shape_indep_rbutton)
655	self.struct_rbutton.SetValue(state.struct_rbutton )
656	self.plugin_rbutton.SetValue(state.plugin_rbutton)
657	##draw sizer containing model parameters value for the current model
658	self._set_model_sizer_selection( self.model )
659	self.set_model_param_sizer(self.model)
660
661	## reset value of combox box
662	self.structurebox.SetSelection(state.structurecombobox )
663	self.formfactorbox.SetSelection(state.formfactorcombobox)
664
665
666	## enable the view 2d button if this is a modelpage type
667	if hasattr(self,"model_view"):
668	if self.enable2D:
669	self.model_view.Disable()
670	else:
671	self.model_view.Enable()
672	## set the select all check box to the a given state
673	if hasattr(self, "cb1"):
674	self.cb1.SetValue(state.cb1)
675
676	## reset state of checkbox,textcrtl and regular parameters value
677
678	self._reset_parameters_state(self.orientation_params_disp,
679	state.orientation_params_disp)
680	self._reset_parameters_state(self.orientation_params,
681	state.orientation_params)
682	self._reset_parameters_state(self.parameters,state.parameters)
683	## display dispersion info layer
684	self.enable_disp.SetValue(state.enable_disp)
685	self.disable_disp.SetValue(state.disable_disp)
686
687	if hasattr(self, "disp_box"):
688
689	self.disp_box.SetSelection(state.disp_box)
690	n= self.disp_box.GetCurrentSelection()
691	dispersity= self.disp_box.GetClientData(n)
692	name= dispersity.__name__
693
694	self._set_dipers_Param(event=None)
695
696	#self.disp_cb_dict = {}
697	#for k,v in self.state.disp_cb_dict.iteritems():
698	# self.disp_cb_dict = copy.deepcopy(state.disp_cb_dict)
699	# self.state.disp_cb_dict = copy.deepcopy(state.disp_cb_dict)
700
701	if name=="ArrayDispersion":
702
703	for item in self.disp_cb_dict.keys():
704
705	if hasattr(self.disp_cb_dict[item],"SetValue") :
706	self.disp_cb_dict[item].SetValue(state.disp_cb_dict[item])
707	# Create the dispersion objects
708	from sans.models.dispersion_models import ArrayDispersion
709	disp_model = ArrayDispersion()
710	if hasattr(state,"values")and self.disp_cb_dict[item].GetValue()==True:
711	if len(state.values)>0:
712	self.values=state.values
713	self.weights=state.weights
714	disp_model.set_weights(self.values, state.weights)
715	else:
716	self._reset_dispersity()
717
718	self._disp_obj_dict[item] = disp_model
719	# Set the new model as the dispersion object for the selected parameter
720	self.model.set_dispersion(item, disp_model)
721
722	#self.model._persistency_dict = {}
723	self.model._persistency_dict[item] = [state.values, state.weights]
724
725	else:
726
727	for k,v in self.state.disp_cb_dict.iteritems():
728	self.disp_cb_dict = copy.deepcopy(state.disp_cb_dict)
729	self.state.disp_cb_dict = copy.deepcopy(state.disp_cb_dict)
730
731	#self._disp_obj_dict={}
732	#for k , v in self.state._disp_obj_dict.iteritems():
733	# self._disp_obj_dict[k]=v
734
735	##plotting range restore
736	self._reset_plotting_range(state)
737	## smearing info restore
738	if hasattr(self,"enable_smearer"):
739	## set smearing value whether or not the data contain the smearing info
740	self.enable_smearer.SetValue(state.enable_smearer)
741	self.disable_smearer.SetValue(state.disable_smearer)
742	self.tcChi.SetLabel(str( state.tcChi))
743	self.sizer5.Layout()
744	self.Layout()
745	self.Refresh()
746	#self.compute_chisqr(smearer= self.smearer)
747	#self._draw_model()
748	## reset state of checkbox,textcrtl and dispersity parameters value
749	self._reset_parameters_state(self.fittable_param,state.fittable_param)
750	self._reset_parameters_state(self.fixed_param,state.fixed_param)
751
752	## draw the model with previous parameters value
753	self._onparamEnter_helper()
754
755	## reset context menu items
756	self._reset_context_menu()
757
758	#self._reset_dispersity()
759	#self._set_dipers_Param(event=None) #to bo removed //resets disper para value
760
761	#self.disp_cb_dict = {}
762	#for k,v in self.state.disp_cb_dict.iteritems():
763	#self.disp_cb_dict = copy.deepcopy(state.disp_cb_dict)
764	#self.state.disp_cb_dict = copy.deepcopy(state.disp_cb_dict)
765
766
767	## set the value of the current state to the state given as parameter
768	self.state = state.clone()
769	self._draw_model()
770
771
772	def _selectDlg(self):
773	"""
774	open a dialog file to selected the customized dispersity
775	"""
776	import os
777	dlg = wx.FileDialog(self, "Choose a weight file",
778	self._default_save_location , "", ".", wx.OPEN)
779	path = None
780	if dlg.ShowModal() == wx.ID_OK:
781	path = dlg.GetPath()
782	dlg.Destroy()
783	return path
784
785
786	def _reset_context_menu(self):
787	"""
788	reset the context menu
789	"""
790	for name, state in self.state.saved_states.iteritems():
791	self.number_saved_state += 1
792	## Add item in the context menu
793	id = wx.NewId()
794	self.popUpMenu.Append(id,name, 'Save model and state %g'%self.number_saved_state)
795	wx.EVT_MENU(self, id, self.onResetModel)
796
797
798	def _reset_plotting_range(self, state):
799	"""
800	Reset the plotting range to a given state
801	"""
802	self.qmin.SetValue(str(state.qmin))
803	self.qmax.SetValue(str(state.qmax))
804	if self.state.npts!=None:
805	self.npts.SetValue(str(state.npts))
806
807
808	def _save_typeOfmodel(self):
809	"""
810	save radiobutton containing the type model that can be selected
811	"""
812	self.state.shape_rbutton = self.shape_rbutton.GetValue()
813	self.state.shape_indep_rbutton = self.shape_indep_rbutton.GetValue()
814	self.state.struct_rbutton = self.struct_rbutton.GetValue()
815	self.state.plugin_rbutton = self.plugin_rbutton.GetValue()
816	self.state.structurebox= self.structurebox.GetCurrentSelection()
817	self.state.formfactorbox = self.formfactorbox.GetCurrentSelection()
818
819	self._undo.Enable(True)
820	## post state to fit panel
821	event = PageInfoEvent(page = self)
822	wx.PostEvent(self.parent, event)
823
824
825	def _save_plotting_range(self ):
826	"""
827	save the state of plotting range
828	"""
829	self.state.qmin = self.qmin_x
830	self.state.qmax = self.qmax_x
831	if self.npts!=None:
832	self.state.npts= self.num_points
833
834
835	def _onparamEnter_helper(self):
836	"""
837	check if values entered by the user are changed and valid to replot
838	model
839	use : _check_value_enter
840	"""
841	self._undo.Enable(True)
842	if self.model !=None:
843
844
845	# Flag to register when a parameter has changed.
846	is_modified = False
847	is_modified =self._check_value_enter( self.fittable_param ,is_modified)
848	is_modified =self._check_value_enter( self.fixed_param ,is_modified)
849	is_modified =self._check_value_enter( self.parameters ,is_modified)
850
851	self.sizer3.Layout()
852	self.Layout()
853	self.Refresh()
854	# Here we should check whether the boundaries have been modified.
855	# If qmin and qmax have been modified, update qmin and qmax and
856	# set the is_modified flag to True
857	from sans.guiframe.utils import check_value
858	if check_value( self.qmin, self.qmax):
859	if float(self.qmin.GetValue()) != self.qmin_x:
860	self.qmin_x = float(self.qmin.GetValue())
861	is_modified = True
862	if float(self.qmax.GetValue()) != self.qmax_x:
863	self.qmax_x = float(self.qmax.GetValue())
864	is_modified = True
865	self.fitrange = True
866	else:
867	self.fitrange = False
868	if self.npts != None:
869	if check_float(self.npts):
870	if float(self.npts.GetValue()) != self.num_points:
871	self.num_points = float(self.npts.GetValue())
872	is_modified = True
873	else:
874	msg= "Cannot Plot :Must enter a number!!! "
875	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
876
877
878	## if any value is modify draw model with new value
879	if is_modified:
880	self.state_change= True
881	self._draw_model()
882	self.save_current_state()
883
884
885	def _reset_parameters_state(self, listtorestore,statelist):
886	"""
887	Reset the parameters at the given state
888	"""
889	if len(statelist)==0 or len(listtorestore)==0 :
890	return
891	if len(statelist)!= len(listtorestore) :
892	return
893	for j in range(len(listtorestore)):
894	item_page = listtorestore[j]
895	item_page_info = statelist[j]
896	##change the state of the check box for simple parameters
897	if item_page[0]!=None:
898	item_page[0].SetValue(item_page_info[0])
899
900	if item_page[2]!=None:
901	item_page[2].SetValue(item_page_info[2])
902
903	if item_page[3]!=None:
904	## show or hide text +/-
905	if item_page_info[2]:
906	item_page[3].Show(True)
907	else:
908	item_page[3].Hide()
909	if item_page[4]!=None:
910	## show of hide the text crtl for fitting error
911	if item_page_info[4][0]:
912	item_page[4].Show(True)
913	item_page[4].SetValue(item_page_info[4][1])
914	else:
915	item_page[3].Hide()
916	if item_page[5]!=None:
917	## show of hide the text crtl for fitting error
918	if item_page_info[5][0]:
919	item_page[5].Show(True)
920	item_page[5].SetValue(item_page_info[4][1])
921	else:
922	item_page[5].Hide()
923
924	if item_page[6]!=None:
925	## show of hide the text crtl for fitting error
926	if item_page_info[6][0]:
927	item_page[6].Show(True)
928	item_page[6].SetValue(item_page_info[6][1])
929	else:
930	item_page[6].Hide()
931
932
933	def _copy_parameters_state(self, listtocopy, statelist):
934	"""
935	copy the state of button
936	@param listtocopy: the list of check button to copy
937	@param statelist: list of state object to store the current state
938	"""
939	if len(listtocopy)==0:
940	return
941
942	for item in listtocopy:
943
944	checkbox_state = None
945	if item[0]!= None:
946	checkbox_state= item[0].GetValue()
947	parameter_name = item[1]
948	parameter_value = None
949	if item[2]!=None:
950	parameter_value = item[2].GetValue()
951	static_text = None
952	if item[3]!=None:
953	static_text = item[3].IsShown()
954	error_value = None
955	error_state = None
956	if item[4]!= None:
957	error_value = item[4].GetValue()
958	error_state = item[4].IsShown()
959
960	min_value = None
961	min_state = None
962	if item[5]!= None:
963	min_value = item[5].GetValue()
964	min_state = item[5].IsShown()
965
966	max_value = None
967	max_state = None
968	if item[6]!= None:
969	max_value = item[6].GetValue()
970	max_state = item[6].IsShown()
971	unit=None
972	if item[7]!=None:
973	unit = item[7].GetLabel()
974
975	statelist.append([checkbox_state, parameter_name, parameter_value,
976	static_text ,[error_state,error_value],
977	[min_state,min_value],[max_state , max_value],unit])
978
979
980
981	def _set_model_sizer_selection(self, model):
982	"""
983	Display the sizer according to the type of the current model
984	"""
985	if model ==None:
986	return
987	if hasattr(model ,"s_model"):
988
989	class_name= model.s_model.__class__
990	name= model.s_model.name
991	flag= name != "NoStructure"
992	if flag and (class_name in self.model_list_box["Structure Factors"]):
993	self.structurebox.Show()
994	self.text2.Show()
995	self.structurebox.Enable()
996	self.text2.Enable()
997	items = self.structurebox.GetItems()
998	self.sizer1.Layout()
999	self.SetScrollbars(20,20,25,65)
1000	for i in range(len(items)):
1001	if items[i]== str(name):
1002	self.structurebox.SetSelection(i)
1003	break
1004
1005	if hasattr(model ,"p_model"):
1006	class_name = model.p_model.__class__
1007	name = model.p_model.name
1008	self.formfactorbox.Clear()
1009
1010	for k, list in self.model_list_box.iteritems():
1011	if k in["P(Q)*S(Q)","Shapes" ] and class_name in self.model_list_box["Shapes"]:
1012	self.shape_rbutton.SetValue(True)
1013	## fill the form factor list with new model
1014	self._populate_box(self.formfactorbox,self.model_list_box["Shapes"])
1015	items = self.formfactorbox.GetItems()
1016	## set comboxbox to the selected item
1017	for i in range(len(items)):
1018	if items[i]== str(name):
1019	self.formfactorbox.SetSelection(i)
1020	break
1021	return
1022	elif k == "Shape-Independent":
1023	self.shape_indep_rbutton.SetValue(True)
1024	elif k == "Structure Factors":
1025	self.struct_rbutton.SetValue(True)
1026	else:
1027	self.plugin_rbutton.SetValue(True)
1028
1029	if class_name in list:
1030	## fill the form factor list with new model
1031	self._populate_box(self.formfactorbox, list)
1032	items = self.formfactorbox.GetItems()
1033	## set comboxbox to the selected item
1034	for i in range(len(items)):
1035	if items[i]== str(name):
1036	self.formfactorbox.SetSelection(i)
1037	break
1038	break
1039	else:
1040
1041	## Select the model from the menu
1042	class_name = model.__class__
1043	name = model.name
1044	self.formfactorbox.Clear()
1045	items = self.formfactorbox.GetItems()
1046
1047	for k, list in self.model_list_box.iteritems():
1048	if k in["P(Q)*S(Q)","Shapes" ] and class_name in self.model_list_box["Shapes"]:
1049	if class_name in self.model_list_box["P(Q)*S(Q)"]:
1050	self.structurebox.Show()
1051	self.text2.Show()
1052	self.structurebox.Enable()
1053	self.structurebox.SetSelection(0)
1054	self.text2.Enable()
1055	else:
1056	self.structurebox.Hide()
1057	self.text2.Hide()
1058	self.structurebox.Disable()
1059	self.structurebox.SetSelection(0)
1060	self.text2.Disable()
1061
1062	self.shape_rbutton.SetValue(True)
1063	## fill the form factor list with new model
1064	self._populate_box(self.formfactorbox,self.model_list_box["Shapes"])
1065	items = self.formfactorbox.GetItems()
1066	## set comboxbox to the selected item
1067	for i in range(len(items)):
1068	if items[i]== str(name):
1069	self.formfactorbox.SetSelection(i)
1070	break
1071	return
1072	elif k == "Shape-Independent":
1073	self.shape_indep_rbutton.SetValue(True)
1074	elif k == "Structure Factors":
1075	self.struct_rbutton.SetValue(True)
1076	else:
1077	self.plugin_rbutton.SetValue(True)
1078	if class_name in list:
1079	self.structurebox.SetSelection(0)
1080	self.structurebox.Disable()
1081	self.text2.Disable()
1082	## fill the form factor list with new model
1083	self._populate_box(self.formfactorbox, list)
1084	items = self.formfactorbox.GetItems()
1085	## set comboxbox to the selected item
1086	for i in range(len(items)):
1087	if items[i]== str(name):
1088	self.formfactorbox.SetSelection(i)
1089	break
1090	break
1091
1092
1093	def _draw_model(self):
1094	"""
1095	Method to draw or refresh a plotted model.
1096	The method will use the data member from the model page
1097	to build a call to the fitting perspective manager.
1098
1099	[Note to coder: This way future changes will be done in only one place.]
1100	"""
1101	if self.model !=None:
1102	temp_smear=None
1103	if hasattr(self, "enable_smearer"):
1104	if self.enable_smearer.GetValue():
1105	temp_smear= self.smearer
1106
1107	self.manager.draw_model(self.model, data=self.data,
1108	smearer= temp_smear,
1109	qmin=float(self.qmin_x), qmax=float(self.qmax_x),
1110	qstep= float(self.num_points),
1111	enable2D=self.enable2D)
1112
1113
1114	def _set_model_sizer(self, sizer, title="", object=None):
1115	"""
1116	Use lists to fill a sizer for model info
1117	"""
1118
1119	sizer.Clear(True)
1120	box_description= wx.StaticBox(self, -1,str(title))
1121	boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
1122	#--------------------------------------------------------
1123	self.shape_rbutton = wx.RadioButton(self, -1, 'Shapes', style=wx.RB_GROUP)
1124	self.shape_indep_rbutton = wx.RadioButton(self, -1, "Shape-Independent")
1125	self.struct_rbutton = wx.RadioButton(self, -1, "Structure Factor ")
1126	self.plugin_rbutton = wx.RadioButton(self, -1, "Customized Models")
1127
1128	self.Bind( wx.EVT_RADIOBUTTON, self._show_combox,
1129	id= self.shape_rbutton.GetId() )
1130	self.Bind( wx.EVT_RADIOBUTTON, self._show_combox,
1131	id= self.shape_indep_rbutton.GetId() )
1132	self.Bind( wx.EVT_RADIOBUTTON, self._show_combox,
1133	id= self.struct_rbutton.GetId() )
1134	self.Bind( wx.EVT_RADIOBUTTON, self._show_combox,
1135	id= self.plugin_rbutton.GetId() )
1136
1137
1138	sizer_radiobutton = wx.GridSizer(2, 2,5, 5)
1139	sizer_radiobutton.Add(self.shape_rbutton)
1140	sizer_radiobutton.Add(self.shape_indep_rbutton)
1141	sizer_radiobutton.Add(self.plugin_rbutton)
1142	sizer_radiobutton.Add(self.struct_rbutton)
1143
1144	sizer_selection = wx.BoxSizer(wx.HORIZONTAL)
1145
1146	self.text1 = wx.StaticText( self,-1,"" )
1147	self.text2 = wx.StaticText( self,-1,"P(Q)*S(Q)" )
1148
1149
1150	self.formfactorbox = wx.ComboBox(self, -1,style=wx.CB_READONLY)
1151	if self.model!=None:
1152	self.formfactorbox.SetValue(self.model.name)
1153
1154
1155	self.structurebox = wx.ComboBox(self, -1,style=wx.CB_READONLY)
1156	wx.EVT_COMBOBOX(self.formfactorbox,-1, self._on_select_model)
1157	wx.EVT_COMBOBOX(self.structurebox,-1, self._on_select_model)
1158
1159
1160	## fill combox box
1161	if len(self.model_list_box)>0:
1162	self._populate_box( self.formfactorbox,self.model_list_box["Shapes"])
1163
1164	if len(self.model_list_box)>0:
1165	self._populate_box( self.structurebox,
1166	self.model_list_box["Structure Factors"])
1167	self.structurebox.Insert("None", 0,None)
1168	self.structurebox.SetSelection(0)
1169	self.structurebox.Hide()
1170	self.text2.Hide()
1171	self.structurebox.Disable()
1172	self.text2.Disable()
1173
1174	if self.model.__class__ in self.model_list_box["P(Q)*S(Q)"]:
1175	self.structurebox.Show()
1176	self.text2.Show()
1177	self.structurebox.Enable()
1178	self.text2.Enable()
1179
1180	## check model type to show sizer
1181	if self.model !=None:
1182	self._set_model_sizer_selection( self.model )
1183
1184	sizer_selection.Add(self.text1)
1185	sizer_selection.Add((5,5))
1186	sizer_selection.Add(self.formfactorbox)
1187	sizer_selection.Add((5,5))
1188	sizer_selection.Add(self.text2)
1189	sizer_selection.Add((5,5))
1190	sizer_selection.Add(self.structurebox)
1191	sizer_selection.Add((5,5))
1192
1193	boxsizer1.Add( sizer_radiobutton )
1194	boxsizer1.Add( (20,20))
1195	boxsizer1.Add( sizer_selection )
1196	if object !=None:
1197	boxsizer1.Add( (-72,-72))
1198	boxsizer1.Add( object, 0, wx.ALIGN_RIGHT\| wx.RIGHT, 10)
1199	boxsizer1.Add( (60,60))
1200	#--------------------------------------------------------
1201	sizer.Add(boxsizer1,0, wx.EXPAND \| wx.ALL, 10)
1202	sizer.Layout()
1203	self.SetScrollbars(20,20,25,65)
1204
1205
1206	def _show_combox(self, event):
1207	"""
1208	Show combox box associate with type of model selected
1209	"""
1210	## Don't want to populate combo box again if the event comes from check box
1211	if self.shape_rbutton.GetValue()and\
1212	event.GetEventObject()==self.shape_rbutton:
1213	##fill the combobox with form factor list
1214	self.structurebox.SetSelection(0)
1215	self.structurebox.Disable()
1216	self.formfactorbox.Clear()
1217	self._populate_box( self.formfactorbox,self.model_list_box["Shapes"])
1218
1219	if self.shape_indep_rbutton.GetValue()and\
1220	event.GetEventObject()==self.shape_indep_rbutton:
1221	##fill the combobox with shape independent factor list
1222	self.structurebox.SetSelection(0)
1223	self.structurebox.Disable()
1224	self.formfactorbox.Clear()
1225	self._populate_box( self.formfactorbox,
1226	self.model_list_box["Shape-Independent"])
1227
1228	if self.struct_rbutton.GetValue() and\
1229	event.GetEventObject()==self.struct_rbutton:
1230	##fill the combobox with structure factor list
1231	self.structurebox.SetSelection(0)
1232	self.structurebox.Disable()
1233	self.formfactorbox.Clear()
1234	self._populate_box( self.formfactorbox,
1235	self.model_list_box["Structure Factors"])
1236
1237	if self.plugin_rbutton.GetValue()and\
1238	event.GetEventObject()==self.plugin_rbutton:
1239
1240	##fill the combobox with form factor list
1241	self.structurebox.Disable()
1242	self.formfactorbox.Clear()
1243	self._populate_box( self.formfactorbox,
1244	self.model_list_box["Customized Models"])
1245
1246	self._on_select_model(event=None)
1247	self._save_typeOfmodel()
1248	self.sizer4_4.Layout()
1249	self.sizer4.Layout()
1250	self.Layout()
1251	self.Refresh()
1252	self.SetScrollbars(20,20,25,65)
1253
1254	def _populate_box(self, combobox, list):
1255	"""
1256	fill combox box with dict item
1257	@param list: contains item to fill the combox
1258	item must model class
1259	"""
1260	for models in list:
1261	model= models()
1262	name = model.__class__.__name__
1263	if models.__name__!="NoStructure":
1264	if hasattr(model, "name"):
1265	name = model.name
1266	combobox.Append(name,models)
1267
1268	return 0
1269
1270
1271	def _on_select_model_helper(self):
1272	"""
1273	call back for model selection
1274	"""
1275	## reset dictionary containing reference to dispersion
1276	self._disp_obj_dict = {}
1277	self.disp_cb_dict ={}
1278	f_id = self.formfactorbox.GetCurrentSelection()
1279	form_factor = self.formfactorbox.GetClientData( f_id )
1280	if not form_factor in self.model_list_box["multiplication"]:
1281	self.structurebox.Hide()
1282	self.text2.Hide()
1283	self.structurebox.Disable()
1284	self.structurebox.SetSelection(0)
1285	self.text2.Disable()
1286	else:
1287	self.structurebox.Show()
1288	self.text2.Show()
1289	self.structurebox.Enable()
1290	self.text2.Enable()
1291
1292	s_id = self.structurebox.GetCurrentSelection()
1293	struct_factor = self.structurebox.GetClientData( s_id )
1294
1295	if struct_factor !=None:
1296	from sans.models.MultiplicationModel import MultiplicationModel
1297	self.model= MultiplicationModel(form_factor(),struct_factor())
1298
1299	else:
1300	if form_factor != None:
1301	self.model= form_factor()
1302	else:
1303	self.model = None
1304	return self.model
1305
1306	## post state to fit panel
1307	self.state.parameters =[]
1308	self.state.model =self.model
1309
1310	self.disp_list =self.model.getDispParamList()
1311	self.state.disp_list = self.disp_list
1312	self.sizer4_4.Layout()
1313	self.sizer4.Layout()
1314	self.Layout()
1315	self.SetScrollbars(20,20,25,65)
1316	self.Refresh()
1317
1318
1319
1320	def _onparamEnter(self,event):
1321	"""
1322	when enter value on panel redraw model according to changed
1323	"""
1324	tcrtl= event.GetEventObject()
1325	## save current state
1326	self.save_current_state()
1327	if event !=None:
1328	self._undo.Enable(True)
1329	event = PageInfoEvent(page = self)
1330	wx.PostEvent(self.parent, event)
1331
1332	if check_float(tcrtl):
1333
1334	self._onparamEnter_helper()
1335	else:
1336	msg= "Cannot Plot :Must enter a number!!! "
1337	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
1338	return
1339
1340
1341	def _check_value_enter(self, list, modified):
1342	"""
1343	@param list: model parameter and panel info
1344	each item of the list should be as follow:
1345	item=[cb state, name, value, "+/-", error of fit, min, max , units]
1346	"""
1347	is_modified = modified
1348	if len(list)==0:
1349	return is_modified
1350	for item in list:
1351	try:
1352	name = str(item[1])
1353	if hasattr(self,"text2_3"):
1354	self.text2_3.Hide()
1355	## check model parameters range
1356	## check minimun value
1357	param_min= None
1358	param_max= None
1359	if item[5]!= None:
1360	if item[5].GetValue().lstrip().rstrip()!="":
1361	param_min = float(item[5].GetValue())
1362
1363	## check maximum value
1364	if item[6]!= None:
1365	if item[6].GetValue().lstrip().rstrip()!="":
1366	param_max = float(item[6].GetValue())
1367
1368	from sans.guiframe.utils import check_value
1369	if param_min != None and param_max !=None:
1370	if not check_value(item[5], item[6]):
1371	msg= "Wrong Fit range entered for parameter "
1372	msg+= "name %s of model %s "%(name, self.model.name)
1373	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
1374	if name in self.model.details.keys():
1375	self.model.details[name][1:]= param_min,param_max
1376
1377	## hide statictext +/-
1378	if item[3]!=None:
1379	item[3].Hide()
1380	## hide textcrtl for error after fit
1381	if item[4]!=None:
1382	item[4].Clear()
1383	item[4].Hide()
1384
1385	value= float(item[2].GetValue())
1386
1387	# If the value of the parameter has changed,
1388	# +update the model and set the is_modified flag
1389	if value != self.model.getParam(name):
1390	self.model.setParam(name,value)
1391	is_modified = True
1392
1393	except:
1394	msg= "Model Drawing Error:wrong value entered : %s"% sys.exc_value
1395	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
1396	return
1397
1398	return is_modified
1399
1400
1401	def _set_dipers_Param(self, event):
1402	"""
1403	respond to self.enable_disp and self.disable_disp radio box.
1404	The dispersity object is reset inside the model into Gaussian.
1405	When the user select yes , this method display a combo box for more selection
1406	when the user selects No,the combo box disappears.
1407	Redraw the model with the default dispersity (Gaussian)
1408	"""
1409
1410	self._reset_dispersity()
1411
1412	if self.model ==None:
1413	self.model_disp.Hide()
1414	self.disp_box.Hide()
1415	self.sizer4_4.Clear(True)
1416
1417	return
1418	#remove previous added details for dispersion parameters.
1419	for item in self.model.details.keys():
1420	if item in self.model.fixed:
1421	del self.model.details [item]
1422
1423	if self.enable_disp.GetValue():
1424	self.model_disp.Show(True)
1425	self.disp_box.Show(True)
1426	## layout for model containing no dispersity parameters
1427
1428	self.disp_list= self.model.getDispParamList()
1429
1430	if len(self.disp_list)==0 and len(self.disp_cb_dict)==0:
1431	self._layout_sizer_noDipers()
1432	else:
1433	## set gaussian sizer
1434	self._on_select_Disp(event=None)
1435	else:
1436	self.model_disp.Hide()
1437	self.disp_box.Hide()
1438	self.disp_box.SetSelection(0)
1439	self.sizer4_4.Clear(True)
1440
1441	## post state to fit panel
1442	self.save_current_state()
1443	if event !=None:
1444	self._undo.Enable(True)
1445	event = PageInfoEvent(page = self)
1446	wx.PostEvent(self.parent, event)
1447	#draw the model with the current dispersity
1448	self._draw_model()
1449	self.sizer4_4.Layout()
1450	self.sizer4.Layout()
1451	self.Layout()
1452	self.Refresh()
1453	self.SetScrollbars(20,20,25,65)
1454
1455
1456	def _layout_sizer_noDipers(self):
1457	"""
1458	Draw a sizer with no dispersity info
1459	"""
1460	ix=0
1461	iy=1
1462	self.fittable_param=[]
1463	self.fixed_param=[]
1464	self.orientation_params_disp=[]
1465
1466	self.model_disp.Hide()
1467	self.disp_box.Hide()
1468	self.sizer4_4.Clear(True)
1469	model_disp = wx.StaticText(self, -1, 'No PolyDispersity for this model')
1470	self.sizer4_4.Add(model_disp,( iy, ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
1471	self.sizer4_4.Layout()
1472	self.sizer4.Layout()
1473	self.SetScrollbars(20,20,25,65)
1474
1475
1476	def _reset_dispersity(self):
1477	"""
1478	put gaussian dispersity into current model
1479	"""
1480	if len(self.param_toFit)>0:
1481	for item in self.fittable_param:
1482	if item in self.param_toFit:
1483	self.param_toFit.remove(item)
1484	for item in self.orientation_params_disp:
1485	if item in self.param_toFit:
1486	self.param_toFit.remove(item)
1487
1488	self.fittable_param=[]
1489	self.fixed_param=[]
1490	self.orientation_params_disp=[]
1491	self.values=[]
1492	self.weights=[]
1493
1494	from sans.models.dispersion_models import GaussianDispersion, ArrayDispersion
1495	if len(self.disp_cb_dict)==0:
1496	self.sizer4_4.Clear(True)
1497	self.sizer4_4.Layout()
1498	self.sizer4.Layout()
1499	self.Layout()
1500	self.Refresh()
1501	self.SetScrollbars(20,20,25,65)
1502	return
1503	if (len(self.disp_cb_dict)>0) :
1504	for p in self.disp_cb_dict:
1505	# The parameter was un-selected. Go back to Gaussian model (with 0 pts)
1506	disp_model = GaussianDispersion()
1507
1508	self._disp_obj_dict[p] = disp_model
1509	# Set the new model as the dispersion object for the selected parameter
1510	try:
1511	self.model.set_dispersion(p, disp_model)
1512	except:
1513	pass
1514	# Redraw the model
1515	#self._draw_model()
1516	## save state into
1517	self.save_current_state()
1518	self.sizer4_4.Layout()
1519	self.sizer4.Layout()
1520	self.Layout()
1521	self.SetScrollbars(20,20,25,65)
1522	self.Refresh()
1523
1524
1525
1526	def _on_select_Disp(self,event):
1527	"""
1528	allow selecting different dispersion
1529	self.disp_list should change type later .now only gaussian
1530	"""
1531
1532	n = self.disp_box.GetCurrentSelection()
1533	name = self.disp_box.GetValue()
1534	dispersity= self.disp_box.GetClientData(n)
1535
1536	self.disp_name = name
1537
1538	if name.lower() == "array":
1539	self._set_sizer_arraydispersion()
1540	else:
1541	self._set_sizer_dispersion(dispersity= dispersity)
1542
1543
1544	self.state.disp_box= n
1545
1546	self._undo.Enable(True)
1547	event = PageInfoEvent(page = self)
1548	wx.PostEvent(self.parent, event)
1549
1550	self.sizer4_4.Layout()
1551	self.sizer4.Layout()
1552	self.Layout()
1553	self.SetScrollbars(20,20,25,65)
1554	self.Refresh()
1555
1556	def _set_sizer_arraydispersion(self):
1557	"""
1558	draw sizer with array dispersity parameters
1559	"""
1560
1561	if len(self.param_toFit)>0:
1562	for item in self.fittable_param:
1563	if item in self.param_toFit:
1564	self.param_toFit.remove(item)
1565	for item in self.orientation_params_disp:
1566	if item in self.param_toFit:
1567	self.param_toFit.remove(item)
1568	for item in self.model.details.keys():
1569	if item in self.model.fixed:
1570	del self.model.details [item]
1571
1572	self.fittable_param=[]
1573	self.fixed_param=[]
1574	self.orientation_params_disp=[]
1575	self.sizer4_4.Clear(True)
1576	self._reset_dispersity()
1577	ix=0
1578	iy=1
1579	disp1 = wx.StaticText(self, -1, 'Array Dispersion')
1580	self.sizer4_4.Add(disp1,( iy, ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
1581
1582	# Look for model parameters to which we can apply an ArrayDispersion model
1583	# Add a check box for each parameter.
1584	self.disp_cb_dict = {}
1585
1586	for p in self.model.dispersion.keys():
1587	if not p in self.model.orientation_params:
1588	ix+=1
1589	self.disp_cb_dict[p] = wx.RadioButton(self, -1, p, (10, 10))
1590	self.state.disp_cb_dict[p]= self.disp_cb_dict[p].GetValue()
1591	self.Bind(wx.EVT_RADIOBUTTON, self.select_disp_angle, id=self.disp_cb_dict[p].GetId())
1592	#wx.EVT_RADIOBUTTON(self, self.disp_cb_dict[p].GetId(), self.select_disp_angle)
1593	self.sizer4_4.Add(self.disp_cb_dict[p], (iy, ix), (1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
1594	for p in self.model.dispersion.keys():
1595	if p in self.model.orientation_params:
1596	ix+=1
1597	self.disp_cb_dict[p] = wx.RadioButton(self, -1, p, (10, 10))
1598	self.state.disp_cb_dict[p]= self.disp_cb_dict[p].GetValue()
1599	#print "self.enable2D",self.enable2D
1600	if not (self.enable2D or self.data.__class__.__name__ =="Data2D"):
1601	self.disp_cb_dict[p].Hide()
1602	#self.disp_cb_dict[p].Disable()
1603	else:
1604	self.disp_cb_dict[p].Show(True)
1605	#self.disp_cb_dict[p].Enable()
1606	self.Bind(wx.EVT_RADIOBUTTON, self.select_disp_angle, id=self.disp_cb_dict[p].GetId())
1607	#wx.EVT_RADIOBUTTON(self, self.disp_cb_dict[p].GetId(), self.select_disp_angle)
1608	self.sizer4_4.Add(self.disp_cb_dict[p], (iy, ix), (1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
1609
1610
1611	ix =0
1612	iy +=1
1613	self.sizer4_4.Add((20,20),(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
1614	self.sizer4_4.Layout()
1615	self.sizer4.Layout()
1616	self.SetScrollbars(20,20,25,65)
1617	## save state into
1618
1619	self.state.orientation_params =[]
1620	self.state.orientation_params_disp =[]
1621	self.state.parameters =[]
1622	self.state.fittable_param =[]
1623	self.state.fixed_param =[]
1624
1625	## save checkbutton state and txtcrtl values
1626
1627	self._copy_parameters_state(self.orientation_params,
1628	self.state.orientation_params)
1629
1630	self._copy_parameters_state(self.orientation_params_disp,
1631	self.state.orientation_params_disp)
1632
1633	self._copy_parameters_state(self.parameters, self.state.parameters)
1634	self._copy_parameters_state(self.fittable_param, self.state.fittable_param)
1635	self._copy_parameters_state(self.fixed_param, self.state.fixed_param)
1636
1637
1638	## post state to fit panel
1639	event = PageInfoEvent(page = self)
1640	wx.PostEvent(self.parent, event)
1641
1642
1643
1644
1645
1646	def _set_range_sizer(self, title, object1=None,object=None):
1647	"""
1648	Fill the
1649	"""
1650	self.sizer5.Clear(True)
1651	box_description= wx.StaticBox(self, -1,str(title))
1652	boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
1653	#--------------------------------------------------------------
1654	self.qmin = BasicPage.ModelTextCtrl(self, -1,size=(_BOX_WIDTH,20))
1655	self.qmin.SetValue(str(self.qmin_x))
1656	self.qmin.SetToolTipString("Minimun value of Q in linear scale.")
1657	#self.qmin.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
1658	#self.qmin.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
1659	#self.qmin.Bind(wx.EVT_TEXT_ENTER, self._onparamEnter)
1660
1661	self.qmax = BasicPage.ModelTextCtrl(self, -1,size=(_BOX_WIDTH,20))
1662	self.qmax.SetValue(str(self.qmax_x))
1663	self.qmax.SetToolTipString("Maximum value of Q in linear scale.")
1664	#self.qmax.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
1665	#self.qmax.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
1666	#self.qmax.Bind(wx.EVT_TEXT_ENTER, self._onparamEnter)
1667
1668	sizer_horizontal=wx.BoxSizer(wx.HORIZONTAL)
1669	sizer= wx.GridSizer(3, 3,5, 5)
1670
1671	sizer.Add((5,5))
1672	sizer.Add(wx.StaticText(self, -1, 'Min'))
1673	sizer.Add(wx.StaticText(self, -1, 'Max'))
1674	sizer.Add(wx.StaticText(self, -1, 'Q range'))
1675
1676	sizer.Add(self.qmin)
1677	sizer.Add(self.qmax)
1678	sizer_horizontal.Add(sizer)
1679	if object!=None:
1680	sizer_horizontal.Add(object)
1681
1682	if object1!=None:
1683	boxsizer1.Add(object1)
1684	boxsizer1.Add((10,10))
1685	boxsizer1.Add(sizer_horizontal)
1686	## save state
1687	self.save_current_state()
1688	#----------------------------------------------------------------
1689	self.sizer5.Add(boxsizer1,0, wx.EXPAND \| wx.ALL, 10)
1690	self.sizer5.Layout()
1691	self.Layout()
1692	self.SetScrollbars(20,20,25,65)
1693
1694
1695	def _fill_save_sizer(self):
1696	"""
1697	Draw the layout for saving option
1698	"""
1699	# Skipping save state functionality for release 0.9.0
1700	#return
1701
1702	self.sizer6.Clear(True)
1703	box_description= wx.StaticBox(self, -1,"Save Status")
1704	boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
1705	sizer_save = wx.BoxSizer(wx.HORIZONTAL)
1706
1707	self.btSave_title = wx.StaticText(self, -1, 'Save the current panel status')
1708	self.btSave = wx.Button(self,wx.NewId(),'Save')
1709	self.btSave.Bind(wx.EVT_BUTTON, self.onSave,id= self.btSave.GetId())
1710	self.btSave.SetToolTipString("Save current panel status")
1711
1712	sizer_save.Add(self.btSave_title)
1713	sizer_save.Add((20,20),0, wx.LEFT\|wx.RIGHT\|wx.EXPAND,45)
1714
1715	sizer_save.Add(self.btSave)
1716
1717	boxsizer1.Add(sizer_save)
1718	self.sizer6.Add(boxsizer1,0, wx.EXPAND \| wx.ALL, 10)
1719	self.sizer6.Layout()
1720	self.SetScrollbars(20,20,25,65)
1721
1722	def on_model_help_clicked(self,event):
1723	"""
1724	#On 'More details' button
1725	"""
1726	from helpPanel import HelpWindow
1727
1728	if self.model == None:
1729	name = 'FuncHelp'
1730	else:
1731	name = self.model.origin_name
1732
1733	frame = HelpWindow(None, -1, pageToOpen="doc/model_functions.html")
1734	frame.Show(True)
1735	if frame.rhelp.HasAnchor(name):
1736	frame.rhelp.ScrollToAnchor(name)
1737	else:
1738	msg= "Model does not contains an available description "
1739	msg +="Please try searching in the Help window"
1740	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
1741

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

SasView

source: sasview/sansview/perspectives/fitting/basepage.py @ 920a6e5

Download in other formats: