swmmobjecttreemodel.h

Go to the documentation of this file.

 
#ifndef SWMMOBJECTTREEMODEL_H
#define SWMMOBJECTTREEMODEL_H
 
#include <QAbstractItemModel>
#include <QPointer>
#include <QVector>
 
#include "layers/swmmmodellayer.h"
#include "selection/selectionmanager.h"
 
class SWMMObjectTreeModel : public QAbstractItemModel
{
    Q_OBJECT
 
public:
    // Extra data-role id's used by the panel when translating model
    // indices into SWMMObjectRef / (Category, row) tuples. Kept outside
    // Qt::UserRole so nothing collides with the delegate's own roles.
    enum Role {
        RoleCategory   = Qt::UserRole + 10,   // int(SWMMModelLayer::Category)
        RoleRow        = Qt::UserRole + 11,   // row within category
        RoleIsLeaf     = Qt::UserRole + 12,   // bool
        RoleObjectRef  = Qt::UserRole + 13,   // QVariant::fromValue<SWMMObjectRef>
        // Slice BM.0 — non-spatial data-objects section.
        RoleSection      = Qt::UserRole + 14, // int(Section)
        RoleDataCategory = Qt::UserRole + 15, // int(SWMMModelLayer::DataCategory)
    };
 
    enum Section : int {
        SectionNetwork = 0, 
        SectionDivider = 1, 
        SectionData    = 2, 
    };
 
    explicit SWMMObjectTreeModel(QObject *parent = nullptr);
    ~SWMMObjectTreeModel() override;
 
    void setLayer(SWMMModelLayer *layer);
 
    SWMMModelLayer::Category categoryAtTopRow(int topRow) const;
 
    int topRowForCategory(SWMMModelLayer::Category c) const;
 
    QModelIndex indexFor(SWMMModelLayer::Category c, int row) const;
 
    QModelIndex indexFor(const SWMMObjectRef &ref) const;
 
    Section sectionAtTopRow(int topRow) const;
 
    SWMMModelLayer::DataCategory dataCategoryAtTopRow(int topRow) const;
 
    int topRowForDataCategory(SWMMModelLayer::DataCategory c) const;
 
    // ── QAbstractItemModel interface ─────────────────────────────────────
    QModelIndex index(int row, int column,
                      const QModelIndex &parent = {}) const override;
    QModelIndex parent(const QModelIndex &child) const override;
    int         rowCount(const QModelIndex &parent = {}) const override;
    int         columnCount(const QModelIndex &parent = {}) const override;
    QVariant    data(const QModelIndex &index,
                     int role = Qt::DisplayRole) const override;
    bool        setData(const QModelIndex &index,
                        const QVariant &value,
                        int role = Qt::EditRole) override;
    Qt::ItemFlags flags(const QModelIndex &index) const override;
    QVariant    headerData(int section, Qt::Orientation orientation,
                           int role = Qt::DisplayRole) const override;
 
    // Drag-and-drop for category reordering (Slice T.2).
    Qt::DropActions supportedDropActions() const override;
    QStringList     mimeTypes() const override;
    QMimeData      *mimeData(const QModelIndexList &indexes) const override;
    bool            canDropMimeData(const QMimeData *data, Qt::DropAction action,
                                     int row, int column,
                                     const QModelIndex &parent) const override;
    bool            dropMimeData(const QMimeData *data, Qt::DropAction action,
                                  int row, int column,
                                  const QModelIndex &parent) override;
 
public slots:
    void reload();
 
    void scheduleReload();
 
private:
    bool m_reloadPending = false;
 
    // internalId sentinels for top-level rows. Leaf rows store their
    // parent row index as internalId instead — leaf internalId is always
    // < kSeparatorId so the discriminator is unambiguous.
    static constexpr quintptr kCategoryId      = std::numeric_limits<quintptr>::max();
    static constexpr quintptr kDataCategoryId  = std::numeric_limits<quintptr>::max() - 1;
    static constexpr quintptr kSeparatorId     = std::numeric_limits<quintptr>::max() - 2;
 
    QPointer<SWMMModelLayer>             m_layer;
 
    // Visible-category list — excludes categories with 0 members so the
    // tree doesn't show empty "Pumps (0)" / etc. headers. Built on
    // reload(); index i in this vector is the model's top-level row for
    // that category.
    QVector<SWMMModelLayer::Category>    m_visible;
 
    // Slice BM.0 — visible data-object categories (non-spatial). Same
    // contract as m_visible: empty categories are omitted. Rendered
    // below m_visible with a non-selectable separator row between when
    // both sections are non-empty.
    QVector<SWMMModelLayer::DataCategory> m_visibleData;
};
 
#endif // SWMMOBJECTTREEMODEL_H