ReadBackCuda.h

Go to the documentation of this file.

 
#ifndef READBACKCUDA_H
#define READBACKCUDA_H
 
#ifdef __APPLE__
#include <OpenGL/gl.h>
#else
#include <GL/gl.h>
#endif
 
#include "export.h"
 
struct cudaGraphicsResource;
typedef unsigned char uchar;
 
namespace vistle {
 
class V_RHREXPORT ReadBackCuda {
public:
    ReadBackCuda();
    ~ReadBackCuda();
 
    bool init();
    bool isInitialized() const;
 
    bool readpixels(GLint x, GLint y, GLint w, GLint pitch, GLint h, GLenum format, int ps, GLubyte *bits, GLint buf,
                    GLenum type = GL_UNSIGNED_BYTE);
    bool readpixelsyuv(GLint x, GLint y, GLint w, GLint pitch, GLint h, GLenum format, int ps, GLubyte *bits, GLint buf,
                       int subx, int suby);
    bool readdepthquant(GLint x, GLint y, GLint w, GLint pitch, GLint h, GLenum format, int ps, GLubyte *bits,
                        GLint buf, GLenum type = GL_UNSIGNED_BYTE);
 
    static bool s_error; 
private:
    bool initPbo(size_t raw, size_t compressed);
    bool pbo2cuda(GLint x, GLint y, GLint w, GLint pitch, GLint h, GLenum format, int ps, GLint buf, GLenum type,
                  uchar **img);
    GLuint pboName; 
    cudaGraphicsResource *imgRes; 
    uchar *outImg; 
    size_t imgSize; 
    size_t compSize; 
    bool m_initialized; 
};
 
#ifndef HAVE_CUDA
ReadBackCuda::ReadBackCuda()
{}
ReadBackCuda::~ReadBackCuda()
{}
bool ReadBackCuda::init()
{
    return false;
}
bool ReadBackCuda::isInitialized() const
{
    return false;
}
bool ReadBackCuda::readpixels(GLint x, GLint y, GLint w, GLint pitch, GLint h, GLenum format, int ps, GLubyte *bits,
                              GLint buf, GLenum type)
{
    return false;
}
bool ReadBackCuda::readpixelsyuv(GLint x, GLint y, GLint w, GLint pitch, GLint h, GLenum format, int ps, GLubyte *bits,
                                 GLint buf, int subx, int suby)
{
    return false;
}
bool ReadBackCuda::readdepthquant(GLint x, GLint y, GLint w, GLint pitch, GLint h, GLenum format, int ps, GLubyte *bits,
                                  GLint buf, GLenum type)
{
    return false;
}
#endif
 
} // namespace vistle
#endif