fix(shader-ast): fix if() handling in scopedChildren()

- include test in returned children otherwise any referenced functions there will not be captured in dependencies
thi-ng · Oct 24, 2023 · 64d23c0 · 64d23c0
1 parent dc7eeab
commit 64d23c0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/packages/shader-ast/src/ast/scope.ts b/packages/shader-ast/src/ast/scope.ts
@@ -30,8 +30,8 @@ export const scopedChildren = (t: Term<any>) =>
 		? (<Func<any>>t).scope.body
 		: t.tag === "if"
 		? (<Branch>t).f
-			? (<Branch>t).t.body.concat((<Branch>t).f!.body)
-			: (<Branch>t).t.body
+			? [(<Branch>t).test, ...(<Branch>t).t.body, ...(<Branch>t).f!.body]
+			: [(<Branch>t).test, ...(<Branch>t).t.body]
 		: undefined;
 
 /**