From d7ae4fc9edecd2f23e90661c281c7f466a3583c8 Mon Sep 17 00:00:00 2001
From: bill <b@i.ll>
Date: Tue, 4 May 2021 22:16:58 +0800
Subject: [PATCH] 	modified:   src/main/java/chocopy/pa3/CodeGenImpl.java

---
 src/main/java/chocopy/pa3/CodeGenImpl.java | 191 +++++++++++++++------
 1 file changed, 137 insertions(+), 54 deletions(-)

diff --git a/src/main/java/chocopy/pa3/CodeGenImpl.java b/src/main/java/chocopy/pa3/CodeGenImpl.java
index 7376ce0..1cb954f 100644
--- a/src/main/java/chocopy/pa3/CodeGenImpl.java
+++ b/src/main/java/chocopy/pa3/CodeGenImpl.java
@@ -4,6 +4,7 @@ import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
+import java.util.Stack;
 
 import chocopy.common.analysis.*;
 import chocopy.common.astnodes.*;
@@ -90,6 +91,8 @@ public class CodeGenImpl extends CodeGenBase
         {
             stmt.dispatch(stmtAnalyzer);
         }
+        stmtAnalyzer.emitSizeLabel();
+
         backend.emitLI(A0, EXIT_ECALL, "Code for ecall: exit");
         backend.emitEcall(null);
     }
@@ -155,10 +158,8 @@ public class CodeGenImpl extends CodeGenBase
         for (Stmt stmt : funcInfo.getStatements()) {
             stmt.dispatch(stmtAnalyzer);
         }
+        stmtAnalyzer.emitSizeLabel();
         backend.emitJ(stmtAnalyzer.epilogue, "Jump to function epilogue");
-
-        
-        
         // --- Epilogue ---
         backend.emitLocalLabel(stmtAnalyzer.epilogue, "Epilogue");
         backend.emitLW(RA, FP, -4, "Get return address");
@@ -206,7 +207,7 @@ public class CodeGenImpl extends CodeGenBase
         private final FuncInfo funcInfo;
 
         /** Label of code that exits from block. */
-        protected Label elseBlock;
+        protected Stack<Label> elseBlock = new Stack<Label>();
 
         /** Variable to keep track of slots of stored variables on stack */
         // example usage: backend.emitLW(A0, SP, varSlotMap.get(varName)) to load the variable using its name
@@ -254,7 +255,9 @@ public class CodeGenImpl extends CodeGenBase
             max_sp = max_sp >= sp_off?max_sp:sp_off;
         }
         
-        
+        public void emitSizeLabel(){
+            backend.defineSym(size_label, max_sp*wordSize);
+        }
         
         
         // *********** functions start ***********
@@ -263,8 +266,7 @@ public class CodeGenImpl extends CodeGenBase
         	    System.out.println("Inside CallExpr: " + node.function.name);
         		backend.emitLW(T6, FP, 0, "Inside CallExpr: " + node.function.name);
         		
-        		// FIXME: Why are we using globalSymbols here but not the sym???
-            SymbolInfo Ty = globalSymbols.get(node.function.name);
+            SymbolInfo Ty = sym.get(node.function.name);
             if(Ty instanceof ClassInfo){
                 //object create
                 ClassInfo cls = (ClassInfo) Ty;
@@ -301,6 +303,18 @@ public class CodeGenImpl extends CodeGenBase
 	        		
 	        		List<Expr> args = node.args;
 	        		int spaceRequiredForArgs = (args.size() + 1)*4;
+                    int depth = calleeFunctionInfo.getDepth();
+                    int tmpHandle = getRegister();
+                    Register tmp = registerPool[tmpHandle];
+                    FuncInfo currFuncInfo = calleeFunctionInfo;
+                    backend.emitMV(tmp, FP, "Load FP");
+                    while (depth > 0) {
+                        final FuncInfo parentFuncInfo = currFuncInfo.getParentFuncInfo();
+                        backend.emitLW(tmp, tmp, parentFuncInfo.getParams().size()*wordSize, "Load static link to " + parentFuncInfo.getFuncName());
+                        currFuncInfo = parentFuncInfo;
+                        --depth;
+                    }
+                    freeRegister(tmpHandle);
 	        		//backend.emitSW(A0, SP, -4, "Put static link");
 	        		for (int i = 0; i < args.size(); i++) {
 	        			int argNum = i + 1;
@@ -312,13 +326,13 @@ public class CodeGenImpl extends CodeGenBase
 	        			StackVarInfo formalParamInfo = (StackVarInfo) calleeFunctionInfo.getSymbolTable().get(formalParamName);
 	        			
 	        			if (expr.getInferredType().equals(Type.INT_TYPE)) {
-	        				if (formalParamInfo.getVarType().equals(Type.OBJECT_TYPE) || formalParamInfo.getVarType().equals(Type.INT_TYPE)) {
+	        				if ((functionId.name.equals("print")) &&(formalParamInfo.getVarType().equals(Type.OBJECT_TYPE) || formalParamInfo.getVarType().equals(Type.INT_TYPE))) {
 	                        backend.emitJAL(makeintLabel, "Box integer");
 	        				} else  {
 	        					// FIXME: passed argument does not match formal parameter
 	        				}
 	                } else if (expr.getInferredType().equals(Type.BOOL_TYPE)) { 
-	                    	if (formalParamInfo.getVarType().equals(Type.OBJECT_TYPE) || formalParamInfo.getVarType().equals(Type.BOOL_TYPE)) {
+	                    	if ((functionId.name.equals("print"))&&(formalParamInfo.getVarType().equals(Type.OBJECT_TYPE) || formalParamInfo.getVarType().equals(Type.BOOL_TYPE))) {
 	                    		backend.emitJAL(makeboolLabel, "Box boolean");
 	                    } else {
 	                    		// FIXME: passed argument does not match formal parameter
@@ -331,7 +345,8 @@ public class CodeGenImpl extends CodeGenBase
 	        		}
 	        		
 	        		backend.emitADDI(SP, SP, -spaceRequiredForArgs, "Set SP to last argument.");
-	        		backend.emitJAL(new Label("$"+functionId.name), "Invoke function: " + functionId.name);
+                    String label_str = "$"+calleeFunctionInfo.getFuncName();
+                    backend.emitJAL(new Label(label_str), "Invoke function: " + label_str);
 	        		backend.emitADDI(SP, SP, spaceRequiredForArgs, "Set SP to stack frame top.");
             }
             return A0;
@@ -355,8 +370,11 @@ public class CodeGenImpl extends CodeGenBase
                     String.format("Push argument %d from last.", n_args - i - 1));
             
             backend.emitLW(A0, FP, (n_args- sp_off) * wordSize, String.format("Peek stack slot %d", sp_off - (n_args + 1)));
-            ClassInfo objectClass = (ClassInfo)sym.get(((Identifier)node.method.object).name);
-
+            ClassInfo objectClass = null;
+            if(node.method.object instanceof CallExpr)
+                objectClass = (ClassInfo)globalSymbols.get(((CallExpr)node.method.object).getInferredType().className());
+            else 
+                objectClass = (ClassInfo)globalSymbols.get(node.method.object.getInferredType().className());
             backend.emitLW(A1, A0, getDispatchTableOffset(), "Load address of object's dispatch table");
             backend.emitLW(A1, A1, getMethodOffset(objectClass, node.method.member.name), 
                 String.format("Load address of method: %s.%s", objectClass.getClassName(), node.method.member.name));
@@ -419,7 +437,54 @@ public class CodeGenImpl extends CodeGenBase
                 backend.emitLI(Register.A0, 0, "Load boolean literal: false ");
             return Register.A0;
         }
-
+        class StackVarRuntimeInfo {
+            public final int off;
+            public final Register sl;
+            public int regHandle;
+            StackVarRuntimeInfo(Register sl, int off){
+                this.off = off;
+                this.sl = sl;
+                regHandle = -1;
+            }
+            StackVarRuntimeInfo(Register sl, int off, int regHandle){
+                this.off = off;
+                this.sl = sl;
+                this.regHandle = regHandle;
+            }
+            public void free(){
+                if(regHandle > 0){
+                    freeRegister(regHandle);
+                    regHandle = -1;
+                }
+            }
+        }
+        private Object getVar(Identifier id){
+            SymbolInfo info = sym.get(id.name);
+            if(info instanceof StackVarInfo)
+                return getStackVar((StackVarInfo)info);
+            else
+                return ((GlobalVarInfo)info).getLabel();
+        }
+        private StackVarRuntimeInfo getStackVar(StackVarInfo info){
+            int curr_depth = funcInfo.getDepth();
+            int d_depth = curr_depth - info.getFuncInfo().getDepth();
+            if(d_depth == 0)
+                return new StackVarRuntimeInfo(SP, varSlotMap.get(info.getVarName()));
+            else
+            {
+                FuncInfo curr = funcInfo;
+                int tmpHandle = getRegister();
+                Register tmp = registerPool[tmpHandle];
+                backend.emitMV(tmp, FP, "tmp = FP");
+                while(d_depth > 0){
+                    curr = curr.getParentFuncInfo();
+                    backend.emitLW(tmp, tmp, curr.getParams().size()*wordSize, "Get static link");
+                    -- d_depth;
+                }
+                // UNSAFE!!!!! PLEASE FREE THE TMP_HANDLE MANUALLY
+                return new StackVarRuntimeInfo(tmp, -wordSize*curr.getVarIndex(info.getVarName()), tmpHandle);
+            }
+        }
         
         @Override
         public Register analyze(AssignStmt node) 
@@ -432,27 +497,24 @@ public class CodeGenImpl extends CodeGenBase
                 Register reg = node.value.dispatch(this);
                 if (reg == null)
                     reg = A0;
-                if (sym.getParent() == null) 
+                for(Expr target: node.targets)
                 {
-                    for(Expr target: node.targets)
+                    if(target instanceof Identifier)
                     {
-                        if(target instanceof Identifier)
-                        {
-                            GlobalVarInfo gvi=(GlobalVarInfo)sym.get(((Identifier)target).name);
-                            backend.emitSW(reg, gvi.getLabel(), Register.T0, "Assign global: "+gvi.getVarName()+"(using tmp register)");
-                        }
+                        Identifier targetID = (Identifier)target;
+                        Object var = getVar(targetID);
+                        if(var instanceof StackVarRuntimeInfo){
+                            StackVarRuntimeInfo rtinfo = (StackVarRuntimeInfo) var;
+                            backend.emitSW(reg, rtinfo.sl, rtinfo.off ,"Store local variable: "+((Identifier)target).name);
+                            rtinfo.free();
+                        } 
                         else
-                        {
-                            Register ret = target.dispatch(this);
-                            backend.emitSW(T1, ret, 0, "Set list element"); 
-                        }
+                            backend.emitSW(reg, (Label)var, T0, "Store Global variable: " + ((Identifier)targetID).name);
                     }
-                }
-                else
-                {
-                    for(Expr target: node.targets)
+                    else
                     {
-                        backend.emitLW(reg, SP, varSlotMap.get(((Identifier)target).name),"Load local variable: "+((Identifier)target).name);
+                        Register ret = target.dispatch(this);
+                        backend.emitSW(T1, ret, 0, "Set list element"); 
                     }
                 }
             //}
@@ -478,8 +540,10 @@ public class CodeGenImpl extends CodeGenBase
             backend.emitBEQZ(result, ln,"Jump to end of loop");
             node.thenExpr.dispatch(this);
             backend.emitJ(ln, "Jump to end of if expression");
-            backend.emitLocalLabel(elseBlock, "Else part of if expression");
+            elseBlock.push(generateLocalLabel());
+            backend.emitLocalLabel(elseBlock.peek(), "Else part of if expression");
             node.elseExpr.dispatch(this);
+            elseBlock.pop();
             backend.emitLocalLabel(ln, "End of if expression");
             return A0;
         }
@@ -495,9 +559,11 @@ public class CodeGenImpl extends CodeGenBase
             for(Stmt s:node.thenBody)
                 s.dispatch(this);
             backend.emitJ(ln, "Jump to end of if statement");
-            backend.emitLocalLabel(elseBlock, "Else part of if statement");
+            elseBlock.push(generateLocalLabel());
+            backend.emitLocalLabel(elseBlock.peek(), "Else part of if statement");
             for(Stmt s:node.elseBody)
                 s.dispatch(this);
+            elseBlock.pop();
             backend.emitLocalLabel(ln, "End of if statement");
             return null;
         }
@@ -782,29 +848,37 @@ public class CodeGenImpl extends CodeGenBase
                 backend.emitJAL(errorNI, "Operator not implemented");
             return Register.A0;
         }
-        
+        private SymbolInfo getSymbolInfo(Identifier id){
+            SymbolTable curr_sym = sym;
+            SymbolInfo info = null;
+            String id_name = id.name;
+            while(curr_sym != null){
+                info = (SymbolInfo) curr_sym.get(id_name);
+                if(info != null)
+                    return info;
+                curr_sym = curr_sym.getParent();
+            }
+            return info;
+        }
         @Override
         public Register analyze(Identifier node) 
         {
         	backend.emitLW(T6, FP, 0, "Inside Identifier: ");
-            if (sym.getParent() == null) 
-            {
-                GlobalVarInfo gvi=(GlobalVarInfo) sym.get(node.name);
-                backend.emitLW(Register.A0, gvi.getLabel(), "Load global: "+gvi.getVarName());
-            }
-            else
-            {
-                backend.emitLW(Register.A0, SP, varSlotMap.get(node.name),"Load local variable: "+node.name);
+
+            Identifier targetID = (Identifier)node;
+            Object var = getVar(targetID);
+            if(var instanceof StackVarRuntimeInfo){
+                StackVarRuntimeInfo rtinfo = (StackVarRuntimeInfo) var;
+                backend.emitLW(A0, rtinfo.sl, rtinfo.off,"Load local variable: "+node.name);
+                rtinfo.free();
             }
+            else 
+                backend.emitLW(A0, (Label) var, "Load local variable: "+node.name);
+
             return Register.A0;
         }
         
-        @Override
-        public Register analyze(VarDef node) 
-        {
-            return null;
-        }
-
+       
         @Override
         public Register analyze(WhileStmt node) 
         {
@@ -812,7 +886,7 @@ public class CodeGenImpl extends CodeGenBase
             Label startLoop = generateLocalLabel(); 
             backend.emitLocalLabel(startLoop, "Beginning of while loop");
             Register result = node.condition.dispatch(this);
-            Label endLoop = elseBlock;
+            Label endLoop = generateLocalLabel();
 
             backend.emitBEQZ(result, endLoop,"Jump to end of the loop");
             for(Stmt stmt:node.body) 
@@ -900,13 +974,18 @@ public class CodeGenImpl extends CodeGenBase
                 backend.emitMUL(iden, iden, ln, "Compute list element offset in bytes");
                 backend.emitADD(iden, l, iden, "Pointer to list element");
                 backend.emitLW(iden, iden, 0, "Set list element");
-                if(sym.getParent() == null)
+                SymbolInfo info = sym.get(node.identifier.name);
+                if(info instanceof GlobalVarInfo)
                 {
-                    GlobalVarInfo gvi=(GlobalVarInfo) sym.get(node.identifier.name);
+                    GlobalVarInfo gvi=(GlobalVarInfo) info;
                     backend.emitSW(iden, gvi.getLabel(), Register.T0, "Assign global: "+node.identifier.name+"(using tmp register)");
                 }
                 else
-                    backend.emitSW(iden, SP, varSlotMap.get(node.identifier.name),"Store local variable: "+node.identifier.name);
+                {
+                    StackVarRuntimeInfo rtinfo = (StackVarRuntimeInfo) getVar(node.identifier);
+                    backend.emitSW(iden, rtinfo.sl, rtinfo.off,"Store local variable: "+node.identifier.name);
+                    rtinfo.free();
+                }
                 for(Stmt stmt:node.body)
                     stmt.dispatch(this);
                 backend.emitADDI(iter, iter, 1, "Increment counter");
@@ -938,14 +1017,18 @@ public class CodeGenImpl extends CodeGenBase
                 
                 backend.emitADDI(iden, iter, 4 * wordSize, "Convert index to offset to char in bytes");
                 backend.emitLBU(iden, iden, 0, "Load character");
-                
-                if(sym.getParent() == null)
+                SymbolInfo info = sym.get(node.identifier.name);
+                if(info instanceof GlobalVarInfo)
                 {
-                    GlobalVarInfo gvi=(GlobalVarInfo) sym.get(node.identifier.name);
+                    GlobalVarInfo gvi=(GlobalVarInfo) info;
                     backend.emitSW(iden, gvi.getLabel(), Register.T0, "Assign global: "+node.identifier.name+"(using tmp register)");
                 }
                 else
-                    backend.emitSW(iden, SP, varSlotMap.get(node.identifier.name),"Store local variable: "+node.identifier.name);
+                {
+                    StackVarRuntimeInfo rtinfo = (StackVarRuntimeInfo) getVar(node.identifier);
+                    backend.emitSW(iden, rtinfo.sl, rtinfo.off,"Store local variable: "+node.identifier.name);
+                    rtinfo.free();
+                }
                 for(Stmt stmt:node.body)
                     stmt.dispatch(this);
                 backend.emitADDI(iter, iter, 1, "Increment counter");