source/programdata.cpp

   1 #include "buffer.h"
   2 #include "color.h"
   3 #include "error.h"
   4 #include "extension.h"
   5 #include "matrix.h"
   6 #include "program.h"
   7 #include "programdata.h"
   8 #include "uniform.h"
   9 #include "uniformblock.h"
  10 #include "vector.h"
  11
  12 using namespace std;
  13
  14 namespace Msp {
  15 namespace GL {
  16
  17 ProgramData::ProgramData():
  18         last_block(0),
  19         buffer(0),
  20         changes(NO_CHANGES)
  21 {
  22         static RequireExtension _ext("GL_ARB_shader_objects");
  23 }
  24
  25 // Blocks are intentionally left uncopied
  26 ProgramData::ProgramData(const ProgramData &other):
  27         uniforms(other.uniforms),
  28         last_block(0),
  29         buffer(0),
  30         changes(NO_CHANGES)
  31 {
  32         for(UniformMap::iterator i=uniforms.begin(); i!=uniforms.end(); ++i)
  33                 i->second = i->second->clone();
  34 }
  35
  36 ProgramData::~ProgramData()
  37 {
  38         for(UniformMap::iterator i=uniforms.begin(); i!=uniforms.end(); ++i)
  39                 delete i->second;
  40         for(BlockMap::iterator i=blocks.begin(); i!=blocks.end(); ++i)
  41                 delete i->second.block;
  42 }
  43
  44 void ProgramData::uniform(const string &name, Uniform *uni)
  45 {
  46         UniformMap::iterator i = uniforms.find(name);
  47         if(i!=uniforms.end())
  48         {
  49                 /* UniformBlock does not copy the uniforms, so existing blocks will be
  50                 left with stale pointers.  This is not a problem as long as no one stores
  51                 pointers to the blocks and expects them to stay valid. */
  52                 delete i->second;
  53                 i->second = uni;
  54                 changes = VALUES_CHANGED;
  55         }
  56         else
  57         {
  58                 uniforms[name] = uni;
  59                 changes = KEYS_CHANGED;
  60         }
  61 }
  62
  63 void ProgramData::uniform(const string &name, int v)
  64 {
  65         uniform(name, new Uniform1i(v));
  66 }
  67
  68 void ProgramData::uniform(const string &name, float v)
  69 {
  70         uniform(name, new Uniform1f(v));
  71 }
  72
  73 void ProgramData::uniform(const string &name, float v0, float v1)
  74 {
  75         float va[2] = { v0, v1 };
  76         uniform2(name, va);
  77 }
  78
  79 void ProgramData::uniform2(const string &name, const float *v)
  80 {
  81         uniform(name, new Uniform2f(v));
  82 }
  83
  84 void ProgramData::uniform(const string &name, float v0, float v1, float v2)
  85 {
  86         float va[3] = { v0, v1, v2 };
  87         uniform3(name, va);
  88 }
  89
  90 void ProgramData::uniform(const string &name, const Vector3 &v)
  91 {
  92         uniform(name, v.x, v.y, v.z);
  93 }
  94
  95 void ProgramData::uniform3(const string &name, const float *v)
  96 {
  97         uniform(name, new Uniform3f(v));
  98 }
  99
 100 void ProgramData::uniform(const string &name, float v0, float v1, float v2, float v3)
 101 {
 102         float va[4] = { v0, v1, v2, v3 };
 103         uniform4(name, va);
 104 }
 105
 106 void ProgramData::uniform(const string &name, const Vector4 &v)
 107 {
 108         uniform(name, v.x, v.y, v.z, v.w);
 109 }
 110
 111 void ProgramData::uniform(const string &name, const Color &c)
 112 {
 113         uniform(name, c.r, c.g, c.b, c.a);
 114 }
 115
 116 void ProgramData::uniform4(const string &name, const float *v)
 117 {
 118         uniform(name, new Uniform4f(v));
 119 }
 120
 121 void ProgramData::uniform_matrix4(const string &name, const float *v)
 122 {
 123         uniform(name, new UniformMatrix4x4f(v));
 124 }
 125
 126 void ProgramData::uniform_matrix4(const string &name, const Matrix &m)
 127 {
 128         float v[16];
 129         copy(m.data(), m.data()+16, v);
 130         uniform_matrix4(name, v);
 131 }
 132
 133 void ProgramData::uniform1_array(const string &name, unsigned n, const float *v)
 134 {
 135         uniform(name, new UniformArray<Uniform1f>(n, v));
 136 }
 137
 138 void ProgramData::uniform2_array(const string &name, unsigned n, const float *v)
 139 {
 140         uniform(name, new UniformArray<Uniform2f>(n, v));
 141 }
 142
 143 void ProgramData::uniform3_array(const string &name, unsigned n, const float *v)
 144 {
 145         uniform(name, new UniformArray<Uniform3f>(n, v));
 146 }
 147
 148 void ProgramData::uniform4_array(const string &name, unsigned n, const float *v)
 149 {
 150         uniform(name, new UniformArray<Uniform4f>(n, v));
 151 }
 152
 153 void ProgramData::uniform_matrix4_array(const string &name, unsigned n, const float *v)
 154 {
 155         uniform(name, new UniformArray<UniformMatrix4x4f>(n, v));
 156 }
 157
 158 void ProgramData::find_uniforms_for_block(Block &block, const Program::UniformBlockInfo &info) const
 159 {
 160         block.uniforms.clear();
 161         for(vector<const Program::UniformInfo *>::const_iterator i=info.uniforms.begin(); i!=info.uniforms.end(); ++i)
 162         {
 163                 // XXX individual array elements
 164                 UniformMap::const_iterator j = uniforms.find((*i)->name);
 165                 if(j!=uniforms.end())
 166                         block.uniforms[(*i)->location] = &j->second;
 167         }
 168 }
 169
 170 UniformBlock *ProgramData::create_block(const Program::UniformBlockInfo &info) const
 171 {
 172         UniformBlock *block = new UniformBlock(info.data_size);
 173         if(!buffer)
 174                 buffer = new Buffer(UNIFORM_BUFFER);
 175         block->use_buffer(buffer, last_block);
 176         last_block = block;
 177         return block;
 178 }
 179
 180 const UniformBlock *ProgramData::get_block(const Program &prog, const Program::UniformBlockInfo *info) const
 181 {
 182         if(changes)
 183         {
 184                 for(BlockMap::iterator i=blocks.begin(); i!=blocks.end(); ++i)
 185                         if(i->second.changes<changes)
 186                                 i->second.changes = changes;
 187                 changes = NO_CHANGES;
 188         }
 189
 190         unsigned layout_hash = (info ? info->layout_hash : prog.get_uniform_layout_hash());
 191
 192         map<unsigned, Block>::iterator i = blocks.find(layout_hash);
 193         if(i==blocks.end())
 194         {
 195                 i = blocks.insert(BlockMap::value_type(layout_hash, Block())).first;
 196                 if(info)
 197                 {
 198                         find_uniforms_for_block(i->second, *info);
 199
 200                         if(!i->second.uniforms.empty())
 201                         {
 202                                 i->second.block = create_block(*info);
 203                                 i->second.changes = VALUES_CHANGED;
 204                         }
 205                 }
 206                 else
 207                 {
 208                         i->second.block = new UniformBlock;
 209                         i->second.changes = VALUES_CHANGED;
 210                 }
 211         }
 212         else if(info && i->second.changes==KEYS_CHANGED)
 213         {
 214                 find_uniforms_for_block(i->second, *info);
 215                 if(!i->second.uniforms.empty())
 216                 {
 217                         if(!i->second.block)
 218                                 i->second.block = create_block(*info);
 219                         i->second.changes = VALUES_CHANGED;
 220                 }
 221                 else
 222                         i->second.changes = NO_CHANGES;
 223         }
 224
 225         if(!i->second.block)
 226                 return 0;
 227
 228         UniformBlock &block = *i->second.block;
 229         if(i->second.changes)
 230         {
 231                 if(info)
 232                 {
 233                         vector<const Program::UniformInfo *>::const_iterator j = info->uniforms.begin();
 234                         map<unsigned, const Uniform *const *>::const_iterator k = i->second.uniforms.begin();
 235                         while(j!=info->uniforms.end() && k!=i->second.uniforms.end())
 236                         {
 237                                 if(k->first==(*j)->location)
 238                                 {
 239                                         block.attach(**j, **k->second);
 240                                         ++k;
 241                                 }
 242                                 ++j;
 243                         }
 244                 }
 245                 else
 246                 {
 247                         for(UniformMap::const_iterator j=uniforms.begin(); j!=uniforms.end(); ++j)
 248                         {
 249                                 int loc = prog.get_uniform_location(j->first);
 250                                 if(loc>=0)
 251                                         block.attach(loc, *j->second);
 252                         }
 253                 }
 254                 i->second.changes = NO_CHANGES;
 255         }
 256
 257         return &block;
 258 }
 259
 260 const UniformBlock *ProgramData::get_block(const Program &prog, const string &name) const
 261 {
 262         if(name.empty())
 263                 return get_block(prog, 0);
 264         else
 265                 return get_block(prog, &prog.get_uniform_block_info(name));
 266 }
 267
 268 void ProgramData::apply() const
 269 {
 270         const Program *prog = Program::current();
 271         if(!prog)
 272                 throw invalid_operation("ProgramData::apply");
 273
 274         const Program::UniformBlockMap &prog_blocks = prog->get_uniform_blocks();
 275         if(!prog_blocks.empty())
 276         {
 277                 typedef pair<const UniformBlock *, unsigned> ApplyBlock;
 278                 list<ApplyBlock> apply_blocks;
 279                 for(Program::UniformBlockMap::const_iterator i=prog_blocks.begin(); i!=prog_blocks.end(); ++i)
 280                         if(const UniformBlock *block = get_block(*prog, &i->second))
 281                                 apply_blocks.push_back(make_pair(block, i->second.bind_point));
 282
 283                 if(buffer)
 284                         buffer->bind();
 285
 286                 for(list<ApplyBlock>::const_iterator i=apply_blocks.begin(); i!=apply_blocks.end(); ++i)
 287                         i->first->apply(i->second);
 288         }
 289
 290         if(const UniformBlock *block = get_block(*prog, 0))
 291                 block->apply(-1);
 292 }
 293
 294
 295 ProgramData::Block::Block():
 296         changes(NO_CHANGES),
 297         block(0)
 298 { }
 299
 300
 301 ProgramData::Loader::Loader(ProgramData &pd):
 302         DataFile::ObjectLoader<ProgramData>(pd)
 303 {
 304         add("uniform1i", &Loader::uniform1i);
 305         add("uniform1f", &Loader::uniform1f);
 306         add("uniform2f", &Loader::uniform2f);
 307         add("uniform3f", &Loader::uniform3f);
 308         add("uniform4f", &Loader::uniform4f);
 309 }
 310
 311 void ProgramData::Loader::uniform1i(const string &n, int v)
 312 {
 313         obj.uniform(n, v);
 314 }
 315
 316 void ProgramData::Loader::uniform1f(const string &n, float v)
 317 {
 318         obj.uniform(n, v);
 319 }
 320
 321 void ProgramData::Loader::uniform2f(const string &n, float v0, float v1)
 322 {
 323         obj.uniform(n, v0, v1);
 324 }
 325
 326 void ProgramData::Loader::uniform3f(const string &n, float v0, float v1, float v2)
 327 {
 328         obj.uniform(n, v0, v1, v2);
 329 }
 330
 331 void ProgramData::Loader::uniform4f(const string &n, float v0, float v1, float v2, float v3)
 332 {
 333         obj.uniform(n, v0, v1, v2, v3);
 334 }
 335
 336 } // namespace GL
 337 } // namespace Msp