Use RED format for ambient occlusion render target

[libs/gl.git] / source / uniform.h

diff --git a/source/uniform.h b/source/uniform.h
index 97a4e9ddf212074eb1e0c093d7cfddcabe40e187..8dc43ab5791a7e48b8055297003d5eb3c7f77e4b 100644 (file)
--- a/source/uniform.h
+++ b/source/uniform.h
@@ -1,11 +1,12 @@
 #ifndef MSP_GL_UNIFORM_H_
 #define MSP_GL_UNIFORM_H_
 
+#include <algorithm>
+#include "program.h"
+
 namespace Msp {
 namespace GL {
 
-class Program;
-
 class Uniform
 {
 protected:
@@ -16,86 +17,152 @@ private:
 public:
        virtual ~Uniform() { }
 
-       virtual void apply(int) const =0;
-       virtual Uniform *clone() const =0;
+       virtual void apply(int) const = 0;
+       virtual void store(const Program::UniformInfo &, void *) const = 0;
+       virtual Uniform *clone() const = 0;
 };
 
 
-class Uniform1i: public Uniform
+template<typename T>
+class UniformScalar: public Uniform
 {
+public:
+       typedef T BaseType;
+       typedef T Type;
+
 private:
-       int v;
+       Type value;
 
 public:
-       Uniform1i(int v_);
+       UniformScalar(Type v): value(v) { }
 
-       virtual void apply(int index) const;
-       virtual Uniform1i *clone() const;
-};
+       virtual void apply(int index) const
+       { apply(index, 1, &value); }
 
+       static void apply(int, unsigned, const T *);
 
-class Uniform1f: public Uniform
-{
-private:
-       float v;
+       virtual void store(const Program::UniformInfo &info, void *buffer) const
+       { store(info, buffer, &value); }
 
-public:
-       Uniform1f(float v_);
+       static void store(const Program::UniformInfo &, void *buffer, const T *value)
+       { *reinterpret_cast<T *>(buffer) = *value; }
 
-       virtual void apply(int index) const;
-       virtual Uniform1f *clone() const;
+       virtual UniformScalar *clone() const
+       { return new UniformScalar(value); }
 };
 
+typedef UniformScalar<int> Uniform1i;
+typedef UniformScalar<float> Uniform1f;
+
 
-class Uniform2f: public Uniform
+template<typename T, unsigned vecsize>
+class UniformVector: public Uniform
 {
+public:
+       typedef T BaseType;
+       typedef T Type[vecsize];
+
 private:
-       float v[2];
+       Type value;
 
 public:
-       Uniform2f(float v0, float v1);
+       UniformVector(const T *vp)
+       { std::copy(vp, vp+vecsize, value); }
 
-       virtual void apply(int index) const;
-       virtual Uniform2f *clone() const;
-};
+       virtual void apply(int index) const
+       { apply(index, 1, value); }
 
+       static void apply(int index, unsigned size, const T *value);
 
-class Uniform3f: public Uniform
-{
-private:
-       float v[3];
+       virtual void store(const Program::UniformInfo &info, void *buffer) const
+       { store(info, buffer, value); }
 
-public:
-       Uniform3f(float v0, float v1, float v2);
+       static void store(const Program::UniformInfo &, void *buffer, const T *value)
+       { std::copy(value, value+vecsize, reinterpret_cast<T *>(buffer)); }
 
-       virtual void apply(int index) const;
-       virtual Uniform3f *clone() const;
+       virtual UniformVector *clone() const
+       { return new UniformVector(value); }
 };
 
+typedef UniformVector<int, 2> Uniform2i;
+typedef UniformVector<float, 2> Uniform2f;
+typedef UniformVector<int, 3> Uniform3i;
+typedef UniformVector<float, 3> Uniform3f;
+typedef UniformVector<int, 4> Uniform4i;
+typedef UniformVector<float, 4> Uniform4f;
+
 
-class Uniform4f: public Uniform
+template<typename T, unsigned rows, unsigned cols>
+class UniformMatrix: public Uniform
 {
+public:
+       typedef T BaseType;
+       typedef T Type[rows*cols];
+
 private:
-       float v[4];
+       Type value;
 
 public:
-       Uniform4f(float v0, float v1, float v2, float v3);
+       UniformMatrix(const T *vp)
+       { std::copy(vp, vp+rows*cols, value); }
+
+       virtual void apply(int index) const
+       { apply(index, 1, value); }
 
-       virtual void apply(int index) const;
-       virtual Uniform4f *clone() const;
+       static void apply(int index, unsigned size, const T *value);
+
+       virtual void store(const Program::UniformInfo &info, void *buffer) const
+       { store(info, buffer, value); }
+
+       static void store(const Program::UniformInfo &info, void *buffer, const T *value)
+       {
+               for(unsigned i=0; i<cols; ++i)
+                       UniformVector<T, rows>::store(info, reinterpret_cast<char *>(buffer)+i*info.matrix_stride, value+i*rows);
+       }
+
+       virtual UniformMatrix *clone() const
+       { return new UniformMatrix(value); }
 };
 
+typedef UniformMatrix<float, 2, 2> UniformMatrix2x2f;
+typedef UniformMatrix<float, 3, 3> UniformMatrix3x3f;
+typedef UniformMatrix<float, 4, 4> UniformMatrix4x4f;
 
-class UniformMatrix4x4f: public Uniform
+
+template<typename T>
+class UniformArray: public Uniform
 {
 private:
-       float v[16];
+       typedef typename T::BaseType BaseType;
+       enum { elemsize = sizeof(typename T::Type)/sizeof(typename T::BaseType) };
 
-public:
-       UniformMatrix4x4f(const float *);
+       BaseType *values;
+       unsigned size;
 
-       virtual void apply(int index) const;
-       virtual UniformMatrix4x4f *clone() const;
+public:
+       UniformArray(unsigned n, const BaseType *vp):
+               size(n)
+       {
+               values = new BaseType[elemsize*size];
+               std::copy(vp, vp+elemsize*size, values);
+       }
+
+       ~UniformArray()
+       {
+               delete[] values;
+       }
+
+       virtual void apply(int index) const
+       { T::apply(index, size, values); }
+
+       virtual void store(const Program::UniformInfo &info, void *buffer) const
+       {
+               for(unsigned i=0; i<size; ++i)
+                       T::store(info, reinterpret_cast<char *>(buffer)+i*info.array_stride, values+i*elemsize);
+       }
+
+       virtual UniformArray *clone() const
+       { return new UniformArray(size, values); }
 };
 
 } // namespace GL