ImageProcessor: Work in progress on GPU code to preprocess images

2026-04-22 21:06:43 +02:00
parent 4ee249fca2
commit f208109717
8 changed files with 313 additions and 76 deletions
@@ -0,0 +1,31 @@
+// 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<int32_t> gpu_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(std::vector<int32_t> &processed_image, const uint8_t *input, T err_value, T sat_value);
+public:
+    ImagePreprocessorGPU(const DiffractionExperiment &experiment, const PixelMask &mask);
+    ImageStatistics Analyze(std::vector<int32_t> &processed_image, const uint8_t *decompressed_image, CompressedImageMode image_mode) override;
+
+    const int32_t* GetImageDevicePtr() const;
+    std::shared_ptr<CudaStream> GetStream() const;
+};
+