#include "EntityPositionInterpolator.hpp"

#include "world/Transform.hpp"
#include "InterpolatedProperty.hpp"
#include "EntityInterpolation.hpp"

#include <spdlog/spdlog.h>

namespace tw::net {

EntityPositionInterpolator::EntityPositionInterpolator(
    entt::registry* registry, size_t bufferingIntervalInMillis
) : m_registry(registry),
    m_bufferingIntervalInMillis(bufferingIntervalInMillis)
{ }

void EntityPositionInterpolator::register_entity(entt::entity entity) {
    m_registry->emplace<EntityPositionInterpolation>(entity, glm::vec3());
}

void EntityPositionInterpolator::set_position(Clock::time_point time_point, entt::entity entity, glm::vec3 position) {
    auto* interpolation = m_registry->try_get<EntityPositionInterpolation>(entity);
    if(interpolation == nullptr) {
        spdlog::warn("Attempt to add position for non-registered entity {}", (uint32_t)entity);
        return;
    }

    interpolation->push(time_point, position);
}

glm::vec3 EntityPositionInterpolator::get_position(Clock::time_point time_point, entt::entity entity) {
    auto* interpolation = m_registry->try_get<EntityPositionInterpolation>(entity);
    if(interpolation == nullptr) {
        spdlog::warn("Attempt to get position for non-registered entity {}", (uint32_t)entity);
        return glm::vec3();
    }

    auto [from, to, value] = interpolation->get_values_around(time_point);
    return glm::mix(from, to, value);
}

void EntityPositionInterpolator::interpolate_smoothed_entities(Clock::time_point time_point) {
    m_registry->view<EntityPositionInterpolation, Transform>()
        .each([&](const auto entity, const EntityPositionInterpolation& interpolation, Transform& ts) {
            auto [from, to, value] = interpolation.get_values_around(time_point);
            ts.set_position(glm::mix(from, to, value));
        });
}

}