CatalogueTreeModel.cpp
201 lines
| 5.2 KiB
| text/x-c
|
CppLexer
| r1259 | #include "Catalogue/CatalogueTreeModel.h" | |||
| r1260 | #include <Catalogue/CatalogueTreeItems/CatalogueAbstractTreeItem.h> | |||
| r1259 | ||||
| r1260 | #include <QMimeData> | |||
| r1259 | #include <memory> | |||
| r1260 | #include <Common/MimeTypesDef.h> | |||
| r1259 | struct CatalogueTreeModel::CatalogueTreeModelPrivate { | |||
| r1260 | std::unique_ptr<CatalogueAbstractTreeItem> m_RootItem = nullptr; | |||
| r1259 | ||||
| r1260 | CatalogueTreeModelPrivate() : m_RootItem{std::make_unique<CatalogueAbstractTreeItem>()} {} | |||
| r1259 | }; | |||
| CatalogueTreeModel::CatalogueTreeModel(QObject *parent) | ||||
| : QAbstractItemModel(parent), impl{spimpl::make_unique_impl<CatalogueTreeModelPrivate>()} | ||||
| { | ||||
| } | ||||
| r1260 | QModelIndex CatalogueTreeModel::addTopLevelItem(CatalogueAbstractTreeItem *item) | |||
| r1259 | { | |||
| r1260 | auto nbTopLevelItems = impl->m_RootItem->children().count(); | |||
| beginInsertRows(QModelIndex(), nbTopLevelItems, nbTopLevelItems); | ||||
| r1259 | impl->m_RootItem->addChild(item); | |||
| endInsertRows(); | ||||
| r1260 | emit dataChanged(QModelIndex(), QModelIndex()); | |||
| r1259 | ||||
| r1260 | return index(nbTopLevelItems, 0); | |||
| r1259 | } | |||
| r1260 | QVector<CatalogueAbstractTreeItem *> CatalogueTreeModel::topLevelItems() const | |||
| r1259 | { | |||
| r1260 | return impl->m_RootItem->children(); | |||
| r1259 | } | |||
| r1260 | void CatalogueTreeModel::addChildItem(CatalogueAbstractTreeItem *child, | |||
| const QModelIndex &parentIndex) | ||||
| r1259 | { | |||
| auto parentItem = item(parentIndex); | ||||
| r1260 | int c = parentItem->children().count(); | |||
| beginInsertRows(parentIndex, c, c); | ||||
| r1259 | parentItem->addChild(child); | |||
| endInsertRows(); | ||||
| r1260 | ||||
| r1261 | emit dataChanged(parentIndex, parentIndex); | |||
| r1259 | } | |||
| r1260 | CatalogueAbstractTreeItem *CatalogueTreeModel::item(const QModelIndex &index) const | |||
| r1259 | { | |||
| r1260 | return static_cast<CatalogueAbstractTreeItem *>(index.internalPointer()); | |||
| r1259 | } | |||
| r1260 | QModelIndex CatalogueTreeModel::indexOf(CatalogueAbstractTreeItem *item, int column) const | |||
| r1259 | { | |||
| auto parentItem = item->parent(); | ||||
| if (!parentItem) { | ||||
| return QModelIndex(); | ||||
| } | ||||
| r1260 | auto row = parentItem->children().indexOf(item); | |||
| r1259 | return createIndex(row, column, item); | |||
| } | ||||
| QModelIndex CatalogueTreeModel::index(int row, int column, const QModelIndex &parent) const | ||||
| { | ||||
| r1260 | if (column > 0) { | |||
| int a = 0; | ||||
| } | ||||
| r1259 | if (!hasIndex(row, column, parent)) { | |||
| return QModelIndex(); | ||||
| } | ||||
| r1260 | CatalogueAbstractTreeItem *parentItem = nullptr; | |||
| r1259 | ||||
| if (!parent.isValid()) { | ||||
| parentItem = impl->m_RootItem.get(); | ||||
| } | ||||
| else { | ||||
| r1260 | parentItem = item(parent); | |||
| r1259 | } | |||
| r1260 | auto childItem = parentItem->children().value(row); | |||
| r1259 | if (childItem) { | |||
| return createIndex(row, column, childItem); | ||||
| } | ||||
| return QModelIndex(); | ||||
| } | ||||
| QModelIndex CatalogueTreeModel::parent(const QModelIndex &index) const | ||||
| { | ||||
| if (!index.isValid()) { | ||||
| return QModelIndex(); | ||||
| } | ||||
| r1260 | auto childItem = item(index); | |||
| r1259 | auto parentItem = childItem->parent(); | |||
| if (parentItem == nullptr || parentItem->parent() == nullptr) { | ||||
| return QModelIndex(); | ||||
| } | ||||
| r1260 | auto row = parentItem->parent()->children().indexOf(parentItem); | |||
| r1259 | return createIndex(row, 0, parentItem); | |||
| } | ||||
| int CatalogueTreeModel::rowCount(const QModelIndex &parent) const | ||||
| { | ||||
| r1260 | CatalogueAbstractTreeItem *parentItem = nullptr; | |||
| r1259 | ||||
| if (!parent.isValid()) { | ||||
| parentItem = impl->m_RootItem.get(); | ||||
| } | ||||
| else { | ||||
| r1260 | parentItem = item(parent); | |||
| r1259 | } | |||
| r1260 | return parentItem->children().count(); | |||
| r1259 | } | |||
| int CatalogueTreeModel::columnCount(const QModelIndex &parent) const | ||||
| { | ||||
| return (int)Column::Count; | ||||
| } | ||||
| Qt::ItemFlags CatalogueTreeModel::flags(const QModelIndex &index) const | ||||
| { | ||||
| r1260 | auto treeItem = item(index); | |||
| if (treeItem) { | ||||
| return treeItem->flags(index.column()); | ||||
| r1259 | } | |||
| return Qt::NoItemFlags; | ||||
| } | ||||
| QVariant CatalogueTreeModel::data(const QModelIndex &index, int role) const | ||||
| { | ||||
| r1260 | auto treeItem = item(index); | |||
| if (treeItem) { | ||||
| return treeItem->data(index.column(), role); | ||||
| r1259 | } | |||
| return QModelIndex(); | ||||
| } | ||||
| bool CatalogueTreeModel::setData(const QModelIndex &index, const QVariant &value, int role) | ||||
| { | ||||
| r1260 | auto treeItem = item(index); | |||
| if (treeItem) { | ||||
| auto result = treeItem->setData(index.column(), role, value); | ||||
| r1259 | ||||
| r1260 | if (result && index.column() == (int)Column::Name) { | |||
| r1259 | emit itemRenamed(index); | |||
| } | ||||
| r1260 | return result; | |||
| r1259 | } | |||
| return false; | ||||
| } | ||||
| r1260 | bool CatalogueTreeModel::canDropMimeData(const QMimeData *data, Qt::DropAction action, int row, | |||
| int column, const QModelIndex &parent) const | ||||
| { | ||||
| auto draggedIndex = parent; | ||||
| auto draggedItem = item(draggedIndex); | ||||
| if (draggedItem) { | ||||
| return draggedItem->canDropMimeData(data, action); | ||||
| } | ||||
| return false; | ||||
| } | ||||
| bool CatalogueTreeModel::dropMimeData(const QMimeData *data, Qt::DropAction action, int row, | ||||
| int column, const QModelIndex &parent) | ||||
| { | ||||
| r1261 | bool result = false; | |||
| auto draggedIndex = parent; | ||||
| auto draggedItem = item(draggedIndex); | ||||
| if (draggedItem) { | ||||
| result = draggedItem->dropMimeData(data, action); | ||||
| if (result) { | ||||
| emit itemDropped(draggedIndex); | ||||
| } | ||||
| } | ||||
| return result; | ||||
| r1260 | } | |||
| Qt::DropActions CatalogueTreeModel::supportedDropActions() const | ||||
| { | ||||
| return Qt::CopyAction | Qt::MoveAction; | ||||
| } | ||||
| QStringList CatalogueTreeModel::mimeTypes() const | ||||
| { | ||||
| return {MIME_TYPE_EVENT_LIST}; | ||||
| } | ||||
