X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fglsl%2Fgenerate.cpp;h=433d506e1a8ab762f50a0e930e314bf387c58be0;hb=d80750e7c20ea061f210b756196cc844b762b852;hp=8626cac6761c6f42f896b76ef47d53111df5c460;hpb=7c7a32e0de7fd8c16c02190a7483a0c2411973c1;p=libs%2Fgl.git

diff --git a/source/glsl/generate.cpp b/source/glsl/generate.cpp
index 8626cac6..433d506e 100644
--- a/source/glsl/generate.cpp
+++ b/source/glsl/generate.cpp
@@ -162,6 +162,13 @@ void TypeResolver::visit(VariableDeclaration &var)
 		var.type_declaration = i->second;
 }
 
+void TypeResolver::visit(FunctionDeclaration &func)
+{
+	map<string, TypeDeclaration *>::iterator i = stage->types.find(func.return_type);
+	func.return_type_declaration = (i!=stage->types.end() ? i->second : 0);
+	TraversingVisitor::visit(func);
+}
+
 
 VariableResolver::VariableResolver():
 	stage(0),