Files
Jungfraujoch/viewer/widgets/JFJochAzIntImageView.cpp

95 lines
2.8 KiB
C++

// SPDX-FileCopyrightText: 2025 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
// SPDX-License-Identifier: GPL-3.0-only
#include "JFJochAzIntImageView.h"
#include <QVBoxLayout>
#include "../../common/JFJochException.h"
JFJochAzIntImageView::JFJochAzIntImageView(QWidget *parent)
: JFJochImage(parent) {
}
void JFJochAzIntImageView::Clear() {
W = 0;
H = 0;
image_fp.clear();
if (scene())
scene()->clear();
}
// data: size = azimuthal_bins * q_bins
// Layout: q varies fastest (i % q_bins == q index)
void JFJochAzIntImageView::SetData(const std::vector<float> &data,
const std::vector<float> &in_phi,
const std::vector<float> &in_q,
int azimuthal_bins) {
if (azimuthal_bins <= 0) {
Clear();
throw std::runtime_error("azimuthal_bins <= 0");
}
int q_bins = data.size() / azimuthal_bins;
if (q_bins <= 0 || in_phi.size() != data.size() || in_q.size() != data.size()) {
Clear();
throw JFJochException(JFJochExceptionCategory::InputParameterInvalid,
"Mismatch in input size");
}
float local_min = range_min, local_max = range_max;
if (auto_range) {
local_min = std::numeric_limits<float>::infinity();
local_max = -std::numeric_limits<float>::infinity();
for (float v : data) {
if (std::isfinite(v)) {
if (v < local_min) local_min = v;
if (v > local_max) local_max = v;
}
}
if (!std::isfinite(local_min) || !std::isfinite(local_max) || local_max <= local_min) {
Clear();
return;
}
} else {
if (!(std::isfinite(local_min) && std::isfinite(local_max) && local_max > local_min)) {
Clear();
return;
}
}
phi = in_phi;
q = in_q;
// Update base class members
W = q_bins;
H = azimuthal_bins;
image_fp = data;
// Update foreground/background for color mapping
background = local_min;
foreground = local_max;
// Generate pixmap and redraw using base class functionality
GeneratePixmap();
Redraw();
}
void JFJochAzIntImageView::mouseHover(QMouseEvent* event) {
if (!scene() || W == 0 || H == 0) return;
QPointF scenePos = mapToScene(event->pos());
int x = static_cast<int>(scenePos.x());
int y = static_cast<int>(scenePos.y());
if (x >= 0 && x < static_cast<int>(W) && y >= 0 && y < static_cast<int>(H)) {
size_t idx = y * W + x;
QString statusText = QString("Q: %1 Å^-1 phi: %2° value: %3")
.arg(QString::number(q[idx], 'f', 3))
.arg(QString::number(phi[idx], 'f', 3))
.arg(QString::number(image_fp[idx], 'f', 3));
emit writeStatusBar(statusText, 0);
}
}