]> git.tdb.fi Git - libs/gl.git/blobdiff - source/core/module.cpp
Move backend information into Device
[libs/gl.git] / source / core / module.cpp
index 6f7fbdb93e9799dc488b7b10646c48cf2bc0c4ac..15bd750cf0b299bd7ea8440ed6d0be5adc3788d1 100644 (file)
@@ -1,4 +1,6 @@
+#include <msp/core/algorithm.h>
 #include <msp/io/print.h>
+#include "device.h"
 #include "module.h"
 #include "resources.h"
 
@@ -36,6 +38,7 @@ enum SpirVConstants
        DECO_SPEC_ID = 1,
        DECO_ARRAY_STRIDE = 6,
        DECO_MATRIX_STRIDE = 7,
+       DECO_BUILTIN = 11,
        DECO_LOCATION = 30,
        DECO_BINDING = 33,
        DECO_DESCRIPTOR_SET = 34,
@@ -47,14 +50,14 @@ namespace GL {
 
 void Module::set_source(const string &src)
 {
-       SL::Compiler compiler;
+       SL::Compiler compiler(create_features());
        compiler.set_source(src);
        compile(compiler);
 }
 
 void Module::load_source(IO::Base &io, Resources *res, const string &name)
 {
-       SL::Compiler compiler;
+       SL::Compiler compiler(create_features());
        compiler.load_source(io, res, name);
        compile(compiler);
 }
@@ -64,6 +67,20 @@ void Module::load_source(IO::Base &io, const string &name)
        load_source(io, 0, name);
 }
 
+SL::Features Module::create_features() const
+{
+       const DeviceInfo &dev_info = Device::get_current().get_info();
+       const SL::Features &device_features = dev_info.glsl_features;
+       SL::Features latest_features = SL::Features::latest(dev_info.api);
+       SL::Features features;
+       features.target_api = latest_features.target_api;
+       features.glsl_version = latest_features.glsl_version;
+       features.constant_id_range = device_features.constant_id_range;
+       features.uniform_binding_range = device_features.uniform_binding_range;
+       features.texture_binding_range = device_features.texture_binding_range;
+       return features;
+}
+
 
 void GlslModule::compile(SL::Compiler &compiler)
 {
@@ -100,23 +117,23 @@ SpirVModule &SpirVModule::operator=(const SpirVModule &other)
 
 void SpirVModule::remap_pointers_from(const SpirVModule &other)
 {
-       for(vector<EntryPoint>::iterator i=entry_points.begin(); i!=entry_points.end(); ++i)
-               for(vector<const Variable *>::iterator j=i->globals.begin(); j!=i->globals.end(); ++j)
-                       *j = &variables[*j-&other.variables.front()];
+       for(EntryPoint &e: entry_points)
+               for(const Variable *&v: e.globals)
+                       v = &variables[v-&other.variables.front()];
 
-       for(vector<Variable>::iterator i=variables.begin(); i!=variables.end(); ++i)
-               if(i->struct_type)
-                       i->struct_type = &structs[i->struct_type-&other.structs.front()];
+       for(Variable &v: variables)
+               if(v.struct_type)
+                       v.struct_type = &structs[v.struct_type-&other.structs.front()];
 
-       for(vector<Structure>::iterator i=structs.begin(); i!=structs.end(); ++i)
-               for(vector<StructMember>::iterator j=i->members.begin(); j!=i->members.end(); ++j)
-                       if(j->struct_type)
-                               j->struct_type = &structs[j->struct_type-&other.structs.front()];
+       for(Structure &s: structs)
+               for(StructMember &m: s.members)
+                       if(m.struct_type)
+                               m.struct_type = &structs[m.struct_type-&other.structs.front()];
 }
 
 void SpirVModule::load_code(IO::Base &io)
 {
-       UInt32 buffer[1024];
+       uint32_t buffer[1024];
        while(1)
        {
                unsigned len = io.read(reinterpret_cast<char *>(buffer), sizeof(buffer));
@@ -127,13 +144,25 @@ void SpirVModule::load_code(IO::Base &io)
                code.insert(code.end(), buffer, buffer+len);
        }
 
+       reflect();
+}
+
+void SpirVModule::compile(SL::Compiler &compiler)
+{
+       compiler.compile(SL::Compiler::SPIRV);
+       code = compiler.get_combined_spirv();
+       reflect();
+}
+
+void SpirVModule::reflect()
+{
        if(code.empty())
                throw invalid_module("Empty SPIR-V code");
 
        if(code[0]==SPIRV_MAGIC_REVERSED)
        {
-               for(vector<UInt32>::iterator i=code.begin(); i!=code.end(); ++i)
-                       *i = ((*i&0xFF)<<24) || ((*i&0xFF00)<<8) | ((*i>>8)&0xFF00) | ((*i>>24)&0xFF);
+               for(uint32_t &c: code)
+                       c = ((c&0xFF)<<24) || ((c&0xFF00)<<8) | ((c>>8)&0xFF00) | ((c>>24)&0xFF);
        }
        else if(code[0]!=SPIRV_MAGIC)
                throw invalid_module("SPIR-V magic number not found");
@@ -141,26 +170,32 @@ void SpirVModule::load_code(IO::Base &io)
        Reflection reflection;
        reflection.reflect_code(code);
 
+       map<const Constant *, unsigned> spec_indices;
+       for(const auto &kvp: reflection.constants)
+               if(kvp.second.constant_id>=0)
+               {
+                       spec_indices[&kvp.second] = spec_constants.size();
+                       spec_constants.push_back(kvp.second);
+               }
+
        map<const Structure *, unsigned> struct_indices;
        structs.reserve(reflection.structs.size());
-       for(map<unsigned, Structure>::const_iterator i=reflection.structs.begin(); i!=reflection.structs.end(); ++i)
+       for(const auto &kvp: reflection.structs)
        {
-               struct_indices[&i->second] = structs.size();
-               structs.push_back(i->second);
+               struct_indices[&kvp.second] = structs.size();
+               structs.push_back(kvp.second);
        }
 
-       for(vector<Structure>::iterator i=structs.begin(); i!=structs.end(); ++i)
+       for(Structure &s: structs)
        {
-               for(vector<StructMember>::iterator j=i->members.begin(); j!=i->members.end(); ++j)
-               {
-                       if(j->struct_type)
+               for(StructMember &m: s.members)
+                       if(m.struct_type)
                        {
-                               map<const Structure *, unsigned>::const_iterator k = struct_indices.find(j->struct_type);
-                               j->struct_type = (k!=struct_indices.end() ? &structs[k->second] : 0);
+                               auto i = struct_indices.find(m.struct_type);
+                               m.struct_type = (i!=struct_indices.end() ? &structs[i->second] : 0);
                        }
-               }
 
-               const StructMember *last_member = &i->members.back();
+               const StructMember *last_member = &s.members.back();
                unsigned last_offset = last_member->offset;
                while(last_member->struct_type)
                {
@@ -171,80 +206,48 @@ void SpirVModule::load_code(IO::Base &io)
                        last_member = lm;
                }
                
-               i->size = last_offset+get_type_size(last_member->type);
-               i->size = (i->size+15)&~15;
+               unsigned last_size = get_type_size(last_member->type); 
+               if(last_member->array_size)
+                       last_size += last_member->array_stride*(last_member->array_size-1);
+               s.size = last_offset+last_size;
+               s.size = (s.size+15)&~15;
        }
 
        map<const Variable *, unsigned> var_indices;
        variables.reserve(reflection.variables.size());
-       for(map<unsigned, Variable>::const_iterator i=reflection.variables.begin(); i!=reflection.variables.end(); ++i)
+       for(const auto &kvp: reflection.variables)
        {
-               int dup_index = -1;
-               for(vector<Variable>::const_iterator j=variables.begin(); (dup_index<0 && j!=variables.end()); ++j)
-                       if(*j==i->second)
-                               dup_index = j-variables.begin();
-
-               if(dup_index>=0)
-                       var_indices[&i->second] = dup_index;
+               auto i = find_if(variables, [&kvp](const Variable &v){ return v==kvp.second; });
+               if(i!=variables.end())
+                       var_indices[&kvp.second] = i-variables.begin();
                else
                {
-                       var_indices[&i->second] = variables.size();
-                       variables.push_back(i->second);
+                       var_indices[&kvp.second] = variables.size();
+                       variables.push_back(kvp.second);
                }
        }
 
-       for(vector<Variable>::iterator i=variables.begin(); i!=variables.end(); ++i)
-               if(i->struct_type)
+       for(Variable &v: variables)
+               if(v.struct_type)
                {
-                       map<const Structure *, unsigned>::const_iterator j = struct_indices.find(i->struct_type);
-                       i->struct_type = (j!=struct_indices.end() ? &structs[j->second] : 0);
+                       auto i = struct_indices.find(v.struct_type);
+                       v.struct_type = (i!=struct_indices.end() ? &structs[i->second] : 0);
                }
 
        entry_points.reserve(reflection.entry_points.size());
-       for(map<unsigned, EntryPoint>::const_iterator i=reflection.entry_points.begin(); i!=reflection.entry_points.end(); ++i)
+       for(const auto &kvp: reflection.entry_points)
        {
-               entry_points.push_back(i->second);
+               entry_points.push_back(kvp.second);
                EntryPoint &entry = entry_points.back();
-               for(vector<const Variable *>::iterator j=entry.globals.begin(); j!=entry.globals.end(); ++j)
+               for(const Variable *&v: entry.globals)
                {
-                       map<const Variable *, unsigned>::const_iterator k = var_indices.find(*j);
-                       *j = (k!=var_indices.end() ? &variables[k->second] : 0);
+                       auto i = var_indices.find(v);
+                       v = (i!=var_indices.end() ? &variables[i->second] : 0);
                }
        }
-
-       for(map<unsigned, SpecConstant>::const_iterator i=reflection.spec_constants.begin(); i!=reflection.spec_constants.end(); ++i)
-               spec_constants.push_back(i->second);
-}
-
-void SpirVModule::compile(SL::Compiler &)
-{
-       throw logic_error("Not implemented yet");
 }
 
 
-SpirVModule::EntryPoint::EntryPoint():
-       stage(VERTEX)
-{ }
-
-
-SpirVModule::StructMember::StructMember():
-       type(VOID),
-       struct_type(0),
-       offset(0),
-       array_size(0),
-       array_stride(0),
-       matrix_stride(0)
-{ }
-
-
-SpirVModule::Variable::Variable():
-       type(VOID),
-       struct_type(0),
-       location(-1),
-       descriptor_set(-1),
-       binding(-1)
-{ }
-
 bool SpirVModule::Variable::operator==(const Variable &other) const
 {
        if(storage!=UNIFORM_CONSTANT && storage!=UNIFORM)
@@ -257,16 +260,7 @@ bool SpirVModule::Variable::operator==(const Variable &other) const
 }
 
 
-SpirVModule::TypeInfo::TypeInfo():
-       type(VOID),
-       struct_type(0),
-       array_size(0),
-       array_stride(0),
-       storage(static_cast<StorageClass>(-1))
-{ }
-
-
-UInt32 SpirVModule::Reflection::get_opcode(UInt32 op)
+uint32_t SpirVModule::Reflection::get_opcode(uint32_t op)
 {
        return op&0xFFFF;
 }
@@ -298,7 +292,7 @@ string SpirVModule::Reflection::read_string(CodeIterator &op, const CodeIterator
        throw invalid_module("Unterminated SPIR-V string literal");
 }
 
-void SpirVModule::Reflection::reflect_code(const vector<UInt32> &code)
+void SpirVModule::Reflection::reflect_code(const vector<uint32_t> &code)
 {
        for(CodeIterator op=code.begin()+5; op!=code.end(); )
        {
@@ -316,18 +310,18 @@ void SpirVModule::Reflection::reflect_code(const vector<UInt32> &code)
                case OP_TYPE_INT: reflect_int_type(op); break;
                case OP_TYPE_FLOAT: reflect_float_type(op); break;
                case OP_TYPE_VECTOR: reflect_vector_type(op); break;
-               case OP_TYPE_MATRIX: reflect_vector_type(op); break;
+               case OP_TYPE_MATRIX: reflect_matrix_type(op); break;
                case OP_TYPE_IMAGE: reflect_image_type(op); break;
                case OP_TYPE_SAMPLED_IMAGE: reflect_sampled_image_type(op); break;
                case OP_TYPE_ARRAY: reflect_array_type(op); break;
                case OP_TYPE_STRUCT: reflect_struct_type(op); break;
                case OP_TYPE_POINTER: reflect_pointer_type(op); break;
-               case OP_CONSTANT_TRUE: constants[*(op+2)] = true; break;
-               case OP_CONSTANT_FALSE: constants[*(op+2)] = false; break;
-               case OP_CONSTANT: reflect_constant(op); break;
+               case OP_CONSTANT_TRUE:
+               case OP_CONSTANT_FALSE:
+               case OP_CONSTANT:
                case OP_SPEC_CONSTANT_TRUE:
                case OP_SPEC_CONSTANT_FALSE:
-               case OP_SPEC_CONSTANT: reflect_spec_constant(op); break;
+               case OP_SPEC_CONSTANT: reflect_constant(op); break;
                case OP_VARIABLE: reflect_variable(op); break;
                case OP_DECORATE: reflect_decorate(op); break;
                case OP_MEMBER_DECORATE: reflect_member_decorate(op); break;
@@ -399,7 +393,7 @@ void SpirVModule::Reflection::reflect_vector_type(CodeIterator op)
        TypeInfo &type = types[*(op+1)];
        DataType component = types[*(op+2)].type;
        unsigned count = *(op+3);
-       type.type = static_cast<DataType>((count<<12) | (component&0xF00) | ((component&0xFF)*count));
+       type.type = static_cast<DataType>(((count-1)<<12) | (component&0xF00) | ((component&0xFF)*count));
 }
 
 void SpirVModule::Reflection::reflect_matrix_type(CodeIterator op)
@@ -407,7 +401,7 @@ void SpirVModule::Reflection::reflect_matrix_type(CodeIterator op)
        TypeInfo &type = types[*(op+1)];
        DataType column = types[*(op+2)].type;
        unsigned count = *(op+3);
-       type.type = static_cast<DataType>((count<<16) | (column&0xF00) | ((column&0xFF)*count));
+       type.type = static_cast<DataType>(((count-1)<<14) | (column&0x3F00) | ((column&0xFF)*count));
 }
 
 void SpirVModule::Reflection::reflect_image_type(CodeIterator op)
@@ -433,11 +427,10 @@ void SpirVModule::Reflection::reflect_array_type(CodeIterator op)
        const TypeInfo &elem = types[*(op+2)];
        type.type = elem.type;
        type.struct_type = elem.struct_type;
-       const Variant &size = constants[*(op+3)];
-       if(size.check_type<int>())
-               type.array_size = size.value<int>();
-       else if(size.check_type<unsigned>())
-               type.array_size = size.value<unsigned>();
+
+       const Constant &size = constants[*(op+3)];
+       if(size.type==INT || size.type==UNSIGNED_INT)
+               type.array_size = size.i_value;
 }
 
 void SpirVModule::Reflection::reflect_struct_type(CodeIterator op)
@@ -450,7 +443,7 @@ void SpirVModule::Reflection::reflect_struct_type(CodeIterator op)
        op += 2;
 
        strct.members.resize(op_end-op);
-       vector<StructMember>::iterator mem = strct.members.begin();
+       auto mem = strct.members.begin();
        for(; op!=op_end; ++op, ++mem)
        {
                TypeInfo &type = types[*op];
@@ -469,23 +462,19 @@ void SpirVModule::Reflection::reflect_pointer_type(CodeIterator op)
 }
 
 void SpirVModule::Reflection::reflect_constant(CodeIterator op)
-{
-       const TypeInfo &type = types[*(op+1)];
-       unsigned id = *(op+2);
-       if(type.type==INT)
-               constants[id] = static_cast<int>(*(op+3));
-       else if(type.type==UNSIGNED_INT)
-               constants[id] = static_cast<unsigned>(*(op+3));
-       else if(type.type==FLOAT)
-               constants[id] = *reinterpret_cast<const float *>(&*(op+3));
-}
-
-void SpirVModule::Reflection::reflect_spec_constant(CodeIterator op)
 {
        unsigned id = *(op+2);
-       SpecConstant &spec = spec_constants[id];
-       spec.name = names[id];
-       spec.type = types[*(op+1)].type;
+       Constant &cnst = constants[id];
+       cnst.name = names[id];
+       cnst.type = types[*(op+1)].type;
+       if(*op==OP_CONSTANT_TRUE || *op==OP_SPEC_CONSTANT_TRUE)
+               cnst.i_value = true;
+       else if(*op==OP_CONSTANT_FALSE || *op==OP_SPEC_CONSTANT_FALSE)
+               cnst.i_value = false;
+       else if(cnst.type==INT || cnst.type==UNSIGNED_INT)
+               cnst.i_value = *(op+3);
+       else if(cnst.type==FLOAT)
+               cnst.f_value = *reinterpret_cast<const float *>(&*(op+3));
 }
 
 void SpirVModule::Reflection::reflect_variable(CodeIterator op)
@@ -509,11 +498,14 @@ void SpirVModule::Reflection::reflect_decorate(CodeIterator op)
        switch(decoration)
        {
        case DECO_SPEC_ID:
-               spec_constants[id].constant_id = *op;
+               constants[id].constant_id = *op;
                break;
        case DECO_ARRAY_STRIDE:
                types[id].array_stride = *op;
                break;
+       case DECO_BUILTIN:
+               variables[id].builtin = static_cast<BuiltinSemantic>(*op);
+               break;
        case DECO_LOCATION:
                variables[id].location = *op;
                break;
@@ -541,6 +533,9 @@ void SpirVModule::Reflection::reflect_member_decorate(CodeIterator op)
        case DECO_MATRIX_STRIDE:
                member.matrix_stride = *op;
                break;
+       case DECO_BUILTIN:
+               member.builtin = static_cast<BuiltinSemantic>(*op);
+               break;
        case DECO_OFFSET:
                member.offset = *op;
                break;