basepage.py @ 9ff861b

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

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

SasView

source: sasview/sansview/perspectives/fitting/basepage.py @ 9ff861b

Download in other formats: