X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fglsl%2Fcompiler.cpp;h=05bf421cd422428dbf1b09b5eef3d0daf737c3f9;hb=2b1dc627491c987b0459226dcb8ac6aadd7ee4d3;hp=4c8c1d03d893d277395b2f64b5def34aa7e55168;hpb=acd7bb6e23e2aff9934ecf32852c62ba72c13574;p=libs%2Fgl.git

diff --git a/source/glsl/compiler.cpp b/source/glsl/compiler.cpp
index 4c8c1d03..05bf421c 100644
--- a/source/glsl/compiler.cpp
+++ b/source/glsl/compiler.cpp
@@ -107,6 +107,19 @@ void Compiler::compile(Mode mode)
 			++i;
 	}
 
+	Stage *prev_stage = 0;
+	for(auto i=module->stages.begin(); i!=module->stages.end(); )
+	{
+		if(i->functions.empty())
+			i = module->stages.erase(i);
+		else
+		{
+			i->previous = prev_stage;
+			prev_stage = &*i;
+			++i;
+		}
+	}
+
 	for(Stage &s: module->stages)
 	{
 		StructuralFeatureConverter().apply(s, features);