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

#include "AzIntCPU.h"

AzIntCPU::AzIntCPU(const AzimuthalIntegration &integration)
    : integration(integration) {}

void AzIntCPU::Run(const std::vector<int32_t> &image, AzimuthalIntegrationProfile &profile) const {
    const auto azint_bins = integration.GetBinNumber();

    std::vector<float> azint_sum(azint_bins);
    std::vector<uint32_t> azint_count(azint_bins);

    for (int i = 0; i < azint_count.size(); i++) {
        azint_sum[i] = 0.0f;
        azint_count[i] = 0;
    }

    if (image.size() != integration.GetPixelToBin().size())
        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_count[bin];
        }
    }
    profile.Clear(integration);
    profile.Add(azint_sum, azint_count);
}