gui_manager.py @ e2c0939

Last change on this file since e2c0939 was e2c0939, checked in by wojciech, 7 years ago
Setting proper CL env even when initiated from None
Property mode set to `100644`
File size: 129.3 KB

Line
1	"""
2	Gui manager: manages the widgets making up an application
3	"""
4	################################################################################
5	# This software was developed by the University of Tennessee as part of the
6	# Distributed Data Analysis of Neutron Scattering Experiments (DANSE)
7	# project funded by the US National Science Foundation.
8	#
9	# See the license text in license.txtz
10	#
11	# copyright 2008, University of Tennessee
12	################################################################################
13
14
15	import wx
16	import wx.aui
17	import os
18	import sys
19	import time
20	import imp
21	import warnings
22	import re
23	import logging
24	import httplib
25	import traceback
26	import urllib
27	import urllib2
28	import json
29
30	from sas.sasgui.guiframe.events import EVT_CATEGORY
31	from sas.sasgui.guiframe.events import EVT_STATUS
32	from sas.sasgui.guiframe.events import EVT_APPEND_BOOKMARK
33	from sas.sasgui.guiframe.events import EVT_PANEL_ON_FOCUS
34	from sas.sasgui.guiframe.events import EVT_NEW_LOAD_DATA
35	from sas.sasgui.guiframe.events import EVT_NEW_COLOR
36	from sas.sasgui.guiframe.events import StatusEvent
37	from sas.sasgui.guiframe.events import NewPlotEvent
38	from sas.sasgui.guiframe.gui_style import GUIFRAME
39	from sas.sasgui.guiframe.gui_style import GUIFRAME_ID
40	from sas.sasgui.guiframe.data_panel import DataPanel
41	from sas.sasgui.guiframe.panel_base import PanelBase
42	from sas.sasgui.guiframe.gui_toolbar import GUIToolBar
43	from sas.sasgui.guiframe.data_processor import GridFrame
44	from sas.sasgui.guiframe.events import EVT_NEW_BATCH
45	from sas.sasgui.guiframe.CategoryManager import CategoryManager
46	from sas.sascalc.dataloader.loader import Loader
47	from sas.sasgui.guiframe.proxy import Connection
48	from matplotlib import _pylab_helpers
49
50	warnings.simplefilter("ignore")
51
52
53	def get_app_dir():
54	"""
55	The application directory is the one where the default custom_config.py
56	file resides.
57
58	:returns: app_path - the path to the applicatin directory
59	"""
60	# First, try the directory of the executable we are running
61	app_path = sys.path[0]
62	if os.path.isfile(app_path):
63	app_path = os.path.dirname(app_path)
64	if os.path.isfile(os.path.join(app_path, "custom_config.py")):
65	app_path = os.path.abspath(app_path)
66	logging.info("Using application path: %s", app_path)
67	return app_path
68
69	# Next, try the current working directory
70	if os.path.isfile(os.path.join(os.getcwd(), "custom_config.py")):
71	logging.info("Using application path: %s", os.getcwd())
72	return os.path.abspath(os.getcwd())
73
74	# Finally, try the directory of the sasview module
75	# TODO: gui_manager will have to know about sasview until we
76	# clean all these module variables and put them into a config class
77	# that can be passed by sasview.py.
78	logging.info(sys.executable)
79	logging.info(str(sys.argv))
80	from sas import sasview as sasview
81	app_path = os.path.dirname(sasview.__file__)
82	logging.info("Using application path: %s", app_path)
83	return app_path
84
85
86	def get_user_directory():
87	"""
88	Returns the user's home directory
89	"""
90	userdir = os.path.join(os.path.expanduser("~"), ".sasview")
91	if not os.path.isdir(userdir):
92	os.makedirs(userdir)
93	return userdir
94
95
96	def _find_local_config(file, path):
97	"""
98	Find configuration file for the current application
99	"""
100	config_module = None
101	fObj = None
102	try:
103	fObj, path_config, descr = imp.find_module(file, [path])
104	config_module = imp.load_module(file, fObj, path_config, descr)
105	except:
106	logging.error("Error loading %s/%s: %s" % (path, file, sys.exc_value))
107	finally:
108	if fObj is not None:
109	fObj.close()
110	logging.info("GuiManager loaded %s/%s" % (path, file))
111	return config_module
112
113	# Get APP folder
114	PATH_APP = get_app_dir()
115	DATAPATH = PATH_APP
116
117	# GUI always starts from the App folder
118	# os.chdir(PATH_APP)
119	# Read in the local config, which can either be with the main
120	# application or in the installation directory
121	config = _find_local_config('local_config', PATH_APP)
122	if config is None:
123	config = _find_local_config('local_config', os.getcwd())
124	if config is None:
125	# Didn't find local config, load the default
126	import sas.sasgui.guiframe.config as config
127	logging.info("using default local_config")
128	else:
129	logging.info("found local_config in %s" % os.getcwd())
130	else:
131	logging.info("found local_config in %s" % PATH_APP)
132
133	from sas.sasgui.guiframe.customdir import SetupCustom
134	c_conf_dir = SetupCustom().setup_dir(PATH_APP)
135	custom_config = _find_local_config('custom_config', c_conf_dir)
136	if custom_config is None:
137	custom_config = _find_local_config('custom_config', os.getcwd())
138	if custom_config is None:
139	msgConfig = "Custom_config file was not imported"
140	logging.info(msgConfig)
141	else:
142	logging.info("using custom_config in %s" % os.getcwd())
143	else:
144	logging.info("using custom_config from %s" % c_conf_dir)
145
146	# read some constants from config
147	APPLICATION_STATE_EXTENSION = config.APPLICATION_STATE_EXTENSION
148	APPLICATION_NAME = config.__appname__
149	SPLASH_SCREEN_PATH = config.SPLASH_SCREEN_PATH
150	WELCOME_PANEL_ON = config.WELCOME_PANEL_ON
151	SPLASH_SCREEN_WIDTH = config.SPLASH_SCREEN_WIDTH
152	SPLASH_SCREEN_HEIGHT = config.SPLASH_SCREEN_HEIGHT
153	SS_MAX_DISPLAY_TIME = config.SS_MAX_DISPLAY_TIME
154	SAS_OPENCL = config.SAS_OPENCL
155	if not WELCOME_PANEL_ON:
156	WELCOME_PANEL_SHOW = False
157	else:
158	WELCOME_PANEL_SHOW = True
159	try:
160	DATALOADER_SHOW = custom_config.DATALOADER_SHOW
161	TOOLBAR_SHOW = custom_config.TOOLBAR_SHOW
162	FIXED_PANEL = custom_config.FIXED_PANEL
163	if WELCOME_PANEL_ON:
164	WELCOME_PANEL_SHOW = custom_config.WELCOME_PANEL_SHOW
165	PLOPANEL_WIDTH = custom_config.PLOPANEL_WIDTH
166	DATAPANEL_WIDTH = custom_config.DATAPANEL_WIDTH
167	GUIFRAME_WIDTH = custom_config.GUIFRAME_WIDTH
168	GUIFRAME_HEIGHT = custom_config.GUIFRAME_HEIGHT
169	CONTROL_WIDTH = custom_config.CONTROL_WIDTH
170	CONTROL_HEIGHT = custom_config.CONTROL_HEIGHT
171	DEFAULT_PERSPECTIVE = custom_config.DEFAULT_PERSPECTIVE
172	CLEANUP_PLOT = custom_config.CLEANUP_PLOT
173	# custom open_path
174	open_folder = custom_config.DEFAULT_OPEN_FOLDER
175	if open_folder is not None and os.path.isdir(open_folder):
176	DEFAULT_OPEN_FOLDER = os.path.abspath(open_folder)
177	else:
178	DEFAULT_OPEN_FOLDER = PATH_APP
179	SAS_OPENCL = custom_config.SAS_OPENCL
180	except:
181	DATALOADER_SHOW = True
182	TOOLBAR_SHOW = True
183	FIXED_PANEL = True
184	WELCOME_PANEL_SHOW = False
185	PLOPANEL_WIDTH = config.PLOPANEL_WIDTH
186	DATAPANEL_WIDTH = config.DATAPANEL_WIDTH
187	GUIFRAME_WIDTH = config.GUIFRAME_WIDTH
188	GUIFRAME_HEIGHT = config.GUIFRAME_HEIGHT
189	CONTROL_WIDTH = -1
190	CONTROL_HEIGHT = -1
191	DEFAULT_PERSPECTIVE = None
192	CLEANUP_PLOT = False
193	DEFAULT_OPEN_FOLDER = PATH_APP
194	SAS_OPENCL = None
195	DEFAULT_STYLE = config.DEFAULT_STYLE
196
197	PLUGIN_STATE_EXTENSIONS = config.PLUGIN_STATE_EXTENSIONS
198	OPEN_SAVE_MENU = config.OPEN_SAVE_PROJECT_MENU
199	VIEW_MENU = config.VIEW_MENU
200	EDIT_MENU = config.EDIT_MENU
201	extension_list = []
202	if APPLICATION_STATE_EXTENSION is not None:
203	extension_list.append(APPLICATION_STATE_EXTENSION)
204	EXTENSIONS = PLUGIN_STATE_EXTENSIONS + extension_list
205	try:
206	PLUGINS_WLIST = '\|'.join(config.PLUGINS_WLIST)
207	except:
208	PLUGINS_WLIST = ''
209	APPLICATION_WLIST = config.APPLICATION_WLIST
210	IS_WIN = True
211	IS_LINUX = False
212	CLOSE_SHOW = True
213	TIME_FACTOR = 2
214	MDI_STYLE = wx.DEFAULT_FRAME_STYLE
215	NOT_SO_GRAPH_LIST = ["BoxSum"]
216	PARENT_FRAME = wx.MDIParentFrame
217	CHILD_FRAME = wx.MDIChildFrame
218	if sys.platform.count("win32") < 1:
219	IS_WIN = False
220	TIME_FACTOR = 2
221	if int(str(wx.__version__).split('.')[0]) == 2:
222	if int(str(wx.__version__).split('.')[1]) < 9:
223	CLOSE_SHOW = False
224	if sys.platform.count("darwin") < 1:
225	IS_LINUX = True
226	PARENT_FRAME = wx.Frame
227	CHILD_FRAME = wx.Frame
228
229	#Initiliaze enviromental variable with custom setting but only if variable not set
230	if SAS_OPENCL and not "SAS_OPENCL" in os.environ:
231	os.environ["SAS_OPENCL"] = SAS_OPENCL
232
233	class ViewerFrame(PARENT_FRAME):
234	"""
235	Main application frame
236	"""
237
238	def __init__(self, parent, title,
239	size=(GUIFRAME_WIDTH, GUIFRAME_HEIGHT),
240	gui_style=DEFAULT_STYLE,
241	style=wx.DEFAULT_FRAME_STYLE,
242	pos=wx.DefaultPosition):
243	"""
244	Initialize the Frame object
245	"""
246	PARENT_FRAME.__init__(self, parent=parent, title=title,
247	pos=pos, size=size)
248	# title
249	self.title = title
250	self.__gui_style = gui_style
251	path = os.path.dirname(__file__)
252	temp_path = os.path.join(path, 'images')
253	ico_file = os.path.join(temp_path, 'ball.ico')
254	if os.path.isfile(ico_file):
255	self.SetIcon(wx.Icon(ico_file, wx.BITMAP_TYPE_ICO))
256	else:
257	temp_path = os.path.join(os.getcwd(), 'images')
258	ico_file = os.path.join(temp_path, 'ball.ico')
259	if os.path.isfile(ico_file):
260	self.SetIcon(wx.Icon(ico_file, wx.BITMAP_TYPE_ICO))
261	else:
262	ico_file = os.path.join(os.path.dirname(os.path.sys.path[0]),
263	'images', 'ball.ico')
264	if os.path.isfile(ico_file):
265	self.SetIcon(wx.Icon(ico_file, wx.BITMAP_TYPE_ICO))
266	self.path = PATH_APP
267	self.application_name = APPLICATION_NAME
268	# Application manager
269	self._input_file = None
270	self.app_manager = None
271	self._mgr = None
272	# add current perpsective
273	self._current_perspective = None
274	self._plotting_plugin = None
275	self._data_plugin = None
276	# Menu bar and item
277	self._menubar = None
278	self._file_menu = None
279	self._data_menu = None
280	self._view_menu = None
281	self._data_panel_menu = None
282	self._help_menu = None
283	self._tool_menu = None
284	self._applications_menu_pos = -1
285	self._applications_menu_name = None
286	self._applications_menu = None
287	self._edit_menu = None
288	self._toolbar_menu = None
289	self._save_appl_menu = None
290	# tool bar
291	self._toolbar = None
292	# Status bar
293	self.sb = None
294	# number of plugins
295	self._num_perspectives = 0
296	# plot duck cleanup option
297	self.cleanup_plots = CLEANUP_PLOT
298	# Find plug-ins
299	# Modify this so that we can specify the directory to look into
300	self.plugins = []
301	# add local plugin
302	self.plugins += self._get_local_plugins()
303	self.plugins += self._find_plugins()
304	# List of panels
305	self.panels = {}
306	# List of plot panels
307	self.plot_panels = {}
308	# default Graph number fot the plotpanel caption
309	self.graph_num = 0
310
311	# Default locations
312	self._default_save_location = DEFAULT_OPEN_FOLDER
313	# Welcome panel
314	self.defaultPanel = None
315	self.welcome_panel_class = None
316	# panel on focus
317	self.panel_on_focus = None
318	# control_panel on focus
319	self.cpanel_on_focus = None
320
321	self.loader = Loader()
322	# data manager
323	self.batch_on = False
324	from sas.sasgui.guiframe.data_manager import DataManager
325	self._data_manager = DataManager()
326	self._data_panel = None # DataPanel(parent=self)
327	if self.panel_on_focus is not None:
328	self._data_panel.set_panel_on_focus(
329	self.panel_on_focus.window_caption)
330	# list of plot panels in schedule to full redraw
331	self.schedule = False
332	# self.callback = True
333	self._idle_count = 0
334	self.schedule_full_draw_list = []
335	self.idletimer = wx.CallLater(TIME_FACTOR, self._onDrawIdle)
336
337	self.batch_frame = GridFrame(parent=self)
338	self.batch_frame.Hide()
339	self.on_batch_selection(event=None)
340	self.add_icon()
341
342	# Register the close event so it calls our own method
343	wx.EVT_CLOSE(self, self.WindowClose)
344	# Register to status events
345	self.Bind(EVT_STATUS, self._on_status_event)
346	# Register add extra data on the same panel event on load
347	self.Bind(EVT_PANEL_ON_FOCUS, self.set_panel_on_focus)
348	self.Bind(EVT_APPEND_BOOKMARK, self.append_bookmark)
349	self.Bind(EVT_NEW_LOAD_DATA, self.on_load_data)
350	self.Bind(EVT_NEW_BATCH, self.on_batch_selection)
351	self.Bind(EVT_NEW_COLOR, self.on_color_selection)
352	self.Bind(EVT_CATEGORY, self.on_change_categories)
353	self.setup_custom_conf()
354	# Preferred window size
355	self._window_width, self._window_height = size
356
357	def add_icon(self):
358	"""
359	get list of child and attempt to add the default icon
360	"""
361
362	list_children = self.GetChildren()
363	for frame in list_children:
364	self.put_icon(frame)
365
366	def put_icon(self, frame):
367	"""
368	Put icon on the tap of a panel
369	"""
370	if hasattr(frame, "IsIconized"):
371	if not frame.IsIconized():
372	try:
373	icon = self.GetIcon()
374	frame.SetIcon(icon)
375	except:
376	logging.error("ViewerFrame.put_icon: could not set icon")
377
378	def get_client_size(self):
379	"""
380	return client size tuple
381	"""
382	width, height = self.GetClientSizeTuple()
383	height -= 45
384	# Adjust toolbar height
385	toolbar = self.GetToolBar()
386	if toolbar is not None:
387	_, tb_h = toolbar.GetSizeTuple()
388	height -= tb_h
389	return width, height
390
391	def on_change_categories(self, evt):
392	# ILL
393	fitpanel = None
394	for item in self.plugins:
395	if hasattr(item, "get_panels"):
396	if hasattr(item, "fit_panel"):
397	fitpanel = item.fit_panel
398
399	if fitpanel is not None:
400	for i in range(0, fitpanel.GetPageCount()):
401	fitpanel.GetPage(i)._populate_listbox()
402
403	def on_set_batch_result(self, data_outputs, data_inputs=None,
404	plugin_name=""):
405	"""
406	Display data into a grid in batch mode and show the grid
407	"""
408	t = time.localtime(time.time())
409	time_str = time.strftime("%b %d %H;%M of %Y", t)
410	details = "File Generated by %s : %s" % (APPLICATION_NAME,
411	str(plugin_name))
412	details += "on %s.\n" % time_str
413	ext = ".csv"
414	file_name = "Batch_" + str(plugin_name) + "_" + time_str + ext
415	file_name = self._default_save_location + str(file_name)
416
417	self.open_with_localapp(file_name=file_name,
418	details=details,
419	data_inputs=data_inputs,
420	data_outputs=data_outputs)
421
422	def open_with_localapp(self, data_inputs=None, details="", file_name=None,
423	data_outputs=None):
424	"""
425	Display value of data into the application grid
426	:param data_inputs: dictionary of string and list of items
427	:param details: descriptive string
428	:param file_name: file name
429	:param data_outputs: Data outputs
430	"""
431	self.batch_frame.set_data(data_inputs=data_inputs,
432	data_outputs=data_outputs,
433	details=details,
434	file_name=file_name)
435	self.show_batch_frame(None)
436
437	def on_read_batch_tofile(self, base):
438	"""
439	Open a file dialog , extract the file to read and display values
440	into a grid
441	"""
442	path = None
443	if self._default_save_location is None:
444	self._default_save_location = os.getcwd()
445	wildcard = "(.csv; .txt)\|.csv; .txt"
446	dlg = wx.FileDialog(base,
447	"Choose a file",
448	self._default_save_location, "",
449	wildcard)
450	if dlg.ShowModal() == wx.ID_OK:
451	path = dlg.GetPath()
452	if path is not None:
453	self._default_save_location = os.path.dirname(path)
454	dlg.Destroy()
455	try:
456	self.read_batch_tofile(file_name=path)
457	except:
458	msg = "Error occurred when reading the file; %s\n" % path
459	msg += "%s\n" % sys.exc_value
460	wx.PostEvent(self, StatusEvent(status=msg,
461	info="error"))
462
463	def read_batch_tofile(self, file_name):
464	"""
465	Extract value from file name and Display them into a grid
466	"""
467	if file_name is None or file_name.strip() == "":
468	return
469	data = {}
470	fd = open(file_name, 'r')
471	_, ext = os.path.splitext(file_name)
472	separator = None
473	if ext.lower() == ".csv":
474	separator = ","
475	fd_buffer = fd.read()
476	lines = fd_buffer.split('\n')
477	fd.close()
478	column_names_line = ""
479	index = None
480	details = ""
481	for index in range(len(lines)):
482	line = lines[index]
483	line.strip()
484	count = 0
485	if separator is None:
486	line.replace('\t', ' ')
487	# found the first line containing the label
488	col_name_toks = line.split()
489	for item in col_name_toks:
490	if item.strip() != "":
491	count += 1
492	else:
493	line = " "
494	elif line.find(separator) != -1:
495	if line.count(separator) >= 2:
496	# found the first line containing the label
497	col_name_toks = line.split(separator)
498	for item in col_name_toks:
499	if item.strip() != "":
500	count += 1
501	else:
502	details += line
503	if count >= 2:
504	column_names_line = line
505	break
506
507	if column_names_line.strip() == "" or index is None:
508	return
509
510	col_name_toks = column_names_line.split(separator)
511	c_index = 0
512	for col_index in range(len(col_name_toks)):
513	c_name = col_name_toks[col_index]
514	if c_name.strip() != "":
515	# distinguish between column name and value
516	try:
517	float(c_name)
518	col_name = "Column %s" % str(col_index + 1)
519	index_min = index
520	except:
521	col_name = c_name
522	index_min = index + 1
523	data[col_name] = [lines[row].split(separator)[c_index]
524	for row in range(index_min, len(lines) - 1)]
525	c_index += 1
526
527	self.open_with_localapp(data_outputs=data, data_inputs=None,
528	file_name=file_name, details=details)
529
530	def write_batch_tofile(self, data, file_name, details=""):
531	"""
532	Helper to write result from batch into cvs file
533	"""
534	self._default_save_location = os.path.dirname(file_name)
535	name = os.path.basename(file_name)
536	if data is None or file_name is None or file_name.strip() == "":
537	return
538	_, ext = os.path.splitext(name)
539	try:
540	fd = open(file_name, 'w')
541	except:
542	# On Permission denied: IOError
543	temp_dir = get_user_directory()
544	temp_file_name = os.path.join(temp_dir, name)
545	fd = open(temp_file_name, 'w')
546	separator = "\t"
547	if ext.lower() == ".csv":
548	separator = ","
549	fd.write(str(details))
550	for col_name in data.keys():
551	fd.write(str(col_name))
552	fd.write(separator)
553	fd.write('\n')
554	max_list = [len(value) for value in data.values()]
555	if len(max_list) == 0:
556	return
557	max_index = max(max_list)
558	index = 0
559	while index < max_index:
560	for value_list in data.values():
561	if index < len(value_list):
562	fd.write(str(value_list[index]))
563	fd.write(separator)
564	else:
565	fd.write('')
566	fd.write(separator)
567	fd.write('\n')
568	index += 1
569	fd.close()
570
571	def open_with_externalapp(self, data, file_name, details=""):
572	"""
573	Display data in the another application , by default Excel
574	"""
575	if not os.path.exists(file_name):
576	self.write_batch_tofile(data=data, file_name=file_name,
577	details=details)
578	try:
579	from win32com.client import Dispatch
580	excel_app = Dispatch('Excel.Application')
581	excel_app.Workbooks.Open(file_name)
582	excel_app.Visible = 1
583	except:
584	msg = "Error occured when calling Excel.\n"
585	msg += "Check that Excel installed in this machine or \n"
586	msg += "check that %s really exists.\n" % str(file_name)
587	wx.PostEvent(self, StatusEvent(status=msg,
588	info="error"))
589
590	def on_batch_selection(self, event=None):
591	"""
592	:param event: contains parameter enable. When enable is set to True
593	the application is in Batch mode otherwise the application is
594	in Single mode.
595	"""
596	if event is not None:
597	self.batch_on = event.enable
598	for plug in self.plugins:
599	plug.set_batch_selection(self.batch_on)
600
601	def on_color_selection(self, event):
602	"""
603	:param event: contains parameters for id and color
604	"""
605	color, event_id = event.color, event.id
606	for plug in self.plugins:
607	plug.add_color(color, event_id)
608
609	def setup_custom_conf(self):
610	"""
611	Set up custom configuration if exists
612	"""
613	if custom_config is None:
614	return
615
616	if not FIXED_PANEL:
617	self.__gui_style &= (~GUIFRAME.FIXED_PANEL)
618	self.__gui_style \|= GUIFRAME.FLOATING_PANEL
619
620	if not DATALOADER_SHOW:
621	self.__gui_style &= (~GUIFRAME.MANAGER_ON)
622
623	if not TOOLBAR_SHOW:
624	self.__gui_style &= (~GUIFRAME.TOOLBAR_ON)
625
626	if WELCOME_PANEL_SHOW:
627	self.__gui_style \|= GUIFRAME.WELCOME_PANEL_ON
628
629	def set_custom_default_perspective(self):
630	"""
631	Set default starting perspective
632	"""
633	if custom_config is None:
634	return
635	for plugin in self.plugins:
636	try:
637	if plugin.sub_menu == DEFAULT_PERSPECTIVE:
638
639	plugin.on_perspective(event=None)
640	frame = plugin.get_frame()
641	frame.Show(True)
642	# break
643	else:
644	frame = plugin.get_frame()
645	frame.Show(False)
646	except:
647	pass
648	return
649
650	def on_load_data(self, event):
651	"""
652	received an event to trigger load from data plugin
653	"""
654	if self._data_plugin is not None:
655	self._data_plugin.load_data(event)
656
657	def get_current_perspective(self):
658	"""
659	return the current perspective
660	"""
661	return self._current_perspective
662
663	def get_save_location(self):
664	"""
665	return the _default_save_location
666	"""
667	return self._default_save_location
668
669	def set_input_file(self, input_file):
670	"""
671	:param input_file: file to read
672	"""
673	self._input_file = input_file
674
675	def get_data_manager(self):
676	"""
677	return the data manager.
678	"""
679	return self._data_manager
680
681	def get_toolbar(self):
682	"""
683	return the toolbar.
684	"""
685	return self._toolbar
686
687	def set_panel_on_focus(self, event):
688	"""
689	Store reference to the last panel on focus
690	update the toolbar if available
691	update edit menu if available
692	"""
693	if event is not None:
694	self.panel_on_focus = event.panel
695	if self.panel_on_focus is not None:
696	# Disable save application if the current panel is in batch mode
697	try:
698	flag = self.panel_on_focus.get_save_flag()
699	if self._save_appl_menu is not None:
700	self._save_appl_menu.Enable(flag)
701
702	if self.panel_on_focus not in self.plot_panels.values():
703	for ID in self.panels.keys():
704	if self.panel_on_focus != self.panels[ID]:
705	self.panels[ID].on_kill_focus(None)
706
707	if self._data_panel is not None and \
708	self.panel_on_focus is not None:
709	self.set_panel_on_focus_helper()
710	# update toolbar
711	self._update_toolbar_helper()
712	# update edit menu
713	self.enable_edit_menu()
714	except wx._core.PyDeadObjectError:
715	pass
716
717	def disable_app_menu(self, p_panel=None):
718	"""
719	Disables all menus in the menubar
720	"""
721	return
722
723	def send_focus_to_datapanel(self, name):
724	"""
725	Send focusing on ID to data explorer
726	"""
727	if self._data_panel is not None:
728	self._data_panel.set_panel_on_focus(name)
729
730	def set_panel_on_focus_helper(self):
731	"""
732	Helper for panel on focus with data_panel
733	"""
734	caption = self.panel_on_focus.window_caption
735	self.send_focus_to_datapanel(caption)
736	# update combo
737	if self.panel_on_focus in self.plot_panels.values():
738	combo = self._data_panel.cb_plotpanel
739	combo_title = str(self.panel_on_focus.window_caption)
740	combo.SetStringSelection(combo_title)
741	combo.SetToolTip(wx.ToolTip(combo_title))
742	elif self.panel_on_focus != self._data_panel:
743	cpanel = self.panel_on_focus
744	if self.cpanel_on_focus != cpanel:
745	cpanel.on_tap_focus()
746	self.cpanel_on_focus = self.panel_on_focus
747
748	def reset_bookmark_menu(self, panel):
749	"""
750	Reset Bookmark menu list
751
752	: param panel: a control panel or tap where the bookmark is
753	"""
754	cpanel = panel
755	if self._toolbar is not None and cpanel._bookmark_flag:
756	for item in self._toolbar.get_bookmark_items():
757	self._toolbar.remove_bookmark_item(item)
758	self._toolbar.add_bookmark_default()
759	pos = 0
760	for bitem in cpanel.popUpMenu.GetMenuItems():
761	pos += 1
762	if pos < 3:
763	continue
764	id = bitem.GetId()
765	label = bitem.GetLabel()
766	self._toolbar.append_bookmark_item(id, label)
767	wx.EVT_MENU(self, id, cpanel._back_to_bookmark)
768	self._toolbar.Realize()
769
770	def build_gui(self):
771	"""
772	Build the GUI by setting up the toolbar, menu and layout.
773	"""
774	# set tool bar
775	self._setup_tool_bar()
776
777	# Create the menu bar. To be filled later.
778	# WX 3.0 needs us to create the menu bar first.
779	self._menubar = wx.MenuBar()
780	if wx.VERSION_STRING >= '3.0.0.0':
781	self.SetMenuBar(self._menubar)
782	self._add_menu_file()
783	self._add_menu_edit()
784	self._add_menu_view()
785	self._add_menu_tool()
786	# Set up the layout
787	self._setup_layout()
788	self._add_menu_application()
789
790	# Set up the menu
791	self._add_current_plugin_menu()
792	self._add_help_menu()
793	# Append item from plugin under menu file if necessary
794	self._populate_file_menu()
795
796	if not wx.VERSION_STRING >= '3.0.0.0':
797	self.SetMenuBar(self._menubar)
798
799	try:
800	self.load_from_cmd(self._input_file)
801	except:
802	msg = "%s Cannot load file %s\n" % (str(APPLICATION_NAME),
803	str(self._input_file))
804	msg += str(sys.exc_value) + '\n'
805	logging.error(msg)
806	if self._data_panel is not None and len(self.plugins) > 0:
807	self._data_panel.fill_cbox_analysis(self.plugins)
808	self.post_init()
809	# Set Custom default
810	self.set_custom_default_perspective()
811	# Set up extra custom tool menu
812	self._setup_extra_custom()
813	self._check_update(None)
814
815	def _setup_extra_custom(self):
816	"""
817	Set up toolbar and welcome view if needed
818	"""
819	style = self.__gui_style & GUIFRAME.TOOLBAR_ON
820	if (style == GUIFRAME.TOOLBAR_ON) & (not self._toolbar.IsShown()):
821	self._on_toggle_toolbar()
822
823	# Set Custom deafult start page
824	welcome_style = self.__gui_style & GUIFRAME.WELCOME_PANEL_ON
825	if welcome_style == GUIFRAME.WELCOME_PANEL_ON:
826	self.show_welcome_panel(None)
827
828	def _setup_layout(self):
829	"""
830	Set up the layout
831	"""
832	# Status bar
833	from sas.sasgui.guiframe.gui_statusbar import StatusBar
834	self.sb = StatusBar(self, wx.ID_ANY)
835	self.SetStatusBar(self.sb)
836	# Load panels
837	self._load_panels()
838
839	def SetStatusText(self, args, *kwds):
840	"""
841	"""
842	number = self.sb.get_msg_position()
843	wx.Frame.SetStatusText(self, number=number, args, *kwds)
844
845	def PopStatusText(self, args, *kwds):
846	"""
847	"""
848	field = self.sb.get_msg_position()
849	wx.Frame.PopStatusText(self, field=field)
850
851	def PushStatusText(self, args, *kwds):
852	"""
853	.. todo:: No message is passed. What is this supposed to do?
854	"""
855	field = self.sb.get_msg_position()
856	wx.Frame.PushStatusText(self, field=field,
857	string=
858	"FIXME - PushStatusText called without text")
859
860	def add_perspective(self, plugin):
861	"""
862	Add a perspective if it doesn't already
863	exist.
864	"""
865	self._num_perspectives += 1
866	is_loaded = False
867	for item in self.plugins:
868	item.set_batch_selection(self.batch_on)
869	if plugin.__class__ == item.__class__:
870	msg = "Plugin %s already loaded" % plugin.sub_menu
871	logging.info(msg)
872	is_loaded = True
873	if not is_loaded:
874	self.plugins.append(plugin)
875	msg = "Plugin %s appended" % plugin.sub_menu
876	logging.info(msg)
877
878	def _get_local_plugins(self):
879	"""
880	get plugins local to guiframe and others
881	"""
882	plugins = []
883	# import guiframe local plugins
884	# check if the style contain guiframe.dataloader
885	style1 = self.__gui_style & GUIFRAME.DATALOADER_ON
886	style2 = self.__gui_style & GUIFRAME.PLOTTING_ON
887	if style1 == GUIFRAME.DATALOADER_ON:
888	try:
889	from sas.sasgui.guiframe.local_perspectives.data_loader \
890	import data_loader
891	self._data_plugin = data_loader.Plugin()
892	plugins.append(self._data_plugin)
893	except:
894	msg = "ViewerFrame._get_local_plugins:"
895	msg += "cannot import dataloader plugin.\n %s" % sys.exc_value
896	logging.error(msg)
897	if style2 == GUIFRAME.PLOTTING_ON:
898	try:
899	from sas.sasgui.guiframe.local_perspectives.plotting \
900	import plotting
901	self._plotting_plugin = plotting.Plugin()
902	plugins.append(self._plotting_plugin)
903	except:
904	msg = "ViewerFrame._get_local_plugins:"
905	msg += "cannot import plotting plugin.\n %s" % sys.exc_value
906	logging.error(msg)
907
908	return plugins
909
910	def _find_plugins(self, dir="perspectives"):
911	"""
912	Find available perspective plug-ins
913
914	:param dir: directory in which to look for plug-ins
915
916	:returns: list of plug-ins
917
918	"""
919	plugins = []
920	# Go through files in panels directory
921	try:
922	if os.path.isdir(dir):
923	file_list = os.listdir(dir)
924	else:
925	file_list = []
926	# the default panel is the panel is the last plugin added
927	for item in file_list:
928	toks = os.path.splitext(os.path.basename(item))
929	name = ''
930	if not toks[0] == '__init__':
931	if toks[1] == '.py' or toks[1] == '':
932	name = toks[0]
933	# check the validity of the module name parsed
934	# before trying to import it
935	if name is None or name.strip() == '':
936	continue
937	path = [os.path.abspath(dir)]
938	file = None
939	try:
940	if toks[1] == '':
941	mod_path = '.'.join([dir, name])
942	module = __import__(mod_path, globals(),
943	locals(), [name])
944	else:
945	(file, path, info) = imp.find_module(name, path)
946	module = imp.load_module(name, file, item, info)
947	if hasattr(module, "PLUGIN_ID"):
948	try:
949	plugins.append(module.Plugin())
950	msg = "Found plug-in: %s" % module.PLUGIN_ID
951	logging.info(msg)
952	except:
953	msg = "Error accessing PluginPanel"
954	msg += " in %s\n %s" % (name, sys.exc_value)
955	config.printEVT(msg)
956	except:
957	msg = "ViewerFrame._find_plugins: %s" % sys.exc_value
958	logging.error(msg)
959	finally:
960	if file is not None:
961	file.close()
962	except:
963	# Should raise and catch at a higher level and
964	# display error on status bar
965	logging.error(sys.exc_value)
966
967	return plugins
968
969	def _get_panels_size(self, p):
970	"""
971	find the proper size of the current panel
972	get the proper panel width and height
973	"""
974	self._window_width, self._window_height = self.get_client_size()
975	# Default size
976	if DATAPANEL_WIDTH < 0:
977	panel_width = int(self._window_width * 0.25)
978	else:
979	panel_width = DATAPANEL_WIDTH
980	panel_height = int(self._window_height)
981	if self._data_panel is not None and (p == self._data_panel):
982	return panel_width, panel_height
983	if hasattr(p, "CENTER_PANE") and p.CENTER_PANE:
984	panel_width = self._window_width * 0.45
985	if CONTROL_WIDTH > 0:
986	panel_width = CONTROL_WIDTH
987	if CONTROL_HEIGHT > 0:
988	panel_height = CONTROL_HEIGHT
989	return panel_width, panel_height
990	elif p == self.defaultPanel:
991	return self._window_width, panel_height
992	return panel_width, panel_height
993
994	def _load_panels(self):
995	"""
996	Load all panels in the panels directory
997	"""
998	# Look for plug-in panels
999	panels = []
1000	if wx.VERSION_STRING >= '3.0.0.0':
1001	mac_pos_y = 85
1002	else:
1003	mac_pos_y = 40
1004	for item in self.plugins:
1005	if hasattr(item, "get_panels"):
1006	ps = item.get_panels(self)
1007	panels.extend(ps)
1008
1009	# Set up welcome panel
1010	# TODO: this needs serious simplification
1011	if self.welcome_panel_class is not None:
1012	welcome_panel = MDIFrame(self, None, 'None', (100, 200))
1013	self.defaultPanel = self.welcome_panel_class(welcome_panel, -1,
1014	style=wx.RAISED_BORDER)
1015	welcome_panel.set_panel(self.defaultPanel)
1016	self.defaultPanel.set_frame(welcome_panel)
1017	welcome_panel.Show(False)
1018
1019	self.panels["default"] = self.defaultPanel
1020	size_t_bar = 70
1021	if IS_LINUX:
1022	size_t_bar = 115
1023	if self.defaultPanel is not None:
1024	w, h = self._get_panels_size(self.defaultPanel)
1025	frame = self.defaultPanel.get_frame()
1026	frame.SetSize((self._window_width, self._window_height))
1027	if not IS_WIN:
1028	frame.SetPosition((0, mac_pos_y + size_t_bar))
1029	frame.Show(True)
1030	# add data panel
1031	win = MDIFrame(self, None, 'None', (100, 200))
1032	data_panel = DataPanel(parent=win, id=-1)
1033	win.set_panel(data_panel)
1034	self.panels["data_panel"] = data_panel
1035	self._data_panel = data_panel
1036	d_panel_width, h = self._get_panels_size(self._data_panel)
1037	win.SetSize((d_panel_width, h))
1038	is_visible = self.__gui_style & \
1039	GUIFRAME.MANAGER_ON == GUIFRAME.MANAGER_ON
1040	if IS_WIN:
1041	win.SetPosition((0, 0))
1042	else:
1043	win.SetPosition((0, mac_pos_y + size_t_bar))
1044	win.Show(is_visible)
1045	# Add the panels to the AUI manager
1046	for panel_class in panels:
1047	frame = panel_class.get_frame()
1048	wx_id = wx.NewId()
1049	# Check whether we need to put this panel in the center pane
1050	if hasattr(panel_class, "CENTER_PANE") and panel_class.CENTER_PANE:
1051	w, h = self._get_panels_size(panel_class)
1052	if panel_class.CENTER_PANE:
1053	self.panels[str(wx_id)] = panel_class
1054	_, pos_y = frame.GetPositionTuple()
1055	frame.SetPosition((d_panel_width + 1, pos_y))
1056	frame.SetSize((w, h))
1057	frame.Show(False)
1058	elif panel_class == self._data_panel:
1059	panel_class.frame.Show(is_visible)
1060	continue
1061	else:
1062	self.panels[str(wx_id)] = panel_class
1063	frame.SetSize((w, h))
1064	frame.Show(False)
1065	if IS_WIN:
1066	frame.SetPosition((d_panel_width + 1, 0))
1067	else:
1068	frame.SetPosition((d_panel_width + 1, mac_pos_y + size_t_bar))
1069
1070	if not IS_WIN:
1071	win_height = mac_pos_y
1072	if IS_LINUX:
1073	if wx.VERSION_STRING >= '3.0.0.0':
1074	win_height = mac_pos_y + 10
1075	else:
1076	win_height = mac_pos_y + 55
1077	self.SetMaxSize((-1, win_height))
1078	else:
1079	self.SetSize((self._window_width, win_height))
1080
1081	def update_data(self, prev_data, new_data):
1082	"""
1083	Update the data.
1084	"""
1085	prev_id, data_state = self._data_manager.update_data(
1086	prev_data=prev_data, new_data=new_data)
1087
1088	self._data_panel.remove_by_id(prev_id)
1089	self._data_panel.load_data_list(data_state)
1090
1091	def update_theory(self, data_id, theory, state=None):
1092	"""
1093	Update the theory
1094	"""
1095	data_state = self._data_manager.update_theory(data_id=data_id,
1096	theory=theory,
1097	state=state)
1098	wx.CallAfter(self._data_panel.load_data_list, data_state)
1099
1100	def onfreeze(self, theory_id):
1101	"""
1102	Saves theory/model and passes to data loader.
1103
1104	..warning:: This seems to be the exact same code as the next
1105	function called simply freeze. This probably needs fixing
1106	"""
1107	data_state_list = self._data_manager.freeze(theory_id)
1108	self._data_panel.load_data_list(list=data_state_list)
1109	for data_state in data_state_list.values():
1110	new_plot = data_state.get_data()
1111
1112	wx.PostEvent(self, NewPlotEvent(plot=new_plot,
1113	title=new_plot.title))
1114
1115	def freeze(self, data_id, theory_id):
1116	"""
1117	Saves theory/model and passes to data loader.
1118
1119	..warning:: This seems to be the exact same code as the next
1120	function called simply freeze. This probably needs fixing
1121	"""
1122	data_state_list = self._data_manager.freeze_theory(data_id=data_id,
1123	theory_id=theory_id)
1124	self._data_panel.load_data_list(list=data_state_list)
1125	for data_state in data_state_list.values():
1126	new_plot = data_state.get_data()
1127	wx.PostEvent(self, NewPlotEvent(plot=new_plot,
1128	title=new_plot.title))
1129
1130	def delete_data(self, data):
1131	"""
1132	Delete the data.
1133	"""
1134	self._current_perspective.delete_data(data)
1135
1136	def get_context_menu(self, plotpanel=None):
1137	"""
1138	Get the context menu items made available
1139	by the different plug-ins.
1140	This function is used by the plotting module
1141	"""
1142	if plotpanel is None:
1143	return
1144	menu_list = []
1145	for item in self.plugins:
1146	menu_list.extend(item.get_context_menu(plotpanel=plotpanel))
1147	return menu_list
1148
1149	def get_current_context_menu(self, plotpanel=None):
1150	"""
1151	Get the context menu items made available
1152	by the current plug-in.
1153	This function is used by the plotting module
1154	"""
1155	if plotpanel is None:
1156	return
1157	menu_list = []
1158	item = self._current_perspective
1159	if item is not None:
1160	menu_list.extend(item.get_context_menu(plotpanel=plotpanel))
1161	return menu_list
1162
1163	def on_panel_close(self, event):
1164	"""
1165	Gets called when the close event for a panel runs.
1166	This will check which panel has been closed and
1167	delete it.
1168	"""
1169	frame = event.GetEventObject()
1170	for ID in self.plot_panels.keys():
1171	if self.plot_panels[ID].window_name == frame.name:
1172	self.disable_app_menu(self.plot_panels[ID])
1173	self.delete_panel(ID)
1174	break
1175	if self.cpanel_on_focus is not None:
1176	self.cpanel_on_focus.SetFocus()
1177
1178	def popup_panel(self, p):
1179	"""
1180	Add a panel object to the AUI manager
1181
1182	:param p: panel object to add to the AUI manager
1183
1184	:returns: ID of the event associated with the new panel [int]
1185
1186	"""
1187	ID = wx.NewId()
1188	self.panels[str(ID)] = p
1189	# Check and set the size
1190	if PLOPANEL_WIDTH < 0:
1191	p_panel_width = int(self._window_width * 0.45)
1192	else:
1193	p_panel_width = PLOPANEL_WIDTH
1194	p_panel_height = int(p_panel_width * 0.76)
1195	p.frame.SetSize((p_panel_width, p_panel_height))
1196	self.graph_num += 1
1197	if p.window_caption.split()[0] in NOT_SO_GRAPH_LIST:
1198	windowcaption = p.window_caption
1199	else:
1200	windowcaption = 'Graph'
1201	windowname = p.window_name
1202
1203	# Append nummber
1204	captions = self._get_plotpanel_captions()
1205	# FIXME: Fix this awful loop
1206	while (1):
1207	caption = windowcaption + '%s' % str(self.graph_num)
1208	if caption not in captions:
1209	break
1210	self.graph_num += 1
1211	# protection from forever-loop: max num = 1000
1212	if self.graph_num > 1000:
1213	break
1214	if p.window_caption.split()[0] not in NOT_SO_GRAPH_LIST:
1215	p.window_caption = caption
1216	p.window_name = windowname + str(self.graph_num)
1217
1218	p.frame.SetTitle(p.window_caption)
1219	p.frame.name = p.window_name
1220	if not IS_WIN:
1221	p.frame.Center()
1222	x_pos, _ = p.frame.GetPositionTuple()
1223	p.frame.SetPosition((x_pos, 112))
1224	p.frame.Show(True)
1225
1226	# Register for showing/hiding the panel
1227	wx.EVT_MENU(self, ID, self.on_view)
1228	if p not in self.plot_panels.values() and p.group_id is not None:
1229	self.plot_panels[ID] = p
1230	if len(self.plot_panels) == 1:
1231	self.panel_on_focus = p
1232	self.set_panel_on_focus(None)
1233	if self._data_panel is not None and \
1234	self._plotting_plugin is not None:
1235	ind = self._data_panel.cb_plotpanel.FindString('None')
1236	if ind != wx.NOT_FOUND:
1237	self._data_panel.cb_plotpanel.Delete(ind)
1238	if caption not in self._data_panel.cb_plotpanel.GetItems():
1239	self._data_panel.cb_plotpanel.Append(str(caption), p)
1240	return ID
1241
1242	def _get_plotpanel_captions(self):
1243	"""
1244	Get all the plotpanel cations
1245
1246	: return: list of captions
1247	"""
1248	captions = []
1249	for Id in self.plot_panels.keys():
1250	captions.append(self.plot_panels[Id].window_caption)
1251
1252	return captions
1253
1254	def _setup_tool_bar(self):
1255	"""
1256	add toolbar to the frame
1257	"""
1258	self._toolbar = GUIToolBar(self)
1259	# The legacy code doesn't work well for wx 3.0
1260	# but the old code produces better results with wx 2.8
1261	if not IS_WIN and wx.VERSION_STRING >= '3.0.0.0':
1262	sizer = wx.BoxSizer(wx.VERTICAL)
1263	sizer.Add(self._toolbar, 0, wx.EXPAND)
1264	self.SetSizer(sizer)
1265	else:
1266	self.SetToolBar(self._toolbar)
1267	self._update_toolbar_helper()
1268	self._on_toggle_toolbar(event=None)
1269
1270	def _update_toolbar_helper(self):
1271	"""
1272	Helping to update the toolbar
1273	"""
1274	application_name = 'No Selected Analysis'
1275	panel_name = 'No Panel on Focus'
1276	c_panel = self.cpanel_on_focus
1277	if self._toolbar is None:
1278	return
1279	if c_panel is not None:
1280	self.reset_bookmark_menu(self.cpanel_on_focus)
1281	if self._current_perspective is not None:
1282	application_name = self._current_perspective.sub_menu
1283	c_panel_state = c_panel
1284	if c_panel is not None:
1285	panel_name = c_panel.window_caption
1286	if not c_panel.IsShownOnScreen():
1287	c_panel_state = None
1288	self._toolbar.update_toolbar(c_panel_state)
1289	self._toolbar.update_button(application_name=application_name,
1290	panel_name=panel_name)
1291	self._toolbar.Realize()
1292
1293	def _add_menu_tool(self):
1294	"""
1295	Tools menu
1296	Go through plug-ins and find tools to populate the tools menu
1297	"""
1298	style = self.__gui_style & GUIFRAME.CALCULATOR_ON
1299	if style == GUIFRAME.CALCULATOR_ON:
1300	self._tool_menu = None
1301	for item in self.plugins:
1302	if hasattr(item, "get_tools"):
1303	for tool in item.get_tools():
1304	# Only create a menu if we have at least one tool
1305	if self._tool_menu is None:
1306	self._tool_menu = wx.Menu()
1307	if tool[0].lower().count('python') > 0:
1308	self._tool_menu.AppendSeparator()
1309	id = wx.NewId()
1310	self._tool_menu.Append(id, tool[0], tool[1])
1311	wx.EVT_MENU(self, id, tool[2])
1312	if self._tool_menu is not None:
1313	self._menubar.Append(self._tool_menu, '&Tools')
1314
1315	def _add_current_plugin_menu(self):
1316	"""
1317	add current plugin menu
1318	Look for plug-in menus
1319	Add available plug-in sub-menus.
1320	"""
1321	if self._menubar is None or self._current_perspective is None \
1322	or self._menubar.GetMenuCount() == 0:
1323	return
1324	# replace or add a new menu for the current plugin
1325	pos = self._menubar.FindMenu(str(self._applications_menu_name))
1326	if pos == -1 and self._applications_menu_pos > 0:
1327	pos = self._applications_menu_pos
1328	if pos != -1:
1329	menu_list = self._current_perspective.populate_menu(self)
1330	if menu_list:
1331	for (menu, name) in menu_list:
1332	self._menubar.Replace(pos, menu, name)
1333	self._applications_menu_name = name
1334	self._applications_menu_pos = pos
1335	else:
1336	self._menubar.Remove(pos)
1337	self._applications_menu_name = None
1338	self._applications_menu_pos = -1
1339	else:
1340	menu_list = self._current_perspective.populate_menu(self)
1341	if menu_list:
1342	for (menu, name) in menu_list:
1343	if self._applications_menu_pos == -1:
1344	# Find the Analysis position and insert just after it
1345	analysis_pos = self._menubar.FindMenu("Analysis")
1346	if analysis_pos == -1:
1347	self._menubar.Append(menu, name)
1348	self._applications_menu_pos = -1
1349	else:
1350	self._menubar.Insert(analysis_pos+1, menu, name)
1351	self._applications_menu_pos = analysis_pos + 1
1352	else:
1353	self._menubar.Insert(self._applications_menu_pos,
1354	menu, name)
1355	self._applications_menu_name = name
1356
1357	def _on_marketplace_click(self, event):
1358	"""
1359	Click event for the help menu item linking to the Marketplace.
1360	"""
1361	import webbrowser
1362	webbrowser.open_new(config.marketplace_url)
1363
1364	def _add_help_menu(self):
1365	"""
1366	add help menu to menu bar. Includes welcome page, about page,
1367	tutorial PDF and documentation pages.
1368	"""
1369	self._help_menu = wx.Menu()
1370
1371	wx_id = wx.NewId()
1372	self._help_menu.Append(wx_id, '&Documentation', '')
1373	wx.EVT_MENU(self, wx_id, self._onSphinxDocs)
1374
1375	if config._do_tutorial and (IS_WIN or sys.platform == 'darwin'):
1376	wx_id = wx.NewId()
1377	self._help_menu.Append(wx_id, '&Tutorial', 'Software tutorial')
1378	wx.EVT_MENU(self, wx_id, self._onTutorial)
1379
1380	if config._do_acknowledge:
1381	wx_id = wx.NewId()
1382	self._help_menu.Append(wx_id, '&Acknowledge',
1383	'Acknowledging SasView')
1384	wx.EVT_MENU(self, wx_id, self._onAcknowledge)
1385
1386	if config._do_aboutbox:
1387	logging.info("Doing help menu")
1388	wx_id = wx.NewId()
1389	self._help_menu.Append(wx_id, '&About', 'Software information')
1390	wx.EVT_MENU(self, wx_id, self._onAbout)
1391
1392	if config.marketplace_url:
1393	wx_id = wx.NewId()
1394	self._help_menu.Append(wx_id, '&Model marketplace', '')
1395	wx.EVT_MENU(self, wx_id, self._on_marketplace_click)
1396
1397	# Checking for updates
1398	wx_id = wx.NewId()
1399	self._help_menu.Append(wx_id, '&Check for update',
1400	'Check for the latest version of %s' %
1401	config.__appname__)
1402	wx.EVT_MENU(self, wx_id, self._check_update)
1403	self._menubar.Append(self._help_menu, '&Help')
1404
1405	def _add_menu_view(self):
1406	"""
1407	add menu items under view menu
1408	"""
1409	if not VIEW_MENU:
1410	return
1411	self._view_menu = wx.Menu()
1412
1413	wx_id = wx.NewId()
1414	hint = "Display the Grid Window for batch results etc."
1415	self._view_menu.Append(wx_id, '&Show Grid Window', hint)
1416	wx.EVT_MENU(self, wx_id, self.show_batch_frame)
1417
1418	self._view_menu.AppendSeparator()
1419	style = self.__gui_style & GUIFRAME.MANAGER_ON
1420	wx_id = wx.NewId()
1421	self._data_panel_menu = self._view_menu.Append(wx_id,
1422	'&Show Data Explorer',
1423	'')
1424	wx.EVT_MENU(self, wx_id, self.show_data_panel)
1425	if style == GUIFRAME.MANAGER_ON:
1426	self._data_panel_menu.SetText('Hide Data Explorer')
1427	else:
1428	self._data_panel_menu.SetText('Show Data Explorer')
1429
1430	self._view_menu.AppendSeparator()
1431	wx_id = wx.NewId()
1432	style1 = self.__gui_style & GUIFRAME.TOOLBAR_ON
1433	if style1 == GUIFRAME.TOOLBAR_ON:
1434	self._toolbar_menu = self._view_menu.Append(wx_id, '&Hide Toolbar',
1435	'')
1436	else:
1437	self._toolbar_menu = self._view_menu.Append(wx_id, '&Show Toolbar',
1438	'')
1439	wx.EVT_MENU(self, wx_id, self._on_toggle_toolbar)
1440
1441	if custom_config is not None:
1442	self._view_menu.AppendSeparator()
1443	wx_id = wx.NewId()
1444	hint_ss = "Select the current/default configuration "
1445	hint_ss += "as a startup setting"
1446	preference_menu = self._view_menu.Append(wx_id, 'Startup Setting',
1447	hint_ss)
1448	wx.EVT_MENU(self, wx_id, self._on_preference_menu)
1449
1450	wx_id = wx.NewId()
1451	self._view_menu.AppendSeparator()
1452	self._view_menu.Append(wx_id, 'Category Manager',
1453	'Edit model categories')
1454	wx.EVT_MENU(self, wx_id, self._on_category_manager)
1455
1456	self._menubar.Append(self._view_menu, '&View')
1457
1458	def show_batch_frame(self, event=None):
1459	"""
1460	show the grid of result
1461	"""
1462	# Show(False) before Show(True) in order to bring it to the front
1463	self.batch_frame.Show(False)
1464	self.batch_frame.Show(True)
1465
1466	def on_category_panel(self, event):
1467	"""
1468	On cat panel
1469	"""
1470	self._on_category_manager(event)
1471
1472	def _on_category_manager(self, event):
1473	"""
1474	Category manager frame
1475	"""
1476	frame = CategoryManager(self, -1, 'Model Category Manager')
1477	icon = self.GetIcon()
1478	frame.SetIcon(icon)
1479
1480	def _on_preference_menu(self, event):
1481	"""
1482	Build a panel to allow to edit Mask
1483	"""
1484	from sas.sasgui.guiframe.startup_configuration \
1485	import StartupConfiguration as ConfDialog
1486
1487	dialog = ConfDialog(parent=self, gui=self.__gui_style)
1488	result = dialog.ShowModal()
1489	if result == wx.ID_OK:
1490	dialog.write_custom_config()
1491	# post event for info
1492	wx.PostEvent(self, StatusEvent(status="Wrote custom configuration",
1493	info='info'))
1494	dialog.Destroy()
1495
1496	def _add_menu_application(self):
1497	"""
1498	# Attach a menu item for each defined perspective or application.
1499	# Only add the perspective menu if there are more than one perspectives
1500	add menu application
1501	"""
1502	if self._num_perspectives > 1:
1503	plug_data_count = False
1504	plug_no_data_count = False
1505	self._applications_menu = wx.Menu()
1506	pos = 0
1507	separator = self._applications_menu.AppendSeparator()
1508	for plug in self.plugins:
1509	if len(plug.get_perspective()) > 0:
1510	id = wx.NewId()
1511	if plug.use_data():
1512	self._applications_menu.InsertCheckItem(pos, id,
1513	plug.sub_menu,
1514	"Switch to analysis: %s" % plug.sub_menu)
1515	plug_data_count = True
1516	pos += 1
1517	else:
1518	plug_no_data_count = True
1519	self._applications_menu.AppendCheckItem(id,
1520	plug.sub_menu,
1521	"Switch to analysis: %s" % plug.sub_menu)
1522	wx.EVT_MENU(self, id, plug.on_perspective)
1523
1524	if not plug_data_count or not plug_no_data_count:
1525	self._applications_menu.RemoveItem(separator)
1526	# Windows introduces a "Window" menu item during the layout process
1527	# somehow. We want it to be next to the last item with Help as
1528	# last. However Analysis gets stuck after Window in normal ordering
1529	# so force it to be next after the Tools menu item. Should we add
1530	# another menu item will need to check if this is still where we
1531	# want Analysis. This is NOT an issue on the Mac which does not
1532	# have the extra Window menu item.
1533	# March 2016 Code Camp -- PDB
1534	Tools_pos = self._menubar.FindMenu("Tools")
1535	self._menubar.Insert(Tools_pos+1, self._applications_menu,
1536	'&Analysis')
1537	self._check_applications_menu()
1538
1539	def _populate_file_menu(self):
1540	"""
1541	Insert menu item under file menu
1542	"""
1543	for plugin in self.plugins:
1544	if len(plugin.populate_file_menu()) > 0:
1545	for item in plugin.populate_file_menu():
1546	m_name, m_hint, m_handler = item
1547	wx_id = wx.NewId()
1548	self._file_menu.Append(wx_id, m_name, m_hint)
1549	wx.EVT_MENU(self, wx_id, m_handler)
1550	self._file_menu.AppendSeparator()
1551
1552	style1 = self.__gui_style & GUIFRAME.MULTIPLE_APPLICATIONS
1553	if OPEN_SAVE_MENU:
1554	wx_id = wx.NewId()
1555	hint_load_file = "read all analysis states saved previously"
1556	self._save_appl_menu = self._file_menu.Append(wx_id,
1557	'&Open Project',
1558	hint_load_file)
1559	wx.EVT_MENU(self, wx_id, self._on_open_state_project)
1560
1561	if style1 == GUIFRAME.MULTIPLE_APPLICATIONS:
1562	# some menu of plugin to be seen under file menu
1563	hint_load_file = "Read a status files and load"
1564	hint_load_file += " them into the analysis"
1565	wx_id = wx.NewId()
1566	self._save_appl_menu = self._file_menu.Append(wx_id,
1567	'&Open Analysis',
1568	hint_load_file)
1569	wx.EVT_MENU(self, wx_id, self._on_open_state_application)
1570	if OPEN_SAVE_MENU:
1571	self._file_menu.AppendSeparator()
1572	wx_id = wx.NewId()
1573	self._file_menu.Append(wx_id, '&Save Project',
1574	'Save the state of the whole analysis')
1575	wx.EVT_MENU(self, wx_id, self._on_save_project)
1576	if style1 == GUIFRAME.MULTIPLE_APPLICATIONS:
1577	wx_id = wx.NewId()
1578	txt = '&Save Analysis'
1579	txt2 = 'Save state of the current active analysis panel'
1580	self._save_appl_menu = self._file_menu.Append(wx_id, txt, txt2)
1581	wx.EVT_MENU(self, wx_id, self._on_save_application)
1582	if not sys.platform == 'darwin':
1583	self._file_menu.AppendSeparator()
1584	wx_id = wx.NewId()
1585	self._file_menu.Append(wx_id, '&Quit', 'Exit')
1586	wx.EVT_MENU(self, wx_id, self.Close)
1587
1588	def _add_menu_file(self):
1589	"""
1590	add menu file
1591	"""
1592	# File menu
1593	self._file_menu = wx.Menu()
1594	# Add sub menus
1595	self._menubar.Append(self._file_menu, '&File')
1596
1597	def _add_menu_edit(self):
1598	"""
1599	add menu edit
1600	"""
1601	if not EDIT_MENU:
1602	return
1603	# Edit Menu
1604	self._edit_menu = wx.Menu()
1605	self._edit_menu.Append(GUIFRAME_ID.UNDO_ID, '&Undo',
1606	'Undo the previous action')
1607	wx.EVT_MENU(self, GUIFRAME_ID.UNDO_ID, self.on_undo_panel)
1608	self._edit_menu.Append(GUIFRAME_ID.REDO_ID, '&Redo',
1609	'Redo the previous action')
1610	wx.EVT_MENU(self, GUIFRAME_ID.REDO_ID, self.on_redo_panel)
1611	self._edit_menu.AppendSeparator()
1612	self._edit_menu.Append(GUIFRAME_ID.COPY_ID, '&Copy Params',
1613	'Copy parameter values')
1614	wx.EVT_MENU(self, GUIFRAME_ID.COPY_ID, self.on_copy_panel)
1615	self._edit_menu.Append(GUIFRAME_ID.PASTE_ID, '&Paste Params',
1616	'Paste parameter values')
1617	wx.EVT_MENU(self, GUIFRAME_ID.PASTE_ID, self.on_paste_panel)
1618
1619	self._edit_menu.AppendSeparator()
1620
1621	self._edit_menu_copyas = wx.Menu()
1622	# Sub menu for Copy As...
1623	self._edit_menu_copyas.Append(GUIFRAME_ID.COPYEX_ID,
1624	'Copy current tab to Excel',
1625	'Copy parameter values in tabular format')
1626	wx.EVT_MENU(self, GUIFRAME_ID.COPYEX_ID, self.on_copy_panel)
1627
1628	self._edit_menu_copyas.Append(GUIFRAME_ID.COPYLAT_ID,
1629	'Copy current tab to LaTeX',
1630	'Copy parameter values in tabular format')
1631	wx.EVT_MENU(self, GUIFRAME_ID.COPYLAT_ID, self.on_copy_panel)
1632
1633	self._edit_menu.AppendMenu(GUIFRAME_ID.COPYAS_ID, 'Copy Params as...',
1634	self._edit_menu_copyas,
1635	'Copy parameter values in various formats')
1636
1637	self._edit_menu.AppendSeparator()
1638
1639	self._edit_menu.Append(GUIFRAME_ID.PREVIEW_ID, '&Report Results',
1640	'Preview current panel')
1641	wx.EVT_MENU(self, GUIFRAME_ID.PREVIEW_ID, self.on_preview_panel)
1642
1643	self._edit_menu.Append(GUIFRAME_ID.RESET_ID, '&Reset Page',
1644	'Reset current panel')
1645	wx.EVT_MENU(self, GUIFRAME_ID.RESET_ID, self.on_reset_panel)
1646
1647	self._menubar.Append(self._edit_menu, '&Edit')
1648	self.enable_edit_menu()
1649
1650	def get_style(self):
1651	"""
1652	Return the gui style
1653	"""
1654	return self.__gui_style
1655
1656	def _add_menu_data(self):
1657	"""
1658	Add menu item item data to menu bar
1659	"""
1660	if self._data_plugin is not None:
1661	menu_list = self._data_plugin.populate_menu(self)
1662	if menu_list:
1663	for (menu, name) in menu_list:
1664	self._menubar.Append(menu, name)
1665
1666	def _on_toggle_toolbar(self, event=None):
1667	"""
1668	hide or show toolbar
1669	"""
1670	if self._toolbar is None:
1671	return
1672	if self._toolbar.IsShown():
1673	if self._toolbar_menu is not None:
1674	self._toolbar_menu.SetItemLabel('Show Toolbar')
1675	self._toolbar.Hide()
1676	else:
1677	if self._toolbar_menu is not None:
1678	self._toolbar_menu.SetItemLabel('Hide Toolbar')
1679	self._toolbar.Show()
1680	self._toolbar.Realize()
1681
1682	def _on_status_event(self, evt):
1683	"""
1684	Display status message
1685	"""
1686	# This CallAfter fixes many crashes on MAC.
1687	wx.CallAfter(self.sb.set_status, evt)
1688
1689	def on_view(self, evt):
1690	"""
1691	A panel was selected to be shown. If it's not already
1692	shown, display it.
1693
1694	:param evt: menu event
1695
1696	"""
1697	panel_id = str(evt.GetId())
1698	self.on_set_plot_focus(self.panels[panel_id])
1699	wx.CallLater(5 * TIME_FACTOR, self.set_schedule(True))
1700	self.set_plot_unfocus()
1701
1702	def show_welcome_panel(self, event):
1703	"""
1704	Display the welcome panel
1705	"""
1706	if self.defaultPanel is None:
1707	return
1708	frame = self.panels['default'].get_frame()
1709	if frame is None:
1710	return
1711	# Show default panel
1712	if not frame.IsShown():
1713	frame.Show(True)
1714
1715	def on_close_welcome_panel(self):
1716	"""
1717	Close the welcome panel
1718	"""
1719	if self.defaultPanel is None:
1720	return
1721	default_panel = self.panels["default"].frame
1722	if default_panel.IsShown():
1723	default_panel.Show(False)
1724
1725	def delete_panel(self, uid):
1726	"""
1727	delete panel given uid
1728	"""
1729	ID = str(uid)
1730	config.printEVT("delete_panel: %s" % ID)
1731	if ID in self.panels.keys():
1732	self.panel_on_focus = None
1733	panel = self.panels[ID]
1734
1735	if hasattr(panel, "connect"):
1736	panel.connect.disconnect()
1737	self._plotting_plugin.delete_panel(panel.group_id)
1738
1739	if panel in self.schedule_full_draw_list:
1740	self.schedule_full_draw_list.remove(panel)
1741
1742	# delete uid number not str(uid)
1743	if ID in self.plot_panels.keys():
1744	del self.plot_panels[ID]
1745	if ID in self.panels.keys():
1746	del self.panels[ID]
1747	else:
1748	logging.error("delete_panel: No such plot id as %s" % ID)
1749
1750	def create_gui_data(self, data, path=None):
1751	"""
1752	"""
1753	return self._data_manager.create_gui_data(data, path)
1754
1755	def get_data(self, path):
1756	"""
1757	"""
1758	log_msg = ''
1759	basename = os.path.basename(path)
1760	_, extension = os.path.splitext(basename)
1761	if extension.lower() not in EXTENSIONS:
1762	log_msg = "File Loader cannot "
1763	log_msg += "load: %s\n" % str(basename)
1764	log_msg += "Try Data opening...."
1765	logging.error(log_msg)
1766	return
1767
1768	# reading a state file
1769	for plug in self.plugins:
1770	reader, ext = plug.get_extensions()
1771	if reader is not None:
1772	# read the state of the single plugin
1773	if extension == ext:
1774	reader.read(path)
1775	return
1776	elif extension == APPLICATION_STATE_EXTENSION:
1777	try:
1778	reader.read(path)
1779	except:
1780	msg = "DataLoader Error: Encounted Non-ASCII character"
1781	msg += "\n(%s)" % sys.exc_value
1782	wx.PostEvent(self, StatusEvent(status=msg,
1783	info="error",
1784	type="stop"))
1785	return
1786
1787	style = self.__gui_style & GUIFRAME.MANAGER_ON
1788	if style == GUIFRAME.MANAGER_ON:
1789	if self._data_panel is not None:
1790	self._data_panel.frame.Show(True)
1791
1792	def load_from_cmd(self, path):
1793	"""
1794	load data from cmd or application
1795	"""
1796	if path is None:
1797	return
1798	else:
1799	path = os.path.abspath(path)
1800	if not os.path.isfile(path) and not os.path.isdir(path):
1801	return
1802
1803	if os.path.isdir(path):
1804	self.load_folder(path)
1805	return
1806
1807	basename = os.path.basename(path)
1808	_, extension = os.path.splitext(basename)
1809	if extension.lower() not in EXTENSIONS:
1810	self.load_data(path)
1811	else:
1812	self.load_state(path)
1813
1814	self._default_save_location = os.path.dirname(path)
1815
1816	def load_state(self, path, is_project=False):
1817	"""
1818	load data from command line or application
1819	"""
1820	if path and (path is not None) and os.path.isfile(path):
1821	basename = os.path.basename(path)
1822	if APPLICATION_STATE_EXTENSION is not None \
1823	and basename.endswith(APPLICATION_STATE_EXTENSION):
1824	if is_project:
1825	for ID in self.plot_panels.keys():
1826	panel = self.plot_panels[ID]
1827	panel.on_close(None)
1828	self.get_data(path)
1829	wx.PostEvent(self, StatusEvent(status="Completed loading."))
1830	else:
1831	wx.PostEvent(self, StatusEvent(status=" "))
1832
1833	def load_data(self, path):
1834	"""
1835	load data from command line
1836	"""
1837	if not os.path.isfile(path):
1838	return
1839	basename = os.path.basename(path)
1840	_, extension = os.path.splitext(basename)
1841	if extension.lower() in EXTENSIONS:
1842	log_msg = "Data Loader cannot "
1843	log_msg += "load: %s\n" % str(path)
1844	log_msg += "Try File opening ...."
1845	logging.error(log_msg)
1846	return
1847	log_msg = ''
1848	output = {}
1849	error_message = ""
1850	try:
1851	logging.info("Loading Data...:\n" + str(path) + "\n")
1852	temp = self.loader.load(path)
1853	if temp.__class__.__name__ == "list":
1854	for item in temp:
1855	data = self.create_gui_data(item, path)
1856	output[data.id] = data
1857	else:
1858	data = self.create_gui_data(temp, path)
1859	output[data.id] = data
1860
1861	self.add_data(data_list=output)
1862	except:
1863	error_message = "Error while loading"
1864	error_message += " Data from cmd:\n %s\n" % str(path)
1865	error_message += str(sys.exc_value) + "\n"
1866	logging.error(error_message)
1867
1868	def load_folder(self, path):
1869	"""
1870	Load entire folder
1871	"""
1872	if not os.path.isdir(path):
1873	return
1874	if self._data_plugin is None:
1875	return
1876	try:
1877	if path is not None:
1878	self._default_save_location = os.path.dirname(path)
1879	file_list = self._data_plugin.get_file_path(path)
1880	self._data_plugin.get_data(file_list)
1881	else:
1882	return
1883	except:
1884	error_message = "Error while loading"
1885	error_message += " Data folder from cmd:\n %s\n" % str(path)
1886	error_message += str(sys.exc_value) + "\n"
1887	logging.error(error_message)
1888
1889	def _on_open_state_application(self, event):
1890	"""
1891	"""
1892	path = None
1893	if self._default_save_location is None:
1894	self._default_save_location = os.getcwd()
1895	wx.PostEvent(self, StatusEvent(status="Loading Analysis file..."))
1896	plug_wlist = self._on_open_state_app_helper()
1897	dlg = wx.FileDialog(self,
1898	"Choose a file",
1899	self._default_save_location, "",
1900	plug_wlist)
1901	if dlg.ShowModal() == wx.ID_OK:
1902	path = dlg.GetPath()
1903	if path is not None:
1904	self._default_save_location = os.path.dirname(path)
1905	dlg.Destroy()
1906	self.load_state(path=path)
1907
1908	def _on_open_state_app_helper(self):
1909	"""
1910	Helps '_on_open_state_application()' to find the extension of
1911	the current perspective/application
1912	"""
1913	# No current perspective or no extension attr
1914	if self._current_perspective is None:
1915	return PLUGINS_WLIST
1916	try:
1917	# Find the extension of the perspective
1918	# and get that as 1st item in list
1919	ind = None
1920	app_ext = self._current_perspective._extensions
1921	plug_wlist = config.PLUGINS_WLIST
1922	for ext in set(plug_wlist):
1923	if ext.count(app_ext) > 0:
1924	ind = ext
1925	break
1926	# Found the extension
1927	if ind is not None:
1928	plug_wlist.remove(ind)
1929	plug_wlist.insert(0, ind)
1930	try:
1931	plug_wlist = '\|'.join(plug_wlist)
1932	except:
1933	plug_wlist = ''
1934
1935	except:
1936	plug_wlist = PLUGINS_WLIST
1937
1938	return plug_wlist
1939
1940	def _on_open_state_project(self, event):
1941	"""
1942	Load in a .svs project file after removing all data from SasView
1943	"""
1944	path = None
1945	if self._default_save_location is None:
1946	self._default_save_location = os.getcwd()
1947	msg = "This operation will set remove all data, plots and analyses from"
1948	msg += " SasView before loading the project. Do you wish to continue?"
1949	msg_box = wx.MessageDialog(None, msg, 'Warning', wx.OK \| wx.CANCEL)
1950	if msg_box.ShowModal() == wx.ID_OK:
1951	self._data_panel.selection_cbox.SetValue('Select all Data')
1952	self._data_panel._on_selection_type(None)
1953	for _, theory_dict in self._data_panel.list_cb_theory.iteritems():
1954	for key, value in theory_dict.iteritems():
1955	item, _, _ = value
1956	item.Check(True)
1957
1958	wx.PostEvent(self, StatusEvent(status="Loading Project file..."))
1959	dlg = wx.FileDialog(self, "Choose a file",
1960	self._default_save_location, "",
1961	APPLICATION_WLIST)
1962	if dlg.ShowModal() == wx.ID_OK:
1963	path = dlg.GetPath()
1964	if path is not None:
1965	self._default_save_location = os.path.dirname(path)
1966	dlg.Destroy()
1967	# Reset to a base state
1968	self._on_reset_state()
1969	self._data_panel.on_remove(None, False)
1970	# Load the project file
1971	self.load_state(path=path, is_project=True)
1972
1973	def _on_reset_state(self):
1974	"""
1975	Resets SasView to its freshly opened state.
1976	:return: None
1977	"""
1978	# Reset all plugins to their base state
1979	self._data_panel.set_panel_on_focus()
1980	# Remove all loaded data
1981	for plugin in self.plugins:
1982	plugin.clear_panel()
1983	# Reset plot number to 0
1984	self.graph_num = 0
1985
1986	def _on_save_application(self, event):
1987	"""
1988	save the state of the current active application
1989	"""
1990	if self.cpanel_on_focus is not None:
1991	try:
1992	wx.PostEvent(self,
1993	StatusEvent(status="Saving Analysis file..."))
1994	self.cpanel_on_focus.on_save(event)
1995	wx.PostEvent(self,
1996	StatusEvent(status="Completed saving."))
1997	except Exception:
1998	msg = "Error occurred while saving: "
1999	msg += traceback.format_exc()
2000	msg += "To save, the application panel should have a data set.."
2001	wx.PostEvent(self, StatusEvent(status=msg))
2002
2003	def _on_save_project(self, event):
2004	"""
2005	save the state of the SasView as *.svs
2006	"""
2007	if self._current_perspective is None:
2008	return
2009	wx.PostEvent(self, StatusEvent(status="Saving Project file..."))
2010	path = None
2011	extension = '*' + APPLICATION_STATE_EXTENSION
2012	dlg = wx.FileDialog(self, "Save Project file",
2013	self._default_save_location, "sasview_proj",
2014	extension,
2015	wx.SAVE)
2016	if dlg.ShowModal() == wx.ID_OK:
2017	path = dlg.GetPath()
2018	self._default_save_location = os.path.dirname(path)
2019	else:
2020	return None
2021	dlg.Destroy()
2022	try:
2023	if path is None:
2024	return
2025	# default cansas xml doc
2026	doc = None
2027	for panel in self.panels.values():
2028	temp = panel.save_project(doc)
2029	if temp is not None:
2030	doc = temp
2031
2032	# Write the XML document
2033	extens = APPLICATION_STATE_EXTENSION
2034	fName = os.path.splitext(path)[0] + extens
2035	if doc is not None:
2036	fd = open(fName, 'w')
2037	fd.write(doc.toprettyxml())
2038	fd.close()
2039	wx.PostEvent(self, StatusEvent(status="Completed Saving."))
2040	else:
2041	msg = "Error occurred while saving the project: "
2042	msg += "To save, at least one application panel "
2043	msg += "should have a data set "
2044	msg += "and model selected. "
2045	msg += "No project was saved to %s" % (str(path))
2046	logging.warning(msg)
2047	wx.PostEvent(self, StatusEvent(status=msg, info="error"))
2048	except Exception:
2049	msg = "Error occurred while saving: "
2050	msg += traceback.format_exc()
2051	msg += "To save, at least one application panel "
2052	msg += "should have a data set.."
2053	wx.PostEvent(self, StatusEvent(status=msg, info="error"))
2054
2055	def on_save_helper(self, doc, reader, panel, path):
2056	"""
2057	Save state into a file
2058	"""
2059	if reader is not None:
2060	# case of a panel with multi-pages
2061	if hasattr(panel, "opened_pages"):
2062	for _, page in panel.opened_pages.iteritems():
2063	data = page.get_data()
2064	# state must be cloned
2065	state = page.get_state().clone()
2066	if data is not None:
2067	new_doc = reader.write_toXML(data, state)
2068	if doc is not None and hasattr(doc, "firstChild"):
2069	child = new_doc.firstChild.firstChild
2070	doc.firstChild.appendChild(child)
2071	else:
2072	doc = new_doc
2073	# case of only a panel
2074	else:
2075	data = panel.get_data()
2076	state = panel.get_state()
2077	if data is not None:
2078	new_doc = reader.write_toXML(data, state)
2079	if doc is not None and hasattr(doc, "firstChild"):
2080	child = new_doc.firstChild.firstChild
2081	doc.firstChild.appendChild(child)
2082	else:
2083	doc = new_doc
2084	return doc
2085
2086	def quit_guiframe(self):
2087	"""
2088	Pop up message to make sure the user wants to quit the application
2089	"""
2090	message = "\nDo you really want to exit this application? \n\n"
2091	dial = wx.MessageDialog(self, message, 'Confirm Exit',
2092	wx.YES_NO \| wx.YES_DEFAULT \| wx.ICON_QUESTION)
2093	if dial.ShowModal() == wx.ID_YES:
2094	return True
2095	else:
2096	return False
2097
2098	def WindowClose(self, event=None):
2099	"""
2100	Quit the application from x icon
2101	"""
2102	flag = self.quit_guiframe()
2103	if flag:
2104	_pylab_helpers.Gcf.figs = {}
2105	self.Close()
2106
2107	def Close(self, event=None):
2108	"""
2109	Quit the application
2110	"""
2111	#IF SAS_OPENCL is set, settings are stored in the custom config file
2112	self._write_opencl_config_file()
2113	logging.info(" --- SasView session was closed --- \n")
2114	wx.Exit()
2115	sys.exit()
2116
2117	def _write_opencl_config_file(self):
2118	#from sas.sasgui.guiframe.customdir import SetupCustom
2119	#from sas.sasgui.guiframe.gui_manager import _find_local_config
2120
2121	sas_opencl = os.environ.get("SAS_OPENCL",None)
2122	#How to store it in file
2123	new_config_lines = []
2124	config_file = open(custom_config.__file__)
2125	if custom_config is not None:
2126	config_lines = config_file.readlines()
2127	for line in config_lines:
2128	if "SAS_OPENCL" in line:
2129	if sas_opencl:
2130	new_config_lines.append("SAS_OPENCL = \""+sas_opencl+"\"")
2131	else:
2132	new_config_lines.append("SAS_OPENCL = None")
2133	else:
2134	new_config_lines.append(line)
2135	config_file.close()
2136	new_config_file = open(custom_config.__file__,"w")
2137	new_config_file.writelines(new_config_lines)
2138	new_config_file.close()
2139
2140
2141	def _check_update(self, event=None):
2142	"""
2143	Check with the deployment server whether a new version
2144	of the application is available.
2145	A thread is started for the connecting with the server. The thread calls
2146	a call-back method when the current version number has been obtained.
2147	"""
2148	version_info = {"version": "0.0.0"}
2149	c = Connection(config.__update_URL__, config.UPDATE_TIMEOUT)
2150	response = c.connect()
2151	if response is not None:
2152	try:
2153	#
2154	content = response.read().strip()
2155	logging.info("Connected to www.sasview.org. Latest version: %s"
2156	% (content))
2157	version_info = json.loads(content)
2158	except:
2159	logging.info("Failed to connect to www.sasview.org")
2160	self._process_version(version_info, standalone=event is None)
2161
2162	def _process_version(self, version_info, standalone=True):
2163	"""
2164	Call-back method for the process of checking for updates.
2165	This methods is called by a VersionThread object once the current
2166	version number has been obtained. If the check is being done in the
2167	background, the user will not be notified unless there's an update.
2168
2169	:param version: version string
2170	:param standalone: True of the update is being checked in
2171	the background, False otherwise.
2172
2173	"""
2174	try:
2175	version = version_info["version"]
2176	if version == "0.0.0":
2177	msg = "Could not connect to the application server."
2178	msg += " Please try again later."
2179	self.SetStatusText(msg)
2180	elif cmp(version, config.__version__) > 0:
2181	msg = "Version %s is available! " % str(version)
2182	if not standalone:
2183	import webbrowser
2184	if "download_url" in version_info:
2185	webbrowser.open(version_info["download_url"])
2186	else:
2187	webbrowser.open(config.__download_page__)
2188	else:
2189	msg += "See the help menu to download it."
2190	self.SetStatusText(msg)
2191	else:
2192	if not standalone:
2193	msg = "You have the latest version"
2194	msg += " of %s" % str(config.__appname__)
2195	self.SetStatusText(msg)
2196	except:
2197	msg = "guiframe: could not get latest application"
2198	msg += " version number\n %s" % sys.exc_value
2199	logging.error(msg)
2200	if not standalone:
2201	msg = "Could not connect to the application server."
2202	msg += " Please try again later."
2203	self.SetStatusText(msg)
2204
2205	def _onAcknowledge(self, evt):
2206	"""
2207	Pop up the acknowledge dialog
2208
2209	:param evt: menu event
2210
2211	"""
2212	if config._do_acknowledge:
2213	import sas.sasgui.guiframe.acknowledgebox as AcknowledgeBox
2214	dialog = AcknowledgeBox.DialogAcknowledge(None, -1, "")
2215	dialog.ShowModal()
2216
2217	def _onAbout(self, evt):
2218	"""
2219	Pop up the about dialog
2220
2221	:param evt: menu event
2222
2223	"""
2224	if config._do_aboutbox:
2225	import sas.sasgui.guiframe.aboutbox as AboutBox
2226	dialog = AboutBox.DialogAbout(None, -1, "")
2227	dialog.ShowModal()
2228
2229	def _onTutorial(self, evt):
2230	"""
2231	Pop up the tutorial dialog
2232
2233	:param evt: menu event
2234
2235	"""
2236	if config._do_tutorial:
2237	path = config.TUTORIAL_PATH
2238	if IS_WIN:
2239	try:
2240	from sas.sasgui.guiframe.pdfview import PDFFrame
2241	dialog = PDFFrame(None, -1, "Tutorial", path)
2242	# put icon
2243	self.put_icon(dialog)
2244	dialog.Show(True)
2245	except:
2246	logging.error("Error in _onTutorial: %s" % sys.exc_value)
2247	try:
2248	# Try an alternate method
2249	logging.error(
2250	"Could not open the tutorial pdf, trying xhtml2pdf")
2251	from xhtml2pdf import pisa
2252	pisa.startViewer(path)
2253	except:
2254	logging.error(
2255	"Could not open the tutorial pdf with xhtml2pdf")
2256	msg = "This feature requires 'PDF Viewer'\n"
2257	wx.MessageBox(msg, 'Error')
2258	else:
2259	try:
2260	command = "open '%s'" % path
2261	os.system(command)
2262	except:
2263	try:
2264	# Try an alternate method
2265	logging.error(
2266	"Could not open the tutorial pdf, trying xhtml2pdf")
2267	from xhtml2pdf import pisa
2268	pisa.startViewer(path)
2269	except:
2270	logging.error(
2271	"Could not open the tutorial pdf with xhtml2pdf")
2272	msg = "This feature requires the Preview application\n"
2273	wx.MessageBox(msg, 'Error')
2274
2275	def _onSphinxDocs(self, evt):
2276	"""
2277	Bring up Sphinx Documentation at top level whenever the menu item
2278	'documentation' is clicked. Calls DocumentationWindow with the top
2279	level path of "index.html"
2280
2281	:param evt: menu event
2282	"""
2283	# Running SasView "in-place" using run.py means the docs will be in a
2284	# different place than they would otherwise.
2285	from documentation_window import DocumentationWindow
2286	_TreeLocation = "user/user.html"
2287	DocumentationWindow(self, -1, _TreeLocation, "",
2288	"SasView Documentation")
2289
2290	def set_manager(self, manager):
2291	"""
2292	Sets the application manager for this frame
2293
2294	:param manager: frame manager
2295	"""
2296	self.app_manager = manager
2297
2298	def post_init(self):
2299	"""
2300	This initialization method is called after the GUI
2301	has been created and all plug-ins loaded. It calls
2302	the post_init() method of each plug-in (if it exists)
2303	so that final initialization can be done.
2304	"""
2305	for item in self.plugins:
2306	if hasattr(item, "post_init"):
2307	item.post_init()
2308
2309	def set_perspective(self, panels):
2310	"""
2311	Sets the perspective of the GUI.
2312	Opens all the panels in the list, and closes
2313	all the others.
2314
2315	:param panels: list of panels
2316	"""
2317	for item in self.panels.keys():
2318	# Check whether this is a sticky panel
2319	if hasattr(self.panels[item], "ALWAYS_ON"):
2320	if self.panels[item].ALWAYS_ON:
2321	continue
2322	if self.panels[item] is None:
2323	continue
2324	if self.panels[item].window_name in panels:
2325	frame = self.panels[item].get_frame()
2326	if not frame.IsShown():
2327	frame.Show(True)
2328	else:
2329	# always show the data panel if enable
2330	style = self.__gui_style & GUIFRAME.MANAGER_ON
2331	if (style == GUIFRAME.MANAGER_ON) \
2332	and self.panels[item] == self._data_panel:
2333	if 'data_panel' in self.panels.keys():
2334	frame = self.panels['data_panel'].get_frame()
2335	if frame is None:
2336	continue
2337	flag = frame.IsShown()
2338	frame.Show(flag)
2339	else:
2340	frame = self.panels[item].get_frame()
2341	if frame is None:
2342	continue
2343
2344	if frame.IsShown():
2345	frame.Show(False)
2346
2347	def show_data_panel(self, event=None, action=True):
2348	"""
2349	show the data panel
2350	"""
2351	if self._data_panel_menu is None:
2352	return
2353	label = self._data_panel_menu.GetText()
2354	pane = self.panels["data_panel"]
2355	frame = pane.get_frame()
2356	if label == 'Show Data Explorer':
2357	if action:
2358	frame.Show(True)
2359	self.__gui_style = self.__gui_style \| GUIFRAME.MANAGER_ON
2360	self._data_panel_menu.SetText('Hide Data Explorer')
2361	else:
2362	if action:
2363	frame.Show(False)
2364	self.__gui_style = self.__gui_style & (~GUIFRAME.MANAGER_ON)
2365	self._data_panel_menu.SetText('Show Data Explorer')
2366
2367	def add_data_helper(self, data_list):
2368	"""
2369	"""
2370	if self._data_manager is not None:
2371	self._data_manager.add_data(data_list)
2372
2373	def add_data(self, data_list):
2374	"""
2375	receive a dictionary of data from loader
2376	store them its data manager if possible
2377	send to data the current active perspective if the data panel
2378	is not active.
2379	:param data_list: dictionary of data's ID and value Data
2380	"""
2381	# Store data into manager
2382	self.add_data_helper(data_list)
2383	# set data in the data panel
2384	if self._data_panel is not None:
2385	data_state = self._data_manager.get_data_state(data_list.keys())
2386	self._data_panel.load_data_list(data_state)
2387	# if the data panel is shown wait for the user to press a button
2388	# to send data to the current perspective. if the panel is not
2389	# show automatically send the data to the current perspective
2390	style = self.__gui_style & GUIFRAME.MANAGER_ON
2391	if style == GUIFRAME.MANAGER_ON:
2392	# wait for button press from the data panel to set_data
2393	if self._data_panel is not None:
2394	self._data_panel.frame.Show(True)
2395	else:
2396	# automatically send that to the current perspective
2397	self.set_data(data_id=data_list.keys())
2398
2399	def set_data(self, data_id, theory_id=None):
2400	"""
2401	set data to current perspective
2402	"""
2403	list_data, _ = self._data_manager.get_by_id(data_id)
2404	if self._current_perspective is not None:
2405	self._current_perspective.set_data(list_data.values())
2406
2407	else:
2408	msg = "Guiframe does not have a current perspective"
2409	logging.info(msg)
2410
2411	def set_theory(self, state_id, theory_id=None):
2412	"""
2413	"""
2414	_, list_theory = self._data_manager.get_by_id(theory_id)
2415	if self._current_perspective is not None:
2416	try:
2417	self._current_perspective.set_theory(list_theory.values())
2418	except:
2419	msg = "Guiframe set_theory: \n" + str(sys.exc_value)
2420	logging.info(msg)
2421	wx.PostEvent(self, StatusEvent(status=msg, info="error"))
2422	else:
2423	msg = "Guiframe does not have a current perspective"
2424	logging.info(msg)
2425
2426	def plot_data(self, state_id, data_id=None,
2427	theory_id=None, append=False):
2428	"""
2429	send a list of data to plot
2430	"""
2431	data_list, _ = self._data_manager.get_by_id(data_id)
2432	_, temp_list_theory = self._data_manager.get_by_id(theory_id)
2433	total_plot_list = data_list.values()
2434	for item in temp_list_theory.values():
2435	theory_data, theory_state = item
2436	total_plot_list.append(theory_data)
2437	GROUP_ID = wx.NewId()
2438	for new_plot in total_plot_list:
2439	if append:
2440	if self.panel_on_focus is None:
2441	message = "cannot append plot. No plot panel on focus!"
2442	message += "please click on any available plot to set focus"
2443	wx.PostEvent(self, StatusEvent(status=message,
2444	info='warning'))
2445	return
2446	else:
2447	if self.enable_add_data(new_plot):
2448	new_plot.group_id = self.panel_on_focus.group_id
2449	else:
2450	message = "Only 1D Data can be append to"
2451	message += " plot panel containing 1D data.\n"
2452	message += "%s not be appended.\n" % str(new_plot.name)
2453	message += "try new plot option.\n"
2454	wx.PostEvent(self, StatusEvent(status=message,
2455	info='warning'))
2456	else:
2457	# if not append then new plot
2458	from sas.sasgui.guiframe.dataFitting import Data2D
2459	if issubclass(Data2D, new_plot.__class__):
2460	# for 2 D always plot in a separated new plot
2461	new_plot.group_id = wx.NewId()
2462	else:
2463	# plot all 1D in a new plot
2464	new_plot.group_id = GROUP_ID
2465	title = "PLOT " + str(new_plot.title)
2466	wx.PostEvent(self, NewPlotEvent(plot=new_plot,
2467	title=title,
2468	group_id=new_plot.group_id))
2469
2470	def remove_data(self, data_id, theory_id=None):
2471	"""
2472	Delete data state if data_id is provide
2473	delete theory created with data of id data_id if theory_id is provide
2474	if delete all true: delete the all state
2475	else delete theory
2476	"""
2477	temp = data_id + theory_id
2478	for plug in self.plugins:
2479	plug.delete_data(temp)
2480	data_list, _ = self._data_manager.get_by_id(data_id)
2481	_, temp_list_theory = self._data_manager.get_by_id(theory_id)
2482	total_plot_list = data_list.values()
2483	for item in temp_list_theory.values():
2484	theory_data, theory_state = item
2485	total_plot_list.append(theory_data)
2486	for new_plot in total_plot_list:
2487	for group_id in new_plot.list_group_id:
2488	wx.PostEvent(self, NewPlotEvent(id=new_plot.id,
2489	group_id=group_id,
2490	action='remove'))
2491	wx.CallAfter(self._remove_res_plot, new_plot.id)
2492	self._data_manager.delete_data(data_id=data_id,
2493	theory_id=theory_id)
2494
2495	def _remove_res_plot(self, id):
2496	"""
2497	Try to remove corresponding res plot
2498
2499	: param id: id of the data
2500	"""
2501	try:
2502	wx.PostEvent(self, NewPlotEvent(id=("res" + str(id)),
2503	group_id=("res" + str(id)),
2504	action='remove'))
2505	except:
2506	logging.error(sys.exc_value)
2507
2508	def save_data1d(self, data, fname):
2509	"""
2510	Save data dialog
2511	"""
2512	default_name = fname
2513	wildcard = "Text files (.txt)\|.txt\|"\
2514	"CanSAS 1D files(.xml)\|.xml"
2515	path = None
2516	dlg = wx.FileDialog(self, "Choose a file",
2517	self._default_save_location,
2518	default_name, wildcard, wx.SAVE)
2519
2520	if dlg.ShowModal() == wx.ID_OK:
2521	path = dlg.GetPath()
2522	# ext_num = 0 for .txt, ext_num = 1 for .xml
2523	# This is MAC Fix
2524	ext_num = dlg.GetFilterIndex()
2525	if ext_num == 0:
2526	ext_format = '.txt'
2527	else:
2528	ext_format = '.xml'
2529	path = os.path.splitext(path)[0] + ext_format
2530	mypath = os.path.basename(path)
2531
2532	# Instantiate a loader
2533	loader = Loader()
2534	ext_format = ".txt"
2535	if os.path.splitext(mypath)[1].lower() == ext_format:
2536	# Make sure the ext included in the file name
2537	# especially on MAC
2538	fName = os.path.splitext(path)[0] + ext_format
2539	self._onsaveTXT(data, fName)
2540	ext_format = ".xml"
2541	if os.path.splitext(mypath)[1].lower() == ext_format:
2542	# Make sure the ext included in the file name
2543	# especially on MAC
2544	fName = os.path.splitext(path)[0] + ext_format
2545	loader.save(fName, data, ext_format)
2546	try:
2547	self._default_save_location = os.path.dirname(path)
2548	except:
2549	pass
2550	dlg.Destroy()
2551
2552	def _onsaveTXT(self, data, path):
2553	"""
2554	Save file as txt
2555
2556	.. todo:: Refactor and remove this method. See 'TODO' in _onSave.
2557	"""
2558	if path is not None:
2559	out = open(path, 'w')
2560	has_errors = True
2561	if data.dy is None or data.dy == []:
2562	has_errors = False
2563	# Sanity check
2564	if has_errors:
2565	try:
2566	if len(data.y) != len(data.dy):
2567	has_errors = False
2568	except:
2569	has_errors = False
2570	if has_errors:
2571	if data.dx is not None and data.dx != []:
2572	out.write("<X> <Y> <dY> <dX>\n")
2573	else:
2574	out.write("<X> <Y> <dY>\n")
2575	else:
2576	out.write("<X> <Y>\n")
2577
2578	for i in range(len(data.x)):
2579	if has_errors:
2580	if data.dx is not None and data.dx != []:
2581	if data.dx[i] is not None:
2582	out.write("%g %g %g %g\n" % (data.x[i],
2583	data.y[i],
2584	data.dy[i],
2585	data.dx[i]))
2586	else:
2587	out.write("%g %g %g\n" % (data.x[i],
2588	data.y[i],
2589	data.dy[i]))
2590	else:
2591	out.write("%g %g %g\n" % (data.x[i],
2592	data.y[i],
2593	data.dy[i]))
2594	else:
2595	out.write("%g %g\n" % (data.x[i],
2596	data.y[i]))
2597	out.close()
2598
2599	def show_data1d(self, data, name):
2600	"""
2601	Show data dialog
2602	"""
2603	try:
2604	xmin = min(data.x)
2605	ymin = min(data.y)
2606	except:
2607	msg = "Unable to find min/max of \n data named %s" % \
2608	data.filename
2609	wx.PostEvent(self, StatusEvent(status=msg,
2610	info="error"))
2611	raise ValueError, msg
2612	# text = str(data)
2613	text = data.__str__()
2614	text += 'Data Min Max:\n'
2615	text += 'X_min = %s: X_max = %s\n' % (xmin, max(data.x))
2616	text += 'Y_min = %s: Y_max = %s\n' % (ymin, max(data.y))
2617	if data.dy is not None:
2618	text += 'dY_min = %s: dY_max = %s\n' % (min(data.dy), max(data.dy))
2619	text += '\nData Points:\n'
2620	x_st = "X"
2621	for index in range(len(data.x)):
2622	if data.dy is not None and len(data.dy) > index:
2623	dy_val = data.dy[index]
2624	else:
2625	dy_val = 0.0
2626	if data.dx is not None and len(data.dx) > index:
2627	dx_val = data.dx[index]
2628	else:
2629	dx_val = 0.0
2630	if data.dxl is not None and len(data.dxl) > index:
2631	if index == 0:
2632	x_st = "Xl"
2633	dx_val = data.dxl[index]
2634	elif data.dxw is not None and len(data.dxw) > index:
2635	if index == 0:
2636	x_st = "Xw"
2637	dx_val = data.dxw[index]
2638
2639	if index == 0:
2640	text += "<index> \t<X> \t<Y> \t<dY> \t<d%s>\n" % x_st
2641	text += "%s \t%s \t%s \t%s \t%s\n" % (index,
2642	data.x[index],
2643	data.y[index],
2644	dy_val,
2645	dx_val)
2646	from pdfview import TextFrame
2647	frame = TextFrame(None, -1, "Data Info: %s" % data.name, text)
2648	# put icon
2649	self.put_icon(frame)
2650	frame.Show(True)
2651
2652	def save_data2d(self, data, fname):
2653	"""
2654	Save data2d dialog
2655	"""
2656	default_name = fname
2657	wildcard = "IGOR/DAT 2D file in Q_map (.dat)\|.DAT"
2658	dlg = wx.FileDialog(self, "Choose a file",
2659	self._default_save_location,
2660	default_name, wildcard, wx.SAVE)
2661
2662	if dlg.ShowModal() == wx.ID_OK:
2663	path = dlg.GetPath()
2664	# ext_num = 0 for .txt, ext_num = 1 for .xml
2665	# This is MAC Fix
2666	ext_num = dlg.GetFilterIndex()
2667	if ext_num == 0:
2668	ext_format = '.dat'
2669	else:
2670	ext_format = ''
2671	path = os.path.splitext(path)[0] + ext_format
2672	mypath = os.path.basename(path)
2673
2674	# Instantiate a loader
2675	loader = Loader()
2676
2677	ext_format = ".dat"
2678	if os.path.splitext(mypath)[1].lower() == ext_format:
2679	# Make sure the ext included in the file name
2680	# especially on MAC
2681	fileName = os.path.splitext(path)[0] + ext_format
2682	loader.save(fileName, data, ext_format)
2683	try:
2684	self._default_save_location = os.path.dirname(path)
2685	except:
2686	pass
2687	dlg.Destroy()
2688
2689	def show_data2d(self, data, name):
2690	"""
2691	Show data dialog
2692	"""
2693	wx.PostEvent(self, StatusEvent(status="Gathering Data2D Info.",
2694	type='start'))
2695	text = data.__str__()
2696	text += 'Data Min Max:\n'
2697	text += 'I_min = %s\n' % min(data.data)
2698	text += 'I_max = %s\n\n' % max(data.data)
2699	text += 'Data (First 2501) Points:\n'
2700	text += 'Data columns include err(I).\n'
2701	text += 'ASCII data starts here.\n'
2702	text += "<index> \t<Qx> \t<Qy> \t<I> \t<dI> \t<dQparal> \t<dQperp>\n"
2703	di_val = 0.0
2704	dx_val = 0.0
2705	dy_val = 0.0
2706	len_data = len(data.qx_data)
2707	for index in xrange(0, len_data):
2708	x_val = data.qx_data[index]
2709	y_val = data.qy_data[index]
2710	i_val = data.data[index]
2711	if data.err_data is not None:
2712	di_val = data.err_data[index]
2713	if data.dqx_data is not None:
2714	dx_val = data.dqx_data[index]
2715	if data.dqy_data is not None:
2716	dy_val = data.dqy_data[index]
2717
2718	text += "%s \t%s \t%s \t%s \t%s \t%s \t%s\n" % (index,
2719	x_val,
2720	y_val,
2721	i_val,
2722	di_val,
2723	dx_val,
2724	dy_val)
2725	# Takes too long time for typical data2d: Break here
2726	if index >= 2500:
2727	text += ".............\n"
2728	break
2729
2730	from pdfview import TextFrame
2731	frame = TextFrame(None, -1, "Data Info: %s" % data.name, text)
2732	# put icon
2733	self.put_icon(frame)
2734	frame.Show(True)
2735	wx.PostEvent(self, StatusEvent(status="Data2D Info Displayed",
2736	type='stop'))
2737
2738	def set_current_perspective(self, perspective):
2739	"""
2740	set the current active perspective
2741	"""
2742	self._current_perspective = perspective
2743	name = "No current analysis selected"
2744	if self._current_perspective is not None:
2745	self._add_current_plugin_menu()
2746	for panel in self.panels.values():
2747	if hasattr(panel, 'CENTER_PANE') and panel.CENTER_PANE:
2748	for name in self._current_perspective.get_perspective():
2749	frame = panel.get_frame()
2750	if frame is not None:
2751	if name == panel.window_name:
2752	panel.on_set_focus(event=None)
2753	frame.Show(True)
2754	else:
2755	frame.Show(False)
2756	name = self._current_perspective.sub_menu
2757	if self._data_panel is not None:
2758	self._data_panel.set_active_perspective(name)
2759	self._check_applications_menu()
2760	# Set the SasView title
2761	self._set_title_name(name)
2762
2763	def _set_title_name(self, name):
2764	"""
2765	Set the SasView title w/ the current application name
2766
2767	: param name: application name [string]
2768	"""
2769	# Set SanView Window title w/ application anme
2770	title = self.title + " - " + name + " -"
2771	self.SetTitle(title)
2772
2773	def _check_applications_menu(self):
2774	"""
2775	check the menu of the current application
2776	"""
2777	if self._applications_menu is not None:
2778	for menu in self._applications_menu.GetMenuItems():
2779	if self._current_perspective is not None:
2780	name = self._current_perspective.sub_menu
2781	if menu.IsCheckable():
2782	if menu.GetLabel() == name:
2783	menu.Check(True)
2784	else:
2785	menu.Check(False)
2786
2787	def enable_add_data(self, new_plot):
2788	"""
2789	Enable append data on a plot panel
2790	"""
2791
2792	if self.panel_on_focus \
2793	not in self._plotting_plugin.plot_panels.values():
2794	return
2795	check = "Theory1D"
2796	is_theory = len(self.panel_on_focus.plots) <= 1 and \
2797	self.panel_on_focus.plots.values()[0].__class__.__name__ == check
2798
2799	is_data2d = hasattr(new_plot, 'data')
2800
2801	is_data1d = self.panel_on_focus.__class__.__name__ == "ModelPanel1D"\
2802	and self.panel_on_focus.group_id is not None
2803	has_meta_data = hasattr(new_plot, 'meta_data')
2804
2805	# disable_add_data if the data is being recovered from a saved state
2806	is_state_data = False
2807	if has_meta_data:
2808	if 'invstate' in new_plot.meta_data:
2809	is_state_data = True
2810	if 'prstate' in new_plot.meta_data:
2811	is_state_data = True
2812	if 'fitstate' in new_plot.meta_data:
2813	is_state_data = True
2814
2815	return is_data1d and not is_data2d and not is_theory \
2816	and not is_state_data
2817
2818	def check_multimode(self, perspective=None):
2819	"""
2820	Check the perspective have batch mode capablitity
2821	"""
2822	if perspective is None or self._data_panel is None:
2823	return
2824	flag = perspective.get_batch_capable()
2825	flag_on = perspective.batch_on
2826	if flag:
2827	self._data_panel.rb_single_mode.SetValue(not flag_on)
2828	self._data_panel.rb_batch_mode.SetValue(flag_on)
2829	else:
2830	self._data_panel.rb_single_mode.SetValue(True)
2831	self._data_panel.rb_batch_mode.SetValue(False)
2832	self._data_panel.rb_single_mode.Enable(flag)
2833	self._data_panel.rb_batch_mode.Enable(flag)
2834
2835	def enable_edit_menu(self):
2836	"""
2837	enable menu item under edit menu depending on the panel on focus
2838	"""
2839	if self.cpanel_on_focus is not None and self._edit_menu is not None:
2840	flag = self.cpanel_on_focus.get_undo_flag()
2841	self._edit_menu.Enable(GUIFRAME_ID.UNDO_ID, flag)
2842	flag = self.cpanel_on_focus.get_redo_flag()
2843	self._edit_menu.Enable(GUIFRAME_ID.REDO_ID, flag)
2844	flag = self.cpanel_on_focus.get_copy_flag()
2845	self._edit_menu.Enable(GUIFRAME_ID.COPY_ID, flag)
2846	flag = self.cpanel_on_focus.get_paste_flag()
2847	self._edit_menu.Enable(GUIFRAME_ID.PASTE_ID, flag)
2848
2849	# Copy menu
2850	flag = self.cpanel_on_focus.get_copy_flag()
2851	self._edit_menu_copyas.Enable(GUIFRAME_ID.COPYEX_ID, flag)
2852	self._edit_menu_copyas.Enable(GUIFRAME_ID.COPYLAT_ID, flag)
2853
2854	flag = self.cpanel_on_focus.get_preview_flag()
2855	self._edit_menu.Enable(GUIFRAME_ID.PREVIEW_ID, flag)
2856	flag = self.cpanel_on_focus.get_reset_flag()
2857	self._edit_menu.Enable(GUIFRAME_ID.RESET_ID, flag)
2858	else:
2859	flag = False
2860	self._edit_menu.Enable(GUIFRAME_ID.UNDO_ID, flag)
2861	self._edit_menu.Enable(GUIFRAME_ID.REDO_ID, flag)
2862	self._edit_menu.Enable(GUIFRAME_ID.COPY_ID, flag)
2863	self._edit_menu.Enable(GUIFRAME_ID.PASTE_ID, flag)
2864	self._edit_menu.Enable(GUIFRAME_ID.PREVIEW_ID, flag)
2865	self._edit_menu.Enable(GUIFRAME_ID.RESET_ID, flag)
2866
2867	def on_undo_panel(self, event=None):
2868	"""
2869	undo previous action of the last panel on focus if possible
2870	"""
2871	if self.cpanel_on_focus is not None:
2872	self.cpanel_on_focus.on_undo(event)
2873
2874	def on_redo_panel(self, event=None):
2875	"""
2876	redo the last cancel action done on the last panel on focus
2877	"""
2878	if self.cpanel_on_focus is not None:
2879	self.cpanel_on_focus.on_redo(event)
2880
2881	def on_copy_panel(self, event=None):
2882	"""
2883	copy the last panel on focus if possible
2884	"""
2885	if self.cpanel_on_focus is not None:
2886	self.cpanel_on_focus.on_copy(event)
2887
2888	def on_paste_panel(self, event=None):
2889	"""
2890	paste clipboard to the last panel on focus
2891	"""
2892	if self.cpanel_on_focus is not None:
2893	self.cpanel_on_focus.on_paste(event)
2894
2895	def on_bookmark_panel(self, event=None):
2896	"""
2897	bookmark panel
2898	"""
2899	if self.cpanel_on_focus is not None:
2900	self.cpanel_on_focus.on_bookmark(event)
2901
2902	def append_bookmark(self, event=None):
2903	"""
2904	Bookmark available information of the panel on focus
2905	"""
2906	self._toolbar.append_bookmark(event)
2907
2908	def on_save_panel(self, event=None):
2909	"""
2910	save possible information on the current panel
2911	"""
2912	if self.cpanel_on_focus is not None:
2913	self.cpanel_on_focus.on_save(event)
2914
2915	def on_preview_panel(self, event=None):
2916	"""
2917	preview information on the panel on focus
2918	"""
2919	if self.cpanel_on_focus is not None:
2920	self.cpanel_on_focus.on_preview(event)
2921
2922	def on_print_panel(self, event=None):
2923	"""
2924	print available information on the last panel on focus
2925	"""
2926	if self.cpanel_on_focus is not None:
2927	self.cpanel_on_focus.on_print(event)
2928
2929	def on_zoom_panel(self, event=None):
2930	"""
2931	zoom on the current panel if possible
2932	"""
2933	if self.cpanel_on_focus is not None:
2934	self.cpanel_on_focus.on_zoom(event)
2935
2936	def on_zoom_in_panel(self, event=None):
2937	"""
2938	zoom in of the panel on focus
2939	"""
2940	if self.cpanel_on_focus is not None:
2941	self.cpanel_on_focus.on_zoom_in(event)
2942
2943	def on_zoom_out_panel(self, event=None):
2944	"""
2945	zoom out on the panel on focus
2946	"""
2947	if self.cpanel_on_focus is not None:
2948	self.cpanel_on_focus.on_zoom_out(event)
2949
2950	def on_drag_panel(self, event=None):
2951	"""
2952	drag apply to the panel on focus
2953	"""
2954	if self.cpanel_on_focus is not None:
2955	self.cpanel_on_focus.on_drag(event)
2956
2957	def on_reset_panel(self, event=None):
2958	"""
2959	reset the current panel
2960	"""
2961	if self.cpanel_on_focus is not None:
2962	self.cpanel_on_focus.on_reset(event)
2963
2964	def on_change_caption(self, name, old_caption, new_caption):
2965	"""
2966	Change the panel caption
2967
2968	:param name: window_name of the pane
2969	:param old_caption: current caption [string]
2970	:param new_caption: new caption [string]
2971	"""
2972	# wx.aui.AuiPaneInfo
2973	pane_info = self.get_paneinfo(old_caption)
2974	# update the data_panel.cb_plotpanel
2975	if 'data_panel' in self.panels.keys():
2976	# remove from data_panel combobox
2977	data_panel = self.panels["data_panel"]
2978	if data_panel.cb_plotpanel is not None:
2979	# Check if any panel has the same caption
2980	has_newstring = data_panel.cb_plotpanel.FindString(
2981	str(new_caption))
2982	caption = new_caption
2983	if has_newstring != wx.NOT_FOUND:
2984	captions = self._get_plotpanel_captions()
2985	# Append nummber
2986	inc = 1
2987	# FIXME: fix this terrible loop
2988	while (1):
2989	caption = new_caption + '_%s' % str(inc)
2990	if caption not in captions:
2991	break
2992	inc += 1
2993	# notify to users
2994	msg = "Found Same Title: Added '_%s'" % str(inc)
2995	wx.PostEvent(self, StatusEvent(status=msg))
2996	# update data_panel cb
2997	pos = data_panel.cb_plotpanel.FindString(str(old_caption))
2998	if pos != wx.NOT_FOUND:
2999	data_panel.cb_plotpanel.SetString(pos, caption)
3000	data_panel.cb_plotpanel.SetStringSelection(caption)
3001	# New Caption
3002	pane_info.SetTitle(caption)
3003	return caption
3004
3005	def get_paneinfo(self, name):
3006	"""
3007	Get pane Caption from window_name
3008
3009	:param name: window_name in AuiPaneInfo
3010	:returns: AuiPaneInfo of the name
3011	"""
3012	for panel in self.plot_panels.values():
3013	if panel.frame.GetTitle() == name:
3014	return panel.frame
3015	return None
3016
3017	def enable_undo(self):
3018	"""
3019	enable undo related control
3020	"""
3021	if self.cpanel_on_focus is not None:
3022	self._toolbar.enable_undo(self.cpanel_on_focus)
3023
3024	def enable_redo(self):
3025	"""
3026	enable redo
3027	"""
3028	if self.cpanel_on_focus is not None:
3029	self._toolbar.enable_redo(self.cpanel_on_focus)
3030
3031	def enable_copy(self):
3032	"""
3033	enable copy related control
3034	"""
3035	if self.cpanel_on_focus is not None:
3036	self._toolbar.enable_copy(self.cpanel_on_focus)
3037
3038	def enable_paste(self):
3039	"""
3040	enable paste
3041	"""
3042	if self.cpanel_on_focus is not None:
3043	self._toolbar.enable_paste(self.cpanel_on_focus)
3044
3045	def enable_bookmark(self):
3046	"""
3047	Bookmark
3048	"""
3049	if self.cpanel_on_focus is not None:
3050	self._toolbar.enable_bookmark(self.cpanel_on_focus)
3051
3052	def enable_save(self):
3053	"""
3054	save
3055	"""
3056	if self.cpanel_on_focus is not None:
3057	self._toolbar.enable_save(self.cpanel_on_focus)
3058
3059	def enable_preview(self):
3060	"""
3061	preview
3062	"""
3063	if self.cpanel_on_focus is not None:
3064	self._toolbar.enable_preview(self.cpanel_on_focus)
3065
3066	def enable_print(self):
3067	"""
3068	print
3069	"""
3070	if self.cpanel_on_focus is not None:
3071	self._toolbar.enable_print(self.cpanel_on_focus)
3072
3073	def enable_zoom(self):
3074	"""
3075	zoom
3076	"""
3077	if self.cpanel_on_focus is not None:
3078	self._toolbar.enable_zoom(self.panel_on_focus)
3079
3080	def enable_zoom_in(self):
3081	"""
3082	zoom in
3083	"""
3084	if self.cpanel_on_focus is not None:
3085	self._toolbar.enable_zoom_in(self.panel_on_focus)
3086
3087	def enable_zoom_out(self):
3088	"""
3089	zoom out
3090	"""
3091	if self.cpanel_on_focus is not None:
3092	self._toolbar.enable_zoom_out(self.panel_on_focus)
3093
3094	def enable_drag(self, event=None):
3095	"""
3096	drag
3097	"""
3098	# Not implemeted
3099
3100	def enable_reset(self):
3101	"""
3102	reset the current panel
3103	"""
3104	if self.cpanel_on_focus is not None:
3105	self._toolbar.enable_reset(self.panel_on_focus)
3106
3107	def get_toolbar_height(self):
3108	"""
3109	"""
3110	size_y = 0
3111	if self.GetToolBar() is not None and self.GetToolBar().IsShown():
3112	if not IS_LINUX:
3113	_, size_y = self.GetToolBar().GetSizeTuple()
3114	return size_y
3115
3116	def set_schedule_full_draw(self, panel=None, func='del'):
3117	"""
3118	Add/subtract the schedule full draw list with the panel given
3119
3120	:param panel: plot panel
3121	:param func: append or del [string]
3122	"""
3123
3124	# append this panel in the schedule list if not in yet
3125	if func == 'append':
3126	if panel not in self.schedule_full_draw_list:
3127	self.schedule_full_draw_list.append(panel)
3128	# remove this panel from schedule list
3129	elif func == 'del':
3130	if len(self.schedule_full_draw_list) > 0:
3131	if panel in self.schedule_full_draw_list:
3132	self.schedule_full_draw_list.remove(panel)
3133
3134	# reset the schdule
3135	if len(self.schedule_full_draw_list) == 0:
3136	self.schedule = False
3137	else:
3138	self.schedule = True
3139
3140	def full_draw(self):
3141	"""
3142	Draw the panels with axes in the schedule to full dwar list
3143	"""
3144
3145	count = len(self.schedule_full_draw_list)
3146	# if not self.schedule:
3147	if count < 1:
3148	self.set_schedule(False)
3149	return
3150
3151	else:
3152	ind = 0
3153	# if any of the panel is shown do full_draw
3154	for panel in self.schedule_full_draw_list:
3155	ind += 1
3156	if panel.frame.IsShown():
3157	break
3158	# otherwise, return
3159	if ind == count:
3160	return
3161	# Simple redraw only for a panel shown
3162
3163	def f_draw(panel):
3164	"""
3165	Draw A panel in the full draw list
3166	"""
3167	try:
3168	# This checking of GetCapture is to stop redrawing
3169	# while any panel is capture.
3170	frame = panel.frame
3171
3172	if not frame.GetCapture():
3173	# draw if possible
3174	panel.set_resizing(False)
3175	# panel.Show(True)
3176	panel.draw_plot()
3177	# Check if the panel is not shown
3178	flag = frame.IsShown()
3179	frame.Show(flag)
3180	except:
3181	pass
3182
3183	# Draw all panels
3184	if count == 1:
3185	f_draw(self.schedule_full_draw_list[0])
3186	else:
3187	map(f_draw, self.schedule_full_draw_list)
3188	# Reset the attr
3189	if len(self.schedule_full_draw_list) == 0:
3190	self.set_schedule(False)
3191	else:
3192	self.set_schedule(True)
3193
3194	def set_schedule(self, schedule=False):
3195	"""
3196	Set schedule
3197	"""
3198	self.schedule = schedule
3199
3200	def get_schedule(self):
3201	"""
3202	Get schedule
3203	"""
3204	return self.schedule
3205
3206	def on_set_plot_focus(self, panel):
3207	"""
3208	Set focus on a plot panel
3209	"""
3210	if panel is None:
3211	return
3212	# self.set_plot_unfocus()
3213	panel.on_set_focus(None)
3214	# set focusing panel
3215	self.panel_on_focus = panel
3216	self.set_panel_on_focus(None)
3217
3218	def set_plot_unfocus(self):
3219	"""
3220	Un focus all plot panels
3221	"""
3222	for plot in self.plot_panels.values():
3223	plot.on_kill_focus(None)
3224
3225	def get_window_size(self):
3226	"""
3227	Get window size
3228
3229	:returns: size
3230	:rtype: tuple
3231	"""
3232	width, height = self.GetSizeTuple()
3233	if not IS_WIN:
3234	# Subtract toolbar height to get real window side
3235	if self._toolbar.IsShown():
3236	height -= 45
3237	return (width, height)
3238
3239	def _onDrawIdle(self, args, *kwargs):
3240	"""
3241	ReDraw with axes
3242	"""
3243	try:
3244	# check if it is time to redraw
3245	if self.GetCapture() is None:
3246	# Draw plot, changes resizing too
3247	self.full_draw()
3248	except:
3249	pass
3250
3251	# restart idle
3252	self._redraw_idle(args, *kwargs)
3253
3254	def _redraw_idle(self, args, *kwargs):
3255	"""
3256	Restart Idle
3257	"""
3258	# restart idle
3259	self.idletimer.Restart(100 * TIME_FACTOR, args, *kwargs)
3260
3261
3262	class DefaultPanel(wx.Panel, PanelBase):
3263	"""
3264	Defines the API for a panels to work with
3265	the GUI manager
3266	"""
3267	# Internal nickname for the window, used by the AUI manager
3268	window_name = "default"
3269	# Name to appear on the window title bar
3270	window_caption = "Welcome panel"
3271	# Flag to tell the AUI manager to put this panel in the center pane
3272	CENTER_PANE = True
3273
3274	def __init__(self, parent, args, *kwds):
3275	wx.Panel.__init__(self, parent, args, *kwds)
3276	PanelBase.__init__(self, parent)
3277
3278
3279	class SasViewApp(wx.App):
3280	"""
3281	SasView application
3282	"""
3283	def OnInit(self):
3284	"""
3285	When initialised
3286	"""
3287	pos, size, self.is_max = self.window_placement((GUIFRAME_WIDTH,
3288	GUIFRAME_HEIGHT))
3289	self.frame = ViewerFrame(parent=None,
3290	title=APPLICATION_NAME,
3291	pos=pos,
3292	gui_style=DEFAULT_STYLE,
3293	size=size)
3294	self.frame.Hide()
3295	if not IS_WIN:
3296	self.frame.EnableCloseButton(False)
3297	self.s_screen = None
3298
3299	try:
3300	self.open_file()
3301	except:
3302	msg = "%s Could not load " % str(APPLICATION_NAME)
3303	msg += "input file from command line.\n"
3304	logging.error(msg)
3305	# Display a splash screen on top of the frame.
3306	try:
3307	if os.path.isfile(SPLASH_SCREEN_PATH):
3308	self.s_screen = \
3309	self.display_splash_screen(parent=self.frame,
3310	path=SPLASH_SCREEN_PATH)
3311	else:
3312	self.frame.Show()
3313	except:
3314	if self.s_screen is not None:
3315	self.s_screen.Close()
3316	msg = "Cannot display splash screen\n"
3317	msg += str(sys.exc_value)
3318	logging.error(msg)
3319	self.frame.Show()
3320
3321	self.SetTopWindow(self.frame)
3322
3323	return True
3324
3325	def maximize_win(self):
3326	"""
3327	Maximize the window after the frame shown
3328	"""
3329	if self.is_max:
3330	if self.frame.IsShown():
3331	# Max window size
3332	self.frame.Maximize(self.is_max)
3333
3334	def open_file(self):
3335	"""
3336	open a state file at the start of the application
3337	"""
3338	input_file = None
3339	if len(sys.argv) >= 2:
3340	cmd = sys.argv[0].lower()
3341	basename = os.path.basename(cmd)
3342	app_base = str(APPLICATION_NAME).lower()
3343	if os.path.isfile(cmd) or basename.lower() == app_base:
3344	app_py = app_base + '.py'
3345	app_exe = app_base + '.exe'
3346	app_app = app_base + '.app'
3347	if basename.lower() in [app_py, app_exe, app_app, app_base]:
3348	data_base = sys.argv[1]
3349	input_file = os.path.normpath(os.path.join(DATAPATH,
3350	data_base))
3351	if input_file is None:
3352	return
3353	if self.frame is not None:
3354	self.frame.set_input_file(input_file=input_file)
3355
3356	def clean_plugin_models(self, path):
3357	"""
3358	Delete plugin models in app folder
3359
3360	:param path: path of the plugin_models folder in app
3361	"""
3362	# do it only the first time app loaded
3363	# delete unused model folder
3364	model_folder = os.path.join(PATH_APP, path)
3365	if os.path.exists(model_folder) and os.path.isdir(model_folder):
3366	if len(os.listdir(model_folder)) > 0:
3367	try:
3368	for filename in os.listdir(model_folder):
3369	file_path = os.path.join(model_folder, filename)
3370	if os.path.isfile(file_path):
3371	os.remove(file_path)
3372	except:
3373	logging.error("gui_manager.clean_plugin_models:\n %s"
3374	% sys.exc_value)
3375
3376	def set_manager(self, manager):
3377	"""
3378	Sets a reference to the application manager
3379	of the GUI manager (Frame)
3380	"""
3381	self.frame.set_manager(manager)
3382
3383	def build_gui(self):
3384	"""
3385	Build the GUI
3386	"""
3387	# try to load file at the start
3388	self.open_file()
3389	self.frame.build_gui()
3390
3391	def set_welcome_panel(self, panel_class):
3392	"""
3393	Set the welcome panel
3394
3395	:param panel_class: class of the welcome panel to be instantiated
3396
3397	"""
3398	self.frame.welcome_panel_class = panel_class
3399
3400	def add_perspective(self, perspective):
3401	"""
3402	Manually add a perspective to the application GUI
3403	"""
3404	self.frame.add_perspective(perspective)
3405
3406	def window_placement(self, size):
3407	"""
3408	Determines the position and size of the application frame such that it
3409	fits on the user's screen without obstructing (or being obstructed by)
3410	the Windows task bar. The maximum initial size in pixels is bounded by
3411	WIDTH x HEIGHT. For most monitors, the application
3412	will be centered on the screen; for very large monitors it will be
3413	placed on the left side of the screen.
3414	"""
3415	is_maximized = False
3416	# Get size of screen without
3417	for screenCount in range(wx.Display().GetCount()):
3418	screen = wx.Display(screenCount)
3419	if screen.IsPrimary():
3420	displayRect = screen.GetClientArea()
3421	break
3422
3423	posX, posY, displayWidth, displayHeight = displayRect
3424	customWidth, customHeight = size
3425
3426	# If the custom size is default, set 90% of the screen size
3427	if customWidth <= 0 and customHeight <= 0:
3428	if customWidth == 0 and customHeight == 0:
3429	is_maximized = True
3430	customWidth = displayWidth * 0.9
3431	customHeight = displayHeight * 0.9
3432	else:
3433	# If the custom screen is bigger than the
3434	# window screen than make maximum size
3435	if customWidth > displayWidth:
3436	customWidth = displayWidth
3437	if customHeight > displayHeight:
3438	customHeight = displayHeight
3439
3440	# Note that when running Linux and using an Xming (X11) server on a PC
3441	# with a dual monitor configuration, the reported display size may be
3442	# that of both monitors combined with an incorrect display count of 1.
3443	# To avoid displaying this app across both monitors, we check for
3444	# screen 'too big'. If so, we assume a smaller width which means the
3445	# application will be placed towards the left hand side of the screen.
3446
3447	# If dual screen registered as 1 screen. Make width half.
3448	# MAC just follows the default behavior of pos
3449	if IS_WIN:
3450	if displayWidth > (displayHeight * 2):
3451	if customWidth == displayWidth:
3452	customWidth = displayWidth / 2
3453	# and set the position to be the corner of the screen.
3454	posX = 0
3455	posY = 0
3456
3457	# Make the position the middle of the screen. (Not 0,0)
3458	else:
3459	posX = (displayWidth - customWidth) / 2
3460	posY = (displayHeight - customHeight) / 2
3461	# Return the suggested position and size for the application frame.
3462	return (posX, posY), (customWidth, customHeight), is_maximized
3463
3464	def display_splash_screen(self, parent,
3465	path=SPLASH_SCREEN_PATH):
3466	"""Displays the splash screen. It will exactly cover the main frame."""
3467
3468	# Prepare the picture. On a 2GHz intel cpu, this takes about a second.
3469	image = wx.Image(path, wx.BITMAP_TYPE_PNG)
3470	image.Rescale(SPLASH_SCREEN_WIDTH,
3471	SPLASH_SCREEN_HEIGHT, wx.IMAGE_QUALITY_HIGH)
3472	bm = image.ConvertToBitmap()
3473
3474	# Create and show the splash screen. It will disappear only when the
3475	# program has entered the event loop AND either the timeout has expired
3476	# or the user has left clicked on the screen. Thus any processing
3477	# performed in this routine (including sleeping) or processing in the
3478	# calling routine (including doing imports) will prevent the splash
3479	# screen from disappearing.
3480	#
3481	# Note that on Linux, the timeout appears to occur immediately in which
3482	# case the splash screen disappears upon entering the event loop.
3483	s_screen = wx.SplashScreen(bitmap=bm,
3484	splashStyle=(wx.SPLASH_TIMEOUT \|
3485	wx.SPLASH_CENTRE_ON_SCREEN),
3486	style=(wx.SIMPLE_BORDER \|
3487	wx.FRAME_NO_TASKBAR \|
3488	wx.FRAME_FLOAT_ON_PARENT),
3489	milliseconds=SS_MAX_DISPLAY_TIME,
3490	parent=parent,
3491	id=wx.ID_ANY)
3492	from sas.sasgui.guiframe.gui_statusbar import SPageStatusbar
3493	statusBar = SPageStatusbar(s_screen)
3494	s_screen.SetStatusBar(statusBar)
3495	s_screen.Bind(wx.EVT_CLOSE, self.on_close_splash_screen)
3496	s_screen.Show()
3497	return s_screen
3498
3499	def on_close_splash_screen(self, event):
3500	"""
3501	When the splash screen is closed.
3502	"""
3503	self.frame.Show(True)
3504	event.Skip()
3505	self.maximize_win()
3506
3507
3508	class MDIFrame(CHILD_FRAME):
3509	"""
3510	Frame for panels
3511	"""
3512	def __init__(self, parent, panel, title="Untitled", size=(300, 200)):
3513	"""
3514	comment
3515	:param parent: parent panel/container
3516	"""
3517	# Initialize the Frame object
3518	CHILD_FRAME.__init__(self, parent=parent, id=wx.ID_ANY,
3519	title=title, size=size)
3520	self.parent = parent
3521	self.name = "Untitled"
3522	self.batch_on = self.parent.batch_on
3523	self.panel = panel
3524	if panel is not None:
3525	self.set_panel(panel)
3526	self.Show(False)
3527
3528	def show_data_panel(self, action):
3529	"""
3530	Turns on the data panel
3531
3532	The the data panel is optional. Most of its functions can be
3533	performed from the menu bar and from the plots.
3534	"""
3535	self.parent.show_data_panel(action)
3536
3537	def set_panel(self, panel):
3538	"""
3539	"""
3540	self.panel = panel
3541	self.name = panel.window_name
3542	self.SetTitle(panel.window_caption)
3543	self.SetHelpText(panel.help_string)
3544	width, height = self.parent._get_panels_size(panel)
3545	if hasattr(panel, "CENTER_PANE") and panel.CENTER_PANE:
3546	width *= 0.6
3547	self.SetSize((width, height))
3548	self.parent.put_icon(self)
3549	self.Bind(wx.EVT_SET_FOCUS, self.set_panel_focus)
3550	self.Bind(wx.EVT_CLOSE, self.OnClose)
3551	self.Show(False)
3552
3553	def set_panel_focus(self, event):
3554	"""
3555	"""
3556	if self.parent.panel_on_focus != self.panel:
3557	self.panel.SetFocus()
3558	self.parent.panel_on_focus = self.panel
3559
3560	def OnClose(self, event):
3561	"""
3562	On Close event
3563	"""
3564	self.panel.on_close(event)
3565
3566	if __name__ == "__main__":
3567	app = SasViewApp(0)
3568	app.MainLoop()

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

SasView

source: sasview/src/sas/sasgui/guiframe/gui_manager.py @ e2c0939

Download in other formats: