1 files changed, 8 insertions, 0 deletions

diff --git a/src/main/java/cuchaz/enigma/analysis/SourceIndex.java b/src/main/java/cuchaz/enigma/analysis/SourceIndex.java
index 73e0431..719930e 100644
--- a/src/main/java/cuchaz/enigma/analysis/SourceIndex.java
+++ b/src/main/java/cuchaz/enigma/analysis/SourceIndex.java
@@ -145,6 +145,14 @@ public class SourceIndex {
         return this.tokenToReference.keySet();
     }
 
+    public Iterable<Token> declarationTokens() {
+        return this.declarationToToken.values();
+    }
+
+    public Iterable<Entry> declarations() {
+        return this.declarationToToken.keySet();
+    }
+
     public Token getDeclarationToken(Entry deobfEntry) {
         return this.declarationToToken.get(deobfEntry);
     }