Jungfraujoch/image_analysis/azint/AzIntEngineGPU.h

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

#pragma once

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

class AzIntEngineGPU : public AzIntEngine {
    std::shared_ptr<CudaStream> stream;
    int threads;
    int blocks;
    size_t shared_needed;
    size_t shared_size;

    CudaDevicePtr<float> gpu_azint_correction;
    CudaDevicePtr<uint16_t> gpu_pixel_to_bin;

    CudaDevicePtr<float> gpu_sum;
    CudaDevicePtr<float> gpu_sum2;
    CudaDevicePtr<uint32_t> gpu_count;
    CudaRegisteredVector<float> cpu_sum_reg;
    CudaRegisteredVector<float> cpu_sum2_reg;
    CudaRegisteredVector<uint32_t> cpu_count_reg;

    CudaDevicePtr<int32_t> gpu_image;
public:
    AzIntEngineGPU(const AzimuthalIntegration& integration, std::shared_ptr<CudaStream> stream);
    void Run(const std::vector<int32_t> &image, AzimuthalIntegrationProfile &profile) override;
};