qualityfunctiontablemodels.h

Go to the documentation of this file.

 
#ifndef OPENSWMMVIS_UI_QUALITYFUNCTIONTABLEMODELS_H
#define OPENSWMMVIS_UI_QUALITYFUNCTIONTABLEMODELS_H
 
#include <QAbstractTableModel>
#include <QStringList>
#include <QStyledItemDelegate>
 
namespace openswmmvis::ui {
 
class QualityFunctionTableModel : public QAbstractTableModel
{
    Q_OBJECT
 
public:
    explicit QualityFunctionTableModel(QObject *parent = nullptr)
        : QAbstractTableModel(parent) {}
 
    void bind(void *engineHandle, int luIndex);
    void refresh();   
 
    int rowCount(const QModelIndex &parent = {}) const override;
 
    [[nodiscard]] void *engineHandle() const { return m_engine; }
    [[nodiscard]] int   landUseIndex() const { return m_luIndex; }
 
protected:
    QString pollutantName(int row) const;
 
    void *m_engine  = nullptr;
    int   m_luIndex = -1;
    int   m_rows    = 0;
};
 
class BuildupTableModel : public QualityFunctionTableModel
{
    Q_OBJECT
 
public:
    enum Column { ColPollutant, ColFunction, ColC1, ColC2, ColC3,
                  ColNormalizer, ColCount };
 
    using QualityFunctionTableModel::QualityFunctionTableModel;
 
    int columnCount(const QModelIndex &parent = {}) const override;
    QVariant data(const QModelIndex &index, int role) const override;
    bool setData(const QModelIndex &index, const QVariant &value,
                 int role) override;
    Qt::ItemFlags flags(const QModelIndex &index) const override;
    QVariant headerData(int section, Qt::Orientation orientation,
                        int role) const override;
 
    static QStringList functionNames();    
    static QStringList normalizerNames();  
};
 
class WashoffTableModel : public QualityFunctionTableModel
{
    Q_OBJECT
 
public:
    enum Column { ColPollutant, ColFunction, ColCoeff, ColExponent,
                  ColSweepEffic, ColBmpEffic, ColCount };
 
    using QualityFunctionTableModel::QualityFunctionTableModel;
 
    int columnCount(const QModelIndex &parent = {}) const override;
    QVariant data(const QModelIndex &index, int role) const override;
    bool setData(const QModelIndex &index, const QVariant &value,
                 int role) override;
    Qt::ItemFlags flags(const QModelIndex &index) const override;
    QVariant headerData(int section, Qt::Orientation orientation,
                        int role) const override;
 
    static QStringList functionNames();    
};
 
class EnumComboDelegate : public QStyledItemDelegate
{
    Q_OBJECT
 
public:
    static constexpr int OptionsRole = Qt::UserRole + 41;
 
    using QStyledItemDelegate::QStyledItemDelegate;
 
    QWidget *createEditor(QWidget *parent, const QStyleOptionViewItem &option,
                          const QModelIndex &index) const override;
    void setEditorData(QWidget *editor, const QModelIndex &index) const override;
    void setModelData(QWidget *editor, QAbstractItemModel *model,
                      const QModelIndex &index) const override;
};
 
} // namespace openswmmvis::ui
 
#endif // OPENSWMMVIS_UI_QUALITYFUNCTIONTABLEMODELS_H