source/programdata.cpp

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