diff options
Diffstat (limited to 'include/cps')
| -rw-r--r-- | include/cps/CodeGen.h | 3 | ||||
| -rw-r--r-- | include/cps/FunctionGen.h | 12 |
2 files changed, 15 insertions, 0 deletions
diff --git a/include/cps/CodeGen.h b/include/cps/CodeGen.h index 268ba7e..b1d3831 100644 --- a/include/cps/CodeGen.h +++ b/include/cps/CodeGen.h @@ -69,6 +69,9 @@ class CodeGen { const SymbolInfo *getSymbolInfo(const std::string &Name) const; const TypeInfo *getExprTypeInfo(ExprAST *Expr) const; llvm::Value *coerceValueToType(llvm::Value *Val, const TypeInfo *TargetInfo); + bool marshalCallArgs(const std::string &Callee, + const std::vector<std::unique_ptr<ExprAST>> &ArgExprs, + std::vector<llvm::Value*> &Out); void emitDeclareStmt(DeclareStmtAST *Stmt); void emitOutputValue(llvm::Value *Val, const TypeInfo *TypeInfo, bool AppendNewline = true); diff --git a/include/cps/FunctionGen.h b/include/cps/FunctionGen.h index 1dc3901..f55db1b 100644 --- a/include/cps/FunctionGen.h +++ b/include/cps/FunctionGen.h @@ -11,6 +11,14 @@ namespace cps { +// Pseudocode-level signature of a FUNCTION/PROCEDURE, kept so call sites can +// dispatch on declared parameter types and BYREF flags instead of inferring +// them back from LLVM types. +struct FuncSig { + std::string ReturnTypeName; + std::vector<std::pair<std::string, bool>> Params; // (TypeName, IsByRef) +}; + class FunctionGen { llvm::LLVMContext &Context; llvm::Module &Module; @@ -21,6 +29,8 @@ class FunctionGen { std::map<std::string, SymbolInfo> &Symbols; bool &HadError; + std::map<std::string, FuncSig> Signatures; + void createArgumentAllocas(llvm::Function *F, const std::vector<std::tuple<std::string, std::string, bool>> &Args); public: @@ -35,6 +45,8 @@ public: llvm::Type *getLLVMType(const std::string &TypeName); + const FuncSig *getSignature(const std::string &Name) const; + llvm::Function *emitPrototype(PrototypeAST *Proto); llvm::Function *emitFunctionDef(FunctionDefAST *FuncAST, const std::function<void(StmtAST*)> &StmtEmitter); |