Allow MethodInstance to keep track of parent class

1 files changed, 9 insertions, 0 deletions

diff --git a/st/AbstractInstance.java b/st/AbstractInstance.java
index d236044..7214e6a 100644
--- a/st/AbstractInstance.java
+++ b/st/AbstractInstance.java
@@ -6,6 +6,7 @@ public abstract class AbstractInstance {
     protected String name;                              // the literal name of the declaration
     protected TypeEnum type;                            // the type of the declaration
     protected ArrayList<AbstractInstance> scope;        // the scope where the instance is valid
+    protected ClassInstance cls;                        // the surrounding class
 
     public AbstractInstance(String name, TypeEnum type) {
         this.type = type;
@@ -41,6 +42,10 @@ public abstract class AbstractInstance {
         }
     }
 
+    protected void addClassInstance(ClassInstance cls) {
+        this.cls = cls;
+    }
+
     public String getName() {
         return this.name;
     }
@@ -53,4 +58,8 @@ public abstract class AbstractInstance {
         return this.scope;
     }
 
+    public ClassInstance getClassInstance() {
+        return this.cls;
+    }
+
 }