#ifndef GLUT2_IMAGE_H
#define GLUT2_IMAGE_H

#include <GL/glew.h>
#include <GL/gl.h>
#include <helper.h>


GLuint glob_program;
GLuint glob_texture;
bool once = false;

void draw() {
    if (once) {
        return;
    }
    once = true;
    gl(glBindImageTexture(0, glob_texture, 0, GL_FALSE, 0, GL_WRITE_ONLY,
        GL_RGBA16UI));

    std::cout << "glob_program is progam "
        << (glIsProgram(glob_program) == GL_TRUE)
        << std::endl;
    GLint uniformTexPosition = gl(glGetUniformLocation(glob_program, "u_image"));
    std::cout << "u_image position " << uniformTexPosition << std::endl;
    gl(glUseProgram(glob_program));
    gl(glUniform1i(uniformTexPosition, 0));
    gl(glDispatchCompute(256, 256, 1));
}

void initTextures() {

    GLushort texturePtr[256 * 256 * 4];

    gl(glGenTextures(1, &glob_texture));
    gl(glBindTexture(GL_TEXTURE_2D, glob_texture));
    gl(glTexStorage2D(GL_TEXTURE_2D, 1, GL_RGBA16UI, 256, 256));
}

void initShader() {
    const GLchar *compShaderT =
        "#version 440 core\n"
        "layout (location = 1, rgba16ui) uniform uimage2D u_image;\n"
        "uniform atomic_uint u_counter;\n"
        "void main() {\n"
            "uint value = atomicCounterIncrement(u_counter);\n"
            "ivec2 size = imageSize(u_image);\n"
            "if (gl_GlobalInvocationID.x < size.x && gl_GlobalInvocationID.y < size.y)"
            "imageStore(u_image, ivec2(gl_GlobalInvocationID.xy), "
                "uvec4(vec3(value), 0xffff));"
        "}\n";

    GLuint vertexShader = gl(glCreateShader(GL_COMPUTE_SHADER));
    gl(glShaderSource(vertexShader, 1, &compShaderT, nullptr));
    gl(glCompileShader(vertexShader));

    glShaderLog(vertexShader);
    glob_program = gl(glCreateProgram());
    gl(glAttachShader(glob_program, vertexShader));
    gl(glLinkProgram(glob_program));

    gl(glDeleteShader(vertexShader));

    glProgramLog(glob_program);
}

#endif