29 lines
945 B
C++
29 lines
945 B
C++
// 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;
|
|
public:
|
|
AzIntEngineGPU(const AzimuthalIntegrationMapping& integration, std::shared_ptr<CudaStream> stream);
|
|
void Run(const ImagePreprocessorBuffer &image, AzimuthalIntegrationProfile &profile) override;
|
|
};
|