FnGLShaderProgram.h

// Copyright (c) 2016 The Foundry Visionmongers Ltd. All Rights Reserved.

#ifndef GLSHADERPROGRAM_H_
#define GLSHADERPROGRAM_H_

#ifdef _WIN32
#include <Windows.h>
#endif

#include <FnAttribute/FnAttribute.h>
#include <FnViewer/plugin/FnMathTypes.h>

#include <GL/glew.h>
#include <string>
#include <map>

namespace Foundry
{
namespace Katana
{
namespace ViewerUtils
{

using Foundry::Katana::ViewerAPI::Matrix44d;

enum ShaderType
{
    VERTEX = GL_VERTEX_SHADER,
    FRAGMENT = GL_FRAGMENT_SHADER,
    GEOMETRY = GL_GEOMETRY_SHADER,
    TESS_CONTROL = GL_TESS_CONTROL_SHADER,
    TESS_EVALUATION = GL_TESS_EVALUATION_SHADER,
    COMPUTE = GL_COMPUTE_SHADER
};

class GLShaderProgram
{
public:
    GLShaderProgram();
    ~GLShaderProgram();

    void cleanup();

    void compileShader(const std::string& filename, ShaderType type);

    void compileShader(const std::string& name, ShaderType type, const std::string& shaderSource);

    GLuint getUniformLocation(const std::string& name);

    void link();

    void use();

    GLuint getHandle() const { return m_handle; }

    bool isLinked() const { return m_isLinked; }

    void bindAttribLocation(GLuint location, const std::string& name);

    void bindFragDataLocation(GLuint location, const std::string& name);

    void setUniform(const std::string& name, float val);

    void setUniform(const std::string& name, float x, float y);

    void setUniform(const std::string& name, float x, float y, float z);

    void setUniform(const std::string& name, float x, float y, float z, float w);

    void setUniform(const std::string& name, int val);

    void setUniform(const std::string& name, bool val);

    void setUniform(const std::string& name, FnAttribute::FloatAttribute val);

    void setUniform(const std::string& name, FnAttribute::DoubleAttribute val);

    void setUniform(const std::string& name, const Matrix44d& matrix);

    void setUniformMatrix4d(const std::string& name, const double* matrix);

private:
    const std::string readFile(const std::string& filename);
    GLuint m_handle;
    bool m_isLinked;
    std::map<std::string, GLuint> m_uniformLocations;
};

} // ViewerUtils
} // Katana
} // Foundry
#endif /* GLSHADERPROGRAM_H_ */