#include "World.hpp"

namespace MC::World {

std::vector<World::ChunkData> World::get_visible_chunks(Vector<3> position) {
    auto visible_chunks = get_visible_chunk_indices(position);

    auto difference = visible_chunks;
    for (auto index : m_visible_chunks) {
        difference.erase(index);
    }

    if (!difference.empty()) {
        for (auto new_index: difference) {
            auto& data = get_or_generate(new_index);
            if (!data.mesh.has_value()) {
                auto mesh = data.chunk->mesh();
                data.mesh = GFX::Binder::load(mesh);
            }
        }
        m_visible_chunks = visible_chunks;
    }

    std::vector<World::ChunkData> chunks{};
    chunks.reserve(visible_chunks.size());
    for (auto index : visible_chunks) {
        chunks.push_back(get_or_generate(index));
    }

    return chunks;
}

std::unordered_set<ChunkIndex> World::get_visible_chunk_indices(Vector<3> position) const {
    int32_t center_x = std::round(position.x() / Chunk::Width);
    int32_t center_y = std::round(position.z() / Chunk::Width);

    std::unordered_set<ChunkIndex> indices{};
    indices.reserve(m_view_distance_radius * m_view_distance_radius * 4);
    auto radius = m_view_distance_radius;
    for (int32_t x = -radius; x <= radius; x++) {
        int32_t height = std::round(std::sqrt(radius * radius - x * x) + 0.5);
        for (int32_t y = -height; y <= height; y++) {
            indices.emplace(x + center_x, y + center_y);
        }
    }

    return indices;
}

World::ChunkData& World::get_or_generate(ChunkIndex index) {
    auto entry = m_chunks.find(index);
    if (entry == m_chunks.end()) {
        auto chunk = m_generator.generate(index.x, index.y);
        ChunkData data{index, std::make_shared<Chunk>(chunk)};

        m_chunks.insert({index, data});
        return m_chunks.at(index);
    }

    return entry->second;
}


}