/*!
 * \file src/Mesh.cpp
 * \brief Textured/Colored Mesh
 *
 * \author xythobuz
 */

#include "global.h"
#include "TextureManager.h"
#include "Mesh.h"

#include <glbinding/gl/gl33.h>

Mesh::Mesh(const std::vector<glm::vec3>& vert,
           const std::vector<IndexedRectangle>& rect,
           const std::vector<IndexedRectangle>& tri,
           const std::vector<IndexedColoredRectangle>& coloredRect,
           const std::vector<IndexedColoredRectangle>& coloredTri) {
    for (auto& t : rect) {
        indicesBuff.push_back(0);
        verticesBuff.push_back(glm::vec3(vert.at(t.v1).x, vert.at(t.v1).y, vert.at(t.v1).z));
        verticesBuff.push_back(glm::vec3(vert.at(t.v2).x, vert.at(t.v2).y, vert.at(t.v2).z));
        verticesBuff.push_back(glm::vec3(vert.at(t.v3).x, vert.at(t.v3).y, vert.at(t.v3).z));
        verticesBuff.push_back(glm::vec3(vert.at(t.v4).x, vert.at(t.v4).y, vert.at(t.v4).z));
        texturesBuff.push_back(t.texture);
    }

    for (auto& t : tri) {
        indicesBuff.push_back(1);
        verticesBuff.push_back(glm::vec3(vert.at(t.v1).x, vert.at(t.v1).y, vert.at(t.v1).z));
        verticesBuff.push_back(glm::vec3(vert.at(t.v2).x, vert.at(t.v2).y, vert.at(t.v2).z));
        verticesBuff.push_back(glm::vec3(vert.at(t.v3).x, vert.at(t.v3).y, vert.at(t.v3).z));
        texturesBuff.push_back(t.texture);
    }

    for (auto& t : coloredRect) {
        indicesColorBuff.push_back(0);
        verticesColorBuff.push_back(glm::vec3(vert.at(t.v1).x, vert.at(t.v1).y, vert.at(t.v1).z));
        verticesColorBuff.push_back(glm::vec3(vert.at(t.v2).x, vert.at(t.v2).y, vert.at(t.v2).z));
        verticesColorBuff.push_back(glm::vec3(vert.at(t.v3).x, vert.at(t.v3).y, vert.at(t.v3).z));
        verticesColorBuff.push_back(glm::vec3(vert.at(t.v4).x, vert.at(t.v4).y, vert.at(t.v4).z));
        colorsBuff.push_back(t.index);
    }

    for (auto& t : coloredTri) {
        indicesColorBuff.push_back(1);
        verticesColorBuff.push_back(glm::vec3(vert.at(t.v1).x, vert.at(t.v1).y, vert.at(t.v1).z));
        verticesColorBuff.push_back(glm::vec3(vert.at(t.v2).x, vert.at(t.v2).y, vert.at(t.v2).z));
        verticesColorBuff.push_back(glm::vec3(vert.at(t.v3).x, vert.at(t.v3).y, vert.at(t.v3).z));
        colorsBuff.push_back(t.index);
    }
}

void Mesh::prepare() {
    std::vector<unsigned short> ind;
    std::vector<glm::vec3> vert;
    std::vector<glm::vec2> uvBuff;
    std::vector<unsigned int> tex;

    int vertIndex = 0;
    for (int i = 0; i < indicesBuff.size(); i++) {
        unsigned int texture = TextureManager::getTile(texturesBuff.at(i)).getTexture();
        for (int x = 0; x < ((indicesBuff.at(i) == 0) ? 4 : 3); x++) {
            int v = x;
            if (v == 0)
                v = 2;
            else if (v == 2)
                v = 0;
            ind.push_back(vert.size());
            vert.push_back(verticesBuff.at(vertIndex + v));
            uvBuff.push_back(TextureManager::getTile(texturesBuff.at(i)).getUV(v));
            tex.push_back(texture);
        }

        if (indicesBuff.at(i) == 0) {
            ind.push_back(ind.at(ind.size() - 4));
            ind.push_back(ind.at(ind.size() - 3));
        }

        vertIndex += (indicesBuff.at(i) == 0) ? 4 : 3;
    }

    orAssertEqual(ind.size() % 3, 0);
    orAssertEqual(vert.size(), tex.size());
    orAssertEqual(vert.size(), uvBuff.size());

    indicesBuff = std::move(ind);
    vertices.bufferData(vert);
    uvs.bufferData(uvBuff);
    texturesBuff = std::move(tex);

    std::vector<unsigned short> indCol;
    std::vector<glm::vec3> vertCol;
    std::vector<glm::vec3> cols;

    vertIndex = 0;
    for (int i = 0; i < indicesColorBuff.size(); i++) {
        for (int x = 0; x < ((indicesColorBuff.at(i) == 0) ? 4 : 3); x++) {
            int v = x;
            if (v == 0)
                v = 2;
            else if (v == 2)
                v = 0;
            indCol.push_back(vertCol.size());
            vertCol.push_back(verticesColorBuff.at(vertIndex + v));
            glm::vec4 c = TextureManager::getPalette(colorsBuff.at(i));
            cols.push_back(glm::vec3(c.x, c.y, c.z));
        }

        if (indicesColorBuff.at(i) == 0) {
            indCol.push_back(indCol.at(indCol.size() - 4));
            indCol.push_back(indCol.at(indCol.size() - 3));
        }

        vertIndex += (indicesColorBuff.at(i) == 0) ? 4 : 3;
    }

    orAssertEqual(indCol.size() % 3, 0);
    orAssertEqual(vertCol.size(), cols.size());

    indicesColor.bufferData(indCol);
    verticesColor.bufferData(vertCol);
    colors.bufferData(cols);

    verticesBuff.clear();
    indicesColorBuff.clear();
    verticesColorBuff.clear();
    colorsBuff.clear();
}

void Mesh::display(glm::mat4 MVP, ShaderTexture* shaderTexture) {
    if (indicesBuff.size() > 0) {
        unsigned int indexStart = 0;
        unsigned int indexPos = 1;
        unsigned int texture = texturesBuff.at(indicesBuff.at(0));

        while ((indexStart != indexPos) && (indexPos < indicesBuff.size())) {
            while ((indexPos < indicesBuff.size())
                   && (texturesBuff.at(indicesBuff.at(indexPos)) == texture)) {
                indexPos++;
            }

            std::vector<unsigned short> ind(indicesBuff.begin() + indexStart,
                                            indicesBuff.begin() + indexPos);
            indices.bufferData(ind);
            Shader::drawGL(vertices, uvs, indices, texture, MVP, TextureStorage::GAME, shaderTexture);

            if (indexPos < indicesBuff.size()) {
                indexStart = indexPos;
                indexPos += 1;
                texture = texturesBuff.at(indicesBuff.at(indexStart));
            }
        }
    }

    if (indicesColor.getSize() > 0)
        Shader::drawGL(verticesColor, colors, indicesColor, MVP, gl::GL_TRIANGLES, shaderTexture);
}