224 lines
9.3 KiB
Python
224 lines
9.3 KiB
Python
#***************************************************************************
|
|
#* Copyright (c) 2002,2003 Jürgen Riegel <juergen.riegel@web.de> *
|
|
#* *
|
|
#* This file is part of the FreeCAD CAx development system. *
|
|
#* *
|
|
#* This program is free software; you can redistribute it and/or modify *
|
|
#* it under the terms of the GNU Lesser General Public License (LGPL) *
|
|
#* as published by the Free Software Foundation; either version 2 of *
|
|
#* the License, or (at your option) any later version. *
|
|
#* for detail see the LICENCE text file. *
|
|
#* *
|
|
#* FreeCAD is distributed in the hope that it will be useful, *
|
|
#* but WITHOUT ANY WARRANTY; without even the implied warranty of *
|
|
#* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
|
|
#* GNU Lesser General Public License for more details. *
|
|
#* *
|
|
#* You should have received a copy of the GNU Library General Public *
|
|
#* License along with FreeCAD; if not, write to the Free Software *
|
|
#* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 *
|
|
#* USA *
|
|
#* *
|
|
#***************************************************************************/
|
|
|
|
# FreeCAD gui init module
|
|
#
|
|
# Gathering all the information to start FreeCAD
|
|
# This is the second one of three init scripts, the third one
|
|
# runs when the gui is up
|
|
|
|
# imports the one and only
|
|
import FreeCAD, FreeCADGui
|
|
|
|
# shortcuts
|
|
Gui = FreeCADGui
|
|
|
|
# this is to keep old code working
|
|
Gui.listCommands = Gui.Command.listAll
|
|
Gui.isCommandActive = lambda cmd: Gui.Command.get(cmd).isActive()
|
|
|
|
# Important definitions
|
|
class Workbench:
|
|
"""The workbench base class."""
|
|
MenuText = ""
|
|
ToolTip = ""
|
|
|
|
def Initialize(self):
|
|
"""Initializes this workbench."""
|
|
App.PrintWarning(str(self) + ": Workbench.Initialize() not implemented in subclass!")
|
|
def ContextMenu(self, recipient):
|
|
pass
|
|
def appendToolbar(self,name,cmds):
|
|
self.__Workbench__.appendToolbar(name, cmds)
|
|
def removeToolbar(self,name):
|
|
self.__Workbench__.removeToolbar(name)
|
|
def listToolbars(self):
|
|
return self.__Workbench__.listToolbars()
|
|
def getToolbarItems(self):
|
|
return self.__Workbench__.getToolbarItems()
|
|
def appendCommandbar(self,name,cmds):
|
|
self.__Workbench__.appendCommandbar(name, cmds)
|
|
def removeCommandbar(self,name):
|
|
self.__Workbench__.removeCommandbar(name)
|
|
def listCommandbars(self):
|
|
return self.__Workbench__.listCommandbars()
|
|
def appendMenu(self,name,cmds):
|
|
self.__Workbench__.appendMenu(name, cmds)
|
|
def removeMenu(self,name):
|
|
self.__Workbench__.removeMenu(name)
|
|
def listMenus(self):
|
|
return self.__Workbench__.listMenus()
|
|
def appendContextMenu(self,name,cmds):
|
|
self.__Workbench__.appendContextMenu(name, cmds)
|
|
def removeContextMenu(self,name):
|
|
self.__Workbench__.removeContextMenu(name)
|
|
def name(self):
|
|
return self.__Workbench__.name()
|
|
def GetClassName(self):
|
|
"""Return the name of the associated C++ class."""
|
|
# as default use this to simplify writing workbenches in Python
|
|
return "Gui::PythonWorkbench"
|
|
|
|
|
|
class StandardWorkbench ( Workbench ):
|
|
"""A workbench defines the tool bars, command bars, menus,
|
|
context menu and dockable windows of the main window.
|
|
"""
|
|
def Initialize(self):
|
|
"""Initialize this workbench."""
|
|
# load the module
|
|
Log ('Init: Loading FreeCAD GUI\n')
|
|
def GetClassName(self):
|
|
"""Return the name of the associated C++ class."""
|
|
return "Gui::StdWorkbench"
|
|
|
|
class NoneWorkbench ( Workbench ):
|
|
"""An empty workbench."""
|
|
MenuText = "<none>"
|
|
ToolTip = "The default empty workbench"
|
|
def Initialize(self):
|
|
"""Initialize this workbench."""
|
|
# load the module
|
|
Log ('Init: Loading FreeCAD GUI\n')
|
|
def GetClassName(self):
|
|
"""Return the name of the associated C++ class."""
|
|
return "Gui::NoneWorkbench"
|
|
|
|
def InitApplications():
|
|
import sys,os,traceback
|
|
try:
|
|
# Python3
|
|
import io as cStringIO
|
|
except ImportError:
|
|
# Python2
|
|
import cStringIO
|
|
# Searching modules dirs +++++++++++++++++++++++++++++++++++++++++++++++++++
|
|
# (additional module paths are already cached)
|
|
ModDirs = FreeCAD.__ModDirs__
|
|
#print ModDirs
|
|
Log('Init: Searching modules...\n')
|
|
for Dir in ModDirs:
|
|
if ((Dir != '') & (Dir != 'CVS') & (Dir != '__init__.py')):
|
|
InstallFile = os.path.join(Dir,"InitGui.py")
|
|
if (os.path.exists(InstallFile)):
|
|
try:
|
|
# XXX: This looks scary securitywise...
|
|
if sys.version_info.major < 3:
|
|
with open(InstallFile) as f:
|
|
exec(f.read())
|
|
else:
|
|
with open(file=InstallFile, encoding="utf-8") as f:
|
|
exec(f.read())
|
|
except Exception as inst:
|
|
Log('Init: Initializing ' + Dir + '... failed\n')
|
|
Log('-'*100+'\n')
|
|
Log(traceback.format_exc())
|
|
Log('-'*100+'\n')
|
|
Err('During initialization the error "' + str(inst) + '" occurred in ' + InstallFile + '\n')
|
|
Err('Please look into the log file for further information\n')
|
|
else:
|
|
Log('Init: Initializing ' + Dir + '... done\n')
|
|
else:
|
|
Log('Init: Initializing ' + Dir + '(InitGui.py not found)... ignore\n')
|
|
|
|
|
|
try:
|
|
import pkgutil
|
|
import importlib
|
|
import freecad
|
|
freecad.gui = FreeCADGui
|
|
for _, freecad_module_name, freecad_module_ispkg in pkgutil.iter_modules(freecad.__path__, "freecad."):
|
|
if freecad_module_ispkg:
|
|
Log('Init: Initializing ' + freecad_module_name + '\n')
|
|
try:
|
|
freecad_module = importlib.import_module(freecad_module_name)
|
|
if any (module_name == 'init_gui' for _, module_name, ispkg in pkgutil.iter_modules(freecad_module.__path__)):
|
|
importlib.import_module(freecad_module_name + '.init_gui')
|
|
Log('Init: Initializing ' + freecad_module_name + '... done\n')
|
|
else:
|
|
Log('Init: No init_gui module found in ' + freecad_module_name + ', skipping\n')
|
|
except Exception as inst:
|
|
Err('During initialization the error "' + str(inst) + '" occurred in ' + freecad_module_name + '\n')
|
|
Err('-'*80+'\n')
|
|
Err(traceback.format_exc())
|
|
Err('-'*80+'\n')
|
|
Log('Init: Initializing ' + freecad_module_name + '... failed\n')
|
|
Log('-'*80+'\n')
|
|
Log(traceback.format_exc())
|
|
Log('-'*80+'\n')
|
|
except ImportError as inst:
|
|
Err('During initialization the error "' + str(inst) + '" occurred\n')
|
|
|
|
Log ('Init: Running FreeCADGuiInit.py start script...\n')
|
|
|
|
|
|
|
|
# init the gui
|
|
|
|
# signal that the gui is up
|
|
App.GuiUp = 1
|
|
App.Gui = FreeCADGui
|
|
FreeCADGui.Workbench = Workbench
|
|
|
|
Gui.addWorkbench(NoneWorkbench())
|
|
|
|
# Monkey patching pivy.coin.SoGroup.removeAllChildren to work around a bug
|
|
# https://bitbucket.org/Coin3D/coin/pull-requests/119/fix-sochildlist-auditing/diff
|
|
|
|
def _SoGroup_init(self,*args):
|
|
import types
|
|
_SoGroup_init_orig(self,*args)
|
|
self.removeAllChildren = \
|
|
types.MethodType(FreeCADGui.coinRemoveAllChildren,self)
|
|
try:
|
|
from pivy import coin
|
|
_SoGroup_init_orig = coin.SoGroup.__init__
|
|
coin.SoGroup.__init__ = _SoGroup_init
|
|
except Exception:
|
|
pass
|
|
|
|
# init modules
|
|
InitApplications()
|
|
|
|
# set standard workbench (needed as fallback)
|
|
Gui.activateWorkbench("NoneWorkbench")
|
|
|
|
# Register .py, .FCScript and .FCMacro
|
|
FreeCAD.addImportType("Inventor V2.1 (*.iv)","FreeCADGui")
|
|
FreeCAD.addImportType("VRML V2.0 (*.wrl *.vrml *.wrz *.wrl.gz)","FreeCADGui")
|
|
FreeCAD.addImportType("Python (*.py *.FCMacro *.FCScript)","FreeCADGui")
|
|
FreeCAD.addExportType("Inventor V2.1 (*.iv)","FreeCADGui")
|
|
FreeCAD.addExportType("VRML V2.0 (*.wrl *.vrml *.wrz *.wrl.gz)","FreeCADGui")
|
|
FreeCAD.addExportType("X3D Extensible 3D (*.x3d *.x3dz)","FreeCADGui")
|
|
FreeCAD.addExportType("WebGL/X3D (*.xhtml)","FreeCADGui")
|
|
#FreeCAD.addExportType("IDTF (for 3D PDF) (*.idtf)","FreeCADGui")
|
|
#FreeCAD.addExportType("3D View (*.svg)","FreeCADGui")
|
|
FreeCAD.addExportType("Portable Document Format (*.pdf)","FreeCADGui")
|
|
|
|
del(InitApplications)
|
|
del(NoneWorkbench)
|
|
del(StandardWorkbench)
|
|
|
|
|
|
Log ('Init: Running FreeCADGuiInit.py start script... done\n')
|