Jungfraujoch/viewer/widgets/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:
    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 = {};
    bool res_ring_auto = false;
    bool show_spots = true;
    bool show_predictions = false;

    bool highlight_ice_rings = true;

    float ice_ring_width_Q_recipA = 0.01;

    void mouseHover(QMouseEvent* event) override;
public slots:
    void loadImage(std::shared_ptr<const JFJochReaderImage> image);
    void setAutoForeground(bool input);

    void setResolutionRing(QVector<float> v);
    void setResolutionRingAuto();
    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);
};