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

#pragma once

#include <memory>

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

class ImagePreprocessorBufferGPU : public ImagePreprocessorBuffer {
    CudaDevicePtr<int32_t> gpu_image;
    CudaRegisteredVector<int32_t> buffer_reg;

public:
    explicit ImagePreprocessorBufferGPU(size_t npixel);

    int32_t *getGPUBuffer() override;
    const int32_t *getGPUBuffer() const override;
};