Quail/quail.cpp at main · ThorGameDev/Quail · 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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
#include "./src/lexer.h"
#include "./src/AST.h"
#include "./src/codegen.h"
#include "./src/BinopsData.h"
#include "./src/logging.h"
#include "./src/output.h"
#include "./src/codegen/optimizations.h"
#include <cstring>
#include <iostream>
#include <string>
#include <vector>

/// top ::= definition | external | expression | ';'
void JitLine() {
    resetLexer();
    std::cout << ">>> ";
    initBuffer();
    try {
        while (true) {
            switch (CurTok) {
            case tok_eof:
                std::cout << "\n";
                return;
            case tok_def:
                CG::HandleDefinitionJit();
                break;
            case tok_extern:
                CG::HandleExtern();
                break;
            default:
                CG::HandleTopLevelExpression();
                break;
            }
        }
    }
    catch (CompileError ce){
        DebugLog("Error Recovered\n");
    }
}

void MainLoop(){
    InitializeBinopPrecedence();
    CG::InitializeCodegen();
    CG::InitializeModuleAndManagers();
    while (true){
        JitLine();
    }
}

void compileFile(char* filepath, char* savename) {
    InitializeBinopPrecedence();
    CG::InitializeCodegen();
    CG::InitializeModuleAndManagers();

    resetLexer();
    readFile(filepath);
    try {
        bool looping = true;
        while (looping) {
            switch (CurTok) {
            case tok_eof:
                std::cout << "\n";
                looping = false;
                break;
            case tok_def:
                CG::HandleDefinitionFile();
                break;
            case tok_extern:
                CG::HandleExtern();
                break;
            default:
                LogError("Invalid Top-level expression");
                break;
            }
        }
        SaveToFile(savename);
    }
    catch (CompileError ce){
        DebugLog("Failed to compile file due to errors\n");
    }
}

int main(int argc, char* argv[]) {
    // Read CLI args and set relevant data
    int argType = 0;
    std::vector<char*> filepaths;
    std::vector<char*> outputs;
    uint optimizationLevel = 2;
    for (int i = 1; i < argc; i++){
        char* arg = argv[i];
        if (strcmp(arg, "-o") == 0){
            argType = 1;
            continue;
        }
        else if (strcmp(arg, "-O0") == 0){
            optimizationLevel = 0;
            continue;
        }
        else if (strcmp(arg, "-O1") == 0){
            optimizationLevel = 1;
            continue;
        }
        else if (strcmp(arg, "-O2") == 0){
            optimizationLevel = 2;
            continue;
        }

        if (argType == 0){
            filepaths.push_back(arg);
        } else if (argType == 1) {
            outputs.push_back(arg);
        }
    }

    SetLevel(optimizationLevel);

    // Run the main "interpreter loop" now.
    if (filepaths.size() == 0) {
        MainLoop();
    }
    if (filepaths.size() == outputs.size() && filepaths.size() > 0){
        for(int i = 0; i < filepaths.size(); i++) {
            compileFile(filepaths[i], outputs[i]);
        }
    }

    CG::CloseCodegen();
    return 0;
}