edgetablemodel.h

Go to the documentation of this file.

 
#pragma once
 
#include <QAbstractTableModel>
#include <QBrush>
#include <QColor>
#include <QStringList>
 
class Graph;

class EdgeTableModel : public QAbstractTableModel
{
    Q_OBJECT
public:
    explicit EdgeTableModel(QObject *parent = nullptr);

    void populate(Graph *graph);

    QStringList attrKeys() const { return m_attrKeys; }
 
    int rowCount(const QModelIndex &parent = QModelIndex()) const override;
    int columnCount(const QModelIndex &parent = QModelIndex()) const override;
    QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override;
    QVariant headerData(int section, Qt::Orientation orientation,
                        int role = Qt::DisplayRole) const override;
    Qt::ItemFlags flags(const QModelIndex &index) const override;
    bool setData(const QModelIndex &index, const QVariant &value,
                 int role = Qt::EditRole) override;
 
private:
    struct EdgeRow {
        int     source;
        int     target;
        QString relation;
        qreal   weight;
        QString label;
        QString color;
        QList<QString> attrValues; 
    };
 
    Graph          *m_graph = nullptr;
    QList<EdgeRow>  m_rows;
    QStringList     m_attrKeys;
 
    static constexpr int FIXED_COLS  = 6;
 
    // Column index constants
    static constexpr int COL_SOURCE   = 0;
    static constexpr int COL_TARGET   = 1;
    static constexpr int COL_RELATION = 2;
    static constexpr int COL_WEIGHT   = 3;
    static constexpr int COL_LABEL    = 4;
    static constexpr int COL_COLOR    = 5;
};