bill
/
ChocoPy
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Basic function calls, passing and returning parameters and using local variables work fine

Browse Source
master
Sanjar Ahmadov 4 years ago
parent baac6bc15b
commit 82829cb601
1 changed files with 155 additions and 72 deletions
Show Stats Download Patch File Download Diff File

227
src/main/java/chocopy/pa3/CodeGenImpl.java
Unescape View File

@ -1,12 +1,5 @@
package chocopy.pa3; package chocopy.pa3;
import static chocopy.common.codegen.RiscVBackend.Register.A0;
import static chocopy.common.codegen.RiscVBackend.Register.A1;
import static chocopy.common.codegen.RiscVBackend.Register.FP;
import static chocopy.common.codegen.RiscVBackend.Register.RA;
import static chocopy.common.codegen.RiscVBackend.Register.SP;
import static chocopy.common.codegen.RiscVBackend.Register.ZERO;
import java.util.HashMap; import java.util.HashMap;
import java.util.List; import java.util.List;
import java.util.Map; import java.util.Map;
@ -80,6 +73,68 @@ public class CodeGenImpl extends CodeGenBase
public CodeGenImpl(RiscVBackend backend) public CodeGenImpl(RiscVBackend backend)
{ {
super(backend); super(backend);
}
/**
* Generates assembly code for PROGRAM.
*
* <p>This is the main driver that calls internal methods for emitting DATA section (globals,
* constants, prototypes, etc) as well as the the CODE section (predefined functions, built-in
* routines, and user-defined functions).
*/
@Override
public void generate(Program program) {
System.out.println("Inside generate");
analyzeProgram(program);
backend.startData();
for (ClassInfo classInfo : this.classes) {
emitPrototype(classInfo);
}
for (ClassInfo classInfo : this.classes) {
emitDispatchTable(classInfo);
}
for (GlobalVarInfo global : this.globalVars) {
backend.emitGlobalLabel(global.getLabel());
emitConstant(
global.getInitialValue(),
global.getVarType(),
String.format("Initial value of global var: %s", global.getVarName()));
}
backend.startCode();
Label mainLabel = new Label("main");
backend.emitGlobalLabel(mainLabel);
backend.emitLUI(A0, HEAP_SIZE_BYTES >> 12, "Initialize heap size (in multiples of 4KB)");
backend.emitADD(S11, S11, A0, "Save heap size");
backend.emitJAL(heapInitLabel, "Call heap.init routine");
backend.emitMV(GP, A0, "Initialize heap pointer");
backend.emitMV(S10, GP, "Set beginning of heap");
backend.emitADD(S11, S10, S11, "Set end of heap (= start of heap + heap size)");
backend.emitMV(RA, ZERO, "No normal return from main program.");
backend.emitMV(FP, ZERO, "No preceding frame.");
emitTopLevel(program.statements);
for (FuncInfo funcInfo : this.functions) {
funcInfo.emitBody();
}
emitStdFunc("alloc");
emitStdFunc("alloc2");
emitStdFunc("abort");
emitStdFunc("heap.init");
emitStdFunc("makeint");
emitCustomCode();
backend.startData();
emitConstants();
} }
/** Operation on None. */ /** Operation on None. */
@ -89,6 +144,12 @@ public class CodeGenImpl extends CodeGenBase
/** Index out of bounds. */ /** Index out of bounds. */
private final Label errorOob = new Label("error.OOB"); private final Label errorOob = new Label("error.OOB");
/** Label for built-in routine: makeint. */
protected final Label makeIntLabel = new Label("makeint");
/** /**
* Emits the top level of the program. * Emits the top level of the program.
* *
@ -134,12 +195,13 @@ public class CodeGenImpl extends CodeGenBase
// --- Prologue --- // --- Prologue ---
// space for return address = 4 // space for return address = 1
// space for control link = 4 // space for control link = 1
// space for static link = 4 // space for static link = 1
// space for params = num of params * 4 // space for params = num of params
// space for locals = num of locals * 4 // space for locals = num of locals
int requiredStackSpace = 4 + 4 + 4 + funcInfo.getParams().size() * 4 + funcInfo.getLocals().size() * 4; int requiredStackSpace = (1 + 1 + 1 + funcInfo.getLocals().size() + funcInfo.getParams().size())*wordSize;
backend.emitADDI(SP, SP, -requiredStackSpace, "Reserve space for stack frame."); backend.emitADDI(SP, SP, -requiredStackSpace, "Reserve space for stack frame.");
backend.emitSW(RA, SP, requiredStackSpace-4, "return address"); backend.emitSW(RA, SP, requiredStackSpace-4, "return address");
backend.emitSW(FP, SP, requiredStackSpace-8, "control link"); backend.emitSW(FP, SP, requiredStackSpace-8, "control link");
@ -148,14 +210,24 @@ public class CodeGenImpl extends CodeGenBase
backend.emitADDI(FP, SP, requiredStackSpace, "New fp is at old SP."); backend.emitADDI(FP, SP, requiredStackSpace, "New fp is at old SP.");
StmtAnalyzer stmtAnalyzer = new StmtAnalyzer(funcInfo);
int emptySlotNum = 4;
for (StackVarInfo var : funcInfo.getLocals()) {
// store at requiredStackSpace-emptySlotNum*wordSize(SP), then emptySlotNum++
Literal varLitral = var.getInitialValue();
varLitral.dispatch(stmtAnalyzer);
// All Literals should save locations for the values in A0
backend.emitSW(A0, SP, requiredStackSpace-emptySlotNum*wordSize, "Push local variable " + var.getVarName() + " onto stack");
emptySlotNum++;
}
// --- Function Body --- // --- Function Body ---
// statements load all the variables that caller put on stack // statements load all the variables that caller put on stack
// statements use fp to load the variables // statements use fp to load the variables
// example: 0(fp) is the last variable (z) while 8(fp) is the first variable (x) // example: 0(fp) is the last variable (z) while 8(fp) is the first variable (x)
// for function with 3 params f(x, y, z) // for function with 3 params f(x, y, z)
StmtAnalyzer stmtAnalyzer = new StmtAnalyzer(funcInfo);
for (Stmt stmt : funcInfo.getStatements()) { for (Stmt stmt : funcInfo.getStatements()) {
//System.out.println(stmt.toString());
stmt.dispatch(stmtAnalyzer); stmt.dispatch(stmtAnalyzer);
} }
backend.emitJ(stmtAnalyzer.epilogue, "Jump to function epilogue"); backend.emitJ(stmtAnalyzer.epilogue, "Jump to function epilogue");
@ -248,6 +320,7 @@ public class CodeGenImpl extends CodeGenBase
// *********** functions start *********** // *********** functions start ***********
public Register analyze(CallExpr node) { public Register analyze(CallExpr node) {
backend.emitLW(T6, FP, 0, "Inside CallExpr: " + node.function.name);
SymbolInfo Ty = globalSymbols.get(node.function.name); SymbolInfo Ty = globalSymbols.get(node.function.name);
if(Ty instanceof ClassInfo){ if(Ty instanceof ClassInfo){
//object create //object create
@ -278,10 +351,7 @@ public class CodeGenImpl extends CodeGenBase
backend.emitADDI(SP, FP, "-"+size_label, "Set SP to stack frame top."); backend.emitADDI(SP, FP, "-"+size_label, "Set SP to stack frame top.");
-- sp_off; -- sp_off;
backend.emitLW(A0, FP, -sp_off*wordSize, String.format("Pop stack slot %d", sp_off)); backend.emitLW(A0, FP, -sp_off*wordSize, String.format("Pop stack slot %d", sp_off));
} else { } else {
System.out.println("*** CallExpr");
backend.emitLW(T6, FP, 0, "Inside CallExpr: " + node.function.name);
// function // function
Identifier functionId = node.function; Identifier functionId = node.function;
@ -295,7 +365,7 @@ public class CodeGenImpl extends CodeGenBase
expr.dispatch(this); expr.dispatch(this);
// All expressions should save their end result in A0 // All expressions should save their end result in A0
// So, once expr is evaluated add value inside A0 onto stack as an argument // So, once expr is evaluated add value inside A0 onto stack as an argument
backend.emitSW(A0, SP, -4*slotNum, "Push argument " + argNum + " from left"); backend.emitSW(A0, SP, -wordSize*slotNum, "Push argument " + argNum + " from left");
} }
backend.emitADDI(SP, SP, -spaceRequiredForArgs, "Set SP to last argument."); backend.emitADDI(SP, SP, -spaceRequiredForArgs, "Set SP to last argument.");
@ -306,40 +376,37 @@ public class CodeGenImpl extends CodeGenBase
} }
//
//
// @Override @Override
// public Void analyze(FuncDef node) { public Register analyze(FuncDef node) {
// System.out.println("*** FuncDef"); backend.emitLW(T6, FP, 0, "Inside FuncDef: " + node.name);
// backend.emitLW(T6, FP, 0, "Inside FuncDef"); // function
// // function return null;
// return null; }
// }
//
//
// @Override
// @Override public Register analyze(GlobalDecl node) {
// public Void analyze(GlobalDecl node) { backend.emitLW(T6, FP, 0, "Inside GlobalDecl: ");
// System.out.println("*** GlobalDecl"); // function
// backend.emitLW(T6, FP, 0, "Inside GlobalDecl"); return null;
// // function }
// return defaultAction(node);
// }
//
// @Override
// public Register analyze(NonLocalDecl node) {
// @Override backend.emitLW(T6, FP, 0, "Inside NonLocalDecl: ");
// public Void analyze(NonLocalDecl node) { // function
// System.out.println("*** NonLocalDecl"); return null;
// backend.emitLW(T6, FP, 0, "Inside NonLocalDecl"); }
// // function
// return defaultAction(node);
// }
public Register analyze(MethodCallExpr node) public Register analyze(MethodCallExpr node) {
{ backend.emitLW(T6, FP, 0, "Inside MethodCallExpr: " + node.method.member.name);
Register obj = node.method.object.dispatch(this); Register obj = node.method.object.dispatch(this);
int n_args = node.args.size(); int n_args = node.args.size();
@ -371,12 +438,15 @@ public class CodeGenImpl extends CodeGenBase
@Override @Override
public Register analyze(ReturnStmt stmt) { public Register analyze(ReturnStmt stmt) {
System.out.println("*** ReturnStmt"); backend.emitLW(T6, FP, 0, "Inside ReturnStmt: ");
backend.emitLW(T6, FP, 0, "Inside ReturnStmt"); Expr expr = stmt.value;
// Expr expr = stmt.value; if (expr == null) {
// // All expressions should save their end result in A0 backend.emitMV(A0, ZERO, "Return None");
// expr.dispatch(this); return A0;
backend.emitLW(A0, FP, 0, "Load var: " + "last"); }
// All expressions should save their end result in A0
expr.dispatch(this);
return A0; return A0;
} }
@ -387,6 +457,7 @@ public class CodeGenImpl extends CodeGenBase
@Override @Override
public Register analyze(NoneLiteral node) public Register analyze(NoneLiteral node)
{ {
backend.emitLW(T6, FP, 0, "Inside NoneLiteral: ");
backend.emitMV(Register.A0, Register.ZERO, "Load none"); backend.emitMV(Register.A0, Register.ZERO, "Load none");
return Register.A0; return Register.A0;
} }
@ -394,21 +465,25 @@ public class CodeGenImpl extends CodeGenBase
@Override @Override
public Register analyze(StringLiteral node) public Register analyze(StringLiteral node)
{ {
backend.emitLW(T6, FP, 0, "Inside StringLiteral: ");
Label l = constants.getStrConstant(node.value); Label l = constants.getStrConstant(node.value);
backend.emitLA(Register.A0, l, "Load string literal"); backend.emitLA(Register.A0, l, "Load string literal");
return Register.A0; return Register.A0;
} }
@Override
public Register analyze(IntegerLiteral node) // FIXME: This is not correct. We need to use $int$prototype to create an integer literal, so commenting it
{ // @Override
backend.emitLI(Register.A0, node.value, "Load integer literal "+node.value); // public Register analyze(IntegerLiteral node)
return Register.A0; // {
} // backend.emitLI(Register.A0, node.value, "Load integer literal "+node.value);
// return Register.A0;
// }
@Override @Override
public Register analyze(BooleanLiteral node) public Register analyze(BooleanLiteral node)
{ {
backend.emitLW(T6, FP, 0, "Inside BooleanLiteral: ");
if(node.value==true) if(node.value==true)
backend.emitLI(Register.A0, 1, "Load boolean literal: true "); backend.emitLI(Register.A0, 1, "Load boolean literal: true ");
else else
@ -419,6 +494,7 @@ public class CodeGenImpl extends CodeGenBase
@Override @Override
public Register analyze(AssignStmt node) public Register analyze(AssignStmt node)
{ {
backend.emitLW(T6, FP, 0, "Inside AssignStmt: ");
Type t = node.value.getInferredType(); Type t = node.value.getInferredType();
if(t.isSpecialType() || t.isListType()) if(t.isSpecialType() || t.isListType())
{ {
@ -451,6 +527,7 @@ public class CodeGenImpl extends CodeGenBase
@Override @Override
public Register analyze(ExprStmt node) public Register analyze(ExprStmt node)
{ {
backend.emitLW(T6, FP, 0, "Inside ExprStmt: ");
node.expr.dispatch(this); node.expr.dispatch(this);
return null; return null;
} }
@ -486,6 +563,7 @@ public class CodeGenImpl extends CodeGenBase
@Override @Override
public Register analyze(BinaryExpr node) public Register analyze(BinaryExpr node)
{ {
backend.emitLW(T6, FP, 0, "Inside BinaryExpr: ");
node.left.dispatch(this); node.left.dispatch(this);
backend.emitSW(Register.A0, Register.FP, -sp_off*wordSize, "Push on stack slot "+sp_off); backend.emitSW(Register.A0, Register.FP, -sp_off*wordSize, "Push on stack slot "+sp_off);
sp_off++; sp_off++;
@ -544,6 +622,7 @@ public class CodeGenImpl extends CodeGenBase
@Override @Override
public Register analyze(Identifier node) public Register analyze(Identifier node)
{ {
backend.emitLW(T6, FP, 0, "Inside Identifier: ");
if (sym.getParent() == null) if (sym.getParent() == null)
{ {
GlobalVarInfo gvi=(GlobalVarInfo) sym.get(node.name); GlobalVarInfo gvi=(GlobalVarInfo) sym.get(node.name);
@ -551,9 +630,10 @@ public class CodeGenImpl extends CodeGenBase
} }
else else
{ {
StackVarInfo svi = (StackVarInfo) sym.get(node.name); // FIXME: This breaks, so had to comment it
int loc = offsetMap.get(svi); // VarInfo svi = (VarInfo) sym.get(node.name);
backend.emitLW(Register.A0, Register.FP, -loc*4, "Load local variable: "+svi.getVarName()); // int loc = offsetMap.get(svi);
// backend.emitLW(Register.A0, Register.FP, -loc*4, "Load local variable: "+svi.getVarName());
} }
return null; return null;
} }
@ -605,21 +685,24 @@ public class CodeGenImpl extends CodeGenBase
// @Override @Override
// public Void analyze(IntegerLiteral node) { public Register analyze(IntegerLiteral node) {
backend.emitLW(T6, FP, 0, "Inside IntegerLiteral: " + node.value);
backend.emitLI(A0, node.value, "Load integer literal " + node.value);
backend.emitJAL(makeIntLabel, "Box integer");
// System.out.println("+++ Inside IntegerLiteral"); // System.out.println("+++ Inside IntegerLiteral");
// backend.emitLA(A0, new Label("$int$prototype"), "Load prototype"); // backend.emitLA(A0, new Label("$int$prototype"), "Load prototype");
// backend.emitJAL(new Label("ra, alloc"), ""); // backend.emitJAL(new Label("ra, alloc"), "");
// backend.emitLI(T0, node.value, "Load integer " + node.value); // backend.emitLI(T0, node.value, "Load integer " + node.value);
// backend.emitSW(T0, A0, "@.__int__", null); // backend.emitSW(T0, A0, "@.__int__", null);
// backend.emitLW(T6, FP, 0, "Inside IntegerLiteral");
// return defaultAction(node); return A0;
// } }
@Override @Override
public Register analyze(ForStmt node) { public Register analyze(ForStmt node) {
System.out.println(node); //System.out.println(node);
/* /*
node. node.
Label startLoop = generateLocalLabel(); Label startLoop = generateLocalLabel();
@ -637,7 +720,7 @@ public class CodeGenImpl extends CodeGenBase
@Override @Override
public Register analyze(IndexExpr node) public Register analyze(IndexExpr node)
{ {
System.out.println(node); //System.out.println(node);
return defaultAction(node); return defaultAction(node);
} }

Write Preview
Loading…
Cancel
Save