81 lines
2.5 KiB
C++
81 lines
2.5 KiB
C++
#include <iostream>
|
|
#include <fstream>
|
|
#include <filesystem>
|
|
|
|
#include <llvm/Support/CommandLine.h>
|
|
|
|
#include "Lexer/Token.h"
|
|
#include "Lexer/Lexer.h"
|
|
|
|
#include "Parser/Parser.h"
|
|
#include "Parser/Statement/Statement.h"
|
|
|
|
#include "Compiler/ModuleBuilder.h"
|
|
#include "Compiler/CodeGenerator.h"
|
|
|
|
#include "Logger.h"
|
|
|
|
using namespace std;
|
|
|
|
string readFile(filesystem::path filePath) {
|
|
ifstream file(filePath, ios::in | ios::binary | ios::ate);
|
|
if (!file.is_open()) {
|
|
cerr << "Cannot open file " << filePath << endl;
|
|
exit(1);
|
|
}
|
|
|
|
streamsize fileSize = file.tellg();
|
|
file.seekg(0, ios::beg);
|
|
vector<char> fileBytes(fileSize);
|
|
file.read(fileBytes.data(), fileSize);
|
|
return string(fileBytes.data(), fileSize);
|
|
}
|
|
|
|
void versionPrinter(llvm::raw_ostream &os) {
|
|
os << "Bits Runner Code, Version 1.0.0 (pre-alpha)\n";
|
|
}
|
|
|
|
int main(int argc, char **argv) {
|
|
llvm::cl::SetVersionPrinter(versionPrinter);
|
|
llvm::cl::extrahelp("\nADDITIONAL HELP:\n\n This is the extra help\n");
|
|
llvm::cl::opt<bool> isVerbose("v", llvm::cl::desc("Verbos output"), llvm::cl::init(false));
|
|
llvm::cl::opt<CodeGenerator::OutputKind> outputKind(
|
|
llvm::cl::desc("Choose generated output:"),
|
|
llvm::cl::init(CodeGenerator::OutputKind::OBJECT),
|
|
llvm::cl::values(
|
|
clEnumValN(CodeGenerator::OutputKind::OBJECT, "c", "Generate object file"),
|
|
clEnumValN(CodeGenerator::OutputKind::ASSEMBLY, "S", "Generate assembly file")
|
|
)
|
|
);
|
|
llvm::cl::opt<string> inputFileName(llvm::cl::Positional, llvm::cl::desc("<input file>"), llvm::cl::Required);
|
|
llvm::cl::ParseCommandLineOptions(argc, argv, "Bits Runner Builder - LLVM based compiler for the Bits Runner Code language");
|
|
|
|
filesystem::path inputFilePath((string(inputFileName)));
|
|
string source = readFile(inputFilePath);
|
|
string moduleName = inputFilePath.filename().replace_extension();
|
|
|
|
Lexer lexer(source);
|
|
vector<shared_ptr<Token>> tokens = lexer.getTokens();
|
|
if (isVerbose) {
|
|
Logger::print(tokens);
|
|
cout << endl;
|
|
}
|
|
|
|
Parser parser(tokens);
|
|
vector<shared_ptr<Statement>> statements = parser.getStatements();
|
|
if (isVerbose) {
|
|
Logger::print(statements);
|
|
cout << endl;
|
|
}
|
|
|
|
ModuleBuilder moduleBuilder(moduleName, inputFilePath, statements);
|
|
shared_ptr<llvm::Module> module = moduleBuilder.getModule();
|
|
if (isVerbose) {
|
|
module->print(llvm::outs(), nullptr);
|
|
}
|
|
|
|
CodeGenerator codeGenerator(module);
|
|
codeGenerator.generateObjectFile(outputKind);
|
|
|
|
return 0;
|
|
} |