[转]OpenGL 使用 PBO 高速复制屏幕图像到内存或者纹理中

#include "glext.h"

PFNGLBINDBUFFERPROC glBindBuffer = NULL;
PFNGLBUFFERDATAPROC glBufferData = NULL;
PFNGLGENBUFFERSPROC glGenBuffers = NULL;
PFNGLMAPBUFFERPROC glMapBuffer = NULL;
PFNGLUNMAPBUFFERPROC glUnmapBuffer = NULL;

void gl_init()
{
    glBindBuffer = (PFNGLBINDBUFFERPROC)wglGetProcAddress("glBindBuffer");
    glBufferData = (PFNGLBUFFERDATAPROC)wglGetProcAddress("glBufferData");
    glGenBuffers = (PFNGLGENBUFFERSPROC)wglGetProcAddress("glGenBuffers");
    glMapBuffer = (PFNGLMAPBUFFERPROC)wglGetProcAddress("glMapBuffer");
    glUnmapBuffer = (PFNGLUNMAPBUFFERPROC)wglGetProcAddress("glUnmapBuffer");
}

GLuint Buffer;
GLuint Texture;

void init()
{
    // 创建1个缓冲区
    glGenBuffers(1, &Buffer);
    
    // 缓冲区刚创建出来的时候还没有分配内存，所以我们要初始化一下它
    // 先绑定..
    glBindBuffer(GL_ARRAY_BUFFER, Buffer);
    
    // 这一步十分重要，第2个参数指定了这个缓冲区的大小，单位是字节，一定要注意
    //  然后第3个参数是初始化用的数据，如果你传个内存指针进去，这个函数就会把你的
    //  数据复制到缓冲区里，我们这里一开始并不需要什么数据，所以传个NULL就行了
    //  GL内部会给缓冲区分配内存，然后什么都不干，第4个参数可以优化显存效率，指定
    //  缓冲区中的数据读写频繁程度，如果缓冲区中的数据不经常读写，可以传入 GL_STATIC_****
    //  这样GL会把缓冲区放在内存数据不经常变动的区域，如果要经常读写缓冲区中的数据，可以传
    //  别的值，具体参考 @https://www.khronos.org/registry/OpenGL-Refpages/gl2.1/ 吧
    // 注意这里的 BUFFER_SIZE 我们假设要复制整个屏幕的像素数据，格式为RGB就行了，那么
    //  大小就是 屏幕宽度×屏幕高度×3，每个像素1字节
    glBufferData(GL_ARRAY_BUFFER, BUFFER_SIZE, NULL, GL_STREAM_COPY);
    
    // 这样我们的缓冲区就已经初始化好了，它现在已经有一块可用的内存
    //  随时可以用 glMapBuffer 来访问
    // 初始化完了那么解绑吧
    glBindBuffer(GL_ARRAY_BUFFER, 0);
    
    
    // 创建1个纹理，等会把屏幕复制到这个纹理
    glGenTextures(1, &Texture);
    // 初始化纹理，不多解释了
    glBindTexture(GL_TEXTURE_2D, Texture);
    // 这里data参数传NULL和上面缓冲区一样，GL仅仅给纹理分配内存而已
    //  ScreenWide和ScreenTall是屏幕的宽度和高度
    // 格式用RGB，因为屏幕不需要透明通道，所以纹理的像素数据大小是和上面的缓冲区大小一样的
    glTexImage2D(GL_TEXTURE_2D, 0, GL_RGB, ScreenWide, ScreenTall, 0, GL_RGB, GL_UNSIGNED_BYTE, NULL);
    glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
    glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
    // 初始化完了解绑
    glBindTexture(GL_TEXTURE_2D, 0);
}

void draw()
{
    // 假装这里画了游戏场景
    
    
    // 首先我们要把缓冲区绑定到 GL_PIXEL_PACK_BUFFER 这个地方
    glBindBuffer(GL_PIXEL_PACK_BUFFER, Buffer);
    // 这个函数会判断 GL_PIXEL_PACK_BUFFER 这个地方有没有绑定一个缓冲区，如果有，那就把数据写入到这个缓冲区里
    // 前4个参数就是要读取的屏幕区域，不多解释
    //  格式是RGB，类型是BYTE，每个像素1字节
    // 如果GL_PIXEL_PACK_BUFFER有绑定缓冲区，最后一个参数就作为偏移值来使用，这里不啰嗦没用的东西了。
    //  传NULL就行
    glReadPixels(0, 0, ScreenWide, ScreenTall, GL_RGB, GL_UNSIGNED_BYTE, NULL);
    // 好了我们已经成功把屏幕的像素数据复制到了缓冲区里
    
    // 这时候，你可以用 glMapBuffer 得到缓冲区的内存指针，来读取里面的像素数据，保存到图片文件
    // 完成截图
    /******
    
    // 注意glMapBuffer的第1个参数不一定要是GL_PIXEL_PACK_BUFFER，你可以把缓冲区绑定到比如上面init函数的GL_ARRAY_BUFFER
    //  然后这里也传GL_ARRAY_BUFFER，由于懒得再绑定一次，就接着用上面绑定的GL_PIXEL_PACK_BUFFER吧
    void *data = glMapBuffer(GL_PIXEL_PACK_BUFFER, GL_READ_WRITE);
    if (data)
    {
        WriteTGA("screenshot.tga", ScreenWide, ScreenTall, data);
        glUnmapBuffer(GL_PIXEL_PACK_BUFFER);    // 不要忘了解除Map
    }
    
    ******/
    
    // 完事了把GL_PIXEL_PACK_BUFFER这个地方的缓冲区解绑掉，以免别的函数误操作
    qglBindBuffer(GL_PIXEL_PACK_BUFFER, 0);
    

    
    // 接着我们演示一下把缓冲区中的像素数据传给纹理
    //  首先我们把缓冲区绑定到 GL_PIXEL_UNPACK_BUFFER 这个地方。这里注意啊！GL_PIXEL_PACK_BUFFER 和 GL_PIXEL_UNPACK_BUFFER 是不同的！
    glBindBuffer(GL_PIXEL_UNPACK_BUFFER, Buffer);
    // 绑定纹理
    glBindTexture(GL_TEXTURE_2D, Texture);
    // 这个函数会判断 GL_PIXEL_UNPACK_BUFFER 这个地方有没有绑定一个缓冲区
    //   如果有，就从这个缓冲区读取数据，而不是data参数指定的那个内存
    // 前面参数很简单就不解释了，最后一个参数和上面glReadPixels同理，传NULL就行
    // 这样glTexSubImage2D就会从我们的缓冲区中读取数据了
    // 这里为什么要用glTexSubImage2D呢，因为如果用glTexImage2D，glTexImage2D会销毁纹理内存重新申请，glTexSubImage2D就仅仅只是更新纹理中的数据
    //  这就提高了速度，并且优化了显存的利用率
    glTexSubImage2D(GL_TEXTURE_2D, 0, 0, 0, ScreenWide, ScreenTall, GL_RGB, GL_UNSIGNED_BYTE, NULL);
    // 完事了把GL_PIXEL_UNPACK_BUFFER这个地方的缓冲区解绑掉，以免别的函数误操作
    glBindBuffer(GL_PIXEL_UNPACK_BUFFER, 0);
    
    // 这时候我们已经更新了纹理，我们可以把纹理画出来看看
    
    // 假装这里有绘制纹理的代码
}