Jungfraujoch/viewer/windows/JFJochMergeStatsWindow.cpp

// SPDX-FileCopyrightText: 2026 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
// SPDX-License-Identifier: GPL-3.0-only

#include "JFJochMergeStatsWindow.h"

#include <cmath>

#include <QButtonGroup>
#include <QComboBox>
#include <QFrame>
#include <QHBoxLayout>
#include <QHeaderView>
#include <QLabel>
#include <QStackedWidget>
#include <QStringList>
#include <QTableWidget>
#include <QToolButton>
#include <QVBoxLayout>
#include <QtCharts/QChart>
#include <QtCharts/QValueAxis>

#include "gemmi/symmetry.hpp"   // find_spacegroup_by_number

#include "../charts/JFJochSimpleChartView.h"
#include "../widgets/ToolbarIcons.h"

namespace {
    // Per-shell metric value (NaN -> dropped by the chart). Keep in sync with the combo entries.
    double ShellMetric(const MergeStatisticsShell &s, int metric) {
        switch (metric) {
            case 0: return s.cc_half * 100.0;                                        // CC1/2 (%)
            case 1: return s.r_meas * 100.0;                                         // R-meas (%)
            case 2: return s.mean_i_over_sigma;                                      // <I/sigma>
            case 3: return s.possible_unique_reflections > 0                          // Completeness (%)
                           ? 100.0 * s.unique_reflections / s.possible_unique_reflections : NAN;
            case 4: return s.unique_reflections > 0                                   // Multiplicity
                           ? static_cast<double>(s.total_observations) / s.unique_reflections : NAN;
            case 5: return s.cc_ref * 100.0;                                         // CCref (%)
            default: return NAN;
        }
    }

    // A big-number "card" for the hero row: large navy value over a small grey caption.
    QWidget *MakeCard(const QString &value, const QString &caption, QWidget *parent) {
        auto *card = new QFrame(parent);
        card->setFrameShape(QFrame::StyledPanel);
        auto *l = new QVBoxLayout(card);
        l->setContentsMargins(12, 8, 12, 8);
        l->setSpacing(0);
        auto *v = new QLabel(value, card);
        QFont f = v->font();
        f.setPointSizeF(f.pointSizeF() * 2.2);
        f.setBold(true);
        v->setFont(f);
        v->setStyleSheet("color: #1F3A5F;");
        v->setAlignment(Qt::AlignCenter);
        auto *c = new QLabel(caption, card);
        c->setStyleSheet("color: gray;");
        c->setAlignment(Qt::AlignCenter);
        l->addWidget(v);
        l->addWidget(c);
        return card;
    }
}

JFJochMergeStatsWindow::JFJochMergeStatsWindow(const QString &title, const MergeStatistics &stats,
                                               double isa, bool has_reference,
                                               const TwinningAnalysisResult &twinning,
                                               std::optional<int64_t> space_group_number,
                                               const std::optional<SearchSpaceGroupResult> &space_group_search,
                                               QWidget *parent)
    : QWidget(parent, Qt::Window), stats_(stats), has_reference_(has_reference) {
    setWindowTitle("Merge statistics — " + title);
    setAttribute(Qt::WA_DeleteOnClose);
    resize(740, 560);

    auto *layout = new QVBoxLayout(this);

    // --- Hero row: overall numbers ---
    const auto &o = stats_.overall;
    const double completeness = o.possible_unique_reflections > 0
        ? 100.0 * o.unique_reflections / o.possible_unique_reflections : NAN;
    const double multiplicity = o.unique_reflections > 0
        ? static_cast<double>(o.total_observations) / o.unique_reflections : NAN;
    auto pct = [](double v) { return std::isfinite(v) ? QString::number(v, 'f', 1) + "%" : QStringLiteral("—"); };
    auto num = [](double v, int d) { return std::isfinite(v) ? QString::number(v, 'f', d) : QStringLiteral("—"); };

    auto *hero = new QHBoxLayout();
    // Resolution range of the merge. When no scaling resolution limit is given, the high-resolution
    // edge is set by the CC1/2 logistic cutoff, so it varies per run - show it first. d_min is the
    // high-res edge (small d), d_max the low-res edge.
    if (o.d_min < o.d_max)
        hero->addWidget(MakeCard(QString::asprintf("%.1f–%.2f", o.d_max, o.d_min), "Resolution [Å]", this));
    hero->addWidget(MakeCard(num(isa, 1), "ISa", this));
    hero->addWidget(MakeCard(pct(o.cc_half * 100.0), "CC½", this));
    hero->addWidget(MakeCard(pct(o.r_meas * 100.0), "R-meas", this));
    hero->addWidget(MakeCard(pct(completeness), "Completeness", this));
    hero->addWidget(MakeCard(num(multiplicity, 1), "Multiplicity", this));
    if (has_reference_)
        hero->addWidget(MakeCard(pct(o.cc_ref * 100.0), "CCref", this));
    if (space_group_number.has_value()) {
        const gemmi::SpaceGroup *sg = gemmi::find_spacegroup_by_number(static_cast<int>(*space_group_number));
        const QString sg_name = sg ? QString::fromStdString(sg->short_name())
                                   : QString::number(*space_group_number);
        hero->addWidget(MakeCard(sg_name, "Space group", this));
    }
    layout->addLayout(hero);

    // De-novo space-group search: the point group + the ranked candidates it scored (only present when
    // the space group was not fixed by the user). A summary line above a compact table.
    if (space_group_search.has_value() && !space_group_search->candidates.empty()) {
        const auto &sgs = *space_group_search;
        QString summary = "Point group " + QString::fromStdString(sgs.point_group_hm.empty() ? "?" : sgs.point_group_hm);
        if (!sgs.alternatives.empty()) {
            QStringList alts;
            for (const auto &alt : sgs.alternatives)
                alts << QString::fromStdString(alt.short_name());
            summary += "  ·  indistinguishable from these data: " + alts.join(", ");
        }
        auto *sgLabel = new QLabel(summary, this);
        sgLabel->setStyleSheet("color: gray;");
        layout->addWidget(sgLabel);

        auto *sgTable = new QTableWidget(this);
        sgTable->setEditTriggers(QAbstractItemView::NoEditTriggers);
        sgTable->verticalHeader()->setVisible(false);
        const QStringList sgHeaders{"Space group", "Absent", "Viol.", "⟨I/σ⟩ abs", "⟨I/σ⟩ pres", "Consistent"};
        sgTable->setColumnCount(sgHeaders.size());
        sgTable->setHorizontalHeaderLabels(sgHeaders);
        sgTable->setRowCount(static_cast<int>(sgs.candidates.size()));
        for (int i = 0; i < static_cast<int>(sgs.candidates.size()); ++i) {
            const auto &c = sgs.candidates[i];
            int col = 0;
            auto set = [&](const QString &t, bool leftAlign = false) {
                auto *it = new QTableWidgetItem(t);
                it->setTextAlignment((leftAlign ? Qt::AlignLeft : Qt::AlignRight) | Qt::AlignVCenter);
                if (c.selected) { QFont f = it->font(); f.setBold(true); it->setFont(f); }
                sgTable->setItem(i, col++, it);
            };
            set(QString::fromStdString(c.space_group.short_name()) + (c.selected ? " *" : ""), true);
            set(QString::number(c.absent_observed));
            set(QString::number(c.absent_violations));
            set(QString::number(c.absent_mean_i_over_sigma, 'f', 2));
            set(QString::number(c.present_mean_i_over_sigma, 'f', 2));
            set(c.consistent ? QStringLiteral("yes") : QStringLiteral("no"));
        }
        sgTable->horizontalHeader()->setSectionResizeMode(QHeaderView::Stretch);
        sgTable->setMaximumHeight(160);   // a secondary detail — keep the shells plot the focus
        layout->addWidget(sgTable);
    }

    // --- Twinning test (Padilla-Yeates L-test + second moment) ---
    // A red-outlined banner flags suspected twinning; otherwise a quiet grey "all clear".
    if (twinning.l_test_pairs > 0) {
        QString text = QString("Twinning:  ⟨|L|⟩ = %1   ⟨I²⟩/⟨I⟩² = %2")
                           .arg(twinning.mean_abs_l, 0, 'f', 3)
                           .arg(twinning.second_moment, 0, 'f', 2);
        if (twinning.twinning_suspected)
            text += QString("   —  twinning suspected (estimated fraction ≈ %1)")
                        .arg(twinning.estimated_twin_fraction, 0, 'f', 2);
        else
            text += "   —  no twinning indicated";

        auto *banner = new QLabel(text, this);
        banner->setContentsMargins(10, 6, 10, 6);
        if (twinning.twinning_suspected)
            banner->setStyleSheet("color: rgb(160,0,0); border: 1px solid red; border-radius: 4px;");
        else
            banner->setStyleSheet("color: gray;");
        layout->addWidget(banner);
    }

    // --- Controls: plot/table view toggle (icon buttons) + per-resolution metric selector ---
    auto *row = new QHBoxLayout();
    auto *plotBtn = new QToolButton(this);
    plotBtn->setIcon(ToolbarIcons::linePlot());
    plotBtn->setCheckable(true);
    plotBtn->setChecked(true);
    plotBtn->setToolTip("Per-resolution plot");
    plotBtn->setStyleSheet(ToolbarIcons::buttonStyle());
    auto *tableBtn = new QToolButton(this);
    tableBtn->setIcon(ToolbarIcons::table());
    tableBtn->setCheckable(true);
    tableBtn->setToolTip("Per-shell table");
    tableBtn->setStyleSheet(ToolbarIcons::buttonStyle());
    auto *viewGroup = new QButtonGroup(this);
    viewGroup->setExclusive(true);
    viewGroup->addButton(plotBtn, 0);
    viewGroup->addButton(tableBtn, 1);
    row->addWidget(plotBtn);
    row->addWidget(tableBtn);
    row->addSpacing(12);
    metricLabel_ = new QLabel("Metric:", this);
    row->addWidget(metricLabel_);
    metric_ = new QComboBox(this);
    metric_->addItem("CC½ (%)", 0);
    metric_->addItem("R-meas (%)", 1);
    metric_->addItem("⟨I/σ⟩", 2);
    metric_->addItem("Completeness (%)", 3);
    metric_->addItem("Multiplicity", 4);
    if (has_reference_)
        metric_->addItem("CCref (%)", 5);
    row->addWidget(metric_);
    row->addStretch();
    layout->addLayout(row);

    // --- Stacked plot / table ---
    stack_ = new QStackedWidget(this);
    chart_ = new JFJochSimpleChartView(this);
    table_ = new QTableWidget(this);
    table_->setEditTriggers(QAbstractItemView::NoEditTriggers);
    table_->verticalHeader()->setVisible(false);
    stack_->addWidget(chart_);   // page 0
    stack_->addWidget(table_);   // page 1
    layout->addWidget(stack_, 1);

    connect(metric_, &QComboBox::currentIndexChanged, this, [this](int) { updatePlot(); });
    connect(viewGroup, &QButtonGroup::idClicked, this, [this](int idx) {
        stack_->setCurrentIndex(idx);
        metric_->setVisible(idx == 0);        // the metric selector + its label only apply to the plot
        metricLabel_->setVisible(idx == 0);
    });

    buildTable();
    updatePlot();
}

void JFJochMergeStatsWindow::updatePlot() {
    const int metric = metric_->currentData().toInt();
    std::vector<float> x, y;
    double dmax = 0.0;
    x.reserve(stats_.shells.size());
    y.reserve(stats_.shells.size());
    for (const auto &s: stats_.shells) {
        x.push_back(s.mean_one_over_d2);
        const double v = ShellMetric(s, metric);
        y.push_back(static_cast<float>(v));
        if (std::isfinite(v))
            dmax = std::max(dmax, v);
    }
    chart_->UpdateData(x, y, "Resolution [Å]", metric_->currentText(), true);

    // Absolute y-axis so positions are comparable: CC1/2 + CCref are 0..100, completeness 0..(>=100),
    // multiplicity / R-meas / I-sigma start at 0. Show the labels (UpdateData hides them by default).
    double ymin = 0.0, ymax = 1.0;
    switch (metric) {
        case 0: case 5: ymax = 100.0; break;                              // CC1/2, CCref
        case 3: ymax = std::max(100.0, dmax * 1.05); break;               // Completeness
        default: ymax = dmax > 0.0 ? dmax * 1.05 : 1.0; break;            // R-meas, I/sigma, multiplicity
    }
    const auto axes = chart_->chart()->axes(Qt::Vertical);
    if (!axes.isEmpty()) {
        if (auto *vax = qobject_cast<QValueAxis *>(axes.first())) {
            vax->setRange(ymin, ymax);
            vax->setLabelsVisible(true);
        }
    }
}

void JFJochMergeStatsWindow::buildTable() {
    QStringList headers{"Res. [Å]", "N_obs", "Unique", "Compl. %", "Mult.", "⟨I/σ⟩", "R-meas %", "CC½ %"};
    if (has_reference_)
        headers << "CCref %";
    table_->setColumnCount(headers.size());
    table_->setHorizontalHeaderLabels(headers);
    table_->setRowCount(static_cast<int>(stats_.shells.size()) + 1);   // shells + overall

    auto fillRow = [&](int rowIdx, const MergeStatisticsShell &s, const QString &resLabel) {
        int c = 0;
        auto set = [&](const QString &t) {
            auto *it = new QTableWidgetItem(t);
            it->setTextAlignment(Qt::AlignRight | Qt::AlignVCenter);   // numbers right-aligned
            table_->setItem(rowIdx, c++, it);
        };
        const double compl_ = s.possible_unique_reflections > 0
            ? 100.0 * s.unique_reflections / s.possible_unique_reflections : NAN;
        const double mult = s.unique_reflections > 0
            ? static_cast<double>(s.total_observations) / s.unique_reflections : NAN;
        set(resLabel);
        set(QString::number(s.total_observations));
        set(QString::number(s.unique_reflections));
        set(std::isfinite(compl_) ? QString::number(compl_, 'f', 1) : QStringLiteral("—"));
        set(std::isfinite(mult) ? QString::number(mult, 'f', 1) : QStringLiteral("—"));
        set(QString::number(s.mean_i_over_sigma, 'f', 1));
        set(QString::number(s.r_meas * 100.0, 'f', 1));
        set(QString::number(s.cc_half * 100.0, 'f', 1));
        if (has_reference_)
            set(QString::number(s.cc_ref * 100.0, 'f', 1));
    };

    int rowIdx = 0;
    for (const auto &s: stats_.shells)
        fillRow(rowIdx++, s, QString::number(s.d_min, 'f', 2));
    fillRow(rowIdx, stats_.overall, QStringLiteral("Overall"));

    // Equal-width columns that fill the table (no super-wide last column).
    table_->horizontalHeader()->setSectionResizeMode(QHeaderView::Stretch);
}