PowerModeler/include/attributeTableModel.h

#ifndef ATTRIBUTETABLEMODEL_H
#define ATTRIBUTETABLEMODEL_H

/**
 * @brief 用来处理attribute数据的Model类
 *
 * 基本功能包括：
 * 1、可以自定义显示数据表中的哪些列
 * 2、可以分页显示并自定义每页展示数量
 * 3、编辑的数据可以突出展示（如加粗、改色）
 * 4、最前方加入一个用于展示行号的列（行号从1开始）
 * 5、点击行号列可以实现选中标识（通过icon）
 * 6、哪一行被编辑，改行的行号指示可以加*进行标识
 *
 */

#include <QAbstractTableModel>
#include <QSqlRecord>
#include "global.h"
#include "messageDialog.h"

class AttributeTableModel : public QAbstractTableModel
{
    Q_OBJECT

public:
    enum EditState
    {
        Clean = 0,
        Modified,
        New,
        Deleted
    };

    struct DataType //数据类型
    {
        int id;
        QString type;
        QString db; //为哪个数据库的可用类型
    };

    explicit AttributeTableModel(const ModelAttributeGroup& modelAttributeGroup
                                 , QObject* parent = nullptr
                                 , const QString& connection = ""
                                 , const QString& tableName = "basic.attribute");
    ~AttributeTableModel();

    QModelIndex index(int row, int column, const QModelIndex& parent) const override;
    QVariant data(const QModelIndex& index, int role) const override;
    QVariant headerData(int section, Qt::Orientation orientation, int role) const override;
    bool setData(const QModelIndex &index, const QVariant &value, int role = Qt::EditRole) override;
    int rowCount(const QModelIndex& parent = QModelIndex()) const override;
    int columnCount(const QModelIndex& parent = QModelIndex()) const override;
    Qt::ItemFlags flags(const QModelIndex &index) const override;

    //分页控制
    void setPageSize(int);
    int pageSize() const;
    void setCurrentPage(int);
    int currentPage() const;
    int totalPages() const;
    void previousPage();
    void nextPage();
    void firstPage();
    void lastPage();

    //数据操作
    //void setTable(const QString&);
    void refresh();
    void insertRecord(int);
    void removeRecord(int);
    void submitChanges(); //提交更改（增、删、改）
    void cancleChanges(); //取消修改

    //展示列控制
    //void setVisibleColumns(const QStringList& columns);

    //others
    QMap<int, DataType> getDataTypes() {return m_dataTypes;}
    void triggerSyncSignal();

signals:
    void syncDataStatus(bool, const PaginationInfo&);
    void showMessage(MessageDialogType,const QString&,const QString&);

private:
    struct FieldInfo //字段信息
    {
        QString originalName; //数据库中的字段命名
        QString displayName; //对外展示的名称
        QString dataType = "unknown type"; //数据类型
    };

    struct RowData
    {
        //QSqlRecord record;
        QVector<QVariant> values;
        QHash<int, bool> cellModified; //记录单元格是否被修改
        EditState state = Clean;
    };

    void iniDisplayField();
    void getDataTypesFromDB();
    void loadPageData(); // 加载当前页数据
    void updateTotalCount(); // 更新总记录数
    QList<RowData> filterRowsByEditState(EditState);
    int getDataTypeID(const QString&);

    QString m_connection;
    QString m_tableName;
    ModelAttributeGroup m_modelAttributeGroup;

    PaginationInfo m_paginationInfo;
    QList<FieldInfo> m_displayField;
    QList<RowData> m_currentPageData;
    QHash<int, RowData> m_modifiedRows; //key:global row number
    QMap<int, DataType> m_dataTypes;
    QHash<QString, int> m_reversalDataTypes; //dataType的反转，用来通过名称快速查找对应id
};

#endif //ATTRIBUTETABLEMODEL_H