X-Git-Url: http://git.tdb.fi/?p=libs%2Fgl.git;a=blobdiff_plain;f=source%2Fcore%2Fprogram.cpp;h=999a94ad75a6d6c3edc1899c9fb0944c69a373eb;hp=4b53b8fe2f28e7e812e605654d6d7d64ec695aeb;hb=2576b61;hpb=ddb8e4ceea602d90ca7d9dc2637b9285610787bd diff --git a/source/core/program.cpp b/source/core/program.cpp index 4b53b8fe..999a94ad 100644 --- a/source/core/program.cpp +++ b/source/core/program.cpp @@ -1,6 +1,6 @@ -#include #include #include +#include #include #include #include @@ -308,16 +308,17 @@ void Program::link() collect_attributes(); } - for(UniformMap::const_iterator i=uniforms.begin(); i!=uniforms.end(); ++i) - require_type(i->second.type); - for(AttributeMap::const_iterator i=attributes.begin(); i!=attributes.end(); ++i) - require_type(i->second.type); + for(vector::const_iterator i=uniforms.begin(); i!=uniforms.end(); ++i) + require_type(i->type); + for(vector::const_iterator i=attributes.begin(); i!=attributes.end(); ++i) + require_type(i->type); } void Program::query_uniforms() { unsigned count = get_program_i(id, GL_ACTIVE_UNIFORMS); - vector uniforms_by_index(count); + uniforms.reserve(count); + vector uniform_names(count); for(unsigned i=0; i3 && !strcmp(name+len-3, "[0]")) name[len-3] = 0; - UniformInfo &info = uniforms[name]; + uniforms.push_back(UniformInfo()); + UniformInfo &info = uniforms.back(); info.name = name; info.array_size = size; info.type = from_gl_type(type); - uniforms_by_index[i] = &info; + uniform_names[i] = name; } } + sort(uniforms, &uniform_name_compare); + if(ARB_uniform_buffer_object) + { + vector uniforms_by_index(count); + for(unsigned i=0; i); query_uniform_blocks(uniforms_by_index); + } - UniformBlockInfo &default_block = uniform_blocks[string()]; + uniform_blocks.push_back(UniformBlockInfo()); + UniformBlockInfo &default_block = uniform_blocks.back(); - for(UniformMap::iterator i=uniforms.begin(); i!=uniforms.end(); ++i) - if(!i->second.block) + for(vector::iterator i=uniforms.begin(); i!=uniforms.end(); ++i) + if(!i->block) { - i->second.location = glGetUniformLocation(id, i->second.name.c_str()); - i->second.block = &default_block; - default_block.uniforms.push_back(&i->second); + i->location = glGetUniformLocation(id, i->name.c_str()); + i->block = &default_block; + default_block.uniforms.push_back(&*i); } default_block.layout_hash = compute_layout_hash(default_block.uniforms); @@ -362,12 +374,15 @@ void Program::query_uniform_blocks(const vector &uniforms_by_inde { std::set used_bind_points; unsigned count = get_program_i(id, GL_ACTIVE_UNIFORM_BLOCKS); + // Reserve an extra index for the default block + uniform_blocks.reserve(count+1); for(unsigned i=0; i &uniforms_by_inde void Program::query_attributes() { unsigned count = get_program_i(id, GL_ACTIVE_ATTRIBUTES); + attributes.reserve(count); for(unsigned i=0; i3 && !strcmp(name+len-3, "[0]")) name[len-3] = 0; - AttributeInfo &info = attributes[name]; + attributes.push_back(AttributeInfo()); + AttributeInfo &info = attributes.back(); info.name = name; info.location = glGetAttribLocation(id, name); info.array_size = size; @@ -455,14 +472,17 @@ void Program::collect_uniforms() { const SpirVModule &mod = static_cast(*module); - UniformBlockInfo &default_block = uniform_blocks[string()]; + // Prepare the default block + uniform_blocks.push_back(UniformBlockInfo()); + vector > block_uniform_names(1); const vector &variables = mod.get_variables(); for(vector::const_iterator i=variables.begin(); i!=variables.end(); ++i) { if(i->storage==SpirVModule::UNIFORM && i->struct_type) { - UniformBlockInfo &info = uniform_blocks[i->struct_type->name]; + uniform_blocks.push_back(UniformBlockInfo()); + UniformBlockInfo &info = uniform_blocks.back(); info.name = i->struct_type->name; info.bind_point = i->binding; info.data_size = i->struct_type->size; @@ -470,28 +490,41 @@ void Program::collect_uniforms() string prefix; if(!i->name.empty()) prefix = i->struct_type->name+"."; - collect_block_uniforms(info, *i->struct_type, prefix, 0); - - info.layout_hash = compute_layout_hash(info.uniforms); + block_uniform_names.push_back(vector()); + collect_block_uniforms(*i->struct_type, prefix, 0, block_uniform_names.back()); } else if(i->storage==SpirVModule::UNIFORM_CONSTANT && i->location>=0) { - UniformInfo &info = uniforms[i->name]; + block_uniform_names[0].push_back(i->name); + uniforms.push_back(UniformInfo()); + UniformInfo &info = uniforms.back(); info.name = i->name; - info.block = &default_block; info.location = i->location; info.array_size = i->array_size; info.type = i->type; - default_block.uniforms.push_back(&info); } } - default_block.layout_hash = compute_layout_hash(default_block.uniforms); + sort(uniforms, &uniform_name_compare); + + for(unsigned i=0; i &names = block_uniform_names[i]; + for(vector::const_iterator j=names.begin(); j!=names.end(); ++j) + { + // The element is already known to be present + UniformInfo &uni = *lower_bound(uniforms, *j, &name_search); + block.uniforms.push_back(&uni); + uni.block = █ + } + block.layout_hash = compute_layout_hash(block.uniforms); + } update_layout_hash(); } -void Program::collect_block_uniforms(UniformBlockInfo &block, const SpirVModule::Structure &strct, const string &prefix, unsigned base_offset) +void Program::collect_block_uniforms(const SpirVModule::Structure &strct, const string &prefix, unsigned base_offset, vector &uniform_names) { for(vector::const_iterator i=strct.members.begin(); i!=strct.members.end(); ++i) { @@ -501,23 +534,23 @@ void Program::collect_block_uniforms(UniformBlockInfo &block, const SpirVModule: if(i->array_size) { for(unsigned j=0; jarray_size; ++j, offset+=i->array_stride) - collect_block_uniforms(block, *i->struct_type, format("%s%s[%d].", prefix, i->name, j), offset); + collect_block_uniforms(*i->struct_type, format("%s%s[%d].", prefix, i->name, j), offset, uniform_names); } else - collect_block_uniforms(block, *i->struct_type, prefix+i->name+".", offset); + collect_block_uniforms(*i->struct_type, prefix+i->name+".", offset, uniform_names); } else { string name = prefix+i->name; - UniformInfo &info = uniforms[name]; + uniform_names.push_back(name); + uniforms.push_back(UniformInfo()); + UniformInfo &info = uniforms.back(); info.name = name; - info.block = █ info.offset = offset; info.array_size = i->array_size; info.array_stride = i->array_stride; info.matrix_stride = i->matrix_stride; info.type = i->type; - block.uniforms.push_back(&info); } } } @@ -533,7 +566,8 @@ void Program::collect_attributes() for(vector::const_iterator j=i->globals.begin(); j!=i->globals.end(); ++j) if((*j)->storage==SpirVModule::INPUT) { - AttributeInfo &info = attributes[(*j)->name]; + attributes.push_back(AttributeInfo()); + AttributeInfo &info = attributes.back(); info.name = (*j)->name; info.location = (*j)->location; info.array_size = (*j)->array_size; @@ -545,8 +579,8 @@ void Program::collect_attributes() void Program::update_layout_hash() { string layout_descriptor; - for(UniformBlockMap::const_iterator i=uniform_blocks.begin(); i!=uniform_blocks.end(); ++i) - layout_descriptor += format("%d:%x\n", i->second.bind_point, i->second.layout_hash); + for(vector::const_iterator i=uniform_blocks.begin(); i!=uniform_blocks.end(); ++i) + layout_descriptor += format("%d:%x\n", i->bind_point, i->layout_hash); uniform_layout_hash = hash32(layout_descriptor); } @@ -563,6 +597,17 @@ bool Program::uniform_location_compare(const UniformInfo *uni1, const UniformInf return uni1->locationlocation; } +bool Program::uniform_name_compare(const UniformInfo &uni1, const UniformInfo &uni2) +{ + return uni1.name +bool Program::name_search(const T &item, const string &name) +{ + return item.name::const_iterator i=uniform_blocks.begin(); i!=uniform_blocks.end(); ++i) + if(i->name==name) + return *i; + throw key_error(name); } const Program::UniformInfo &Program::get_uniform_info(const string &name) const { - return get_item(uniforms, name); + vector::const_iterator i = lower_bound(uniforms, name, &name_search); + if(i==uniforms.end() || i->name!=name) + throw key_error(name); + return *i; } int Program::get_uniform_location(const string &name) const @@ -587,16 +638,16 @@ int Program::get_uniform_location(const string &name) const if(name[name.size()-1]==']') throw invalid_argument("Program::get_uniform_location"); - UniformMap::const_iterator i = uniforms.find(name); - if(i==uniforms.end()) - return -1; - - return i->second.block->bind_point<0 ? i->second.location : -1; + vector::const_iterator i = lower_bound(uniforms, name, &name_search); + return i!=uniforms.end() && i->name==name && i->block->bind_point<0 ? i->location : -1; } const Program::AttributeInfo &Program::get_attribute_info(const string &name) const { - return get_item(attributes, name); + vector::const_iterator i = lower_bound(attributes, name, &name_search); + if(i==attributes.end() || i->name!=name) + throw key_error(name); + return *i; } int Program::get_attribute_location(const string &name) const @@ -604,8 +655,8 @@ int Program::get_attribute_location(const string &name) const if(name[name.size()-1]==']') throw invalid_argument("Program::get_attribute_location"); - AttributeMap::const_iterator i = attributes.find(name); - return i!=attributes.end() ? i->second.location : -1; + vector::const_iterator i = lower_bound(attributes, name, &name_search); + return i!=attributes.end() && i->name==name ? i->location : -1; } void Program::bind() const