Most simple methods reimplemented in HeatVisitor

1 files changed, 37 insertions, 20 deletions

diff --git a/heat/HeatVisitor.java b/heat/HeatVisitor.java
index fdc71ad..ba739c9 100644
--- a/heat/HeatVisitor.java
+++ b/heat/HeatVisitor.java
@@ -113,7 +113,6 @@ public class HeatVisitor extends GJDepthFirst<TypeBundle,String> {
         this.symt.setActive(TypeEnum.classname, symt.getClass(id));
         this.symt.setActive(TypeEnum.method, symt.getMethod(n.f6.tokenImage));
         ///////////////////////////////////////////////////////////////
-        n.f1.accept(this, argu);
         n.f14.accept(this, argu);
         n.f15.accept(this, argu);
         ///////////////////////////////////////////////////////////////
@@ -158,7 +157,6 @@ public class HeatVisitor extends GJDepthFirst<TypeBundle,String> {
                                          id));
         this.symt.setActive(TypeEnum.classname, symt.getClass(id));
         ///////////////////////////////////////////////////////////////
-        n.f1.accept(this, argu);
         n.f3.accept(this, argu);
         n.f4.accept(this, argu);
         ///////////////////////////////////////////////////////////////
@@ -182,13 +180,12 @@ public class HeatVisitor extends GJDepthFirst<TypeBundle,String> {
     public TypeBundle visit(ClassExtendsDeclaration n, String argu) {
         TypeBundle _ret=null;
         String id = n.f1.f0.tokenImage;
+        String id0 = n.f3.f0.tokenImage;
         MinimalLogger.info(String.format("-> %s (%s)",
                                          n.getClass().getSimpleName(),
                                          id));
         this.symt.setActive(TypeEnum.classname, symt.getClass(id));
         ///////////////////////////////////////////////////////////////
-        n.f1.accept(this, argu);
-        n.f3.accept(this, argu);
         n.f5.accept(this, argu);
         n.f6.accept(this, argu);
         ///////////////////////////////////////////////////////////////
@@ -211,8 +208,7 @@ public class HeatVisitor extends GJDepthFirst<TypeBundle,String> {
                                          n.getClass().getSimpleName(),
                                          id));
         ///////////////////////////////////////////////////////////////
-        n.f0.accept(this, argu);
-        n.f1.accept(this, argu);
+        _ret = n.f0.accept(this, argu);
         ///////////////////////////////////////////////////////////////
         MinimalLogger.info(String.format("<- %s (%s)",
                                          n.getClass().getSimpleName(),
@@ -244,7 +240,6 @@ public class HeatVisitor extends GJDepthFirst<TypeBundle,String> {
         this.symt.setActive(TypeEnum.method, symt.getMethod(id));
         ///////////////////////////////////////////////////////////////
         n.f1.accept(this, argu);
-        n.f2.accept(this, argu);
         n.f4.accept(this, argu);
         n.f7.accept(this, argu);
         n.f8.accept(this, argu);
@@ -266,6 +261,7 @@ public class HeatVisitor extends GJDepthFirst<TypeBundle,String> {
         MinimalLogger.info(String.format("-> %s",
                                          n.getClass().getSimpleName()));
         ///////////////////////////////////////////////////////////////
+        // FIXME Check distinct!
         n.f0.accept(this, argu);
         n.f1.accept(this, argu);
         ///////////////////////////////////////////////////////////////
@@ -286,8 +282,7 @@ public class HeatVisitor extends GJDepthFirst<TypeBundle,String> {
                                          n.getClass().getSimpleName(),
                                          id));
         ///////////////////////////////////////////////////////////////
-        n.f0.accept(this, argu);
-        n.f1.accept(this, argu);
+        _ret = n.f0.accept(this, argu);
         ///////////////////////////////////////////////////////////////
         MinimalLogger.info(String.format("<- %s (%s)",
                                          n.getClass().getSimpleName(),
@@ -305,7 +300,7 @@ public class HeatVisitor extends GJDepthFirst<TypeBundle,String> {
         MinimalLogger.info(String.format("-> %s",
                                          n.getClass().getSimpleName()));
         ///////////////////////////////////////////////////////////////
-        n.f1.accept(this, argu);
+        _ret = n.f1.accept(this, argu);
         ///////////////////////////////////////////////////////////////
         MinimalLogger.info(String.format("<- %s with %s",
                                          n.getClass().getSimpleName(),
@@ -324,7 +319,7 @@ public class HeatVisitor extends GJDepthFirst<TypeBundle,String> {
         MinimalLogger.info(String.format("-> %s",
                                          n.getClass().getSimpleName()));
         ///////////////////////////////////////////////////////////////
-        n.f0.accept(this, argu);
+        _ret = n.f0.accept(this, argu);
         ///////////////////////////////////////////////////////////////
         MinimalLogger.info(String.format("<- %s with %s",
                                          n.getClass().getSimpleName(),
@@ -342,6 +337,7 @@ public class HeatVisitor extends GJDepthFirst<TypeBundle,String> {
         MinimalLogger.info(String.format("-> %s",
                                          n.getClass().getSimpleName()));
         ///////////////////////////////////////////////////////////////
+        _ret = new TypeBundle(TypeEnum.intarray, null);
         ///////////////////////////////////////////////////////////////
         MinimalLogger.info(String.format("<- %s with %s",
                                          n.getClass().getSimpleName(),
@@ -357,6 +353,7 @@ public class HeatVisitor extends GJDepthFirst<TypeBundle,String> {
         MinimalLogger.info(String.format("-> %s",
                                          n.getClass().getSimpleName()));
         ///////////////////////////////////////////////////////////////
+        _ret = new TypeBundle(TypeEnum.bool, null);
         ///////////////////////////////////////////////////////////////
         MinimalLogger.info(String.format("<- %s with %s",
                                          n.getClass().getSimpleName(),
@@ -372,6 +369,7 @@ public class HeatVisitor extends GJDepthFirst<TypeBundle,String> {
         MinimalLogger.info(String.format("-> %s",
                                          n.getClass().getSimpleName()));
         ///////////////////////////////////////////////////////////////
+        _ret = new TypeBundle(TypeEnum.integer, null);
         ///////////////////////////////////////////////////////////////
         MinimalLogger.info(String.format("<- %s with %s",
                                          n.getClass().getSimpleName(),
@@ -392,7 +390,7 @@ public class HeatVisitor extends GJDepthFirst<TypeBundle,String> {
         MinimalLogger.info(String.format("-> %s",
                                          n.getClass().getSimpleName()));
         ///////////////////////////////////////////////////////////////
-        n.f0.accept(this, argu);
+        _ret = n.f0.accept(this, argu);
         ///////////////////////////////////////////////////////////////
         MinimalLogger.info(String.format("<- %s with %s",
                                          n.getClass().getSimpleName(),
@@ -410,7 +408,7 @@ public class HeatVisitor extends GJDepthFirst<TypeBundle,String> {
         MinimalLogger.info(String.format("-> %s",
                                          n.getClass().getSimpleName()));
         ///////////////////////////////////////////////////////////////
-        n.f1.accept(this, argu);
+        _ret = n.f1.accept(this, argu);
         ///////////////////////////////////////////////////////////////
         MinimalLogger.info(String.format("<- %s with %s",
                                          n.getClass().getSimpleName(),
@@ -430,8 +428,16 @@ public class HeatVisitor extends GJDepthFirst<TypeBundle,String> {
         MinimalLogger.info(String.format("-> %s",
                                          n.getClass().getSimpleName()));
         ///////////////////////////////////////////////////////////////
-        n.f0.accept(this, argu);
-        n.f2.accept(this, argu);
+        TypeBundle tb = new TypeBundle(this.symt.getType(id).getType(), null);
+        if (tb.getType() == null)
+            throw new TypecheckException(String.format("%s glowers at symbol table!",
+                                                       n.getClass().getSimpleName()));
+
+        // Fixme---add subtyping?
+        _ret = n.f2.accept(this, argu);
+        if (!tb.equals(_ret))
+            throw new TypecheckException(String.format("%s tried to assign unequal types!",
+                                                       n.getClass().getSimpleName()));
         ///////////////////////////////////////////////////////////////
         MinimalLogger.info(String.format("<- %s with %s",
                                          n.getClass().getSimpleName(),
@@ -450,12 +456,20 @@ public class HeatVisitor extends GJDepthFirst<TypeBundle,String> {
      */
     public TypeBundle visit(ArrayAssignmentStatement n, String argu) {
         TypeBundle _ret=null;
+        String id = n.f0.f0.tokenImage;
         MinimalLogger.info(String.format("-> %s",
                                          n.getClass().getSimpleName()));
         ///////////////////////////////////////////////////////////////
-        n.f0.accept(this, argu);
-        n.f2.accept(this, argu);
-        n.f5.accept(this, argu);
+        TypeInstance t = this.symt.getType(id);
+        if (t == null || t.getType() != TypeEnum.intarray)
+            throw new TypecheckException(String.format("%s called on non-array!",
+                                                       n.getClass().getSimpleName()));
+
+        TypeBundle tb = n.f2.accept(this, argu);
+        _ret = n.f5.accept(this, argu);
+        if (!tb.isInt() || _ret.isInt())
+            throw new TypecheckException(String.format("%s tried with something other than an integer!",
+                                                       n.getClass().getSimpleName()));
         ///////////////////////////////////////////////////////////////
         MinimalLogger.info(String.format("<- %s with %s",
                                          n.getClass().getSimpleName(),
@@ -477,7 +491,10 @@ public class HeatVisitor extends GJDepthFirst<TypeBundle,String> {
         MinimalLogger.info(String.format("-> %s",
                                          n.getClass().getSimpleName()));
         ///////////////////////////////////////////////////////////////
-        n.f2.accept(this, argu);
+        _ret = n.f2.accept(this, argu);
+        if (!_ret.isBool())
+            throw new TypecheckException(String.format("%s called on non-bool conditional!",
+                                                       n.getClass().getSimpleName()));
         n.f4.accept(this, argu);
         n.f6.accept(this, argu);
         ///////////////////////////////////////////////////////////////
@@ -754,7 +771,7 @@ public class HeatVisitor extends GJDepthFirst<TypeBundle,String> {
         }
 
         if (t == null || t.getClassInstance() == null)
-            throw new TypecheckException(String.format("%s called on a missing/non class variable!",
+            throw new TypecheckException(String.format("%s called on a non class variable!",
                                                        n.getClass().getSimpleName()));
 
         MethodInstance m = this.symt.getMethod(n.f2.f0.tokenImage, t.getClassInstance());