67 lines
1.5 KiB
C++
67 lines
1.5 KiB
C++
|
|
||
|
|
#include <cmath>
|
||
|
|
#include <memory>
|
||
|
|
|
||
|
|
#include "llvm/IR/Value.h"
|
||
|
|
|
||
|
|
#include "LLVMCodeGen.hpp"
|
||
|
|
#include "bolt/CST.hpp"
|
||
|
|
#include "bolt/CSTVisitor.hpp"
|
||
|
|
|
||
|
|
namespace bolt {
|
||
|
|
|
||
|
|
LLVMCodeGen::LLVMCodeGen(llvm::LLVMContext* TheContext):
|
||
|
|
TheContext(TheContext) {}
|
||
|
|
|
||
|
|
llvm::Value* LLVMCodeGen::generateExpression(Expression* E) {
|
||
|
|
|
||
|
|
switch (E->getKind()) {
|
||
|
|
|
||
|
|
case NodeKind::LiteralExpression:
|
||
|
|
{
|
||
|
|
auto Lit = static_cast<LiteralExpression*>(E);
|
||
|
|
switch (Lit->Token->getKind()) {
|
||
|
|
case NodeKind::IntegerLiteral:
|
||
|
|
{
|
||
|
|
auto V = static_cast<IntegerLiteral*>(Lit->Token)->V;
|
||
|
|
// TODO assert that V fits in the APInt
|
||
|
|
return llvm::ConstantInt::get(*TheContext, llvm::APInt(32, V));
|
||
|
|
}
|
||
|
|
case NodeKind::StringLiteral:
|
||
|
|
{
|
||
|
|
auto Str = static_cast<StringLiteral*>(Lit->Token)->Text;
|
||
|
|
return Builder->CreateGlobalStringPtr(llvm::StringRef(Str));
|
||
|
|
}
|
||
|
|
default:
|
||
|
|
ZEN_UNREACHABLE
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
default:
|
||
|
|
ZEN_UNREACHABLE
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
void LLVMCodeGen::generateElement(Node* N) {
|
||
|
|
switch (N->getKind()) {
|
||
|
|
case NodeKind::ExpressionStatement:
|
||
|
|
{
|
||
|
|
auto Stmt = static_cast<ExpressionStatement*>(N);
|
||
|
|
generateExpression(Stmt->Expression);
|
||
|
|
}
|
||
|
|
default:
|
||
|
|
ZEN_UNREACHABLE
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void LLVMCodeGen::generate(SourceFile* SF) {
|
||
|
|
Module = std::make_unique<llvm::Module>(SF->File.getPath(), *TheContext);
|
||
|
|
for (auto Element: SF->Elements) {
|
||
|
|
generateElement(Element);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|