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

#include <cmath>
#include "ROIAzimuthal.h"
#include "JFJochException.h"

ROIAzimuthal::ROIAzimuthal(const std::string &in_name, float in_d_min_A, float in_d_max_A)
        : ROIElement(in_name) {
    if ((in_d_min_A <= 0) || (in_d_max_A <= 0))
        throw JFJochException(JFJochExceptionCategory::InputParameterInvalid,
                              "Resolution cannot be zero or negative");
    if (in_d_min_A > in_d_max_A) {
        d_max_A = in_d_min_A;
        d_min_A = in_d_max_A;
    } else {
        d_max_A = in_d_max_A;
        d_min_A = in_d_min_A;
    }
}

float ROIAzimuthal::GetDMin_A() const {
    return d_min_A;
}

float ROIAzimuthal::GetDMax_A() const {
    return d_max_A;
}

void ROIAzimuthal::MarkROI(std::vector<uint16_t> &v, uint16_t value_to_mark, int64_t xpixel, int64_t ypixel,
                           const std::vector<float> &resolution_map) const {
    if (v.size() != xpixel * ypixel)
        throw JFJochException(JFJochExceptionCategory::InputParameterInvalid,
                              "ROIRectangle: MarkROI mismatch in input array size");

    if (resolution_map.size() != v.size())
        throw JFJochException(JFJochExceptionCategory::InputParameterInvalid,
                              "ROIRectangle: MarkROI mismatch in resolution array size");

    if (value_to_mark >= 16)
        throw JFJochException(JFJochExceptionCategory::InputParameterInvalid,
                              "ROIRectangle: Only 16-bit available for ROI");

    for (int64_t y = 0; y <= ypixel * xpixel; y++) {
            if (resolution_map[y] >= d_min_A && resolution_map[y] <= d_max_A)
                v[y] |= (1<<value_to_mark);
    }
}

float ROIAzimuthal::GetQMax_recipA() const {
    return 2.0f * M_PI / d_min_A;
}

float ROIAzimuthal::GetQMin_recipA() const {
    return 2.0f * M_PI / d_max_A;
}

ROIConfig ROIAzimuthal::ExportMetadata() const {
    double qmin = GetQMin_recipA();
    double qmax = GetQMax_recipA();
    return ROIConfig{
        .type = ROIConfig::ROIType::Azim,
        .name = name,
        .azim = ROIConfigAzim{.qmin = qmin, .qmax = qmax}
    };
}