]> git.tdb.fi Git - libs/gl.git/blobdiff - source/glsl/finalize.cpp
Use default member initializers for simple types
[libs/gl.git] / source / glsl / finalize.cpp
index fb76ab6d57dc49c22997ad7a64b07e030ed45b34..4068037e919e81b5cc514d12584bec7df8088360 100644 (file)
@@ -1,8 +1,10 @@
 #include <msp/core/algorithm.h>
+#include <msp/core/hash.h>
 #include <msp/core/raii.h>
 #include <msp/strings/lexicalcast.h>
 #include "finalize.h"
 #include "glsl_error.h"
+#include "reflect.h"
 
 using namespace std;
 
@@ -10,86 +12,306 @@ namespace Msp {
 namespace GL {
 namespace SL {
 
-DefaultPrecisionGenerator::DefaultPrecisionGenerator():
-       stage(0)
-{ }
+void StructOrganizer::visit(StructDeclaration &strct)
+{
+       SetForScope<int> set_offset(offset, 0);
+       TraversingVisitor::visit(strct);
+}
 
-void DefaultPrecisionGenerator::apply(Stage &s)
+void StructOrganizer::visit(VariableDeclaration &var)
 {
-       stage = &s;
-       s.content.visit(*this);
+       if(offset>=0)
+       {
+               int *layout_offset = 0;
+               bool has_matrix_order = false;
+               if(var.layout)
+               {
+                       for(Layout::Qualifier &q: var.layout->qualifiers)
+                       {
+                               if(q.name=="offset" && q.has_value)
+                               {
+                                       layout_offset = &q.value;
+                                       if(q.value>=offset)
+                                               offset = q.value;
+                               }
+                               else if(q.name=="column_major" || q.name=="row_major")
+                                       has_matrix_order = true;
+                       }
+               }
+
+               MemoryRequirementsCalculator::Result mem_reqs = MemoryRequirementsCalculator().apply(var);
+               offset += mem_reqs.alignment-1;
+               offset -= offset%mem_reqs.alignment;
+
+               if(layout_offset)
+                       *layout_offset = offset;
+               else
+               {
+                       if(!var.layout)
+                               var.layout = new Layout;
+
+                       var.layout->qualifiers.push_back(Layout::Qualifier("offset", offset));
+               }
+
+               if(!has_matrix_order)
+               {
+                       const BasicTypeDeclaration *basic = dynamic_cast<const BasicTypeDeclaration *>(var.type_declaration);
+                       while(basic && basic->kind==BasicTypeDeclaration::ARRAY)
+                               basic = dynamic_cast<const BasicTypeDeclaration *>(basic->base_type);
+                       if(basic && basic->kind==BasicTypeDeclaration::MATRIX)
+                               var.layout->qualifiers.push_back(Layout::Qualifier("column_major"));
+               }
+
+               offset += mem_reqs.size;
+       }
 }
 
-void DefaultPrecisionGenerator::visit(Block &block)
+
+void LocationAllocator::apply(Module &module, const Features &features)
 {
-       for(NodeList<Statement>::iterator i=block.body.begin(); i!=block.body.end(); ++i)
+       for(Stage &s: module.stages)
+               apply(s);
+       allocate_locations("uniform");
+
+       for(InterfaceBlock *b: unbound_blocks)
+               bind_uniform(b->layout, b->block_name, features.uniform_binding_range);
+       for(VariableDeclaration *t: unbound_textures)
+               bind_uniform(t->layout, t->name, features.texture_binding_range);
+}
+
+void LocationAllocator::apply(Stage &stage)
+{
+       swap(used_locations["in"], used_locations["out"]);
+       used_locations["out"].clear();
+
+       stage.content.visit(*this);
+
+       allocate_locations("in");
+       allocate_locations("out");
+}
+
+void LocationAllocator::allocate_locations(const string &iface)
+{
+       auto write = unplaced_variables.begin();
+       unsigned next = 0;
+       for(auto i=unplaced_variables.begin(); i!=unplaced_variables.end(); ++i)
        {
-               if(&block==&stage->content)
-                       insert_point = i;
-               (*i)->visit(*this);
+               if((*i)->interface!=iface)
+               {
+                       if(write!=i)
+                               *write = *i;
+                       ++write;
+                       continue;
+               }
+
+               if((*i)->interface=="uniform")
+               {
+                       auto j = uniforms.find((*i)->name);
+                       if(j!=uniforms.end() && j->second.location>=0)
+                       {
+                               add_layout_value((*i)->layout, "location", j->second.location);
+                               continue;
+                       }
+               }
+
+               set<unsigned> &used = used_locations[(*i)->interface];
+
+               unsigned size = LocationCounter().apply(**i);
+               while(1)
+               {
+                       int blocking = -1;
+                       for(unsigned j=0; j<size; ++j)
+                               if(used.count(next+j))
+                                       blocking = next+j;
+                       if(blocking<0)
+                               break;
+                       next = blocking+1;
+               }
+
+               add_layout_value((*i)->layout, "location", next);
+               if((*i)->interface=="uniform")
+                       uniforms[(*i)->name].location = next;
+
+               for(unsigned j=0; j<size; ++j)
+                       used.insert(next+j);
+               next += size;
+       }
+
+       unplaced_variables.erase(write, unplaced_variables.end());
+}
+
+void LocationAllocator::bind_uniform(RefPtr<Layout> &layout, const string &name, unsigned range)
+{
+       auto i = uniforms.find(name);
+       if(i!=uniforms.end() && i->second.bind_point>=0)
+               add_layout_value(layout, "binding", i->second.bind_point);
+       else
+       {
+               set<unsigned> &used = used_bindings[0];
+
+               unsigned bind_point = fold32(hash64(name))%range;
+               while(used.count(bind_point))
+                       bind_point = (bind_point+1)%range;
+
+               add_layout_value(layout, "binding", bind_point);
+               uniforms[name].bind_point = bind_point;
+               used.insert(bind_point);
        }
 }
 
-void DefaultPrecisionGenerator::visit(Precision &prec)
+void LocationAllocator::add_layout_value(RefPtr<Layout> &layout, const string &name, unsigned value)
 {
-       have_default.insert(prec.type);
+       if(!layout)
+               layout = new Layout;
+
+       layout->qualifiers.push_back(Layout::Qualifier(name, value));
 }
 
-void DefaultPrecisionGenerator::visit(VariableDeclaration &var)
+void LocationAllocator::visit(VariableDeclaration &var)
 {
-       if(var.type_declaration)
+       if(!var.name.compare(0, 3, "gl_"))
                return;
 
-       string type = var.type;
-       if(!type.compare(0, 3, "vec") || !type.compare(0, 3, "mat"))
-               type = "float";
-       else if(!type.compare(0, 3, "ivec") || type=="uint")
-               type = "int";
-
-       if(!have_default.count(type))
+       if(!var.interface.empty())
        {
-               Precision *prec = new Precision;
-               if(!type.compare(0, 7, "sampler"))
-                       prec->precision = "lowp";
-               else if(stage->type==Stage::FRAGMENT)
-                       prec->precision = "mediump";
+               int location = (var.layout ? get_layout_value(*var.layout, "location") : -1);
+
+               if(location<0 && var.linked_declaration && var.linked_declaration->layout)
+               {
+                       location = get_layout_value(*var.linked_declaration->layout, "location");
+                       if(location>=0)
+                               add_layout_value(var.layout, "location", location);
+               }
+
+               if(location>=0)
+               {
+                       unsigned size = LocationCounter().apply(var);
+                       for(unsigned i=0; i<size; ++i)
+                               used_locations[var.interface].insert(location+i);
+                       if(var.interface=="uniform")
+                               uniforms[var.name].location = location;
+               }
                else
-                       prec->precision = "highp";
-               prec->type = type;
-               stage->content.body.insert(insert_point, prec);
+                       unplaced_variables.push_back(&var);
+       }
 
-               have_default.insert(type);
+       if(var.interface=="uniform")
+       {
+               const TypeDeclaration *type = var.type_declaration;
+               while(const BasicTypeDeclaration *basic = dynamic_cast<const BasicTypeDeclaration *>(type))
+                       type = basic->base_type;
+               if(dynamic_cast<const ImageTypeDeclaration *>(type))
+               {
+                       int bind_point = (var.layout ? get_layout_value(*var.layout, "binding") : -1);
+                       if(bind_point>=0)
+                       {
+                               used_bindings[0].insert(bind_point);
+                               uniforms[var.name].bind_point = bind_point;
+                       }
+                       else
+                               unbound_textures.push_back(&var);
+               }
        }
 }
 
+void LocationAllocator::visit(InterfaceBlock &iface)
+{
+       if(!iface.instance_name.compare(0, 3, "gl_"))
+               return;
+
+       if(iface.interface=="uniform")
+       {
+               int bind_point = (iface.layout ? get_layout_value(*iface.layout, "binding") : -1);
+
+               if(bind_point>=0)
+               {
+                       used_bindings[0].insert(bind_point);
+                       uniforms[iface.block_name].bind_point = bind_point;
+               }
+               else
+                       unbound_blocks.push_back(&iface);
+       }
+}
 
-void PrecisionRemover::apply(Stage &stage)
+
+void PrecisionConverter::apply(Stage &s)
 {
-       stage.content.visit(*this);
-       NodeRemover().apply(stage, nodes_to_remove);
+       stage = &s;
+       s.content.visit(*this);
+       NodeRemover().apply(s, nodes_to_remove);
 }
 
-void PrecisionRemover::visit(Precision &prec)
+void PrecisionConverter::visit(Block &block)
 {
-       nodes_to_remove.insert(&prec);
+       for(auto i=block.body.begin(); i!=block.body.end(); ++i)
+       {
+               if(&block==&stage->content)
+                       insert_point = i;
+               (*i)->visit(*this);
+       }
 }
 
-void PrecisionRemover::visit(VariableDeclaration &var)
+void PrecisionConverter::visit(Precision &prec)
 {
-       var.precision.clear();
+       if(stage->required_features.target_api==OPENGL_ES)
+               have_default.insert(prec.type);
+       else
+               nodes_to_remove.insert(&prec);
 }
 
+void PrecisionConverter::visit(VariableDeclaration &var)
+{
+       if(stage->required_features.target_api!=OPENGL_ES)
+       {
+               var.precision.clear();
+               return;
+       }
+
+       const char *default_prec = (stage->type==Stage::FRAGMENT ? "mediump" : "highp");
+       const TypeDeclaration *type = var.type_declaration;
+       while(type)
+       {
+               if(dynamic_cast<const ImageTypeDeclaration *>(type))
+               {
+                       default_prec = "lowp";
+                       break;
+               }
+               else if(const BasicTypeDeclaration *basic = dynamic_cast<const BasicTypeDeclaration *>(type))
+               {
+                       if(basic->kind==BasicTypeDeclaration::INT || basic->kind==BasicTypeDeclaration::FLOAT)
+                               break;
+                       type = basic->base_type;
+               }
+               else
+                       return;
+       }
+       if(!type)
+               return;
+
+       if(!have_default.count(type->name))
+       {
+               Precision *prec = new Precision;
+               prec->precision = default_prec;
+               prec->type = type->name;
+               stage->content.body.insert(insert_point, prec);
+
+               have_default.insert(type->name);
+       }
+}
 
-LegacyConverter::LegacyConverter():
-       frag_out(0)
-{ }
 
 void LegacyConverter::apply(Stage &s, const Features &feat)
 {
        stage = &s;
        features = feat;
        if(supports_stage(s.type))
+       {
                s.content.visit(*this);
+               NodeRemover().apply(s, nodes_to_remove);
+
+               if(!stage->required_features.glsl_version)
+                       stage->required_features.glsl_version = Version(1, (stage->required_features.target_api==OPENGL_ES ? 0 : 10));
+       }
        else
                unsupported(format("Stage %s is not supported", Stage::get_stage_name(s.type)));
 }
@@ -106,7 +328,7 @@ void LegacyConverter::unsupported(const string &reason)
 
 void LegacyConverter::visit(Block &block)
 {
-       for(NodeList<Statement>::iterator i=block.body.begin(); i!=block.body.end(); ++i)
+       for(auto i=block.body.begin(); i!=block.body.end(); ++i)
        {
                if(&block==&stage->content)
                        uniform_insert_point = i;
@@ -138,7 +360,7 @@ bool LegacyConverter::supports_stage(Stage::Type st) const
 {
        if(st==Stage::GEOMETRY)
        {
-               if(features.gl_api==OPENGL_ES2)
+               if(features.target_api==OPENGL_ES)
                        return check_version(Version(3, 20));
                else
                        return check_version(Version(1, 50));
@@ -149,7 +371,7 @@ bool LegacyConverter::supports_stage(Stage::Type st) const
 
 bool LegacyConverter::supports_unified_interface_syntax() const
 {
-       if(features.gl_api==OPENGL_ES2)
+       if(features.target_api==OPENGL_ES)
                return check_version(Version(3, 0));
        else
                return check_version(Version(1, 30));
@@ -161,12 +383,7 @@ void LegacyConverter::visit(VariableReference &var)
        {
                var.name = "gl_FragColor";
                var.declaration = 0;
-               r_type = "vec4";
        }
-       else if(var.declaration)
-               r_type = var.declaration->type;
-       else
-               r_type.clear();
 }
 
 void LegacyConverter::visit(Assignment &assign)
@@ -178,7 +395,7 @@ void LegacyConverter::visit(Assignment &assign)
 
 bool LegacyConverter::supports_unified_sampling_functions() const
 {
-       if(features.gl_api==OPENGL_ES2)
+       if(features.target_api==OPENGL_ES)
                return check_version(Version(3, 0));
        else
                return check_version(Version(1, 30));
@@ -186,73 +403,66 @@ bool LegacyConverter::supports_unified_sampling_functions() const
 
 void LegacyConverter::visit(FunctionCall &call)
 {
-       if(call.name=="texture")
+       if(call.declaration && call.declaration->source==BUILTIN_SOURCE)
        {
-               string sampler_type;
-               r_type.clear();
-               NodeArray<Expression>::iterator i = call.arguments.begin();
-               if(i!=call.arguments.end())
-               {
-                       (*i)->visit(*this);
-                       sampler_type = r_type;
-
-                       for(; i!=call.arguments.end(); ++i)
-                               (*i)->visit(*this);
-               }
-
-               if(!supports_unified_sampling_functions())
+               if(!call.name.compare(0, 7, "texture") && call.arguments.size()>=1)
                {
-                       if(sampler_type=="sampler1D")
-                               call.name = "texture1D";
-                       else if(sampler_type=="sampler2D")
-                               call.name = "texture2D";
-                       else if(sampler_type=="sampler3D")
-                               call.name = "texture3D";
-                       else if(sampler_type=="samplerCube")
-                               call.name = "textureCube";
-                       else if(sampler_type=="sampler1DShadow")
-                               call.name = "shadow1D";
-                       else if(sampler_type=="sampler2DShadow")
-                               call.name = "shadow2D";
-                       else if(sampler_type=="sampler1DArray")
-                       {
-                               check_extension(&Features::ext_texture_array);
-                               call.name = "texture1DArray";
-                       }
-                       else if(sampler_type=="sampler2DArray")
+                       const ImageTypeDeclaration *arg_image = dynamic_cast<const ImageTypeDeclaration *>(call.arguments.front()->type);
+                       if(arg_image && !supports_unified_sampling_functions())
                        {
-                               check_extension(&Features::ext_texture_array);
-                               call.name = "texture2DArray";
-                       }
-                       else if(sampler_type=="sampler1DArrayShadow")
-                       {
-                               check_extension(&Features::ext_texture_array);
-                               call.name = "shadow1DArray";
-                       }
-                       else if(sampler_type=="sampler2DArrayShadow")
-                       {
-                               check_extension(&Features::ext_texture_array);
-                               call.name = "shadow2DArray";
+                               string suffix = call.name.substr(7);
+                               call.name = (arg_image->shadow ? "shadow" : "texture");
+
+                               switch(arg_image->dimensions)
+                               {
+                               case ImageTypeDeclaration::ONE: call.name += "1D"; break;
+                               case ImageTypeDeclaration::TWO: call.name += "2D"; break;
+                               case ImageTypeDeclaration::THREE: call.name += "3D"; break;
+                               case ImageTypeDeclaration::CUBE: call.name += "Cube"; break;
+                               }
+
+                               if(arg_image->array)
+                               {
+                                       /* Array textures and the unified sampling function name were
+                                       both introduced in GLSL 1.30. */
+                                       if(arg_image->dimensions==ImageTypeDeclaration::ONE || arg_image->dimensions==ImageTypeDeclaration::TWO)
+                                               check_extension(&Features::ext_texture_array);
+                                       call.name += "Array";
+                               }
+
+                               call.name += suffix;
                        }
                }
        }
-       else
-               TraversingVisitor::visit(call);
+
+       TraversingVisitor::visit(call);
 }
 
 bool LegacyConverter::supports_interface_layouts() const
 {
-       if(features.gl_api==OPENGL_ES2)
+       if(features.target_api==OPENGL_ES)
                return check_version(Version(3, 0));
        else if(check_version(Version(3, 30)))
                return true;
-       else
+       else if(check_version(Version(1, 30)))
                return check_extension(&Features::arb_explicit_attrib_location);
+       else
+               return false;
+}
+
+bool LegacyConverter::supports_stage_interface_layouts() const
+{
+       if(features.target_api==OPENGL_ES)
+               return check_version(Version(3, 10));
+       else if(check_version(Version(4, 10)))
+               return true;
+       else
+               return check_extension(&Features::arb_separate_shader_objects);
 }
 
 bool LegacyConverter::supports_centroid_sampling() const
 {
-       if(features.gl_api==OPENGL_ES2)
+       if(features.target_api==OPENGL_ES)
                return check_version(Version(3, 0));
        else if(check_version(Version(1, 20)))
                return true;
@@ -262,7 +472,7 @@ bool LegacyConverter::supports_centroid_sampling() const
 
 bool LegacyConverter::supports_sample_sampling() const
 {
-       if(features.gl_api==OPENGL_ES2)
+       if(features.target_api==OPENGL_ES)
                return check_version(Version(3, 20));
        else if(check_version(Version(4, 0)))
                return true;
@@ -270,31 +480,78 @@ bool LegacyConverter::supports_sample_sampling() const
                return check_extension(&Features::arb_gpu_shader5);
 }
 
+bool LegacyConverter::supports_uniform_location() const
+{
+       if(features.target_api==OPENGL_ES)
+               return check_version(Version(3, 10));
+       else if(check_version(Version(4, 30)))
+               return true;
+       else
+               return check_extension(&Features::arb_explicit_uniform_location);
+}
+
+bool LegacyConverter::supports_binding() const
+{
+       if(features.target_api==OPENGL_ES)
+               return check_version(Version(3, 10));
+       else
+               return check_version(Version(4, 20));
+}
+
 void LegacyConverter::visit(VariableDeclaration &var)
 {
-       if(var.layout && !supports_interface_layouts())
+       if(var.layout)
        {
-               vector<Layout::Qualifier>::iterator i;
-               for(i=var.layout->qualifiers.begin(); (i!=var.layout->qualifiers.end() && i->name!="location"); ++i) ;
-               if(i!=var.layout->qualifiers.end())
+               for(auto i=var.layout->qualifiers.begin(); i!=var.layout->qualifiers.end(); )
                {
-                       if(stage->type==Stage::VERTEX && var.interface=="in")
+                       if(i->name=="location")
                        {
-                               stage->locations[var.name] = i->value;
-                               var.layout->qualifiers.erase(i);
+                               bool supported = true;
+                               bool external = false;
+                               if(var.interface=="in")
+                               {
+                                       external = (stage->type==Stage::VERTEX);
+                                       supported = (external ? supports_interface_layouts() : supports_stage_interface_layouts());
+                               }
+                               else if(var.interface=="out")
+                               {
+                                       external = (stage->type==Stage::FRAGMENT);
+                                       supported = (external ? supports_interface_layouts() : supports_stage_interface_layouts());
+                                       if(external && !supported && !check_extension(&Features::ext_gpu_shader4))
+                                       {
+                                               external = false;
+                                               if(i->value!=0)
+                                                       unsupported("EXT_gpu_shader4 required for multiple fragment shader outputs");
+                                       }
+                               }
+                               else if(var.interface=="uniform")
+                                       supported = supports_uniform_location();
+
+                               if(!supported)
+                               {
+                                       if(external)
+                                               stage->locations[var.name] = i->value;
+                                       i = var.layout->qualifiers.erase(i);
+                               }
+                               else
+                                       ++i;
                        }
-                       else if(stage->type==Stage::FRAGMENT && var.interface=="out")
+                       else if(i->name=="binding" && !supports_binding())
                        {
-                               if(check_extension(&Features::ext_gpu_shader4))
-                                       stage->locations[var.name] = i->value;
-                               else if(i->value!=0)
-                                       unsupported("EXT_gpu_shader4 required for multiple fragment shader outputs");
-                               var.layout->qualifiers.erase(i);
-                       }
+                               const TypeDeclaration *type = var.type_declaration;
+                               while(const BasicTypeDeclaration *basic = dynamic_cast<const BasicTypeDeclaration *>(type))
+                                       type = basic->base_type;
+                               if(dynamic_cast<const ImageTypeDeclaration *>(type))
+                                       stage->texture_bindings[var.name] = i->value;
 
-                       if(var.layout->qualifiers.empty())
-                               var.layout = 0;
+                               i = var.layout->qualifiers.erase(i);
+                       }
+                       else
+                               ++i;
                }
+
+               if(var.layout->qualifiers.empty())
+                       var.layout = 0;
        }
 
        if(var.sampling=="centroid")
@@ -322,7 +579,7 @@ void LegacyConverter::visit(VariableDeclaration &var)
 
 bool LegacyConverter::supports_interface_blocks(const string &iface) const
 {
-       if(features.gl_api==OPENGL_ES2)
+       if(features.target_api==OPENGL_ES)
        {
                if(iface=="uniform")
                        return check_version(Version(3, 0));
@@ -337,8 +594,37 @@ bool LegacyConverter::supports_interface_blocks(const string &iface) const
                return false;
 }
 
+bool LegacyConverter::supports_interface_block_location() const
+{
+       if(features.target_api==OPENGL_ES)
+               return check_version(Version(3, 20));
+       else if(check_version(Version(4, 40)))
+               return true;
+       else
+               return check_extension(&Features::arb_enhanced_layouts);
+}
+
 void LegacyConverter::visit(InterfaceBlock &iface)
 {
+       if(iface.layout)
+       {
+               for(auto i=iface.layout->qualifiers.begin(); i!=iface.layout->qualifiers.end(); )
+               {
+                       if(i->name=="location" && !supports_interface_block_location())
+                               i = iface.layout->qualifiers.erase(i);
+                       else if(i->name=="binding" && !supports_binding())
+                       {
+                               stage->uniform_block_bindings[iface.block_name] = i->value;
+                               i = iface.layout->qualifiers.erase(i);
+                       }
+                       else
+                               ++i;
+               }
+
+               if(iface.layout->qualifiers.empty())
+                       iface.layout = 0;
+       }
+
        if(!supports_interface_blocks(iface.interface) && iface.type_declaration)
        {
                if(!iface.instance_name.empty())