--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/E5Gui/E5ModelToolBar.py Wed Jan 13 18:53:58 2010 +0000
@@ -0,0 +1,289 @@
+# -*- coding: utf-8 -*-
+
+# Copyright (c) 2009 - 2010 Detlev Offenbach <detlev@die-offenbachs.de>
+#
+
+"""
+Module implementing a tool bar populated from a QAbstractItemModel.
+"""
+
+from PyQt4.QtCore import *
+from PyQt4.QtGui import *
+
+from .E5ModelMenu import E5ModelMenu
+
+class E5ModelToolBar(QToolBar):
+ """
+ Class implementing a tool bar populated from a QAbstractItemModel.
+ """
+ def __init__(self, title = None, parent = None):
+ """
+ Constructor
+
+ @param title title for the tool bar (string)
+ @param parent reference to the parent widget (QWidget)
+ """
+ if title is not None:
+ QToolBar.__init__(self, title, parent)
+ else:
+ QToolBar.__init__(self, parent)
+
+ self.__model = None
+
+ self.__root = QModelIndex()
+ self.__dragStartPosition = QPoint()
+
+ if self.isVisible():
+ self._build()
+
+ self.setAcceptDrops(True)
+
+ self._mouseButton = Qt.NoButton
+ self._keyboardModifiers = Qt.KeyboardModifiers(Qt.NoModifier)
+ self.__dropRow = -1
+ self.__dropIndex = None
+
+ def setModel(self, model):
+ """
+ Public method to set the model for the tool bar.
+
+ @param model reference to the model (QAbstractItemModel)
+ """
+ if self.__model is not None:
+ self.disconnect(self.__model,
+ SIGNAL("modelReset()"),
+ self._build)
+ self.disconnect(self.__model,
+ SIGNAL("rowsInserted(const QModelIndex&, int, int)"),
+ self._build)
+ self.disconnect(self.__model,
+ SIGNAL("rowsRemoved(const QModelIndex&, int, int)"),
+ self._build)
+ self.disconnect(self.__model,
+ SIGNAL("dataChanged(const QModelIndex&, const QModelIndex&)"),
+ self._build)
+
+ self.__model = model
+
+ if self.__model is not None:
+ self.connect(self.__model,
+ SIGNAL("modelReset()"),
+ self._build)
+ self.connect(self.__model,
+ SIGNAL("rowsInserted(const QModelIndex&, int, int)"),
+ self._build)
+ self.connect(self.__model,
+ SIGNAL("rowsRemoved(const QModelIndex&, int, int)"),
+ self._build)
+ self.connect(self.__model,
+ SIGNAL("dataChanged(const QModelIndex&, const QModelIndex&)"),
+ self._build)
+
+ def model(self):
+ """
+ Public method to get a reference to the model.
+
+ @return reference to the model (QAbstractItemModel)
+ """
+ return self.__model
+
+ def setRootIndex(self, idx):
+ """
+ Public method to set the root index.
+
+ @param idx index to be set as the root index (QModelIndex)
+ """
+ self.__root = idx
+
+ def rootIndex(self):
+ """
+ Public method to get the root index.
+
+ @return root index (QModelIndex)
+ """
+ return self.__root
+
+ def _build(self):
+ """
+ Protected slot to build the tool bar.
+ """
+ assert self.__model is not None
+
+ self.clear()
+
+ for i in range(self.__model.rowCount(self.__root)):
+ idx = self.__model.index(i, 0, self.__root)
+ v = idx
+
+ title = idx.data(Qt.DisplayRole)
+ icon = idx.data(Qt.DecorationRole)
+ if icon == NotImplemented or icon is None:
+ icon = QIcon()
+ folder = self.__model.hasChildren(idx)
+
+ act = self.addAction(icon, title)
+ act.setData(v)
+
+ button = self.widgetForAction(act)
+ button.installEventFilter(self)
+
+ if folder:
+ menu = self._createMenu()
+ menu.setModel(self.__model)
+ menu.setRootIndex(idx)
+ act.setMenu(menu)
+ button.setPopupMode(QToolButton.InstantPopup)
+ button.setToolButtonStyle(Qt.ToolButtonTextBesideIcon)
+
+ def index(self, action):
+ """
+ Public method to get the index of an action.
+
+ @param action reference to the action to get the index for (QAction)
+ @return index of the action (QModelIndex)
+ """
+ if action is None:
+ return QModelIndex()
+
+ idx = action.data()
+ if idx is None:
+ return QModelIndex()
+
+ if not isinstance(idx, QModelIndex):
+ return QModelIndex()
+
+ return idx
+
+ def _createMenu(self):
+ """
+ Protected method to create the menu for a tool bar action.
+
+ @return menu for a tool bar action (E5ModelMenu)
+ """
+ return E5ModelMenu(self)
+
+ def eventFilter(self, obj, evt):
+ """
+ Public method to handle event for other objects.
+
+ @param obj reference to the object (QObject)
+ @param evt reference to the event (QEvent)
+ @return flag indicating that the event should be filtered out (boolean)
+ """
+ if evt.type() == QEvent.MouseButtonRelease:
+ self._mouseButton = evt.button()
+ self._keyboardModifiers = evt.modifiers()
+ act = obj.defaultAction()
+ idx = self.index(act)
+ if idx.isValid():
+ self.emit(SIGNAL("activated(const QModelIndex&)"), idx)
+ elif evt.type() == QEvent.MouseButtonPress:
+ if evt.buttons() & Qt.LeftButton:
+ self.__dragStartPosition = self.mapFromGlobal(evt.globalPos())
+
+ return False
+
+ def dragEnterEvent(self, evt):
+ """
+ Protected method to handle drag enter events.
+
+ @param evt reference to the event (QDragEnterEvent)
+ """
+ if self.__model is not None:
+ mimeTypes = self.__model.mimeTypes()
+ for mimeType in mimeTypes:
+ if evt.mimeData().hasFormat(mimeType):
+ evt.acceptProposedAction()
+
+ QToolBar.dragEnterEvent(self, evt)
+
+ def dropEvent(self, evt):
+ """
+ Protected method to handle drop events.
+
+ @param evt reference to the event (QDropEvent)
+ """
+ if self.__model is not None:
+ act = self.actionAt(evt.pos())
+ parentIndex = self.__root
+ if act is None:
+ row = self.__model.rowCount(self.__root)
+ else:
+ idx = self.index(act)
+ assert idx.isValid()
+ row = idx.row()
+ if self.__model.hasChildren(idx):
+ parentIndex = idx
+ row = self.__model.rowCount(idx)
+
+ self.__dropRow = row
+ self.__dropIndex = parentIndex
+ evt.acceptProposedAction()
+ self.__model.dropMimeData(evt.mimeData(), evt.dropAction(),
+ row, 0, parentIndex)
+
+ QToolBar.dropEvent(self, evt)
+
+ def mouseMoveEvent(self, evt):
+ """
+ Protected method to handle mouse move events.
+
+ @param evt reference to the event (QMouseEvent)
+ """
+ if self.__model is None:
+ QToolBar.mouseMoveEvent(self, evt)
+ return
+
+ if not (evt.buttons() & Qt.LeftButton):
+ QToolBar.mouseMoveEvent(self, evt)
+ return
+
+ manhattanLength = (evt.pos() - self.__dragStartPosition).manhattanLength()
+ if manhattanLength <= QApplication.startDragDistance():
+ QToolBar.mouseMoveEvent(self, evt)
+ return
+
+ act = self.actionAt(self.__dragStartPosition)
+ if act is None:
+ QToolBar.mouseMoveEvent(self, evt)
+ return
+
+ idx = self.index(act)
+ assert idx.isValid()
+
+ drag = QDrag(self)
+ drag.setMimeData(self.__model.mimeData([idx]))
+ actionRect = self.actionGeometry(act)
+ drag.setPixmap(QPixmap.grabWidget(self, actionRect))
+
+ if drag.exec_() == Qt.MoveAction:
+ row = idx.row()
+ if self.__dropIndex == idx.parent() and self.__dropRow <= row:
+ row += 1
+ self.__model.removeRow(row, self.__root)
+
+ def hideEvent(self, evt):
+ """
+ Protected method to handle hide events.
+
+ @param evt reference to the hide event (QHideEvent)
+ """
+ self.clear()
+ QToolBar.hideEvent(self, evt)
+
+ def showEvent(self, evt):
+ """
+ Protected method to handle show events.
+
+ @param evt reference to the hide event (QHideEvent)
+ """
+ if len(self.actions()) == 0:
+ self._build()
+ QToolBar.showEvent(self, evt)
+
+ def resetFlags(self):
+ """
+ Public method to reset the saved internal state.
+ """
+ self._mouseButton = Qt.NoButton
+ self._keyboardModifiers = Qt.KeyboardModifiers(Qt.NoModifier)
\ No newline at end of file
