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

#ifndef JUNGFRAUJOCH_AZIMUTHALINTEGRATIONMAPPING_H
#define JUNGFRAUJOCH_AZIMUTHALINTEGRATIONMAPPING_H

#include <optional>
#include "DiffractionExperiment.h"
#include "PixelMask.h"

class AzimuthalIntegration {
protected:
    const AzimuthalIntegrationSettings settings;
    const float wavelength;
    const size_t width, height;

    std::vector<float> bin_to_q;
    std::vector<float> bin_to_2theta;
    std::vector<float> bin_to_d;
    std::vector<float> bin_to_phi;
    std::vector<uint16_t> pixel_to_bin;

    std::vector<float> pixel_resolution;
    std::vector<float> corrections;

    std::optional<float> polarization_correction;

    void UpdateMaxBinNumber();
    AzimuthalIntegration(const AzimuthalIntegrationSettings& settings,
                         size_t width, size_t height, float wavelength);

    void SetupRawGeom(const DiffractionExperiment& experiment,
                      const std::vector<uint32_t> &mask);

    void SetupConvGeom(const DiffractionGeometry &geom,
                       const std::vector<uint32_t> &mask);

    void SetupPixel(const DiffractionGeometry &geom, const std::vector<uint32_t> &mask,
                    uint32_t pxl, uint32_t col, uint32_t row);
public:

    AzimuthalIntegration(const DiffractionExperiment& experiment,
                         const PixelMask& mask);

    [[nodiscard]] uint16_t GetBinNumber() const;
    [[nodiscard]] const std::vector<uint16_t>&GetPixelToBin() const;
    [[nodiscard]] const std::vector<float> &GetBinToQ() const;
    [[nodiscard]] const std::vector<float> &GetBinToD() const;
    [[nodiscard]] const std::vector<float> &GetBinToTwoTheta() const;
    [[nodiscard]] const std::vector<float> &GetBinToPhi() const;
    [[nodiscard]] uint16_t QToBin(float q) const;
    [[nodiscard]] const std::vector<float> &Corrections() const;
    [[nodiscard]] const std::vector<float> &Resolution() const;
    [[nodiscard]] size_t GetWidth() const;
    [[nodiscard]] size_t GetHeight() const;
    [[nodiscard]] const AzimuthalIntegrationSettings& Settings() const;
    [[nodiscard]] int32_t GetAzimuthalBinCount() const;
    [[nodiscard]] int32_t GetQBinCount() const;
};


#endif //JUNGFRAUJOCH_AZIMUTHALINTEGRATIONMAPPING_H