source/programdata.cpp

   1 #include "color.h"
   2 #include "error.h"
   3 #include "extension.h"
   4 #include "matrix.h"
   5 #include "program.h"
   6 #include "programdata.h"
   7 #include "uniform.h"
   8 #include "uniformblock.h"
   9 #include "vector.h"
  10
  11 using namespace std;
  12
  13 namespace Msp {
  14 namespace GL {
  15
  16 ProgramData::ProgramData():
  17         modified(false)
  18 {
  19         static RequireExtension _ext("GL_ARB_shader_objects");
  20 }
  21
  22 // Blocks are intentionally left uncopied
  23 ProgramData::ProgramData(const ProgramData &other):
  24         uniforms(other.uniforms),
  25         modified(false)
  26 {
  27         for(UniformMap::iterator i=uniforms.begin(); i!=uniforms.end(); ++i)
  28                 i->second = i->second->clone();
  29 }
  30
  31 ProgramData::~ProgramData()
  32 {
  33         for(UniformMap::iterator i=uniforms.begin(); i!=uniforms.end(); ++i)
  34                 delete i->second;
  35         for(BlockMap::iterator i=blocks.begin(); i!=blocks.end(); ++i)
  36                 delete i->second.block;
  37 }
  38
  39 void ProgramData::uniform(const string &name, Uniform *uni)
  40 {
  41         UniformMap::iterator i = uniforms.find(name);
  42         if(i!=uniforms.end())
  43         {
  44                 delete i->second;
  45                 i->second = uni;
  46         }
  47         else
  48                 uniforms[name] = uni;
  49
  50         modified = true;
  51 }
  52
  53 void ProgramData::uniform(const string &name, int v)
  54 {
  55         uniform(name, new Uniform1i(v));
  56 }
  57
  58 void ProgramData::uniform(const string &name, float v)
  59 {
  60         uniform(name, new Uniform1f(v));
  61 }
  62
  63 void ProgramData::uniform(const string &name, float v0, float v1)
  64 {
  65         uniform(name, new Uniform2f(v0, v1));
  66 }
  67
  68 void ProgramData::uniform2(const string &name, const float *v)
  69 {
  70         uniform(name, v[0], v[1]);
  71 }
  72
  73 void ProgramData::uniform(const string &name, float v0, float v1, float v2)
  74 {
  75         uniform(name, new Uniform3f(v0, v1, v2));
  76 }
  77
  78 void ProgramData::uniform(const string &name, const Vector3 &v)
  79 {
  80         uniform(name, v.x, v.y, v.z);
  81 }
  82
  83 void ProgramData::uniform3(const string &name, const float *v)
  84 {
  85         uniform(name, v[0], v[1], v[2]);
  86 }
  87
  88 void ProgramData::uniform(const string &name, float v0, float v1, float v2, float v3)
  89 {
  90         uniform(name, new Uniform4f(v0, v1, v2, v3));
  91 }
  92
  93 void ProgramData::uniform(const string &name, const Vector4 &v)
  94 {
  95         uniform(name, v.x, v.y, v.z, v.w);
  96 }
  97
  98 void ProgramData::uniform(const string &name, const Color &c)
  99 {
 100         uniform(name, c.r, c.g, c.b, c.a);
 101 }
 102
 103 void ProgramData::uniform4(const string &name, const float *v)
 104 {
 105         uniform(name, v[0], v[1], v[2], v[3]);
 106 }
 107
 108 void ProgramData::uniform_matrix4(const string &name, const float *v)
 109 {
 110         uniform(name, new UniformMatrix4x4f(v));
 111 }
 112
 113 void ProgramData::uniform_matrix4(const string &name, const Matrix &m)
 114 {
 115         float v[16];
 116         copy(m.data(), m.data()+16, v);
 117         uniform_matrix4(name, v);
 118 }
 119
 120 const UniformBlock &ProgramData::get_block(const Program &prog) const
 121 {
 122         if(modified)
 123         {
 124                 for(BlockMap::iterator i=blocks.begin(); i!=blocks.end(); ++i)
 125                         i->second.dirty = true;
 126                 modified = false;
 127         }
 128
 129         unsigned layout_hash = prog.get_uniform_layout_hash();
 130         map<unsigned, Block>::iterator i = blocks.find(layout_hash);
 131         if(i==blocks.end())
 132         {
 133                 i = blocks.insert(BlockMap::value_type(layout_hash, Block())).first;
 134                 i->second.dirty = true;
 135                 i->second.block = new UniformBlock;
 136         }
 137
 138         UniformBlock &block = *i->second.block;
 139         if(i->second.dirty)
 140         {
 141                 for(UniformMap::const_iterator j=uniforms.begin(); j!=uniforms.end(); ++j)
 142                 {
 143                         int loc = prog.get_uniform_location(j->first);
 144                         if(loc>=0)
 145                                 block.uniform(loc, *j->second);
 146                 }
 147                 i->second.dirty = false;
 148         }
 149
 150         return block;
 151 }
 152
 153 void ProgramData::apply() const
 154 {
 155         const Program *prog = Program::current();
 156         if(!prog)
 157                 throw invalid_operation("ProgramData::apply");
 158
 159         const UniformBlock &block = get_block(*prog);
 160         block.apply(-1);
 161 }
 162
 163
 164 ProgramData::Block::Block():
 165         dirty(false),
 166         block(0)
 167 { }
 168
 169
 170 ProgramData::Loader::Loader(ProgramData &pd):
 171         DataFile::ObjectLoader<ProgramData>(pd)
 172 {
 173         add("uniform1i", &Loader::uniform1i);
 174         add("uniform1f", &Loader::uniform1f);
 175         add("uniform2f", &Loader::uniform2f);
 176         add("uniform3f", &Loader::uniform3f);
 177         add("uniform4f", &Loader::uniform4f);
 178 }
 179
 180 void ProgramData::Loader::uniform1i(const string &n, int v)
 181 {
 182         obj.uniform(n, v);
 183 }
 184
 185 void ProgramData::Loader::uniform1f(const string &n, float v)
 186 {
 187         obj.uniform(n, v);
 188 }
 189
 190 void ProgramData::Loader::uniform2f(const string &n, float v0, float v1)
 191 {
 192         obj.uniform(n, v0, v1);
 193 }
 194
 195 void ProgramData::Loader::uniform3f(const string &n, float v0, float v1, float v2)
 196 {
 197         obj.uniform(n, v0, v1, v2);
 198 }
 199
 200 void ProgramData::Loader::uniform4f(const string &n, float v0, float v1, float v2, float v3)
 201 {
 202         obj.uniform(n, v0, v1, v2, v3);
 203 }
 204
 205 } // namespace GL
 206 } // namespace Msp