abouttreesummaryrefslogcommitdiff
path: root/src/toc.h
diff options
context:
space:
mode:
Diffstat (limited to 'src/toc.h')
-rw-r--r--src/toc.h274
1 files changed, 274 insertions, 0 deletions
diff --git a/src/toc.h b/src/toc.h
new file mode 100644
index 0000000..b36d878
--- /dev/null
+++ b/src/toc.h
@@ -0,0 +1,274 @@
+#pragma once
+
+#include <iostream>
+
+#include "TocParser.h"
+
+void toc(std::ostream & o, TocParser::ProgContext * ctx);
+void toc(std::ostream & o, TocParser::VarDeclContext * ctx);
+void toc(std::ostream & o, TocParser::FuncContext * ctx);
+void toc(std::ostream & o, TocParser::StructDeclContext * ctx);
+void toc(std::ostream & o, TocParser::BodyContext * ctx);
+void toc(std::ostream & o, TocParser::StmtContext * ctx);
+void toc(std::ostream & o, TocParser::IfCondContext * ctx);
+void toc(std::ostream & o, TocParser::WhileLoopContext * ctx);
+void toc(std::ostream & o, TocParser::AssignmentContext * ctx);
+void toc(std::ostream & o, TocParser::ReturnStmtContext * ctx);
+void toc(std::ostream & o, TocParser::ExprContext * ctx);
+void toc(std::ostream & o, TocParser::NonOpExprContext * ctx);
+void toc(std::ostream & o, TocParser::NonSubscriptExprContext * ctx);
+void toc(std::ostream & o, TocParser::FuncCallContext * ctx);
+void toc(std::ostream & o, TocParser::IdentifierContext * ctx);
+void toc(std::ostream & o, TocParser::LiteralContext * ctx);
+void toc(std::ostream & o, TocParser::SubscriptContext * ctx);
+void toc(std::ostream & o, TocParser::MemberAccessContext * ctx);
+void toc(std::ostream & o, TocParser::ParenExprContext * ctx);
+void toc(std::ostream & o, TocParser::BinaryOperatorContext * ctx);
+
+void toc_stub(std::ostream & o, TocParser::FuncContext * ctx);
+void toc_stub(std::ostream & o, TocParser::StructDeclContext * ctx);
+
+
+void toc(std::ostream & o, TocParser::ProgContext * ctx) {
+ for (auto * decl : ctx->decl()) {
+ /**/ if (decl->structDecl() != nullptr) toc_stub(o, decl->structDecl());
+ else if (decl->funcDecl() != nullptr) toc_stub(o, decl->funcDecl()->func());
+ }
+ for (auto * decl : ctx->decl()) {
+ if (decl->varDecl() != nullptr) {
+ toc(o, decl->varDecl());
+ o << ";\n";
+ }
+ else if (decl->structDecl() != nullptr) toc(o, decl->structDecl());
+ else if (decl->funcDecl() != nullptr) toc(o, decl->funcDecl()->func());
+ }
+}
+void toc(std::ostream & o, TocParser::VarDeclContext * ctx) {
+ o
+ << ctx->var()->type()->getText()
+ << " "
+ << ctx->var()->varName()->getText();
+
+ if (ctx->var()->expr() != nullptr) {
+ o << " = ";
+ toc(o, ctx->var()->expr());
+ }
+}
+void toc(std::ostream & o, TocParser::FuncContext * ctx) {
+ o
+ << ctx->type()->getText()
+ << " "
+ << ctx->funcName()->getText()
+ << "(";
+
+ if (ctx->parameter()->firstParameter() != nullptr) {
+ o
+ << ctx->parameter()->firstParameter()->var()->type()->getText()
+ << " "
+ << ctx->parameter()->firstParameter()->var()->varName()->getText();
+
+ for (auto * par : ctx->parameter()->additionalParameter()) {
+ o
+ << ", "
+ << par->var()->type()->getText()
+ << " "
+ << par->var()->varName()->getText();
+ }
+ }
+
+ o << ")\n{\n";
+
+ toc(o, ctx->body());
+
+ o << "}\n";
+}
+void toc(std::ostream & o, TocParser::StructDeclContext * ctx) {
+ o
+ << "typedef struct "
+ << ctx->structName()->getText()
+ << "\n{\n";
+
+ for (auto * member : ctx->structMember()) {
+ if (member->structVar() != nullptr) {
+ o
+ << member->structVar()->var()->type()->getText()
+ << " "
+ << member->structVar()->var()->varName()->getText()
+ << ";\n";
+ }
+ }
+ o << "} "
+ << ctx->structName()->getText()
+ << ";\n";
+ for (auto * member : ctx->structMember()) {
+ if (member->structMethod() != nullptr) {
+ o
+ << member->structMethod()->func()->type()->getText()
+ << " "
+ << ctx->structName()->getText()
+ << "_"
+ << member->structMethod()->func()->funcName()->getText()
+ << "("
+ << ctx->structName()->getText()
+ << " * this";
+
+ if (member->structMethod()->func()->parameter()->firstParameter() != nullptr) {
+ o
+ << ", "
+ << member->structMethod()->func()->parameter()->firstParameter()->var()->type()->getText()
+ << " "
+ << member->structMethod()->func()->parameter()->firstParameter()->var()->varName()->getText();
+
+ for (auto * par : member->structMethod()->func()->parameter()->additionalParameter()) {
+ o
+ << ", "
+ << par->var()->type()->getText()
+ << " "
+ << par->var()->varName()->getText();
+ }
+ }
+
+ o << ")\n{\n";
+
+ toc(o, member->structMethod()->func()->body());
+
+ o << "}\n";
+ }
+ }
+}
+void toc(std::ostream & o, TocParser::BodyContext * ctx) {
+ for (auto * stmt : ctx->stmt()) {
+ toc(o, stmt);
+ o << "\n";
+ }
+}
+void toc(std::ostream & o, TocParser::StmtContext * ctx) {
+ /**/ if (ctx->varDecl() != nullptr) toc(o, ctx->varDecl());
+ else if (ctx->conditional() != nullptr) toc(o, ctx->conditional()->ifCond());
+ else if (ctx->loop() != nullptr) toc(o, ctx->loop()->whileLoop());
+ else if (ctx->assignment() != nullptr) toc(o, ctx->assignment());
+ else if (ctx->returnStmt() != nullptr) toc(o, ctx->returnStmt());
+ else if (ctx->expr() != nullptr) toc(o, ctx->expr());
+
+ if (ctx->conditional() == nullptr && ctx->loop() == nullptr)
+ o << ";";
+}
+void toc(std::ostream & o, TocParser::IfCondContext * ctx) {
+ o << "if (";
+ toc(o, ctx->expr());
+ o << ")\n{\n";
+ toc(o, ctx->body());
+ o << "}\n";
+}
+void toc(std::ostream & o, TocParser::WhileLoopContext * ctx) {
+ o << "while (";
+ toc(o, ctx->expr());
+ o << ")\n{\n";
+ toc(o, ctx->body());
+ o << "}\n";
+}
+void toc(std::ostream & o, TocParser::AssignmentContext * ctx) {
+ toc(o, ctx->identifier());
+ o << " = ";
+ toc(o, ctx->expr());
+}
+void toc(std::ostream & o, TocParser::ReturnStmtContext * ctx) {
+ o << "return ";
+ toc(o, ctx->expr());
+}
+void toc(std::ostream & o, TocParser::ExprContext * ctx) {
+ /**/ if (ctx->funcCall() != nullptr) toc(o, ctx->funcCall());
+ else if (ctx->identifier() != nullptr) toc(o, ctx->identifier());
+ else if (ctx->literal() != nullptr) toc(o, ctx->literal());
+ else if (ctx->subscript() != nullptr) toc(o, ctx->subscript());
+ else if (ctx->memberAccess() != nullptr) toc(o, ctx->memberAccess());
+ else if (ctx->parenExpr() != nullptr) toc(o, ctx->parenExpr());
+ else if (ctx->operatorExpr() != nullptr) toc(o, ctx->operatorExpr()->binaryOperator());
+}
+void toc(std::ostream & o, TocParser::NonOpExprContext * ctx) {
+ /**/ if (ctx->funcCall() != nullptr) toc(o, ctx->funcCall());
+ else if (ctx->identifier() != nullptr) toc(o, ctx->identifier());
+ else if (ctx->literal() != nullptr) toc(o, ctx->literal());
+ else if (ctx->subscript() != nullptr) toc(o, ctx->subscript());
+ else if (ctx->memberAccess() != nullptr) toc(o, ctx->memberAccess());
+ else if (ctx->parenExpr() != nullptr) toc(o, ctx->parenExpr());
+}
+void toc(std::ostream & o, TocParser::NonSubscriptExprContext * ctx) {
+ /**/ if (ctx->funcCall() != nullptr) toc(o, ctx->funcCall());
+ else if (ctx->identifier() != nullptr) toc(o, ctx->identifier());
+ else if (ctx->memberAccess() != nullptr) toc(o, ctx->memberAccess());
+ else if (ctx->parenExpr() != nullptr) toc(o, ctx->parenExpr());
+}
+void toc(std::ostream & o, TocParser::FuncCallContext * ctx) {
+ o
+ << ctx->funcName()->getText()
+ << "(";
+ for (int i = 0; i < ctx->expr().size(); i++) {
+ if (i != 0) o << ", ";
+ toc(o, ctx->expr(i));
+ }
+ o << ")";
+}
+void toc(std::ostream & o, TocParser::IdentifierContext * ctx) {
+ o << ctx->getText();
+}
+void toc(std::ostream & o, TocParser::LiteralContext * ctx) {
+ if (ctx->INTLIT() != nullptr) o << ctx->INTLIT()->getText();
+}
+void toc(std::ostream & o, TocParser::SubscriptContext * ctx) {
+ toc(o, ctx->nonSubscriptExpr());
+ o << "[";
+ toc(o, ctx->expr());
+ o << "]";
+}
+void toc(std::ostream & o, TocParser::MemberAccessContext * ctx) {
+ toc(o, ctx->identifier(0));
+ o << ".";
+ toc(o, ctx->identifier(1));
+}
+void toc(std::ostream & o, TocParser::ParenExprContext * ctx) {
+ o << "(";
+ toc(o, ctx->expr());
+ o << ")";
+}
+void toc(std::ostream & o, TocParser::BinaryOperatorContext * ctx) {
+ for (int i = 0; i < ctx->BINARY_OPERATOR().size(); i++) {
+ toc(o, ctx->nonOpExpr(i));
+ o
+ << " "
+ << ctx->BINARY_OPERATOR(i)->getText()
+ << " ";
+ toc(o, ctx->nonOpExpr(i + 1));
+ }
+}
+
+void toc_stub(std::ostream & o, TocParser::FuncContext * ctx) {
+ o
+ << ctx->type()->getText()
+ << " "
+ << ctx->funcName()->getText()
+ << "(";
+
+ if (ctx->parameter()->firstParameter() != nullptr) {
+ o
+ << ctx->parameter()->firstParameter()->var()->type()->getText()
+ << " "
+ << ctx->parameter()->firstParameter()->var()->varName()->getText();
+
+ for (auto * par : ctx->parameter()->additionalParameter()) {
+ o
+ << ", "
+ << par->var()->type()->getText()
+ << " "
+ << par->var()->varName()->getText();
+ }
+ }
+
+ o << ");\n";
+}
+void toc_stub(std::ostream & o, TocParser::StructDeclContext * ctx) {
+ o
+ << "struct "
+ << ctx->structName()->getText()
+ << ";\n";
+}
generated by ps-cgit (a fork of cgit), using a theme by hackitude