1 #include <msp/core/maputils.h>
2 #include <msp/gl/extensions/arb_direct_state_access.h>
8 #include "programdata.h"
10 #include "uniformblock.h"
18 ProgramData::ProgramData(const Program *p):
25 // Blocks are intentionally left uncopied
26 ProgramData::ProgramData(const ProgramData &other):
28 uniforms(other.uniforms),
33 for(vector<NamedUniform>::iterator i=uniforms.begin(); i!=uniforms.end(); ++i)
34 i->value = i->value->clone();
37 ProgramData &ProgramData::operator=(const ProgramData &other)
39 tied_program = other.tied_program;
41 uniforms = other.uniforms;
42 for(vector<NamedUniform>::iterator i=uniforms.begin(); i!=uniforms.end(); ++i)
43 i->value = i->value->clone();
45 for(BlockMap::iterator i=blocks.begin(); i!=blocks.end(); ++i)
46 delete i->second.block;
56 ProgramData::~ProgramData()
58 for(vector<NamedUniform>::iterator i=uniforms.begin(); i!=uniforms.end(); ++i)
60 for(BlockMap::iterator i=blocks.begin(); i!=blocks.end(); ++i)
62 if(i->second.indices.type_flag==0xFE)
63 delete i->second.indices.dynamic.values;
64 delete i->second.block;
69 void ProgramData::uniform(const string &name, Uniform *uni)
74 tied_program->get_uniform_info(name);
75 else if(name[name.size()-1]==']')
76 throw invalid_argument("ProgramData::uniform");
84 int i = find_uniform_index(name);
87 uniforms[i].replace_value(uni);
89 if(static_cast<unsigned>(i)<MASK_BITS)
91 else // Force a full update if the mask isn't wide enough
97 vector<NamedUniform>::iterator j = lower_bound(uniforms.begin(), uniforms.end(), name, uniform_name_compare);
102 uniforms.insert(j, nu);
107 void ProgramData::uniform(const string &name, const Uniform &u)
109 uniform(name, u.clone());
112 void ProgramData::uniform(const string &name, int v)
114 uniform(name, new Uniform1i(v));
117 void ProgramData::uniform(const string &name, float v)
119 uniform(name, new Uniform1f(v));
122 void ProgramData::uniform(const string &name, int v0, int v1)
124 int va[2] = { v0, v1 };
128 void ProgramData::uniform(const string &name, float v0, float v1)
130 float va[2] = { v0, v1 };
134 void ProgramData::uniform2(const string &name, const int *v)
136 uniform(name, new Uniform2i(v));
139 void ProgramData::uniform2(const string &name, const float *v)
141 uniform(name, new Uniform2f(v));
144 void ProgramData::uniform(const string &name, int v0, int v1, int v2)
146 int va[3] = { v0, v1, v2 };
150 void ProgramData::uniform(const string &name, float v0, float v1, float v2)
152 float va[3] = { v0, v1, v2 };
156 void ProgramData::uniform(const string &name, const Vector3 &v)
158 uniform(name, v.x, v.y, v.z);
161 void ProgramData::uniform3(const string &name, const int *v)
163 uniform(name, new Uniform3i(v));
166 void ProgramData::uniform3(const string &name, const float *v)
168 uniform(name, new Uniform3f(v));
171 void ProgramData::uniform(const string &name, int v0, int v1, int v2, int v3)
173 int va[4] = { v0, v1, v2, v3 };
177 void ProgramData::uniform(const string &name, float v0, float v1, float v2, float v3)
179 float va[4] = { v0, v1, v2, v3 };
183 void ProgramData::uniform(const string &name, const Vector4 &v)
185 uniform(name, v.x, v.y, v.z, v.w);
188 void ProgramData::uniform(const string &name, const Color &c)
190 uniform(name, c.r, c.g, c.b, c.a);
193 void ProgramData::uniform4(const string &name, const int *v)
195 uniform(name, new Uniform4i(v));
198 void ProgramData::uniform4(const string &name, const float *v)
200 uniform(name, new Uniform4f(v));
203 void ProgramData::uniform(const string &name, const LinAl::Matrix<float, 2, 2> &m)
205 uniform_matrix2(name, &m(0, 0));
208 void ProgramData::uniform_matrix2(const string &name, const float *v)
210 uniform(name, new UniformMatrix2x2f(v));
213 void ProgramData::uniform(const string &name, const LinAl::Matrix<float, 3, 3> &m)
215 uniform_matrix3(name, &m(0, 0));
218 void ProgramData::uniform_matrix3(const string &name, const float *v)
220 uniform(name, new UniformMatrix3x3f(v));
223 void ProgramData::uniform(const string &name, const Matrix &m)
225 uniform_matrix4(name, m.data());
228 void ProgramData::uniform_matrix4(const string &name, const float *v)
230 uniform(name, new UniformMatrix4x4f(v));
233 void ProgramData::uniform1_array(const string &name, unsigned n, const int *v)
235 uniform(name, new UniformArray<Uniform1i>(n, v));
238 void ProgramData::uniform1_array(const string &name, unsigned n, const float *v)
240 uniform(name, new UniformArray<Uniform1f>(n, v));
243 void ProgramData::uniform2_array(const string &name, unsigned n, const int *v)
245 uniform(name, new UniformArray<Uniform2i>(n, v));
248 void ProgramData::uniform2_array(const string &name, unsigned n, const float *v)
250 uniform(name, new UniformArray<Uniform2f>(n, v));
253 void ProgramData::uniform3_array(const string &name, unsigned n, const int *v)
255 uniform(name, new UniformArray<Uniform3i>(n, v));
258 void ProgramData::uniform3_array(const string &name, unsigned n, const float *v)
260 uniform(name, new UniformArray<Uniform3f>(n, v));
263 void ProgramData::uniform4_array(const string &name, unsigned n, const int *v)
265 uniform(name, new UniformArray<Uniform4i>(n, v));
268 void ProgramData::uniform4_array(const string &name, unsigned n, const float *v)
270 uniform(name, new UniformArray<Uniform4f>(n, v));
273 void ProgramData::uniform_matrix2_array(const string &name, unsigned n, const float *v)
275 uniform(name, new UniformArray<UniformMatrix2x2f>(n, v));
278 void ProgramData::uniform_matrix3_array(const string &name, unsigned n, const float *v)
280 uniform(name, new UniformArray<UniformMatrix3x3f>(n, v));
283 void ProgramData::uniform_matrix4_array(const string &name, unsigned n, const float *v)
285 uniform(name, new UniformArray<UniformMatrix4x4f>(n, v));
288 void ProgramData::remove_uniform(const string &name)
290 vector<NamedUniform>::const_iterator i = lower_bound(uniforms.begin(), uniforms.end(), name, uniform_name_compare);
291 if(i==uniforms.end() || i->name!=name)
300 vector<string> ProgramData::get_uniform_names() const
302 vector<string> names;
303 names.reserve(uniforms.size());
304 for(vector<NamedUniform>::const_iterator i=uniforms.begin(); i!=uniforms.end(); ++i)
305 names.push_back(i->name);
309 const Uniform &ProgramData::get_uniform(const string &name) const
311 int i = find_uniform_index(name);
313 throw key_error(name);
314 return *uniforms[i].value;
317 bool ProgramData::uniform_name_compare(const NamedUniform &nu, const string &name)
322 int ProgramData::find_uniform_index(const string &name) const
324 vector<NamedUniform>::const_iterator i = lower_bound(uniforms.begin(), uniforms.end(), name, uniform_name_compare);
325 return ((i!=uniforms.end() && i->name==name) ? i-uniforms.begin() : -1);
328 void ProgramData::update_block_uniform_indices(SharedBlock &block, const Program::UniformBlockInfo &info) const
330 UInt8 *indices = block.indices.values;
331 if(info.uniforms.size()>16)
333 if(block.indices.type_flag==0xFD)
335 block.indices.dynamic.values = new UInt8[info.uniforms.size()];
336 block.indices.type_flag = 0xFE;
338 indices = block.indices.dynamic.values;
342 for(unsigned i=0; i<info.uniforms.size(); ++i)
344 int j = find_uniform_index(info.uniforms[i]->name);
348 if(static_cast<unsigned>(j)<MASK_BITS)
355 block.dirty = block.used;
358 void ProgramData::update_block(SharedBlock &block, const Program::UniformBlockInfo &info) const
360 const UInt8 *indices = block.get_uniform_indices();
361 for(unsigned i=0; i<info.uniforms.size(); ++i)
363 block.block->attach(*info.uniforms[i], *uniforms[indices[i]].value);
366 ProgramData::SharedBlock *ProgramData::get_shared_block(const Program::UniformBlockInfo &info) const
368 BlockMap::iterator i = blocks.find(info.layout_hash);
371 bool any_found = false;
372 for(vector<const Program::UniformInfo *>::const_iterator j=info.uniforms.begin(); (!any_found && j!=info.uniforms.end()); ++j)
373 any_found = (find_uniform_index((*j)->name)>=0);
375 // TODO throw if all uniforms for a buffer-backed block are not found
380 if(info.bind_point>=0)
384 buffer = new Buffer(UNIFORM_BUFFER);
385 buffer->set_usage(STREAM_DRAW);
388 block = new UniformBlock(info.data_size);
389 block->use_buffer(buffer, last_block);
393 block = new UniformBlock;
395 i = blocks.insert(BlockMap::value_type(info.layout_hash, SharedBlock(block))).first;
396 update_block_uniform_indices(i->second, info);
402 void ProgramData::apply() const
404 const Program *prog = Program::current();
406 throw invalid_operation("ProgramData::apply");
408 Program::LayoutHash layout = prog->get_uniform_layout_hash();
409 ProgramUniforms &pu = programs[layout];
411 Mask force_dirty = (dirty==ALL_ONES ? ALL_ONES : 0U);
412 Mask affected = (dirty&pu.used) | force_dirty;
413 if(affected|pu.dirty)
415 /* If the global dirty flag affects this program, add it to per-program
416 dirty flags and clear the global flag. A previously unseen program will
417 always cause this to happen. */
420 for(BlockMap::iterator i=blocks.begin(); i!=blocks.end(); ++i)
421 i->second.dirty |= (dirty&i->second.used) | force_dirty;
422 for(ProgramMap::iterator i=programs.begin(); i!=programs.end(); ++i)
423 i->second.dirty |= (dirty&i->second.used) | force_dirty;
427 const Program::UniformBlockMap &prog_blocks = prog->get_uniform_blocks();
429 if(pu.dirty==ALL_ONES)
431 /* The set of uniforms has changed since this program was last used.
432 Regenerate the list of uniform blocks. */
434 pu.blocks.reserve(prog_blocks.size());
437 for(Program::UniformBlockMap::const_iterator i=prog_blocks.begin(); i!=prog_blocks.end(); ++i)
439 SharedBlock *shared = get_shared_block(i->second);
442 if(shared->dirty==ALL_ONES)
443 update_block_uniform_indices(*shared, i->second);
444 pu.used |= shared->used;
447 pu.blocks.push_back(ProgramBlock(i->second.bind_point, shared));
451 // Update the contents of all dirty blocks.
452 bool buffered_blocks_updated = false;
453 std::vector<ProgramBlock>::iterator j = pu.blocks.begin();
454 for(Program::UniformBlockMap::const_iterator i=prog_blocks.begin(); i!=prog_blocks.end(); ++i, ++j)
456 if(!j->shared || !j->shared->dirty)
459 update_block(*j->shared, i->second);
460 j->shared->dirty = 0;
461 buffered_blocks_updated |= (j->bind_point>=0);
466 /* If any blocks stored in the buffer were updated, bind the buffer here
467 to avoid state thrashing. */
468 if(buffered_blocks_updated && !ARB_direct_state_access)
472 for(vector<ProgramBlock>::iterator i=pu.blocks.begin(); i!=pu.blocks.end(); ++i)
474 i->block->apply(i->bind_point);
478 ProgramData::NamedUniform::NamedUniform():
482 void ProgramData::NamedUniform::replace_value(Uniform *v)
484 /* UniformBlock does not copy the uniforms, so existing default blocks
485 will be left with stale pointers. This is not a problem as long as no
486 one stores pointers to the blocks and expects them to stay valid. */
492 ProgramData::SharedBlock::SharedBlock(UniformBlock *b):
497 indices.type_flag = 0xFD;
500 const UInt8 *ProgramData::SharedBlock::get_uniform_indices() const
502 return (indices.type_flag==0xFE ? indices.dynamic.values : indices.values);
506 ProgramData::ProgramBlock::ProgramBlock():
512 ProgramData::ProgramBlock::ProgramBlock(int p, SharedBlock *b):
514 block((b && b->used) ? b->block : 0),
519 ProgramData::ProgramUniforms::ProgramUniforms():
525 ProgramData::Loader::Loader(ProgramData &pd):
526 DataFile::ObjectLoader<ProgramData>(pd)
528 add("uniform", &Loader::uniform1i);
529 add("uniform1i", &Loader::uniform1i);
530 add("uniform", &Loader::uniform1f);
531 add("uniform1f", &Loader::uniform1f);
532 add("uniform", &Loader::uniform2i);
533 add("uniform2i", &Loader::uniform2i);
534 add("uniform", &Loader::uniform2f);
535 add("uniform2f", &Loader::uniform2f);
536 add("uniform", &Loader::uniform3i);
537 add("uniform3i", &Loader::uniform3i);
538 add("uniform", &Loader::uniform3f);
539 add("uniform3f", &Loader::uniform3f);
540 add("uniform", &Loader::uniform4i);
541 add("uniform4i", &Loader::uniform4i);
542 add("uniform", &Loader::uniform4f);
543 add("uniform4f", &Loader::uniform4f);
544 add("uniform1i_array", &Loader::uniform1i_array);
545 add("uniform1f_array", &Loader::uniform1f_array);
546 add("uniform2f_array", &Loader::uniform2f_array);
547 add("uniform3f_array", &Loader::uniform3f_array);
548 add("uniform4f_array", &Loader::uniform4f_array);
549 add("uniform_array", &Loader::uniform_array);
552 void ProgramData::Loader::uniform1i(const string &n, int v)
557 void ProgramData::Loader::uniform1f(const string &n, float v)
562 void ProgramData::Loader::uniform2i(const string &n, int v0, int v1)
564 obj.uniform(n, v0, v1);
567 void ProgramData::Loader::uniform2f(const string &n, float v0, float v1)
569 obj.uniform(n, v0, v1);
572 void ProgramData::Loader::uniform3i(const string &n, int v0, int v1, int v2)
574 obj.uniform(n, v0, v1, v2);
577 void ProgramData::Loader::uniform3f(const string &n, float v0, float v1, float v2)
579 obj.uniform(n, v0, v1, v2);
582 void ProgramData::Loader::uniform4i(const string &n, int v0, int v1, int v2, int v3)
584 obj.uniform(n, v0, v1, v2, v3);
587 void ProgramData::Loader::uniform4f(const string &n, float v0, float v1, float v2, float v3)
589 obj.uniform(n, v0, v1, v2, v3);
592 void ProgramData::Loader::uniform_array_(const string &n, DataType t, unsigned e)
594 ArrayLoader ldr(t, e);
596 unsigned size = ldr.get_size();
598 throw logic_error("empty uniform array");
600 DataType type = ldr.get_data_type();
601 unsigned elem_size = ldr.get_element_size();
604 const int *data = reinterpret_cast<const int *>(ldr.get_data());
606 obj.uniform1_array(n, size, data);
607 else if(elem_size==2)
608 obj.uniform2_array(n, size, data);
609 else if(elem_size==3)
610 obj.uniform3_array(n, size, data);
611 else if(elem_size==4)
612 obj.uniform4_array(n, size, data);
614 throw logic_error("unsupported combination of array type and element size");
618 const float *data = reinterpret_cast<const float *>(ldr.get_data());
620 obj.uniform1_array(n, size, data);
621 else if(elem_size==2)
622 obj.uniform2_array(n, size, data);
623 else if(elem_size==3)
624 obj.uniform3_array(n, size, data);
625 else if(elem_size==4)
626 obj.uniform4_array(n, size, data);
628 throw logic_error("unsupported combination of array type and element size");
631 throw logic_error("unsupported array type");
634 void ProgramData::Loader::uniform1i_array(const string &n)
636 uniform_array_(n, INT, 1);
639 void ProgramData::Loader::uniform1f_array(const string &n)
641 uniform_array_(n, FLOAT, 1);
644 void ProgramData::Loader::uniform2i_array(const string &n)
646 uniform_array_(n, INT, 2);
649 void ProgramData::Loader::uniform2f_array(const string &n)
651 uniform_array_(n, FLOAT, 2);
654 void ProgramData::Loader::uniform3i_array(const string &n)
656 uniform_array_(n, INT, 3);
659 void ProgramData::Loader::uniform3f_array(const string &n)
661 uniform_array_(n, FLOAT, 3);
664 void ProgramData::Loader::uniform4i_array(const string &n)
666 uniform_array_(n, INT, 4);
669 void ProgramData::Loader::uniform4f_array(const string &n)
671 uniform_array_(n, FLOAT, 4);
674 void ProgramData::Loader::uniform_array(const string &n)
676 uniform_array_(n, static_cast<DataType>(0), 0);
680 ProgramData::ArrayLoader::ArrayLoader(DataType t, unsigned e):
684 add("uniform", &ArrayLoader::uniform1i);
685 add("uniform1i", &ArrayLoader::uniform1i);
686 add("uniform", &ArrayLoader::uniform1f);
687 add("uniform1f", &ArrayLoader::uniform1f);
688 add("uniform", &ArrayLoader::uniform2f);
689 add("uniform2f", &ArrayLoader::uniform2f);
690 add("uniform", &ArrayLoader::uniform3f);
691 add("uniform3f", &ArrayLoader::uniform3f);
692 add("uniform", &ArrayLoader::uniform4f);
693 add("uniform4f", &ArrayLoader::uniform4f);
696 void ProgramData::ArrayLoader::uniform(DataType t, unsigned e, const void *v)
698 if(element_size && (t!=type || e!=element_size))
699 throw logic_error("heterogeneous array contents");
707 const char *cv = reinterpret_cast<const char *>(v);
708 data.insert(data.end(), cv, cv+element_size*4);
711 void ProgramData::ArrayLoader::uniform1i(int v)
716 void ProgramData::ArrayLoader::uniform1f(float v)
718 uniform(FLOAT, 1, &v);
721 void ProgramData::ArrayLoader::uniform2i(int v0, int v1)
723 int va[2] = { v0, v1 };
727 void ProgramData::ArrayLoader::uniform2f(float v0, float v1)
729 float va[2] = { v0, v1 };
730 uniform(FLOAT, 2, va);
733 void ProgramData::ArrayLoader::uniform3i(int v0, int v1, int v2)
735 int va[3] = { v0, v1, v2 };
739 void ProgramData::ArrayLoader::uniform3f(float v0, float v1, float v2)
741 float va[3] = { v0, v1, v2 };
742 uniform(FLOAT, 3, va);
745 void ProgramData::ArrayLoader::uniform4i(int v0, int v1, int v2, int v3)
747 int va[4] = { v0, v1, v2, v3 };
751 void ProgramData::ArrayLoader::uniform4f(float v0, float v1, float v2, float v3)
753 float va[4] = { v0, v1, v2, v3 };
754 uniform(FLOAT, 4, va);