AzIntEngine: Azimuthal integration runs on CPU/GPU

2026-04-22 16:05:06 +02:00
parent 6f361afae3
commit b7c5497c2c
11 changed files with 269 additions and 33 deletions
@@ -0,0 +1,35 @@
+// SPDX-FileCopyrightText: 2025 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
+// SPDX-License-Identifier: GPL-3.0-only
+
+#include "AzIntEngineCPU.h"
+
+AzIntEngineCPU::AzIntEngineCPU(const AzimuthalIntegration &integration)
+    : AzIntEngine(integration) {}
+
+void AzIntEngineCPU::Run(const std::vector<int32_t> &image, AzimuthalIntegrationProfile &profile){
+
+    for (int i = 0; i < azint_count.size(); i++) {
+        azint_sum[i] = 0.0f;
+        azint_sum2[i] = 0.0f;
+        azint_count[i] = 0;
+    }
+
+    if (image.size() != npixel)
+        throw std::runtime_error("ImageSpotFinder::AzimIntegration: Mismatch in size");
+
+    const uint16_t *pixel_to_bin = integration.GetPixelToBin().data();
+    const float *corrections = integration.Corrections().data();
+
+    for (int i = 0; i < image.size(); i++) {
+        const uint16_t bin = pixel_to_bin[i];
+        if (bin < azint_bins) {
+            float val = static_cast<float>(image[i]) * corrections[i];
+            azint_sum[bin] += val;
+            azint_sum2[bin] += val * val;
+            ++azint_count[bin];
+        }
+    }
+
+    profile.Clear(integration);
+    profile.Add(azint_sum, azint_count);
+}