autohide_dock.py

from PySide.QtCore import Qt, QEvent, QPointF
from PySide.QtGui import *

class QAutoHideDockWidgets(QToolBar):
    """
    QMainWindow "mixin" which provides auto-hiding support for dock widgets
    (not toolbars).
    """
    DOCK_AREA_TO_TB = {
        Qt.LeftDockWidgetArea: Qt.LeftToolBarArea,
        Qt.RightDockWidgetArea: Qt.RightToolBarArea,
        Qt.TopDockWidgetArea: Qt.TopToolBarArea,
        Qt.BottomDockWidgetArea: Qt.BottomToolBarArea,
    }

    def __init__(self, area, parent, name="AUTO_HIDE"):
        QToolBar.__init__(self, parent)
        assert isinstance(parent, QMainWindow)
        assert area in self.DOCK_AREA_TO_TB
        self._area = area
        self.setObjectName(name)
        self.setWindowTitle(name)
        
        self.setFloatable(False)
        self.setMovable(False)
        w = QWidget(None)
        w.resize(10, 100)
        self.setSizePolicy(QSizePolicy(QSizePolicy.Fixed, QSizePolicy.MinimumExpanding))
        self.addWidget(w)

        self.setAllowedAreas(self.DOCK_AREA_TO_TB[self._area])
        self.parent().addToolBar(self.DOCK_AREA_TO_TB[self._area], self)
        self.parent().centralWidget().installEventFilter(self)
        
        self.currentState = None
        self.exceptions = []
        self.activeTab = 0
        self.setVisible(False)
        self.hideDockWidgets()

    def _dockWidgets(self):
        mw = self.parent()
        for w in mw.findChildren(QDockWidget, None):
            if mw.dockWidgetArea(w) == self._area and not w.isFloating():
                yield w

    def paintEvent(self, event):
        p = QPainter(self)
        p.setPen(Qt.black)
        p.setBrush(Qt.black)
        if self._area == Qt.LeftDockWidgetArea:
            p.translate(QPointF(0, self.height() / 2 - 5))
            p.drawPolygon((QPointF(2,0), QPointF(8,5), QPointF(2,10)))
        elif self._area == Qt.RightDockWidgetArea:
            p.translate(QPointF(0, self.height() / 2 - 5))
            p.drawPolygon((QPointF(8,0), QPointF(2,5), QPointF(8,10)))

    def _multiSetVisible(self, widgets, state):
        if self.currentState == state:
            return
        widgetsNotInExceptions = [w for w in widgets if w not in self.exceptions]
        if len(widgetsNotInExceptions) == 0:
            return
        newExceptions = []
        if state:
            self.setVisible(False)
        else:
            if self.currentState != None:
                newExceptions = [w for w in widgets if not w.isVisible()]
                # get the tab bar to reset the active tab position
                tabList = self.parent().findChildren(QTabBar, None)
                if len(tabList) > 0:
                    tabBar = tabList[0]
                    self.activeTab = tabBar.currentIndex()
        self.currentState = state

        for w in widgets:
            w.setUpdatesEnabled(False)
        for w in widgetsNotInExceptions:
            w.setVisible(state)
        for w in widgets:
            w.setUpdatesEnabled(True)

        if not state and widgets:
            self.setVisible(True)
        self.exceptions = newExceptions

        if state:
            tabList = self.parent().findChildren(QTabBar, None)
            if len(tabList) > 0:
                tabBar = tabList[0]
                tabBar.setCurrentIndex(self.activeTab)

    def enterEvent(self, event):
        self.showDockWidgets()

    def eventFilter(self, obj, event):
        if event.type() == QEvent.Enter:
            assert obj == self.parent().centralWidget()
            self.hideDockWidgets()
        return False

    def setDockWidgetsVisible(self, state):
        self._multiSetVisible(list(self._dockWidgets()), state)

    def showDockWidgets(self): self.setDockWidgetsVisible(True)
    def hideDockWidgets(self): self.setDockWidgetsVisible(False)