add storage collector

TauMC · Oct 28, 2024 · 7ed010b · 7ed010b
1 parent 85fbb44
commit 7ed010b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/main/java/org/taumc/glsl/Transformer.java b/src/main/java/org/taumc/glsl/Transformer.java
@@ -221,4 +221,10 @@ public void rewriteStructArrays() {
         ParseTreeWalker.DEFAULT.walk(new StructArrayRewriter(), root);
     }
 
+    public List<TerminalNode> collectStorage() {
+        List<TerminalNode> tokens = new ArrayList<>();
+        ParseTreeWalker.DEFAULT.walk(new StorageCollector(tokens), root);
+        return tokens;
+    }
+
 }