Linker : - Fixed vertex flag propagation when a shader is forced into fragment

hxsl/Linker.hx

@@ -296,21 +296,6 @@ class Linker {
 		s.deps = new Map();
 		for( r in s.readVars )
 			buildDependency(s, r, s.writeMap.exists(r.id));
-		// propagate fragment flag
-		if( s.vertex == null )
-			for( d in s.deps.keys() )
-				if( d.vertex == false ) {
-					debug(s.name + " marked as fragment because of " + d.name);
-					s.vertex = false;
-					break;
-				}
-		// propagate vertex flag
-		if( s.vertex )
-			for( d in s.deps.keys() )
-				if( d.vertex == null ) {
-					debug(d.name + " marked as vertex because of " + s.name);
-					d.vertex = true;
-				}
 	}
 
 	function collect( cur : ShaderInfos, out : Array<ShaderInfos>, vertex : Bool ) {
@@ -448,11 +433,31 @@ class Linker {
 					break;
 				}
 			if( onlyParams ) {
-				debug("Force " + s.name+" into fragment since it only reads params");
+				debug("Force " + s.name + " into fragment since it only reads params");
 				s.vertex = false;
 			}
 		}
 
+		for( s in shaders ) {
+			if ( s.deps == null)
+				continue;
+			// propagate fragment flag
+			if( s.vertex == null )
+				for( d in s.deps.keys() )
+					if( d.vertex == false ) {
+						debug(s.name + " marked as fragment because of " + d.name);
+						s.vertex = false;
+						break;
+					}
+			// propagate vertex flag
+			if( s.vertex )
+				for( d in s.deps.keys() )
+					if( d.vertex == null ) {
+						debug(d.name + " marked as vertex because of " + s.name);
+						d.vertex = true;
+					}
+		}
+
 		// collect needed dependencies
 		var v = [], f = [];
 		collect(entry, v, true);