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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
#include <memory>
#include <vector>
#include <llvm/IR/IRBuilder.h>
#include <llvm/IR/Verifier.h>
#include "DeclNode.hh"
#include "FunctionNode.hh"
#include "Log.hh"
#include "TypeIdentifier.hh"
namespace pcc {
FunctionNode::FunctionNode(
Context* context, std::string&& name, std::list<Declaration>&& arguments,
std::list<std::shared_ptr<DeclNode>>&& LocalDeclarations,
std::shared_ptr<TypeIdentifier> ReturnTypeIdentifier,
std::shared_ptr<StatementListNode> body)
: BaseNode(context)
, name_(std::move(name))
, arguments_(std::move(arguments))
, LocalDeclarations_(std::move(LocalDeclarations))
, ReturnTypeIdentifier_(ReturnTypeIdentifier)
, locals_(context->GetGlobals())
, body_(body) {
for (auto&& decl : LocalDeclarations_) {
decl->SetScope(&locals_);
}
}
Value FunctionNode::CodeGen() {
llvm::Module* module = context_->GetModule();
llvm::IRBuilder<>* builder = context_->GetBuilder();
TypeManager* manager = context_->GetTypeManager();
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)->GetType(); });
std::shared_ptr<Type> ReturnType = ReturnTypeIdentifier_->GetType();
std::shared_ptr<FunctionType> type =
manager->CreateFunctionType(ReturnType, params);
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);
context_->AddFunction(name_, {type, function});
llvm::BasicBlock* block =
llvm::BasicBlock::Create(GlobalLLVMContext, name_, function);
Log(LogLevel::PCC_INFO, "generating function %s of type %s", name_.c_str(),
type->GetCommonName());
builder->SetInsertPoint(block);
context_->PushScope(&locals_);
// alloc arguments
llvm::Function::arg_iterator ArgIt = function->arg_begin();
decltype(params)::iterator ParamIt = params.begin();
for (auto&& arg : arguments_) {
std::shared_ptr<Type> ArgType = *ParamIt;
Value ArgValue = builder->CreateAlloca(ArgType->GetLLVMType());
builder->CreateStore(ArgIt, ArgValue);
locals_.AddVariable(std::get<0>(arg),
{manager->GetPointerType(ArgType), ArgValue});
++ArgIt, ++ParamIt;
}
// alloc local variables
for (auto&& decl : LocalDeclarations_) {
decl->CodeGen();
}
Value rv = nullptr;
bool IsVoid = ReturnType == manager->GetBuiltinType(BuiltinType::VOID);
if (IsVoid == false) {
// create a variable with the same name as the function for return value
rv = builder->CreateAlloca(ReturnType->GetLLVMType());
locals_.AddVariable(name_, {manager->GetPointerType(ReturnType), rv});
}
// push the function scope variables
body_->CodeGen();
if (IsVoid == false) {
builder->CreateRet(builder->CreateLoad(ReturnType->GetLLVMType(), rv));
} else {
builder->CreateRetVoid();
}
context_->PopScope();
std::string msg;
llvm::raw_string_ostream MsgStream(msg);
llvm::verifyFunction(*function, &MsgStream);
MsgStream.flush();
if (!msg.empty()) {
Log(LogLevel::PCC_INFO, "%s", msg.c_str());
}
return nullptr;
}
ExternNode::ExternNode(Context* context, std::string&& name,
std::list<Declaration>&& arguments,
std::shared_ptr<TypeIdentifier> ReturnTypeIdentifier,
bool IsVariadic)
: FunctionNode(context, std::move(name), std::move(arguments),
std::list<std::shared_ptr<DeclNode>>(), ReturnTypeIdentifier,
nullptr)
, IsVariadic_(IsVariadic) {}
Value ExternNode::CodeGen() {
llvm::Module* module = context_->GetModule();
llvm::IRBuilder<>* builder = context_->GetBuilder();
TypeManager* manager = context_->GetTypeManager();
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)->GetType(); });
std::shared_ptr<Type> ReturnType = ReturnTypeIdentifier_->GetType();
std::shared_ptr<FunctionType> type =
manager->CreateFunctionType(ReturnType, params, IsVariadic_);
llvm::FunctionType* LLVMType =
llvm::cast<llvm::FunctionType>(type->GetLLVMType());
llvm::FunctionCallee callee = module->getOrInsertFunction(name_, LLVMType);
llvm::Function* function = llvm::cast<llvm::Function>(callee.getCallee());
function->setCallingConv(llvm::CallingConv::C);
context_->AddFunction(name_, {type, function});
Log(LogLevel::PCC_INFO, "extern function %s of type %s", name_.c_str(),
type->GetCommonName());
return nullptr;
}
} // namespace pcc
|
