1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
#include <memory>
#include <vector>
#include <llvm/IR/IRBuilder.h>
#include "DeclNode.hh"
#include "FunctionNode.hh"
namespace pcc {
FunctionNode::FunctionNode(
Context* context, std::string name, std::list<Declaration> arguments,
std::list<std::shared_ptr<BaseNode>> LocalDeclarations,
std::shared_ptr<Type> type, std::shared_ptr<StatementListNode> body)
: BaseNode(context)
, name_(name)
, arguments_(arguments)
, LocalDeclarations_(LocalDeclarations)
, type_(type)
, locals_(context->GetGlobals())
, body_(body) {
for (auto& decl : LocalDeclarations_) {
std::shared_ptr<VarDeclNode> VarDecl =
std::dynamic_pointer_cast<VarDeclNode>(decl);
if (VarDecl != nullptr) {
// set the inherited attribute
VarDecl->SetScope(&locals_);
}
}
}
Value FunctionNode::CodeGen() {
llvm::Module* module = context_->GetModule();
llvm::IRBuilder<>* builder = context_->GetBuilder();
std::vector<std::shared_ptr<Type>> params(arguments_.size());
std::transform(arguments_.cbegin(), arguments_.cend(), params.begin(),
[](const Declaration& decl) { return std::get<1>(decl); });
std::shared_ptr<Type> type =
context_->GetTypeManager()->CreateFunctionType(type_, params);
context_->AddFunctionType(name_, type);
llvm::FunctionType* LLVMType =
llvm::cast<llvm::FunctionType>(type->GetLLVMType());
llvm::Function* function = llvm::Function::Create(
LLVMType, llvm::GlobalValue::ExternalLinkage, name_, *module);
function->setCallingConv(llvm::CallingConv::C);
llvm::BasicBlock* block =
llvm::BasicBlock::Create(GlobalLLVMContext, name_, function);
builder->SetInsertPoint(block);
// alloc arguments
llvm::Function::arg_iterator ArgIt;
for (auto& arg : arguments_) {
Value ArgValue = builder->CreateAlloca(std::get<1>(arg)->GetLLVMType());
builder->CreateStore(ArgIt++, ArgValue);
locals_.Add(std::get<0>(arg), {std::get<1>(arg), ArgValue});
}
// alloc local variables
for (auto& decl : LocalDeclarations_) {
decl->CodeGen();
}
// create a variable with the same name as the function for return value
Value rv = builder->CreateAlloca(type_->GetLLVMType());
locals_.Add(name_, {type_, rv});
// push the function scope variables
context_->PushScope(&locals_);
body_->CodeGen();
builder->CreateRet(rv);
context_->PopScope();
return nullptr;
}
} // namespace pcc
|
