nodetablemodel.h

Go to the documentation of this file.

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

class NodeTableModel : public QAbstractTableModel
{
    Q_OBJECT
public:
    explicit NodeTableModel(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 NodeRow {
        int     number;
        QString label;
        bool    visible;
        QString shape;
        int     size;
        QString color;          
        QList<QString> attrValues; 
    };
 
    Graph          *m_graph = nullptr;
    QList<NodeRow>  m_rows;
    QStringList     m_attrKeys;
 
    static constexpr int FIXED_COLS = 6;
 
    // Column index constants
    static constexpr int COL_NUMBER  = 0;
    static constexpr int COL_LABEL   = 1;
    static constexpr int COL_VISIBLE = 2;
    static constexpr int COL_SHAPE   = 3;
    static constexpr int COL_SIZE    = 4;
    static constexpr int COL_COLOR   = 5;
};