X-Git-Url: https://gitweb.ps.run/toc/blobdiff_plain/45409c781a9e35df68c43b1e2f028d30bf90c0a0..7f83e1b208e87e3808b268303bb633a8fda203f5:/src/toc.h

diff --git a/src/toc.h b/src/toc.h
index b36d878..43ab1ba 100644
--- a/src/toc.h
+++ b/src/toc.h
@@ -1,274 +1,324 @@
 #pragma once
 
 #include <iostream>
+#include <sstream>
 
-#include "TocParser.h"
+#include "repr.h"
+#include "typeInfo.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);
+template<typename T>
+std::string vectorStr (const std::vector<T> & v, const std::string & separator, bool end = false)
+{
+  std::stringstream sstr;
 
-void toc_stub(std::ostream & o, TocParser::FuncContext * ctx);
-void toc_stub(std::ostream & o, TocParser::StructDeclContext * ctx);
+  bool putSeparator = false;
+  for (auto t : v)
+  {
+    if (putSeparator) sstr << separator;
+    else putSeparator = true;
+    sstr << t;
+  }
+  if (end && !v.empty())
+    sstr << separator;
 
+  return sstr.str();
+}
 
-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());
-  }
+std::ostream & operator<< (std::ostream & out, const Type & t);
+std::ostream & operator<< (std::ostream & out, const Variable & v);
+std::ostream & operator<< (std::ostream & out, const Body & b);
+std::ostream & operator<< (std::ostream & out, const Expr & e);
+std::ostream & operator<< (std::ostream & out, const Stmt & s);
+
+void tocFunction (std::ostream & out, const Function & f, bool stub);
+void tocStruct   (std::ostream & out, const Struct & s, bool stub);
+void tocProgram  (std::ostream & out, const Program & p);
+void tocNamespace  (std::ostream & out, const Namespace & n, bool stub);
+
+static const int TAB_WIDTH = 2;
+static int indentation = 0;
+static void indent(std::ostream & out, int change = 0)
+{
+  indentation += change;
+  out << std::string(indentation, ' ');
 }
-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());
+
+static std::vector<std::string> namespaces;
+static std::string namespacePrefix() {
+  std::stringstream sstr;
+  for (auto n : namespaces)
+  {
+    sstr << n << "_";
   }
+  return sstr.str();
+}
+
+static Program globalPrg;
+
+std::ostream & operator<< (std::ostream & out, const Type & t)
+{
+  out << vectorStr(t.namespacePrefixes, "_", true) << t.name;
+
+  return out;
 }
-void toc(std::ostream & o, TocParser::FuncContext * ctx) {
-  o
-    << ctx->type()->getText()
-    << " "
-    << ctx->funcName()->getText()
-    << "(";
+std::ostream & operator<< (std::ostream & out, const Variable & v)
+{
+  out << v.type << " ";
 
-  if (ctx->parameter()->firstParameter() != nullptr) {
-    o
-      << ctx->parameter()->firstParameter()->var()->type()->getText()
-      << " "
-      << ctx->parameter()->firstParameter()->var()->varName()->getText();
+  std::stringstream sstr;
+  std::string s = v.name;
 
-    for (auto * par : ctx->parameter()->additionalParameter()) {
-      o
-        << ", "
-        << par->var()->type()->getText()
-        << " "
-        << par->var()->varName()->getText();
+  for (auto m = v.type.modifiers.rbegin(); m != v.type.modifiers.rend(); m++)
+  {
+    if (m->type == TypeModifierType::Pointer)
+    {
+      sstr.str(std::string());
+      sstr << "*(" << s << ")";
+      s = sstr.str();
+    }
+    else
+    {
+      sstr.str(std::string());
+      sstr << "(" << s << ")[";
+      if (m->_staticArray)
+        sstr << m->_arraySize;
+      sstr << "]";
+      s = sstr.str();
     }
   }
+  out << s;
 
-  o << ")\n{\n";
+  return out;
+}
+std::ostream & operator<< (std::ostream & out, const Body & b)
+{
+  indent(out);
+  out << "{\n";
+  indentation += 2;
 
-  toc(o, ctx->body());
+  for (auto v : b.ctx->variables)
+  {
+    indent(out);
+    out << v << ";\n";
+  }
 
-  o << "}\n";
+  out << "\n";
+  
+  for (auto s : b.statements)
+  {
+    indent(out);
+    out << s << "\n";
+  }
+
+  indent(out, -2);
+  out << "}\n";
+
+  return out;
 }
-void toc(std::ostream & o, TocParser::StructDeclContext * ctx) {
-  o
-    << "typedef struct "
-    << ctx->structName()->getText()
-    << "\n{\n";
+std::ostream & operator<< (std::ostream & out, const Expr & e)
+{
+  switch (e.type)
+  {
+  case ExprType::Func:
+    out << vectorStr(e._func.namespacePrefixes, "_", true) << e._func.functionName << "(" << vectorStr(e._func.arguments, ", ") << ")"; break;
+  case ExprType::Method:
+  {
+    TypeInfo ti = typeExpr(globalPrg, namespaces, *e._method.expr);
+    out <<
+      vectorStr(ti.type.namespacePrefixes, "_", true) <<
+      ti.type.name << "_" << e._method.methodName << "(" << *e._method.expr << vectorStr(e._method.arguments, ", ") << ")"; break;
+  }
+  case ExprType::Lit:
+    /**/ if (e._lit.type == LitType::Int) out << e._lit._int;
+    else if (e._lit.type == LitType::Decimal) out << e._lit._decimal;
+    else if (e._lit.type == LitType::String) out << e._lit._string;
+    else if (e._lit.type == LitType::Bool) out << e._lit._bool;
+    break;
+  case ExprType::Paren:
+    out << "(" << e._paren.expr << ")"; break;
+  case ExprType::Dot:
+    out << *e._dot.expr << "." << e._dot.identifier; break;
+  case ExprType::PrefixOp:
+    out << PrefixOperatorTypeStrings[(int)e._prefixOp.type] << *e._prefixOp.expr; break;
+  case ExprType::PostfixOp:
+    out << *e._postfixOp.expr << PostfixOperatorTypeStrings[(int)e._postfixOp.type]; break;
+  case ExprType::BinaryOp:
+    out << *e._binaryOp.lexpr <<
+    " " << BinaryOperatorTypeStrings[(int)e._binaryOp.type] << " " <<
+    *e._binaryOp.rexpr; break;
+  case ExprType::TernaryOp:
+    out << *e._ternaryOp.lexpr <<
+      " ? " << *e._ternaryOp.rexprTrue <<
+      " : " << *e._ternaryOp.rexprFalse; break;
+  case ExprType::Bracket:
+    out << *e._brackets.lexpr << "[" << *e._brackets.rexpr << "]"; break;
+  case ExprType::Identifier:
+    out << vectorStr(e._identifier.namespacePrefixes, "_", true) << e._identifier.identifier; break;
+  }
 
-  for (auto * member : ctx->structMember()) {
-    if (member->structVar() != nullptr) {
-      o
-        << member->structVar()->var()->type()->getText()
-        << " "
-        << member->structVar()->var()->varName()->getText()
-        << ";\n";
+  return out;
+}
+std::ostream & operator<< (std::ostream & out, const Stmt & s)
+{
+  switch (s.type)
+  {
+  case StmtType::If:
+    out << "if (" << s._if.condition << ")\n" << s._if.body; break;
+  case StmtType::Switch:
+    out << "switch (" << s._switch.ident << ")\n{\n";
+    for (auto c : s._switch.cases)
+    {
+      indent(out, 2);
+      out << "case " << *c.expr << ": " << c.body << "break;";
     }
+    indent(out, -2);
+    out << "}\n";
+    break;
+  case StmtType::For:
+    out << "for (" <<
+      s._for.init << "; " <<
+      *s._for.condition << "; " <<
+      *s._for.action <<
+      ")\n" << s._for.body; break;
+  case StmtType::While:
+    out << "while (" << s._while.condition << ")\n" << s._while.body; break;
+  case StmtType::Assign:
+    out << s._assign.lexpr << " = " << s._assign.rexpr << ";"; break;
+  case StmtType::Return:
+    out << "return " << s._return.expr << ";"; break;
+  case StmtType::Expr:
+    out << s._expr << ";"; break;
   }
-  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();
+  return out;
+}
 
-        for (auto * par : member->structMethod()->func()->parameter()->additionalParameter()) {
-          o
-            << ", "
-            << par->var()->type()->getText()
-            << " "
-            << par->var()->varName()->getText();
-        }
-      }
 
-      o << ")\n{\n";
+void tocFunction (std::ostream & out, const Function & f, bool stub)
+{
+  out << f.returnType << " " << namespacePrefix() << f.name << " (" << vectorStr(f.parameters, ", ") << ")";
 
-      toc(o, member->structMethod()->func()->body());
+  if (stub)
+  {
+    out << ";\n";
+  }
+  else
+  {
+    out << "\n" << f.body;
+  }
+}
+void tocStruct (std::ostream & out, const Struct & s, bool stub)
+{
+  out << "struct " << namespacePrefix() << s.name;
+  if (stub)
+  {
+    out << ";\n";
+    for (auto m : s.methods)
+    {
+      Function f = m;
 
-      o << "}\n";
+      f.parameters.insert(f.parameters.begin(),
+      {"this",
+        {
+          namespaces,
+          s.name,
+          {
+            {TypeModifierType::Pointer, false, -1}
+          }
+        }
+      });
+      out << f.returnType << " " <<
+        namespacePrefix() << s.name << "_" << f.name <<
+        " (" << vectorStr(f.parameters, ", ") << ");\n";
     }
+    return;
   }
-}
-void toc(std::ostream & o, TocParser::BodyContext * ctx) {
-  for (auto * stmt : ctx->stmt()) {
-    toc(o, stmt);
-    o << "\n";
+  out << "\n{\n";
+  indentation += 2;
+
+  for (auto m : s.members)
+  {
+    indent(out);
+    out << m << ";\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));
+  indent(out, -2);
+  out << "};\n";
+  
+  for (auto m : s.methods)
+  {
+    Function f = m;
+    f.parameters.insert(f.parameters.begin(),
+      {"this",
+        {
+          namespaces,
+          s.name,
+          {
+            {TypeModifierType::Pointer, false, -1}
+          }
+        }
+      });
+    out << f.returnType << " " <<
+    namespacePrefix() << s.name << "_" << f.name <<
+    " (" << vectorStr(f.parameters, ", ") << ")\n" << f.body;
   }
-  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 tocProgram (std::ostream & out, const Program & p)
+{
+  globalPrg = p;
+  for (auto n : p.namespaces)
+  {
+    tocNamespace(out, n, true);
+  }
+  for (auto s : p.structs)
+  {
+    tocStruct(out, s, true);
+  }
+  for (auto f : p.functions)
+  {
+    tocFunction(out, f, true);
   }
-}
 
-void toc_stub(std::ostream & o, TocParser::FuncContext * ctx) {
-  o
-    << ctx->type()->getText()
-    << " "
-    << ctx->funcName()->getText()
-    << "(";
+  for (auto v : p.ctx->variables)
+  {
+    out << v << ";\n";
+  }
+  for (auto n : p.namespaces)
+  {
+    tocNamespace(out, n, false);
+  }
+  for (auto s : p.structs)
+  {
+    tocStruct(out, s, false);
+  }
+  for (auto f : p.functions)
+  {
+    tocFunction(out, f, false);
+  }
+}
 
-  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();
+void tocNamespace  (std::ostream & out, const Namespace & n, bool stub)
+{
+  namespaces.push_back(n.name);
+  if (!stub)
+  {
+    for (auto v : n.ctx->variables)
+    {
+      out << v << ";\n";
     }
   }
-
-  o << ");\n";
-}
-void toc_stub(std::ostream & o, TocParser::StructDeclContext * ctx) {
-  o
-    << "struct "
-    << ctx->structName()->getText()
-    << ";\n";
-}
+  for (auto n : n.namespaces)
+  {
+    tocNamespace(out, n, stub);
+  }
+  for (auto s : n.structs)
+  {
+    tocStruct(out, s, stub);
+  }
+  for (auto f : n.functions)
+  {
+    tocFunction(out, f, stub);
+  }
+  namespaces.pop_back();
+}
\ No newline at end of file