package chocopy.pa3; 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.*; import chocopy.common.analysis.types.*; import chocopy.common.codegen.*; import chocopy.common.codegen.RiscVBackend.Register; import static chocopy.common.codegen.RiscVBackend.Register.*; /** * This is where the main implementation of PA3 will live. * *

* A large part of the functionality has already been implemented in the base * class, CodeGenBase. Make sure to read through that class, since you will want * to use many of its fields and utility methods in this class when emitting * code. * *

* Also read the PDF spec for details on what the base class does and what APIs * it exposes for its sub-class (this one). Of particular importance is knowing * what all the SymbolInfo classes contain. */ public class CodeGenImpl extends CodeGenBase { /** Label for built-in routines. */ protected final Label makeintLabel = new Label("makeint"); protected final Label strneqlLabel = new Label("strneql"); protected final Label streqlLabel = new Label("streql"); protected final Label makeboolLabel = new Label("makebool"); protected final Label strcatLabel = new Label("strcat"); protected final Label concatLabel = new Label("concat"); protected final Label conslistLabel = new Label("conslist"); /** Operation on None. */ private final Label errorNone = new Label("error.None"); /** Division by zero. */ private final Label errorDiv = new Label("error.Div"); /** Index out of bounds. */ private final Label errorOob = new Label("error.OOB"); /** Let's try to implement everything first.*/ private final Label errorNI = new Label("error.NI"); /** A code generator emitting instructions to BACKEND. */ public CodeGenImpl(RiscVBackend backend) { super(backend); backend.defineSym("listHeaderWords", "4"); backend.defineSym("strHeaderWords", "4"); backend.defineSym("bool.True", "const_1"); backend.defineSym("bool.False", "const_0"); } /** * Emits the top level of the program. * *

* This method is invoked exactly once, and is surrounded by some boilerplate * code that: (1) initializes the heap before the top-level begins and (2) exits * after the top-level ends. * *

* You only need to generate code for statements. * * @param statements top level statements */ protected void emitTopLevel(List statements) { StmtAnalyzer stmtAnalyzer = new StmtAnalyzer(null); backend.emitADDI(SP, SP, -2 * backend.getWordSize(), "Saved FP and saved RA (unused at top level)."); backend.emitSW(ZERO, SP, 0, "Top saved FP is 0."); backend.emitSW(ZERO, SP, 4, "Top saved RA is 0."); backend.emitADDI(FP, SP, 2 * backend.getWordSize(), "Set FP to previous SP."); for (Stmt stmt : statements) { stmt.dispatch(stmtAnalyzer); } stmtAnalyzer.emitSizeLabel(); backend.emitLI(A0, EXIT_ECALL, "Code for ecall: exit"); backend.emitEcall(null); } /** * Emits the code for a function described by FUNCINFO. * *

* This method is invoked once per function and method definition. At the code * generation stage, nested functions are emitted as separate functions of their * own. So if function `bar` is nested within function `foo`, you only emit * `foo`'s code for `foo` and only emit `bar`'s code for `bar`. */ protected void emitUserDefinedFunction(FuncInfo funcInfo) { backend.emitGlobalLabel(funcInfo.getCodeLabel()); // --- Prologue --- // space for return address = 1 // space for control link = 1 // space for static link = 1 // space for locals = num of locals int requiredStackSpace = (1 + 1 + 1 + funcInfo.getLocals().size())*wordSize; backend.emitADDI(SP, SP, -requiredStackSpace, "Reserve space for stack frame."); backend.emitSW(RA, SP, requiredStackSpace-4, "return address"); backend.emitSW(FP, SP, requiredStackSpace-8, "control link"); // if we want to add static link //backend.emitSW(FP, SP, requiredStackSpace-12, "static link"); backend.emitADDI(FP, SP, requiredStackSpace, "New fp is at old SP."); StmtAnalyzer stmtAnalyzer = new StmtAnalyzer(funcInfo); int emptySlotNum = 3; for (StackVarInfo var : funcInfo.getLocals()) { 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 --- // statements load all the variables that caller put on stack // statements use fp to load the variables // 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 (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"); backend.emitLW(FP, FP, -8, "Use control link to restore caller's fp"); backend.emitADDI(SP, SP, requiredStackSpace, "Restore stack pointer"); backend.emitJR(RA, "Return to caller"); } /** An analyzer that encapsulates code generation for statements. */ private class StmtAnalyzer extends AbstractNodeAnalyzer { /* * The symbol table has all the info you need to determine what a given * identifier 'x' in the current scope is. You can use it as follows: SymbolInfo * x = sym.get("x"); * * A SymbolInfo can be one the following: - ClassInfo: a descriptor for classes * - FuncInfo: a descriptor for functions/methods - AttrInfo: a descriptor for * attributes - GlobalVarInfo: a descriptor for global variables - StackVarInfo: * a descriptor for variables allocated on the stack, such as locals and * parameters * * Since the input program is assumed to be semantically valid and well-typed at * this stage, you can always assume that the symbol table contains valid * information. For example, in an expression `foo()` you KNOW that * sym.get("foo") will either be a FuncInfo or ClassInfo, but not any of the * other infos and never null. * * The symbol table in funcInfo has already been populated in the base class: * CodeGenBase. You do not need to add anything to the symbol table. Simply * query it with an identifier name to get a descriptor for a function, class, * variable, etc. * * The symbol table also maps nonlocal and global vars, so you only need to * lookup one symbol table and it will fetch the appropriate info for the var * that is currently in scope. */ /** Symbol table for my statements. */ private final SymbolTable sym; /** Label of code that exits from procedure. */ protected final Label epilogue; /** The descriptor for the current function, or null at the top level. */ private final FuncInfo funcInfo; /** Label of code that exits from block. */ protected Stack

This method is called after emitting the top level and the function bodies for each * function. * *

You can use this method to emit anything you want outside of the top level or functions, * e.g. custom routines that you may want to call from within your code to do common tasks. This * is not strictly needed. You might not modify this at all and still complete the assignment. * *

To start you off, here is an implementation of three routines that will be commonly needed * from within the code you will generate for statements. * *

The routines are error handlers for operations on None, index out of bounds, and division * by zero. They never return to their caller. Just jump to one of these routines to throw an * error and exit the program. For example, to throw an OOB error: backend.emitJ(errorOob, "Go * to out-of-bounds error and abort"); */ protected void emitCustomCode() { emitStdFunc("concat"); emitStdFunc("conslist"); emitStdFunc("strcat"); emitStdFunc("streql"); emitStdFunc("strneql"); emitStdFunc("makeint"); emitStdFunc("makebool"); emitErrorFunc(errorNone, "Operation on None"); emitErrorFunc(errorDiv, "Division by zero"); emitErrorFunc(errorOob, "Index out of bounds"); emitErrorFunc(errorNI, "Not Implemented."); } /** Emit an error routine labeled ERRLABEL that aborts with message MSG. */ private void emitErrorFunc(Label errLabel, String msg) { backend.emitGlobalLabel(errLabel); if(errLabel==errorOob) backend.emitLI(A0, ERROR_OOB, "Exit code for: " + msg); else if(errLabel==errorDiv) backend.emitLI(A0, ERROR_DIV_ZERO, "Exit code for: " + msg); else if(errLabel==errorNI) backend.emitLI(A0, ERROR_NYI, "Exit code for: " + msg); else backend.emitLI(A0, ERROR_NONE, "Exit code for: " + msg); backend.emitLA(A1, constants.getStrConstant(msg), "Load error message as str"); backend.emitADDI( A1, A1, getAttrOffset(strClass, "__str__"), "Load address of attribute __str__"); backend.emitJ(abortLabel, "Abort"); } }