diff --git a/src/main/java/org/taumc/glsl/Transformer.java b/src/main/java/org/taumc/glsl/Transformer.java
index 301d3d7..232e3db 100644
--- 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;
+    }
+
 }