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

#ifndef JUNGFRAUJOCH_DIFFRACTIONGEOMETRY_H
#define JUNGFRAUJOCH_DIFFRACTIONGEOMETRY_H

#include "JFJochException.h"
#include "Coord.h"

class DiffractionGeometry {
    float beam_x_pxl = 0.0;
    float beam_y_pxl = 0.0;
    float det_distance_mm = 100.0;
    float pixel_size_mm = 0.075;
    float wavelength_A = 1.0;
    float poni_rot_1 = 0.0f;
    float poni_rot_2 = 0.0f;
    float poni_rot_3 = 0.0f;
    RotMatrix poni_rot;
    void UpdatePoniRotMatrix();
public:
    DiffractionGeometry &BeamX_pxl(float input);
    DiffractionGeometry &BeamY_pxl(float input);
    DiffractionGeometry &DetectorDistance_mm(float input);
    DiffractionGeometry &PixelSize_mm(float input);
    DiffractionGeometry &Wavelength_A(float input);
    DiffractionGeometry &PoniRot1_rad(float input);
    DiffractionGeometry &PoniRot2_rad(float input);
    DiffractionGeometry &PoniRot3_rad(float input);

    [[nodiscard]] float GetBeamX_pxl() const;
    [[nodiscard]] float GetBeamY_pxl() const;
    [[nodiscard]] float GetDetectorDistance_mm() const;
    [[nodiscard]] float GetPixelSize_mm() const;
    [[nodiscard]] float GetWavelength_A() const;
    [[nodiscard]] Coord GetScatteringVector() const;
    [[nodiscard]] float GetPoniRot1_rad() const;
    [[nodiscard]] float GetPoniRot2_rad() const;
    [[nodiscard]] float GetPoniRot3_rad() const;
    [[nodiscard]] std::pair<float, float> GetDirectBeam_pxl() const;

    [[nodiscard]] Coord LabCoord(float x, float y) const;
    [[nodiscard]] Coord DetectorToRecip(float x, float y) const;
    [[nodiscard]] std::pair<float, float> RecipToDector(const Coord &recip) const;
    [[nodiscard]] float TwoTheta_rad(float x, float y) const;
    [[nodiscard]] float Phi_rad(float x, float y) const;
    [[nodiscard]] float PxlToRes(float x, float y) const;
    [[nodiscard]] float PxlToQ(float x, float y) const;
    [[nodiscard]] float PxlToRes(float dist_pxl) const;
    [[nodiscard]] float ResToPxl(float d_A) const;
    [[nodiscard]] Coord ResToPxl(float d_A, float phi) const;
    [[nodiscard]] float DistFromEwaldSphere(const Coord& recip) const;
    [[nodiscard]] float CalcAzIntSolidAngleCorr(float q) const;
    [[nodiscard]] float CalcAzIntSolidAngleCorr(float x, float y) const;
    [[nodiscard]] float CalcAzIntPolarizationCorr(float x, float y, float coeff) const;
    [[nodiscard]] std::pair<float, float> ResPhiToPxl(float d_A, float phi_rad) const;
    [[nodiscard]] Coord ProjectToEwaldSphere(const Coord &p0) const;

    // eq. 18 in https://journals.iucr.org/d/issues/2014/08/00/dz5332/index.html
    [[nodiscard]] float AngleFromEwaldSphere_deg(const Coord &p0) const;

    [[nodiscard]] const RotMatrix& GetPoniRotMatrix() const;
};

#endif //JUNGFRAUJOCH_DIFFRACTIONGEOMETRY_H