AIR/cpp-peglib
1
0
Fork 0
mirror of https://github.com/yhirose/cpp-peglib.git synced 2024-12-22 20:05:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Cleanup pl0.cc

Browse Source
This commit is contained in:
yhirose 2017-08-17 17:44:29 -04:00
parent dcdfd9fdb7
commit 6ee4326723
1 changed files with 27 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

63
pl0/pl0.cc
View File

@ -1,5 +1,5 @@
// //
// pl0.cc - PL/0 interpreter (https://en.wikipedia.org/wiki/PL/0) // pl0.cc - PL/0 language (https://en.wikipedia.org/wiki/PL/0)
// //
// Copyright (c) 2015 Yuji Hirose. All rights reserved. // Copyright (c) 2015 Yuji Hirose. All rights reserved.
// MIT License // MIT License
@ -79,7 +79,6 @@ struct SymbolScope;
struct Annotation { struct Annotation {
shared_ptr<SymbolScope> scope; shared_ptr<SymbolScope> scope;
shared_ptr<vector<string>> freeVariables;
}; };
typedef AstBase<Annotation> AstPL0; typedef AstBase<Annotation> AstPL0;
@ -538,20 +537,10 @@ struct LLVM {
void exec() { void exec() {
unique_ptr<ExecutionEngine> ee(EngineBuilder(std::move(module_)).create()); unique_ptr<ExecutionEngine> ee(EngineBuilder(std::move(module_)).create());
std::vector<GenericValue> noargs; std::vector<GenericValue> noargs;
auto fn = ee->FindFunctionNamed("__main__"); auto fn = ee->FindFunctionNamed("main");
auto ret = ee->runFunction(fn, noargs); auto ret = ee->runFunction(fn, noargs);
} }
static void dump(const shared_ptr<AstPL0> ast) {
LLVM compiler(ast);
compiler.dump();
}
static void exec(const shared_ptr<AstPL0> ast) {
LLVM compiler(ast);
compiler.exec();
}
private: private:
LLVMContext context_; LLVMContext context_;
IRBuilder<> builder_; IRBuilder<> builder_;
@ -631,7 +620,7 @@ struct LLVM {
void compile_program(const shared_ptr<AstPL0> ast) { void compile_program(const shared_ptr<AstPL0> ast) {
auto fn = cast<Function>(module_->getOrInsertFunction( auto fn = cast<Function>(module_->getOrInsertFunction(
"__main__", builder_.getVoidTy(), nullptr)); "main", builder_.getVoidTy(), nullptr));
{ {
auto BB = BasicBlock::Create(context_, "entry", fn); auto BB = BasicBlock::Create(context_, "entry", fn);
builder_.SetInsertPoint(BB); builder_.SetInsertPoint(BB);
@ -901,14 +890,31 @@ struct LLVM {
*/ */
int main(int argc, const char** argv) { int main(int argc, const char** argv) {
if (argc < 2) { if (argc < 2) {
cout << "usage: pl0 PATH [--ast]" << endl; cout << "usage: pl0 PATH [--ast] [--llvm] [--jit]" << endl;
return 1; return 1;
} }
// Read a source file into memory // Parser commandline parameters
auto path = argv[1]; auto path = argv[1];
vector<char> source; bool opt_jit = false;
bool opt_ast = false;
bool opt_llvm = false;
{
auto argi = 2;
while (argi < argc) {
if (string("--ast") == argv[argi]) {
opt_ast = true;
} else if (string("--jit") == argv[argi]) {
opt_jit = true;
} else if (string("--llvm") == argv[argi]) {
opt_llvm = true;
}
argi++;
}
}
// Read a source file into memory
vector<char> source;
ifstream ifs(path, ios::in | ios::binary); ifstream ifs(path, ios::in | ios::binary);
if (ifs.fail()) { if (ifs.fail()) {
cerr << "can't open the source file." << endl; cerr << "can't open the source file." << endl;
@ -930,23 +936,6 @@ int main(int argc, const char** argv) {
// Parse the source and make an AST // Parse the source and make an AST
shared_ptr<AstPL0> ast; shared_ptr<AstPL0> ast;
if (parser.parse_n(source.data(), source.size(), ast, path)) { if (parser.parse_n(source.data(), source.size(), ast, path)) {
bool opt_jit = false;
bool opt_ast = false;
bool opt_llvm = false;
{
auto argi = 2;
while (argi < argc) {
if (string("--ast") == argv[argi]) {
opt_ast = true;
} else if (string("--jit") == argv[argi]) {
opt_jit = true;
} else if (string("--llvm") == argv[argi]) {
opt_llvm = true;
}
argi++;
}
}
try { try {
SymbolTable::build_on_ast(ast); SymbolTable::build_on_ast(ast);
@ -955,11 +944,13 @@ int main(int argc, const char** argv) {
} }
if (opt_llvm || opt_jit) { if (opt_llvm || opt_jit) {
LLVM compiler(ast);
if (opt_llvm) { if (opt_llvm) {
LLVM::dump(ast); compiler.dump();
} }
if (opt_jit) { if (opt_jit) {
LLVM::exec(ast); compiler.exec();
} }
} else { } else {
Interpreter::exec(ast); Interpreter::exec(ast);