meshattributetablemodel.h

Go to the documentation of this file.

 
#ifndef OPENSWMMVIS_UI_PANELS_MESHATTRIBUTETABLEMODEL_H
#define OPENSWMMVIS_UI_PANELS_MESHATTRIBUTETABLEMODEL_H
 
#include "mesh/meshinfil.h"                      // mesh::InfilProvenance
#include "selection/selectionmanager.h"
#include "ui/panels/swmmattributetablemodel.h"   // openswmmvis::ColumnSpec
 
#include <QAbstractTableModel>
#include <QHash>
#include <QPointer>
#include <QRectF>
#include <QVector>
 
class MapCanvas;
class SWMM2DMeshLayer;
class SWMMModelLayer;
 
class MeshAttributeTableModel : public QAbstractTableModel
{
    Q_OBJECT
public:
    enum class Kind { Vertex, Edge, Cell };
 
    explicit MeshAttributeTableModel(QObject *parent = nullptr);
 
    void setSource(SWMM2DMeshLayer *layer, Kind kind);
 
    [[nodiscard]] SWMM2DMeshLayer *layer() const { return m_layer.data(); }
    [[nodiscard]] Kind             kind()  const { return m_kind; }
 
    void setCanvas(MapCanvas *canvas);
 
    void setModelLayer(SWMMModelLayer *layer);
 
    [[nodiscard]] QList<openswmmvis::ColumnSpec> columnSpecs() const
    { return m_columnSpecs; }
 
    [[nodiscard]] SWMMObjectRef refForRow(int row) const;
 
    [[nodiscard]] int rowForRef(const SWMMObjectRef &ref) const;
 
    [[nodiscard]] QRectF elementExtent(int row, bool *ok = nullptr) const;
 
    // QAbstractTableModel ----------------------------------------------------
    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;
 
public slots:
    void reload();
 
signals:
    void objectEdited(const QString &refName);
 
private:
    void rebuildColumns();
    void rebuildEdgeRows();
    void connectLayer(SWMM2DMeshLayer *layer);
 
    void onLayerAttributeChanged(const QString &refName);
 
    [[nodiscard]] bool rowIsBoundaryEdge(int row) const;
 
    [[nodiscard]] bool cellInfilParamApplies(int row, const QByteArray &key) const;
 
    [[nodiscard]] mesh::InfilProvenance cellInfilProvenance(int row) const;
 
    [[nodiscard]] int slotForRow(int row) const;
 
    [[nodiscard]] bool canPick() const;
 
    [[nodiscard]] QVariant pickerRef(const QString &key,
                                     const QString &current) const;
 
    QPointer<SWMM2DMeshLayer> m_layer;
    QPointer<SWMMModelLayer>  m_modelLayer;
    QPointer<MapCanvas>       m_canvas;
    Kind                      m_kind = Kind::Vertex;
 
    QList<openswmmvis::ColumnSpec> m_columnSpecs;
 
    QVector<int>   m_edgeSlots;
    QHash<int,int> m_slotRow;
 
    int m_firstCellParamCol = -1;
};
 
#endif // OPENSWMMVIS_UI_PANELS_MESHATTRIBUTETABLEMODEL_H