Store types and interface blocks in Stage

[libs/gl.git] / source / glsl / generate.cpp

diff --git a/source/glsl/generate.cpp b/source/glsl/generate.cpp
index 751043d9231a64d4c72af4894ec29c39b8143d48..c073d870ac4d779591fa2c896724737a5de278ca 100644 (file)
--- a/source/glsl/generate.cpp
+++ b/source/glsl/generate.cpp
@@ -80,6 +80,7 @@ void BlockResolver::enter(Block &block)
 
 
 VariableResolver::VariableResolver():
+       stage(0),
        builtins(0),
        members(0),
        record_target(false),
@@ -87,11 +88,12 @@ VariableResolver::VariableResolver():
        self_referencing(false)
 { }
 
-void VariableResolver::apply(Stage &stage)
+void VariableResolver::apply(Stage &s)
 {
-       Stage *builtin_stage = get_builtins(stage.type);
+       stage = &s;
+       Stage *builtin_stage = get_builtins(s.type);
        builtins = (builtin_stage ? &builtin_stage->content : 0);
-       stage.content.visit(*this);
+       s.content.visit(*this);
 }
 
 Block *VariableResolver::next_block(Block &block)
@@ -108,38 +110,38 @@ void VariableResolver::visit(VariableReference &var)
 {
        var.declaration = 0;
        members = 0;
-       for(Block *block=current_block; block; block=next_block(*block))
+       for(Block *block=current_block; (!var.declaration && block); block=next_block(*block))
        {
                map<string, VariableDeclaration *>::iterator i = block->variables.find(var.name);
                if(i!=block->variables.end())
                        var.declaration = i->second;
-               else
-               {
-                       const map<string, InterfaceBlock *> &ifaces = block->interfaces;
-                       for(map<string, InterfaceBlock *>::const_iterator j=ifaces.begin(); (!var.declaration && j!=ifaces.end()); ++j)
-                               if(j->second->instance_name.empty())
-                               {
-                                       i = j->second->members.variables.find(var.name);
-                                       if(i!=j->second->members.variables.end())
-                                               var.declaration = i->second;
-                               }
-               }
-
-               if(var.declaration)
-               {
-                       if(var.declaration->type_declaration)
-                               members = &var.declaration->type_declaration->members.variables;
-                       break;
-               }
+       }
 
-               map<string, InterfaceBlock *>::iterator j = block->interfaces.find(var.name);
-               if(j!=block->interfaces.end() && j->second->instance_name==var.name)
+       if(var.declaration)
+       {
+               if(var.declaration->type_declaration)
+                       members = &var.declaration->type_declaration->members.variables;
+       }
+       else
+       {
+               const map<string, InterfaceBlock *> &blocks = stage->interface_blocks;
+               map<string, InterfaceBlock *>::const_iterator i = blocks.find(var.name);
+               if(i!=blocks.end() && i->second->instance_name==var.name)
                {
                        iface_ref = new InterfaceBlockReference;
                        iface_ref->name = var.name;
-                       iface_ref->declaration = j->second;
-                       members = &j->second->members.variables;
-                       break;
+                       iface_ref->declaration = i->second;
+                       members = &i->second->members.variables;
+               }
+               else
+               {
+                       for(i=blocks.begin(); (!var.declaration && i!=blocks.end()); ++i)
+                               if(i->second->instance_name.empty())
+                               {
+                                       map<string, VariableDeclaration *>::iterator j = i->second->members.variables.find(var.name);
+                                       if(j!=i->second->members.variables.end())
+                                               var.declaration = j->second;
+                               }
                }
        }
 
@@ -162,11 +164,11 @@ void VariableResolver::visit(InterfaceBlockReference &iface)
        iface.declaration = 0;
        for(Block *block=current_block; block; block=next_block(*block))
        {
-               map<string, InterfaceBlock *>::iterator j = block->interfaces.find(iface.name);
-               if(j!=block->interfaces.end())
+               map<string, InterfaceBlock *>::iterator i = stage->interface_blocks.find(iface.name);
+               if(i!=stage->interface_blocks.end())
                {
-                       iface.declaration = j->second;
-                       members = &j->second->members.variables;
+                       iface.declaration = i->second;
+                       members = &i->second->members.variables;
                        break;
                }
        }
@@ -237,17 +239,14 @@ void VariableResolver::visit(Assignment &assign)
 void VariableResolver::visit(StructDeclaration &strct)
 {
        TraversingVisitor::visit(strct);
-       current_block->types[strct.name] = &strct;
+       stage->types[strct.name] = &strct;
 }
 
 void VariableResolver::visit(VariableDeclaration &var)
 {
-       for(Block *block=current_block; block; block=next_block(*block))
-       {
-               map<string, StructDeclaration *>::iterator j = block->types.find(var.type);
-               if(j!=block->types.end())
-                       var.type_declaration = j->second;
-       }
+       map<string, StructDeclaration *>::iterator i = stage->types.find(var.type);
+       if(i!=stage->types.end())
+               var.type_declaration = i->second;
 
        if(!block_interface.empty() && var.interface.empty())
                var.interface = block_interface;
@@ -260,9 +259,9 @@ void VariableResolver::visit(InterfaceBlock &iface)
 {
        /* Block names can't be used for any other identifiers so we can put them
        in the same map with instance names. */
-       current_block->interfaces[iface.name] = &iface;
+       stage->interface_blocks[iface.name] = &iface;
        if(!iface.instance_name.empty())
-               current_block->interfaces[iface.instance_name] = &iface;
+               stage->interface_blocks[iface.instance_name] = &iface;
 
        SetForScope<string> set_iface(block_interface, iface.interface);
        TraversingVisitor::visit(iface);
@@ -376,7 +375,7 @@ bool InterfaceGenerator::generate_interface(VariableDeclaration &var, const stri
 
 bool InterfaceGenerator::generate_interface(InterfaceBlock &out_block)
 {
-       if(stage->content.interfaces.count(out_block.name))
+       if(stage->interface_blocks.count(out_block.name))
                return false;
 
        InterfaceBlock *in_block = new InterfaceBlock;
@@ -398,9 +397,9 @@ bool InterfaceGenerator::generate_interface(InterfaceBlock &out_block)
        }
 
        iface_target_block->body.insert(iface_insert_point, in_block);
-       iface_target_block->interfaces[in_block->name] = in_block;
+       stage->interface_blocks[in_block->name] = in_block;
        if(!in_block->instance_name.empty())
-               stage->content.interfaces[in_block->instance_name] = in_block;
+               stage->interface_blocks[in_block->instance_name] = in_block;
 
        SetFlag set_scope(function_scope, false);
        SetForScope<Block *> set_block(current_block, &stage->content);
@@ -446,7 +445,7 @@ void InterfaceGenerator::visit(VariableReference &var)
                return;
        }
 
-       const map<string, InterfaceBlock *> &prev_blocks = stage->previous->content.interfaces;
+       const map<string, InterfaceBlock *> &prev_blocks = stage->previous->interface_blocks;
        map<string, InterfaceBlock *>::const_iterator j = prev_blocks.find(var.name);
        if(j!=prev_blocks.end() && j->second->interface=="out" && j->second->instance_name==var.name)
        {
@@ -527,7 +526,7 @@ void InterfaceGenerator::visit(InterfaceBlock &iface)
        {
                if(!iface.linked_block && stage->previous)
                {
-                       const map<string, InterfaceBlock *> &prev_blocks = stage->previous->content.interfaces;
+                       const map<string, InterfaceBlock *> &prev_blocks = stage->previous->interface_blocks;
                        map<string, InterfaceBlock *>::const_iterator i = prev_blocks.find(iface.name);
                        if(i!=prev_blocks.end() && i->second->interface=="out" && i->second->name==iface.name)
                        {