// SPDX-FileCopyrightText: 2025 Filip Leonarski, Paul Scherrer Institute
// SPDX-License-Identifier: GPL-3.0-only

#include "JFJochViewerReflectionListWindow.h"

#include <QVBoxLayout>
#include <QHeaderView>

JFJochViewerReflectionListWindow::JFJochViewerReflectionListWindow(QWidget* parent)
    : QMainWindow(parent) {
    QWidget* centralWidget = new QWidget(this);
    setCentralWidget(centralWidget);

    QVBoxLayout* layout = new QVBoxLayout(centralWidget);

    tableView = new QTableView(this);

    tableModel = new QStandardItemModel(this);
    setupTableModel();

    proxyModel = new QSortFilterProxyModel(this);
    proxyModel->setSourceModel(tableModel);
    proxyModel->setSortRole(Qt::UserRole); // numeric sort on underlying values

    tableView->setModel(proxyModel);
    tableView->sortByColumn(7, Qt::DescendingOrder); // default: I (Intensity) desc
    tableView->setEditTriggers(QAbstractItemView::NoEditTriggers);
    tableView->setSortingEnabled(true);
    tableView->verticalHeader()->setVisible(false);
    tableView->horizontalHeader()->setSectionsClickable(true);
    tableView->horizontalHeader()->setSortIndicatorShown(true);
    tableView->setStyleSheet("background-color: white;");
    tableView->setSelectionBehavior(QAbstractItemView::SelectRows);
    tableView->setSelectionMode(QAbstractItemView::SingleSelection);

    tableView->horizontalHeader()->setSectionResizeMode(QHeaderView::Stretch);

    layout->addWidget(tableView);

    setWindowTitle("Reflections");
    resize(900, 600);

    connect(tableView, &QTableView::doubleClicked,
            this, &JFJochViewerReflectionListWindow::onTableRowDoubleClicked);
}

void JFJochViewerReflectionListWindow::setupTableModel() {
    tableModel->setColumnCount(9);
    tableModel->setHeaderData(0,  Qt::Horizontal, "h");
    tableModel->setHeaderData(1,  Qt::Horizontal, "k");
    tableModel->setHeaderData(2,  Qt::Horizontal, "l");
    tableModel->setHeaderData(3,  Qt::Horizontal, "Pred. x");
    tableModel->setHeaderData(4,  Qt::Horizontal, "Pred. y");
    tableModel->setHeaderData(5,  Qt::Horizontal, "d [Å]");
    tableModel->setHeaderData(6,  Qt::Horizontal, "I");
    tableModel->setHeaderData(7, Qt::Horizontal, "Sigma");
    tableModel->setHeaderData(8,  Qt::Horizontal, "Bkg");
}

static QStandardItem* mkNumItem(const QVariant& userValue, const QString& displayText = QString()) {
    auto* it = new QStandardItem();
    if (displayText.isEmpty()) {
        it->setData(userValue, Qt::DisplayRole);
    } else {
        it->setData(displayText, Qt::DisplayRole);
    }
    it->setData(userValue, Qt::UserRole);
    it->setTextAlignment(Qt::AlignRight | Qt::AlignVCenter);
    return it;
}

void JFJochViewerReflectionListWindow::addReflectionRow(int index, const Reflection& r) {
    QList<QStandardItem*> row;

    row.append(mkNumItem(static_cast<qlonglong>(r.h)));
    row.append(mkNumItem(static_cast<qlonglong>(r.k)));
    row.append(mkNumItem(static_cast<qlonglong>(r.l)));
    row.append(mkNumItem(static_cast<double>(r.predicted_x),
                         QString::number(r.predicted_x, 'f', 2)));
    row.append(mkNumItem(static_cast<double>(r.predicted_y),
                         QString::number(r.predicted_y, 'f', 2)));
    row.append(mkNumItem(static_cast<double>(r.d),
                         QString::number(r.d, 'f', 3)));
    row.append(mkNumItem(static_cast<double>(r.I),
    QString::number(r.I, 'f', 1)));
    row.append(mkNumItem(static_cast<double>(r.sigma),
                         QString::number(r.sigma, 'f', 2)));
    row.append(mkNumItem(static_cast<double>(r.bkg),
                         QString::number(r.bkg, 'f', 1)));

    tableModel->appendRow(row);
}


void JFJochViewerReflectionListWindow::open() {
    show();
    raise();
    activateWindow();
}

void JFJochViewerReflectionListWindow::onTableRowDoubleClicked(const QModelIndex& index) {
    if (!index.isValid())
        return;

    QModelIndex src = proxyModel->mapToSource(index);
    int reflIndex = src.row();

    if (!image)
        return;
    if (reflIndex >= 0 && reflIndex < image->ImageData().reflections.size())
        emit reflectionSelected(image->ImageData().reflections[reflIndex].predicted_x,
            image->ImageData().reflections[reflIndex].predicted_y);
}

void JFJochViewerReflectionListWindow::closeEvent(QCloseEvent* event) {
    event->accept();
    emit closing();
}


void JFJochViewerReflectionListWindow::datasetLoaded(std::shared_ptr<const JFJochReaderDataset> dataset) {
    image = {};
    tableModel->removeRows(0, tableModel->rowCount());
}

void JFJochViewerReflectionListWindow::imageLoaded(std::shared_ptr<const JFJochReaderImage> in_image) {
    tableModel->removeRows(0, tableModel->rowCount());
    image = in_image;
    if (image) {
        for (int i = 0; i < image->ImageData().reflections.size(); ++i)
            addReflectionRow(i, image->ImageData().reflections[i]);
    }
}