Persistent toolbars for FreeCAD
This commit is contained in:
triplus
@@ -5,6 +5,8 @@ ENDIF (BUILD_GUI)
|
||||
SET(Tux_SRCS
|
||||
InitGui.py
|
||||
NavigationIndicatorGui.py
|
||||
PersistentToolbars.py
|
||||
PersistentToolbarsGui.py
|
||||
)
|
||||
SOURCE_GROUP("" FILES ${Tux_SRCS})
|
||||
|
||||
|
||||
@@ -20,7 +20,17 @@
|
||||
|
||||
p = FreeCAD.ParamGet("User parameter:Tux")
|
||||
|
||||
|
||||
# Navigation indicator
|
||||
if p.GetGroup("NavigationIndicator").GetBool("Enabled", 1):
|
||||
import NavigationIndicatorGui
|
||||
else:
|
||||
pass
|
||||
|
||||
|
||||
# Persistent toolbars
|
||||
if p.GetGroup("PersistentToolbars").GetBool("Enabled", 1):
|
||||
import PersistentToolbarsGui
|
||||
else:
|
||||
pass
|
||||
|
||||
|
||||
89
src/Mod/Tux/PersistentToolbars.py
Normal file
89
src/Mod/Tux/PersistentToolbars.py
Normal file
@@ -0,0 +1,89 @@
|
||||
# Persistent toolbars for FreeCAD.
|
||||
# Copyright (C) 2016, 2017 triplus @ FreeCAD
|
||||
#
|
||||
#
|
||||
# This library is free software; you can redistribute it and/or
|
||||
# modify it under the terms of the GNU Lesser General Public
|
||||
# License as published by the Free Software Foundation; either
|
||||
# version 2.1 of the License, or (at your option) any later version.
|
||||
#
|
||||
# This library 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 Lesser General Public
|
||||
# License along with this library; if not, write to the Free Software
|
||||
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
||||
|
||||
"""Persistent toolbars for FreeCAD."""
|
||||
|
||||
import FreeCAD as App
|
||||
|
||||
|
||||
def addTop(name, toolbars):
|
||||
"""addTop("name", ["toolbars"])
|
||||
|
||||
Example for Sketcher workbench (InitGui.py):
|
||||
|
||||
try:
|
||||
import PersistentToolbars
|
||||
|
||||
toolbars = ["Sketcher constraints",
|
||||
"Break",
|
||||
"Sketcher geometries"]
|
||||
|
||||
PersistentToolbars.addRight("SketcherWorkbench", toolbars)
|
||||
except ImportError:
|
||||
pass
|
||||
|
||||
Description:
|
||||
Provides ability to preset toolbar position for any workbench.
|
||||
Settings are deleted when FreeCAD exits normally. After user
|
||||
customizes workbench toolbar position this setting has no
|
||||
effect anymore. Customized toolbar position set by the user is
|
||||
used instead."""
|
||||
|
||||
p = App.ParamGet("User parameter:Tux/PersistentToolbars/System")
|
||||
p.GetGroup(name).SetBool("Saved", 1)
|
||||
p.GetGroup(name).SetString("Top", ",".join(toolbars))
|
||||
|
||||
|
||||
def addRight(name, toolbars):
|
||||
"""addRight("name", ["toolbars"])
|
||||
|
||||
Description:
|
||||
Look at addTop for more information."""
|
||||
|
||||
p = App.ParamGet("User parameter:Tux/PersistentToolbars/System")
|
||||
p.GetGroup(name).SetBool("Saved", 1)
|
||||
p.GetGroup(name).SetString("Right", ",".join(toolbars))
|
||||
|
||||
|
||||
def addLeft(name, toolbars):
|
||||
"""addLeft("name", ["toolbars"])
|
||||
|
||||
Description:
|
||||
Look at addTop for more information."""
|
||||
|
||||
p = App.ParamGet("User parameter:Tux/PersistentToolbars/System")
|
||||
p.GetGroup(name).SetBool("Saved", 1)
|
||||
p.GetGroup(name).SetString("Left", ",".join(toolbars))
|
||||
|
||||
|
||||
def addBottom(name, toolbars):
|
||||
"""addBottom("name", ["toolbars"])
|
||||
|
||||
Description:
|
||||
Look at addTop for more information."""
|
||||
|
||||
p = App.ParamGet("User parameter:Tux/PersistentToolbars/System")
|
||||
p.GetGroup(name).SetBool("Saved", 1)
|
||||
p.GetGroup(name).SetString("Bottom", ",".join(toolbars))
|
||||
|
||||
|
||||
def clear():
|
||||
"""Delete all user and system toolbar position data."""
|
||||
|
||||
p = App.ParamGet("User parameter:Tux")
|
||||
p.RemGroup("PersistentToolbars")
|
||||
251
src/Mod/Tux/PersistentToolbarsGui.py
Normal file
251
src/Mod/Tux/PersistentToolbarsGui.py
Normal file
@@ -0,0 +1,251 @@
|
||||
# Persistent toolbars for FreeCAD
|
||||
# Copyright (C) 2016, 2017 triplus @ FreeCAD
|
||||
#
|
||||
#
|
||||
# This library is free software; you can redistribute it and/or
|
||||
# modify it under the terms of the GNU Lesser General Public
|
||||
# License as published by the Free Software Foundation; either
|
||||
# version 2.1 of the License, or (at your option) any later version.
|
||||
#
|
||||
# This library 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 Lesser General Public
|
||||
# License along with this library; if not, write to the Free Software
|
||||
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
||||
|
||||
"""Persistent toolbars for FreeCAD."""
|
||||
|
||||
import operator
|
||||
import FreeCAD as App
|
||||
import FreeCADGui as Gui
|
||||
from PySide import QtCore
|
||||
from PySide import QtGui
|
||||
|
||||
conectedToolbars = []
|
||||
timer = QtCore.QTimer()
|
||||
mw = Gui.getMainWindow()
|
||||
|
||||
|
||||
def pythonToolbars():
|
||||
"""Manage Python based toolbars in Arch and Draft workbench."""
|
||||
|
||||
active = Gui.activeWorkbench().__class__.__name__
|
||||
|
||||
if active == "DraftWorkbench" or active == "ArchWorkbench":
|
||||
if hasattr(Gui, "draftToolBar"):
|
||||
Gui.draftToolBar.Activated()
|
||||
if hasattr(Gui, "Snapper"):
|
||||
Gui.Snapper.show()
|
||||
else:
|
||||
pass
|
||||
|
||||
|
||||
def isConnected(i):
|
||||
"""Connect toolbar to onSave function."""
|
||||
|
||||
if i not in conectedToolbars:
|
||||
conectedToolbars.append(i)
|
||||
i.topLevelChanged.connect(onSave)
|
||||
else:
|
||||
pass
|
||||
|
||||
|
||||
def onRestore(active):
|
||||
"""Restore current workbench toolbars position."""
|
||||
|
||||
toolbars = {}
|
||||
tb = mw.findChildren(QtGui.QToolBar)
|
||||
pUser = App.ParamGet("User parameter:Tux/PersistentToolbars/User")
|
||||
pSystem = App.ParamGet("User parameter:Tux/PersistentToolbars/System")
|
||||
|
||||
for i in tb:
|
||||
|
||||
isConnected(i)
|
||||
|
||||
if i.objectName() and not i.isFloating():
|
||||
toolbars[i.objectName()] = i
|
||||
else:
|
||||
pass
|
||||
|
||||
if pUser.GetGroup(active).GetBool("Saved"):
|
||||
group = pUser.GetGroup(active)
|
||||
elif pSystem.GetGroup(active).GetBool("Saved"):
|
||||
group = pSystem.GetGroup(active)
|
||||
else:
|
||||
group = None
|
||||
|
||||
if group:
|
||||
topRestore = group.GetString("Top").split(",")
|
||||
leftRestore = group.GetString("Left").split(",")
|
||||
rightRestore = group.GetString("Right").split(",")
|
||||
bottomRestore = group.GetString("Bottom").split(",")
|
||||
|
||||
# Reduce flickering.
|
||||
for i in toolbars:
|
||||
if (i not in topRestore and
|
||||
i not in leftRestore and
|
||||
i not in rightRestore and
|
||||
i not in bottomRestore):
|
||||
|
||||
area = mw.toolBarArea(toolbars[i])
|
||||
|
||||
if area == QtCore.Qt.ToolBarArea.LeftToolBarArea:
|
||||
leftRestore.append(i)
|
||||
elif area == QtCore.Qt.ToolBarArea.RightToolBarArea:
|
||||
rightRestore.append(i)
|
||||
elif area == QtCore.Qt.ToolBarArea.BottomToolBarArea:
|
||||
bottomRestore.append(i)
|
||||
else:
|
||||
topRestore.append(i)
|
||||
else:
|
||||
pass
|
||||
|
||||
for i in topRestore:
|
||||
if i == "Break":
|
||||
mw.addToolBarBreak(QtCore.Qt.TopToolBarArea)
|
||||
elif i in toolbars:
|
||||
mw.addToolBar(QtCore.Qt.TopToolBarArea, toolbars[i])
|
||||
else:
|
||||
pass
|
||||
|
||||
for i in leftRestore:
|
||||
if i == "Break":
|
||||
mw.addToolBarBreak(QtCore.Qt.LeftToolBarArea)
|
||||
elif i in toolbars:
|
||||
mw.addToolBar(QtCore.Qt.LeftToolBarArea, toolbars[i])
|
||||
else:
|
||||
pass
|
||||
|
||||
for i in rightRestore:
|
||||
if i == "Break":
|
||||
mw.addToolBarBreak(QtCore.Qt.RightToolBarArea)
|
||||
elif i in toolbars:
|
||||
mw.addToolBar(QtCore.Qt.RightToolBarArea, toolbars[i])
|
||||
else:
|
||||
pass
|
||||
|
||||
for i in bottomRestore:
|
||||
if i == "Break":
|
||||
mw.addToolBarBreak(QtCore.Qt.BottomToolBarArea)
|
||||
elif i in toolbars:
|
||||
mw.addToolBar(QtCore.Qt.BottomToolBarArea, toolbars[i])
|
||||
else:
|
||||
pass
|
||||
else:
|
||||
pass
|
||||
|
||||
|
||||
def onSave():
|
||||
"""Save current workbench toolbars position."""
|
||||
|
||||
tb = mw.findChildren(QtGui.QToolBar)
|
||||
active = Gui.activeWorkbench().__class__.__name__
|
||||
p = App.ParamGet("User parameter:Tux/PersistentToolbars/User")
|
||||
group = p.GetGroup(active)
|
||||
|
||||
top = []
|
||||
left = []
|
||||
right = []
|
||||
bottom = []
|
||||
|
||||
for i in tb:
|
||||
if i.objectName() and i.isVisible() and not i.isFloating():
|
||||
|
||||
area = mw.toolBarArea(i)
|
||||
|
||||
x = i.geometry().x()
|
||||
y = i.geometry().y()
|
||||
b = mw.toolBarBreak(i)
|
||||
n = i.objectName()
|
||||
|
||||
if area == QtCore.Qt.ToolBarArea.TopToolBarArea:
|
||||
top.append([x, y, b, n])
|
||||
elif area == QtCore.Qt.ToolBarArea.LeftToolBarArea:
|
||||
left.append([x, y, b, n])
|
||||
elif area == QtCore.Qt.ToolBarArea.RightToolBarArea:
|
||||
right.append([-x, y, b, n])
|
||||
elif area == QtCore.Qt.ToolBarArea.BottomToolBarArea:
|
||||
bottom.append([x, -y, b, n])
|
||||
else:
|
||||
pass
|
||||
else:
|
||||
pass
|
||||
|
||||
top = sorted(top, key=operator.itemgetter(1, 0))
|
||||
left = sorted(left, key=operator.itemgetter(0, 1))
|
||||
right = sorted(right, key=operator.itemgetter(0, 1))
|
||||
bottom = sorted(bottom, key=operator.itemgetter(1, 0))
|
||||
|
||||
topSave = []
|
||||
leftSave = []
|
||||
rightSave = []
|
||||
bottomSave = []
|
||||
|
||||
for i in top:
|
||||
if i[2] is True:
|
||||
topSave.append("Break")
|
||||
topSave.append(i[3])
|
||||
else:
|
||||
topSave.append(i[3])
|
||||
|
||||
for i in left:
|
||||
if i[2] is True:
|
||||
leftSave.append("Break")
|
||||
leftSave.append(i[3])
|
||||
else:
|
||||
leftSave.append(i[3])
|
||||
|
||||
for i in right:
|
||||
if i[2] is True:
|
||||
rightSave.append("Break")
|
||||
rightSave.append(i[3])
|
||||
else:
|
||||
rightSave.append(i[3])
|
||||
|
||||
for i in bottom:
|
||||
if i[2] is True:
|
||||
bottomSave.append("Break")
|
||||
bottomSave.append(i[3])
|
||||
else:
|
||||
bottomSave.append(i[3])
|
||||
|
||||
group.SetBool("Saved", 1)
|
||||
group.SetString("Top", ",".join(topSave))
|
||||
group.SetString("Left", ",".join(leftSave))
|
||||
group.SetString("Right", ",".join(rightSave))
|
||||
group.SetString("Bottom", ",".join(bottomSave))
|
||||
|
||||
|
||||
def onWorkbenchActivated():
|
||||
"""When workbench gets activated restore toolbar position."""
|
||||
|
||||
active = Gui.activeWorkbench().__class__.__name__
|
||||
|
||||
if active:
|
||||
pythonToolbars()
|
||||
onRestore(active)
|
||||
else:
|
||||
pass
|
||||
|
||||
|
||||
def onStart():
|
||||
"""Start persistent toolbars."""
|
||||
|
||||
onWorkbenchActivated()
|
||||
mw.mainWindowClosed.connect(onClose)
|
||||
mw.workbenchActivated.connect(onWorkbenchActivated)
|
||||
|
||||
|
||||
def onClose():
|
||||
"""Remove system provided presets on FreeCAD close."""
|
||||
|
||||
p = App.ParamGet("User parameter:Tux/PersistentToolbars")
|
||||
p.RemGroup("System")
|
||||
|
||||
|
||||
timer.setSingleShot(True)
|
||||
timer.timeout.connect(onStart)
|
||||
timer.start()
|
||||
Reference in New Issue
Block a user