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

#ifndef JFJOCH_GONIOMETERAXIS_H
#define JFJOCH_GONIOMETERAXIS_H

#include <optional>
#include <vector>

#include "Coord.h"

class GoniometerAxis {
    std::string name;
    float start;
    float increment;
    Coord axis;
    std::optional<Coord> helical_step;
    std::optional<float> screening_wedge;
public:
    GoniometerAxis(const std::string& name,
                   float start,
                   float increment,
                   const Coord &axis,
                   const std::optional<Coord> &helical_step);

    GoniometerAxis& Axis(const Coord &input);
    GoniometerAxis& ScreeningWedge(const std::optional<float>& input);

    [[nodiscard]] std::string GetName() const;
    [[nodiscard]] float GetStart_deg() const;
    [[nodiscard]] float GetIncrement_deg() const;
    [[nodiscard]] std::optional<float> GetScreeningWedge() const;
    [[nodiscard]] float GetWedge_deg() const;
    [[nodiscard]] float GetAngle_deg(float image_number) const;

    [[nodiscard]] Coord GetAxis() const;
    [[nodiscard]] std::optional<Coord> GetHelicalStep() const;
    [[nodiscard]] Coord GetPosition(int64_t image_number) const;

    [[nodiscard]] std::vector<double> GetAngleContainer(int64_t max_image_number) const;
    [[nodiscard]] std::vector<double> GetAngleContainerEnd(int64_t max_image_number) const;
    [[nodiscard]] std::vector<double> GetXContainer_m(int64_t max_image_number) const;
    [[nodiscard]] std::vector<double> GetYContainer_m(int64_t max_image_number) const;
    [[nodiscard]] std::vector<double> GetZContainer_m(int64_t max_image_number) const;

    [[nodiscard]] std::vector<double> GetAxisVector() const;

    [[nodiscard]] RotMatrix GetTransformationAngle(float angle_deg) const;
};


#endif //JFJOCH_GONIOMETERAXIS_H