basepage.py @ e16e5c3

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 e16e5c3 was baf1ef3, checked in by Jae Cho <jhjcho@…>, 13 years ago
remember the last folder opened/saved
Property mode set to `100644`
File size: 117.5 KB

Line
1
2	import sys
3	import os
4	import wx
5	import numpy
6	import time
7	import copy
8	import math
9	import string
10	from wx.lib.scrolledpanel import ScrolledPanel
11	from sans.guiframe.panel_base import PanelBase
12	from sans.guiframe.utils import format_number,check_float
13	from sans.guiframe.events import PanelOnFocusEvent
14	from sans.guiframe.events import StatusEvent
15	from sans.guiframe.events import AppendBookmarkEvent
16	from sans.guiframe.dataFitting import Data2D
17	from sans.guiframe.dataFitting import Data1D
18	from sans.guiframe.dataFitting import check_data_validity
19	from sans.dataloader.data_info import Detector
20	from sans.dataloader.data_info import Source
21	import pagestate
22	from pagestate import PageState
23
24	(PageInfoEvent, EVT_PAGE_INFO) = wx.lib.newevent.NewEvent()
25	(PreviousStateEvent, EVT_PREVIOUS_STATE) = wx.lib.newevent.NewEvent()
26	(NextStateEvent, EVT_NEXT_STATE) = wx.lib.newevent.NewEvent()
27
28	_BOX_WIDTH = 76
29	_QMIN_DEFAULT = 0.0005
30	_QMAX_DEFAULT = 0.5
31	_NPTS_DEFAULT = 50
32	#Control panel width
33	if sys.platform.count("win32")> 0:
34	PANEL_WIDTH = 450
35	FONT_VARIANT = 0
36	ON_MAC = False
37	else:
38	PANEL_WIDTH = 500
39	FONT_VARIANT = 1
40	ON_MAC = True
41
42
43
44	class BasicPage(ScrolledPanel, PanelBase):
45	"""
46	This class provide general structure of fitpanel page
47	"""
48	## Internal name for the AUI manager
49	window_name = "Fit Page"
50	## Title to appear on top of the window
51	window_caption = "Fit Page "
52
53	def __init__(self, parent,color='blue', **kwargs):
54	"""
55	"""
56	ScrolledPanel.__init__(self, parent, **kwargs)
57	PanelBase.__init__(self, parent)
58	self.SetupScrolling()
59	#Set window's font size
60	self.SetWindowVariant(variant=FONT_VARIANT)
61
62	self.SetBackgroundColour(color)
63	## parent of the page
64	self.parent = parent
65	## manager is the fitting plugin
66	## owner of the page (fitting plugin)
67	self.event_owner = None
68	## current model
69	self.model = None
70	self.index_model = None
71	## data
72	self.data = None
73	#list of available data
74	self.data_list = []
75	self.mask = None
76	self.uid = wx.NewId()
77	#Q range for data set
78	self.qmin_data_set = None
79	self.qmax_data_set = None
80	self.npts_data_set = 0
81	## Q range
82	self.qmin = None
83	self.qmax = None
84	self.qmax_x = _QMAX_DEFAULT
85	self.qmin_x = _QMIN_DEFAULT
86	self.npts_x = _NPTS_DEFAULT
87	## total number of point: float
88	self.npts = None
89	## default fitengine type
90	self.engine_type = 'scipy'
91	## smear default
92	self.current_smearer = None
93	## 2D smear accuracy default
94	self.smear2d_accuracy = 'Low'
95	## slit smear:
96	self.dxl = None
97	self.dxw = None
98	## pinhole smear
99	self.dx_min = None
100	self.dx_max = None
101
102	self.disp_cb_dict = {}
103
104	self.state = PageState(parent=parent)
105	## dictionary containing list of models
106	self.model_list_box = {}
107
108	## Data member to store the dispersion object created
109	self._disp_obj_dict = {}
110	## selected parameters to apply dispersion
111	self.disp_cb_dict ={}
112	## smearer object
113	self.enable2D = False
114	self.is_mac = ON_MAC
115
116	##list of model parameters. each item must have same length
117	## each item related to a given parameters
118	##[cb state, name, value, "+/-", error of fit, min, max , units]
119	self.parameters = []
120	# non-fittable parameter whose value is astring
121	self.str_parameters = []
122	## list of parameters to fit , must be like self.parameters
123	self.param_toFit = []
124	## list of looking like parameters but with non fittable parameters info
125	self.fixed_param = []
126	## list of looking like parameters but with fittable parameters info
127	self.fittable_param = []
128	##list of dispersion parameters
129	self.disp_list = []
130	self.disp_name = ""
131
132	## list of orientation parameters
133	self.orientation_params = []
134	self.orientation_params_disp = []
135	if self.model != None:
136	self.disp_list = self.model.getDispParamList()
137	self.temp_multi_functional = False
138	##enable model 2D draw
139	self.enable2D = False
140	## check that the fit range is correct to plot the model again
141	self.fitrange = True
142	## Create memento to save the current state
143	self.state = PageState(parent=self.parent,
144	model=self.model, data=self.data)
145	## flag to determine if state has change
146	self.state_change = False
147	## save customized array
148	self.values = []
149	self.weights = []
150	## retrieve saved state
151	self.number_saved_state = 0
152	## dictionary of saved state
153	self.saved_states = {}
154	## Create context menu for page
155	self.popUpMenu = wx.Menu()
156
157	id = wx.NewId()
158	self._keep = wx.MenuItem(self.popUpMenu,id,"Add bookmark",
159	" Keep the panel status to recall it later")
160	self.popUpMenu.AppendItem(self._keep)
161	self._keep.Enable(False)
162	self._set_bookmark_flag(False)
163	self._set_save_flag(False)
164	wx.EVT_MENU(self, id, self.on_bookmark)
165	self.popUpMenu.AppendSeparator()
166
167	## Default locations
168	self._default_save_location = os.getcwd()
169	## save initial state on context menu
170	#self.onSave(event=None)
171	self.Bind(wx.EVT_CONTEXT_MENU, self.onContextMenu)
172
173	# bind key event
174	self.Bind(wx.EVT_LEFT_DOWN, self.on_left_down)
175
176	## create the basic structure of the panel with empty sizer
177	self.define_page_structure()
178	## drawing Initial dispersion parameters sizer
179	self.set_dispers_sizer()
180
181	## layout
182	self.set_layout()
183
184	def set_index_model(self, index):
185	"""
186	Index related to this page
187	"""
188	self.index_model = index
189
190	def create_default_data(self):
191	"""
192	Given the user selection, creates a 1D or 2D data
193	Only when the page is on theory mode.
194	"""
195	if not hasattr(self, "model_view"):
196	return
197	toggle_mode_on = self.model_view.IsEnabled()
198	if toggle_mode_on:
199	if self.enable2D and not check_data_validity(self.data):
200	self._create_default_2d_data()
201	else:
202	self._create_default_1d_data()
203
204	def _create_default_1d_data(self):
205	"""
206	Create default data for fitting perspective
207	Only when the page is on theory mode.
208	:warning: This data is never plotted.
209
210	"""
211	x = numpy.linspace(start=self.qmin_x, stop=self.qmax_x,
212	num=self.npts_x, endpoint=True)
213	self.data = Data1D(x=x)
214	self.data.xaxis('\\rm{Q}',"A^{-1}")
215	self.data.yaxis('\\rm{Intensity}', "cm^{-1}")
216	self.data.is_data = False
217	self.data.id = str(self.uid) + " data"
218	self.data.group_id = str(self.uid) + " Model1D"
219
220	def _create_default_2d_data(self):
221	"""
222	Create 2D data by default
223	Only when the page is on theory mode.
224	:warning: This data is never plotted.
225	"""
226	self.data = Data2D()
227	qmax = self.qmax_x / math.sqrt(2)
228	self.data.xaxis('\\rm{Q_{x}}', 'A^{-1}')
229	self.data.yaxis('\\rm{Q_{y}}', 'A^{-1}')
230	self.data.is_data = False
231	self.data.id = str(self.uid) + " data"
232	self.data.group_id = str(self.uid) + " Model2D"
233	## Default values
234	self.data.detector.append(Detector())
235	index = len(self.data.detector) - 1
236	self.data.detector[index].distance = 8000 # mm
237	self.data.source.wavelength= 6 # A
238	self.data.detector[index].pixel_size.x = 5 # mm
239	self.data.detector[index].pixel_size.y = 5 # mm
240	self.data.detector[index].beam_center.x = qmax
241	self.data.detector[index].beam_center.y = qmax
242	## create x_bins and y_bins of the model 2D
243	pixel_width_x = self.data.detector[index].pixel_size.x
244	pixel_width_y = self.data.detector[index].pixel_size.y
245	center_x = self.data.detector[index].beam_center.x/pixel_width_x
246	center_y = self.data.detector[index].beam_center.y/pixel_width_y
247	# theory default: assume the beam
248	#center is located at the center of sqr detector
249	xmax = qmax
250	xmin = -qmax
251	ymax = qmax
252	ymin = -qmax
253	qstep = self.npts_x
254
255	x = numpy.linspace(start=xmin, stop=xmax, num=qstep, endpoint=True)
256	y = numpy.linspace(start=ymin, stop=ymax, num=qstep, endpoint=True)
257	## use data info instead
258	new_x = numpy.tile(x, (len(y), 1))
259	new_y = numpy.tile(y, (len(x), 1))
260	new_y = new_y.swapaxes(0,1)
261	# all data reuire now in 1d array
262	qx_data = new_x.flatten()
263	qy_data = new_y.flatten()
264	q_data = numpy.sqrt(qx_dataqx_data + qy_dataqy_data)
265	# set all True (standing for unmasked) as default
266	mask = numpy.ones(len(qx_data), dtype=bool)
267	# calculate the range of qx and qy: this way,
268	# it is a little more independent
269	x_size = xmax - xmin
270	y_size = ymax - ymin
271	# store x and y bin centers in q space
272	x_bins = x
273	y_bins = y
274	# bin size: x- & y-directions
275	xstep = x_size/len(x_bins-1)
276	ystep = y_size/len(y_bins-1)
277
278	self.data.source = Source()
279	self.data.data = numpy.ones(len(mask))
280	self.data.err_data = numpy.ones(len(mask))
281	self.data.qx_data = qx_data
282	self.data.qy_data = qy_data
283	self.data.q_data = q_data
284	self.data.mask = mask
285	self.data.x_bins = x_bins
286	self.data.y_bins = y_bins
287	# max and min taking account of the bin sizes
288	self.data.xmin = xmin
289	self.data.xmax = xmax
290	self.data.ymin = ymin
291	self.data.ymax = ymax
292
293	def on_set_focus(self, event):
294	"""
295	"""
296	if self._manager is not None:
297	wx.PostEvent(self._manager.parent, PanelOnFocusEvent(panel=self))
298
299	class ModelTextCtrl(wx.TextCtrl):
300	"""
301	Text control for model and fit parameters.
302	Binds the appropriate events for user interactions.
303	Default callback methods can be overwritten on initialization
304
305	:param kill_focus_callback: callback method for EVT_KILL_FOCUS event
306	:param set_focus_callback: callback method for EVT_SET_FOCUS event
307	:param mouse_up_callback: callback method for EVT_LEFT_UP event
308	:param text_enter_callback: callback method for EVT_TEXT_ENTER event
309
310	"""
311	## Set to True when the mouse is clicked while the whole string is selected
312	full_selection = False
313	## Call back for EVT_SET_FOCUS events
314	_on_set_focus_callback = None
315
316	def __init__(self, parent, id=-1,
317	value=wx.EmptyString,
318	pos=wx.DefaultPosition,
319	size=wx.DefaultSize,
320	style=0,
321	validator=wx.DefaultValidator,
322	name=wx.TextCtrlNameStr,
323	kill_focus_callback=None,
324	set_focus_callback=None,
325	mouse_up_callback=None,
326	text_enter_callback = None):
327
328	wx.TextCtrl.__init__(self, parent, id, value, pos,
329	size, style, validator, name)
330
331	# Bind appropriate events
332	self._on_set_focus_callback = parent.onSetFocus \
333	if set_focus_callback is None else set_focus_callback
334	self.Bind(wx.EVT_SET_FOCUS, self._on_set_focus)
335	self.Bind(wx.EVT_KILL_FOCUS, self._silent_kill_focus \
336	if kill_focus_callback is None else kill_focus_callback)
337	self.Bind(wx.EVT_TEXT_ENTER, parent._onparamEnter \
338	if text_enter_callback is None else text_enter_callback)
339	if not ON_MAC :
340	self.Bind(wx.EVT_LEFT_UP, self._highlight_text \
341	if mouse_up_callback is None else mouse_up_callback)
342
343	def _on_set_focus(self, event):
344	"""
345	Catch when the text control is set in focus to highlight the whole
346	text if necessary
347
348	:param event: mouse event
349
350	"""
351	event.Skip()
352	self.full_selection = True
353	return self._on_set_focus_callback(event)
354
355
356
357	def _highlight_text(self, event):
358	"""
359	Highlight text of a TextCtrl only of no text has be selected
360
361	:param event: mouse event
362
363	"""
364	# Make sure the mouse event is available to other listeners
365	event.Skip()
366	control = event.GetEventObject()
367	if self.full_selection:
368	self.full_selection = False
369	# Check that we have a TextCtrl
370	if issubclass(control.__class__, wx.TextCtrl):
371	# Check whether text has been selected,
372	# if not, select the whole string
373	(start, end) = control.GetSelection()
374	if start==end:
375	control.SetSelection(-1,-1)
376
377	def _silent_kill_focus(self,event):
378	"""
379	Save the state of the page
380	"""
381
382	event.Skip()
383	pass
384
385	def set_page_info(self, page_info):
386	"""
387	set some page important information at once
388	"""
389	##window_name
390	self.window_name = page_info.window_name
391	##window_caption
392	self.window_caption = page_info.window_caption
393	## manager is the fitting plugin
394	self._manager= page_info.manager
395	## owner of the page (fitting plugin)
396	self.event_owner= page_info.event_owner
397	## current model
398	self.model = page_info.model
399	## data
400	self.data = page_info.data
401	## dictionary containing list of models
402	self.model_list_box = page_info.model_list_box
403	## Data member to store the dispersion object created
404	self.populate_box(dict=self.model_list_box)
405
406	def onContextMenu(self, event):
407	"""
408	Retrieve the state selected state
409	"""
410	# Skipping the save state functionality for release 0.9.0
411	#return
412
413	pos = event.GetPosition()
414	pos = self.ScreenToClient(pos)
415
416	self.PopupMenu(self.popUpMenu, pos)
417
418
419	def onUndo(self, event):
420	"""
421	Cancel the previous action
422	"""
423	event = PreviousStateEvent(page = self)
424	wx.PostEvent(self.parent, event)
425
426	def onRedo(self, event):
427	"""
428	Restore the previous action cancelled
429	"""
430	event = NextStateEvent(page= self)
431	wx.PostEvent(self.parent, event)
432
433	def define_page_structure(self):
434	"""
435	Create empty sizer for a panel
436	"""
437	self.vbox = wx.BoxSizer(wx.VERTICAL)
438	self.sizer0 = wx.BoxSizer(wx.VERTICAL)
439	self.sizer1 = wx.BoxSizer(wx.VERTICAL)
440	self.sizer2 = wx.BoxSizer(wx.VERTICAL)
441	self.sizer3 = wx.BoxSizer(wx.VERTICAL)
442	self.sizer4 = wx.BoxSizer(wx.VERTICAL)
443	self.sizer5 = wx.BoxSizer(wx.VERTICAL)
444	self.sizer6 = wx.BoxSizer(wx.VERTICAL)
445
446	self.sizer0.SetMinSize((PANEL_WIDTH,-1))
447	self.sizer1.SetMinSize((PANEL_WIDTH,-1))
448	self.sizer2.SetMinSize((PANEL_WIDTH,-1))
449	self.sizer3.SetMinSize((PANEL_WIDTH,-1))
450	self.sizer4.SetMinSize((PANEL_WIDTH,-1))
451	self.sizer5.SetMinSize((PANEL_WIDTH,-1))
452	self.sizer6.SetMinSize((PANEL_WIDTH,-1))
453
454	self.vbox.Add(self.sizer0)
455	self.vbox.Add(self.sizer1)
456	self.vbox.Add(self.sizer2)
457	self.vbox.Add(self.sizer3)
458	self.vbox.Add(self.sizer4)
459	self.vbox.Add(self.sizer5)
460	self.vbox.Add(self.sizer6)
461
462	def set_layout(self):
463	"""
464	layout
465	"""
466	self.vbox.Layout()
467	self.vbox.Fit(self)
468	self.SetSizer(self.vbox)
469	self.Centre()
470
471	def set_owner(self,owner):
472	"""
473	set owner of fitpage
474
475	:param owner: the class responsible of plotting
476
477	"""
478	self.event_owner = owner
479	self.state.event_owner = owner
480
481	def get_state(self):
482	"""
483	"""
484	return self.state
485
486	def get_data(self):
487	"""
488	return the current data
489	"""
490	return self.data
491
492	def get_data_list(self):
493	"""
494	return the current data
495	"""
496	return self.data_list
497
498	def set_manager(self, manager):
499	"""
500	set panel manager
501
502	:param manager: instance of plugin fitting
503
504	"""
505	self._manager = manager
506	self.state.manager = manager
507
508	def populate_box(self, dict):
509	"""
510	Store list of model
511
512	:param dict: dictionary containing list of models
513
514	"""
515	self.model_list_box = dict
516	self.state.model_list_box = self.model_list_box
517	self.initialize_combox()
518
519	def initialize_combox(self):
520	"""
521	put default value in the combobox
522	"""
523	## fill combox box
524	if self.model_list_box is None:
525	return
526	if len(self.model_list_box) > 0:
527	self._populate_box(self.formfactorbox,
528	self.model_list_box["Shapes"])
529
530	if len(self.model_list_box) > 0:
531	self._populate_box(self.structurebox,
532	self.model_list_box["Structure Factors"])
533	self.structurebox.Insert("None", 0, None)
534	self.structurebox.SetSelection(0)
535	self.structurebox.Hide()
536	self.text2.Hide()
537	self.structurebox.Disable()
538	self.text2.Disable()
539
540	if self.model.__class__ in self.model_list_box["P(Q)*S(Q)"]:
541	self.structurebox.Show()
542	self.text2.Show()
543	self.structurebox.Enable()
544	self.text2.Enable()
545
546	def set_dispers_sizer(self):
547	"""
548	fill sizer containing dispersity info
549	"""
550	self.sizer4.Clear(True)
551	name="Polydispersity and Orientational Distribution"
552	box_description= wx.StaticBox(self, -1,name)
553	boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
554	#----------------------------------------------------
555	self.disable_disp = wx.RadioButton(self, -1, 'Off', (10, 10),
556	style=wx.RB_GROUP)
557	self.enable_disp = wx.RadioButton(self, -1, 'On', (10, 30))
558	# best size for MAC and PC
559	if ON_MAC:
560	size_q = (30, 20)
561	else:
562	size_q = (20, 15)
563	self.disp_help_bt = wx.Button(self,wx.NewId(),'?',
564	style = wx.BU_EXACTFIT,
565	size=size_q)
566	self.disp_help_bt.Bind(wx.EVT_BUTTON,
567	self.on_pd_help_clicked,id= self.disp_help_bt.GetId())
568	self.disp_help_bt.SetToolTipString("Helps for Polydispersion.")
569
570	self.Bind(wx.EVT_RADIOBUTTON, self._set_dipers_Param,
571	id=self.disable_disp.GetId())
572	self.Bind(wx.EVT_RADIOBUTTON, self._set_dipers_Param,
573	id=self.enable_disp.GetId())
574	#MAC needs SetValue
575	self.disable_disp.SetValue(True)
576	sizer_dispersion = wx.BoxSizer(wx.HORIZONTAL)
577	sizer_dispersion.Add((20,20))
578	name=""#Polydispersity and \nOrientational Distribution "
579	sizer_dispersion.Add(wx.StaticText(self,-1,name))
580	sizer_dispersion.Add(self.enable_disp )
581	sizer_dispersion.Add((20,20))
582	sizer_dispersion.Add(self.disable_disp )
583	sizer_dispersion.Add((25,20))
584	sizer_dispersion.Add(self.disp_help_bt)
585
586	## fill a sizer for dispersion
587	boxsizer1.Add( sizer_dispersion,0,
588	wx.TOP\|wx.BOTTOM\|wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE,border=5)
589	#boxsizer1.Add( (10,10) )
590	#boxsizer1.Add( sizer_select_dispers )
591	self.sizer4_4 = wx.GridBagSizer(6,5)
592
593	boxsizer1.Add( self.sizer4_4 )
594	#-----------------------------------------------------
595	self.sizer4.Add(boxsizer1,0, wx.EXPAND \| wx.ALL, 10)
596	self.sizer4_4.Layout()
597	self.sizer4.Layout()
598	self.Layout()
599
600	self.Refresh()
601	## saving the state of enable dispersity button
602	self.state.enable_disp= self.enable_disp.GetValue()
603	self.state.disable_disp= self.disable_disp.GetValue()
604	self.SetupScrolling()
605
606
607	def onResetModel(self, event):
608	"""
609	Reset model state
610	"""
611	menu = event.GetEventObject()
612	## post help message for the selected model
613	msg = menu.GetHelpString(event.GetId())
614	msg +=" reloaded"
615	wx.PostEvent(self.parent.parent, StatusEvent(status=msg))
616
617	name = menu.GetLabel(event.GetId())
618	self._on_select_model_helper()
619
620	if name in self.saved_states.keys():
621	previous_state = self.saved_states[name]
622	## reset state of checkbox,textcrtl and regular parameters value
623	self.Show(False)
624	self.reset_page(previous_state)
625	self.Show(True)
626
627	def on_preview(self, event):
628	"""
629	Report the current fit results
630	"""
631	# Get plot image from plotpanel
632	images, canvases = self.get_images()
633	# get the report dialog
634	self.state.report(images, canvases)
635
636
637	def on_save(self, event):
638	"""
639	Save the current state into file
640	"""
641	self.save_current_state()
642	new_state = self.state.clone()
643	# Ask the user the location of the file to write to.
644	path = None
645	if self.parent != None:
646	self._default_save_location = \
647	self.parent.parent._default_save_location
648	dlg = wx.FileDialog(self, "Choose a file", self._default_save_location,
649	self.window_caption, "*.fitv", wx.SAVE)
650
651	if dlg.ShowModal() == wx.ID_OK:
652	path = dlg.GetPath()
653	self._default_save_location = os.path.dirname(path)
654	self.parent.parent._default_save_location =\
655	self._default_save_location
656	else:
657	return None
658	# MAC always needs the extension for saving
659	extens = ".fitv"
660	# Make sure the ext included in the file name
661	fName = os.path.splitext(path)[0] + extens
662	#the manager write the state into file
663	self._manager.save_fit_state(filepath=fName, fitstate=new_state)
664	return new_state
665
666	def on_copy(self, event):
667	"""
668	Copy Parameter values to the clipboad
669	"""
670	if event != None:
671	event.Skip()
672	# It seems MAC needs wxCallAfter
673	wx.CallAfter(self.get_copy)
674
675	# messages depending on the flag
676	self._copy_info(None)
677
678	def on_paste(self, event):
679	"""
680	Paste Parameter values to the panel if possible
681	"""
682	if event != None:
683	event.Skip()
684	# It seems MAC needs wxCallAfter for the setvalues
685	# for multiple textctrl items, otherwise it tends to crash once a while.
686	wx.CallAfter(self.get_paste)
687	# messages depending on the flag
688	#self._copy_info(True)
689
690	def _copy_info(self, flag):
691	"""
692	Send event dpemding on flag
693
694	: Param flag: flag that distinguish event
695	"""
696	# messages depending on the flag
697	if flag == None:
698	msg = " Parameter values are copied to the clipboard..."
699	infor = 'warning'
700	elif flag:
701	msg = " Parameter values are pasted from the clipboad..."
702	infor = "warning"
703	else:
704	msg = "Error was occured "
705	msg += ": No valid parameter values to paste from the clipboard..."
706	infor = "error"
707	# inform msg to wx
708	wx.PostEvent( self.parent.parent,
709	StatusEvent(status= msg, info=infor))
710
711	def _get_time_stamp(self):
712	"""
713	return time and date stings
714	"""
715	# date and time
716	year, month, day,hour,minute,second,tda,ty,tm_isdst= time.localtime()
717	current_time= str(hour)+":"+str(minute)+":"+str(second)
718	current_date= str( month)+"/"+str(day)+"/"+str(year)
719	return current_time, current_date
720
721	def on_bookmark(self, event):
722	"""
723	save history of the data and model
724	"""
725	if self.model==None:
726	msg="Can not bookmark; Please select Data and Model first..."
727	wx.MessageBox(msg, 'Info')
728	return
729	self.save_current_state()
730	new_state = self.state.clone()
731	##Add model state on context menu
732	self.number_saved_state += 1
733	current_time, current_date = self._get_time_stamp()
734	#name= self.model.name+"[%g]"%self.number_saved_state
735	name = "Fitting: %g]" % self.number_saved_state
736	name += self.model.__class__.__name__
737	name += "bookmarked at %s on %s" % (current_time, current_date)
738	self.saved_states[name]= new_state
739
740	## Add item in the context menu
741	msg = "Model saved at %s on %s"%(current_time, current_date)
742	## post help message for the selected model
743	msg +=" Saved! right click on this page to retrieve this model"
744	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
745
746	id = wx.NewId()
747	self.popUpMenu.Append(id,name,str(msg))
748	wx.EVT_MENU(self, id, self.onResetModel)
749	wx.PostEvent(self.parent.parent,
750	AppendBookmarkEvent(title=name,
751	hint=str(msg), handler=self._back_to_bookmark))
752	def _back_to_bookmark(self, event):
753	"""
754	Back to bookmark
755	"""
756	self._manager.on_perspective(event)
757	self.onResetModel(event)
758	def old_on_bookmark(self, event):
759	"""
760	save history of the data and model
761	"""
762	if self.model==None:
763	msg="Can not bookmark; Please select Data and Model first..."
764	wx.MessageBox(msg, 'Info')
765	return
766	if hasattr(self,"enable_disp"):
767	self.state.enable_disp = copy.deepcopy(self.enable_disp.GetValue())
768	if hasattr(self, "disp_box"):
769	self.state.disp_box = copy.deepcopy(self.disp_box.GetSelection())
770
771	self.state.model.name= self.model.name
772
773	#Remember fit engine_type for fit panel
774	if self.engine_type == None:
775	self.engine_type = "scipy"
776	if self._manager !=None:
777	self._manager._on_change_engine(engine=self.engine_type)
778
779	self.state.engine_type = self.engine_type
780
781	new_state = self.state.clone()
782	new_state.model.name = self.state.model.name
783
784	new_state.enable2D = copy.deepcopy(self.enable2D)
785	##Add model state on context menu
786	self.number_saved_state += 1
787	#name= self.model.name+"[%g]"%self.number_saved_state
788	name= self.model.__class__.__name__+"[%g]"%self.number_saved_state
789	self.saved_states[name]= new_state
790
791	## Add item in the context menu
792
793	year, month, day,hour,minute,second,tda,ty,tm_isdst= time.localtime()
794	my_time= str(hour)+" : "+str(minute)+" : "+str(second)+" "
795	date= str( month)+"\|"+str(day)+"\|"+str(year)
796	msg= "Model saved at %s on %s"%(my_time, date)
797	## post help message for the selected model
798	msg +=" Saved! right click on this page to retrieve this model"
799	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
800
801	id = wx.NewId()
802	self.popUpMenu.Append(id,name,str(msg))
803	wx.EVT_MENU(self, id, self.onResetModel)
804
805	def onSetFocus(self, evt):
806	"""
807	highlight the current textcrtl and hide the error text control shown
808	after fitting
809	"""
810	return
811
812	def read_file(self, path):
813	"""
814	Read two columns file
815
816	:param path: the path to the file to read
817
818	"""
819	try:
820	if path==None:
821	wx.PostEvent(self.parent.parent, StatusEvent(status=\
822	" Selected Distribution was not loaded: %s"%path))
823	return None, None
824	input_f = open(path, 'r')
825	buff = input_f.read()
826	lines = buff.split('\n')
827
828	angles = []
829	weights=[]
830	for line in lines:
831	toks = line.split()
832	try:
833	angle = float(toks[0])
834	weight = float(toks[1])
835	except:
836	# Skip non-data lines
837	pass
838	angles.append(angle)
839	weights.append(weight)
840	return numpy.array(angles), numpy.array(weights)
841	except:
842	raise
843
844	def createMemento(self):
845	"""
846	return the current state of the page
847	"""
848	return self.state.clone()
849
850
851	def save_current_state(self):
852	"""
853	Store current state
854	"""
855	self.state.engine_type = copy.deepcopy(self.engine_type)
856	## save model option
857	if self.model!= None:
858	self.disp_list= self.model.getDispParamList()
859	self.state.disp_list= copy.deepcopy(self.disp_list)
860	self.state.model = self.model.clone()
861	#save radiobutton state for model selection
862	self.state.shape_rbutton = self.shape_rbutton.GetValue()
863	self.state.shape_indep_rbutton = self.shape_indep_rbutton.GetValue()
864	self.state.struct_rbutton = self.struct_rbutton.GetValue()
865	self.state.plugin_rbutton = self.plugin_rbutton.GetValue()
866	#model combobox
867	self.state.structurebox = self.structurebox.GetSelection()
868	self.state.formfactorbox = self.formfactorbox.GetSelection()
869
870	self.state.enable2D = copy.deepcopy(self.enable2D)
871	self.state.values= copy.deepcopy(self.values)
872	self.state.weights = copy.deepcopy( self.weights)
873	## save data
874	self.state.data= copy.deepcopy(self.data)
875	self.state.qmax_x = self.qmax_x
876	self.state.qmin_x = self.qmin_x
877	self.state.dI_noweight = copy.deepcopy(self.dI_noweight.GetValue())
878	self.state.dI_didata = copy.deepcopy(self.dI_didata.GetValue())
879	self.state.dI_sqrdata = copy.deepcopy(self.dI_sqrdata.GetValue())
880	self.state.dI_idata = copy.deepcopy(self.dI_idata.GetValue())
881	if hasattr(self,"enable_disp"):
882	self.state.enable_disp= self.enable_disp.GetValue()
883	self.state.disable_disp = self.disable_disp.GetValue()
884
885	self.state.smearer = copy.deepcopy(self.current_smearer)
886	if hasattr(self,"enable_smearer"):
887	self.state.enable_smearer = \
888	copy.deepcopy(self.enable_smearer.GetValue())
889	self.state.disable_smearer = \
890	copy.deepcopy(self.disable_smearer.GetValue())
891
892	self.state.pinhole_smearer = \
893	copy.deepcopy(self.pinhole_smearer.GetValue())
894	self.state.slit_smearer = copy.deepcopy(self.slit_smearer.GetValue())
895
896	if len(self._disp_obj_dict)>0:
897	for k , v in self._disp_obj_dict.iteritems():
898	self.state._disp_obj_dict[k]= v
899
900
901	self.state.values = copy.deepcopy(self.values)
902	self.state.weights = copy.deepcopy(self.weights)
903	## save plotting range
904	self._save_plotting_range()
905
906	self.state.orientation_params = []
907	self.state.orientation_params_disp = []
908	self.state.parameters = []
909	self.state.fittable_param = []
910	self.state.fixed_param = []
911	self.state.str_parameters = []
912
913
914	## save checkbutton state and txtcrtl values
915	self._copy_parameters_state(self.str_parameters,
916	self.state.str_parameters)
917	self._copy_parameters_state(self.orientation_params,
918	self.state.orientation_params)
919	self._copy_parameters_state(self.orientation_params_disp,
920	self.state.orientation_params_disp)
921
922	self._copy_parameters_state(self.parameters, self.state.parameters)
923	self._copy_parameters_state(self.fittable_param,
924	self.state.fittable_param)
925	self._copy_parameters_state(self.fixed_param, self.state.fixed_param)
926	#save chisqr
927	self.state.tcChi = self.tcChi.GetValue()
928
929	def save_current_state_fit(self):
930	"""
931	Store current state for fit_page
932	"""
933	## save model option
934	if self.model!= None:
935	self.disp_list= self.model.getDispParamList()
936	self.state.disp_list= copy.deepcopy(self.disp_list)
937	self.state.model = self.model.clone()
938	if hasattr(self, "engine_type"):
939	self.state.engine_type = copy.deepcopy(self.engine_type)
940
941	self.state.enable2D = copy.deepcopy(self.enable2D)
942	self.state.values= copy.deepcopy(self.values)
943	self.state.weights = copy.deepcopy( self.weights)
944	## save data
945	self.state.data= copy.deepcopy(self.data)
946
947	if hasattr(self,"enable_disp"):
948	self.state.enable_disp= self.enable_disp.GetValue()
949	self.state.disable_disp = self.disable_disp.GetValue()
950
951	self.state.smearer = copy.deepcopy(self.current_smearer)
952	if hasattr(self,"enable_smearer"):
953	self.state.enable_smearer = \
954	copy.deepcopy(self.enable_smearer.GetValue())
955	self.state.disable_smearer = \
956	copy.deepcopy(self.disable_smearer.GetValue())
957
958	self.state.pinhole_smearer = \
959	copy.deepcopy(self.pinhole_smearer.GetValue())
960	self.state.slit_smearer = copy.deepcopy(self.slit_smearer.GetValue())
961	self.state.dI_noweight = copy.deepcopy(self.dI_noweight.GetValue())
962	self.state.dI_didata = copy.deepcopy(self.dI_didata.GetValue())
963	self.state.dI_sqrdata = copy.deepcopy(self.dI_sqrdata.GetValue())
964	self.state.dI_idata = copy.deepcopy(self.dI_idata.GetValue())
965	if hasattr(self,"disp_box"):
966	self.state.disp_box = self.disp_box.GetCurrentSelection()
967
968	if len(self.disp_cb_dict) > 0:
969	for k, v in self.disp_cb_dict.iteritems():
970
971	if v == None :
972	self.state.disp_cb_dict[k] = v
973	else:
974	try:
975	self.state.disp_cb_dict[k] = v.GetValue()
976	except:
977	self.state.disp_cb_dict[k] = None
978
979	if len(self._disp_obj_dict) > 0:
980	for k , v in self._disp_obj_dict.iteritems():
981
982	self.state._disp_obj_dict[k] = v
983
984
985	self.state.values = copy.deepcopy(self.values)
986	self.state.weights = copy.deepcopy(self.weights)
987
988	## save plotting range
989	self._save_plotting_range()
990
991	## save checkbutton state and txtcrtl values
992	self._copy_parameters_state(self.orientation_params,
993	self.state.orientation_params)
994	self._copy_parameters_state(self.orientation_params_disp,
995	self.state.orientation_params_disp)
996	self._copy_parameters_state(self.parameters, self.state.parameters)
997	self._copy_parameters_state(self.fittable_param,
998	self.state.fittable_param)
999	self._copy_parameters_state(self.fixed_param, self.state.fixed_param)
1000
1001
1002	def check_invalid_panel(self):
1003	"""
1004	check if the user can already perform some action with this panel
1005	"""
1006	flag = False
1007	if self.data is None:
1008	self.disable_smearer.SetValue(True)
1009	self.disable_disp.SetValue(True)
1010	msg = "Please load Data and select Model to start..."
1011	wx.MessageBox(msg, 'Info')
1012	return True
1013
1014	def set_model_state(self, state):
1015	"""
1016	reset page given a model state
1017	"""
1018	self.disp_cb_dict = state.disp_cb_dict
1019	self.disp_list = state.disp_list
1020
1021	## set the state of the radio box
1022	self.shape_rbutton.SetValue(state.shape_rbutton )
1023	self.shape_indep_rbutton.SetValue(state.shape_indep_rbutton)
1024	self.struct_rbutton.SetValue(state.struct_rbutton)
1025	self.plugin_rbutton.SetValue(state.plugin_rbutton)
1026
1027	## fill model combobox
1028	self._show_combox_helper()
1029	#select the current model
1030	self.formfactorbox.Select(int(state.formfactorcombobox))
1031	self.structurebox.SetSelection(state.structurecombobox )
1032	if state.multi_factor != None:
1033	self.multifactorbox.SetSelection(state.multi_factor)
1034
1035	## reset state of checkbox,textcrtl and regular parameters value
1036	self._reset_parameters_state(self.orientation_params_disp,
1037	state.orientation_params_disp)
1038	self._reset_parameters_state(self.orientation_params,
1039	state.orientation_params)
1040	self._reset_parameters_state(self.str_parameters,
1041	state.str_parameters)
1042	self._reset_parameters_state(self.parameters,state.parameters)
1043	## display dispersion info layer
1044	self.enable_disp.SetValue(state.enable_disp)
1045	self.disable_disp.SetValue(state.disable_disp)
1046
1047	if hasattr(self, "disp_box"):
1048
1049	self.disp_box.SetSelection(state.disp_box)
1050	n= self.disp_box.GetCurrentSelection()
1051	dispersity= self.disp_box.GetClientData(n)
1052	name = dispersity.__name__
1053
1054	self._set_dipers_Param(event=None)
1055
1056	if name == "ArrayDispersion":
1057
1058	for item in self.disp_cb_dict.keys():
1059
1060	if hasattr(self.disp_cb_dict[item], "SetValue") :
1061	self.disp_cb_dict[item].SetValue(\
1062	state.disp_cb_dict[item])
1063	# Create the dispersion objects
1064	from sans.models.dispersion_models import ArrayDispersion
1065	disp_model = ArrayDispersion()
1066	if hasattr(state,"values")and\
1067	self.disp_cb_dict[item].GetValue() == True:
1068	if len(state.values)>0:
1069	self.values=state.values
1070	self.weights=state.weights
1071	disp_model.set_weights(self.values,
1072	state.weights)
1073	else:
1074	self._reset_dispersity()
1075
1076	self._disp_obj_dict[item] = disp_model
1077	# Set the new model as the dispersion object
1078	#for the selected parameter
1079	self.model.set_dispersion(item, disp_model)
1080
1081	self.model._persistency_dict[item] = \
1082	[state.values, state.weights]
1083
1084	else:
1085	keys = self.model.getParamList()
1086	for item in keys:
1087	if item in self.disp_list and \
1088	not self.model.details.has_key(item):
1089	self.model.details[item] = ["", None, None]
1090	for k,v in self.state.disp_cb_dict.iteritems():
1091	self.disp_cb_dict = copy.deepcopy(state.disp_cb_dict)
1092	self.state.disp_cb_dict = copy.deepcopy(state.disp_cb_dict)
1093	## smearing info restore
1094	if hasattr(self, "enable_smearer"):
1095	## set smearing value whether or not the data
1096	#contain the smearing info
1097	self.enable_smearer.SetValue(state.enable_smearer)
1098	self.disable_smearer.SetValue(state.disable_smearer)
1099	self.onSmear(event=None)
1100	self.pinhole_smearer.SetValue(state.pinhole_smearer)
1101	self.slit_smearer.SetValue(state.slit_smearer)
1102
1103	self.dI_noweight.SetValue(state.dI_noweight)
1104	self.dI_didata.SetValue(state.dI_didata)
1105	self.dI_sqrdata.SetValue(state.dI_sqrdata)
1106	self.dI_idata.SetValue(state.dI_idata)
1107
1108	## we have two more options for smearing
1109	if self.pinhole_smearer.GetValue(): self.onPinholeSmear(event=None)
1110	elif self.slit_smearer.GetValue(): self.onSlitSmear(event=None)
1111
1112	## reset state of checkbox,textcrtl and dispersity parameters value
1113	self._reset_parameters_state(self.fittable_param,state.fittable_param)
1114	self._reset_parameters_state(self.fixed_param,state.fixed_param)
1115
1116	## draw the model with previous parameters value
1117	self._onparamEnter_helper()
1118	self.select_param(event=None)
1119	#Save state_fit
1120	self.save_current_state_fit()
1121	self._lay_out()
1122	self.Refresh()
1123
1124	def reset_page_helper(self, state):
1125	"""
1126	Use page_state and change the state of existing page
1127
1128	:precondition: the page is already drawn or created
1129
1130	:postcondition: the state of the underlying data change as well as the
1131	state of the graphic interface
1132	"""
1133	if state == None:
1134	#self._undo.Enable(False)
1135	return
1136	# set data, etc. from the state
1137	# reset page between theory and fitting from bookmarking
1138	#if state.data == None:
1139	# data = None
1140	#else:
1141	data = state.data
1142
1143	#if data != None:
1144
1145	if data == None:
1146	data_min = state.qmin
1147	data_max = state.qmax
1148	self.qmin_x = data_min
1149	self.qmax_x = data_max
1150	#self.minimum_q.SetValue(str(data_min))
1151	#self.maximum_q.SetValue(str(data_max))
1152	self.qmin.SetValue(str(data_min))
1153	self.qmax.SetValue(str(data_max))
1154
1155	self.state.data = data
1156	self.state.qmin = self.qmin_x
1157	self.state.qmax = self.qmax_x
1158	else:
1159	self.set_data(data)
1160
1161	self.enable2D= state.enable2D
1162	self.engine_type = state.engine_type
1163
1164	self.disp_cb_dict = state.disp_cb_dict
1165	self.disp_list = state.disp_list
1166
1167	## set the state of the radio box
1168	self.shape_rbutton.SetValue(state.shape_rbutton )
1169	self.shape_indep_rbutton.SetValue(state.shape_indep_rbutton)
1170	self.struct_rbutton.SetValue(state.struct_rbutton)
1171	self.plugin_rbutton.SetValue(state.plugin_rbutton)
1172
1173	## fill model combobox
1174	self._show_combox_helper()
1175	#select the current model
1176	self.formfactorbox.Select(int(state.formfactorcombobox))
1177	self.structurebox.SetSelection(state.structurecombobox )
1178	if state.multi_factor != None:
1179	self.multifactorbox.SetSelection(state.multi_factor)
1180
1181	#reset the fitting engine type
1182	self.engine_type = state.engine_type
1183	#draw the pnael according to the new model parameter
1184	self._on_select_model(event=None)
1185	# take care of 2D button
1186	if data == None and self.model_view.IsEnabled():
1187	if self.enable2D:
1188	self.model_view.SetLabel("2D Mode")
1189	else:
1190	self.model_view.SetLabel("1D Mode")
1191	# else:
1192
1193	if self._manager !=None:
1194	self._manager._on_change_engine(engine=self.engine_type)
1195	## set the select all check box to the a given state
1196	self.cb1.SetValue(state.cb1)
1197
1198	## reset state of checkbox,textcrtl and regular parameters value
1199	self._reset_parameters_state(self.orientation_params_disp,
1200	state.orientation_params_disp)
1201	self._reset_parameters_state(self.orientation_params,
1202	state.orientation_params)
1203	self._reset_parameters_state(self.str_parameters,
1204	state.str_parameters)
1205	self._reset_parameters_state(self.parameters,state.parameters)
1206	## display dispersion info layer
1207	self.enable_disp.SetValue(state.enable_disp)
1208	self.disable_disp.SetValue(state.disable_disp)
1209	# If the polydispersion is ON
1210	if state.enable_disp:
1211	# reset dispersion according the state
1212	self._set_dipers_Param(event=None)
1213	self._reset_page_disp_helper(state)
1214	##plotting range restore
1215	self._reset_plotting_range(state)
1216	## smearing info restore
1217	if hasattr(self, "enable_smearer"):
1218	## set smearing value whether or not the data
1219	#contain the smearing info
1220	self.enable_smearer.SetValue(state.enable_smearer)
1221	self.disable_smearer.SetValue(state.disable_smearer)
1222	self.onSmear(event=None)
1223	self.pinhole_smearer.SetValue(state.pinhole_smearer)
1224	self.slit_smearer.SetValue(state.slit_smearer)
1225
1226	self.dI_noweight.SetValue(state.dI_noweight)
1227	self.dI_didata.SetValue(state.dI_didata)
1228	self.dI_sqrdata.SetValue(state.dI_sqrdata)
1229	self.dI_idata.SetValue(state.dI_idata)
1230
1231
1232	## we have two more options for smearing
1233	if self.pinhole_smearer.GetValue(): self.onPinholeSmear(event=None)
1234	elif self.slit_smearer.GetValue(): self.onSlitSmear(event=None)
1235
1236	## reset state of checkbox,textcrtl and dispersity parameters value
1237	self._reset_parameters_state(self.fittable_param,state.fittable_param)
1238	self._reset_parameters_state(self.fixed_param,state.fixed_param)
1239
1240	## draw the model with previous parameters value
1241	self._onparamEnter_helper()
1242	#reset the value of chisqr when not consistent with the value computed
1243	self.tcChi.SetValue(str(self.state.tcChi))
1244	## reset context menu items
1245	self._reset_context_menu()
1246
1247	## set the value of the current state to the state given as parameter
1248	self.state = state.clone()
1249
1250	def _reset_page_disp_helper(self, state):
1251	"""
1252	Help to rest page for dispersions
1253	"""
1254	keys = self.model.getParamList()
1255	for item in keys:
1256	if item in self.disp_list and \
1257	not self.model.details.has_key(item):
1258	self.model.details[item] = ["", None, None]
1259	#for k,v in self.state.disp_cb_dict.iteritems():
1260	self.disp_cb_dict = copy.deepcopy(state.disp_cb_dict)
1261	self.state.disp_cb_dict = copy.deepcopy(state.disp_cb_dict)
1262	self.values = copy.deepcopy(state.values)
1263	self.weights = copy.deepcopy(state.weights)
1264
1265	for key, disp in state._disp_obj_dict.iteritems():
1266	# From saved file, disp_model can not be sent in model obj.
1267	# it will be sent as a string here, then converted to model object.
1268	if disp.__class__.__name__ == 'str':
1269	com_str = "from sans.models.dispersion_models "
1270	com_str += "import %s as disp_func"
1271	exec com_str % disp
1272	disp_model = disp_func()
1273	else:
1274	disp_model = disp
1275
1276	self._disp_obj_dict[key] = disp_model
1277	param_name = key.split('.')[0]
1278	# Try to set dispersion only when available
1279	# for eg., pass the orient. angles for 1D Cal
1280	try:
1281	self.model.set_dispersion(param_name, disp_model)
1282	self.model._persistency_dict[key] = \
1283	[state.values, state.weights]
1284	except:
1285	pass
1286	selection = self._find_polyfunc_selection(disp_model)
1287	for list in self.fittable_param:
1288	if list[1] == key and list[7] != None:
1289	list[7].SetSelection(selection)
1290	# For the array disp_model, set the values and weights
1291	if selection == 1:
1292	disp_model.set_weights(self.values[key],
1293	self.weights[key])
1294	try:
1295	# Diables all fittable params for array
1296	list[0].SetValue(False)
1297	list[0].Disable()
1298	list[2].Disable()
1299	list[5].Disable()
1300	list[6].Disable()
1301	except:
1302	pass
1303	# For array, disable all fixed params
1304	if selection == 1:
1305	for item in self.fixed_param:
1306	if item[1].split(".")[0] == key.split(".")[0]:
1307	# try it and pass it for the orientation for 1D
1308	try:
1309	item[2].Disable()
1310	except:
1311	pass
1312
1313	# Make sure the check box updated when all checked
1314	if self.cb1.GetValue():
1315	self.select_all_param(None)
1316
1317	def _selectDlg(self):
1318	"""
1319	open a dialog file to selected the customized dispersity
1320	"""
1321	import os
1322	if self.parent != None:
1323	self._default_save_location = \
1324	self.parent.parent._default_save_location
1325	dlg = wx.FileDialog(self, "Choose a weight file",
1326	self._default_save_location , "",
1327	".", wx.OPEN)
1328	path = None
1329	if dlg.ShowModal() == wx.ID_OK:
1330	path = dlg.GetPath()
1331	dlg.Destroy()
1332	return path
1333
1334	def _reset_context_menu(self):
1335	"""
1336	reset the context menu
1337	"""
1338	for name, state in self.state.saved_states.iteritems():
1339	self.number_saved_state += 1
1340	## Add item in the context menu
1341	id = wx.NewId()
1342	msg = 'Save model and state %g' % self.number_saved_state
1343	self.popUpMenu.Append(id, name, msg)
1344	wx.EVT_MENU(self, id, self.onResetModel)
1345
1346	def _reset_plotting_range(self, state):
1347	"""
1348	Reset the plotting range to a given state
1349	"""
1350	# if self.check_invalid_panel():
1351	# return
1352	self.qmin.SetValue(str(state.qmin))
1353	self.qmax.SetValue(str(state.qmax))
1354
1355	def _save_typeOfmodel(self):
1356	"""
1357	save radiobutton containing the type model that can be selected
1358	"""
1359	self.state.shape_rbutton = self.shape_rbutton.GetValue()
1360	self.state.shape_indep_rbutton = self.shape_indep_rbutton.GetValue()
1361	self.state.struct_rbutton = self.struct_rbutton.GetValue()
1362	self.state.plugin_rbutton = self.plugin_rbutton.GetValue()
1363	self.state.structurebox= self.structurebox.GetCurrentSelection()
1364	self.state.formfactorbox = self.formfactorbox.GetCurrentSelection()
1365
1366	#self._undo.Enable(True)
1367	## post state to fit panel
1368	event = PageInfoEvent(page = self)
1369	wx.PostEvent(self.parent, event)
1370
1371	def _save_plotting_range(self ):
1372	"""
1373	save the state of plotting range
1374	"""
1375	self.state.qmin = self.qmin_x
1376	self.state.qmax = self.qmax_x
1377	self.state.npts = self.npts_x
1378
1379	def _onparamEnter_helper(self):
1380	"""
1381	check if values entered by the user are changed and valid to replot
1382	model
1383	"""
1384	# Flag to register when a parameter has changed.
1385	is_modified = False
1386	self.fitrange = True
1387	is_2Ddata = False
1388	#self._undo.Enable(True)
1389	# check if 2d data
1390	if self.data.__class__.__name__ == "Data2D":
1391	is_2Ddata = True
1392	if self.model !=None:
1393	try:
1394	is_modified = self._check_value_enter(self.fittable_param,
1395	is_modified)
1396	is_modified = self._check_value_enter(self.fixed_param,
1397	is_modified)
1398	is_modified = self._check_value_enter(self.parameters,
1399	is_modified)
1400	except:
1401	pass
1402	#if is_modified:
1403
1404	# Here we should check whether the boundaries have been modified.
1405	# If qmin and qmax have been modified, update qmin and qmax and
1406	# set the is_modified flag to True
1407	if self._validate_qrange(self.qmin, self.qmax):
1408	tempmin = float(self.qmin.GetValue())
1409	if tempmin != self.qmin_x:
1410	self.qmin_x = tempmin
1411	is_modified = True
1412	tempmax = float(self.qmax.GetValue())
1413	if tempmax != self.qmax_x:
1414	self.qmax_x = tempmax
1415	is_modified = True
1416
1417	if is_2Ddata:
1418	# set mask
1419	is_modified = self._validate_Npts()
1420
1421	else:
1422	self.fitrange = False
1423
1424	## if any value is modify draw model with new value
1425	if not self.fitrange:
1426	#self.btFit.Disable()
1427	if is_2Ddata: self.btEditMask.Disable()
1428	else:
1429	#self.btFit.Enable(True)
1430	if is_2Ddata: self.btEditMask.Enable(True)
1431	if is_modified and self.fitrange:
1432	#if self.data == None:
1433	# Theory case: need to get npts value to draw
1434	self.npts_x = float(self.Npts_total.GetValue())
1435	self.create_default_data()
1436	self.state_change= True
1437	self._draw_model()
1438	self.Refresh()
1439	return is_modified
1440
1441	def _update_paramv_on_fit(self):
1442	"""
1443	make sure that update param values just before the fitting
1444	"""
1445	#flag for qmin qmax check values
1446	flag = True
1447	self.fitrange = True
1448	is_modified = False
1449
1450	#wx.PostEvent(self._manager.parent, StatusEvent(status=" \
1451	#updating ... ",type="update"))
1452
1453	##So make sure that update param values on_Fit.
1454	#self._undo.Enable(True)
1455	if self.model !=None:
1456	##Check the values
1457	self._check_value_enter( self.fittable_param ,is_modified)
1458	self._check_value_enter( self.fixed_param ,is_modified)
1459	self._check_value_enter( self.parameters ,is_modified)
1460
1461	# If qmin and qmax have been modified, update qmin and qmax and
1462	# Here we should check whether the boundaries have been modified.
1463	# If qmin and qmax have been modified, update qmin and qmax and
1464	# set the is_modified flag to True
1465	self.fitrange = self._validate_qrange(self.qmin, self.qmax)
1466	if self.fitrange:
1467	tempmin = float(self.qmin.GetValue())
1468	if tempmin != self.qmin_x:
1469	self.qmin_x = tempmin
1470	tempmax = float(self.qmax.GetValue())
1471	if tempmax != self.qmax_x:
1472	self.qmax_x = tempmax
1473	if tempmax == tempmin:
1474	flag = False
1475	temp_smearer = None
1476	if not self.disable_smearer.GetValue():
1477	temp_smearer= self.current_smearer
1478	if self.slit_smearer.GetValue():
1479	flag = self.update_slit_smear()
1480	elif self.pinhole_smearer.GetValue():
1481	flag = self.update_pinhole_smear()
1482	else:
1483	self._manager.set_smearer(smearer=temp_smearer,
1484	uid=self.uid,
1485	fid=self.data.id,
1486	qmin=float(self.qmin_x),
1487	qmax=float(self.qmax_x),
1488	enable_smearer=not self.disable_smearer.GetValue(),
1489	draw=False)
1490	elif not self._is_2D():
1491	self._manager.set_smearer(smearer=temp_smearer,
1492	qmin=float(self.qmin_x),
1493	uid=self.uid,
1494	fid=self.data.id,
1495	qmax= float(self.qmax_x),
1496	enable_smearer=not self.disable_smearer.GetValue(),
1497	draw=False)
1498	if self.data != None:
1499	index_data = ((self.qmin_x <= self.data.x)&\
1500	(self.data.x <= self.qmax_x))
1501	val = str(len(self.data.x[index_data==True]))
1502	self.Npts_fit.SetValue(val)
1503	else:
1504	# No data in the panel
1505	try:
1506	self.npts_x = float(self.Npts_total.GetValue())
1507	except:
1508	flag = False
1509	return flag
1510	flag = True
1511	if self._is_2D():
1512	# only 2D case set mask
1513	flag = self._validate_Npts()
1514	if not flag:
1515	return flag
1516	else: flag = False
1517	else:
1518	flag = False
1519
1520	#For invalid q range, disable the mask editor and fit button, vs.
1521	if not self.fitrange:
1522	#self.btFit.Disable()
1523	if self._is_2D():
1524	self.btEditMask.Disable()
1525	else:
1526	#self.btFit.Enable(True)
1527	if self._is_2D() and self.data != None:
1528	self.btEditMask.Enable(True)
1529
1530	if not flag:
1531	msg = "Cannot Plot or Fit :Must select a "
1532	msg += " model or Fitting range is not valid!!! "
1533	wx.PostEvent(self.parent.parent, StatusEvent(status=msg))
1534
1535	self.save_current_state()
1536
1537	return flag
1538
1539	def _is_modified(self, is_modified):
1540	"""
1541	return to self._is_modified
1542	"""
1543	return is_modified
1544
1545	def _reset_parameters_state(self, listtorestore, statelist):
1546	"""
1547	Reset the parameters at the given state
1548	"""
1549	if len(statelist) == 0 or len(listtorestore) == 0:
1550	return
1551	if len(statelist) != len(listtorestore):
1552	return
1553
1554	for j in range(len(listtorestore)):
1555	item_page = listtorestore[j]
1556	item_page_info = statelist[j]
1557	##change the state of the check box for simple parameters
1558	if item_page[0]!=None:
1559	item_page[0].SetValue(item_page_info[0])
1560	if item_page[2]!=None:
1561	item_page[2].SetValue(item_page_info[2])
1562	if item_page[2].__class__.__name__ == "ComboBox":
1563	if self.model.fun_list.has_key(item_page_info[2]):
1564	fun_val = self.model.fun_list[item_page_info[2]]
1565	self.model.setParam(item_page_info[1],fun_val)
1566	if item_page[3]!=None:
1567	## show or hide text +/-
1568	if item_page_info[2]:
1569	item_page[3].Show(True)
1570	else:
1571	item_page[3].Hide()
1572	if item_page[4]!=None:
1573	## show of hide the text crtl for fitting error
1574	if item_page_info[4][0]:
1575	item_page[4].Show(True)
1576	item_page[4].SetValue(item_page_info[4][1])
1577	else:
1578	item_page[3].Hide()
1579	if item_page[5]!=None:
1580	## show of hide the text crtl for fitting error
1581	item_page[5].Show(item_page_info[5][0])
1582	item_page[5].SetValue(item_page_info[5][1])
1583
1584	if item_page[6]!=None:
1585	## show of hide the text crtl for fitting error
1586	item_page[6].Show(item_page_info[6][0])
1587	item_page[6].SetValue(item_page_info[6][1])
1588
1589
1590	def _reset_strparam_state(self, listtorestore, statelist):
1591	"""
1592	Reset the string parameters at the given state
1593	"""
1594	if len(statelist) == 0:
1595	return
1596
1597	listtorestore = copy.deepcopy(statelist)
1598
1599	for j in range(len(listtorestore)):
1600	item_page = listtorestore[j]
1601	item_page_info = statelist[j]
1602	##change the state of the check box for simple parameters
1603
1604	if item_page[0] != None:
1605	item_page[0].SetValue(format_number(item_page_info[0], True))
1606
1607	if item_page[2] != None:
1608	param_name = item_page_info[1]
1609	value = item_page_info[2]
1610	selection = value
1611	if self.model.fun_list.has_key(value):
1612	selection = self.model.fun_list[value]
1613	item_page[2].SetValue(selection)
1614	self.model.setParam(param_name, selection)
1615
1616	def _copy_parameters_state(self, listtocopy, statelist):
1617	"""
1618	copy the state of button
1619
1620	:param listtocopy: the list of check button to copy
1621	:param statelist: list of state object to store the current state
1622
1623	"""
1624	if len(listtocopy)==0:
1625	return
1626
1627	for item in listtocopy:
1628
1629	checkbox_state = None
1630	if item[0]!= None:
1631	checkbox_state= item[0].GetValue()
1632	parameter_name = item[1]
1633	parameter_value = None
1634	if item[2]!=None:
1635	parameter_value = item[2].GetValue()
1636	static_text = None
1637	if item[3]!=None:
1638	static_text = item[3].IsShown()
1639	error_value = None
1640	error_state = None
1641	if item[4]!= None:
1642	error_value = item[4].GetValue()
1643	error_state = item[4].IsShown()
1644
1645	min_value = None
1646	min_state = None
1647	if item[5]!= None:
1648	min_value = item[5].GetValue()
1649	min_state = item[5].IsShown()
1650
1651	max_value = None
1652	max_state = None
1653	if item[6]!= None:
1654	max_value = item[6].GetValue()
1655	max_state = item[6].IsShown()
1656	unit=None
1657	if item[7]!=None:
1658	unit = item[7].GetLabel()
1659
1660	statelist.append([checkbox_state, parameter_name, parameter_value,
1661	static_text ,[error_state, error_value],
1662	[min_state, min_value],
1663	[max_state, max_value], unit])
1664
1665	def _set_model_sizer_selection(self, model):
1666	"""
1667	Display the sizer according to the type of the current model
1668	"""
1669	if model == None:
1670	return
1671	if hasattr(model ,"s_model"):
1672
1673	class_name = model.s_model.__class__
1674	name = model.s_model.name
1675	flag = (name != "NoStructure")
1676	if flag and \
1677	(class_name in self.model_list_box["Structure Factors"]):
1678	self.structurebox.Show()
1679	self.text2.Show()
1680	self.structurebox.Enable()
1681	self.text2.Enable()
1682	items = self.structurebox.GetItems()
1683	self.sizer1.Layout()
1684
1685	for i in range(len(items)):
1686	if items[i]== str(name):
1687	self.structurebox.SetSelection(i)
1688	break
1689
1690	if hasattr(model ,"p_model"):
1691	class_name = model.p_model.__class__
1692	name = model.p_model.name
1693	self.formfactorbox.Clear()
1694
1695	for k, list in self.model_list_box.iteritems():
1696	if k in["P(Q)*S(Q)","Shapes" ] and class_name in self.model_list_box["Shapes"]:
1697	self.shape_rbutton.SetValue(True)
1698	## fill the form factor list with new model
1699	self._populate_box(self.formfactorbox,self.model_list_box["Shapes"])
1700	items = self.formfactorbox.GetItems()
1701	## set comboxbox to the selected item
1702	for i in range(len(items)):
1703	if items[i]== str(name):
1704	self.formfactorbox.SetSelection(i)
1705	break
1706	return
1707	elif k == "Shape-Independent":
1708	self.shape_indep_rbutton.SetValue(True)
1709	elif k == "Structure Factors":
1710	self.struct_rbutton.SetValue(True)
1711	elif k == "Multi-Functions":
1712	continue
1713	else:
1714	self.plugin_rbutton.SetValue(True)
1715
1716	if class_name in list:
1717	## fill the form factor list with new model
1718	self._populate_box(self.formfactorbox, list)
1719	items = self.formfactorbox.GetItems()
1720	## set comboxbox to the selected item
1721	for i in range(len(items)):
1722	if items[i]== str(name):
1723	self.formfactorbox.SetSelection(i)
1724	break
1725	break
1726	else:
1727
1728	## Select the model from the menu
1729	class_name = model.__class__
1730	name = model.name
1731	self.formfactorbox.Clear()
1732	items = self.formfactorbox.GetItems()
1733
1734	for k, list in self.model_list_box.iteritems():
1735	if k in["P(Q)*S(Q)","Shapes" ] and class_name in self.model_list_box["Shapes"]:
1736	if class_name in self.model_list_box["P(Q)*S(Q)"]:
1737	self.structurebox.Show()
1738	self.text2.Show()
1739	self.structurebox.Enable()
1740	self.structurebox.SetSelection(0)
1741	self.text2.Enable()
1742	else:
1743	self.structurebox.Hide()
1744	self.text2.Hide()
1745	self.structurebox.Disable()
1746	self.structurebox.SetSelection(0)
1747	self.text2.Disable()
1748
1749	self.shape_rbutton.SetValue(True)
1750	## fill the form factor list with new model
1751	self._populate_box(self.formfactorbox,self.model_list_box["Shapes"])
1752	items = self.formfactorbox.GetItems()
1753	## set comboxbox to the selected item
1754	for i in range(len(items)):
1755	if items[i]== str(name):
1756	self.formfactorbox.SetSelection(i)
1757	break
1758	return
1759	elif k == "Shape-Independent":
1760	self.shape_indep_rbutton.SetValue(True)
1761	elif k == "Structure Factors":
1762	self.struct_rbutton.SetValue(True)
1763	elif k == "Multi-Functions":
1764	continue
1765	else:
1766	self.plugin_rbutton.SetValue(True)
1767	if class_name in list:
1768	self.structurebox.SetSelection(0)
1769	self.structurebox.Disable()
1770	self.text2.Disable()
1771	## fill the form factor list with new model
1772	self._populate_box(self.formfactorbox, list)
1773	items = self.formfactorbox.GetItems()
1774	## set comboxbox to the selected item
1775	for i in range(len(items)):
1776	if items[i]== str(name):
1777	self.formfactorbox.SetSelection(i)
1778	break
1779	break
1780
1781	def _draw_model(self, update_chisqr=True):
1782	"""
1783	Method to draw or refresh a plotted model.
1784	The method will use the data member from the model page
1785	to build a call to the fitting perspective manager.
1786
1787	:param chisqr: update chisqr value [bool]
1788	"""
1789	#if self.check_invalid_panel():
1790	# return
1791	if self.model !=None:
1792	temp_smear=None
1793	if hasattr(self, "enable_smearer"):
1794	if not self.disable_smearer.GetValue():
1795	temp_smear= self.current_smearer
1796	# compute weight for the current data
1797	from .utils import get_weight
1798	flag = self.get_weight_flag()
1799	weight = get_weight(data=self.data, is2d=self._is_2D(), flag=flag)
1800	toggle_mode_on = self.model_view.IsEnabled()
1801	self._manager.draw_model(self.model,
1802	data=self.data,
1803	smearer= temp_smear,
1804	qmin=float(self.qmin_x),
1805	qmax=float(self.qmax_x),
1806	page_id=self.uid,
1807	toggle_mode_on=toggle_mode_on,
1808	state = self.state,
1809	enable2D=self.enable2D,
1810	update_chisqr=update_chisqr,
1811	weight=weight)
1812
1813
1814	def _on_show_sld(self, event=None):
1815	"""
1816	Plot SLD profile
1817	"""
1818	# get profile data
1819	x,y=self.model.getProfile()
1820
1821	from danse.common.plottools import Data1D
1822	#from sans.perspectives.theory.profile_dialog import SLDPanel
1823	from sans.guiframe.local_perspectives.plotting.profile_dialog \
1824	import SLDPanel
1825	sld_data = Data1D(x,y)
1826	sld_data.name = 'SLD'
1827	sld_data.axes = self.sld_axes
1828	self.panel = SLDPanel(self, data=sld_data, axes =self.sld_axes,id =-1)
1829	self.panel.ShowModal()
1830
1831	def _set_multfactor_combobox(self, multiplicity=10):
1832	"""
1833	Set comboBox for muitfactor of CoreMultiShellModel
1834	:param multiplicit: no. of multi-functionality
1835	"""
1836	# build content of the combobox
1837	for idx in range(0,multiplicity):
1838	self.multifactorbox.Append(str(idx),int(idx))
1839	#self.multifactorbox.SetSelection(1)
1840	self._hide_multfactor_combobox()
1841
1842	def _show_multfactor_combobox(self):
1843	"""
1844	Show the comboBox of muitfactor of CoreMultiShellModel
1845	"""
1846	if not self.mutifactor_text.IsShown():
1847	self.mutifactor_text.Show(True)
1848	self.mutifactor_text1.Show(True)
1849	if not self.multifactorbox.IsShown():
1850	self.multifactorbox.Show(True)
1851
1852	def _hide_multfactor_combobox(self):
1853	"""
1854	Hide the comboBox of muitfactor of CoreMultiShellModel
1855	"""
1856	if self.mutifactor_text.IsShown():
1857	self.mutifactor_text.Hide()
1858	self.mutifactor_text1.Hide()
1859	if self.multifactorbox.IsShown():
1860	self.multifactorbox.Hide()
1861
1862
1863	def _show_combox_helper(self):
1864	"""
1865	Fill panel's combo box according to the type of model selected
1866	"""
1867	if self.shape_rbutton.GetValue():
1868	##fill the combobox with form factor list
1869	self.structurebox.SetSelection(0)
1870	self.structurebox.Disable()
1871	self.formfactorbox.Clear()
1872	self._populate_box( self.formfactorbox,self.model_list_box["Shapes"])
1873	if self.shape_indep_rbutton.GetValue():
1874	##fill the combobox with shape independent factor list
1875	self.structurebox.SetSelection(0)
1876	self.structurebox.Disable()
1877	self.formfactorbox.Clear()
1878	self._populate_box( self.formfactorbox,
1879	self.model_list_box["Shape-Independent"])
1880	if self.struct_rbutton.GetValue():
1881	##fill the combobox with structure factor list
1882	self.structurebox.SetSelection(0)
1883	self.structurebox.Disable()
1884	self.formfactorbox.Clear()
1885	self._populate_box( self.formfactorbox,
1886	self.model_list_box["Structure Factors"])
1887	if self.plugin_rbutton.GetValue():
1888	##fill the combobox with form factor list
1889	self.structurebox.Disable()
1890	self.formfactorbox.Clear()
1891	self._populate_box( self.formfactorbox,
1892	self.model_list_box["Customized Models"])
1893
1894	def _show_combox(self, event=None):
1895	"""
1896	Show combox box associate with type of model selected
1897	"""
1898	#if self.check_invalid_panel():
1899	# self.shape_rbutton.SetValue(True)
1900	# return
1901
1902	self._show_combox_helper()
1903	self._on_select_model(event=None)
1904	self._save_typeOfmodel()
1905	self.sizer4_4.Layout()
1906	self.sizer4.Layout()
1907	self.Layout()
1908	self.Refresh()
1909
1910	def _populate_box(self, combobox, list):
1911	"""
1912	fill combox box with dict item
1913
1914	:param list: contains item to fill the combox
1915	item must model class
1916	"""
1917	for models in list:
1918	model= models()
1919	name = model.__class__.__name__
1920	if models.__name__!="NoStructure":
1921	if hasattr(model, "name"):
1922	name = model.name
1923	combobox.Append(name,models)
1924	return 0
1925
1926	def _onQrangeEnter(self, event):
1927	"""
1928	Check validity of value enter in the Q range field
1929
1930	"""
1931	tcrtl = event.GetEventObject()
1932	#Clear msg if previously shown.
1933	msg = ""
1934	wx.PostEvent(self.parent, StatusEvent(status=msg))
1935	# Flag to register when a parameter has changed.
1936	is_modified = False
1937	if tcrtl.GetValue().lstrip().rstrip() != "":
1938	try:
1939	value = float(tcrtl.GetValue())
1940	tcrtl.SetBackgroundColour(wx.WHITE)
1941	# If qmin and qmax have been modified, update qmin and qmax
1942	if self._validate_qrange(self.qmin, self.qmax):
1943	tempmin = float(self.qmin.GetValue())
1944	if tempmin != self.qmin_x:
1945	self.qmin_x = tempmin
1946	tempmax = float(self.qmax.GetValue())
1947	if tempmax != self.qmax_x:
1948	self.qmax_x = tempmax
1949	else:
1950	tcrtl.SetBackgroundColour("pink")
1951	msg = "Model Error:wrong value entered : %s" % sys.exc_value
1952	wx.PostEvent(self.parent, StatusEvent(status=msg))
1953	return
1954	except:
1955	tcrtl.SetBackgroundColour("pink")
1956	msg = "Model Error:wrong value entered : %s" % sys.exc_value
1957	wx.PostEvent(self.parent, StatusEvent(status=msg))
1958	return
1959	#Check if # of points for theory model are valid(>0).
1960	if self.npts != None:
1961	if check_float(self.npts):
1962	temp_npts = float(self.npts.GetValue())
1963	if temp_npts != self.num_points:
1964	self.num_points = temp_npts
1965	is_modified = True
1966	else:
1967	msg = "Cannot Plot :No npts in that Qrange!!! "
1968	wx.PostEvent(self.parent, StatusEvent(status=msg))
1969	else:
1970	tcrtl.SetBackgroundColour("pink")
1971	msg = "Model Error:wrong value entered!!!"
1972	wx.PostEvent(self.parent, StatusEvent(status=msg))
1973	#self._undo.Enable(True)
1974	self.save_current_state()
1975	event = PageInfoEvent(page=self)
1976	wx.PostEvent(self.parent, event)
1977	self.state_change = False
1978	#Draw the model for a different range
1979	self.create_default_data()
1980	self._draw_model()
1981
1982	def _theory_qrange_enter(self, event):
1983	"""
1984	Check validity of value enter in the Q range field
1985	"""
1986
1987	tcrtl= event.GetEventObject()
1988	#Clear msg if previously shown.
1989	msg= ""
1990	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
1991	# Flag to register when a parameter has changed.
1992	is_modified = False
1993	if tcrtl.GetValue().lstrip().rstrip()!="":
1994	try:
1995	value = float(tcrtl.GetValue())
1996	tcrtl.SetBackgroundColour(wx.WHITE)
1997
1998	# If qmin and qmax have been modified, update qmin and qmax
1999	if self._validate_qrange(self.theory_qmin, self.theory_qmax):
2000	tempmin = float(self.theory_qmin.GetValue())
2001	if tempmin != self.theory_qmin_x:
2002	self.theory_qmin_x = tempmin
2003	tempmax = float(self.theory_qmax.GetValue())
2004	if tempmax != self.qmax_x:
2005	self.theory_qmax_x = tempmax
2006	else:
2007	tcrtl.SetBackgroundColour("pink")
2008	msg= "Model Error:wrong value entered : %s"% sys.exc_value
2009	wx.PostEvent(self._manager.parent, StatusEvent(status = msg ))
2010	return
2011	except:
2012	tcrtl.SetBackgroundColour("pink")
2013	msg= "Model Error:wrong value entered : %s"% sys.exc_value
2014	wx.PostEvent(self._manager.parent, StatusEvent(status = msg ))
2015	return
2016	#Check if # of points for theory model are valid(>0).
2017	if self.Npts_total.IsEditable() :
2018	if check_float(self.Npts_total):
2019	temp_npts = float(self.Npts_total.GetValue())
2020	if temp_npts != self.num_points:
2021	self.num_points = temp_npts
2022	is_modified = True
2023	else:
2024	msg= "Cannot Plot :No npts in that Qrange!!! "
2025	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
2026	else:
2027	tcrtl.SetBackgroundColour("pink")
2028	msg = "Model Error:wrong value entered!!!"
2029	wx.PostEvent(self.parent.parent, StatusEvent(status=msg))
2030	#self._undo.Enable(True)
2031	self.save_current_state()
2032	event = PageInfoEvent(page = self)
2033	wx.PostEvent(self.parent, event)
2034	self.state_change= False
2035	#Draw the model for a different range
2036	self.create_default_data()
2037	self._draw_model()
2038
2039	def _on_select_model_helper(self):
2040	"""
2041	call back for model selection
2042	"""
2043	## reset dictionary containing reference to dispersion
2044	self._disp_obj_dict = {}
2045	self.disp_cb_dict ={}
2046	self.temp_multi_functional = False
2047	f_id = self.formfactorbox.GetCurrentSelection()
2048	#For MAC
2049	form_factor = None
2050	if f_id >= 0:
2051	form_factor = self.formfactorbox.GetClientData(f_id)
2052
2053	if not form_factor in self.model_list_box["multiplication"]:
2054	self.structurebox.Hide()
2055	self.text2.Hide()
2056	self.structurebox.Disable()
2057	self.structurebox.SetSelection(0)
2058	self.text2.Disable()
2059	else:
2060	self.structurebox.Show()
2061	self.text2.Show()
2062	self.structurebox.Enable()
2063	self.text2.Enable()
2064
2065	if form_factor != None:
2066	# set multifactor for Mutifunctional models
2067	if form_factor().__class__ in self.model_list_box["Multi-Functions"]:
2068	m_id = self.multifactorbox.GetCurrentSelection()
2069	multiplicity = form_factor().multiplicity_info[0]
2070	self.multifactorbox.Clear()
2071	#self.mutifactor_text.SetLabel(form_factor().details[])
2072	self._set_multfactor_combobox(multiplicity)
2073	self._show_multfactor_combobox()
2074	#ToDo: this info should be called directly from the model
2075	text = form_factor().multiplicity_info[1]#'No. of Shells: '
2076
2077	#self.mutifactor_text.Clear()
2078	self.mutifactor_text.SetLabel(text)
2079	if m_id > multiplicity -1:
2080	# default value
2081	m_id = 1
2082
2083	self.multi_factor = self.multifactorbox.GetClientData(m_id)
2084	if self.multi_factor == None: self.multi_factor =0
2085	form_factor = form_factor(int(self.multi_factor))
2086	self.multifactorbox.SetSelection(m_id)
2087	# Check len of the text1 and max_multiplicity
2088	text = ''
2089	if form_factor.multiplicity_info[0] == len(form_factor.multiplicity_info[2]):
2090	text = form_factor.multiplicity_info[2][self.multi_factor]
2091	self.mutifactor_text1.SetLabel(text)
2092	# Check if model has get sld profile.
2093	if len(form_factor.multiplicity_info[3]) > 0:
2094	self.sld_axes = form_factor.multiplicity_info[3]
2095	self.show_sld_button.Show(True)
2096	else:
2097	self.sld_axes = ""
2098
2099	else:
2100	self._hide_multfactor_combobox()
2101	self.show_sld_button.Hide()
2102	form_factor = form_factor()
2103	self.multi_factor = None
2104	else:
2105	self._hide_multfactor_combobox()
2106	self.show_sld_button.Hide()
2107	self.multi_factor = None
2108
2109	s_id = self.structurebox.GetCurrentSelection()
2110	struct_factor = self.structurebox.GetClientData( s_id )
2111
2112	if struct_factor !=None:
2113	from sans.models.MultiplicationModel import MultiplicationModel
2114	self.model= MultiplicationModel(form_factor,struct_factor())
2115	# multifunctional form factor
2116	if len(form_factor.non_fittable) > 0:
2117	self.temp_multi_functional = True
2118	else:
2119	if form_factor != None:
2120	self.model= form_factor
2121	else:
2122	self.model = None
2123	return self.model
2124
2125
2126	## post state to fit panel
2127	self.state.parameters =[]
2128	self.state.model =self.model
2129	self.state.qmin = self.qmin_x
2130	self.state.multi_factor = self.multi_factor
2131	self.disp_list =self.model.getDispParamList()
2132	self.state.disp_list = self.disp_list
2133	self.on_set_focus(None)
2134	self.Layout()
2135
2136	def _validate_qrange(self, qmin_ctrl, qmax_ctrl):
2137	"""
2138	Verify that the Q range controls have valid values
2139	and that Qmin < Qmax.
2140
2141	:param qmin_ctrl: text control for Qmin
2142	:param qmax_ctrl: text control for Qmax
2143
2144	:return: True is the Q range is value, False otherwise
2145
2146	"""
2147	qmin_validity = check_float(qmin_ctrl)
2148	qmax_validity = check_float(qmax_ctrl)
2149	if not (qmin_validity and qmax_validity):
2150	return False
2151	else:
2152	qmin = float(qmin_ctrl.GetValue())
2153	qmax = float(qmax_ctrl.GetValue())
2154	if qmin < qmax:
2155	#Make sure to set both colours white.
2156	qmin_ctrl.SetBackgroundColour(wx.WHITE)
2157	qmin_ctrl.Refresh()
2158	qmax_ctrl.SetBackgroundColour(wx.WHITE)
2159	qmax_ctrl.Refresh()
2160	else:
2161	qmin_ctrl.SetBackgroundColour("pink")
2162	qmin_ctrl.Refresh()
2163	qmax_ctrl.SetBackgroundColour("pink")
2164	qmax_ctrl.Refresh()
2165	msg= "Invalid Q range: Q min must be smaller than Q max"
2166	wx.PostEvent(self.parent.parent, StatusEvent(status = msg))
2167	return False
2168	return True
2169
2170	def _validate_Npts(self):
2171	"""
2172	Validate the number of points for fitting is more than 10 points.
2173	If valid, setvalues Npts_fit otherwise post msg.
2174	"""
2175	#default flag
2176	flag = True
2177	# Theory
2178	if self.data == None and self.enable2D:
2179	return flag
2180
2181	# q value from qx and qy
2182	radius= numpy.sqrt( self.data.qx_data * self.data.qx_data +
2183	self.data.qy_data * self.data.qy_data )
2184	#get unmasked index
2185	index_data = (float(self.qmin.GetValue()) <= radius) & \
2186	(radius <= float(self.qmax.GetValue()))
2187	index_data = (index_data) & (self.data.mask)
2188	index_data = (index_data) & (numpy.isfinite(self.data.data))
2189
2190	if len(index_data[index_data]) < 10:
2191	# change the color pink.
2192	self.qmin.SetBackgroundColour("pink")
2193	self.qmin.Refresh()
2194	self.qmax.SetBackgroundColour("pink")
2195	self.qmax.Refresh()
2196	msg= "Cannot Plot :No or too little npts in that data range!!! "
2197	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
2198	self.fitrange = False
2199	flag = False
2200	else:
2201	self.Npts_fit.SetValue(str(len(self.data.mask[index_data==True])))
2202	self.fitrange = True
2203
2204	return flag
2205
2206	def _check_value_enter(self, list, modified):
2207	"""
2208	:param list: model parameter and panel info
2209	:Note: each item of the list should be as follow:
2210	item=[check button state, parameter's name,
2211	paramater's value, string="+/-",
2212	parameter's error of fit,
2213	parameter's minimum value,
2214	parrameter's maximum value ,
2215	parameter's units]
2216	"""
2217	is_modified = modified
2218	if len(list)==0:
2219	return is_modified
2220	for item in list:
2221	#skip angle parameters for 1D
2222	if not self.enable2D:#self.data.__class__.__name__ !="Data2D":
2223	if item in self.orientation_params:
2224	continue
2225	#try:
2226	name = str(item[1])
2227
2228	if string.find(name,".npts") ==-1 and \
2229	string.find(name,".nsigmas")==-1:
2230	## check model parameters range
2231	param_min= None
2232	param_max= None
2233
2234	## check minimun value
2235	if item[5]!= None and item[5]!= "":
2236	if item[5].GetValue().lstrip().rstrip()!="":
2237	try:
2238
2239	param_min = float(item[5].GetValue())
2240	if not self._validate_qrange(item[5],item[2]):
2241	if numpy.isfinite(param_min):
2242	item[2].SetValue(format_number(param_min))
2243
2244	item[5].SetBackgroundColour(wx.WHITE)
2245	item[2].SetBackgroundColour(wx.WHITE)
2246
2247	except:
2248	msg = "Wrong Fit parameter range entered "
2249	wx.PostEvent(self.parent.parent,
2250	StatusEvent(status = msg))
2251	raise ValueError, msg
2252	is_modified = True
2253	## check maximum value
2254	if item[6]!= None and item[6]!= "":
2255	if item[6].GetValue().lstrip().rstrip()!="":
2256	try:
2257	param_max = float(item[6].GetValue())
2258	if not self._validate_qrange(item[2],item[6]):
2259	if numpy.isfinite(param_max):
2260	item[2].SetValue(format_number(param_max))
2261
2262	item[6].SetBackgroundColour(wx.WHITE)
2263	item[2].SetBackgroundColour(wx.WHITE)
2264	except:
2265	msg = "Wrong Fit parameter range entered "
2266	wx.PostEvent(self.parent.parent,
2267	StatusEvent(status = msg))
2268	raise ValueError, msg
2269	is_modified = True
2270
2271
2272	if param_min != None and param_max !=None:
2273	if not self._validate_qrange(item[5], item[6]):
2274	msg= "Wrong Fit range entered for parameter "
2275	msg+= "name %s of model %s "%(name, self.model.name)
2276	wx.PostEvent(self.parent.parent,
2277	StatusEvent(status = msg))
2278
2279	if name in self.model.details.keys():
2280	self.model.details[name][1:3] = param_min, param_max
2281	is_modified = True
2282
2283	else:
2284	self.model.details [name] = ["", param_min, param_max]
2285	is_modified = True
2286	try:
2287	# Check if the textctr is enabled
2288	if item[2].IsEnabled():
2289	value= float(item[2].GetValue())
2290	item[2].SetBackgroundColour("white")
2291	# If the value of the parameter has changed,
2292	# +update the model and set the is_modified flag
2293	if value != self.model.getParam(name) and \
2294	numpy.isfinite(value):
2295	self.model.setParam(name, value)
2296
2297	except:
2298	item[2].SetBackgroundColour("pink")
2299	msg = "Wrong Fit parameter value entered "
2300	wx.PostEvent(self.parent.parent, StatusEvent(status = msg))
2301
2302	return is_modified
2303
2304
2305	def _set_dipers_Param(self, event):
2306	"""
2307	respond to self.enable_disp and self.disable_disp radio box.
2308	The dispersity object is reset inside the model into Gaussian.
2309	When the user select yes , this method display a combo box for more selection
2310	when the user selects No,the combo box disappears.
2311	Redraw the model with the default dispersity (Gaussian)
2312	"""
2313	#if self.check_invalid_panel():
2314	# return
2315	## On selction if no model exists.
2316	if self.model ==None:
2317	self.disable_disp.SetValue(True)
2318	msg="Please select a Model first..."
2319	wx.MessageBox(msg, 'Info')
2320	wx.PostEvent(self._manager.parent, StatusEvent(status=\
2321	"Polydispersion: %s"%msg))
2322	return
2323
2324	self._reset_dispersity()
2325
2326	if self.model ==None:
2327	self.model_disp.Hide()
2328	self.sizer4_4.Clear(True)
2329	return
2330
2331	if self.enable_disp.GetValue():
2332	## layout for model containing no dispersity parameters
2333
2334	self.disp_list= self.model.getDispParamList()
2335
2336	if len(self.disp_list)==0 and len(self.disp_cb_dict)==0:
2337	self._layout_sizer_noDipers()
2338	else:
2339	## set gaussian sizer
2340	self._on_select_Disp(event=None)
2341	else:
2342	self.sizer4_4.Clear(True)
2343
2344	## post state to fit panel
2345	self.save_current_state()
2346	if event !=None:
2347	#self._undo.Enable(True)
2348	event = PageInfoEvent(page = self)
2349	wx.PostEvent(self.parent, event)
2350	#draw the model with the current dispersity
2351	self._draw_model()
2352	self.sizer4_4.Layout()
2353	self.sizer5.Layout()
2354	self.Layout()
2355	self.Refresh()
2356
2357
2358	def _layout_sizer_noDipers(self):
2359	"""
2360	Draw a sizer with no dispersity info
2361	"""
2362	ix=0
2363	iy=1
2364	self.fittable_param=[]
2365	self.fixed_param=[]
2366	self.orientation_params_disp=[]
2367
2368	self.sizer4_4.Clear(True)
2369	text = "No polydispersity available for this model"
2370	text = "No polydispersity available for this model"
2371	model_disp = wx.StaticText(self, -1, text)
2372	self.sizer4_4.Add(model_disp,( iy, ix),(1,1),
2373	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 10)
2374	self.sizer4_4.Layout()
2375	self.sizer4.Layout()
2376
2377	def _reset_dispersity(self):
2378	"""
2379	put gaussian dispersity into current model
2380	"""
2381	if len(self.param_toFit)>0:
2382	for item in self.fittable_param:
2383	if item in self.param_toFit:
2384	self.param_toFit.remove(item)
2385
2386	for item in self.orientation_params_disp:
2387	if item in self.param_toFit:
2388	self.param_toFit.remove(item)
2389
2390	self.fittable_param=[]
2391	self.fixed_param=[]
2392	self.orientation_params_disp=[]
2393	self.values={}
2394	self.weights={}
2395
2396	from sans.models.dispersion_models import GaussianDispersion, ArrayDispersion
2397	if len(self.disp_cb_dict)==0:
2398	self.save_current_state()
2399	self.sizer4_4.Clear(True)
2400	self.Layout()
2401
2402	return
2403	if (len(self.disp_cb_dict)>0) :
2404	for p in self.disp_cb_dict:
2405	# The parameter was un-selected. Go back to Gaussian model (with 0 pts)
2406	disp_model = GaussianDispersion()
2407
2408	self._disp_obj_dict[p] = disp_model
2409	# Set the new model as the dispersion object for the selected parameter
2410	try:
2411	self.model.set_dispersion(p, disp_model)
2412	except:
2413
2414	pass
2415
2416	## save state into
2417	self.save_current_state()
2418	self.Layout()
2419	self.Refresh()
2420
2421	def _on_select_Disp(self,event):
2422	"""
2423	allow selecting different dispersion
2424	self.disp_list should change type later .now only gaussian
2425	"""
2426	self._set_sizer_dispersion()
2427
2428	## Redraw the model
2429	self._draw_model()
2430	#self._undo.Enable(True)
2431	event = PageInfoEvent(page = self)
2432	wx.PostEvent(self.parent, event)
2433
2434	self.sizer4_4.Layout()
2435	self.sizer4.Layout()
2436	self.SetupScrolling()
2437
2438	def _on_disp_func(self, event=None):
2439	"""
2440	Select a distribution function for the polydispersion
2441
2442	:Param event: ComboBox event
2443	"""
2444	# get ready for new event
2445	if event != None:
2446	event.Skip()
2447	# Get event object
2448	disp_box = event.GetEventObject()
2449
2450	# Try to select a Distr. function
2451	try:
2452	disp_box.SetBackgroundColour("white")
2453	selection = disp_box.GetCurrentSelection()
2454	param_name = disp_box.Name.split('.')[0]
2455	disp_name = disp_box.GetValue()
2456	dispersity= disp_box.GetClientData(selection)
2457
2458	#disp_model = GaussianDispersion()
2459	disp_model = dispersity()
2460	# Get param names to reset the values of the param
2461	name1 = param_name + ".width"
2462	name2 = param_name + ".npts"
2463	name3 = param_name + ".nsigmas"
2464	# Check Disp. function whether or not it is 'array'
2465	if disp_name.lower() == "array":
2466	value2= ""
2467	value3= ""
2468	value1 = self._set_array_disp(name=name1, disp=disp_model)
2469	else:
2470	self._del_array_values(name1)
2471	#self._reset_array_disp(param_name)
2472	self._disp_obj_dict[name1] = disp_model
2473	self.model.set_dispersion(param_name, disp_model)
2474	self.state._disp_obj_dict[name1]= disp_model
2475
2476	value1= str(format_number(self.model.getParam(name1), True))
2477	value2= str(format_number(self.model.getParam(name2)))
2478	value3= str(format_number(self.model.getParam(name3)))
2479	# Reset fittable polydispersin parameter value
2480	for item in self.fittable_param:
2481	if item[1] == name1:
2482	item[2].SetValue(value1)
2483	item[5].SetValue("")
2484	item[6].SetValue("")
2485	# Disable for array
2486	if disp_name.lower() == "array":
2487	item[0].SetValue(False)
2488	item[0].Disable()
2489	item[2].Disable()
2490	item[5].Disable()
2491	item[6].Disable()
2492	else:
2493	item[0].Enable()
2494	item[2].Enable()
2495	item[5].Enable()
2496	item[6].Enable()
2497	break
2498	# Reset fixed polydispersion params
2499	for item in self.fixed_param:
2500	if item[1] == name2:
2501	item[2].SetValue(value2)
2502	# Disable Npts for array
2503	if disp_name.lower() == "array":
2504	item[2].Disable()
2505	else:
2506	item[2].Enable()
2507	if item[1] == name3:
2508	item[2].SetValue(value3)
2509	# Disable Nsigs for array
2510	if disp_name.lower() == "array":
2511	item[2].Disable()
2512	else:
2513	item[2].Enable()
2514
2515	# Make sure the check box updated when all checked
2516	if self.cb1.GetValue():
2517	self.select_all_param(None)
2518
2519	# update params
2520	self._update_paramv_on_fit()
2521	# draw
2522	self._draw_model()
2523	self.Refresh()
2524	except:
2525	# Error msg
2526	msg = "Error occurred:"
2527	msg += " Could not select the distribution function..."
2528	msg += " Please select another distribution function."
2529	disp_box.SetBackgroundColour("pink")
2530	# Focus on Fit button so that users can see the pinky box
2531	self.btFit.SetFocus()
2532	wx.PostEvent(self.parent.parent,
2533	StatusEvent(status=msg, info="error"))
2534
2535
2536	def _set_array_disp(self, name=None, disp=None):
2537	"""
2538	Set array dispersion
2539
2540	:param name: name of the parameter for the dispersion to be set
2541	:param disp: the polydisperion object
2542	"""
2543	# The user wants this parameter to be averaged.
2544	# Pop up the file selection dialog.
2545	path = self._selectDlg()
2546	# Array data
2547	values = []
2548	weights = []
2549	# If nothing was selected, just return
2550	if path is None:
2551	self.disp_cb_dict[name].SetValue(False)
2552	#self.noDisper_rbox.SetValue(True)
2553	return
2554	self._default_save_location = os.path.dirname(path)
2555	if self.parent != None:
2556	self.parent.parent._default_save_location =\
2557	self._default_save_location
2558
2559	basename = os.path.basename(path)
2560	values,weights = self.read_file(path)
2561
2562	# If any of the two arrays is empty, notify the user that we won't
2563	# proceed
2564	if len(self.param_toFit)>0:
2565	if name in self.param_toFit:
2566	self.param_toFit.remove(name)
2567
2568	# Tell the user that we are about to apply the distribution
2569	msg = "Applying loaded %s distribution: %s" % (name, path)
2570	wx.PostEvent(self.parent.parent, StatusEvent(status=msg))
2571
2572	disp.set_weights(values, weights)
2573	self._disp_obj_dict[name] = disp
2574	self.model.set_dispersion(name.split('.')[0], disp)
2575	self.state._disp_obj_dict[name]= disp
2576	self.values[name] = values
2577	self.weights[name] = weights
2578	# Store the object to make it persist outside the
2579	# scope of this method
2580	#TODO: refactor model to clean this up?
2581	self.state.values = {}
2582	self.state.weights = {}
2583	self.state.values = copy.deepcopy(self.values)
2584	self.state.weights = copy.deepcopy(self.weights)
2585
2586	# Set the new model as the dispersion object for the
2587	#selected parameter
2588	#self.model.set_dispersion(p, disp_model)
2589	# Store a reference to the weights in the model object
2590	#so that
2591	# it's not lost when we use the model within another thread.
2592	#TODO: total hack - fix this
2593	self.state.model= self.model.clone()
2594	self.model._persistency_dict[name.split('.')[0]] = \
2595	[values, weights]
2596	self.state.model._persistency_dict[name.split('.')[0]] = \
2597	[values,weights]
2598	return basename
2599
2600	def _del_array_values(self, name=None):
2601	"""
2602	Reset array dispersion
2603
2604	:param name: name of the parameter for the dispersion to be set
2605	"""
2606	# Try to delete values and weight of the names array dic if exists
2607	try:
2608	del self.values[name]
2609	del self.weights[name]
2610	# delete all other dic
2611	del self.state.values[name]
2612	del self.state.weights[name]
2613	del self.model._persistency_dict[name.split('.')[0]]
2614	del self.state.model._persistency_dict[name.split('.')[0]]
2615	except:
2616	pass
2617
2618	def _lay_out(self):
2619	"""
2620	returns self.Layout
2621
2622	:Note: Mac seems to like this better when self.
2623	Layout is called after fitting.
2624	"""
2625	self._sleep4sec()
2626	self.Layout()
2627	return
2628
2629	def _sleep4sec(self):
2630	"""
2631	sleep for 1 sec only applied on Mac
2632	Note: This 1sec helps for Mac not to crash on self.:ayout after self._draw_model
2633	"""
2634	if ON_MAC == True:
2635	time.sleep(1)
2636
2637	def _find_polyfunc_selection(self, disp_func = None):
2638	"""
2639	FInd Comboox selection from disp_func
2640
2641	:param disp_function: dispersion distr. function
2642	"""
2643	# List of the poly_model name in the combobox
2644	list = ["RectangleDispersion", "ArrayDispersion",
2645	"LogNormalDispersion", "GaussianDispersion",
2646	"SchulzDispersion"]
2647
2648	# Find the selection
2649	try:
2650	selection = list.index(disp_func.__class__.__name__)
2651	return selection
2652	except:
2653	return 3
2654
2655	def on_reset_clicked(self,event):
2656	"""
2657	On 'Reset' button for Q range clicked
2658	"""
2659	flag = True
2660	#if self.check_invalid_panel():
2661	# return
2662	##For 3 different cases: Data2D, Data1D, and theory
2663	if self.model == None:
2664	msg="Please select a model first..."
2665	wx.MessageBox(msg, 'Info')
2666	flag = False
2667	return
2668
2669	elif self.data.__class__.__name__ == "Data2D":
2670	data_min= 0
2671	x= max(math.fabs(self.data.xmin), math.fabs(self.data.xmax))
2672	y= max(math.fabs(self.data.ymin), math.fabs(self.data.ymax))
2673	self.qmin_x = data_min
2674	self.qmax_x = math.sqrt(xx + yy)
2675	# check smearing
2676	if not self.disable_smearer.GetValue():
2677	temp_smearer= self.current_smearer
2678	## set smearing value whether or not the data contain the smearing info
2679	if self.pinhole_smearer.GetValue():
2680	flag = self.update_pinhole_smear()
2681	else:
2682	flag = True
2683
2684	elif self.data == None:
2685	self.qmin_x = _QMIN_DEFAULT
2686	self.qmax_x = _QMAX_DEFAULT
2687	self.num_points = _NPTS_DEFAULT
2688	self.state.npts = self.num_points
2689
2690	elif self.data.__class__.__name__ != "Data2D":
2691	self.qmin_x = min(self.data.x)
2692	self.qmax_x = max(self.data.x)
2693	# check smearing
2694	if not self.disable_smearer.GetValue():
2695	temp_smearer= self.current_smearer
2696	## set smearing value whether or not the data contain the smearing info
2697	if self.slit_smearer.GetValue():
2698	flag = self.update_slit_smear()
2699	elif self.pinhole_smearer.GetValue():
2700	flag = self.update_pinhole_smear()
2701	else:
2702	flag = True
2703	else:
2704	flag = False
2705
2706	if flag == False:
2707	msg= "Cannot Plot :Must enter a number!!! "
2708	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
2709	else:
2710	# set relative text ctrs.
2711	self.qmin.SetValue(str(self.qmin_x))
2712	self.qmax.SetValue(str(self.qmax_x))
2713	self.set_npts2fit()
2714	# At this point, some button and variables satatus (disabled?) should be checked
2715	# such as color that should be reset to white in case that it was pink.
2716	self._onparamEnter_helper()
2717
2718	self.save_current_state()
2719	self.state.qmin = self.qmin_x
2720	self.state.qmax = self.qmax_x
2721
2722	#reset the q range values
2723	self._reset_plotting_range(self.state)
2724	#self.compute_chisqr(smearer=self.current_smearer)
2725	#Re draw plot
2726	self._draw_model()
2727
2728	def get_images(self):
2729	"""
2730	Get the images of the plots corresponding this panel for report
2731
2732	: return graphs: list of figures
2733	: TODO: Move to guiframe
2734	"""
2735	# set list of graphs
2736	graphs = []
2737	canvases = []
2738	# call gui_manager
2739	gui_manager = self.parent.parent
2740	# loops through the panels [dic]
2741	for item1, item2 in gui_manager.panels.iteritems():
2742	data_title = self.data.group_id
2743	data_name = str(self.data.name).split(" [")[0]
2744	# try to get all plots belonging to this control panel
2745	try:
2746	title = ''
2747	# check titles (main plot)
2748	if hasattr(item2,"data2D"):
2749	title = item2.data2D.title
2750	# and data_names (model plot[2D], and residuals)
2751	if item2.group_id == data_title or \
2752	title.count(data_name) or \
2753	item2.window_name.count(data_name) or \
2754	item2.window_caption.count(data_name):
2755	#panel = gui_manager._mgr.GetPane(item2.window_name)
2756	# append to the list
2757	graphs.append(item2.figure)
2758	canvases.append(item2.canvas)
2759	except:
2760	# Not for control panels
2761	pass
2762	# return the list of graphs
2763	return graphs, canvases
2764
2765	def on_model_help_clicked(self,event):
2766	"""
2767	on 'More details' button
2768	"""
2769	from help_panel import HelpWindow
2770	from sans.models import get_data_path
2771
2772	# Get models help model_function path
2773	path = get_data_path(media='media')
2774	model_path = os.path.join(path,"model_functions.html")
2775	if self.model == None:
2776	name = 'FuncHelp'
2777	else:
2778	name = self.formfactorbox.GetValue()
2779	#name = self.model.__class__.__name__
2780	frame = HelpWindow(None, -1, pageToOpen=model_path)
2781	frame.Show(True)
2782	if frame.rhelp.HasAnchor(name):
2783	frame.rhelp.ScrollToAnchor(name)
2784	else:
2785	msg= "Model does not contains an available description "
2786	msg +="Please try searching in the Help window"
2787	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
2788
2789	def on_pd_help_clicked(self, event):
2790	"""
2791	Button event for PD help
2792	"""
2793	from help_panel import HelpWindow
2794	import sans.models as models
2795
2796	# Get models help model_function path
2797	path = models.get_data_path(media='media')
2798	pd_path = os.path.join(path,"pd_help.html")
2799
2800	frame = HelpWindow(None, -1, pageToOpen=pd_path)
2801	frame.Show(True)
2802
2803	def on_left_down(self, event):
2804	"""
2805	Get key stroke event
2806	"""
2807	# Figuring out key combo: Cmd for copy, Alt for paste
2808	if event.CmdDown() and event.ShiftDown():
2809	flag = self.get_paste()
2810	elif event.CmdDown():
2811	flag = self.get_copy()
2812	else:
2813	event.Skip()
2814	return
2815	# make event free
2816	event.Skip()
2817
2818
2819	def get_copy(self):
2820	"""
2821	Get the string copies of the param names and values in the tap
2822	"""
2823	content = 'sansview_parameter_values:'
2824	# Do it if params exist
2825	if self.parameters !=[]:
2826
2827	# go through the parameters
2828	string = self._get_copy_helper(self.parameters,
2829	self.orientation_params)
2830	content += string
2831
2832	# go through the fittables
2833	string = self._get_copy_helper(self.fittable_param,
2834	self.orientation_params_disp)
2835	content += string
2836
2837	# go through the fixed params
2838	string = self._get_copy_helper(self.fixed_param,
2839	self.orientation_params_disp)
2840	content += string
2841
2842	# go through the str params
2843	string = self._get_copy_helper(self.str_parameters,
2844	self.orientation_params)
2845	content += string
2846
2847	if wx.TheClipboard.Open():
2848	wx.TheClipboard.SetData(wx.TextDataObject(str(content)))
2849	data = wx.TextDataObject()
2850	success = wx.TheClipboard.GetData(data)
2851	text = data.GetText()
2852	wx.TheClipboard.Close()
2853
2854	return None
2855
2856	def _get_copy_helper(self, param, orient_param):
2857	"""
2858	Helping get value and name of the params
2859
2860	: param param: parameters
2861	: param orient_param = oritational params
2862	: return content: strings [list] [name,value:....]
2863	"""
2864	content = ''
2865	# go through the str params
2866	for item in param:
2867	# 2D
2868	if self.data.__class__.__name__== "Data2D":
2869	name = item[1]
2870	value = item[2].GetValue()
2871	# 1D
2872	else:
2873	## for 1D all parameters except orientation
2874	if not item[1] in orient_param:
2875	name = item[1]
2876	value = item[2].GetValue()
2877	# add to the content
2878	content += name + ',' + value + ':'
2879
2880	return content
2881
2882	def get_paste(self):
2883	"""
2884	Get the string copies of the param names and values in the tap
2885	"""
2886	context = {}
2887	text = ""
2888
2889	# Get text from the clip board
2890	if wx.TheClipboard.Open():
2891	if wx.TheClipboard.IsSupported(wx.DataFormat(wx.DF_TEXT)):
2892	data = wx.TextDataObject()
2893	# get wx dataobject
2894	success = wx.TheClipboard.GetData(data)
2895	# get text
2896	text = data.GetText()
2897	# close clipboard
2898	wx.TheClipboard.Close()
2899
2900	# put the text into dictionary
2901	lines = text.split(':')
2902	if lines[0] != 'sansview_parameter_values':
2903	self._copy_info(False)
2904	return False
2905	for line in lines[1:-1]:
2906	if len(line) != 0:
2907	item =line.split(',')
2908	name = item[0]
2909	value = item[1]
2910	# Transfer the text to content[dictionary]
2911	context[name] = value
2912
2913	# Do it if params exist
2914	if self.parameters != []:
2915	# go through the parameters
2916	self._get_paste_helper(self.parameters,
2917	self.orientation_params, context)
2918
2919	# go through the fittables
2920	self._get_paste_helper(self.fittable_param,
2921	self.orientation_params_disp, context)
2922
2923	# go through the fixed params
2924	self._get_paste_helper(self.fixed_param,
2925	self.orientation_params_disp, context)
2926
2927	# go through the str params
2928	self._get_paste_helper(self.str_parameters,
2929	self.orientation_params, context)
2930	self._copy_info(True)
2931	return True
2932
2933	def _get_paste_helper(self, param, orient_param, content):
2934	"""
2935	Helping set values of the params
2936
2937	: param param: parameters
2938	: param orient_param: oritational params
2939	: param content: dictionary [ name, value: name1.value1,...]
2940	"""
2941	# go through the str params
2942	for item in param:
2943	# 2D
2944	if self.data.__class__.__name__== "Data2D":
2945	name = item[1]
2946	if name in content.keys():
2947	item[2].SetValue(content[name])
2948	# 1D
2949	else:
2950	## for 1D all parameters except orientation
2951	if not item[1] in orient_param:
2952	name = item[1]
2953	if name in content.keys():
2954	# Avoid changing combox content which needs special care
2955	value = content[name]
2956	item[2].SetValue(value)
2957	if item[2].__class__.__name__ == "ComboBox":
2958	if self.model.fun_list.has_key(value):
2959	fun_val = self.model.fun_list[value]
2960	self.model.setParam(name,fun_val)
2961	# save state
2962	#self._copy_parameters_state(self.str_parameters,
2963	# self.state.str_parameters)
2964

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

SasView

source: sasview/fittingview/src/sans/perspectives/fitting/basepage.py @ e16e5c3

Download in other formats: