// Copyright (2019-2023) Paul Scherrer Institute

#include <cmath>

#include "AzimuthalIntegrationMapping.h"
#include "../common/JFJochException.h"
#include "../common/DiffractionGeometry.h"

AzimuthalIntegrationMapping::AzimuthalIntegrationMapping(const DiffractionExperiment& experiment) :
        low_q(experiment.GetLowQForAzimInt_recipA()),
        high_q(experiment.GetHighQForAzimInt_recipA()),
        q_spacing(experiment.GetQSpacingForAzimInt_recipA()),
        pixel_to_bin_raw(experiment.GetModulesNum() * RAW_MODULE_SIZE),
        max_bin_number(0)
{

    if (q_spacing < 0.0)
        throw JFJochException(JFJochExceptionCategory::InputParameterInvalid,
                              "Q-spacing has to be >= 0.0");

    if (low_q >= high_q)
        throw JFJochException(JFJochExceptionCategory::InputParameterInvalid,
                              "Low-Q must be smaller than high-Q");

    if (low_q <= 0)
        throw JFJochException(JFJochExceptionCategory::InputParameterInvalid,
                              "Low-Q must be > 0");

    if (std::ceil((high_q-low_q) / q_spacing ) >= UINT16_MAX)
        throw JFJochException(JFJochExceptionCategory::InputParameterInvalid,
                              "Cannot accommodate more than 65536 rings");

    for (int m = 0; m < experiment.GetModulesNum(); m++) {
        for (int pxl = 0; pxl < RAW_MODULE_SIZE; pxl++) {
            auto [x,y] = RawToConvertedCoordinate(experiment, m, pxl);
            double pixel_q = 2 * M_PI / PxlToRes(experiment, x, y);
            if ((pixel_q < low_q) || (pixel_q >= high_q))
                pixel_to_bin_raw[m * RAW_MODULE_SIZE + pxl] = UINT16_MAX;
            else
                pixel_to_bin_raw[m * RAW_MODULE_SIZE + pxl] = std::floor((pixel_q - low_q) / q_spacing);
        }
    }

    // In principle max bin number is equal to std::floor((high_q - low_q) / q_spacing), but it might be lower than this
    // depending on detector distance and beam center position
    for (const auto &i: pixel_to_bin_raw) {
        if ((i != UINT16_MAX) && (i > max_bin_number))
            max_bin_number = i;
    }

    bin_to_q.resize(max_bin_number + 1);
    for (int i = 0; i < max_bin_number + 1; i++)
        bin_to_q[i] = static_cast<float>(q_spacing * (i + 0.5) + low_q);
}

uint16_t AzimuthalIntegrationMapping::GetBinNumber() const {
    return max_bin_number + 1;
}

const std::vector<uint16_t> &AzimuthalIntegrationMapping::GetPixelToBinMappingRaw() const {
    return pixel_to_bin_raw;
}

const std::vector<float> &AzimuthalIntegrationMapping::GetBinToQ() const {
    return bin_to_q;
}

double AzimuthalIntegrationMapping::QToBin(double q) const {
    return std::min(static_cast<double>(max_bin_number), std::max(0.0, (q - low_q) / q_spacing));
}