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

#pragma once

#include "ImagePreprocessor.h"
#include "../indexing/CUDAMemHelpers.h"

class ImagePreprocessorGPU : public ImagePreprocessor {
    std::shared_ptr<CudaStream> stream;
    int threads;
    int blocks;
    CudaDevicePtr<uint8_t> gpu_mask;
    CudaDevicePtr<uint8_t> gpu_decompressed_image;
    CudaDevicePtr<ImageStatistics> gpu_stats;

    std::vector<ImageStatistics> cpu_stats;
    CudaRegisteredVector<ImageStatistics> cpu_stats_reg;

    std::vector<int32_t> cpu_image;

    template <class T> ImageStatistics Analyze(ImagePreprocessorBuffer &processed_image, const uint8_t *input, T err_value, T sat_value);
public:
    ImagePreprocessorGPU(const DiffractionExperiment &experiment, const PixelMask &mask, std::shared_ptr<CudaStream> stream);
    ImageStatistics Analyze(ImagePreprocessorBuffer &processed_image, const uint8_t *decompressed_image, CompressedImageMode image_mode) override;
};