From 45409c781a9e35df68c43b1e2f028d30bf90c0a0 Mon Sep 17 00:00:00 2001
From: Patrick Schönberger
Date: Wed, 28 Jul 2021 09:07:53 +0200
Subject: Initial commit

---
 src/toc.h | 274 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 274 insertions(+)
 create mode 100644 src/toc.h

(limited to 'src/toc.h')

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";
+}
-- 
cgit v1.2.3