basepage.py @ 6e9150d

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 6e9150d was 6e9150d, checked in by Gervaise Alina <gervyh@…>, 15 years ago
separe the 2 radio group of selecting dispersiton type and enable apply dispersion
Property mode set to `100644`
File size: 70.4 KB

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

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

SasView

source: sasview/sansview/perspectives/fitting/basepage.py @ 6e9150d

Download in other formats: