#include <mbgl/renderer/render_static_data.hpp>
#include <mbgl/gfx/context.hpp>
#include <mbgl/gfx/upload_pass.hpp>
#include <mbgl/programs/program_parameters.hpp>

namespace mbgl {

static gfx::VertexVector<gfx::Vertex<PositionOnlyLayoutAttributes>> tileVertices() {
    gfx::VertexVector<gfx::Vertex<PositionOnlyLayoutAttributes>> result;
    result.emplace_back(gfx::Vertex<PositionOnlyLayoutAttributes>({{{ 0,            0 }}}));
    result.emplace_back(gfx::Vertex<PositionOnlyLayoutAttributes>({{{ util::EXTENT, 0 }}}));
    result.emplace_back(gfx::Vertex<PositionOnlyLayoutAttributes>({{{ 0, util::EXTENT }}}));
    result.emplace_back(gfx::Vertex<PositionOnlyLayoutAttributes>({{{ util::EXTENT, util::EXTENT }}}));
    return result;
}

static gfx::IndexVector<gfx::Triangles> quadTriangleIndices() {
    gfx::IndexVector<gfx::Triangles> result;
    result.emplace_back(0, 1, 2);
    result.emplace_back(1, 2, 3);
    return result;
}

static gfx::IndexVector<gfx::LineStrip> tileLineStripIndices() {
    gfx::IndexVector<gfx::LineStrip> result;
    result.emplace_back(0);
    result.emplace_back(1);
    result.emplace_back(3);
    result.emplace_back(2);
    result.emplace_back(0);
    return result;
}

static gfx::VertexVector<RasterLayoutVertex> rasterVertices() {
    gfx::VertexVector<RasterLayoutVertex> result;
    result.emplace_back(RasterProgram::layoutVertex({ 0, 0 }, { 0, 0 }));
    result.emplace_back(RasterProgram::layoutVertex({ util::EXTENT, 0 }, { util::EXTENT, 0 }));
    result.emplace_back(RasterProgram::layoutVertex({ 0, util::EXTENT }, { 0, util::EXTENT }));
    result.emplace_back(RasterProgram::layoutVertex({ util::EXTENT, util::EXTENT }, { util::EXTENT, util::EXTENT }));
    return result;
}

static gfx::VertexVector<HeatmapTextureLayoutVertex> heatmapTextureVertices() {
    gfx::VertexVector<HeatmapTextureLayoutVertex> result;
    result.emplace_back(HeatmapTextureProgram::layoutVertex({ 0, 0 }));
    result.emplace_back(HeatmapTextureProgram::layoutVertex({ 1, 0 }));
    result.emplace_back(HeatmapTextureProgram::layoutVertex({ 0, 1 }));
    result.emplace_back(HeatmapTextureProgram::layoutVertex({ 1, 1 }));
    return result;
}

RenderStaticData::RenderStaticData(gfx::Context& context, float pixelRatio)
    : programs(context, ProgramParameters{pixelRatio, false}),
      clippingMaskSegments(tileTriangleSegments())
#ifndef NDEBUG
      ,
      overdrawPrograms(context, ProgramParameters{pixelRatio, true})
#endif
{
}

SegmentVector<BackgroundAttributes> RenderStaticData::tileTriangleSegments() {
    SegmentVector<BackgroundAttributes> segments;
    segments.emplace_back(0, 0, 4, 6);
    return segments;
}

SegmentVector<DebugAttributes> RenderStaticData::tileBorderSegments() {
    SegmentVector<DebugAttributes> segments;
    segments.emplace_back(0, 0, 4, 5);
    return segments;
}

SegmentVector<RasterAttributes> RenderStaticData::rasterSegments() {
    SegmentVector<RasterAttributes> segments;
    segments.emplace_back(0, 0, 4, 6);
    return segments;
}

SegmentVector<HeatmapTextureAttributes> RenderStaticData::heatmapTextureSegments() {
    SegmentVector<HeatmapTextureAttributes> segments;
    segments.emplace_back(0, 0, 4, 6);
    return segments;
}

void RenderStaticData::upload(gfx::UploadPass& uploadPass) {
    if (!uploaded) {
        tileVertexBuffer = uploadPass.createVertexBuffer(tileVertices());
        rasterVertexBuffer = uploadPass.createVertexBuffer(rasterVertices());
        heatmapTextureVertexBuffer = uploadPass.createVertexBuffer(heatmapTextureVertices());
        quadTriangleIndexBuffer = uploadPass.createIndexBuffer(quadTriangleIndices());
        tileBorderIndexBuffer = uploadPass.createIndexBuffer(tileLineStripIndices());
        uploaded = true;
    }
}

} // namespace mbgl