basepage.py @ 79ac6f8

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 79ac6f8 was bb70474, checked in by Gervaise Alina <gervyh@…>, 14 years ago
working on save state
Property mode set to `100644`
File size: 90.4 KB

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

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

SasView

source: sasview/sansview/perspectives/fitting/basepage.py @ 79ac6f8

Download in other formats: