legendlayertreemodel.h

Go to the documentation of this file.

 
#ifndef OPENSWMMVIS_UI_MODELS_LEGENDLAYERTREEMODEL_H
#define OPENSWMMVIS_UI_MODELS_LEGENDLAYERTREEMODEL_H
 
#include <QAbstractItemModel>
#include <QColor>
#include <QList>
#include <QPointer>
#include <QString>
#include <QVector>
 
class MapCanvas;
class OpenSWMMVisLayer;
 
namespace OpenSWMM::Render { class LegendOverlayStyle; }
 
namespace openswmmvis::ui {
 
class LegendLayerTreeModel : public QAbstractItemModel
{
    Q_OBJECT
public:
    enum Column {
        ColItem  = 0,
        ColColor = 1,
        ColSize  = 2,   
        ColCount
    };
 
    enum CustomRole {
        ClassKeyRole = Qt::UserRole + 1,
        LayerPtrRole,                     
        EditableRole,                     
        SublayerIdRole                    
    };
 
    explicit LegendLayerTreeModel(MapCanvas *canvas, QObject *parent = nullptr);
    ~LegendLayerTreeModel() override;
 
    void setOverlayStyle(OpenSWMM::Render::LegendOverlayStyle *style);
 
    // QAbstractItemModel ─────────────────────────────────────────────
    int columnCount(const QModelIndex &parent = {}) const override;
    int rowCount(const QModelIndex &parent = {}) const override;
    QModelIndex index(int row, int column, const QModelIndex &parent = {}) const override;
    QModelIndex parent(const QModelIndex &child) const override;
    QVariant data(const QModelIndex &idx, int role = Qt::DisplayRole) const override;
    bool setData(const QModelIndex &idx, const QVariant &value, int role = Qt::EditRole) override;
    Qt::ItemFlags flags(const QModelIndex &idx) const override;
    QVariant headerData(int section, Qt::Orientation orientation,
                        int role = Qt::DisplayRole) const override;
 
private slots:
    void onCanvasLayersChanged();   // layerAdded/Removed/OrderChanged
    void onLayerRepaintRequested(); // any layer's renderer state shifted
 
private:
    // Internal nodes: layer headers (top-level) + item rows (one per
    // LegendSymbolItem inside a layer). Stored flat for fast lookup;
    // index parent/child computed via internalId.
    struct ItemRow {
        QString classKey;
        QString label;
        QColor  color;
        bool    visible = true;
        qreal   size    = -1.0;  
        QString sublayerId;      
    };
    struct LayerNode {
        QPointer<OpenSWMMVisLayer> layer;
        QString          name;
        QVector<ItemRow> items;
        bool             editableColor = false;  
        bool             editableSize  = false;  
    };
 
    void rebuildLayerCache();
    void connectLayer(OpenSWMMVisLayer *layer);
    void disconnectAllLayers();
 
    QPointer<MapCanvas>                       m_canvas;
    QPointer<OpenSWMM::Render::LegendOverlayStyle> m_style;
    QVector<LayerNode>                        m_layers;
    QList<QPointer<OpenSWMMVisLayer>>         m_connectedLayers;
    // VS.9 — sublayer-host layers route each sublayer's invalidated()
    // signal here so style/visibility edits on a sublayer refresh the
    // legend live. Tracked as QObject* (ISublayer derives QObject) so the
    // header needs no isublayer.h include; disconnected on every rebuild.
    QList<QPointer<QObject>>                  m_connectedSublayers;
};
 
} // namespace openswmmvis::ui
 
#endif // OPENSWMMVIS_UI_MODELS_LEGENDLAYERTREEMODEL_H