fix(shader-ast): allChildren() (while loop support)

thi-ng · Jun 24, 2019 · 3a559cf · 3a559cf
1 parent bb424b0
commit 3a559cf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/packages/shader-ast/src/ast.ts b/packages/shader-ast/src/ast.ts
@@ -150,7 +150,7 @@ export const scopeChildren = (t: Term<any>) =>
 export const allChildren = (t: Term<any>) =>
     t.tag === "scope"
         ? (<Scope>t).body
-        : t.tag === "fn" || t.tag === "for"
+        : t.tag === "fn" || t.tag === "for" || t.tag == "while"
         ? (<Func<any>>t).scope.body
         : t.tag === "if"
         ? (<Branch>t).f