gen_scatter_panel.py @ b199493

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 b199493 was 7a04dbb, checked in by krzywon, 10 years ago

Removed the last vestiges of the SANS references and ran run.py. Tested
loading multiple file types and ran fits, p(r) and inversion. Help
windows loaded properly. Saving and loading projects and analyses acted
the same as before the conversion.

All unit tests run by utest_sasview.py passed.

Property mode set to 100644

File size: 77.0 KB

Line
1	"""
2	Generic Scattering panel.
3	This module relies on guiframe manager.
4	"""
5
6	import wx
7	import sys
8	import os
9	import numpy
10	#import math
11	import wx.aui as aui
12	#import wx.lib.agw.aui as aui
13	import logging
14	import time
15
16	import matplotlib
17	matplotlib.interactive(False)
18	#Use the WxAgg back end. The Wx one takes too long to render
19	matplotlib.use('WXAgg')
20
21	#from sas.guiframe.gui_manager import MDIFrame
22	from sas.data_util.calcthread import CalcThread
23	from sas.guiframe.local_perspectives.plotting.SimplePlot import PlotFrame
24	from sas.guiframe.dataFitting import Data2D
25	from sas.guiframe.dataFitting import Data1D
26	from sas.dataloader.data_info import Detector
27	from sas.dataloader.data_info import Source
28	from sas.guiframe.panel_base import PanelBase
29	from sas.guiframe.utils import format_number
30	from sas.guiframe.events import StatusEvent
31	from sas.calculator import sas_gen
32	from sas.perspectives.calculator.calculator_widgets import OutputTextCtrl
33	from sas.perspectives.calculator.calculator_widgets import InputTextCtrl
34	from wx.lib.scrolledpanel import ScrolledPanel
35	from sas.perspectives.calculator.load_thread import GenReader
36	from sas.plottools.arrow3d import Arrow3D
37	from sas.perspectives.calculator import calculator_widgets as widget
38	from sas.guiframe.events import NewPlotEvent
39
40	_BOX_WIDTH = 76
41	#Slit length panel size
42	if sys.platform.count("win32") > 0:
43	PANEL_WIDTH = 570
44	PANEL_HEIGHT = 370
45	FONT_VARIANT = 0
46	else:
47	PANEL_WIDTH = 620
48	PANEL_HEIGHT = 370
49	FONT_VARIANT = 1
50	_QMAX_DEFAULT = 0.3
51	_NPTS_DEFAULT = 50
52	_Q1D_MIN = 0.001
53
54	def add_icon(parent, frame):
55	"""
56	Add icon in the frame
57	"""
58	if parent != None:
59	if hasattr(frame, "IsIconized"):
60	if not frame.IsIconized():
61	try:
62	icon = parent.GetIcon()
63	frame.SetIcon(icon)
64	except:
65	pass
66
67	def _set_error(panel, item, show_msg=False):
68	"""
69	Set_error dialog
70	"""
71	if item != None:
72	item.SetBackgroundColour("pink")
73	item.Refresh()
74	if show_msg:
75	msg = "Error: wrong (or out of range) value entered."
76	if panel.parent.parent != None:
77	wx.PostEvent(panel.parent.parent,
78	StatusEvent(status=msg, info='Error' ))
79	panel.SetFocus()
80	return False
81
82	class CalcGen(CalcThread):
83	"""
84	Computation
85	"""
86	def __init__(self,
87	id=-1,
88	input = None,
89	completefn = None,
90	updatefn = None,
91	#elapsed = 0,
92	yieldtime = 0.01,
93	worktime = 0.01):
94	"""
95	"""
96	CalcThread.__init__(self, completefn,
97	updatefn,
98	yieldtime,
99	worktime)
100	self.starttime = 0
101	self.id = id
102	self.input = input
103	self.update_fn = updatefn
104
105	def compute(self):
106	"""
107	excuting computation
108	"""
109	#elapsed = time.time() - self.starttime
110	self.starttime = time.time()
111	self.complete(input=self.input, update=self.update_fn)
112
113	class SasGenPanel(ScrolledPanel, PanelBase):
114	"""
115	Provides the sas gen calculator GUI.
116	"""
117	## Internal nickname for the window, used by the AUI manager
118	window_name = "Generic SAS Calculator"
119	## Name to appear on the window title bar
120	window_caption = "Generic SAS "
121	## Flag to tell the AUI manager to put this panel in the center pane
122	CENTER_PANE = True
123
124	def __init__(self, parent, args, *kwds):
125	ScrolledPanel.__init__(self, parent, style=wx.RAISED_BORDER,
126	args, *kwds)
127	#kwds['style'] = wx.SUNKEN_BORDER
128	PanelBase.__init__(self)
129	#Font size
130	self.SetWindowVariant(variant=FONT_VARIANT)
131	self.SetupScrolling()
132	#thread to read data
133	self.reader = None
134	self.ext = None
135	self.id = 'GenSAS'
136	self.file_name = ''
137	self.time_text = None
138	self.orient_combo = None
139	self.omfreader = sas_gen.OMFReader()
140	self.sldreader = sas_gen.SLDReader()
141	self.pdbreader = sas_gen.PDBReader()
142	self.model = sas_gen.GenSAS()
143	self.param_dic = self.model.params
144	self.parameters = []
145	self.data = None
146	self.scale2d = None
147	self.is_avg = False
148	self.plot_frame = None
149	self.qmax_x = _QMAX_DEFAULT
150	self.npts_x = _NPTS_DEFAULT
151	self.sld_data = None
152	self.graph_num = 1
153	self.default_shape = 'rectangular'
154	# Object that receive status event
155	self.parent = parent
156	self._do_layout()
157	self._create_default_sld_data()
158	self._create_default_2d_data()
159	wx.CallAfter(self._set_sld_data_helper)
160
161	def _define_structure(self):
162	"""
163	Define the main sizers building to build this application.
164	"""
165	self.main_sizer = wx.BoxSizer(wx.VERTICAL)
166	self.box_source = wx.StaticBox(self, -1, str("SLD Data File"))
167	self.box_parameters = wx.StaticBox(self, -1, str("Input Parameters"))
168	self.box_qrange = wx.StaticBox(self, -1, str("Q Range"))
169	self.boxsizer_source = wx.StaticBoxSizer(self.box_source,
170	wx.VERTICAL)
171	self.boxsizer_parameters = wx.StaticBoxSizer(self.box_parameters,
172	wx.VERTICAL)
173	self.boxsizer_qrange = wx.StaticBoxSizer(self.box_qrange,
174	wx.VERTICAL)
175	self.data_name_sizer = wx.BoxSizer(wx.HORIZONTAL)
176	self.param_sizer = wx.BoxSizer(wx.HORIZONTAL)
177	self.shape_sizer = wx.BoxSizer(wx.HORIZONTAL)
178	self.hint_sizer = wx.BoxSizer(wx.HORIZONTAL)
179	self.qrange_sizer = wx.BoxSizer(wx.HORIZONTAL)
180	self.button_sizer = wx.BoxSizer(wx.HORIZONTAL)
181
182	def _layout_data_name(self):
183	"""
184	Fill the sizer containing data's name
185	"""
186	data_name_txt = wx.StaticText(self, -1, 'Data: ')
187	self.data_name_tcl = OutputTextCtrl(self, -1,
188	size=(_BOX_WIDTH * 4, -1))
189	data_hint = "Loaded data"
190	self.data_name_tcl.SetToolTipString(data_hint)
191	#control that triggers importing data
192	id = wx.NewId()
193	self.browse_button = wx.Button(self, id, "Load")
194	hint_on_browse = "Click on this button to load a data in this panel."
195	#self.browse_button.SetToolTipString(hint_on_browse)
196	self.Bind(wx.EVT_BUTTON, self.on_load_data, id=id)
197	self.data_name_sizer.AddMany([(data_name_txt, 0, wx.LEFT, 15),
198	(self.data_name_tcl, 0, wx.LEFT, 10),
199	(self.browse_button, 0, wx.LEFT, 10)])
200	def _layout_param_size(self):
201	"""
202	Fill the sizer containing slit size information
203	"""
204	self.parameters = []
205	sizer = wx.GridBagSizer(3, 6)
206	model = self.model
207	details = self.model.details
208	params = self.model.params
209	ix = 0
210	iy = 0
211	param_title = wx.StaticText(self, -1, 'Parameter')
212	sizer.Add(param_title, (iy, ix), (1, 1), \
213	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
214	ix += 1
215	value_title = wx.StaticText(self, -1, 'Value')
216	sizer.Add(value_title, (iy, ix), (1, 1), \
217	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
218	ix += 1
219	unit_title = wx.StaticText(self, -1, 'Unit')
220	sizer.Add(unit_title, (iy, ix), (1, 1), \
221	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
222	key_list = params.keys()
223	key_list.sort()
224	for param in key_list:
225	iy += 1
226	ix = 0
227	p_name = wx.StaticText(self, -1, param)
228	sizer.Add(p_name, (iy, ix), (1, 1), \
229	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
230	## add parameter value
231	ix += 1
232	value = model.getParam(param)
233	ctl = InputTextCtrl(self, -1, size=(_BOX_WIDTH * 2, 20),
234	style=wx.TE_PROCESS_ENTER)
235	#ctl.SetToolTipString(\
236	# "Hit 'Enter' after typing to update the plot.")
237	ctl.SetValue(format_number(value, True))
238	sizer.Add(ctl, (iy, ix), (1, 1), wx.EXPAND)
239	## add unit
240	ix += 1
241	unit = wx.StaticText(self, -1, details[param][0])
242	sizer.Add(unit, (iy, ix), (1, 1), \
243	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
244	self.parameters.append([p_name, ctl, unit])
245
246	self.param_sizer.Add(sizer, 0, wx.LEFT, 10)
247
248	def _layout_hint(self):
249	"""
250	Fill the sizer containing hint
251	"""
252	hint_msg = "We support omf, sld or pdb data files only."
253	hint_msg += " "
254	if FONT_VARIANT < 1:
255	hint_msg += "Very "
256	hint_msg += "SLOW drawing -->"
257	hint_txt = wx.StaticText(self, -1, hint_msg)
258
259	id = wx.NewId()
260	self.draw_button = wx.Button(self, id, "Arrow Draw")
261	hint_on_draw = "Draw with arrows. Caution: it is a very slow drawing."
262	self.draw_button.SetToolTipString(hint_on_draw)
263	self.draw_button.Bind(wx.EVT_BUTTON, self.sld_draw, id=id)
264
265	self.draw_button.Enable(False)
266	self.hint_sizer.AddMany([(hint_txt, 0, wx.LEFT, 15),
267	(self.draw_button, 0, wx.LEFT, 7)])
268
269	def _layout_shape(self):
270	"""
271	Fill the shape sizer
272	"""
273	label_txt = wx.StaticText(self, -1, "Shape:")
274	self.shape_combo = self._fill_shape_combo()
275	self.shape_sizer.AddMany([(label_txt, 0, wx.LEFT, 15),
276	(self.shape_combo, 0, wx.LEFT, 5)])
277
278	def _fill_shape_combo(self):
279	"""
280	Fill up the shape combo box
281	"""
282	shape_combo = wx.ComboBox(self, -1, size=(150, -1),
283	style=wx.CB_READONLY)
284	shape_combo.Append('Rectangular')
285	shape_combo.Append('Ellipsoid')
286	shape_combo.Bind(wx.EVT_COMBOBOX, self._on_shape_select)
287	shape_combo.SetSelection(0)
288	return shape_combo
289
290	def _on_shape_select(self, event):
291	"""
292	On selecting a shape
293	"""
294	event.Skip()
295	label = event.GetEventObject().GetValue().lower()
296	self.default_shape = label
297	self.parent.set_omfpanel_default_shap(self.default_shape)
298	self.parent.set_omfpanel_npts()
299
300	def _fill_orient_combo(self):
301	"""
302	Fill up the orientation combo box: used only for atomic structure
303	"""
304	orient_combo = wx.ComboBox(self, -1, size=(150, -1),
305	style=wx.CB_READONLY)
306	orient_combo.Append('Fixed orientation')
307	orient_combo.Append('Debye full avg.')
308	#orient_combo.Append('Debye sph. sym.')
309
310	orient_combo.Bind(wx.EVT_COMBOBOX, self._on_orient_select)
311	orient_combo.SetSelection(0)
312	return orient_combo
313
314	def _on_orient_select(self, event):
315	"""
316	On selecting a orientation
317	"""
318	event.Skip()
319	cb = event.GetEventObject()
320	if cb.GetCurrentSelection() == 2:
321	self.is_avg = None
322	else:
323	is_avg = cb.GetCurrentSelection() == 1
324	self.is_avg = is_avg
325	self.model.set_is_avg(self.is_avg)
326	self.set_est_time()
327
328	def _layout_qrange(self):
329	"""
330	Fill the sizer containing qrange
331	"""
332	sizer = wx.GridBagSizer(2, 3)
333	ix = 0
334	iy = 0
335	#key_list.sort()
336	name = wx.StaticText(self, -1, 'No. of Qx (Qy) bins: ')
337	sizer.Add(name, (iy, ix), (1, 1), \
338	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
339	## add parameter value
340	ix += 1
341	self.npt_ctl = InputTextCtrl(self, -1, size=(_BOX_WIDTH * 1.5, 20),
342	style=wx.TE_PROCESS_ENTER)
343	self.npt_ctl.Bind(wx.EVT_TEXT, self._onparamEnter)
344	self.npt_ctl.SetValue(format_number(self.npts_x, True))
345	sizer.Add(self.npt_ctl, (iy, ix), (1, 1), wx.EXPAND)
346	## add unit
347	ix += 1
348	unit = wx.StaticText(self, -1, '')
349	sizer.Add(unit, (iy, ix), (1, 1), \
350	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
351	iy += 1
352	ix = 0
353	name = wx.StaticText(self, -1, 'Qx (Qy) Max: ')
354	sizer.Add(name, (iy, ix), (1, 1), \
355	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
356	## add parameter value
357	ix += 1
358	self.qmax_ctl = InputTextCtrl(self, -1, size=(_BOX_WIDTH * 1.5, 20),
359	style=wx.TE_PROCESS_ENTER)
360	self.qmax_ctl.Bind(wx.EVT_TEXT, self._onparamEnter )
361	self.qmax_ctl.SetValue(format_number(self.qmax_x, True))
362	sizer.Add(self.qmax_ctl, (iy, ix), (1, 1), wx.EXPAND)
363	## add unit
364	ix += 1
365	unit = wx.StaticText(self, -1, '[1/A]')
366	sizer.Add(unit, (iy, ix), (1, 1), \
367	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
368	self.qrange_sizer.Add(sizer, 0, wx.LEFT, 10)
369
370	def _layout_button(self):
371	"""
372	Do the layout for the button widgets
373	"""
374	self.est_time = '*Estimated Computation time :\n %s'
375	self.time_text = wx.StaticText(self, -1, self.est_time% str('2 sec') )
376	self.orient_combo = self._fill_orient_combo()
377	self.orient_combo.Show(False)
378	self.bt_compute = wx.Button(self, wx.NewId(),'Compute')
379	self.bt_compute.Bind(wx.EVT_BUTTON, self.on_compute)
380	self.bt_compute.SetToolTipString("Compute 2D Scattering Pattern.")
381	self.button_sizer.AddMany([(self.time_text , 0, wx.LEFT, 20),
382	(self.orient_combo , 0, wx.LEFT, 40),
383	(self.bt_compute, 0, wx.LEFT, 20)])
384
385	def estimate_ctime(self):
386	"""
387	Calculation time estimation
388	"""
389	# magic equation: not very accurate
390	factor = 1
391	n_qbins = float(self.npt_ctl.GetValue())
392	n_qbins *= n_qbins
393	n_pixs = float(self.parent.get_npix())
394	if self.is_avg:
395	factor = 6
396	n_pixs *= (n_pixs / 200)
397	x_in = n_qbins * n_pixs / 100000
398	etime = factor + 0.085973 * x_in
399	return int(etime)
400
401	def set_est_time(self):
402	"""
403	Set text for est. computation time
404	"""
405	unit = 'sec'
406	if self.time_text != None:
407	self.time_text.SetForegroundColour('black')
408	etime = self.estimate_ctime()
409	if etime > 60:
410	etime /= 60
411	unit = 'min'
412	self.time_text.SetForegroundColour('red')
413	time_str = str(etime) + ' ' + unit
414	self.time_text.SetLabel(self.est_time% time_str)
415
416	def _do_layout(self):
417	"""
418	Draw window content
419	"""
420	self._define_structure()
421	self._layout_data_name()
422	self._layout_param_size()
423	self._layout_qrange()
424	self._layout_hint()
425	self._layout_shape()
426	self._layout_button()
427	self.boxsizer_source.AddMany([(self.data_name_sizer, 0,
428	wx.EXPAND\|wx.TOP\|wx.BOTTOM, 5),
429	(self.hint_sizer, 0,
430	wx.EXPAND\|wx.TOP\|wx.BOTTOM, 5),
431	(self.shape_sizer, 0,
432	wx.EXPAND\|wx.TOP\|wx.BOTTOM, 5)])
433	self.boxsizer_parameters.AddMany([(self.param_sizer, 0,
434	wx.EXPAND\|wx.TOP\|wx.BOTTOM, 5),])
435	self.boxsizer_qrange.AddMany([(self.qrange_sizer, 0,
436	wx.EXPAND\|wx.TOP\|wx.BOTTOM, 5),])
437	self.main_sizer.AddMany([(self.boxsizer_source, 0,
438	wx.EXPAND\|wx.ALL, 10),
439	(self.boxsizer_parameters, 0,
440	wx.EXPAND\|wx.ALL, 10),
441	(self.boxsizer_qrange, 0,
442	wx.EXPAND\|wx.ALL, 10),
443	(self.button_sizer, 0,
444	wx.EXPAND\|wx.TOP\|wx.BOTTOM, 5)])
445	self.SetSizer(self.main_sizer)
446	self.SetAutoLayout(True)
447
448	def _create_default_sld_data(self):
449	"""
450	Making default sld-data
451	"""
452	sld_n_default = 6.97e-06
453	omfdata = sas_gen.OMFData()
454	omf2sld = sas_gen.OMF2SLD()
455	omf2sld.set_data(omfdata, self.default_shape)
456	self.sld_data = omf2sld.output
457	self.sld_data.is_data = False
458	self.sld_data.filename = "Default SLD Profile"
459	self.sld_data.set_sldn(sld_n_default)
460	self.data_name_tcl.SetValue(self.sld_data.filename)
461
462	def choose_data_file(self, location=None):
463	"""
464	Choosing a dtata file
465	"""
466	path = None
467	filename = ''
468	if location == None:
469	location = os.getcwd()
470
471	exts = "*" + self.omfreader.ext[0]
472	exts += ", *" + self.sldreader.ext[0]
473	exts += ", *" + self.pdbreader.ext[0]
474	all_type = "All GEN files (%s, %s) \| %s"% (exts.upper(), exts.lower(),
475	exts.lower().replace(',', ';'))
476	wildcard = [all_type]
477	omf_type = self.omfreader.type
478	sld_type = self.sldreader.type
479	pdb_type = self.pdbreader.type
480
481	for type in sld_type:
482	wildcard.append(type)
483	for type in omf_type:
484	wildcard.append(type)
485	for type in pdb_type:
486	wildcard.append(type)
487	wildcard = '\|'.join(wildcard)
488	dlg = wx.FileDialog(self, "Choose a file", location,
489	"", wildcard, wx.OPEN)
490	if dlg.ShowModal() == wx.ID_OK:
491	path = dlg.GetPath()
492	filename = os.path.basename(path)
493	dlg.Destroy()
494	return path
495
496	def on_load_data(self, event):
497	"""
498	Open a file dialog to allow the user to select a given file.
499	The user is only allow to load file with extension .omf, .txt, .sld.
500	Display the slit size corresponding to the loaded data.
501	"""
502	location = self.parent.get_path()
503	path = self.choose_data_file(location=location)
504	if path is None:
505	return
506
507	self.shape_sizer.ShowItems(False)
508	self.default_shape = 'rectangular'
509	self.parent.set_omfpanel_default_shap(self.default_shape)
510
511	self.parent.set_file_location(os.path.dirname(path))
512	try:
513	#Load data
514	self.ext = os.path.splitext(path)[-1]
515	if self.ext in self.omfreader.ext:
516	loader = self.omfreader
517	elif self.ext in self.sldreader.ext:
518	loader = self.sldreader
519	elif self.ext in self.pdbreader.ext:
520	loader = self.pdbreader
521	else:
522	loader = None
523	if self.reader is not None and self.reader.isrunning():
524	self.reader.stop()
525	self.browse_button.Enable(False)
526	self.browse_button.SetLabel("Loading...")
527	if self.parent.parent is not None:
528	wx.PostEvent(self.parent.parent,
529	StatusEvent(status="Loading...",
530	type="progress"))
531	self.reader = GenReader(path=path, loader=loader,
532	completefn=self.complete_loading,
533	updatefn=self.load_update)
534	self.reader.queue()
535	#self.load_update()
536	except:
537	self.ext = None
538	if self.parent.parent is None:
539	return
540	msg = "Generic SAS Calculator: %s" % (sys.exc_value)
541	wx.PostEvent(self.parent.parent,
542	StatusEvent(status=msg, type='stop'))
543	self.SetFocus()
544	return
545
546	def load_update(self):
547	"""
548	print update on the status bar
549	"""
550	if self.parent.parent is None:
551	return
552	if self.reader.isrunning():
553	type = "progress"
554	else:
555	type = "stop"
556	wx.PostEvent(self.parent.parent, StatusEvent(status="",
557	type=type))
558
559	def complete_loading(self, data=None, filename=''):
560	"""
561	Complete the loading
562	"""
563	#compute the slit size
564	self.browse_button.Enable(True)
565	self.browse_button.SetLabel('Load')
566	try:
567	is_pdbdata = False
568	filename = data.filename
569	self.data_name_tcl.SetValue(str(filename))
570	self.file_name = filename.split('.')[0]
571	self.orient_combo.SetSelection(0)
572	self.is_avg = False
573	if self.ext in self.omfreader.ext:
574	gen = sas_gen.OMF2SLD()
575	gen.set_data(data)
576	#omf_data = data
577	self.sld_data = gen.get_magsld()
578	elif self.ext in self.sldreader.ext:
579	self.sld_data = data
580	elif self.ext in self.pdbreader.ext:
581	self.sld_data = data
582	is_pdbdata = True
583	#omf_data = None
584	else:
585	raise
586	self.orient_combo.Show(is_pdbdata)
587	self.button_sizer.Layout()
588	self._set_sld_data_helper(True)
589	except:
590	if self.parent.parent is None:
591	raise
592	msg = "Loading Error: This file format is not supported "
593	msg += "for GenSAS."
594	wx.PostEvent(self.parent.parent,
595	StatusEvent(status=msg, type='stop', info='Error'))
596	self.SetFocus()
597	return
598	if self.parent.parent is None:
599	return
600
601	msg = "Load Complete"
602	wx.PostEvent(self.parent.parent, StatusEvent(status=msg, type='stop'))
603	self.SetFocus()
604
605	def _set_sld_data_helper(self, is_draw=False):
606	"""
607	Set sld data helper
608	"""
609	#is_avg = self.orient_combo.GetCurrentSelection() == 1
610	self.model.set_is_avg(self.is_avg)
611	self.model.set_sld_data(self.sld_data)
612
613	self.draw_button.Enable(self.sld_data!=None)
614	wx.CallAfter(self.parent.set_sld_data, self.sld_data)
615	self._update_model_params()
616	if is_draw:
617	wx.CallAfter(self.sld_draw, None, False)
618
619	def _update_model_params(self):
620	"""
621	Update the model parameter values
622	"""
623	for list in self.parameters:
624	param_name = list[0].GetLabelText()
625	val = str(self.model.params[param_name])
626	list[1].SetValue(val)
627
628	def set_volume_ctl_val(self, val):
629	"""
630	Set volume txtctrl value
631	"""
632	for list in self.parameters:
633	param_name = list[0].GetLabelText()
634	if param_name.lower() == 'total_volume':
635	list[1].SetValue(val)
636	list[1].Refresh()
637	break
638
639	def _onparamEnter(self, event):
640	"""
641	On param enter
642	"""
643	try:
644	item = event.GetEventObject()
645	self._check_value()
646	item.Refresh()
647	except:
648	pass
649
650	def sld_draw(self, event=None, has_arrow=True):
651	"""
652	Draw 3D sld profile
653	"""
654	flag = self.parent.check_omfpanel_inputs()
655	if not flag:
656	infor = 'Error'
657	msg = 'Error: Wrong inputs in the SLD info panel.'
658	# inform msg to wx
659	wx.PostEvent(self.parent.parent,
660	StatusEvent(status=msg, info=infor))
661	self.SetFocus()
662	return
663
664	self.sld_data = self.parent.get_sld_from_omf()
665	output = self.sld_data
666	#frame_size = wx.Size(470, 470)
667	self.plot_frame = PlotFrame(self, -1, 'testView')
668	frame = self.plot_frame
669	frame.Show(False)
670	add_icon(self.parent, frame)
671	panel = frame.plotpanel
672	try:
673	# mpl >= 1.0.0
674	ax = panel.figure.gca(projection='3d')
675	except:
676	# mpl < 1.0.0
677	try:
678	from mpl_toolkits.mplot3d import Axes3D
679	ax = Axes3D(panel.figure)
680	except:
681	logging.error("PlotPanel could not import Axes3D")
682	raise
683	panel.dimension = 3
684	graph_title = self._sld_plot_helper(ax, output, has_arrow)
685	# Use y, z axes (in mpl 3d) as z, y axes
686	# that consistent with our SAS detector coords.
687	ax.set_xlabel('x ($\A%s$)'% output.pos_unit)
688	ax.set_ylabel('z ($\A%s$)'% output.pos_unit)
689	ax.set_zlabel('y ($\A%s$)'% output.pos_unit)
690	panel.subplot.figure.subplots_adjust(left=0.05, right=0.95,
691	bottom=0.05, top=0.96)
692	if output.pix_type == 'atom':
693	ax.legend(loc='upper left', prop={'size':10})
694	num_graph = str(self.graph_num)
695	frame.SetTitle('Graph %s: %s'% (num_graph, graph_title))
696	wx.CallAfter(frame.Show, True)
697	self.graph_num += 1
698
699	def _sld_plot_helper(self, ax, output, has_arrow=False):
700	"""
701	Actual plot definition happens here
702	:Param ax: axis3d
703	:Param output: sld_data [MagSLD]
704	:Param has_arrow: whether or not draws M vector [bool]
705	"""
706	# Set the locals
707	color_dic = {'H':'blue', 'D':'purple', 'N': 'orange',
708	'O':'red', 'C':'green', 'P':'cyan', 'Other':'k'}
709	marker = ','
710	m_size = 2
711	graph_title = self.file_name
712	graph_title += " 3D SLD Profile "
713	pos_x = output.pos_x
714	pos_y = output.pos_y
715	pos_z = output.pos_z
716	sld_mx = output.sld_mx
717	sld_my = output.sld_my
718	sld_mz = output.sld_mz
719	pix_symbol = output.pix_symbol
720	if output.pix_type == 'atom':
721	marker = 'o'
722	m_size = 3.5
723	sld_tot = (numpy.fabs(sld_mx) + numpy.fabs(sld_my) + \
724	numpy.fabs(sld_mz) + numpy.fabs(output.sld_n))
725	is_nonzero = sld_tot > 0.0
726	is_zero = sld_tot == 0.0
727	# I. Plot null points
728	if is_zero.any():
729	ax.plot(pos_x[is_zero], pos_z[is_zero], pos_y[is_zero], marker,
730	c="y", alpha=0.5, markeredgecolor='y', markersize=m_size)
731	pos_x = pos_x[is_nonzero]
732	pos_y = pos_y[is_nonzero]
733	pos_z = pos_z[is_nonzero]
734	sld_mx = sld_mx[is_nonzero]
735	sld_my = sld_my[is_nonzero]
736	sld_mz = sld_mz[is_nonzero]
737	pix_symbol = output.pix_symbol[is_nonzero]
738	# II. Plot selective points in color
739	other_color = numpy.ones(len(pix_symbol), dtype='bool')
740	for key in color_dic.keys():
741	chosen_color = pix_symbol == key
742	if numpy.any(chosen_color):
743	other_color = other_color & (chosen_color != True)
744	color = color_dic[key]
745	ax.plot(pos_x[chosen_color], pos_z[chosen_color],
746	pos_y[chosen_color], marker, c=color, alpha=0.5,
747	markeredgecolor=color, markersize=m_size, label=key)
748	# III. Plot All others
749	if numpy.any(other_color):
750	a_name = ''
751	if output.pix_type == 'atom':
752	# Get atom names not in the list
753	a_names = [symb for symb in pix_symbol \
754	if symb not in color_dic.keys()]
755	a_name = a_names[0]
756	for name in a_names:
757	new_name = ", " + name
758	if a_name.count(name) == 0:
759	a_name += new_name
760	# plot in black
761	ax.plot(pos_x[other_color], pos_z[other_color], pos_y[other_color],
762	marker, c="k", alpha=0.5, markeredgecolor="k",
763	markersize=m_size, label=a_name)
764	# IV. Draws atomic bond with grey lines if any
765	if output.has_conect:
766	for ind in range(len(output.line_x)):
767	ax.plot(output.line_x[ind], output.line_z[ind],
768	output.line_y[ind], '-', lw=0.6, c="grey", alpha=0.3)
769	# V. Draws magnetic vectors
770	if has_arrow and len(pos_x) > 0:
771	graph_title += " - Magnetic Vector as Arrow -"
772	panel = self.plot_frame.plotpanel
773	def _draw_arrow(input=None, update=None):
774	"""
775	draw magnetic vectors w/arrow
776	"""
777	max_mx = max(numpy.fabs(sld_mx))
778	max_my = max(numpy.fabs(sld_my))
779	max_mz = max(numpy.fabs(sld_mz))
780	max_m = max(max_mx, max_my, max_mz)
781	try:
782	max_step = max(output.xstepsize, output.ystepsize,
783	output.zstepsize)
784	except:
785	max_step = 0
786	if max_step <= 0:
787	max_step = 5
788	try:
789	if max_m != 0:
790	unit_x2 = sld_mx / max_m
791	unit_y2 = sld_my / max_m
792	unit_z2 = sld_mz / max_m
793	# 0.8 is for avoiding the color becomes white=(1,1,1))
794	color_x = numpy.fabs(unit_x2 * 0.8)
795	color_y = numpy.fabs(unit_y2 * 0.8)
796	color_z = numpy.fabs(unit_z2 * 0.8)
797	x2 = pos_x + unit_x2 * max_step
798	y2 = pos_y + unit_y2 * max_step
799	z2 = pos_z + unit_z2 * max_step
800	x_arrow = numpy.column_stack((pos_x, x2))
801	y_arrow = numpy.column_stack((pos_y, y2))
802	z_arrow = numpy.column_stack((pos_z, z2))
803	colors = numpy.column_stack((color_x, color_y, color_z))
804	arrows = Arrow3D(panel, x_arrow, z_arrow, y_arrow,
805	colors, mutation_scale=10, lw=1,
806	arrowstyle="->", alpha = 0.5)
807	ax.add_artist(arrows)
808	except:
809	pass
810	msg = "Arrow Drawing completed.\n"
811	status_type = 'stop'
812	self._status_info(msg, status_type)
813	msg = "Arrow Drawing is in progress..."
814	status_type = 'progress'
815	self._status_info(msg, status_type)
816	draw_out = CalcGen(input=ax,
817	completefn=_draw_arrow, updatefn=self._update)
818	draw_out.queue()
819	return graph_title
820
821	def set_input_params(self):
822	"""
823	Set model parameters
824	"""
825	for list in self.parameters:
826	param_name = list[0].GetLabelText()
827	param_value = float(list[1].GetValue())
828	self.model.setParam(param_name, param_value)
829
830	def on_compute(self, event):
831	"""
832	Compute I(qx, qy)
833	"""
834	flag = self.parent.check_omfpanel_inputs()
835	if not flag and self.parent.parent != None:
836	infor = 'Error'
837	msg = 'Error: Wrong inputs in the SLD info panel.'
838	# inform msg to wx
839	wx.PostEvent(self.parent.parent,
840	StatusEvent(status=msg, info=infor))
841	self.SetFocus()
842	return
843	self.sld_data = self.parent.get_sld_from_omf()
844	if self.sld_data == None:
845	if self.parent.parent != None:
846	infor = 'Error'
847	msg = 'Error: No data has been selected.'
848	# inform msg to wx
849	wx.PostEvent(self.parent.parent,
850	StatusEvent(status=msg, info=infor))
851	self.SetFocus()
852	return
853	flag = self._check_value()
854	if not flag:
855	_set_error(self, None, True)
856	return
857	try:
858	self.model.set_sld_data(self.sld_data)
859	self.set_input_params()
860	if self.is_avg or self.is_avg == None:
861	self._create_default_1d_data()
862	i_out = numpy.zeros(len(self.data.y))
863	inputs = [self.data.x, [], i_out]
864	else:
865	self._create_default_2d_data()
866	i_out = numpy.zeros(len(self.data.data))
867	inputs=[self.data.qx_data, self.data.qy_data, i_out]
868
869	msg = "Computation is in progress..."
870	status_type = 'progress'
871	self._status_info(msg, status_type)
872	cal_out = CalcGen(input=inputs,
873	completefn=self.complete,
874	updatefn=self._update)
875	cal_out.queue()
876
877	except:
878	msg = "%s."% sys.exc_value
879	status_type = 'stop'
880	self._status_info(msg, status_type)
881	wx.PostEvent(self.parent.parent,
882	StatusEvent(status=msg, info='Error'))
883	self.SetFocus()
884
885	def _check_value(self):
886	"""
887	Check input values if float
888	"""
889	flag = True
890	self.npt_ctl.SetBackgroundColour("white")
891	self.qmax_ctl.SetBackgroundColour("white")
892	try:
893	npt_val = float(self.npt_ctl.GetValue())
894	if npt_val < 2 or npt_val > 1000:
895	raise
896	self.npt_ctl.SetValue(str(int(npt_val)))
897	self.set_est_time()
898	except:
899	flag = _set_error(self, self.npt_ctl)
900	try:
901	qmax_val = float(self.qmax_ctl.GetValue())
902	if qmax_val <= 0 or qmax_val > 1000:
903	raise
904	except:
905	flag = _set_error(self, self.qmax_ctl)
906	for list in self.parameters:
907	list[1].SetBackgroundColour("white")
908	param_name = list[0].GetLabelText()
909	try:
910	param_val = float(list[1].GetValue())
911	if param_name.count('frac') > 0:
912	if param_val < 0 or param_val > 1:
913	raise
914	except:
915	flag = _set_error(self, list[1])
916	return flag
917
918	def _status_info(self, msg = '', type = "update"):
919	"""
920	Status msg
921	"""
922	if type == "stop":
923	label = "Compute"
924	able = True
925	else:
926	label = "Wait..."
927	able = False
928	self.bt_compute.Enable(able)
929	self.bt_compute.SetLabel(label)
930	self.bt_compute.SetToolTipString(label)
931	if self.parent.parent != None:
932	wx.PostEvent(self.parent.parent,
933	StatusEvent(status = msg, type = type ))
934
935	def _update(self, time=None):
936	"""
937	Update the progress bar
938	"""
939	if self.parent.parent == None:
940	return
941	type = "progress"
942	msg = "Please wait. Computing... (Note: Window may look frozen.)"
943	wx.PostEvent(self.parent.parent, StatusEvent(status=msg,
944	type=type))
945
946	def complete(self, input, update=None):
947	"""
948	Gen compute complete function
949	:Param input: input list [qx_data, qy_data, i_out]
950	"""
951	out = numpy.empty(0)
952	#s = time.time()
953	for ind in range(len(input[0])):
954	if self.is_avg:
955	if ind % 1 == 0 and update != None:
956	update()
957	time.sleep(0.1)
958	inputi = [input[0][ind:ind+1], [], input[2][ind:ind+1]]
959	outi = self.model.run(inputi)
960	out = numpy.append(out, outi)
961	else:
962	if ind % 50 == 0 and update != None:
963	update()
964	time.sleep(0.001)
965	inputi = [input[0][ind:ind+1], input[1][ind:ind+1],
966	input[2][ind:ind+1]]
967	outi = self.model.runXY(inputi)
968	out = numpy.append(out, outi)
969	#print time.time() - s
970	if self.is_avg or self.is_avg == None:
971	self._draw1D(out)
972	else:
973	#out = self.model.runXY(input)
974	self._draw2D(out)
975
976	msg = "Gen computation completed.\n"
977	status_type = 'stop'
978	self._status_info(msg, status_type)
979
980	def _create_default_2d_data(self):
981	"""
982	Create 2D data by default
983	Only when the page is on theory mode.
984	:warning: This data is never plotted.
985	"""
986	self.qmax_x = float(self.qmax_ctl.GetValue())
987	self.npts_x = int(float(self.npt_ctl.GetValue()))
988	self.data = Data2D()
989	qmax = self.qmax_x #/ numpy.sqrt(2)
990	self.data.xaxis('\\rm{Q_{x}}', '\AA^{-1}')
991	self.data.yaxis('\\rm{Q_{y}}', '\AA^{-1}')
992	self.data.is_data = False
993	self.data.id = str(self.uid) + " GenData"
994	self.data.group_id = str(self.uid) + " Model2D"
995	## Default values
996	self.data.detector.append(Detector())
997	index = len(self.data.detector) - 1
998	self.data.detector[index].distance = 8000 # mm
999	self.data.source.wavelength = 6 # A
1000	self.data.detector[index].pixel_size.x = 5 # mm
1001	self.data.detector[index].pixel_size.y = 5 # mm
1002	self.data.detector[index].beam_center.x = qmax
1003	self.data.detector[index].beam_center.y = qmax
1004	xmax = qmax
1005	xmin = -qmax
1006	ymax = qmax
1007	ymin = -qmax
1008	qstep = self.npts_x
1009
1010	x = numpy.linspace(start=xmin, stop=xmax, num=qstep, endpoint=True)
1011	y = numpy.linspace(start=ymin, stop=ymax, num=qstep, endpoint=True)
1012	## use data info instead
1013	new_x = numpy.tile(x, (len(y), 1))
1014	new_y = numpy.tile(y, (len(x), 1))
1015	new_y = new_y.swapaxes(0, 1)
1016	# all data reuire now in 1d array
1017	qx_data = new_x.flatten()
1018	qy_data = new_y.flatten()
1019	q_data = numpy.sqrt(qx_data * qx_data + qy_data * qy_data)
1020	# set all True (standing for unmasked) as default
1021	mask = numpy.ones(len(qx_data), dtype=bool)
1022	# store x and y bin centers in q space
1023	x_bins = x
1024	y_bins = y
1025	self.data.source = Source()
1026	self.data.data = numpy.ones(len(mask))
1027	self.data.err_data = numpy.ones(len(mask))
1028	self.data.qx_data = qx_data
1029	self.data.qy_data = qy_data
1030	self.data.q_data = q_data
1031	self.data.mask = mask
1032	self.data.x_bins = x_bins
1033	self.data.y_bins = y_bins
1034	# max and min taking account of the bin sizes
1035	self.data.xmin = xmin
1036	self.data.xmax = xmax
1037	self.data.ymin = ymin
1038	self.data.ymax = ymax
1039
1040	def _create_default_1d_data(self):
1041	"""
1042	Create 2D data by default
1043	Only when the page is on theory mode.
1044	:warning: This data is never plotted.
1045	residuals.x = data_copy.x[index]
1046	residuals.dy = numpy.ones(len(residuals.y))
1047	residuals.dx = None
1048	residuals.dxl = None
1049	residuals.dxw = None
1050	"""
1051	self.qmax_x = float(self.qmax_ctl.GetValue())
1052	self.npts_x = int(float(self.npt_ctl.GetValue()))
1053	qmax = self.qmax_x #/ numpy.sqrt(2)
1054	## Default values
1055	xmax = qmax
1056	xmin = qmax * _Q1D_MIN
1057	qstep = self.npts_x
1058	x = numpy.linspace(start=xmin, stop=xmax, num=qstep, endpoint=True)
1059	# store x and y bin centers in q space
1060	#self.data.source = Source()
1061	y = numpy.ones(len(x))
1062	dy = numpy.zeros(len(x))
1063	dx = numpy.zeros(len(x))
1064	self.data = Data1D(x=x, y=y)
1065	self.data.dx = dx
1066	self.data.dy = dy
1067
1068	def _draw1D(self, y_out):
1069	"""
1070	Complete get the result of modelthread and create model 2D
1071	that can be plot.
1072	"""
1073	page_id = self.id
1074	data = self.data
1075
1076	model = self.model
1077	state = None
1078
1079	new_plot = Data1D(x=data.x, y=y_out)
1080	new_plot.dx = data.dx
1081	new_plot.dy = data.dy
1082	new_plot.xaxis('\\rm{Q_{x}}', '\AA^{-1}')
1083	new_plot.yaxis('\\rm{Intensity}', 'cm^{-1}')
1084	new_plot.is_data = False
1085	new_plot.id = str(self.uid) + " GenData1D"
1086	new_plot.group_id = str(self.uid) + " Model1D"
1087	new_plot.name = model.name + '1d'
1088	new_plot.title = "Generic model1D "
1089	new_plot.id = str(page_id) + ': ' + self.file_name \
1090	+ ' #%s'% str(self.graph_num) + "_1D"
1091	new_plot.group_id = str(page_id) + " Model1D" +\
1092	' #%s'% str(self.graph_num) + "_1D"
1093	new_plot.is_data = False
1094
1095	title = new_plot.title
1096	_yaxis, _yunit = new_plot.get_yaxis()
1097	_xaxis, _xunit = new_plot.get_xaxis()
1098	new_plot.xaxis(str(_xaxis), str(_xunit))
1099	new_plot.yaxis(str(_yaxis), str(_yunit))
1100
1101	if new_plot.is_data:
1102	data_name = str(new_plot.name)
1103	else:
1104	data_name = str(model.__class__.__name__) + '1d'
1105
1106	if len(title) > 1:
1107	new_plot.title = "Gen Theory for %s "% model.name + data_name
1108	new_plot.name = new_plot.id
1109	new_plot.label = new_plot.id
1110	#theory_data = deepcopy(new_plot)
1111	if self.parent.parent != None:
1112	self.parent.parent.update_theory(data_id=new_plot.id,
1113	theory=new_plot,
1114	state=state)
1115	title = new_plot.title
1116	num_graph = str(self.graph_num)
1117	wx.CallAfter(self.parent.draw_graph, new_plot,
1118	title="GEN Graph %s: "% num_graph + new_plot.id )
1119	self.graph_num += 1
1120
1121	def _draw2D(self, image):
1122	"""
1123	Complete get the result of modelthread and create model 2D
1124	that can be plot.
1125	"""
1126	page_id = self.id
1127	data = self.data
1128
1129	model = self.model
1130	qmin = 0.0
1131	state = None
1132
1133	numpy.nan_to_num(image)
1134	new_plot = Data2D(image=image, err_image=data.err_data)
1135	new_plot.name = model.name + '2d'
1136	new_plot.title = "Generic model 2D "
1137	new_plot.id = str(page_id) + ': ' + self.file_name \
1138	+ ' #%s'% str(self.graph_num) + "_2D"
1139	new_plot.group_id = str(page_id) + " Model2D" \
1140	+ ' #%s'% str(self.graph_num) + "_2D"
1141	new_plot.detector = data.detector
1142	new_plot.source = data.source
1143	new_plot.is_data = False
1144	new_plot.qx_data = data.qx_data
1145	new_plot.qy_data = data.qy_data
1146	new_plot.q_data = data.q_data
1147	new_plot.mask = data.mask
1148	## plot boundaries
1149	new_plot.ymin = data.ymin
1150	new_plot.ymax = data.ymax
1151	new_plot.xmin = data.xmin
1152	new_plot.xmax = data.xmax
1153	title = data.title
1154	_yaxis, _yunit = data.get_yaxis()
1155	_xaxis, _xunit = data.get_xaxis()
1156	new_plot.xaxis(str(_xaxis), str(_xunit))
1157	new_plot.yaxis(str(_yaxis), str(_yunit))
1158
1159	new_plot.is_data = False
1160	if data.is_data:
1161	data_name = str(data.name)
1162	else:
1163	data_name = str(model.__class__.__name__) + '2d'
1164
1165	if len(title) > 1:
1166	new_plot.title = "Gen Theory for %s "% model.name + data_name
1167	new_plot.name = new_plot.id
1168	new_plot.label = new_plot.id
1169	#theory_data = deepcopy(new_plot)
1170	if self.parent.parent != None:
1171	self.parent.parent.update_theory(data_id=data.id,
1172	theory=new_plot,
1173	state=state)
1174	title = new_plot.title
1175	num_graph = str(self.graph_num)
1176	wx.CallAfter(self.parent.draw_graph, new_plot,
1177	title="GEN Graph %s: "% num_graph + new_plot.id )
1178	self.graph_num += 1
1179
1180	def set_scale2d(self, scale):
1181	"""
1182	Set SLD plot scale
1183	"""
1184	self.scale2d = None
1185
1186	def on_panel_close(self, event):
1187	"""
1188	On Close SLD panel
1189	"""
1190	#Not implemented
1191
1192	class OmfPanel(ScrolledPanel, PanelBase):
1193	"""
1194	Provides the sas gen calculator GUI.
1195	"""
1196	## Internal nickname for the window, used by the AUI manager
1197	window_name = "SLD Pixel Info"
1198	## Name to appear on the window title bar
1199	window_caption = "SLD Pixel Info "
1200	## Flag to tell the AUI manager to put this panel in the center pane
1201	CENTER_PANE = False
1202
1203	def __init__(self, parent, args, *kwds):
1204	ScrolledPanel.__init__(self, parent, style=wx.RAISED_BORDER,
1205	args, *kwds)
1206	PanelBase.__init__(self)
1207	#Font size
1208	self.SetWindowVariant(variant=FONT_VARIANT)
1209	self.SetupScrolling()
1210	# Object that receive status event
1211	self.parent = parent
1212	self.sld_data = sas_gen.MagSLD([0], [0], [0])
1213	self.sld_ctl = None
1214	self.default_shape = 'rectangular'
1215	self._do_layout()
1216
1217	def set_slddata(self, slddata):
1218	"""
1219	Set sld data related items
1220	"""
1221	self.sld_data = slddata
1222	self._set_slddata_ctr_val(slddata)
1223	# Make sure that self._set_slddata_ctr_val() is finished
1224	wx.CallAfter(self._set_omfdata_ctr, slddata)
1225
1226	def get_sld_val(self):
1227	"""
1228	Set sld_n of slddata on sld input
1229	"""
1230	sld_sets = {}
1231	if not self.sld_data.is_data:
1232	self._get_other_val()
1233	for list in self.slds:
1234	if list[1].IsEnabled():
1235	list[1].SetBackgroundColour("white")
1236	list[1].Refresh()
1237	try:
1238	val = float(list[1].GetValue())
1239	sld_sets[list[0]] = val
1240	except:
1241	flag = _set_error(self, list[1])
1242	if not flag:
1243	return self.sld_data
1244	else:
1245	sld_sets[list[0]] = None
1246	for key in sld_sets.keys():
1247	key_low = key.lower()
1248	if key_low.count('mx') > 0:
1249	if sld_sets[key] == None:
1250	sld_sets[key] = self.sld_data.sld_mx
1251	mx = sld_sets[key]
1252	elif key_low.count('my') > 0:
1253	if sld_sets[key] == None:
1254	sld_sets[key] = self.sld_data.sld_my
1255	my = sld_sets[key]
1256	elif key_low.count('mz') > 0:
1257	if sld_sets[key] == None:
1258	sld_sets[key] = self.sld_data.sld_mz
1259	mz = sld_sets[key]
1260	else:
1261	if sld_sets[key] != None:
1262	self.sld_data.set_sldn(sld_sets[key])
1263	self.sld_data.set_sldms(mx, my, mz)
1264	self._set_slddata_ctr_val(self.sld_data)
1265
1266	return self.sld_data
1267
1268	def get_pix_volumes(self):
1269	"""
1270	Get the pixel volume
1271	"""
1272	vol = self.sld_data.vol_pix
1273
1274	return vol
1275
1276	def _get_other_val(self):
1277	"""
1278	"""
1279	omfdata = sas_gen.OMFData()
1280	sets = {}
1281	try:
1282	for lst in self.stepsize:
1283	if lst[1].IsEnabled():
1284	val = float(lst[1].GetValue())
1285	sets[lst[0]] = val
1286	else:
1287	sets[lst[0]] = None
1288	return
1289	for lst in self.nodes:
1290	if lst[1].IsEnabled():
1291	val = float(lst[1].GetValue())
1292	sets[lst[0]] = val
1293	else:
1294	sets[lst[0]] = None
1295	return
1296
1297	for key in sets.keys():
1298	exec "omfdata.%s = sets['%s']"% (key, key)
1299
1300	omf2sld = sas_gen.OMF2SLD()
1301	omf2sld.set_data(omfdata, self.default_shape)
1302	self.sld_data = omf2sld.output
1303	self.sld_data.is_data = False
1304	self.sld_data.filename = "Default SLD Profile"
1305	except:
1306	msg = "OMF Panel: %s"% sys.exc_value
1307	infor = 'Error'
1308	#logging.error(msg)
1309	if self.parent.parent != None:
1310	# inform msg to wx
1311	wx.PostEvent(self.parent.parent,
1312	StatusEvent(status=msg, info=infor))
1313	self.SetFocus()
1314
1315	def _set_slddata_ctr_val(self, slddata):
1316	"""
1317	Set slddata crl
1318	"""
1319	try:
1320	val = str(len(slddata.sld_n))
1321	except:
1322	val = 'Unknown'
1323	self.npix_ctl.SetValue(val)
1324
1325	def _set_omfdata_ctr(self, omfdata):
1326	"""
1327	Set the textctr box values
1328	"""
1329
1330	if omfdata == None:
1331	self._set_none_text()
1332	return
1333	nodes_list = self._get_nodes_key_list(omfdata)
1334	step_list = self._get_step_key_list(omfdata)
1335	for ctr_list in self.nodes:
1336	for key in nodes_list.keys():
1337	if ctr_list[0] == key:
1338	ctr_list[1].SetValue(format_number(nodes_list[key], True))
1339	ctr_list[1].Enable(not omfdata.is_data)
1340	break
1341	for ctr_list in self.stepsize:
1342	for key in step_list.keys():
1343	if ctr_list[0] == key:
1344	ctr_list[1].SetValue(format_number(step_list[key], True))
1345	ctr_list[1].Enable(not omfdata.is_data)
1346	break
1347
1348	def _set_none_text(self):
1349	"""
1350	Set Unknown in textctrls
1351	"""
1352	val = 'Unknown'
1353	for ctr_list in self.nodes:
1354	ctr_list[1].SetValue(val)
1355	for ctr_list in self.stepsize:
1356	ctr_list[1].SetValue(val)
1357
1358	def _define_structure(self):
1359	"""
1360	Define the main sizers building to build this application.
1361	"""
1362	self.main_sizer = wx.BoxSizer(wx.VERTICAL)
1363
1364	self.npixels_sizer = wx.BoxSizer(wx.HORIZONTAL)
1365	self.box_sld = wx.StaticBox(self, -1,
1366	str("Mean SLD"))
1367	self.box_node = wx.StaticBox(self, -1, str("Nodes"))
1368	self.boxsizer_sld = wx.StaticBoxSizer(self.box_sld, wx.VERTICAL)
1369	self.box_stepsize = wx.StaticBox(self, -1, str("Step Size"))
1370	self.boxsizer_node = wx.StaticBoxSizer(self.box_node, wx.VERTICAL)
1371	self.boxsizer_stepsize = wx.StaticBoxSizer(self.box_stepsize,
1372	wx.VERTICAL)
1373	self.sld_sizer = wx.BoxSizer(wx.HORIZONTAL)
1374	self.node_sizer = wx.BoxSizer(wx.HORIZONTAL)
1375	self.step_sizer = wx.BoxSizer(wx.HORIZONTAL)
1376	self.hint_sizer = wx.BoxSizer(wx.HORIZONTAL)
1377	self.button_sizer = wx.BoxSizer(wx.HORIZONTAL)
1378
1379	def _layout_npix(self):
1380	"""
1381	Build No of pixels sizer
1382	"""
1383	num_pix_text = wx.StaticText(self, -1, "No. of Pixels: ")
1384	self.npix_ctl = OutputTextCtrl(self, -1, size=(_BOX_WIDTH, 20),
1385	style=wx.TE_PROCESS_ENTER)
1386	self._set_slddata_ctr_val(self.sld_data)
1387	self._set_omfdata_ctr(self.sld_data)
1388	self.npixels_sizer.AddMany([(num_pix_text, 0,
1389	wx.EXPAND\|wx.LEFT\|wx.TOP, 5),
1390	(self.npix_ctl, 0,
1391	wx.EXPAND\|wx.TOP, 5)])
1392
1393	def _layout_slds(self):
1394	"""
1395	Build nuclear sld sizer
1396	"""
1397	self.slds = []
1398	omfdata = self.sld_data
1399	if omfdata == None:
1400	raise
1401	sld_key_list = self._get_slds_key_list(omfdata)
1402	# Dic is not sorted
1403	key_list = [key for key in sld_key_list.keys()]
1404	# Sort here
1405	key_list.sort()
1406	is_data = self.sld_data.is_data
1407	sizer = wx.GridBagSizer(2, 3)
1408	ix = 0
1409	iy = -1
1410	for key in key_list:
1411	value = sld_key_list[key]
1412	iy += 1
1413	ix = 0
1414	name = wx.StaticText(self, -1, key)
1415	sizer.Add(name, (iy, ix), (1, 1), \
1416	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
1417	## add parameter value
1418	ix += 1
1419	ctl = InputTextCtrl(self, -1, size=(_BOX_WIDTH, 20),
1420	style=wx.TE_PROCESS_ENTER)
1421	ctl.SetValue(format_number(value, True))
1422	ctl.Enable(not is_data)
1423	sizer.Add(ctl, (iy, ix), (1, 1), wx.EXPAND)
1424	## add unit
1425	ix += 1
1426	s_unit = '[' + omfdata.sld_unit + ']'
1427	unit = wx.StaticText(self, -1, s_unit)
1428	sizer.Add(unit, (iy, ix), (1, 1), \
1429	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
1430	self.slds.append([key, ctl, unit])
1431	self.sld_sizer.Add(sizer, 0, wx.LEFT, 10)
1432
1433	def _layout_nodes(self):
1434	"""
1435	Fill the sizer containing data's name
1436	"""
1437	self.nodes = []
1438	omfdata = self.sld_data
1439	if omfdata == None:
1440	raise
1441	key_list = self._get_nodes_key_list(omfdata)
1442	is_data = self.sld_data.is_data
1443	sizer = wx.GridBagSizer(2, 3)
1444	ix = 0
1445	iy = -1
1446	for key, value in key_list.iteritems():
1447	iy += 1
1448	ix = 0
1449	name = wx.StaticText(self, -1, key)
1450	sizer.Add(name, (iy, ix), (1, 1), \
1451	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
1452	## add parameter value
1453	ix += 1
1454	ctl = InputTextCtrl(self, -1, size=(_BOX_WIDTH, 20),
1455	style=wx.TE_PROCESS_ENTER)
1456	ctl.Bind(wx.EVT_TEXT, self._onparamEnter )
1457	ctl.SetValue(format_number(value, True))
1458	ctl.Enable(not is_data)
1459	sizer.Add(ctl, (iy, ix), (1, 1), wx.EXPAND)
1460	## add unit
1461	ix += 1
1462	unit = wx.StaticText(self, -1, '')
1463	sizer.Add(unit, (iy, ix), (1, 1), \
1464	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
1465	self.nodes.append([key, ctl, unit])
1466	self.node_sizer.Add(sizer, 0, wx.LEFT, 10)
1467
1468	def _layout_stepsize(self):
1469	"""
1470	Fill the sizer containing slit size information
1471	"""
1472	self.stepsize = []
1473	omfdata = self.sld_data
1474	if omfdata == None:
1475	raise
1476	key_list = self._get_step_key_list(omfdata)
1477	is_data = self.sld_data.is_data
1478	sizer = wx.GridBagSizer(2, 3)
1479	ix = 0
1480	iy = -1
1481	#key_list.sort()
1482	for key, value in key_list.iteritems():
1483	iy += 1
1484	ix = 0
1485	name = wx.StaticText(self, -1, key)
1486	sizer.Add(name, (iy, ix), (1, 1), \
1487	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
1488	## add parameter value
1489	ix += 1
1490	ctl = InputTextCtrl(self, -1, size=(_BOX_WIDTH, 20),
1491	style=wx.TE_PROCESS_ENTER)
1492	ctl.Bind(wx.EVT_TEXT, self._onstepsize )
1493	ctl.SetValue(format_number(value, True))
1494	ctl.Enable(not is_data)
1495	sizer.Add(ctl, (iy, ix), (1, 1), wx.EXPAND)
1496	## add unit
1497	ix += 1
1498	p_unit = '[' + omfdata.pos_unit + ']'
1499	unit = wx.StaticText(self, -1, p_unit)
1500	sizer.Add(unit, (iy, ix), (1, 1), \
1501	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
1502	self.stepsize.append([key, ctl, unit])
1503	self.step_sizer.Add(sizer, 0, wx.LEFT, 10)
1504
1505	def _layout_hint(self):
1506	"""
1507	Fill the sizer containing hint
1508	"""
1509	hint_msg = "Load an omf or 3d sld profile data file."
1510	self.hint_txt = wx.StaticText(self, -1, hint_msg)
1511	self.hint_sizer.AddMany([(self.hint_txt, 0, wx.LEFT, 15)])
1512
1513	def _layout_button(self):
1514	"""
1515	Do the layout for the button widgets
1516	"""
1517	self.bt_draw = wx.Button(self, wx.NewId(),'Draw Points')
1518	self.bt_draw.Bind(wx.EVT_BUTTON, self.on_sld_draw)
1519	self.bt_draw.SetToolTipString("Draw a scatter plot for sld profile.")
1520	self.bt_save = wx.Button(self, wx.NewId(),'Save SLD Data')
1521	self.bt_save.Bind(wx.EVT_BUTTON, self.on_save)
1522	self.bt_save.Enable(False)
1523	self.bt_save.SetToolTipString("Save SLD data.")
1524	self.button_sizer.AddMany([(self.bt_draw, 0, wx.LEFT, 10),
1525	(self.bt_save, 0, wx.LEFT, 10)])
1526
1527	def _do_layout(self):
1528	"""
1529	Draw window content
1530	"""
1531	self._define_structure()
1532	self._layout_nodes()
1533	self._layout_stepsize()
1534	self._layout_npix()
1535	self._layout_slds()
1536	#self._layout_hint()
1537	self._layout_button()
1538	self.boxsizer_node.AddMany([(self.node_sizer, 0,
1539	wx.EXPAND\|wx.TOP, 5),
1540	(self.hint_sizer, 0,
1541	wx.EXPAND\|wx.TOP\|wx.BOTTOM, 5)])
1542	self.boxsizer_stepsize.AddMany([(self.step_sizer, 0,
1543	wx.EXPAND\|wx.TOP\|wx.BOTTOM, 5),])
1544	self.boxsizer_sld.AddMany([(self.sld_sizer, 0,
1545	wx.EXPAND\|wx.BOTTOM, 5),])
1546	self.main_sizer.AddMany([(self.npixels_sizer, 0, wx.EXPAND\|wx.ALL, 10),
1547	(self.boxsizer_sld, 0, wx.EXPAND\|wx.ALL, 10),
1548	(self.boxsizer_node, 0, wx.EXPAND\|wx.ALL, 10),
1549	(self.boxsizer_stepsize, 0, wx.EXPAND\|wx.ALL, 10),
1550	(self.button_sizer, 0, wx.EXPAND\|wx.TOP\|wx.BOTTOM, 5)])
1551	self.SetSizer(self.main_sizer)
1552	self.SetAutoLayout(True)
1553
1554	def _get_nodes_key_list(self, data):
1555	"""
1556	Return nodes key list
1557
1558	:Param data: OMFData
1559	"""
1560	key_list = {'xnodes' : data.xnodes,
1561	'ynodes' : data.ynodes,
1562	'znodes' : data.znodes}
1563	return key_list
1564
1565	def _get_slds_key_list(self, data):
1566	"""
1567	Return nodes key list
1568
1569	:Param data: OMFData
1570	"""
1571	key_list = {'Nucl.' : data.sld_n,
1572	'Mx' : data.sld_mx,
1573	'My' : data.sld_my,
1574	'Mz' : data.sld_mz}
1575	return key_list
1576
1577	def _get_step_key_list(self, data):
1578	"""
1579	Return step key list
1580
1581	:Param data: OMFData
1582	"""
1583	key_list = {'xstepsize' : data.xstepsize,
1584	'ystepsize' : data.ystepsize,
1585	'zstepsize' : data.zstepsize}
1586	return key_list
1587
1588	def set_sld_ctr(self, sld_data):
1589	"""
1590	Set sld textctrls
1591	"""
1592	if sld_data == None:
1593	for ctr_list in self.slds:
1594	ctr_list[1].Enable(False)
1595	#break
1596	return
1597
1598	self.sld_data = sld_data
1599	sld_list = self._get_slds_key_list(sld_data)
1600	for ctr_list in self.slds:
1601	for key in sld_list.keys():
1602	if ctr_list[0] == key:
1603	min_val = numpy.min(sld_list[key])
1604	max_val = numpy.max(sld_list[key])
1605	mean_val = numpy.mean(sld_list[key])
1606	enable = (min_val == max_val) and \
1607	sld_data.pix_type == 'pixel'
1608	ctr_list[1].SetValue(format_number(mean_val, True))
1609	ctr_list[1].Enable(enable)
1610	#ctr_list[2].SetLabel("[" + sld_data.sld_unit + "]")
1611	break
1612
1613	def on_sld_draw(self, event):
1614	"""
1615	Draw sld profile as scattered plot
1616	"""
1617	self.parent.sld_draw()
1618
1619	def on_save(self, event):
1620	"""
1621	Close the window containing this panel
1622	"""
1623	flag = True
1624	flag = self.check_inputs()
1625	if not flag:
1626	return
1627	self.sld_data = self.get_sld_val()
1628	self.parent.set_main_panel_sld_data(self.sld_data)
1629
1630	reader = sas_gen.SLDReader()
1631	extension = '*.sld'
1632	path = None
1633	data= None
1634	location = self.parent.get_path()
1635	dlg = wx.FileDialog(self, "Save sld file",
1636	location, "sld_file",
1637	extension,
1638	wx.SAVE)
1639	if dlg.ShowModal() == wx.ID_OK:
1640	path = dlg.GetPath()
1641	self.parent.set_file_location(os.path.dirname(path))
1642	else:
1643	return None
1644	dlg.Destroy()
1645	try:
1646	if path is None:
1647	return
1648
1649	data = self.parent.get_sld_data()
1650	fName = os.path.splitext(path)[0] + '.' + extension.split('.')[-1]
1651	if data != None:
1652	try:
1653	reader.write(fName, data)
1654	except:
1655	raise
1656	else:
1657	msg = "%s cannot write %s\n" % ('Generic Scattering', str(path))
1658	infor = 'Error'
1659	#logging.error(msg)
1660	if self.parent.parent != None:
1661	# inform msg to wx
1662	wx.PostEvent(self.parent.parent,
1663	StatusEvent(status=msg, info=infor))
1664	self.SetFocus()
1665	return
1666	except:
1667	msg = "Error occurred while saving. "
1668	infor = 'Error'
1669	if self.parent.parent != None:
1670	# inform msg to wx
1671	wx.PostEvent(self.parent.parent,
1672	StatusEvent(status=msg, info=infor))
1673	self.SetFocus()
1674
1675	def _onparamEnter(self, event):
1676	"""
1677	"""
1678	flag = True
1679	if event != None:
1680	event.Skip()
1681	ctl = event.GetEventObject()
1682	ctl.SetBackgroundColour("white")
1683	#_set_error(self, ctl)
1684	try:
1685	float(ctl.GetValue())
1686	except:
1687	flag = _set_error(self, ctl)
1688	if flag:
1689	npts = 1
1690	for item in self.nodes:
1691	n_val = float(item[1].GetValue())
1692	if n_val <= 0:
1693	item[1].SetBackgroundColour("pink")
1694	npts = -1
1695	break
1696	if numpy.isfinite(n_val):
1697	npts *= int(n_val)
1698	if npts > 0:
1699	nop = self.set_npts_from_slddata()
1700	if nop == None:
1701	nop = npts
1702	self.display_npts(nop)
1703
1704	ctl.Refresh()
1705	return flag
1706
1707	def _set_volume_ctr_val(self, npts):
1708	"""
1709	Set total volume
1710	"""
1711	total_volume = npts * self.sld_data.vol_pix[0]
1712	self.parent.set_volume_ctr_val(total_volume)
1713
1714	def _onstepsize(self, event):
1715	"""
1716	On stepsize event
1717	"""
1718	flag = True
1719	if event != None:
1720	event.Skip()
1721	ctl = event.GetEventObject()
1722	ctl.SetBackgroundColour("white")
1723
1724	if flag and not self.sld_data.is_data:#ctl.IsEnabled():
1725	s_size = 1.0
1726	try:
1727	for item in self.stepsize:
1728	s_val = float(item[1].GetValue())
1729	if s_val <= 0:
1730	item[1].SetBackgroundColour("pink")
1731	ctl.Refresh()
1732	return
1733	if numpy.isfinite(s_val):
1734	s_size *= s_val
1735	self.sld_data.set_pixel_volumes(s_size)
1736	if ctl.IsEnabled():
1737	total_volume = sum(self.sld_data.vol_pix)
1738	self.parent.set_volume_ctr_val(total_volume)
1739	except:
1740	pass
1741	ctl.Refresh()
1742
1743
1744	def set_npts_from_slddata(self):
1745	"""
1746	Set total n. of points form the sld data
1747	"""
1748	try:
1749	sld_data = self.parent.get_sld_from_omf()
1750	#nop = (nop * numpy.pi) / 6
1751	nop = len(sld_data.sld_n)
1752	except:
1753	nop = None
1754	return nop
1755
1756	def display_npts(self, nop):
1757	"""
1758	Displays Npts ctrl
1759	"""
1760	try:
1761	self.npix_ctl.SetValue(str(nop))
1762	self.npix_ctl.Refresh()
1763	self.parent.set_etime()
1764	wx.CallAfter(self._set_volume_ctr_val, nop)
1765	except:
1766	# On Init
1767	pass
1768
1769	def check_inputs(self):
1770	"""
1771	check if the inputs are valid
1772	"""
1773	flag = self._check_input_helper(self.slds)
1774	if flag:
1775	flag = self._check_input_helper(self.nodes)
1776	if flag:
1777	flag = self._check_input_helper(self.stepsize)
1778	return flag
1779
1780	def _check_input_helper(self, list):
1781	"""
1782	Check list values
1783	"""
1784	flag = True
1785	for item in list:
1786	item[1].SetBackgroundColour("white")
1787	item[1].Refresh()
1788	try:
1789	float(item[1].GetValue())
1790	except:
1791	flag = _set_error(self, item[1])
1792	break
1793	return flag
1794
1795	class SasGenWindow(widget.CHILD_FRAME):
1796	"""
1797	GEN SAS main window
1798	"""
1799	def __init__(self, parent=None, manager= None, title="Generic Scattering Calculator",
1800	size=(PANEL_WIDTH * 1.3, PANEL_HEIGHT * 1.65), args, *kwds):
1801	"""
1802	Init
1803	"""
1804	kwds['size'] = size
1805	kwds['title'] = title
1806	widget.CHILD_FRAME.__init__(self, parent, args, *kwds)
1807	self.parent = parent
1808	self.base = manager
1809	self.omfpanel = OmfPanel(parent=self)
1810	self.panel = SasGenPanel(parent=self)
1811	self.data = None
1812	self.omfdata = sas_gen.OMFData()
1813	self.sld_data = None
1814	self._default_save_location = os.getcwd()
1815
1816	self._mgr = aui.AuiManager(self)
1817	self._mgr.SetDockSizeConstraint(0.5, 0.5)
1818	self._plot_title = ''
1819	self.scale2d = 'log_{10}'
1820	self.Bind(wx.EVT_CLOSE, self.on_close)
1821
1822	self._build_toolbar()
1823
1824	self.build_panels()
1825	self.SetPosition((25, 150))
1826	self.Show(True)
1827
1828	def _build_toolbar(self):
1829	"""
1830	Build toolbar
1831	"""
1832	tsize = (20, 20)
1833	# The legacy code doesn't work well for wx 3.0
1834	# but the old code produces better results with wx 2.8
1835	if wx.VERSION_STRING >= '3.0.0.0':
1836	tb = wx.ToolBar(self, style=wx.TB_HORIZONTAL \| wx.TB_FLAT)
1837	sizer = wx.BoxSizer(wx.VERTICAL)
1838	sizer.Add(tb, 0, wx.EXPAND)
1839	self.SetSizer(sizer)
1840	else:
1841	tb = self.CreateToolBar(wx.TB_HORIZONTAL \| wx.TB_FLAT)
1842	open_bmp = wx.ArtProvider.GetBitmap(wx.ART_FILE_OPEN, wx.ART_TOOLBAR,
1843	tsize)
1844	save_bmp = wx.ArtProvider.GetBitmap(wx.ART_FILE_SAVE_AS, wx.ART_TOOLBAR,
1845	tsize)
1846	close_bmp = wx.ArtProvider.GetBitmap(wx.ART_QUIT, wx.ART_TOOLBAR,
1847	tsize)
1848	help_bmp = wx.ArtProvider.GetBitmap(wx.ART_HELP, wx.ART_TOOLBAR,
1849	(17, 20))
1850
1851	id = wx.NewId()
1852	tb.AddLabelTool(id, "Open", open_bmp, shortHelp="Open",
1853	longHelp="Open sld/omf file")
1854	self.Bind(wx.EVT_TOOL, self.on_open_file, id=id)
1855
1856	id = wx.NewId()
1857	tb.AddSimpleTool(id, save_bmp, "Save", "Save as sld file")
1858	self.Bind(wx.EVT_TOOL, self.on_save_file, id=id)
1859
1860	tb.AddSeparator()
1861	id = wx.NewId()
1862	tb.AddSimpleTool(id, close_bmp, "Quit", "Quit")
1863	self.Bind(wx.EVT_TOOL, self.on_close, id=id)
1864
1865	tb.AddSeparator()
1866	id = wx.NewId()
1867	tb.AddSimpleTool(id, help_bmp, "Help", "Help")
1868	self.Bind(wx.EVT_TOOL, self.on_help, id=id)
1869
1870	tb.Realize()
1871
1872	def _build_menubar(self):
1873	"""
1874	Build menubar
1875	"""
1876	tsize = (13, 13)
1877	open_bmp = wx.ArtProvider.GetBitmap(wx.ART_FILE_OPEN, wx.ART_TOOLBAR,
1878	tsize)
1879	save_bmp = wx.ArtProvider.GetBitmap(wx.ART_FILE_SAVE_AS, wx.ART_TOOLBAR,
1880	tsize)
1881	quit_bmp = wx.ArtProvider.GetBitmap(wx.ART_QUIT, wx.ART_TOOLBAR,
1882	tsize)
1883	help_bmp = wx.ArtProvider.GetBitmap(wx.ART_HELP, wx.ART_TOOLBAR,
1884	(13, 15))
1885
1886	menu_bar = wx.MenuBar()
1887
1888	menu = wx.Menu()
1889	id = wx.NewId()
1890	item = wx.MenuItem(menu, id, "&Open sld/omf file")
1891	item.SetBitmap(open_bmp)
1892	menu.AppendItem(item)
1893	wx.EVT_MENU(self, id, self.on_open_file)
1894
1895	id = wx.NewId()
1896	item = wx.MenuItem(menu, id, "&Save as sld file")
1897	item.SetBitmap(save_bmp)
1898	menu.AppendItem(item)
1899	wx.EVT_MENU(self, id, self.on_save_file)
1900
1901	menu.AppendSeparator()
1902	id = wx.NewId()
1903	item = wx.MenuItem(menu, id, "&Quit")
1904	item.SetBitmap(quit_bmp)
1905	menu.AppendItem(item)
1906
1907	menu_bar.Append(menu, "&File")
1908	wx.EVT_MENU(self, id, self.on_close)
1909
1910	menu_help = wx.Menu()
1911	id = wx.NewId()
1912	item = wx.MenuItem(menu_help, id, "&Theory and GUI")
1913	item.SetBitmap(help_bmp)
1914	menu_help.AppendItem(item)
1915	wx.EVT_MENU(self, id, self.on_help)
1916
1917	menu_bar.Append(menu_help, "&Help")
1918
1919	self.SetMenuBar(menu_bar)
1920
1921	def build_panels(self):
1922	"""
1923	"""
1924
1925	self.set_sld_data(self.sld_data)
1926
1927	self._mgr.AddPane(self.panel, aui.AuiPaneInfo().
1928	Name(self.panel.window_name).
1929	CenterPane().
1930	# This is where we set the size of
1931	# the application window
1932	BestSize(wx.Size(PANEL_WIDTH,
1933	PANEL_HEIGHT)).
1934	Show())
1935	self._mgr.AddPane(self.omfpanel, aui.AuiPaneInfo().
1936	Name(self.omfpanel.window_name).
1937	Caption(self.omfpanel.window_caption).
1938	CloseButton(False).
1939	Right().
1940	Floatable(False).
1941	BestSize(wx.Size(PANEL_WIDTH/2.5, PANEL_HEIGHT)).
1942	Show())
1943	self._mgr.Update()
1944
1945	def get_sld_data(self):
1946	"""
1947	Return slddata
1948	"""
1949	return self.sld_data
1950
1951	def get_sld_from_omf(self):
1952	"""
1953	"""
1954	self.sld_data = self.omfpanel.get_sld_val()
1955	return self.sld_data
1956
1957	def set_sld_n(self, sld):
1958	"""
1959	"""
1960	self.panel.sld_data = sld
1961	self.panel.model.set_sld_data(sld)
1962
1963	def set_sld_data(self, data):
1964	"""
1965	Set omfdata
1966	"""
1967	if data == None:
1968	return
1969	self.sld_data = data
1970	enable = (not data==None)
1971	self._set_omfpanel_sld_data(self.sld_data)
1972	self.omfpanel.bt_save.Enable(enable)
1973	self.set_etime()
1974
1975	def set_omfpanel_npts(self):
1976	"""
1977	Set Npts in omf panel
1978	"""
1979	nop = self.omfpanel.set_npts_from_slddata()
1980	self.omfpanel.display_npts(nop)
1981
1982	def _set_omfpanel_sld_data(self, data):
1983	"""
1984	Set sld_data in omf panel
1985	"""
1986	self.omfpanel.set_slddata(data)
1987	self.omfpanel.set_sld_ctr(data)
1988
1989	def check_omfpanel_inputs(self):
1990	"""
1991	Check OMF panel inputs
1992	"""
1993	return self.omfpanel.check_inputs()
1994
1995	def set_main_panel_sld_data(self, sld_data):
1996	"""
1997	"""
1998	self.sld_data = sld_data
1999
2000	def set_file_location(self, path):
2001	"""
2002	File location
2003	"""
2004	self._default_save_location = path
2005
2006	def get_path(self):
2007	"""
2008	File location
2009	"""
2010	return self._default_save_location
2011
2012	def draw_graph(self, plot, title=''):
2013	"""
2014	"""
2015	try:
2016	wx.PostEvent(self.parent, NewPlotEvent(plot=plot, title=title))
2017	except:
2018	# standalone
2019	frame = PlotFrame(self, -1, 'testView', self.scale2d)
2020	#add_icon(self.parent, frame)
2021	frame.add_plot(plot)
2022	frame.SetTitle(title)
2023	frame.Show(True)
2024	frame.SetFocus()
2025
2026	def set_schedule_full_draw(self, panel=None, func='del'):
2027	"""
2028	Send full draw to gui frame
2029	"""
2030	if self.parent != None:
2031	self.parent.set_schedule_full_draw(panel, func)
2032
2033	def get_npix(self):
2034	"""
2035	Get no. of pixels from omf panel
2036	"""
2037	n_pix = self.omfpanel.npix_ctl.GetValue()
2038	return n_pix
2039
2040	def get_pix_volumes(self):
2041	"""
2042	Get a pixel volume
2043	"""
2044	vol = self.omfpanel.get_pix_volumes()
2045	return vol
2046
2047	def set_volume_ctr_val(self, val):
2048	"""
2049	Set volume txtctl value
2050	"""
2051	try:
2052	self.panel.set_volume_ctl_val(str(val))
2053	except:
2054	print "self.panel is not initialized yet"
2055
2056	def set_omfpanel_default_shap(self, shape):
2057	"""
2058	Set default_shape in omfpanel
2059	"""
2060	self.omfpanel.default_shape = shape
2061
2062	def set_etime(self):
2063	"""
2064	Sets est. computation time on panel
2065	"""
2066	self.panel.set_est_time()
2067
2068	def get_sld_data_from_omf(self):
2069	"""
2070	"""
2071	data = self.omfpanel.get_sld_val()
2072	return data
2073
2074	def set_scale2d(self, scale):
2075	"""
2076	"""
2077	self.scale2d = scale
2078
2079	def on_panel_close(self, event):
2080	"""
2081	"""
2082	#Not implemented
2083
2084	def on_open_file(self, event):
2085	"""
2086	On Open
2087	"""
2088	self.panel.on_load_data(event)
2089
2090	def sld_draw(self):
2091	"""
2092	sld draw
2093	"""
2094	self.panel.sld_draw(event=None, has_arrow=False)
2095
2096	def on_save_file(self, event):
2097	"""
2098	On Close
2099	"""
2100	self.omfpanel.on_save(event)
2101
2102	def on_close(self, event):
2103	"""
2104	Close
2105	"""
2106	if self.base != None:
2107	self.base.gen_frame = None
2108	self.Destroy()
2109
2110	def on_help(self, event):
2111	"""
2112	Gen scatter angle help panel
2113	"""
2114	from sas.perspectives.calculator.help_panel import HelpWindow
2115	# Get models help model_function path
2116	import sas.perspectives.calculator as calmedia
2117
2118	media = calmedia.get_data_path(media='media')
2119	path = os.path.join(media,"gen_sas_help.html")
2120	name = "Generic Scattering Calculator"
2121	frame = HelpWindow(self, -1,
2122	title=' Help: GenSAS',
2123	pageToOpen=path, size=(865, 450))
2124	try:
2125	frame.splitter.DetachWindow(frame.lpanel)
2126	# Display only the right side one
2127	frame.lpanel.Hide()
2128	frame.Show(True)
2129	add_icon(self.parent, frame)
2130	except:
2131	frame.Destroy()
2132	msg = 'Display Error\n'
2133	info = "Info"
2134	wx.MessageBox(msg, info)
2135
2136	if __name__ == "__main__":
2137	app = wx.PySimpleApp()
2138	widget.CHILD_FRAME = wx.Frame
2139	SGframe = SasGenWindow()
2140	SGframe.Show(True)
2141	app.MainLoop()

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

SasView

source: sasview/src/sas/perspectives/calculator/gen_scatter_panel.py @ b199493

Download in other formats: