diff -r 31463df17fd5 -r b5c84934de9c E4Gui/E4ModelToolBar.py
--- a/E4Gui/E4ModelToolBar.py Tue Jan 12 18:39:15 2010 +0000
+++ /dev/null Thu Jan 01 00:00:00 1970 +0000
@@ -1,289 +0,0 @@
-# -*- 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 .E4ModelMenu import E4ModelMenu
-
-class E4ModelToolBar(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 (E4ModelMenu)
- """
- return E4ModelMenu(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)
