basepage.py @ abf2d75

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 abf2d75 was abf2d75, checked in by Gervaise Alina <gervyh@…>, 15 years ago
remove save button and rename context menu as bookmark
Property mode set to `100644`
File size: 74.5 KB

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

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

SasView

source: sasview/sansview/perspectives/fitting/basepage.py @ abf2d75

Download in other formats: