#include "floor-mesh.hpp"
#include "shaders/tile-shader.hpp"
#include "tile.hpp"
#include "chunk.hpp"
#include "tile-atlas.hpp"
#include <Magnum/GL/MeshView.h>

namespace Magnum::Examples {

constexpr auto quad_index_count = 6;

floor_mesh::floor_mesh()
{
    _mesh.setCount((int)(quad_index_count * TILE_COUNT))
         .addVertexBuffer(_positions_buffer, 0, tile_shader::Position{})
         .addVertexBuffer(_vertex_buffer, 0, tile_shader::TextureCoordinates{})
         .setIndexBuffer(_index_buffer, 0, GL::MeshIndexType::UnsignedShort);
}

void floor_mesh::set_tile(quad_data& data, tile& x)
{
    CORRADE_INTERNAL_ASSERT(x.ground_image);

    auto texcoords = x.ground_image.atlas->texcoords_for_id(x.ground_image.variant);
    for (size_t i = 0; i < 4; i++)
        data[i] = { texcoords[i] };
}

void floor_mesh::draw(tile_shader& shader, chunk& c)
{
    std::array<quad_data, TILE_COUNT> data;
    c.foreach_tile([&](tile& x, std::size_t idx, local_coords) {
      set_tile(data[idx], x);
    });
    _vertex_buffer.setSubData(0, data);
    Magnum::GL::MeshView mesh{_mesh};
    mesh.setCount(quad_index_count);
    const tile_atlas* last_tile_atlas = nullptr;
    c.foreach_tile([&](tile& x, std::size_t i, local_coords) {
      mesh.setIndexRange((int)(i*quad_index_count), 0, quad_index_count*TILE_COUNT - 1);
      if (auto* atlas = x.ground_image.atlas.get(); atlas != last_tile_atlas)
      {
          atlas->texture().bind(0);
          last_tile_atlas = atlas;
      }
      shader.draw(mesh);
    });
}

std::array<std::array<UnsignedShort, 6>, TILE_COUNT> floor_mesh::make_index_array()
{
    // NOLINTNEXTLINE(cppcoreguidelines-pro-type-member-init)
    std::array<std::array<UnsignedShort, quad_index_count>, TILE_COUNT> array;
    for (std::size_t i = 0; i < std::size(array); i++)
        array[i] = tile_atlas::indices(i);
    return array;
}

std::array<std::array<Vector3, 4>, TILE_COUNT> floor_mesh::make_position_array()
{
    std::array<std::array<Vector3, 4>, TILE_COUNT> array;
    constexpr float X = TILE_SIZE[0], Y = TILE_SIZE[1];
    for (std::size_t j = 0, k = 0; j < TILE_MAX_DIM; j++)
        for (std::size_t i = 0; i < TILE_MAX_DIM; i++, k++)
        {
            Vector3 center {(float)(X*i), (float)(Y*j), 0};
            array[k] = { tile_atlas::floor_quad(center, {X, Y}) };
        }
    return array;
}

} // namespace Magnum::Examples