mesh2daquifermodel.h

Go to the documentation of this file.

 
#ifndef OPENSWMMVIS_UI_DIALOGS_MESH2DAQUIFERMODEL_H
#define OPENSWMMVIS_UI_DIALOGS_MESH2DAQUIFERMODEL_H
 
#include <QAbstractTableModel>
#include <QList>
#include <QString>
 
#include <openswmm/engine/openswmm_engine.h>
 
namespace openswmmvis::ui {
 
class Mesh2DAquiferModel : public QAbstractTableModel
{
    Q_OBJECT
public:
    enum Column {
        ColScope = 0,   
        ColTarget,      
        ColKs,
        ColZs,
        ColThetaS,
        ColThetaR,
        ColAlpha,
        ColSoil,
        ColClosure,
        ColHg0,
        ColCLoss,
        ColCount
    };
 
    explicit Mesh2DAquiferModel(QObject *parent = nullptr);
 
    [[nodiscard]] static QStringList soilTokens();
    [[nodiscard]] static QStringList closureTokens();
    [[nodiscard]] static int closureCodeAt(int index);
    [[nodiscard]] static int closureIndexOf(int code);
 
    int rowCount(const QModelIndex &parent = QModelIndex()) const override;
    int columnCount(const QModelIndex &parent = QModelIndex()) const override;
    QVariant data(const QModelIndex &index, int role) const override;
    bool setData(const QModelIndex &index, const QVariant &value,
                 int role = Qt::EditRole) override;
    QVariant headerData(int section, Qt::Orientation orientation,
                        int role = Qt::DisplayRole) const override;
    Qt::ItemFlags flags(const QModelIndex &index) const override;
    bool removeRows(int row, int count,
                    const QModelIndex &parent = QModelIndex()) override;
 
    void load(SWMM_Engine engine);
    QString commit(SWMM_Engine engine);
    [[nodiscard]] bool isDirty() const;
 
    int appendRow();
 
    void setUnitLabels(const QString &lengthLabel, const QString &rateLabel);
 
private:
    struct Row {
        int     scope   = 0;      // SWMM_GW2D_SCOPE_*
        QString tag;
        int     cell    = -1;     // 0-based
        double  ks      = 0.5;
        double  zs      = 5.0;
        double  thetaS  = 0.45;
        double  thetaR  = 0.10;
        double  alpha   = 2.0;
        int     soil    = 0;      // SWMM_GW2D_SOIL_*
        bool    soilSet = false;
        int     closure = -1;     // SWMM_GW2D_CLOSURE_*
        bool    closureSet = false;
        double  hg0     = -1.0;   // < 0 = "use the option default"
        double  cLoss   = 0.0;
        double  psiB    = 0.20;
        double  lambda  = 0.40;
        double  vgN     = 1.6;
        double  vgL     = 0.5;
        int     mLayers = -1;
        bool operator==(const Row &o) const;
    };
 
    QList<Row> m_rows;
    QList<Row> m_loaded;
    QString    m_lengthLabel = QStringLiteral("m");
    QString    m_rateLabel   = QStringLiteral("mm/hr");
};
 
class Mesh2DAquiferNodeModel : public QAbstractTableModel
{
    Q_OBJECT
public:
    enum Column { ColNode = 0, ColCell, ColKc, ColDc, ColArea, ColCount };
 
    explicit Mesh2DAquiferNodeModel(QObject *parent = nullptr);
 
    int rowCount(const QModelIndex &parent = QModelIndex()) const override;
    int columnCount(const QModelIndex &parent = QModelIndex()) const override;
    QVariant data(const QModelIndex &index, int role) const override;
    bool setData(const QModelIndex &index, const QVariant &value,
                 int role = Qt::EditRole) override;
    QVariant headerData(int section, Qt::Orientation orientation,
                        int role = Qt::DisplayRole) const override;
    Qt::ItemFlags flags(const QModelIndex &index) const override;
    bool removeRows(int row, int count,
                    const QModelIndex &parent = QModelIndex()) override;
 
    void load(SWMM_Engine engine);
    QString commit(SWMM_Engine engine);
    [[nodiscard]] bool isDirty() const;
    int appendRow(const QString &node, int cell);
 
    void setUnitLabels(const QString &lengthLabel, const QString &rateLabel,
                       const QString &areaLabel);
 
private:
    struct Row {
        QString node;
        int     cell = 0;    // 0-based
        double  kc   = 0.0;  // 0 = direct Darcy, no semi-confining bed
        double  dc   = 0.0;
        double  area = 0.0;  // 0 = the cell's own area
        bool operator==(const Row &o) const;
    };
 
    QList<Row> m_rows;
    QList<Row> m_loaded;
    QString    m_lengthLabel = QStringLiteral("m");
    QString    m_rateLabel   = QStringLiteral("mm/hr");
    QString    m_areaLabel   = QStringLiteral("m²");
};
 
} // namespace openswmmvis::ui
 
#endif // OPENSWMMVIS_UI_DIALOGS_MESH2DAQUIFERMODEL_H