GLSLProgram.h

Go to the documentation of this file.

//Code from : OpenGL 4 Shading Language Cookbook Second Edition, by David Wolff
// [MK] Modified to only support simple vertex/fragment shaders
#ifndef GLSLPROGRAM_H
#define GLSLPROGRAM_H
 
#include <GL/glew.h>
#include <string>
#include <stdexcept>
#include <Util/exceptions.h>
 
#pragma warning( disable : 4290 )
 
namespace GLSLShaderInfo
{
    class shader_file_extension
    {
    public:
        std::string ext;
        GLenum type;
    };
 
    const shader_file_extension extensions[] =
    {
        { ".vs"   , GL_VERTEX_SHADER },
        { ".vert" , GL_VERTEX_SHADER },
        { ".fs"   , GL_FRAGMENT_SHADER },
        { ".frag" , GL_FRAGMENT_SHADER },
    };
}
 
class GLSLProgram
{
    template< unsigned int Dim > static void glUniformiv      ( GLint location , GLsizei count ,                       const GLint*   value );
 
    template< unsigned int Dim > static void glUniformfv      ( GLint location , GLsizei count ,                       const GLfloat* value );
 
    template< unsigned int Dim > static void glUniformMatrixfv( GLint location , GLsizei count , GLboolean transpose , const GLfloat* value );
 
    // Make these private in order to make the object non-copyable
    GLSLProgram( const GLSLProgram & other ) { }
 
    GLSLProgram & operator=( const GLSLProgram &other ){ return *this; }
 
    GLuint _handle;
 
    bool _linked;
 
    std::string _vertexSource;
 
    std::string _fragmentSource;
public:
    static bool FileExists( const std::string & fileName );
 
    static std::string GetExtension( const std::string & fileName );
 
    static GLenum GetShaderType( const std::string &fileName );
 
    static std::string GetShaderTypeString( GLenum type );
 
    GLSLProgram( const std::string &vs_filename , const std::string &fs_filename );
 
    ~GLSLProgram( void );
 
    void compileShader( const std::string & source , GLenum type );
 
    void link( void );
 
    void validate( void );
 
    void use( void );
 
    GLuint handle( void ) const { return _handle; }
 
    bool linked( void ) const { return _linked; }
 
    void setUniform( const std::string &name , int    val , bool showWarning=true );
 
    void setUniform( const std::string &name , float  val , bool showWarning=true );
 
    void setUniform( const std::string &name , double val , bool showWarning=true );
 
    template< unsigned int > void setUniform( const std::string &name , const int*    v , bool showWarning=true );
 
    template< unsigned int > void setUniform( const std::string &name , const float*  v , bool showWarning=true );
 
    template< unsigned int > void setUniform( const std::string &name , const double* v , bool showWarning=true );
 
    template< unsigned int > void setUniformMatrix( const std::string &name , const float*  m , bool showWarning=true );
 
    template< unsigned int > void setUniformMatrix( const std::string &name , const double* m , bool showWarning=true );
 
    void init( void );
};
 
template< unsigned int Dim >
void GLSLProgram::setUniform( const std::string &name , const int* v , bool showWarning )
{
    GLint loc = glGetUniformLocation( _handle , name.c_str() );
    if( loc>=0 ) glUniformiv< Dim >( loc , 1 , v );
    else if( showWarning ) WARN( "Non-existant uniform: %s" , name.c_str() );
}
 
template< unsigned int Dim >
void GLSLProgram::setUniform( const std::string &name , const float* v , bool showWarning )
{
    GLint loc = glGetUniformLocation( _handle , name.c_str() );
    if( loc>=0 ) glUniformfv< Dim >( loc , 1 , v );
    else if( showWarning ) WARN( "Non-existant uniform: %s" , name.c_str() );
}
 
template< unsigned int Dim >
void GLSLProgram::setUniform( const std::string &name , const double* v , bool showWarning )
{
    float _v[Dim];
    for( int i=0 ; i<Dim ; i++ ) _v[i] = (float)v[i];
    setUniform< Dim >( name , _v , showWarning );
}
 
template< unsigned int Dim >
void GLSLProgram::setUniformMatrix( const std::string &name , const float* m , bool showWarning )
{
    GLint loc = glGetUniformLocation( _handle , name.c_str() );
    if( loc>=0 ) glUniformMatrixfv< Dim >( loc , 1 , GL_FALSE , m );
    else if( showWarning ) WARN( "Non-existant uniform: %s" , name.c_str() );
};
 
template< unsigned int Dim >
void GLSLProgram::setUniformMatrix( const std::string &name , const double* m , bool showWarning )
{
    float _m[Dim*Dim];
    for( int i=0 ; i<Dim*Dim ; i++ ) _m[i] = (float)m[i];
    setUniformMatrix< Dim >( name , _m , showWarning );
}
#endif // GLSLPROGRAM_H