basepage.py @ 43db4a8

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

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

SasView

source: sasview/sansview/perspectives/fitting/basepage.py @ 43db4a8

Download in other formats: