Jungfraujoch/image_analysis/MXAnalysisAfterFPGA.h

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

#pragma once

#include "../common/DiffractionExperiment.h"
#include "bragg_prediction/BraggPrediction.h"
#include "indexing/IndexerThreadPool.h"
#include "spot_finding/StrongPixelSet.h"
#include "IndexAndRefine.h"

class MXAnalysisAfterFPGA {
    mutable std::mutex read_from_cpu_mutex;
    const DiffractionExperiment &experiment;
    IndexAndRefine &indexer;
    std::unique_ptr<BraggPrediction> prediction;

    bool find_spots = false;
    std::vector<DiffractionSpot> spots;
    constexpr static const float spot_distance_threshold_pxl = 2.0f;

    std::vector<float> arr_mean;
    std::vector<float> arr_sttdev;
    std::vector<uint32_t> arr_valid_count;
    std::vector<uint32_t> arr_strong_pixel;

    enum class State {Idle, Disabled, Enabled} state = State::Idle;

    std::chrono::duration<double, std::micro> spot_finding_time_total{0.0};
    bool spot_finding_timing_active = false;
public:

    MXAnalysisAfterFPGA(const DiffractionExperiment& experiment, IndexAndRefine &indexer);

    void ReadFromFPGA(const DeviceOutput* output,
                      const SpotFindingSettings& settings,
                      size_t module_number);

    void ReadFromCPU(DeviceOutput *output,
                     const SpotFindingSettings &settings,
                     size_t module_number);

    void Process(DataMessage &message, const SpotFindingSettings& settings);
};