basepage.py @ acd0bda3

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

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

SasView

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

Download in other formats: