FreeCAD / src /Mod /Tux /PersistentToolbarsGui.py

Upload folder using huggingface_hub

985c397 verified 9 months ago

7.2 kB

# SPDX-License-Identifier: LGPL-2.1-or-later

# 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 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 i.parentWidget() == mw 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:
        onRestore(active)
    else:
        pass


def onStart():
    """Start persistent toolbars."""
    if mw.property("eventLoop"):
        start = False
        try:
            mw.mainWindowClosed
            mw.workbenchActivated
            start = True
        except AttributeError:
            pass
        if start:
            timer.stop()
            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.timeout.connect(onStart)
timer.start(500)

	# SPDX-License-Identifier: LGPL-2.1-or-later

	# 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 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 i.parentWidget() == mw 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:
	onRestore(active)
	else:
	pass


	def onStart():
	"""Start persistent toolbars."""
	if mw.property("eventLoop"):
	start = False
	try:
	mw.mainWindowClosed
	mw.workbenchActivated
	start = True
	except AttributeError:
	pass
	if start:
	timer.stop()
	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.timeout.connect(onStart)
	timer.start(500)