gui_manager.py @ f89fa84

ESS_GUIESS_GUI_DocsESS_GUI_batch_fittingESS_GUI_bumps_abstractionESS_GUI_iss1116ESS_GUI_iss879ESS_GUI_iss959ESS_GUI_openclESS_GUI_orderingESS_GUI_sync_sascalccostrafo411magnetic_scattrelease-4.1.1release-4.1.2release-4.2.2release_4.0.1ticket-1009ticket-1094-headlessticket-1242-2d-resolutionticket-1243ticket-1249ticket885unittest-saveload

Last change on this file since f89fa84 was 064723b, checked in by krzywon, 10 years ago
Found one small issue with the last commit.
Property mode set to `100644`
File size: 127.1 KB

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

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

SasView

source: sasview/src/sas/guiframe/gui_manager.py @ f89fa84

Download in other formats: