tuto_compute_buffer.cpp

Go to the documentation of this file.

 
#include <vector>
 
#include "app.h"
#include "program.h"
 
struct ComputeBuffer : public App
{
    ComputeBuffer( ) : App(1280, 768, 4,3) {}
    
    int init( )
    {
        m_program= read_program("gkit2_tutos/M2/compute_buffer.glsl");
        program_print_errors(m_program);
        
        // initialise un tableau de valeurs
        m_data= std::vector<int>(1024);
        
        // cree les buffers pour les parametres du shader
        glGenBuffers(1, &m_gpu_buffer1);
        glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 0, m_gpu_buffer1);
        glBufferData(GL_SHADER_STORAGE_BUFFER, sizeof(int) * m_data.size(), m_data.data(), GL_STATIC_COPY);
        
        // buffer resultat, meme taille, mais pas de donnees...
        glGenBuffers(1, &m_gpu_buffer2);
        glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 1, m_gpu_buffer2);
        glBufferData(GL_SHADER_STORAGE_BUFFER, sizeof(int) * m_data.size(), nullptr, GL_STATIC_COPY);
        
        return 0;
    }
    
    int quit( )
    {
        release_program(m_program);
        glDeleteBuffers(1, &m_gpu_buffer1);
        glDeleteBuffers(1, &m_gpu_buffer2);
        
        return 0;
    }
    
    int render( )
    {
        glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 0, m_gpu_buffer1);
        glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 1, m_gpu_buffer2);
        
        // execute les shaders
        glUseProgram(m_program);
        
        // recupere le nombre de threads declare par le shader
        int threads[3]= {};
        glGetProgramiv(m_program, GL_COMPUTE_WORK_GROUP_SIZE, threads);
        
        int n= m_data.size() / threads[0];
        // nombre de groupes de threads, arrondi...
        if(m_data.size() % threads[0])
            n++;
        // oui on peut calculer ca de maniere plus directe... 
        // ou utiliser directement 256, comme dans le shader...
        
        // go !
        glDispatchCompute(n, 1, 1);
        
        // attendre que les resultats soient disponibles
        glMemoryBarrier(GL_BUFFER_UPDATE_BARRIER_BIT);
        
        // relire le resultat
        std::vector<int> tmp(1024);
        glBindBuffer(GL_SHADER_STORAGE_BUFFER, m_gpu_buffer2);
        glGetBufferSubData(GL_SHADER_STORAGE_BUFFER, 0, sizeof(int) * tmp.size(), tmp.data());
        
        for(unsigned i= 0; i < tmp.size(); i++)
            printf("%d ", tmp[i]);
        printf("\n");
        
        return 0;
    }
    
    std::vector<int> m_data;
    GLuint m_gpu_buffer1;
    GLuint m_gpu_buffer2;
    GLuint m_read_buffer;
    GLuint m_program;
};
 
int main( )
{
    ComputeBuffer app;
    app.run();
    
    return 0;
}