#ifndef CUSTOMMODEL_H
#define CUSTOMMODEL_H

#include <QAbstractTableModel>
#include <QVector>
#include <QString>
#include <QStyledItemDelegate>
#include <QTableView>
#include <QMainWindow>
#include <QApplication>
#include <QHeaderView>
#include <QPainter>



class CustomModel : public QAbstractTableModel {
public:
    explicit CustomModel(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 = Qt::DisplayRole) const override;
    bool setData(const QModelIndex &index, const QVariant &value, int role = Qt::EditRole) override;
    Qt::ItemFlags flags(const QModelIndex &index) const override;
private:
    QVector<QVector<QString>> tableData;
};

class CustomTableDelegate : public QStyledItemDelegate {
public:
    explicit CustomTableDelegate(QObject *parent = nullptr);
    void paint(QPainter *painter, const QStyleOptionViewItem &option, const QModelIndex &index) const override;
};


class CustomTableView : public QTableView {
public:
    explicit CustomTableView(QWidget *parent = nullptr);
};





#endif // CUSTOMMODEL_H


CustomModel::CustomModel(QObject *parent)
    : QAbstractTableModel(parent) {
    tableData = {
        {"Item 1-1", "Item 1-2", "Item 1-3"},
        {"Item 2-1", "Item 2-2", "Item 2-3"},
        {"Item 3-1", "Item 3-2", "Item 3-3"}
    };
}

int CustomModel::rowCount(const QModelIndex &) const {
    return tableData.size();
}

int CustomModel::columnCount(const QModelIndex &) const {
    return tableData.isEmpty() ? 0 : tableData[0].size();
}

QVariant CustomModel::data(const QModelIndex &index, int role) const {
    if (!index.isValid())
        return QVariant();

    if (role == Qt::DisplayRole || role == Qt::EditRole) {
        return tableData[index.row()][index.column()];
    }

    return QVariant();
}

bool CustomModel::setData(const QModelIndex &index, const QVariant &value, int role) {
    if (index.isValid() && role == Qt::EditRole) {
        tableData[index.row()][index.column()] = value.toString();
        emit dataChanged(index, index);
        return true;
    }
    return false;
}

Qt::ItemFlags CustomModel::flags(const QModelIndex &index) const {
    if (!index.isValid())
        return Qt::NoItemFlags;

    return Qt::ItemIsSelectable | Qt::ItemIsEditable | Qt::ItemIsEnabled;
}
CustomTableDelegate::CustomTableDelegate(QObject *parent) : QStyledItemDelegate(parent) {}

void CustomTableDelegate::paint(QPainter *painter, const QStyleOptionViewItem &option, const QModelIndex &index) const {
    painter->save();

    if (index.row() % 2 == 0) {
        painter->fillRect(option.rect, QColor(220, 220, 220));
    } else {
        painter->fillRect(option.rect, Qt::white);
    }

    painter->drawText(option.rect, Qt::AlignCenter, index.data().toString());

    painter->restore();
}

CustomTableView::CustomTableView(QWidget *parent) : QTableView(parent) {
    setAlternatingRowColors(true);
    setSelectionBehavior(QAbstractItemView::SelectRows);
    setSelectionMode(QAbstractItemView::SingleSelection);
    horizontalHeader()->setSectionResizeMode(QHeaderView::Stretch);
}



int main(int argc, char *argv[]) {
    QApplication app(argc, argv);

    auto *tableView = new CustomTableView;
    auto *model = new CustomModel(tableView);
    auto *delegate = new CustomTableDelegate(tableView);

    tableView->setModel(model);
    tableView->setItemDelegate(delegate);

    tableView->show();


    return app.exec();
}