solid-lang/IRGenerator.h at main · m-maresch/solid-lang · GitHub

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
#ifndef SOLID_LANG_IRGENERATOR_H
#define SOLID_LANG_IRGENERATOR_H

#include "llvm/IR/IRBuilder.h"
#include "llvm/IR/LLVMContext.h"
#include "llvm/IR/Module.h"
#include <llvm/IR/LegacyPassManager.h>
#include <map>
#include <utility>

#include "ExpressionVisitor.h"

using namespace llvm;

class IRGenerator : public ExpressionVisitor {
    LLVMContext &Context;

    IRBuilder<> &Builder;

    Module &Module;

    std::unique_ptr<legacy::FunctionPassManager> PassManager;

    std::map<std::string, AllocaInst *> &ValuesByName;

    std::map<std::string, std::unique_ptr<FunctionDeclaration>> &FunctionDeclarations;

    Value *Current;

    Function *LookupFunction(std::string Name);

    AllocaInst *CreateAlloca(Function *Func, StringRef Name);

public:
    explicit IRGenerator(LLVMContext &Context, IRBuilder<> &Builder, class Module &Module,
                         std::unique_ptr<legacy::FunctionPassManager> PassManager,
                         std::map<std::string, AllocaInst *> &ValuesByName,
                         std::map<std::string, std::unique_ptr<FunctionDeclaration>> &FunctionDeclarations)
            : Context(Context), Builder(Builder), Module(Module), PassManager(std::move(PassManager)),
              ValuesByName(ValuesByName), FunctionDeclarations(FunctionDeclarations) {}

    void Visit(VariableExpression &Expression) override;

    void Visit(VariableDefinition &Expression) override;

    void Visit(FunctionCall &Expression) override;

    void Visit(FunctionDeclaration &Expression) override;

    void Visit(FunctionDefinition &Expression) override;

    void Visit(UnaryExpression &Expression) override;

    void Visit(BinaryExpression &Expression) override;

    void Visit(NumExpression &Expression) override;

    void Visit(ConditionalExpression &Expression) override;

    void Visit(LoopExpression &Expression) override;

    void Register(std::unique_ptr<FunctionDeclaration> Declaration) override;

    Value *GetValue() {
        return Current;
    }

    void LogError(const char *Message) {
        fprintf(stderr, "Error: %s\n", Message);
    }
};

class IRPrinter : public ExpressionVisitor {
    std::unique_ptr<class IRGenerator> IRGenerator;

    void Print() {
        if (auto *IR = IRGenerator->GetValue()) {
            errs() << "\n" << "Generated LLVM IR:" << "\n";
            IR->print(errs());
            errs() << "\n";
        }
    }

public:
    explicit IRPrinter(std::unique_ptr<class IRGenerator> IRGenerator) : IRGenerator(std::move(IRGenerator)) {}

    void Visit(VariableExpression &Expression) override;

    void Visit(VariableDefinition &Expression) override;

    void Visit(FunctionCall &Expression) override;

    void Visit(FunctionDeclaration &Expression) override;

    void Visit(FunctionDefinition &Expression) override;

    void Visit(UnaryExpression &Expression) override;

    void Visit(BinaryExpression &Expression) override;

    void Visit(NumExpression &Expression) override;

    void Visit(ConditionalExpression &Expression) override;

    void Visit(LoopExpression &Expression) override;

    void Register(std::unique_ptr<FunctionDeclaration> Declaration) override;
};


#endif