a083fea057
Also other small fixes: cleanup for getting the icon of the workbench; we don't need to test for PySide because it should already exist as part of the system for the entire interface to work; cleanup of the printing functions.
207 lines
10 KiB
Python
207 lines
10 KiB
Python
"""Initialization of the Draft workbench (graphical interface)."""
|
|
# ***************************************************************************
|
|
# * Copyright (c) 2009 Yorik van Havre <yorik@uncreated.net> *
|
|
# * *
|
|
# * 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. *
|
|
# * *
|
|
# * This program 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 Library General Public License for more details. *
|
|
# * *
|
|
# * You should have received a copy of the GNU Library General Public *
|
|
# * License along with this program; if not, write to the Free Software *
|
|
# * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 *
|
|
# * USA *
|
|
# * *
|
|
# ***************************************************************************
|
|
import os
|
|
import FreeCAD
|
|
import FreeCADGui
|
|
|
|
__title__ = "FreeCAD Draft Workbench - Init file"
|
|
__author__ = "Yorik van Havre <yorik@uncreated.net>"
|
|
__url__ = "https://www.freecadweb.org"
|
|
|
|
|
|
class DraftWorkbench(FreeCADGui.Workbench):
|
|
"""The Draft Workbench definition."""
|
|
|
|
def __init__(self):
|
|
def QT_TRANSLATE_NOOP(context, text):
|
|
return text
|
|
|
|
__dirname__ = os.path.join(FreeCAD.getResourceDir(), "Mod", "Draft")
|
|
_tooltip = "The Draft workbench is used for 2D drafting on a grid"
|
|
self.__class__.Icon = os.path.join(__dirname__,
|
|
"Resources", "icons",
|
|
"DraftWorkbench.svg")
|
|
self.__class__.MenuText = QT_TRANSLATE_NOOP("draft", "Draft")
|
|
self.__class__.ToolTip = QT_TRANSLATE_NOOP("draft", _tooltip)
|
|
|
|
def Initialize(self):
|
|
"""When the workbench is first loaded."""
|
|
|
|
def QT_TRANSLATE_NOOP(context, text):
|
|
return text
|
|
|
|
# Run self-tests
|
|
dependencies_OK = False
|
|
try:
|
|
from pivy import coin
|
|
if FreeCADGui.getSoDBVersion() != coin.SoDB.getVersion():
|
|
raise AssertionError("FreeCAD and Pivy use different versions "
|
|
"of Coin. "
|
|
"This will lead to unexpected behaviour.")
|
|
except AssertionError:
|
|
FreeCAD.Console.PrintWarning("Error: FreeCAD and Pivy "
|
|
"use different versions of Coin. "
|
|
"This will lead to unexpected "
|
|
"behaviour.\n")
|
|
except ImportError:
|
|
FreeCAD.Console.PrintWarning("Error: Pivy not found, "
|
|
"Draft Workbench will be disabled.\n")
|
|
except Exception:
|
|
FreeCAD.Console.PrintWarning("Error: Unknown error "
|
|
"while trying to load Pivy.\n")
|
|
else:
|
|
dependencies_OK = True
|
|
|
|
if not dependencies_OK:
|
|
return
|
|
|
|
# Import Draft tools, icons
|
|
try:
|
|
import Draft_rc
|
|
import DraftTools
|
|
import DraftGui
|
|
import DraftFillet
|
|
from draftguitools import gui_circulararray
|
|
from draftguitools import gui_polararray
|
|
FreeCADGui.addLanguagePath(":/translations")
|
|
FreeCADGui.addIconPath(":/icons")
|
|
except Exception as exc:
|
|
FreeCAD.Console.PrintError(exc)
|
|
FreeCAD.Console.PrintError("Error: Initializing one or more "
|
|
"of the Draft modules failed, "
|
|
"Draft will not work as expected.\n")
|
|
|
|
# Set up command lists
|
|
self.cmdList = ["Draft_Line", "Draft_Wire", "Draft_Fillet", "Draft_ArcTools",
|
|
"Draft_Circle", "Draft_Ellipse", "Draft_Rectangle", "Draft_Polygon",
|
|
"Draft_BSpline", "Draft_BezierTools", "Draft_Point",
|
|
"Draft_Facebinder"]
|
|
|
|
self.annotation_tools = ["Draft_Text", "Draft_ShapeString", "Draft_Dimension",
|
|
"Draft_Label"]
|
|
|
|
self.modList = ["Draft_Move", "Draft_Rotate", "Draft_Offset",
|
|
"Draft_Trimex", "Draft_Join", "Draft_Split",
|
|
"Draft_Upgrade", "Draft_Downgrade", "Draft_Scale",
|
|
"Draft_Edit", "Draft_SubelementHighlight",
|
|
"Draft_WireToBSpline", "Draft_AddPoint",
|
|
"Draft_DelPoint", "Draft_Shape2DView",
|
|
"Draft_Draft2Sketch", "Draft_Array", "Draft_LinkArray",
|
|
"Draft_PolarArray", "Draft_CircularArray",
|
|
"Draft_PathArray", "Draft_PathLinkArray", "Draft_PointArray", "Draft_Clone",
|
|
"Draft_Drawing", "Draft_Mirror", "Draft_Stretch"]
|
|
|
|
self.treecmdList = ["Draft_ApplyStyle", "Draft_ToggleDisplayMode",
|
|
"Draft_AddToGroup", "Draft_SelectGroup",
|
|
"Draft_SelectPlane", "Draft_ShowSnapBar",
|
|
"Draft_ToggleGrid", "Draft_AutoGroup"]
|
|
|
|
self.lineList = ["Draft_UndoLine", "Draft_FinishLine",
|
|
"Draft_CloseLine"]
|
|
|
|
self.utils = ["Draft_Layer", "Draft_Heal", "Draft_FlipDimension",
|
|
"Draft_ToggleConstructionMode",
|
|
"Draft_ToggleContinueMode", "Draft_Edit",
|
|
"Draft_Slope", "Draft_SetWorkingPlaneProxy",
|
|
"Draft_AddConstruction"]
|
|
|
|
self.snapList = ['Draft_Snap_Lock', 'Draft_Snap_Midpoint',
|
|
'Draft_Snap_Perpendicular',
|
|
'Draft_Snap_Grid', 'Draft_Snap_Intersection',
|
|
'Draft_Snap_Parallel',
|
|
'Draft_Snap_Endpoint', 'Draft_Snap_Angle',
|
|
'Draft_Snap_Center',
|
|
'Draft_Snap_Extension', 'Draft_Snap_Near',
|
|
'Draft_Snap_Ortho', 'Draft_Snap_Special',
|
|
'Draft_Snap_Dimensions', 'Draft_Snap_WorkingPlane']
|
|
|
|
# Set up toolbars
|
|
self.appendToolbar("Draft creation tools", self.cmdList)
|
|
self.appendToolbar("Draft annotation tools", self.annotation_tools)
|
|
self.appendToolbar("Draft modification tools", self.modList)
|
|
|
|
# Set up menu
|
|
self.appendMenu(QT_TRANSLATE_NOOP("draft","&2D Drafting"),self.cmdList)
|
|
self.appendMenu(QT_TRANSLATE_NOOP("draft","&Modify"),self.modList)
|
|
self.appendMenu(QT_TRANSLATE_NOOP("draft","&Annotation"),self.annotation_tools)
|
|
self.appendMenu(QT_TRANSLATE_NOOP("draft", "&Utilities"), self.utils + self.treecmdList)
|
|
|
|
if hasattr(FreeCADGui, "draftToolBar"):
|
|
if not hasattr(FreeCADGui.draftToolBar, "loadedPreferences"):
|
|
FreeCADGui.addPreferencePage(":/ui/preferences-draft.ui", QT_TRANSLATE_NOOP("draft", "Draft"))
|
|
FreeCADGui.addPreferencePage(":/ui/preferences-draftsnap.ui", QT_TRANSLATE_NOOP("draft", "Draft"))
|
|
FreeCADGui.addPreferencePage(":/ui/preferences-draftvisual.ui", QT_TRANSLATE_NOOP("draft", "Draft"))
|
|
FreeCADGui.addPreferencePage(":/ui/preferences-drafttexts.ui", QT_TRANSLATE_NOOP("draft", "Draft"))
|
|
FreeCADGui.draftToolBar.loadedPreferences = True
|
|
FreeCAD.Console.PrintLog('Loading Draft module, done.\n')
|
|
|
|
def Activated(self):
|
|
"""When entering the workbench."""
|
|
if hasattr(FreeCADGui, "draftToolBar"):
|
|
FreeCADGui.draftToolBar.Activated()
|
|
if hasattr(FreeCADGui, "Snapper"):
|
|
FreeCADGui.Snapper.show()
|
|
FreeCAD.Console.PrintLog("Draft workbench activated.\n")
|
|
|
|
def Deactivated(self):
|
|
"""When quitting the workbench."""
|
|
if hasattr(FreeCADGui, "draftToolBar"):
|
|
FreeCADGui.draftToolBar.Deactivated()
|
|
if hasattr(FreeCADGui, "Snapper"):
|
|
FreeCADGui.Snapper.hide()
|
|
FreeCAD.Console.PrintLog("Draft workbench deactivated.\n")
|
|
|
|
def ContextMenu(self, recipient):
|
|
"""Define an optional custom context menu."""
|
|
from DraftGui import translate
|
|
if recipient == "View":
|
|
if FreeCAD.activeDraftCommand is None:
|
|
if FreeCADGui.Selection.getSelection():
|
|
self.appendContextMenu("Draft", self.cmdList + self.modList)
|
|
self.appendContextMenu("Utilities", self.treecmdList)
|
|
else:
|
|
self.appendContextMenu("Draft", self.cmdList)
|
|
else:
|
|
if FreeCAD.activeDraftCommand.featureName == translate("draft","Line"):
|
|
# BUG: line subcommands are not usable while another command is active
|
|
self.appendContextMenu("", self.lineList)
|
|
else:
|
|
if FreeCADGui.Selection.getSelection():
|
|
self.appendContextMenu("Utilities", self.treecmdList)
|
|
|
|
def GetClassName(self):
|
|
"""Type of workbench."""
|
|
return "Gui::PythonWorkbench"
|
|
|
|
|
|
FreeCADGui.addWorkbench(DraftWorkbench)
|
|
|
|
# Preference pages for importing and exporting various file formats
|
|
# are independent of the loading of the workbench and can be loaded at startup
|
|
import Draft_rc
|
|
FreeCADGui.addPreferencePage(":/ui/preferences-dxf.ui", "Import-Export")
|
|
FreeCADGui.addPreferencePage(":/ui/preferences-dwg.ui", "Import-Export")
|
|
FreeCADGui.addPreferencePage(":/ui/preferences-svg.ui", "Import-Export")
|
|
FreeCADGui.addPreferencePage(":/ui/preferences-oca.ui", "Import-Export")
|
|
|
|
FreeCAD.__unit_test__ += ["TestDraft"]
|