basepage.py @ 3fe701a

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

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

SasView

source: sasview/sansview/perspectives/fitting/basepage.py @ 3fe701a

Download in other formats: