Assign Stmt

.gitignore

@@ -147,3 +147,5 @@ tramp
 Session.vim
 .netrwhist
 *~
+
+*.gi

src/main/java/chocopy/pa3/CodeGenImpl.java

@@ -9,6 +9,7 @@ import java.util.Stack;
 import chocopy.common.analysis.SymbolTable;
 import chocopy.common.analysis.AbstractNodeAnalyzer;
 import chocopy.common.analysis.types.SymbolType;
+import chocopy.common.analysis.types.ValueType;
 import chocopy.common.astnodes.*;
 import chocopy.common.codegen.*;
 
@@ -220,7 +221,7 @@ public class CodeGenImpl extends CodeGenBase {
     }
 
     /** An analyzer that encapsulates code generation for statments. */
-    private class StmtAnalyzer extends AbstractNodeAnalyzer<Object> {
+    private class StmtAnalyzer extends AbstractNodeAnalyzer<Void> {
         /*
          * The symbol table has all the info you need to determine
          * what a given identifier 'x' in the current scope is. You can
@@ -302,7 +303,6 @@ public class CodeGenImpl extends CodeGenBase {
             stmt.value.dispatch(this);
             backend.emitLW(A0, SP, 0, "Load return value into A0");
             return null;
-
         }
 
         @Override
@@ -311,6 +311,51 @@ public class CodeGenImpl extends CodeGenBase {
             return null;
         }
 
+        @Override
+        public Void analyze(AssignStmt node) {
+            node.value.dispatch(this);
+            RiscVBackend.Register tmpReg = /*regMgr.borrowOneTmp()*/ T6;
+            RiscVBackend.Register tmpReg2 = /*regMgr.borrowOneTmp()*/ T5;
+            betterBackend.emitPop(tmpReg, "ASSIGN: Get the value to assign: Pop the expr result from stack");
+
+            SymbolType symbolType = null;
+            for(Expr expr : node.targets) {
+                if(! (expr instanceof Identifier)) {
+                    throw new RuntimeException("assignment target should be Identifier.");
+                }
+                SymbolInfo symbolInfo = sym.get(((Identifier) expr).name);
+                Label label = null;
+                if(symbolType == null) {
+                    // first run:
+                    if (symbolInfo instanceof GlobalVarInfo) {
+                        GlobalVarInfo globalVarInfo = (GlobalVarInfo) symbolInfo;
+                        symbolType = globalVarInfo.getVarType();
+                    } else if (symbolInfo instanceof StackVarInfo) {
+                        StackVarInfo stackVarInfo = (StackVarInfo) symbolInfo;
+                        symbolType = stackVarInfo.getVarType();
+                    }
+                    if (symbolType.equals(SymbolType.INT_TYPE) || symbolType.equals(SymbolType.BOOL_TYPE)) {
+                        // first run:
+                        /*
+                        backend.emitLW(tmpReg, tmpReg, 3 * backend.getWordSize(), "ASSIGN: Load the value of INT or BOOL");
+                        */
+                    }
+                }
+
+                // do assign
+                if (symbolInfo instanceof GlobalVarInfo) {
+                    GlobalVarInfo globalVarInfo = (GlobalVarInfo) symbolInfo;
+                    symbolType = globalVarInfo.getVarType();
+                    backend.emitSW(tmpReg, globalVarInfo.getLabel(), tmpReg2, "ASSIGN: Assign to global var");
+                } else if (symbolInfo instanceof StackVarInfo) {
+                    StackVarInfo stackVarInfo = (StackVarInfo) symbolInfo;
+                    symbolType = stackVarInfo.getVarType();
+                    throw new RuntimeException("ASSIGN to stack variable: NOT IMPLEMENTED: StackVarInfo not implemented.");
+                }
+            }
+            return null;
+        }
+
 
         @Override
         public Void analyze(CallExpr callExpr) {
@@ -485,10 +530,6 @@ public class CodeGenImpl extends CodeGenBase {
         }
 
 
-        @Override
-        public Void analyze(AssignStmt assignStmt) {
-            return null;
-        }
         // FIXME: More, of course.
 
     }