]> git.tdb.fi Git - libs/gl.git/blobdiff - source/core/program.cpp
Split reflection data from Program to a separate struct
[libs/gl.git] / source / core / program.cpp
index 8bec2f6f488295474804d641516b825de8ef309a..82f438cdbf02112ff3362018b7587b066f251fbd 100644 (file)
@@ -1,7 +1,6 @@
 #include <cstring>
 #include <set>
 #include <msp/core/algorithm.h>
-#include <msp/core/hash.h>
 #include <msp/core/maputils.h>
 #include <msp/core/raii.h>
 #include <msp/gl/extensions/arb_es2_compatibility.h>
@@ -16,7 +15,6 @@
 #include <msp/gl/extensions/khr_debug.h>
 #include <msp/gl/extensions/nv_non_square_matrices.h>
 #include <msp/io/print.h>
-#include <msp/strings/format.h>
 #include "buffer.h"
 #include "error.h"
 #include "program.h"
@@ -236,9 +234,7 @@ void Program::link()
        if(!has_stages())
                throw invalid_operation("Program::link");
 
-       uniforms.clear();
-       uniform_blocks.clear();
-       attributes.clear();
+       reflect_data = ReflectData();
 
        glLinkProgram(id);
        int status = 0;
@@ -266,13 +262,13 @@ void Program::link()
                query_attributes();
                if(transient)
                {
-                       for(unsigned i=0; i<uniform_blocks.size(); ++i)
+                       for(unsigned i=0; i<reflect_data.uniform_blocks.size(); ++i)
                        {
-                               auto j = transient->blocks.find(uniform_blocks[i].name);
+                               auto j = transient->blocks.find(reflect_data.uniform_blocks[i].name);
                                if(j!=transient->blocks.end())
                                {
                                        glUniformBlockBinding(id, i, j->second);
-                                       uniform_blocks[i].bind_point = j->second;
+                                       reflect_data.uniform_blocks[i].bind_point = j->second;
                                }
                        }
 
@@ -300,9 +296,9 @@ void Program::link()
        delete transient;
        transient = 0;
 
-       for(const UniformInfo &u: uniforms)
+       for(const ReflectData::UniformInfo &u: reflect_data.uniforms)
                require_type(u.type);
-       for(const AttributeInfo &a: attributes)
+       for(const ReflectData::AttributeInfo &a: reflect_data.attributes)
                require_type(a.type);
 }
 
@@ -310,7 +306,7 @@ void Program::query_uniforms()
 {
        unsigned count = 0;
        glGetProgramiv(id, GL_ACTIVE_UNIFORMS, reinterpret_cast<int *>(&count));
-       uniforms.reserve(count);
+       reflect_data.uniforms.reserve(count);
        vector<string> uniform_names(count);
        for(unsigned i=0; i<count; ++i)
        {
@@ -326,8 +322,8 @@ void Program::query_uniforms()
                        if(len>3 && !strcmp(name+len-3, "[0]"))
                                name[len-3] = 0;
 
-                       uniforms.push_back(UniformInfo());
-                       UniformInfo &info = uniforms.back();
+                       reflect_data.uniforms.push_back(ReflectData::UniformInfo());
+                       ReflectData::UniformInfo &info = reflect_data.uniforms.back();
                        info.name = name;
                        info.tag = name;
                        info.array_size = size;
@@ -336,22 +332,22 @@ void Program::query_uniforms()
                }
        }
 
-       sort_member(uniforms, &UniformInfo::tag);
+       sort_member(reflect_data.uniforms, &ReflectData::UniformInfo::tag);
 
        if(ARB_uniform_buffer_object)
        {
-               vector<UniformInfo *> uniforms_by_index(count);
+               vector<ReflectData::UniformInfo *> uniforms_by_index(count);
                for(unsigned i=0; i<count; ++i)
                        if(!uniform_names[i].empty())
                                // The element is already known to be present
-                               uniforms_by_index[i] = &*lower_bound_member(uniforms, Tag(uniform_names[i]), &UniformInfo::tag);
+                               uniforms_by_index[i] = &*lower_bound_member(reflect_data.uniforms, Tag(uniform_names[i]), &ReflectData::UniformInfo::tag);
                query_uniform_blocks(uniforms_by_index);
        }
 
-       uniform_blocks.push_back(UniformBlockInfo());
-       UniformBlockInfo &default_block = uniform_blocks.back();
+       reflect_data.uniform_blocks.push_back(ReflectData::UniformBlockInfo());
+       ReflectData::UniformBlockInfo &default_block = reflect_data.uniform_blocks.back();
 
-       for(UniformInfo &u: uniforms)
+       for(ReflectData::UniformInfo &u: reflect_data.uniforms)
                if(!u.block)
                {
                        u.location = glGetUniformLocation(id, u.name.c_str());
@@ -362,24 +358,23 @@ void Program::query_uniforms()
                                glGetUniformiv(id, u.location, &u.binding);
                }
 
-       default_block.layout_hash = compute_layout_hash(default_block.uniforms);
-
-       update_layout_hash();
+       default_block.update_layout_hash();
+       reflect_data.update_layout_hash();
 }
 
-void Program::query_uniform_blocks(const vector<UniformInfo *> &uniforms_by_index)
+void Program::query_uniform_blocks(const vector<ReflectData::UniformInfo *> &uniforms_by_index)
 {
        unsigned count = 0;
        glGetProgramiv(id, GL_ACTIVE_UNIFORM_BLOCKS, reinterpret_cast<int *>(&count));
        // Reserve an extra index for the default block
-       uniform_blocks.reserve(count+1);
+       reflect_data.uniform_blocks.reserve(count+1);
        for(unsigned i=0; i<count; ++i)
        {
                char name[128];
                int len;
                glGetActiveUniformBlockName(id, i, sizeof(name), &len, name);
-               uniform_blocks.push_back(UniformBlockInfo());
-               UniformBlockInfo &info = uniform_blocks.back();
+               reflect_data.uniform_blocks.push_back(ReflectData::UniformBlockInfo());
+               ReflectData::UniformBlockInfo &info = reflect_data.uniform_blocks.back();
                info.name = name;
 
                int value;
@@ -431,8 +426,8 @@ void Program::query_uniform_blocks(const vector<UniformInfo *> &uniforms_by_inde
                                uniforms_by_index[query_indices[j]]->matrix_stride = values[j];
                }
 
-               sort(info.uniforms, uniform_location_compare);
-               info.layout_hash = compute_layout_hash(info.uniforms);
+               info.sort_uniforms();
+               info.update_layout_hash();
        }
 }
 
@@ -440,7 +435,7 @@ void Program::query_attributes()
 {
        unsigned count = 0;
        glGetProgramiv(id, GL_ACTIVE_ATTRIBUTES, reinterpret_cast<int *>(&count));
-       attributes.reserve(count);
+       reflect_data.attributes.reserve(count);
        for(unsigned i=0; i<count; ++i)
        {
                char name[128];
@@ -453,8 +448,8 @@ void Program::query_attributes()
                        if(len>3 && !strcmp(name+len-3, "[0]"))
                                name[len-3] = 0;
 
-                       attributes.push_back(AttributeInfo());
-                       AttributeInfo &info = attributes.back();
+                       reflect_data.attributes.push_back(ReflectData::AttributeInfo());
+                       ReflectData::AttributeInfo &info = reflect_data.attributes.back();
                        info.name = name;
                        info.location = glGetAttribLocation(id, name);
                        info.array_size = size;
@@ -468,15 +463,15 @@ void Program::collect_uniforms()
        const SpirVModule &mod = static_cast<const SpirVModule &>(*module);
 
        // Prepare the default block
-       uniform_blocks.push_back(UniformBlockInfo());
+       reflect_data.uniform_blocks.push_back(ReflectData::UniformBlockInfo());
        vector<vector<string> > block_uniform_names(1);
 
        for(const SpirVModule::Variable &v: mod.get_variables())
        {
                if(v.storage==SpirVModule::UNIFORM && v.struct_type)
                {
-                       uniform_blocks.push_back(UniformBlockInfo());
-                       UniformBlockInfo &info = uniform_blocks.back();
+                       reflect_data.uniform_blocks.push_back(ReflectData::UniformBlockInfo());
+                       ReflectData::UniformBlockInfo &info = reflect_data.uniform_blocks.back();
                        info.name = v.struct_type->name;
                        info.bind_point = v.binding;
                        info.data_size = v.struct_type->size;
@@ -490,8 +485,8 @@ void Program::collect_uniforms()
                else if(v.storage==SpirVModule::UNIFORM_CONSTANT && v.location>=0)
                {
                        block_uniform_names[0].push_back(v.name);
-                       uniforms.push_back(UniformInfo());
-                       UniformInfo &info = uniforms.back();
+                       reflect_data.uniforms.push_back(ReflectData::UniformInfo());
+                       ReflectData::UniformInfo &info = reflect_data.uniforms.back();
                        info.name = v.name;
                        info.tag = v.name;
                        info.location = v.location;
@@ -501,23 +496,23 @@ void Program::collect_uniforms()
                }
        }
 
-       sort_member(uniforms, &UniformInfo::tag);
+       sort_member(reflect_data.uniforms, &ReflectData::UniformInfo::tag);
 
-       for(unsigned i=0; i<uniform_blocks.size(); ++i)
+       for(unsigned i=0; i<reflect_data.uniform_blocks.size(); ++i)
        {
-               UniformBlockInfo &block = uniform_blocks[i];
+               ReflectData::UniformBlockInfo &block = reflect_data.uniform_blocks[i];
                for(const string &n: block_uniform_names[i])
                {
                        // The element is already known to be present
-                       UniformInfo &uni = *lower_bound_member(uniforms, Tag(n), &UniformInfo::tag);
+                       ReflectData::UniformInfo &uni = *lower_bound_member(reflect_data.uniforms, Tag(n), &ReflectData::UniformInfo::tag);
                        block.uniforms.push_back(&uni);
                        uni.block = &block;
                }
-               sort(block.uniforms, uniform_location_compare);
-               block.layout_hash = compute_layout_hash(block.uniforms);
+               block.sort_uniforms();
+               block.update_layout_hash();
        }
 
-       update_layout_hash();
+       reflect_data.update_layout_hash();
 }
 
 void Program::collect_block_uniforms(const SpirVModule::Structure &strct, const string &prefix, unsigned base_offset, vector<string> &uniform_names)
@@ -551,8 +546,8 @@ void Program::collect_block_uniforms(const SpirVModule::Structure &strct, const
                {
                        string name = prefix+m.name;
                        uniform_names.push_back(name);
-                       uniforms.push_back(UniformInfo());
-                       UniformInfo &info = uniforms.back();
+                       reflect_data.uniforms.push_back(ReflectData::UniformInfo());
+                       ReflectData::UniformInfo &info = reflect_data.uniforms.back();
                        info.name = name;
                        info.tag = name;
                        info.offset = offset;
@@ -574,8 +569,8 @@ void Program::collect_attributes()
                        for(const SpirVModule::Variable *v: e.globals)
                                if(v->storage==SpirVModule::INPUT)
                                {
-                                       attributes.push_back(AttributeInfo());
-                                       AttributeInfo &info = attributes.back();
+                                       reflect_data.attributes.push_back(ReflectData::AttributeInfo());
+                                       ReflectData::AttributeInfo &info = reflect_data.attributes.back();
                                        info.name = v->name;
                                        info.location = v->location;
                                        info.array_size = v->array_size;
@@ -584,47 +579,26 @@ void Program::collect_attributes()
                }
 }
 
-void Program::update_layout_hash()
-{
-       string layout_descriptor;
-       for(const UniformBlockInfo &b: uniform_blocks)
-               layout_descriptor += format("%d:%x\n", b.bind_point, b.layout_hash);
-       uniform_layout_hash = hash32(layout_descriptor);
-}
-
-Program::LayoutHash Program::compute_layout_hash(const vector<const UniformInfo *> &uniforms)
-{
-       string layout_descriptor;
-       for(const UniformInfo *u: uniforms)
-               layout_descriptor += format("%d:%s:%x:%d\n", u->location, u->name, u->type, u->array_size);
-       return hash32(layout_descriptor);
-}
-
-bool Program::uniform_location_compare(const UniformInfo *uni1, const UniformInfo *uni2)
-{
-       return uni1->location<uni2->location;
-}
-
-const Program::UniformBlockInfo &Program::get_uniform_block_info(const string &name) const
+const ReflectData::UniformBlockInfo &Program::get_uniform_block_info(const string &name) const
 {
-       auto i = find_member(uniform_blocks, name, &UniformBlockInfo::name);
-       if(i==uniform_blocks.end())
+       auto i = find_member(reflect_data.uniform_blocks, name, &ReflectData::UniformBlockInfo::name);
+       if(i==reflect_data.uniform_blocks.end())
                throw key_error(name);
        return *i;
 }
 
-const Program::UniformInfo &Program::get_uniform_info(const string &name) const
+const ReflectData::UniformInfo &Program::get_uniform_info(const string &name) const
 {
-       auto i = lower_bound_member(uniforms, Tag(name), &UniformInfo::tag);
-       if(i==uniforms.end() || i->name!=name)
+       auto i = lower_bound_member(reflect_data.uniforms, Tag(name), &ReflectData::UniformInfo::tag);
+       if(i==reflect_data.uniforms.end() || i->name!=name)
                throw key_error(name);
        return *i;
 }
 
-const Program::UniformInfo &Program::get_uniform_info(Tag tag) const
+const ReflectData::UniformInfo &Program::get_uniform_info(Tag tag) const
 {
-       auto i = lower_bound_member(uniforms, tag, &UniformInfo::tag);
-       if(i==uniforms.end() || i->tag!=tag)
+       auto i = lower_bound_member(reflect_data.uniforms, tag, &ReflectData::UniformInfo::tag);
+       if(i==reflect_data.uniforms.end() || i->tag!=tag)
                throw key_error(tag);
        return *i;
 }
@@ -634,26 +608,26 @@ int Program::get_uniform_location(const string &name) const
        if(name[name.size()-1]==']')
                throw invalid_argument("Program::get_uniform_location");
 
-       auto i = lower_bound_member(uniforms, Tag(name), &UniformInfo::tag);
-       return i!=uniforms.end() && i->name==name && i->block->bind_point<0 ? i->location : -1;
+       auto i = lower_bound_member(reflect_data.uniforms, Tag(name), &ReflectData::UniformInfo::tag);
+       return i!=reflect_data.uniforms.end() && i->name==name && i->block->bind_point<0 ? i->location : -1;
 }
 
 int Program::get_uniform_location(Tag tag) const
 {
-       auto i = lower_bound_member(uniforms, tag, &UniformInfo::tag);
-       return i!=uniforms.end() && i->tag==tag && i->block->bind_point<0 ? i->location : -1;
+       auto i = lower_bound_member(reflect_data.uniforms, tag, &ReflectData::UniformInfo::tag);
+       return i!=reflect_data.uniforms.end() && i->tag==tag && i->block->bind_point<0 ? i->location : -1;
 }
 
 int Program::get_uniform_binding(Tag tag) const
 {
-       auto i = lower_bound_member(uniforms, tag, &UniformInfo::tag);
-       return i!=uniforms.end() && i->tag==tag ? i->binding : -1;
+       auto i = lower_bound_member(reflect_data.uniforms, tag, &ReflectData::UniformInfo::tag);
+       return i!=reflect_data.uniforms.end() && i->tag==tag ? i->binding : -1;
 }
 
-const Program::AttributeInfo &Program::get_attribute_info(const string &name) const
+const ReflectData::AttributeInfo &Program::get_attribute_info(const string &name) const
 {
-       auto i = lower_bound_member(attributes, name, &AttributeInfo::name);
-       if(i==attributes.end() || i->name!=name)
+       auto i = lower_bound_member(reflect_data.attributes, name, &ReflectData::AttributeInfo::name);
+       if(i==reflect_data.attributes.end() || i->name!=name)
                throw key_error(name);
        return *i;
 }
@@ -663,8 +637,8 @@ int Program::get_attribute_location(const string &name) const
        if(name[name.size()-1]==']')
                throw invalid_argument("Program::get_attribute_location");
 
-       auto i = lower_bound_member(attributes, name, &AttributeInfo::name);
-       return i!=attributes.end() && i->name==name ? i->location : -1;
+       auto i = lower_bound_member(reflect_data.attributes, name, &ReflectData::AttributeInfo::name);
+       return i!=reflect_data.attributes.end() && i->name==name ? i->location : -1;
 }
 
 void Program::set_debug_name(const string &name)
@@ -695,31 +669,6 @@ void Program::set_stage_debug_name(unsigned stage_id, Stage type)
 }
 
 
-Program::UniformInfo::UniformInfo():
-       block(0),
-       location(-1),
-       array_size(0),
-       array_stride(0),
-       matrix_stride(0),
-       type(VOID),
-       binding(-1)
-{ }
-
-
-Program::UniformBlockInfo::UniformBlockInfo():
-       data_size(0),
-       bind_point(-1),
-       layout_hash(0)
-{ }
-
-
-Program::AttributeInfo::AttributeInfo():
-       location(-1),
-       array_size(0),
-       type(VOID)
-{ }
-
-
 Program::Loader::Loader(Program &p, Collection &c):
        DataFile::CollectionObjectLoader<Program>(p, &c)
 {