#include "CvMatRender.hpp"
#include <cstdint>

using namespace pixelarium::imaging;

pixelarium::render::CvMatRender::CvMatRender(const Image& img)
{
    this->_img = img.GetImage();
    this->_texture = 0;

    cv::cvtColor(this->_img, this->_img, cv::COLOR_BGR2RGBA);
}

/*static*/ void pixelarium::render::matToTexture(const cv::Mat& image, GLuint* texture)
{
    glGenTextures(1, texture);
    glBindTexture(GL_TEXTURE_2D, *texture);
    glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
    glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);

    // auto image = img.GetImage();
    switch (image.type())
    {
        case CV_16U:
        case CV_16UC3:
        case 26:
            glPixelStorei(GL_UNPACK_ALIGNMENT, 1);
            glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA, image.cols, image.rows, 0,
                         GL_RGBA, GL_UNSIGNED_SHORT, image.data);
            break;
        case 5:
        case 29:
            glPixelStorei(GL_UNPACK_ALIGNMENT, 1);
            glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA, image.cols, image.rows, 0,
                         GL_RGBA, GL_FLOAT, image.data);
            break;
        default:
            glPixelStorei(GL_UNPACK_ROW_LENGTH, 0);
            glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA, image.cols, image.rows, 0,
                         GL_RGBA, GL_UNSIGNED_BYTE, image.data);
            break;
    }
}

GLuint pixelarium::render::CvMatRender::Render()
{
    matToTexture(this->_img, &this->_texture);
    return this->_texture;
}