#include "Pathfinding.hpp"

#include "world/Level.hpp"
#include "voxels/GlobalChunks.hpp"
#include "voxels/Chunk.hpp"
#include "voxels/blocks_agent.hpp"
#include "content/Content.hpp"

inline constexpr float SQRT2 = 1.4142135623730951f; // sqrt(2)

using namespace voxels;

static float heuristic(const glm::ivec3& a, const glm::ivec3& b) {
    return glm::distance(glm::vec3(a), glm::vec3(b));
}

Pathfinding::Pathfinding(const Level& level)
    : level(level), chunks(*level.chunks) {
}

static bool check_passability(
    const Agent& agent,
    const GlobalChunks& chunks,
    const Node& node,
    const glm::ivec2& offset,
    bool diagonal
) {
    if (!diagonal) {
        return true;
    }
    auto a = node.pos + glm::ivec3(offset.x, 0, 0);
    auto b = node.pos + glm::ivec3(0, 0, offset.y);

    for (int i = 0; i < agent.height; i++) {
        if (blocks_agent::is_obstacle_at(chunks, a.x, a.y + i, a.z))
            return false;
        if (blocks_agent::is_obstacle_at(chunks, b.x, b.y + i, b.z))
            return false;
    }
    return true;
}

static void restore_route(
    Route& route,
    const glm::ivec3& lastPos,
    const std::unordered_map<glm::ivec3, Node>& parents
) {
    auto pos = lastPos;
    while (true) {
        const auto& found = parents.find(pos);
        if (found == parents.end()) {
            route.nodes.push_back({pos});
            break;
        }
        route.nodes.push_back({pos});
        pos = found->second.pos;
    }
}

int Pathfinding::createAgent() {
    int id = nextAgent++;
    agents[id] = Agent();
    return id;
}

bool Pathfinding::removeAgent(int id) {
    auto found = agents.find(id);
    if (found != agents.end()) {
        agents.erase(found);
        return true;
    }
    return false;
}

void Pathfinding::performAllAsync(int stepsPerAgent) {
    for (auto& [id, agent] : agents) {
        if (agent.state.finished) {
            continue;
        }
        perform(agent, stepsPerAgent);
    }
}

Route Pathfinding::perform(Agent& agent, int maxVisited) {
    using namespace blocks_agent;

    Route route {};

    State state = std::move(agent.state);
    if (state.queue.empty()) {
        state.queue.push({agent.start, {}, 0, heuristic(agent.start, agent.target)});
    }

    const auto& chunks = *level.chunks;
    int height = std::max(agent.height, 1);

    if (state.nearest == glm::ivec3(0)) {
        state.nearest = agent.start;
        state.minHScore = heuristic(agent.start, agent.target);
    }
    int visited = -1;

    while (!state.queue.empty()) {
        if (state.blocked.size() == agent.maxVisitedBlocks) {
            if (agent.mayBeIncomplete) {
                restore_route(route, state.nearest, state.parents);
                route.nodes.push_back({agent.start});
                route.found = true;
                state.finished = true;
                agent.state = std::move(state);
                agent.route = route;
                return route;
            }
            break;
        }
        visited++;
        if (visited == maxVisited) {
            state.finished = false;
            agent.state = std::move(state);
            return {};
        }

        auto node = state.queue.top();
        state.queue.pop();

        if (node.pos.x == agent.target.x &&
            glm::abs((node.pos.y - agent.target.y) / height) == 0 &&
            node.pos.z == agent.target.z) {
            restore_route(route, node.pos, state.parents);
            route.nodes.push_back({agent.start});
            route.found = true;
            state.finished = true;
            agent.state = std::move(state);
            agent.route = route;
            return route;
        }

        state.blocked.emplace(node.pos);
        glm::ivec2 neighbors[8] {
            {0, 1}, {1, 0}, {0, -1}, {-1, 0},
            {-1, -1}, {1, -1}, {1, 1}, {-1, 1},
        };

        for (int i = 0; i < sizeof(neighbors) / sizeof(glm::ivec2); i++) {
            auto offset = neighbors[i];
            auto pos = node.pos;

            int surface = getSurfaceAt(pos + glm::ivec3(offset.x, 0, offset.y), 1);

            if (surface == -1) {
                continue;
            }
            pos.y = surface;
            auto point = pos + glm::ivec3(offset.x, 0, offset.y);
            if (state.blocked.find(point) != state.blocked.end()) {
                continue;
            }

            if (is_obstacle_at(chunks, pos.x, pos.y + agent.jumpHeight, pos.z)) {
                continue;
            }
            if (!check_passability(agent, chunks, node, offset, i >= 4)) {
                continue;
            }

            int score = glm::abs(node.pos.y - pos.y) * 10;
            float sum = glm::abs(offset.x) + glm::abs(offset.y);
            float gScore =
                node.gScore + sum + score;
            const auto& found = state.parents.find(point);
            if (found == state.parents.end()) {
                float hScore = heuristic(point, agent.target);
                if (hScore < state.minHScore) {
                    state.minHScore = hScore;
                    state.nearest = point;
                }
                float fScore = gScore * 0.75f + hScore;
                Node nNode {point, node.pos, gScore, fScore};
                state.parents[point] = node;
                state.queue.push(nNode);
            }
        }
    }
    state.finished = true;
    agent.state = std::move(state);
    return {};
}

Agent* Pathfinding::getAgent(int id) {
    const auto& found = agents.find(id);
    if (found != agents.end()) {
        return &found->second;
    }
    return nullptr;
}

const std::unordered_map<int, Agent>& Pathfinding::getAgents() const {
    return agents;
}

static int check_point(
    const ContentUnitIndices<Block>& defs,
    const GlobalChunks& chunks,
    int x,
    int y,
    int z
) {
    auto vox = blocks_agent::get(chunks, x, y, z);
    if (vox == nullptr) {
        return 0;
    }
    const auto& def = defs.require(vox->id);
    if (def.obstacle) {
        return 0;
    }
    if (def.translucent) {
        return -1;
    }
    return 1;
}

int Pathfinding::getSurfaceAt(const glm::ivec3& pos, int maxDelta) {
    using namespace blocks_agent;

    const auto& defs = level.content.getIndices()->blocks;

    int status;
    int surface = pos.y;
    if (check_point(defs, chunks, pos.x, surface, pos.z) <= 0) {
        if (check_point(defs, chunks, pos.x, surface + 1, pos.z) <= 0)
            return -1;
        else
            return surface + 1;
    } else if ((status = check_point(defs, chunks, pos.x, surface - 1, pos.z)) <= 0) {
        if (status == -1)
            return -1;
        return surface;
    } else if (check_point(defs, chunks, pos.x, surface - 2, pos.z) == 0) {
        return surface - 1;
    }
    return -1;
}