#include "conewobble.h"
#include "coordinates.h"

namespace motions {
    WobbleCone::WobbleCone(const double delta, const double eta, const double chi) : BaseMotion("Wobble in Cone", delta, eta), m_angle(chi) {}
    WobbleCone::WobbleCone() : BaseMotion("Wobble in Cone") {}

    void WobbleCone::initialize(std::mt19937_64& rng) {
        m_axis = draw_position(rng);
    }

    double WobbleCone::jump(std::mt19937_64& rng) {
        const double real_angle = m_uni_dist(rng) * m_angle;
        const double phi = 2 * M_PI * m_uni_dist(rng);
        return omega_q(rotate(m_axis, real_angle, phi));
    }

    std::unique_ptr<BaseMotion> WobbleCone::clone() const {
        return std::make_unique<WobbleCone>(*this);
    }

    void WobbleCone::setParameters(const std::unordered_map<std::string, double> &parameters) {
        BaseMotion::setParameters(parameters);
        m_angle = parameters.at("angle");
    }

    std::unordered_map<std::string, double> WobbleCone::getParameters() const {
        auto parameter =  BaseMotion::getParameters();
        parameter["angle"] = m_angle;

        return parameter;
    }

    std::string WobbleCone::toString() const {
        return std::string("ConeWobble/angle=") + std::to_string(m_angle);
    }
}