filtermanagertreemodel.py

Go to the documentation of this file.

# SPDX-License-Identifier: CC0-1.0
 
try:
    from PyQt6.QtCore import QAbstractItemModel, QModelIndex, Qt
    from PyQt6.QtGui import QPixmap
except:
    from PyQt5.QtCore import QAbstractItemModel, QModelIndex, Qt
    from PyQt5.QtGui import QPixmap
from builtins import i18n
from . import filtermanagertreeitem
 
 
class FilterManagerTreeModel(QAbstractItemModel):
 
    TYPE_COLUMN = 1
    NODE_COLUMN = 3
    DOCUMENT_COLUMN = 4
 
    def __init__(self, uiFilterManager, parent=None):
        super(FilterManagerTreeModel, self).__init__(parent)
 
        self.rootItem = filtermanagertreeitem.FilterManagerTreeItem(
            (i18n("Name"), i18n("Type"), i18n("Thumbnail")))
        self.uiFilterManager = uiFilterManager
        self._loadTreeModel(self.rootItem)
 
    def index(self, row, column, parent):
        if not self.hasIndex(row, column, parent):
            return QModelIndex()
 
        if parent.isValid():
            parentItem = parent.internalPointer()
        else:
            parentItem = self.rootItem
 
        # It's a FilterManagerTreeItem
        childItem = parentItem.child(row)
        if childItem:
            return self.createIndex(row, column, childItem)
        else:
            return QModelIndex()
 
    def parent(self, index):
        if not index.isValid():
            return QModelIndex()
 
        childItem = index.internalPointer()
        parentItem = childItem.parent()
 
        if parentItem == self.rootItem:
            return QModelIndex()
 
        return self.createIndex(parentItem.row(), 0, parentItem)
 
    def rowCount(self, parent):
        if parent.column() > 0:
            return 0
 
        if not parent.isValid():
            parentItem = self.rootItem
        else:
            parentItem = parent.internalPointer()
        return parentItem.childCount()
 
    def columnCount(self, parent):
        if parent.isValid():
            return parent.internalPointer().columnCount()
        else:
            return self.rootItem.columnCount()
 
    def data(self, index, role):
        if not index.isValid():
            return None
 
        item = index.internalPointer()
 
        if role == Qt.ItemDataRole.UserRole + 1:
            return item.data(self.NODE_COLUMNNODE_COLUMN)
 
        if role == Qt.ItemDataRole.UserRole + 2:
            return item.data(self.DOCUMENT_COLUMNDOCUMENT_COLUMN)
 
        if role == Qt.ItemDataRole.UserRole + 3:
            return item.data(self.TYPE_COLUMNTYPE_COLUMN)
 
        if role != Qt.ItemDataRole.DisplayRole and role != Qt.ItemDataRole.DecorationRole:
            return None
 
        return item.data(index.column())
 
    def flags(self, index):
        if not index.isValid():
            return Qt.ItemFlag.NoItemFlags
 
        return Qt.ItemFlag.ItemIsEnabled | Qt.ItemFlag.ItemIsSelectable
 
    def headerData(self, section, orientation, role):
        if orientation == Qt.Orientation.Horizontal and role == Qt.ItemDataRole.DisplayRole:
            return self.rootItem.data(section)
 
        return None
 
    def _loadTreeModel(self, parent):
        for index, document in enumerate(self.uiFilterManager.documents):
            rootNode = document.rootNode()
            columnData = (document.fileName(),
                          "Document",
                          QPixmap.fromImage(document.thumbnail(30, 30)),
                          rootNode, index)
            item = filtermanagertreeitem.FilterManagerTreeItem(
                columnData, parent)
            parent.appendChild(item)
 
            childNodes = rootNode.childNodes()
            if len(childNodes):
                self._addSubNodes(childNodes[::-1], item, index)
 
    def _addSubNodes(self, nodes, parent, documentIndex):
        for node in nodes:
            nodeName = node.name()
            nodeType = node.type()
            columnData = ("Unnamed" if nodeName == '' else nodeName,
                          "Untyped" if nodeType == '' else nodeType,
                          QPixmap.fromImage(node.thumbnail(30, 30)),
                          node, documentIndex)
            item = filtermanagertreeitem.FilterManagerTreeItem(
                columnData, parent)
            parent.appendChild(item)
 
            childNodes = node.childNodes()
            if len(childNodes):
                self._addSubNodes(childNodes[::-1], item, documentIndex)