forked from dubstar-04/TurningAddon
-
Notifications
You must be signed in to change notification settings - Fork 0
/
InitTurningAddon.py
99 lines (76 loc) · 4.01 KB
/
InitTurningAddon.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
# -*- coding: utf-8 -*-
# ***************************************************************************
# * *
# * Copyright (c) 2020 Daniel Wood <s.d.wood.82@googlemail.com> *
# * *
# * 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 FreeCADGui
from PySide import QtGui
# import all turning operations
from PathTurnScripts import PathTurnFaceGui # noqa: F401
from PathTurnScripts import PathTurnPartoffGui # noqa: F401
from PathTurnScripts import PathTurnProfileGui # noqa: F401
from PathTurnScripts import PathTurnRoughGui # noqa: F401
from PathTurnScripts import PathTurnToolHelperGui # noqa: F401
import os
import re
__dir__ = os.path.dirname(__file__)
iconPath = os.path.join(__dir__, 'Gui/Resources/Icons')
def getIcon(iconName):
return os.path.join(iconPath, iconName)
def updateMenu(workbench):
if workbench == 'PathWorkbench':
print('FreeCAD Turning Addon loaded:', workbench)
mw = FreeCADGui.getMainWindow()
addonMenu = None
# Find the main path menu
pathMenu = mw.findChild(QtGui.QMenu, "&Path")
for menu in pathMenu.actions():
if menu.text() == "Turning Addon":
# create a new addon menu
addonMenu = menu.menu()
break
if addonMenu is None:
addonMenu = QtGui.QMenu("Turning Addon")
addonMenu.setObjectName("Turning_Addons")
addonMenu.setIcon(QtGui.QPixmap(getIcon('Path-TurningAddon.svg')))
# Find the dressup menu entry
dressupMenu = mw.findChild(QtGui.QMenu, "Path Dressup")
pathMenu.insertMenu(dressupMenu.menuAction(), addonMenu)
# list of commands to add to the menu, add new commands here:
commands = ['TurnFace', 'TurnProfile', 'TurnPartoff', 'TurnRough', 'TurnToolHelper']
# load the commands to the FreeCAD Menu
for command in commands:
print('Loading Command: ', command)
# create an action for this addon
action = createAction(addonMenu, command)
# append this addon to addon menu
addonMenu.addAction(action)
def createAction(menu, actionName):
# space the actionName at caps
text = re.sub(r"(\w)([A-Z])", r"\1 \2", actionName)
command = "Path_" + actionName
icon = "Path-" + actionName + ".svg"
action = QtGui.QAction(menu)
action.setText(text)
action.setIcon(QtGui.QPixmap(getIcon(icon)))
action.setStatusTip("Path " + actionName + " Operation")
action.triggered.connect(lambda: FreeCADGui.runCommand(command, 0))
return action
FreeCADGui.getMainWindow().workbenchActivated.connect(updateMenu)