1 #include <msp/core/algorithm.h>
2 #include <msp/core/maputils.h>
3 #include <msp/debug/demangle.h>
4 #include <msp/gl/extensions/arb_direct_state_access.h>
5 #include <msp/io/print.h>
11 #include "programdata.h"
13 #include "uniformblock.h"
21 ProgramData::ProgramData(const Program *p):
28 // Blocks are intentionally left uncopied
29 ProgramData::ProgramData(const ProgramData &other):
30 tied_program(other.tied_program),
31 uniforms(other.uniforms),
36 for(vector<TaggedUniform>::iterator i=uniforms.begin(); i!=uniforms.end(); ++i)
37 i->value = i->value->clone();
40 ProgramData::ProgramData(const ProgramData &other, const Program *p):
48 for(vector<TaggedUniform>::const_iterator i=other.uniforms.begin(); i!=other.uniforms.end(); ++i)
52 uniforms = other.uniforms;
53 for(vector<TaggedUniform>::iterator i=uniforms.begin(); i!=uniforms.end(); ++i)
54 i->value = i->value->clone();
57 ProgramData &ProgramData::operator=(const ProgramData &other)
59 tied_program = other.tied_program;
61 uniforms = other.uniforms;
62 for(vector<TaggedUniform>::iterator i=uniforms.begin(); i!=uniforms.end(); ++i)
63 i->value = i->value->clone();
65 for(BlockMap::iterator i=blocks.begin(); i!=blocks.end(); ++i)
66 delete i->second.block;
76 ProgramData::~ProgramData()
78 for(vector<TaggedUniform>::iterator i=uniforms.begin(); i!=uniforms.end(); ++i)
80 for(BlockMap::iterator i=blocks.begin(); i!=blocks.end(); ++i)
82 if(i->second.indices.type_flag==0xFE)
83 delete[] i->second.indices.dynamic.values;
84 delete i->second.block;
89 void ProgramData::uniform(Tag tag, Uniform *uni)
93 if(!validate_tag(tag))
105 int i = find_uniform_index(tag);
107 return add_uniform(tag, uni);
109 uniforms[i].replace_value(uni);
113 template<typename T, typename V>
114 void ProgramData::uniform(Tag tag, V value)
116 if(!validate_tag(tag))
119 int i = find_uniform_index(tag);
121 return add_uniform(tag, new T(value));
123 if(T *uni = dynamic_cast<T *>(uniforms[i].value))
126 uniforms[i].replace_value(new T(value));
131 template<typename T, typename V>
132 void ProgramData::uniform_array(Tag tag, unsigned n, V value)
134 if(!validate_tag(tag))
137 int i = find_uniform_index(tag);
139 return add_uniform(tag, new UniformArray<T>(n, value));
141 UniformArray<T> *uni = dynamic_cast<UniformArray<T> *>(uniforms[i].value);
142 if(uni && n==uni->size())
145 uniforms[i].replace_value(new UniformArray<T>(n, value));
150 bool ProgramData::validate_tag(Tag tag) const
157 tied_program->get_uniform_info(tag);
161 catch(const exception &e)
163 IO::print(IO::cerr, "Error while setting uniform %s: %s: %s\n", tag, Debug::demangle(typeid(e).name()), e.what());
169 void ProgramData::add_uniform(Tag tag, Uniform *uni)
171 if(uniforms.size()>=MASK_BITS)
174 throw too_many_uniforms(tag.str());
177 vector<TaggedUniform>::iterator j = lower_bound_member(uniforms, tag, &TaggedUniform::tag);
182 uniforms.insert(j, nu);
187 void ProgramData::uniform(Tag tag, const Uniform &u)
189 uniform(tag, u.clone());
192 void ProgramData::uniform(Tag tag, int v)
194 uniform<Uniform1i>(tag, v);
197 void ProgramData::uniform(Tag tag, float v)
199 uniform<Uniform1f>(tag, v);
202 void ProgramData::uniform(Tag tag, int v0, int v1)
204 int va[2] = { v0, v1 };
208 void ProgramData::uniform(Tag tag, float v0, float v1)
210 float va[2] = { v0, v1 };
214 void ProgramData::uniform2(Tag tag, const int *v)
216 uniform<Uniform2i>(tag, v);
219 void ProgramData::uniform2(Tag tag, const float *v)
221 uniform<Uniform2f>(tag, v);
224 void ProgramData::uniform(Tag tag, int v0, int v1, int v2)
226 int va[3] = { v0, v1, v2 };
230 void ProgramData::uniform(Tag tag, float v0, float v1, float v2)
232 float va[3] = { v0, v1, v2 };
236 void ProgramData::uniform(Tag tag, const Vector3 &v)
238 uniform(tag, v.x, v.y, v.z);
241 void ProgramData::uniform3(Tag tag, const int *v)
243 uniform<Uniform3i>(tag, v);
246 void ProgramData::uniform3(Tag tag, const float *v)
248 uniform<Uniform3f>(tag, v);
251 void ProgramData::uniform(Tag tag, int v0, int v1, int v2, int v3)
253 int va[4] = { v0, v1, v2, v3 };
257 void ProgramData::uniform(Tag tag, float v0, float v1, float v2, float v3)
259 float va[4] = { v0, v1, v2, v3 };
263 void ProgramData::uniform(Tag tag, const Vector4 &v)
265 uniform(tag, v.x, v.y, v.z, v.w);
268 void ProgramData::uniform(Tag tag, const Color &c)
270 uniform(tag, c.r, c.g, c.b, c.a);
273 void ProgramData::uniform4(Tag tag, const int *v)
275 uniform<Uniform4i>(tag, v);
278 void ProgramData::uniform4(Tag tag, const float *v)
280 uniform<Uniform4f>(tag, v);
283 void ProgramData::uniform(Tag tag, const LinAl::Matrix<float, 2, 2> &m)
285 uniform_matrix2(tag, &m(0, 0));
288 void ProgramData::uniform_matrix2(Tag tag, const float *v)
290 uniform<UniformMatrix2x2f>(tag, v);
293 void ProgramData::uniform(Tag tag, const LinAl::Matrix<float, 2, 3> &m)
295 uniform_matrix3x2(tag, &m(0, 0));
298 void ProgramData::uniform_matrix3x2(Tag tag, const float *v)
300 uniform<UniformMatrix3x2f>(tag, v);
303 void ProgramData::uniform(Tag tag, const LinAl::Matrix<float, 2, 4> &m)
305 uniform_matrix4x2(tag, &m(0, 0));
308 void ProgramData::uniform_matrix4x2(Tag tag, const float *v)
310 uniform<UniformMatrix4x2f>(tag, v);
313 void ProgramData::uniform(Tag tag, const LinAl::Matrix<float, 3, 2> &m)
315 uniform_matrix2x3(tag, &m(0, 0));
318 void ProgramData::uniform_matrix2x3(Tag tag, const float *v)
320 uniform<UniformMatrix2x3f>(tag, v);
323 void ProgramData::uniform(Tag tag, const LinAl::Matrix<float, 3, 3> &m)
325 uniform_matrix3(tag, &m(0, 0));
328 void ProgramData::uniform_matrix3(Tag tag, const float *v)
330 uniform<UniformMatrix3x3f>(tag, v);
333 void ProgramData::uniform(Tag tag, const LinAl::Matrix<float, 3, 4> &m)
335 uniform_matrix4x3(tag, &m(0, 0));
338 void ProgramData::uniform_matrix4x3(Tag tag, const float *v)
340 uniform<UniformMatrix4x3f>(tag, v);
343 void ProgramData::uniform(Tag tag, const LinAl::Matrix<float, 4, 2> &m)
345 uniform_matrix2x4(tag, &m(0, 0));
348 void ProgramData::uniform_matrix2x4(Tag tag, const float *v)
350 uniform<UniformMatrix2x4f>(tag, v);
353 void ProgramData::uniform(Tag tag, const LinAl::Matrix<float, 4, 3> &m)
355 uniform_matrix3x4(tag, &m(0, 0));
358 void ProgramData::uniform_matrix3x4(Tag tag, const float *v)
360 uniform<UniformMatrix3x4f>(tag, v);
363 void ProgramData::uniform(Tag tag, const Matrix &m)
365 uniform_matrix4(tag, m.data());
368 void ProgramData::uniform_matrix4(Tag tag, const float *v)
370 uniform<UniformMatrix4x4f>(tag, v);
373 void ProgramData::uniform1_array(Tag tag, unsigned n, const int *v)
375 uniform_array<Uniform1i>(tag, n, v);
378 void ProgramData::uniform1_array(Tag tag, unsigned n, const float *v)
380 uniform_array<Uniform1f>(tag, n, v);
383 void ProgramData::uniform2_array(Tag tag, unsigned n, const int *v)
385 uniform_array<Uniform2i>(tag, n, v);
388 void ProgramData::uniform2_array(Tag tag, unsigned n, const float *v)
390 uniform_array<Uniform2f>(tag, n, v);
393 void ProgramData::uniform3_array(Tag tag, unsigned n, const int *v)
395 uniform_array<Uniform3i>(tag, n, v);
398 void ProgramData::uniform3_array(Tag tag, unsigned n, const float *v)
400 uniform_array<Uniform3f>(tag, n, v);
403 void ProgramData::uniform4_array(Tag tag, unsigned n, const int *v)
405 uniform_array<Uniform4i>(tag, n, v);
408 void ProgramData::uniform4_array(Tag tag, unsigned n, const float *v)
410 uniform_array<Uniform4f>(tag, n, v);
413 void ProgramData::uniform_matrix2_array(Tag tag, unsigned n, const float *v)
415 uniform_array<UniformMatrix2x2f>(tag, n, v);
418 void ProgramData::uniform_matrix3x2_array(Tag tag, unsigned n, const float *v)
420 uniform_array<UniformMatrix3x2f>(tag, n, v);
423 void ProgramData::uniform_matrix4x2_array(Tag tag, unsigned n, const float *v)
425 uniform_array<UniformMatrix4x2f>(tag, n, v);
428 void ProgramData::uniform_matrix2x3_array(Tag tag, unsigned n, const float *v)
430 uniform_array<UniformMatrix2x3f>(tag, n, v);
433 void ProgramData::uniform_matrix3_array(Tag tag, unsigned n, const float *v)
435 uniform_array<UniformMatrix3x3f>(tag, n, v);
438 void ProgramData::uniform_matrix4x3_array(Tag tag, unsigned n, const float *v)
440 uniform_array<UniformMatrix4x3f>(tag, n, v);
443 void ProgramData::uniform_matrix2x4_array(Tag tag, unsigned n, const float *v)
445 uniform_array<UniformMatrix2x4f>(tag, n, v);
448 void ProgramData::uniform_matrix3x4_array(Tag tag, unsigned n, const float *v)
450 uniform_array<UniformMatrix3x4f>(tag, n, v);
453 void ProgramData::uniform_matrix4_array(Tag tag, unsigned n, const float *v)
455 uniform_array<UniformMatrix4x4f>(tag, n, v);
458 void ProgramData::remove_uniform(Tag tag)
460 vector<TaggedUniform>::const_iterator i = lower_bound_member(uniforms, tag, &TaggedUniform::tag);
461 if(i==uniforms.end() || i->tag!=tag)
470 vector<Tag> ProgramData::get_uniform_tags() const
473 tags.reserve(uniforms.size());
474 for(vector<TaggedUniform>::const_iterator i=uniforms.begin(); i!=uniforms.end(); ++i)
475 tags.push_back(i->tag);
479 const Uniform &ProgramData::get_uniform(Tag tag) const
481 int i = find_uniform_index(tag);
483 throw key_error(tag);
484 return *uniforms[i].value;
487 const Uniform *ProgramData::find_uniform(Tag tag) const
489 int i = find_uniform_index(tag);
490 return (i>=0 ? uniforms[i].value : 0);
493 int ProgramData::find_uniform_index(Tag tag) const
495 vector<TaggedUniform>::const_iterator i = lower_bound_member(uniforms, tag, &TaggedUniform::tag);
496 return ((i!=uniforms.end() && i->tag==tag) ? i-uniforms.begin() : -1);
499 void ProgramData::update_block_uniform_indices(SharedBlock &block, const Program::UniformBlockInfo &info) const
501 UInt8 *indices = block.indices.values;
502 if(info.uniforms.size()>16)
504 if(block.indices.type_flag==0xFD)
506 block.indices.dynamic.values = new UInt8[info.uniforms.size()];
507 block.indices.type_flag = 0xFE;
509 indices = block.indices.dynamic.values;
513 for(unsigned i=0; i<info.uniforms.size(); ++i)
515 int j = find_uniform_index(info.uniforms[i]->tag);
519 if(static_cast<unsigned>(j)<MASK_BITS)
526 block.dirty = block.used;
529 void ProgramData::update_block(SharedBlock &block, const Program::UniformBlockInfo &info) const
531 const UInt8 *indices = block.get_uniform_indices();
532 for(unsigned i=0; i<info.uniforms.size(); ++i)
534 block.block->attach(*info.uniforms[i], *uniforms[indices[i]].value);
537 ProgramData::SharedBlock *ProgramData::get_shared_block(const Program::UniformBlockInfo &info) const
539 BlockMap::iterator i = blocks.find(info.layout_hash);
542 bool any_found = false;
543 bool all_found = true;
544 for(vector<const Program::UniformInfo *>::const_iterator j=info.uniforms.begin(); j!=info.uniforms.end(); ++j)
546 if(find_uniform_index((*j)->name)>=0)
554 else if(!all_found && info.bind_point>=0)
557 IO::print(IO::cerr, "Warning: not all uniforms for block %s are present\n", info.name);
559 throw incomplete_uniform_block(info.name);
564 if(info.bind_point>=0)
567 buffer = new Buffer(UNIFORM_BUFFER);
569 block = new UniformBlock(info.data_size);
570 block->use_buffer(buffer, last_block);
574 block = new UniformBlock;
576 i = blocks.insert(BlockMap::value_type(info.layout_hash, SharedBlock(block))).first;
577 update_block_uniform_indices(i->second, info);
583 void ProgramData::apply() const
585 const Program *prog = Program::current();
587 throw invalid_operation("ProgramData::apply");
589 Program::LayoutHash layout = prog->get_uniform_layout_hash();
590 ProgramUniforms &pu = programs[layout];
592 Mask force_dirty = (dirty==ALL_ONES ? ALL_ONES : 0U);
593 Mask affected = (dirty&pu.used) | force_dirty;
594 if(affected|pu.dirty)
596 /* If the global dirty flag affects this program, add it to per-program
597 dirty flags and clear the global flag. A previously unseen program will
598 cause this to happen if there's any dirty uniforms. */
601 for(BlockMap::iterator i=blocks.begin(); i!=blocks.end(); ++i)
602 i->second.dirty |= (dirty&i->second.used) | force_dirty;
603 for(ProgramMap::iterator i=programs.begin(); i!=programs.end(); ++i)
604 i->second.dirty |= (dirty&i->second.used) | force_dirty;
608 const vector<Program::UniformBlockInfo> &prog_blocks = prog->get_uniform_blocks();
610 UniformBlock *old_last_block = last_block;
611 if(pu.dirty==ALL_ONES)
613 /* The set of uniforms has changed since this program was last used.
614 Regenerate the list of uniform blocks. */
616 pu.blocks.reserve(prog_blocks.size());
619 for(vector<Program::UniformBlockInfo>::const_iterator i=prog_blocks.begin(); i!=prog_blocks.end(); ++i)
621 SharedBlock *shared = get_shared_block(*i);
624 if(shared->dirty==ALL_ONES)
625 update_block_uniform_indices(*shared, *i);
626 pu.used |= shared->used;
629 pu.blocks.push_back(ProgramBlock(i->bind_point, shared));
633 // Update the contents of all dirty blocks.
634 bool buffered_blocks_updated = false;
635 vector<ProgramBlock>::iterator j = pu.blocks.begin();
636 for(vector<Program::UniformBlockInfo>::const_iterator i=prog_blocks.begin(); i!=prog_blocks.end(); ++i, ++j)
638 if(!j->shared || !j->shared->dirty)
641 update_block(*j->shared, *i);
642 j->shared->dirty = 0;
643 buffered_blocks_updated |= (j->bind_point>=0);
648 /* If any blocks stored in the buffer were updated, bind the buffer here
649 to avoid state thrashing. */
650 if(buffered_blocks_updated && !ARB_direct_state_access)
653 if(last_block!=old_last_block)
655 unsigned required_size = last_block->get_required_buffer_size();
656 if(last_block->get_required_buffer_size()>buffer->get_size())
658 if(buffer->get_size()>0)
661 buffer = new Buffer(UNIFORM_BUFFER);
662 last_block->change_buffer(buffer);
665 buffer->storage(required_size);
670 for(vector<ProgramBlock>::iterator i=pu.blocks.begin(); i!=pu.blocks.end(); ++i)
672 i->block->apply(i->bind_point);
676 ProgramData::TaggedUniform::TaggedUniform():
680 void ProgramData::TaggedUniform::replace_value(Uniform *v)
682 /* UniformBlock does not copy the uniforms, so existing default blocks
683 will be left with stale pointers. This is not a problem as long as no
684 one stores pointers to the blocks and expects them to stay valid. */
690 ProgramData::SharedBlock::SharedBlock(UniformBlock *b):
695 indices.type_flag = 0xFD;
698 const UInt8 *ProgramData::SharedBlock::get_uniform_indices() const
700 return (indices.type_flag==0xFE ? indices.dynamic.values : indices.values);
704 ProgramData::ProgramBlock::ProgramBlock():
710 ProgramData::ProgramBlock::ProgramBlock(int p, SharedBlock *b):
712 block((b && b->used) ? b->block : 0),
717 ProgramData::ProgramUniforms::ProgramUniforms():
723 ProgramData::Loader::Loader(ProgramData &pd):
724 DataFile::ObjectLoader<ProgramData>(pd)
726 add("uniform", &Loader::uniform1i);
727 add("uniform1i", &Loader::uniform1i);
728 add("uniform", &Loader::uniform1f);
729 add("uniform1f", &Loader::uniform1f);
730 add("uniform", &Loader::uniform2i);
731 add("uniform2i", &Loader::uniform2i);
732 add("uniform", &Loader::uniform2f);
733 add("uniform2f", &Loader::uniform2f);
734 add("uniform", &Loader::uniform3i);
735 add("uniform3i", &Loader::uniform3i);
736 add("uniform", &Loader::uniform3f);
737 add("uniform3f", &Loader::uniform3f);
738 add("uniform", &Loader::uniform4i);
739 add("uniform4i", &Loader::uniform4i);
740 add("uniform", &Loader::uniform4f);
741 add("uniform4f", &Loader::uniform4f);
742 add("uniform1i_array", &Loader::uniform1i_array);
743 add("uniform1f_array", &Loader::uniform1f_array);
744 add("uniform2f_array", &Loader::uniform2f_array);
745 add("uniform3f_array", &Loader::uniform3f_array);
746 add("uniform4f_array", &Loader::uniform4f_array);
747 add("uniform_array", &Loader::uniform_array);
750 void ProgramData::Loader::uniform1i(const string &n, int v)
755 void ProgramData::Loader::uniform1f(const string &n, float v)
760 void ProgramData::Loader::uniform2i(const string &n, int v0, int v1)
762 obj.uniform(n, v0, v1);
765 void ProgramData::Loader::uniform2f(const string &n, float v0, float v1)
767 obj.uniform(n, v0, v1);
770 void ProgramData::Loader::uniform3i(const string &n, int v0, int v1, int v2)
772 obj.uniform(n, v0, v1, v2);
775 void ProgramData::Loader::uniform3f(const string &n, float v0, float v1, float v2)
777 obj.uniform(n, v0, v1, v2);
780 void ProgramData::Loader::uniform4i(const string &n, int v0, int v1, int v2, int v3)
782 obj.uniform(n, v0, v1, v2, v3);
785 void ProgramData::Loader::uniform4f(const string &n, float v0, float v1, float v2, float v3)
787 obj.uniform(n, v0, v1, v2, v3);
790 void ProgramData::Loader::uniform_array_(const string &n, DataType t, unsigned e)
792 ArrayLoader ldr(t, e);
794 unsigned size = ldr.get_size();
796 throw logic_error("empty uniform array");
798 DataType type = ldr.get_data_type();
799 unsigned elem_size = ldr.get_element_size();
802 const int *data = reinterpret_cast<const int *>(ldr.get_data());
804 obj.uniform1_array(n, size, data);
805 else if(elem_size==2)
806 obj.uniform2_array(n, size, data);
807 else if(elem_size==3)
808 obj.uniform3_array(n, size, data);
809 else if(elem_size==4)
810 obj.uniform4_array(n, size, data);
812 throw logic_error("unsupported combination of array type and element size");
816 const float *data = reinterpret_cast<const float *>(ldr.get_data());
818 obj.uniform1_array(n, size, data);
819 else if(elem_size==2)
820 obj.uniform2_array(n, size, data);
821 else if(elem_size==3)
822 obj.uniform3_array(n, size, data);
823 else if(elem_size==4)
824 obj.uniform4_array(n, size, data);
826 throw logic_error("unsupported combination of array type and element size");
829 throw logic_error("unsupported array type");
832 void ProgramData::Loader::uniform1i_array(const string &n)
834 uniform_array_(n, INT, 1);
837 void ProgramData::Loader::uniform1f_array(const string &n)
839 uniform_array_(n, FLOAT, 1);
842 void ProgramData::Loader::uniform2i_array(const string &n)
844 uniform_array_(n, INT, 2);
847 void ProgramData::Loader::uniform2f_array(const string &n)
849 uniform_array_(n, FLOAT, 2);
852 void ProgramData::Loader::uniform3i_array(const string &n)
854 uniform_array_(n, INT, 3);
857 void ProgramData::Loader::uniform3f_array(const string &n)
859 uniform_array_(n, FLOAT, 3);
862 void ProgramData::Loader::uniform4i_array(const string &n)
864 uniform_array_(n, INT, 4);
867 void ProgramData::Loader::uniform4f_array(const string &n)
869 uniform_array_(n, FLOAT, 4);
872 void ProgramData::Loader::uniform_array(const string &n)
874 uniform_array_(n, static_cast<DataType>(0), 0);
878 ProgramData::ArrayLoader::ArrayLoader(DataType t, unsigned e):
882 add("uniform", &ArrayLoader::uniform1i);
883 add("uniform1i", &ArrayLoader::uniform1i);
884 add("uniform", &ArrayLoader::uniform1f);
885 add("uniform1f", &ArrayLoader::uniform1f);
886 add("uniform", &ArrayLoader::uniform2f);
887 add("uniform2f", &ArrayLoader::uniform2f);
888 add("uniform", &ArrayLoader::uniform3f);
889 add("uniform3f", &ArrayLoader::uniform3f);
890 add("uniform", &ArrayLoader::uniform4f);
891 add("uniform4f", &ArrayLoader::uniform4f);
894 void ProgramData::ArrayLoader::uniform(DataType t, unsigned e, const void *v)
896 if(element_size && (t!=type || e!=element_size))
897 throw logic_error("heterogeneous array contents");
905 const char *cv = reinterpret_cast<const char *>(v);
906 data.insert(data.end(), cv, cv+element_size*4);
909 void ProgramData::ArrayLoader::uniform1i(int v)
914 void ProgramData::ArrayLoader::uniform1f(float v)
916 uniform(FLOAT, 1, &v);
919 void ProgramData::ArrayLoader::uniform2i(int v0, int v1)
921 int va[2] = { v0, v1 };
925 void ProgramData::ArrayLoader::uniform2f(float v0, float v1)
927 float va[2] = { v0, v1 };
928 uniform(FLOAT, 2, va);
931 void ProgramData::ArrayLoader::uniform3i(int v0, int v1, int v2)
933 int va[3] = { v0, v1, v2 };
937 void ProgramData::ArrayLoader::uniform3f(float v0, float v1, float v2)
939 float va[3] = { v0, v1, v2 };
940 uniform(FLOAT, 3, va);
943 void ProgramData::ArrayLoader::uniform4i(int v0, int v1, int v2, int v3)
945 int va[4] = { v0, v1, v2, v3 };
949 void ProgramData::ArrayLoader::uniform4f(float v0, float v1, float v2, float v3)
951 float va[4] = { v0, v1, v2, v3 };
952 uniform(FLOAT, 4, va);