InitGui.py

# 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."""


def persistentToolbars():
    """Persistent toolbars for FreeCAD."""

    import operator
    import FreeCAD as App
    import FreeCADGui as Gui
    from PySide import QtCore
    from PySide import QtGui
    import PersistentToolbarsPath

    conectedToolbars = []
    mw = Gui.getMainWindow()
    t = PersistentToolbarsPath.timer()

    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"):
                try:
                    Gui.draftToolBar.Activated()
                except:
                    m = "Persistent toolbars: draftToolBar toolbar not managed.\n"
                    App.Console.PrintMessage(m)
            if hasattr(Gui, "Snapper"):
                try:
                    Gui.Snapper.show()
                except:
                    m = "Persistent toolbars: Snapper toolbar not managed.\n"
                    App.Console.PrintMessage(m)
        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)
        p = App.ParamGet("User parameter:BaseApp/PersistentToolbars")

        for i in tb:

            isConnected(i)

            if i.objectName() and not i.isFloating():
                toolbars[i.objectName()] = i
            else:
                pass

        if p.GetGroup(active).GetBool("Saved"):

            group = p.GetGroup(active)

            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:BaseApp/PersistentToolbars")
        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.workbenchActivated.connect(onWorkbenchActivated)

    t.setSingleShot(True)
    t.timeout.connect(onStart)
    t.start()


try:
    if FreeCAD.Version()[1] <= "16":
        persistentToolbars()
    else:
        pass
except:
    pass