X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fglsl%2Fgenerate.cpp;h=8626cac6761c6f42f896b76ef47d53111df5c460;hb=44792dd6fb56a709d9a1d33d7fccc2ab8de905d2;hp=ad8515a94392568d4823d84bc3595c44dc9945ee;hpb=d72d8a9163e342167f35360c807ee9ea5ebacdc2;p=libs%2Fgl.git

diff --git a/source/glsl/generate.cpp b/source/glsl/generate.cpp
index ad8515a9..8626cac6 100644
--- a/source/glsl/generate.cpp
+++ b/source/glsl/generate.cpp
@@ -114,6 +114,55 @@ void BlockHierarchyResolver::enter(Block &block)
 }
 
 
+TypeResolver::TypeResolver():
+	stage(0)
+{ }
+
+void TypeResolver::apply(Stage &s)
+{
+	stage = &s;
+	s.types.clear();
+	s.content.visit(*this);
+}
+
+void TypeResolver::visit(BasicTypeDeclaration &type)
+{
+	map<string, TypeDeclaration *>::iterator i = stage->types.find(type.base);
+	type.base_type = (i!=stage->types.end() ? i->second : 0);
+
+	if(type.kind==BasicTypeDeclaration::VECTOR && type.base_type)
+		if(BasicTypeDeclaration *basic_base = dynamic_cast<BasicTypeDeclaration *>(type.base_type))
+			if(basic_base->kind==BasicTypeDeclaration::VECTOR)
+			{
+				type.kind = BasicTypeDeclaration::MATRIX;
+				type.size |= basic_base->size<<16;
+			}
+
+	stage->types.insert(make_pair(type.name, &type));
+}
+
+void TypeResolver::visit(ImageTypeDeclaration &type)
+{
+	map<string, TypeDeclaration *>::iterator i = stage->types.find(type.base);
+	type.base_type = (i!=stage->types.end() ? i->second : 0);
+
+	stage->types.insert(make_pair(type.name, &type));
+}
+
+void TypeResolver::visit(StructDeclaration &strct)
+{
+	stage->types.insert(make_pair(strct.name, &strct));
+	TraversingVisitor::visit(strct);
+}
+
+void TypeResolver::visit(VariableDeclaration &var)
+{
+	map<string, TypeDeclaration *>::iterator i = stage->types.find(var.type);
+	if(i!=stage->types.end())
+		var.type_declaration = i->second;
+}
+
+
 VariableResolver::VariableResolver():
 	stage(0),
 	r_members(0),
@@ -125,7 +174,6 @@ VariableResolver::VariableResolver():
 void VariableResolver::apply(Stage &s)
 {
 	stage = &s;
-	s.types.clear();
 	s.interface_blocks.clear();
 	s.content.visit(*this);
 }
@@ -149,12 +197,7 @@ void VariableResolver::visit(VariableReference &var)
 			var.declaration = i->second;
 	}
 
-	if(var.declaration)
-	{
-		if(var.declaration->type_declaration)
-			r_members = &var.declaration->type_declaration->members.variables;
-	}
-	else
+	if(!var.declaration)
 	{
 		const map<string, InterfaceBlock *> &blocks = stage->interface_blocks;
 		map<string, InterfaceBlock *>::const_iterator i = blocks.find("_"+var.name);
@@ -163,6 +206,8 @@ void VariableResolver::visit(VariableReference &var)
 			/* The name refers to an interface block with an instance name rather
 			than a variable.  Prepare a new syntax tree node accordingly. */
 			r_iface_ref = new InterfaceBlockReference;
+			r_iface_ref->source = var.source;
+			r_iface_ref->line = var.line;
 			r_iface_ref->name = var.name;
 			r_iface_ref->declaration = i->second;
 			r_members = &i->second->members.variables;
@@ -180,6 +225,10 @@ void VariableResolver::visit(VariableReference &var)
 		}
 	}
 
+	if(var.declaration)
+		if(StructDeclaration *strct = dynamic_cast<StructDeclaration *>(var.declaration->type_declaration))
+			r_members = &strct->members.variables;
+
 	if(record_target)
 	{
 		if(r_assignment_target)
@@ -228,8 +277,8 @@ void VariableResolver::visit(MemberAccess &memacc)
 		if(i!=r_members->end())
 		{
 			memacc.declaration = i->second;
-			if(i->second->type_declaration)
-				r_members = &i->second->type_declaration->members.variables;
+			if(StructDeclaration *strct = dynamic_cast<StructDeclaration *>(i->second->type_declaration))
+				r_members = &strct->members.variables;
 		}
 		else
 			r_members = 0;
@@ -292,18 +341,8 @@ void VariableResolver::visit(FunctionCall &call)
 	r_iface_ref = 0;
 }
 
-void VariableResolver::visit(StructDeclaration &strct)
-{
-	TraversingVisitor::visit(strct);
-	stage->types.insert(make_pair(strct.name, &strct));
-}
-
 void VariableResolver::visit(VariableDeclaration &var)
 {
-	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;
 
@@ -558,7 +597,10 @@ void InterfaceGenerator::visit(VariableDeclaration &var)
 			const map<string, VariableDeclaration *> &linked_vars = iface_block->linked_block->members.variables;
 			map<string, VariableDeclaration *>::const_iterator i = linked_vars.find(var.name);
 			if(i!=linked_vars.end())
+			{
 				var.linked_declaration = i->second;
+				var.linked_declaration->linked_declaration = &var;
+			}
 		}
 		return;
 	}