直接從設備（視頻）顯示當前窗口中的位圖內存

Question

假設我已在設備（視頻）內存中使用cudaMalloc()和cudaMemcpy()的位圖（圖像）。

直接在屏幕上顯示此位圖（當前窗口）的最簡單方法是什麼？

我猜應該有一個替代GDI的BitBlt() ......但任何方法（Direct3D，OpenGL甚至GDI）都可以。

CUDA SDK（OpenGL版本）中的示例使用Texture，它使用緩衝區（像素緩衝區對象）來保存數據，該數據註冊爲CUDA資源，每幀應該映射/未映射（glutDisplayFunc()調用）。而這一切似乎只是一點點複雜和不必要的。

Answer 1

好的，會試着回答我自己的問題。看起來像通過OpenGL顯示位圖有一個更簡單的方法。我們可以使用支持紋理格式GL_RGBA8（自CUDA Toolkit 4.0以來）的cudaGraphicsGLRegisterImage()，而不是使用紋理+像素緩衝區對象。

爲簡單起見，示例沒有cutilSafeCall()或錯誤檢查。我對OpenGL知之甚少，很高興聽到建議。

#include <stdio.h> 
#include <string.h> 

#include <GL/glew.h> 
#include <GL/freeglut.h> 

#include <cuda_runtime_api.h> 
#include <cuda_gl_interop.h> 

/* Handles OpenGL-CUDA exchange. */ 
cudaGraphicsResource *cuda_texture; 

/* Registers (+ resizes) CUDA-Texture (aka Renderbuffer). */ 
void resizeTexture(int w, int h) { 
    static GLuint gl_texture = 0; 

    /* Delete old CUDA-Texture. */ 
    if (gl_texture) { 
     cudaGraphicsUnregisterResource(cuda_texture); 
     glDeleteTextures(1, &gl_texture); 
    } else glEnable(GL_TEXTURE_2D); 

    glGenTextures(1, &gl_texture); 
    glBindTexture(GL_TEXTURE_2D, gl_texture); 

    glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_NEAREST); 
    glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_NEAREST); 

    glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA8, w, h, 0, GL_RGBA, GL_UNSIGNED_BYTE, NULL); 

    cudaGraphicsGLRegisterImage(&cuda_texture, gl_texture, GL_TEXTURE_2D, cudaGraphicsMapFlagsWriteDiscard); 
} 

void updateMemDevice(cudaArray *memDevice, int w, int h) { 
    struct Color { 
     unsigned char r, g, b, a; 
    } *memHost = new Color[w*h]; 

    memset(memHost, 128, w*h*4); 
    for (int y = 0; y<h; ++y) { 
     for (int x = 0; x<w; ++x) { 
      Color &c = memHost[y*w + x]; 
      c.r = c.b = 255 * x/w; 
     } 
    } 
    cudaMemcpyToArray(memDevice, 0, 0, memHost, w*h*4, cudaMemcpyHostToDevice); 
    delete [] memHost; 
} 

void editTexture(int w, int h) { 
    cudaGraphicsMapResources(1, &cuda_texture); 
    cudaArray* memDevice; 
    cudaGraphicsSubResourceGetMappedArray(&memDevice, cuda_texture, 0, 0); 
    updateMemDevice(memDevice, w, h); 
    cudaGraphicsUnmapResources(1, &cuda_texture); 
} 


void windowResizeFunc(int w, int h) { 
    glViewport(-w, -h, w*2, h*2); 
} 

void displayFunc() { 
    glBegin(GL_QUADS); 
    glTexCoord2i(0, 0); glVertex2i(0, 0); 
    glTexCoord2i(1, 0); glVertex2i(1, 0); 
    glTexCoord2i(1, 1); glVertex2i(1, 1); 
    glTexCoord2i(0, 1); glVertex2i(0, 1); 
    glEnd(); 

    glFlush(); 
} 


int main(int argc, char *argv[]) { 
    /* Initialize OpenGL context. */ 
    glutInit(&argc, argv); 
    glutInitDisplayMode(GLUT_RGB); 
    glutInitWindowSize(400, 300); 
    glutCreateWindow("Bitmap in Device Memory"); 
    glutReshapeFunc(windowResizeFunc); 
    glutDisplayFunc(displayFunc); 

    glewInit(); 

    cudaGLSetGLDevice(0); 

    int width = 5, height = 5; 
    resizeTexture(width, height); 
    editTexture(width, height); 

    glutMainLoop(); 
    return 0; 
}