Jungfraujoch/viewer/JFJochImageReadingWorker.h

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

#ifndef JFJOCHIMAGEREADINGWORKER_H
#define JFJOCHIMAGEREADINGWORKER_H

#include <QObject>
#include <QString>
#include <QVector>
#include <QRect>
#include <QMutex>

#include "JFJochViewerToolbar.h"

#include "../reader/JFJochHDF5Reader.h"
#include "../common/Logger.h"
#include "../reader/JFJochHttpReader.h"
#include "../image_analysis/MXAnalysisWithoutFPGA.h"

Q_DECLARE_METATYPE(std::shared_ptr<const JFJochReaderDataset>)
Q_DECLARE_METATYPE(std::shared_ptr<const JFJochReaderImage>)
Q_DECLARE_METATYPE(DiffractionExperiment)
Q_DECLARE_METATYPE(SpotFindingSettings)
Q_DECLARE_METATYPE(IndexingSettings)
Q_DECLARE_METATYPE(UnitCell)

class JFJochImageReadingWorker : public QObject {
    Q_OBJECT

    mutable QMutex m;
    bool http_mode = false;

    JFJochHDF5Reader file_reader;
    JFJochHttpReader http_reader;

    QString current_file;

    DiffractionExperiment curr_experiment;
    IndexingSettings indexing_settings;
    std::unique_ptr<IndexerThreadPool> indexing;

    std::shared_ptr<JFJochReaderImage> current_image_ptr;
    std::unique_ptr<AzimuthalIntegration> azint_mapping;
    std::unique_ptr<MXAnalysisWithoutFPGA> image_analysis;

    std::unique_ptr<ROIElement> roi;

    SpotFindingSettings spot_finding_settings;

    std::optional<int64_t> current_image;
    int64_t current_summation = 1;
    int64_t total_images = 0;
    Logger logger{"jfjoch_viewer"};

    bool auto_reanalyze = false;

    void LoadImage_i(int64_t image_number, int64_t summation);
    void CalcROI_i();
    void ReanalyzeImage_i();
    void UpdateDataset_i(const DiffractionExperiment& experiment);
    void UpdateAzint_i(const JFJochReaderDataset *dataset);
signals:
    void datasetLoaded(std::shared_ptr<const JFJochReaderDataset>);
    void imageLoaded(std::shared_ptr<const JFJochReaderImage>);
    void imageStatsUpdated(std::shared_ptr<const JFJochReaderImage>);
    void imageNumberChanged(int64_t total_images, int64_t current_image);
    void setToolbarMode(JFJochViewerToolbar::ToolbarMode input);
    void setRings(const QVector<float> &v);
public:
    JFJochImageReadingWorker(const SpotFindingSettings &settings, const DiffractionExperiment& experiment, QObject *parent = nullptr);
    ~JFJochImageReadingWorker() override = default;
public slots:
    void LoadFile(const QString &filename, qint64 image_number, qint64 summation);
    void CloseFile();
    void LoadImage(int64_t image_number, int64_t summation);
    void SetROIBox(QRect box);
    void SetROICircle(double x, double y, double radius);

    void UpdateDataset(const DiffractionExperiment& experiment);

    void FindCenter(const UnitCell& calibrant, bool guess);

    void Analyze();

    void UpdateSpotFindingSettings(const SpotFindingSettings &settings, const IndexingSettings &indexing, int64_t max_spots, bool reanalyze);
};


#endif //JFJOCHIMAGEREADINGWORKER_H