mirror of
https://gitlab.com/libeigen/eigen.git
synced 2026-04-10 11:34:33 +08:00
add support for uniforms
This commit is contained in:
Gael Guennebaud
@@ -202,45 +202,114 @@ template<typename Derived> void glRotate(const RotationBase<Derived,3>& rot)
|
||||
#define EIGEN_GL_EVAL(X) X
|
||||
|
||||
#define EIGEN_GL_FUNC1_DECLARATION(FUNC,ARG1,CONST) \
|
||||
template< typename XprType, \
|
||||
typename Scalar = typename XprType::Scalar, \
|
||||
int Rows = XprType::RowsAtCompileTime, \
|
||||
int Cols = XprType::ColsAtCompileTime, \
|
||||
bool IsGLCompatible = bool(XprType::Flags&LinearAccessBit) \
|
||||
&& bool(XprType::Flags&DirectAccessBit) \
|
||||
&& (XprType::IsVectorAtCompileTime || (XprType::Flags&RowMajorBit)==0)> \
|
||||
struct EIGEN_CAT(EIGEN_CAT(ei_gl_,FUNC),_impl); \
|
||||
\
|
||||
template<typename XprType, typename Scalar, int Rows, int Cols> \
|
||||
struct EIGEN_CAT(EIGEN_CAT(ei_gl_,FUNC),_impl)<XprType,Scalar,Rows,Cols,false> { \
|
||||
inline static void run(ARG1 a,EIGEN_GL_EVAL(EIGEN_GL_MAKE_CONST_##CONST) XprType& p) { \
|
||||
EIGEN_CAT(EIGEN_CAT(ei_gl_,FUNC),_impl)<typename ei_plain_matrix_type_column_major<XprType>::type>::run(a,p); } \
|
||||
}; \
|
||||
\
|
||||
template<typename Derived> inline void FUNC(ARG1 a,EIGEN_GL_EVAL(EIGEN_GL_MAKE_CONST_##CONST) Eigen::DenseBase<Derived>& p) { \
|
||||
EIGEN_CAT(EIGEN_CAT(ei_gl_,FUNC),_impl)<Derived>::run(a,p.derived()); \
|
||||
template< typename XprType, \
|
||||
typename Scalar = typename XprType::Scalar, \
|
||||
int Rows = XprType::RowsAtCompileTime, \
|
||||
int Cols = XprType::ColsAtCompileTime, \
|
||||
bool IsGLCompatible = bool(XprType::Flags&LinearAccessBit) \
|
||||
&& bool(XprType::Flags&DirectAccessBit) \
|
||||
&& (XprType::IsVectorAtCompileTime || (XprType::Flags&RowMajorBit)==0)> \
|
||||
struct EIGEN_CAT(EIGEN_CAT(ei_gl_,FUNC),_impl); \
|
||||
\
|
||||
template<typename XprType, typename Scalar, int Rows, int Cols> \
|
||||
struct EIGEN_CAT(EIGEN_CAT(ei_gl_,FUNC),_impl)<XprType,Scalar,Rows,Cols,false> { \
|
||||
inline static void run(ARG1 a,EIGEN_GL_EVAL(EIGEN_GL_MAKE_CONST_##CONST) XprType& p) { \
|
||||
EIGEN_CAT(EIGEN_CAT(ei_gl_,FUNC),_impl)<typename ei_plain_matrix_type_column_major<XprType>::type>::run(a,p); } \
|
||||
}; \
|
||||
\
|
||||
template<typename Derived> inline void FUNC(ARG1 a,EIGEN_GL_EVAL(EIGEN_GL_MAKE_CONST_##CONST) Eigen::DenseBase<Derived>& p) { \
|
||||
EIGEN_CAT(EIGEN_CAT(ei_gl_,FUNC),_impl)<Derived>::run(a,p.derived()); \
|
||||
}
|
||||
|
||||
|
||||
#define EIGEN_GL_FUNC1_SPECIALIZATION_MAT(FUNC,ARG1,CONST,SCALAR,ROWS,COLS,SUFFIX) \
|
||||
template< typename XprType> struct EIGEN_CAT(EIGEN_CAT(ei_gl_,FUNC),_impl)<XprType, SCALAR, ROWS, COLS, true> { \
|
||||
inline static void run(ARG1 a, EIGEN_GL_EVAL(EIGEN_GL_MAKE_CONST_##CONST) XprType& p) { FUNC##SUFFIX(a,p.data()); } \
|
||||
template< typename XprType> struct EIGEN_CAT(EIGEN_CAT(ei_gl_,FUNC),_impl)<XprType, SCALAR, ROWS, COLS, true> { \
|
||||
inline static void run(ARG1 a, EIGEN_GL_EVAL(EIGEN_GL_MAKE_CONST_##CONST) XprType& p) { FUNC##SUFFIX(a,p.data()); } \
|
||||
};
|
||||
|
||||
|
||||
#define EIGEN_GL_FUNC1_SPECIALIZATION_VEC(FUNC,ARG1,CONST,SCALAR,SIZE,SUFFIX) \
|
||||
template< typename XprType> struct EIGEN_CAT(EIGEN_CAT(ei_gl_,FUNC),_impl)<XprType, SCALAR, SIZE, 1, true> { \
|
||||
inline static void run(ARG1 a, EIGEN_GL_EVAL(EIGEN_GL_MAKE_CONST_##CONST) XprType& p) { FUNC##SUFFIX(a,p.data()); } \
|
||||
}; \
|
||||
template< typename XprType> struct EIGEN_CAT(EIGEN_CAT(ei_gl_,FUNC),_impl)<XprType, SCALAR, 1, SIZE, true> { \
|
||||
inline static void run(ARG1 a, EIGEN_GL_EVAL(EIGEN_GL_MAKE_CONST_##CONST) XprType& p) { FUNC##SUFFIX(a,p.data()); } \
|
||||
template< typename XprType> struct EIGEN_CAT(EIGEN_CAT(ei_gl_,FUNC),_impl)<XprType, SCALAR, SIZE, 1, true> { \
|
||||
inline static void run(ARG1 a, EIGEN_GL_EVAL(EIGEN_GL_MAKE_CONST_##CONST) XprType& p) { FUNC##SUFFIX(a,p.data()); } \
|
||||
}; \
|
||||
template< typename XprType> struct EIGEN_CAT(EIGEN_CAT(ei_gl_,FUNC),_impl)<XprType, SCALAR, 1, SIZE, true> { \
|
||||
inline static void run(ARG1 a, EIGEN_GL_EVAL(EIGEN_GL_MAKE_CONST_##CONST) XprType& p) { FUNC##SUFFIX(a,p.data()); } \
|
||||
};
|
||||
|
||||
|
||||
EIGEN_GL_FUNC1_DECLARATION (glGet,GLenum,_)
|
||||
EIGEN_GL_FUNC1_SPECIALIZATION_MAT(glGet,GLenum,_,float, 4,4,Floatv)
|
||||
EIGEN_GL_FUNC1_SPECIALIZATION_MAT(glGet,GLenum,_,double, 4,4,Doublev)
|
||||
|
||||
// glUniform API
|
||||
|
||||
#ifdef GL_VERSION_2_0
|
||||
|
||||
void glUniform2fv_ei (GLint loc, const float* v) { glUniform2fv(loc,1,v); }
|
||||
void glUniform2iv_ei (GLint loc, const int* v) { glUniform2iv(loc,1,v); }
|
||||
|
||||
void glUniform3fv_ei (GLint loc, const float* v) { glUniform3fv(loc,1,v); }
|
||||
void glUniform3iv_ei (GLint loc, const int* v) { glUniform3iv(loc,1,v); }
|
||||
|
||||
void glUniform4fv_ei (GLint loc, const float* v) { glUniform4fv(loc,1,v); }
|
||||
void glUniform4iv_ei (GLint loc, const int* v) { glUniform4iv(loc,1,v); }
|
||||
|
||||
void glUniformMatrix2fv_ei (GLint loc, const float* v) { glUniformMatrix2fv(loc,1,false,v); }
|
||||
void glUniformMatrix3fv_ei (GLint loc, const float* v) { glUniformMatrix3fv(loc,1,false,v); }
|
||||
void glUniformMatrix4fv_ei (GLint loc, const float* v) { glUniformMatrix4fv(loc,1,false,v); }
|
||||
|
||||
|
||||
EIGEN_GL_FUNC1_DECLARATION (glUniform,GLint,const)
|
||||
EIGEN_GL_FUNC1_SPECIALIZATION_VEC(glUniform,GLint,const,float, 2,2fv_ei)
|
||||
EIGEN_GL_FUNC1_SPECIALIZATION_VEC(glUniform,GLint,const,int, 2,2iv_ei)
|
||||
EIGEN_GL_FUNC1_SPECIALIZATION_VEC(glUniform,GLint,const,float, 3,3fv_ei)
|
||||
EIGEN_GL_FUNC1_SPECIALIZATION_VEC(glUniform,GLint,const,int, 3,3iv_ei)
|
||||
EIGEN_GL_FUNC1_SPECIALIZATION_VEC(glUniform,GLint,const,float, 4,4fv_ei)
|
||||
EIGEN_GL_FUNC1_SPECIALIZATION_VEC(glUniform,GLint,const,int, 4,4iv_ei)
|
||||
|
||||
EIGEN_GL_FUNC1_SPECIALIZATION_MAT(glUniform,GLint,const,float, 2,2,Matrix2fv_ei)
|
||||
EIGEN_GL_FUNC1_SPECIALIZATION_MAT(glUniform,GLint,const,float, 3,3,Matrix3fv_ei)
|
||||
EIGEN_GL_FUNC1_SPECIALIZATION_MAT(glUniform,GLint,const,float, 4,4,Matrix4fv_ei)
|
||||
|
||||
#endif
|
||||
|
||||
#ifdef GL_VERSION_2_1
|
||||
|
||||
void glUniformMatrix2x3fv_ei(GLint loc, const float* v) { glUniformMatrix2x3fv(loc,1,false,v); }
|
||||
void glUniformMatrix3x2fv_ei(GLint loc, const float* v) { glUniformMatrix3x2fv(loc,1,false,v); }
|
||||
void glUniformMatrix2x4fv_ei(GLint loc, const float* v) { glUniformMatrix2x4fv(loc,1,false,v); }
|
||||
void glUniformMatrix4x2fv_ei(GLint loc, const float* v) { glUniformMatrix4x2fv(loc,1,false,v); }
|
||||
void glUniformMatrix3x4fv_ei(GLint loc, const float* v) { glUniformMatrix3x4fv(loc,1,false,v); }
|
||||
void glUniformMatrix4x3fv_ei(GLint loc, const float* v) { glUniformMatrix4x3fv(loc,1,false,v); }
|
||||
|
||||
EIGEN_GL_FUNC1_SPECIALIZATION_MAT(glUniform,GLint,const,float, 2,3,Matrix2x3fv_ei)
|
||||
EIGEN_GL_FUNC1_SPECIALIZATION_MAT(glUniform,GLint,const,float, 3,2,Matrix3x2fv_ei)
|
||||
EIGEN_GL_FUNC1_SPECIALIZATION_MAT(glUniform,GLint,const,float, 2,4,Matrix2x4fv_ei)
|
||||
EIGEN_GL_FUNC1_SPECIALIZATION_MAT(glUniform,GLint,const,float, 4,2,Matrix4x2fv_ei)
|
||||
EIGEN_GL_FUNC1_SPECIALIZATION_MAT(glUniform,GLint,const,float, 3,4,Matrix3x4fv_ei)
|
||||
EIGEN_GL_FUNC1_SPECIALIZATION_MAT(glUniform,GLint,const,float, 4,3,Matrix4x3fv_ei)
|
||||
|
||||
#endif
|
||||
|
||||
#ifdef GL_VERSION_3_0
|
||||
|
||||
// void glUniform2dv_ei (GLint loc, const double* v) { glUniform2dv(loc,1,v); }
|
||||
void glUniform2uiv_ei (GLint loc, const unsigned int* v) { glUniform2uiv(loc,1,v); }
|
||||
|
||||
// void glUniform3dv_ei (GLint loc, const double* v) { glUniform3dv(loc,1,v); }
|
||||
void glUniform3uiv_ei (GLint loc, const unsigned int* v) { glUniform3uiv(loc,1,v); }
|
||||
|
||||
// void glUniform4dv_ei (GLint loc, const double* v) { glUniform4dv(loc,1,v); }
|
||||
void glUniform4uiv_ei (GLint loc, const unsigned int* v) { glUniform4uiv(loc,1,v); }
|
||||
|
||||
// EIGEN_GL_FUNC1_SPECIALIZATION_VEC(glUniform,GLint,const,double, 2,2dv_ei)
|
||||
EIGEN_GL_FUNC1_SPECIALIZATION_VEC(glUniform,GLint,const,unsigned int, 2,2uiv_ei)
|
||||
// EIGEN_GL_FUNC1_SPECIALIZATION_VEC(glUniform,GLint,const,double, 3,3dv_ei)
|
||||
EIGEN_GL_FUNC1_SPECIALIZATION_VEC(glUniform,GLint,const,unsigned int, 3,3uiv_ei)
|
||||
// EIGEN_GL_FUNC1_SPECIALIZATION_VEC(glUniform,GLint,const,double, 4,4dv_ei)
|
||||
EIGEN_GL_FUNC1_SPECIALIZATION_VEC(glUniform,GLint,const,unsigned int, 4,4uiv_ei)
|
||||
|
||||
#endif
|
||||
|
||||
//@}
|
||||
|
||||
}
|
||||
|
||||
@@ -87,9 +87,11 @@ endif()
|
||||
|
||||
find_package(OpenGL)
|
||||
find_package(GLUT)
|
||||
if(OPENGL_FOUND AND GLUT_FOUND AND GLUT_LIBRARIES)
|
||||
find_package(GLEW)
|
||||
if(OPENGL_FOUND AND GLUT_FOUND AND GLUT_LIBRARIES AND GLEW_FOUND)
|
||||
ei_add_property(EIGEN_TESTED_BACKENDS "opengl, ")
|
||||
ei_add_test(openglsupport "" "${GLUT_LIBRARIES}" )
|
||||
set(EIGEN_GL_LIB ${GLUT_LIBRARIES} ${GLEW_LIBRARIES})
|
||||
ei_add_test(openglsupport "" "${EIGEN_GL_LIB}" )
|
||||
else()
|
||||
ei_add_property(EIGEN_MISSING_BACKENDS "opengl, ")
|
||||
endif()
|
||||
|
||||
@@ -24,8 +24,9 @@
|
||||
|
||||
#include <main.h>
|
||||
#include <iostream>
|
||||
#include <GL/glut.h>
|
||||
#include <GL/glew.h>
|
||||
#include <Eigen/OpenGLSupport>
|
||||
#include <GL/glut.h>
|
||||
using namespace Eigen;
|
||||
|
||||
|
||||
@@ -42,6 +43,84 @@ using namespace Eigen;
|
||||
VERIFY_IS_APPROX((REF).cast<float>(), m); \
|
||||
}
|
||||
|
||||
#define VERIFY_UNIFORM(NAME,TYPE) { \
|
||||
TYPE value; value.setRandom(); \
|
||||
TYPE data; \
|
||||
int loc = glGetUniformLocation(prg_id, #NAME); \
|
||||
VERIFY((loc!=-1) && "uniform not found"); \
|
||||
glUniform(loc,value); \
|
||||
glGetUniformfv(prg_id,loc,data.data()); \
|
||||
if(!value.isApprox(data)) { \
|
||||
std::cerr << "Expected:\n" << value << "\n" << "got\n" << data << "\n\n"; \
|
||||
} \
|
||||
VERIFY_IS_APPROX(value, data); \
|
||||
}
|
||||
|
||||
#define VERIFY_UNIFORMi(NAME,TYPE) { \
|
||||
TYPE value; value.setRandom(); \
|
||||
TYPE data; \
|
||||
int loc = glGetUniformLocation(prg_id, #NAME); \
|
||||
VERIFY((loc!=-1) && "uniform not found"); \
|
||||
glUniform(loc,value); \
|
||||
glGetUniformiv(prg_id,loc,(GLint*)data.data()); \
|
||||
if(!value.isApprox(data)) { \
|
||||
std::cerr << "Expected:\n" << value << "\n" << "got\n" << data << "\n\n"; \
|
||||
} \
|
||||
VERIFY_IS_APPROX(value, data); \
|
||||
}
|
||||
|
||||
void printInfoLog(GLuint objectID)
|
||||
{
|
||||
int infologLength, charsWritten;
|
||||
GLchar *infoLog;
|
||||
glGetProgramiv(objectID,GL_INFO_LOG_LENGTH, &infologLength);
|
||||
if(infologLength > 0)
|
||||
{
|
||||
infoLog = new GLchar[infologLength];
|
||||
glGetProgramInfoLog(objectID, infologLength, &charsWritten, infoLog);
|
||||
if (charsWritten>0)
|
||||
std::cerr << "Shader info : \n" << infoLog << std::endl;
|
||||
delete[] infoLog;
|
||||
}
|
||||
}
|
||||
|
||||
GLint createShader(const char* vtx, const char* frg)
|
||||
{
|
||||
GLint prg_id = glCreateProgram();
|
||||
GLint vtx_id = glCreateShader(GL_VERTEX_SHADER);
|
||||
GLint frg_id = glCreateShader(GL_FRAGMENT_SHADER);
|
||||
GLint ok;
|
||||
|
||||
glShaderSource(vtx_id, 1, &vtx, 0);
|
||||
glCompileShader(vtx_id);
|
||||
glGetShaderiv(vtx_id,GL_COMPILE_STATUS,&ok);
|
||||
if(!ok)
|
||||
{
|
||||
std::cerr << "vtx compilation failed\n";
|
||||
}
|
||||
|
||||
glShaderSource(frg_id, 1, &frg, 0);
|
||||
glCompileShader(frg_id);
|
||||
glGetShaderiv(frg_id,GL_COMPILE_STATUS,&ok);
|
||||
if(!ok)
|
||||
{
|
||||
std::cerr << "frg compilation failed\n";
|
||||
}
|
||||
|
||||
glAttachShader(prg_id, vtx_id);
|
||||
glAttachShader(prg_id, frg_id);
|
||||
glLinkProgram(prg_id);
|
||||
glGetProgramiv(prg_id,GL_LINK_STATUS,&ok);
|
||||
if(!ok)
|
||||
{
|
||||
std::cerr << "linking failed\n";
|
||||
}
|
||||
printInfoLog(prg_id);
|
||||
|
||||
glUseProgram(prg_id);
|
||||
return prg_id;
|
||||
}
|
||||
|
||||
void test_openglsupport()
|
||||
{
|
||||
int argc = 0;
|
||||
@@ -52,9 +131,15 @@ void test_openglsupport()
|
||||
|
||||
if(glutCreateWindow("Eigen") <= 0)
|
||||
{
|
||||
std::cerr << "Unable to create GLUT Window.\n";
|
||||
std::cerr << "Error: Unable to create GLUT Window.\n";
|
||||
exit(1);
|
||||
}
|
||||
|
||||
glewExperimental = GL_TRUE;
|
||||
if(glewInit() != GLEW_OK)
|
||||
{
|
||||
std::cerr << "Warning: Failed to initialize GLEW\n";
|
||||
}
|
||||
|
||||
Vector3f v3f;
|
||||
Matrix3f rot;
|
||||
@@ -144,4 +229,90 @@ void test_openglsupport()
|
||||
VERIFY_MATRIX(glScale(usd), Projective3d(usd).matrix());
|
||||
}
|
||||
|
||||
// uniform
|
||||
{
|
||||
const char* vtx = "void main(void) { gl_Position = gl_Vertex; }\n";
|
||||
|
||||
if(GLEW_VERSION_2_0)
|
||||
{
|
||||
const char* frg = ""
|
||||
"uniform vec2 v2f;\n"
|
||||
"uniform vec3 v3f;\n"
|
||||
"uniform vec4 v4f;\n"
|
||||
"uniform ivec2 v2i;\n"
|
||||
"uniform ivec3 v3i;\n"
|
||||
"uniform ivec4 v4i;\n"
|
||||
"uniform mat2 m2f;\n"
|
||||
"uniform mat3 m3f;\n"
|
||||
"uniform mat4 m4f;\n"
|
||||
"void main(void) { gl_FragColor = vec4(v2f[0]+v3f[0]+v4f[0])+vec4(v2i[0]+v3i[0]+v4i[0])+vec4(m2f[0][0]+m3f[0][0]+m4f[0][0]); }\n";
|
||||
|
||||
GLint prg_id = createShader(vtx,frg);
|
||||
|
||||
VERIFY_UNIFORM(v2f, Vector2f);
|
||||
VERIFY_UNIFORM(v3f, Vector3f);
|
||||
VERIFY_UNIFORM(v4f, Vector4f);
|
||||
VERIFY_UNIFORMi(v2i, Vector2i);
|
||||
VERIFY_UNIFORMi(v3i, Vector3i);
|
||||
VERIFY_UNIFORMi(v4i, Vector4i);
|
||||
VERIFY_UNIFORM(m2f, Matrix2f);
|
||||
VERIFY_UNIFORM(m3f, Matrix3f);
|
||||
VERIFY_UNIFORM(m4f, Matrix4f);
|
||||
}
|
||||
else
|
||||
std::cerr << "Warning: opengl 2.0 was not tested\n";
|
||||
|
||||
if(GLEW_VERSION_2_1)
|
||||
{
|
||||
const char* frg = "#version 120\n"
|
||||
"uniform mat2x3 m23f;\n"
|
||||
"uniform mat3x2 m32f;\n"
|
||||
"uniform mat2x4 m24f;\n"
|
||||
"uniform mat4x2 m42f;\n"
|
||||
"uniform mat3x4 m34f;\n"
|
||||
"uniform mat4x3 m43f;\n"
|
||||
"void main(void) { gl_FragColor = vec4(m23f[0][0]+m32f[0][0]+m24f[0][0]+m42f[0][0]+m34f[0][0]+m43f[0][0]); }\n";
|
||||
|
||||
GLint prg_id = createShader(vtx,frg);
|
||||
|
||||
typedef Matrix<float,2,3> Matrix23f;
|
||||
typedef Matrix<float,3,2> Matrix32f;
|
||||
typedef Matrix<float,2,4> Matrix24f;
|
||||
typedef Matrix<float,4,2> Matrix42f;
|
||||
typedef Matrix<float,3,4> Matrix34f;
|
||||
typedef Matrix<float,4,3> Matrix43f;
|
||||
|
||||
VERIFY_UNIFORM(m23f, Matrix23f);
|
||||
VERIFY_UNIFORM(m32f, Matrix32f);
|
||||
VERIFY_UNIFORM(m24f, Matrix24f);
|
||||
VERIFY_UNIFORM(m42f, Matrix42f);
|
||||
VERIFY_UNIFORM(m34f, Matrix34f);
|
||||
VERIFY_UNIFORM(m43f, Matrix43f);
|
||||
}
|
||||
else
|
||||
std::cerr << "Warning: opengl 2.1 was not tested\n";
|
||||
|
||||
if(GLEW_VERSION_3_0)
|
||||
{
|
||||
const char* frg = "#version 150\n"
|
||||
"uniform uvec2 v2ui;\n"
|
||||
"uniform uvec3 v3ui;\n"
|
||||
"uniform uvec4 v4ui;\n"
|
||||
"out vec4 data;\n"
|
||||
"void main(void) { data = vec4(v2ui[0]+v3ui[0]+v4ui[0]); }\n";
|
||||
|
||||
GLint prg_id = createShader(vtx,frg);
|
||||
|
||||
typedef Matrix<unsigned int,2,1> Vector2ui;
|
||||
typedef Matrix<unsigned int,3,1> Vector3ui;
|
||||
typedef Matrix<unsigned int,4,1> Vector4ui;
|
||||
|
||||
VERIFY_UNIFORMi(v2ui, Vector2ui);
|
||||
// VERIFY_UNIFORMi(v3ui, Vector3ui);
|
||||
// VERIFY_UNIFORMi(v4ui, Vector4ui);
|
||||
}
|
||||
else
|
||||
std::cerr << "Warning: opengl 3.0 was not tested\n";
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user