Jungfraujoch/viewer/image_viewer/JFJochDiffractionImage.h

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

#pragma once

#include "JFJochImage.h"
#include "../../reader/JFJochReaderImage.h"

class JFJochDiffractionImage : public JFJochImage {
Q_OBJECT

    QColor spot_color = Qt::green;
    QColor prediction_color = Qt::darkRed;
    QColor ice_ring_color = Qt::cyan;

public:
    enum class RingMode {Auto, Estimation, Manual, None, IceRings};
    Q_ENUM(RingMode)

    JFJochDiffractionImage(QWidget *parent = nullptr);
private:

    void addCustomOverlay() override;
    void LoadImageInternal();
    void DrawResolutionRings();
    void DrawSpots();
    void DrawPredictions();
    void DrawBeamCenter();
    void DrawTopPixels();
    void DrawSaturation();
    void DrawCross(float x, float y, float size, float width, float z = 1);

    std::shared_ptr<const JFJochReaderImage> image;

    int32_t show_highest_pixels = 0;

    QVector<float> res_ring = {};

    RingMode ring_mode = RingMode::Estimation;

    bool show_spots = false;
    bool show_predictions = false;

    bool highlight_ice_rings = true;

    float ice_ring_width_Q_recipA = 0.01;

    void mouseHover(QMouseEvent* event) override;

signals:
    void autoForegroundChanged(bool input);
public slots:
    void loadImage(std::shared_ptr<const JFJochReaderImage> image);
    void setAutoForeground(bool input);

    void setResolutionRing(QVector<float> v);
    void setResolutionRingMode(RingMode mode);

    void showSpots(bool input);
    void showPredictions(bool input);

    void setSpotColor(QColor input);
    void setPredictionColor(QColor input);

    void showHighestPixels(int32_t v);
    void showSaturation(bool input);

    void highlightIceRings(bool input);
    void changeForeground(float val) override;
};