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