pre change

[toc] / src / toc.h

diff --git a/src/toc.h b/src/toc.h
index b36d878cc580c33d673e1ed0410b4b13de83664c..95c24d8beb795b7926e7705dacd11f0de7bc7723 100644 (file)
--- a/src/toc.h
+++ b/src/toc.h
@@ -1,274 +1,324 @@
 #pragma once\r
 \r
 #include <iostream>\r
 #pragma once\r
 \r
 #include <iostream>\r

+#include <sstream>\r

 \r
 \r

-#include "TocParser.h"\r
+#include "repr.h"\r
+#include "typeInfo.h"\r

 \r
 \r

-void toc(std::ostream & o, TocParser::ProgContext * ctx);\r
-void toc(std::ostream & o, TocParser::VarDeclContext * ctx);\r
-void toc(std::ostream & o, TocParser::FuncContext * ctx);\r
-void toc(std::ostream & o, TocParser::StructDeclContext * ctx);\r
-void toc(std::ostream & o, TocParser::BodyContext * ctx);\r
-void toc(std::ostream & o, TocParser::StmtContext * ctx);\r
-void toc(std::ostream & o, TocParser::IfCondContext * ctx);\r
-void toc(std::ostream & o, TocParser::WhileLoopContext * ctx);\r
-void toc(std::ostream & o, TocParser::AssignmentContext * ctx);\r
-void toc(std::ostream & o, TocParser::ReturnStmtContext * ctx);\r
-void toc(std::ostream & o, TocParser::ExprContext * ctx);\r
-void toc(std::ostream & o, TocParser::NonOpExprContext * ctx);\r
-void toc(std::ostream & o, TocParser::NonSubscriptExprContext * ctx);\r
-void toc(std::ostream & o, TocParser::FuncCallContext * ctx);\r
-void toc(std::ostream & o, TocParser::IdentifierContext * ctx);\r
-void toc(std::ostream & o, TocParser::LiteralContext * ctx);\r
-void toc(std::ostream & o, TocParser::SubscriptContext * ctx);\r
-void toc(std::ostream & o, TocParser::MemberAccessContext * ctx);\r
-void toc(std::ostream & o, TocParser::ParenExprContext * ctx);\r
-void toc(std::ostream & o, TocParser::BinaryOperatorContext * ctx);\r
+template<typename T>\r
+std::string vectorStr (const std::vector<T> & v, const std::string & separator, bool end = false)\r
+{\r
+  std::stringstream sstr;\r

 \r
 \r

-void toc_stub(std::ostream & o, TocParser::FuncContext * ctx);\r
-void toc_stub(std::ostream & o, TocParser::StructDeclContext * ctx);\r
+  bool putSeparator = false;\r
+  for (auto t : v)\r
+  {\r
+    if (putSeparator) sstr << separator;\r
+    else putSeparator = true;\r
+    sstr << t;\r
+  }\r
+  if (end && !v.empty())\r
+    sstr << separator;\r

 \r
 \r

+  return sstr.str();\r
+}\r

 \r
 \r

-void toc(std::ostream & o, TocParser::ProgContext * ctx) {\r
-  for (auto * decl : ctx->decl()) {\r
-    /**/ if (decl->structDecl() != nullptr) toc_stub(o, decl->structDecl());\r
-    else if (decl->funcDecl()   != nullptr) toc_stub(o, decl->funcDecl()->func());\r
-  }\r
-  for (auto * decl : ctx->decl()) {\r
-    if (decl->varDecl()    != nullptr) {\r
-      toc(o, decl->varDecl());\r
-      o << ";\n";\r
-    }\r
-    else if (decl->structDecl() != nullptr) toc(o, decl->structDecl());\r
-    else if (decl->funcDecl()   != nullptr) toc(o, decl->funcDecl()->func());\r
-  }\r
+std::ostream & operator<< (std::ostream & out, const Type & t);\r
+std::ostream & operator<< (std::ostream & out, const Variable & v);\r
+std::ostream & operator<< (std::ostream & out, const Body & b);\r
+std::ostream & operator<< (std::ostream & out, const Expr & e);\r
+std::ostream & operator<< (std::ostream & out, const Stmt & s);\r
+\r
+void tocFunction (std::ostream & out, const Function & f, bool stub);\r
+void tocStruct   (std::ostream & out, const Struct & s, bool stub);\r
+void tocProgram  (std::ostream & out, const Program & p);\r
+void tocNamespace  (std::ostream & out, const Namespace & n, bool stub);\r
+\r
+static const int TAB_WIDTH = 2;\r
+static int indentation = 0;\r
+static void indent(std::ostream & out, int change = 0)\r
+{\r
+  indentation += change;\r
+  out << std::string(indentation, ' ');\r

 }\r
 }\r

-void toc(std::ostream & o, TocParser::VarDeclContext * ctx) {\r
-  o\r
-    << ctx->var()->type()->getText()\r
-    << " "\r
-    << ctx->var()->varName()->getText();\r
-  \r
-  if (ctx->var()->expr() != nullptr) {\r
-    o << " = ";\r
-    toc(o, ctx->var()->expr());\r
+\r
+static std::vector<std::string> namespaces;\r
+static std::string namespacePrefix() {\r
+  std::stringstream sstr;\r
+  for (auto n : namespaces)\r
+  {\r
+    sstr << n << "_";\r

   }\r
   }\r

+  return sstr.str();\r
+}\r
+\r
+static Program globalPrg;\r
+\r
+std::ostream & operator<< (std::ostream & out, const Type & t)\r
+{\r
+  out << vectorStr(t.namespacePrefixes, "_", true) << t.name;\r
+\r
+  return out;\r

 }\r
 }\r

-void toc(std::ostream & o, TocParser::FuncContext * ctx) {\r
-  o\r
-    << ctx->type()->getText()\r
-    << " "\r
-    << ctx->funcName()->getText()\r
-    << "(";\r
+std::ostream & operator<< (std::ostream & out, const Variable & v)\r
+{\r
+  out << v.type << " ";\r

 \r
 \r

-  if (ctx->parameter()->firstParameter() != nullptr) {\r
-    o\r
-      << ctx->parameter()->firstParameter()->var()->type()->getText()\r
-      << " "\r
-      << ctx->parameter()->firstParameter()->var()->varName()->getText();\r
+  std::stringstream sstr;\r
+  std::string s = v.name;\r

 \r
 \r

-    for (auto * par : ctx->parameter()->additionalParameter()) {\r
-      o\r
-        << ", "\r
-        << par->var()->type()->getText()\r
-        << " "\r
-        << par->var()->varName()->getText();\r
+  for (auto m = v.type.modifiers.rbegin(); m != v.type.modifiers.rend(); m++)\r
+  {\r
+    if (m->type == TypeModifierType::Pointer)\r
+    {\r
+      sstr.str(std::string());\r
+      sstr << "*(" << s << ")";\r
+      s = sstr.str();\r
+    }\r
+    else\r
+    {\r
+      sstr.str(std::string());\r
+      sstr << "(" << s << ")[";\r
+      if (m->_staticArray)\r
+        sstr << m->_arraySize;\r
+      sstr << "]";\r
+      s = sstr.str();\r

     }\r
   }\r
     }\r
   }\r

+  out << s;\r

 \r
 \r

-  o << ")\n{\n";\r
+  return out;\r
+}\r
+std::ostream & operator<< (std::ostream & out, const Body & b)\r
+{\r
+  indent(out);\r
+  out << "{\n";\r
+  indentation += 2;\r

 \r
 \r

-  toc(o, ctx->body());\r
+  for (auto v : b.variables)\r
+  {\r
+    indent(out);\r
+    out << v << ";\n";\r
+  }\r

 \r
 \r

-  o << "}\n";\r
+  out << "\n";\r
+  \r
+  for (auto s : b.statements)\r
+  {\r
+    indent(out);\r
+    out << s << "\n";\r
+  }\r
+\r
+  indent(out, -2);\r
+  out << "}\n";\r
+\r
+  return out;\r

 }\r
 }\r

-void toc(std::ostream & o, TocParser::StructDeclContext * ctx) {\r
-  o\r
-    << "typedef struct "\r
-    << ctx->structName()->getText()\r
-    << "\n{\n";\r
+std::ostream & operator<< (std::ostream & out, const Expr & e)\r
+{\r
+  switch (e.type)\r
+  {\r
+  case ExprType::Func:\r
+    out << vectorStr(e._func.namespacePrefixes, "_", true) << e._func.functionName << "(" << vectorStr(e._func.arguments, ", ") << ")"; break;\r
+  case ExprType::Method:\r
+  {\r
+    TypeInfo ti = typeExpr(globalPrg, namespaces, *e._method.expr);\r
+    out <<\r
+      vectorStr(ti.type.namespacePrefixes, "_", true) <<\r
+      ti.type.name << "_" << e._method.methodName << "(" << *e._method.expr << vectorStr(e._method.arguments, ", ") << ")"; break;\r
+  }\r
+  case ExprType::Lit:\r
+    /**/ if (e._lit.type == LitType::Int) out << e._lit._int;\r
+    else if (e._lit.type == LitType::Decimal) out << e._lit._decimal;\r
+    else if (e._lit.type == LitType::String) out << e._lit._string;\r
+    else if (e._lit.type == LitType::Bool) out << e._lit._bool;\r
+    break;\r
+  case ExprType::Paren:\r
+    out << "(" << e._paren.expr << ")"; break;\r
+  case ExprType::Dot:\r
+    out << *e._dot.expr << "." << e._dot.identifier; break;\r
+  case ExprType::PrefixOp:\r
+    out << PrefixOperatorTypeStrings[(int)e._prefixOp.type] << *e._prefixOp.expr; break;\r
+  case ExprType::PostfixOp:\r
+    out << *e._postfixOp.expr << PostfixOperatorTypeStrings[(int)e._postfixOp.type]; break;\r
+  case ExprType::BinaryOp:\r
+    out << *e._binaryOp.lexpr <<\r
+    " " << BinaryOperatorTypeStrings[(int)e._binaryOp.type] << " " <<\r
+    *e._binaryOp.rexpr; break;\r
+  case ExprType::TernaryOp:\r
+    out << *e._ternaryOp.lexpr <<\r
+      " ? " << *e._ternaryOp.rexprTrue <<\r
+      " : " << *e._ternaryOp.rexprFalse; break;\r
+  case ExprType::Bracket:\r
+    out << *e._brackets.lexpr << "[" << *e._brackets.rexpr << "]"; break;\r
+  case ExprType::Identifier:\r
+    out << vectorStr(e._identifier.namespacePrefixes, "_", true) << e._identifier.identifier; break;\r
+  }\r

 \r
 \r

-  for (auto * member : ctx->structMember()) {\r
-    if (member->structVar() != nullptr) {\r
-      o\r
-        << member->structVar()->var()->type()->getText()\r
-        << " "\r
-        << member->structVar()->var()->varName()->getText()\r
-        << ";\n";\r
+  return out;\r
+}\r
+std::ostream & operator<< (std::ostream & out, const Stmt & s)\r
+{\r
+  switch (s.type)\r
+  {\r
+  case StmtType::If:\r
+    out << "if (" << s._if.condition << ")\n" << s._if.body; break;\r
+  case StmtType::Switch:\r
+    out << "switch (" << s._switch.ident << ")\n{\n";\r
+    for (auto c : s._switch.cases)\r
+    {\r
+      indent(out, 2);\r
+      out << "case " << *c.expr << ": " << c.body << "break;";\r

     }\r
     }\r

+    indent(out, -2);\r
+    out << "}\n";\r
+    break;\r
+  case StmtType::For:\r
+    out << "for (" <<\r
+      s._for.init << "; " <<\r
+      *s._for.condition << "; " <<\r
+      *s._for.action <<\r
+      ")\n" << s._for.body; break;\r
+  case StmtType::While:\r
+    out << "while (" << s._while.condition << ")\n" << s._while.body; break;\r
+  case StmtType::Assign:\r
+    out << s._assign.lexpr << " = " << s._assign.rexpr << ";"; break;\r
+  case StmtType::Return:\r
+    out << "return " << s._return.expr << ";"; break;\r
+  case StmtType::Expr:\r
+    out << s._expr << ";"; break;\r

   }\r
   }\r

-  o << "} "\r
-    << ctx->structName()->getText()\r
-    << ";\n";\r
-  for (auto * member : ctx->structMember()) {\r
-    if (member->structMethod() != nullptr) {\r
-      o\r
-        << member->structMethod()->func()->type()->getText()\r
-        << " "\r
-        << ctx->structName()->getText()\r
-        << "_"\r
-        << member->structMethod()->func()->funcName()->getText()\r
-        << "("\r
-        << ctx->structName()->getText()\r
-        << " * this";\r

 \r
 \r

-      if (member->structMethod()->func()->parameter()->firstParameter() != nullptr) {\r
-        o\r
-          << ", "\r
-          << member->structMethod()->func()->parameter()->firstParameter()->var()->type()->getText()\r
-          << " "\r
-          << member->structMethod()->func()->parameter()->firstParameter()->var()->varName()->getText();\r
+  return out;\r
+}\r

 \r
 \r

-        for (auto * par : member->structMethod()->func()->parameter()->additionalParameter()) {\r
-          o\r
-            << ", "\r
-            << par->var()->type()->getText()\r
-            << " "\r
-            << par->var()->varName()->getText();\r
-        }\r
-      }\r

 \r
 \r

-      o << ")\n{\n";\r
+void tocFunction (std::ostream & out, const Function & f, bool stub)\r
+{\r
+  out << f.returnType << " " << namespacePrefix() << f.name << " (" << vectorStr(f.parameters, ", ") << ")";\r

 \r
 \r

-      toc(o, member->structMethod()->func()->body());\r
+  if (stub)\r
+  {\r
+    out << ";\n";\r
+  }\r
+  else\r
+  {\r
+    out << "\n" << f.body;\r
+  }\r
+}\r
+void tocStruct (std::ostream & out, const Struct & s, bool stub)\r
+{\r
+  out << "struct " << namespacePrefix() << s.name;\r
+  if (stub)\r
+  {\r
+    out << ";\n";\r
+    for (auto m : s.methods)\r
+    {\r
+      Function f = m;\r

 \r
 \r

-      o << "}\n";\r
+      f.parameters.insert(f.parameters.begin(),\r
+      {"this",\r
+        {\r
+          namespaces,\r
+          s.name,\r
+          {\r
+            {TypeModifierType::Pointer, false, -1}\r
+          }\r
+        }\r
+      });\r
+      out << f.returnType << " " <<\r
+        namespacePrefix() << s.name << "_" << f.name <<\r
+        " (" << vectorStr(f.parameters, ", ") << ");\n";\r

     }\r
     }\r

+    return;\r

   }\r
   }\r

-}\r
-void toc(std::ostream & o, TocParser::BodyContext * ctx) {\r
-  for (auto * stmt : ctx->stmt()) {\r
-    toc(o, stmt);\r
-    o << "\n";\r
+  out << "\n{\n";\r
+  indentation += 2;\r
+\r
+  for (auto m : s.members)\r
+  {\r
+    indent(out);\r
+    out << m << ";\n";\r

   }\r
   }\r

-}\r
-void toc(std::ostream & o, TocParser::StmtContext * ctx) {\r
-  /**/ if (ctx->varDecl() != nullptr) toc(o, ctx->varDecl());\r
-  else if (ctx->conditional() != nullptr) toc(o, ctx->conditional()->ifCond());\r
-  else if (ctx->loop() != nullptr) toc(o, ctx->loop()->whileLoop());\r
-  else if (ctx->assignment() != nullptr) toc(o, ctx->assignment());\r
-  else if (ctx->returnStmt() != nullptr) toc(o, ctx->returnStmt());\r
-  else if (ctx->expr() != nullptr) toc(o, ctx->expr());\r

 \r
 \r

-  if (ctx->conditional() == nullptr && ctx->loop() == nullptr)\r
-    o << ";";\r
-}\r
-void toc(std::ostream & o, TocParser::IfCondContext * ctx) {\r
-  o << "if (";\r
-  toc(o, ctx->expr());\r
-  o << ")\n{\n";\r
-  toc(o, ctx->body());\r
-  o << "}\n";\r
-}\r
-void toc(std::ostream & o, TocParser::WhileLoopContext * ctx) {\r
-  o << "while (";\r
-  toc(o, ctx->expr());\r
-  o << ")\n{\n";\r
-  toc(o, ctx->body());\r
-  o << "}\n";\r
-}\r
-void toc(std::ostream & o, TocParser::AssignmentContext * ctx) {\r
-  toc(o, ctx->identifier());\r
-  o << " = ";\r
-  toc(o, ctx->expr());\r
-}\r
-void toc(std::ostream & o, TocParser::ReturnStmtContext * ctx) {\r
-  o << "return ";\r
-  toc(o, ctx->expr());\r
-}\r
-void toc(std::ostream & o, TocParser::ExprContext * ctx) {\r
-    /**/ if (ctx->funcCall()     != nullptr) toc(o, ctx->funcCall());\r
-    else if (ctx->identifier()   != nullptr) toc(o, ctx->identifier());\r
-    else if (ctx->literal()      != nullptr) toc(o, ctx->literal());\r
-    else if (ctx->subscript()    != nullptr) toc(o, ctx->subscript());\r
-    else if (ctx->memberAccess() != nullptr) toc(o, ctx->memberAccess());\r
-    else if (ctx->parenExpr()    != nullptr) toc(o, ctx->parenExpr());\r
-    else if (ctx->operatorExpr() != nullptr) toc(o, ctx->operatorExpr()->binaryOperator());\r
-}\r
-void toc(std::ostream & o, TocParser::NonOpExprContext * ctx) {\r
-    /**/ if (ctx->funcCall()     != nullptr) toc(o, ctx->funcCall());\r
-    else if (ctx->identifier()   != nullptr) toc(o, ctx->identifier());\r
-    else if (ctx->literal()      != nullptr) toc(o, ctx->literal());\r
-    else if (ctx->subscript()    != nullptr) toc(o, ctx->subscript());\r
-    else if (ctx->memberAccess() != nullptr) toc(o, ctx->memberAccess());\r
-    else if (ctx->parenExpr()    != nullptr) toc(o, ctx->parenExpr());\r
-}\r
-void toc(std::ostream & o, TocParser::NonSubscriptExprContext * ctx) {\r
-    /**/ if (ctx->funcCall()     != nullptr) toc(o, ctx->funcCall());\r
-    else if (ctx->identifier()   != nullptr) toc(o, ctx->identifier());\r
-    else if (ctx->memberAccess() != nullptr) toc(o, ctx->memberAccess());\r
-    else if (ctx->parenExpr()    != nullptr) toc(o, ctx->parenExpr());\r
-}\r
-void toc(std::ostream & o, TocParser::FuncCallContext * ctx) {\r
-  o\r
-    << ctx->funcName()->getText()\r
-    << "(";\r
-  for (int i = 0; i < ctx->expr().size(); i++) {\r
-    if (i != 0) o << ", ";\r
-    toc(o, ctx->expr(i));\r
+  indent(out, -2);\r
+  out << "};\n";\r
+  \r
+  for (auto m : s.methods)\r
+  {\r
+    Function f = m;\r
+    f.parameters.insert(f.parameters.begin(),\r
+      {"this",\r
+        {\r
+          namespaces,\r
+          s.name,\r
+          {\r
+            {TypeModifierType::Pointer, false, -1}\r
+          }\r
+        }\r
+      });\r
+    out << f.returnType << " " <<\r
+    namespacePrefix() << s.name << "_" << f.name <<\r
+    " (" << vectorStr(f.parameters, ", ") << ")\n" << f.body;\r

   }\r
   }\r

-  o << ")";\r
-}\r
-void toc(std::ostream & o, TocParser::IdentifierContext * ctx) {\r
-  o << ctx->getText();\r
-}\r
-void toc(std::ostream & o, TocParser::LiteralContext * ctx) {\r
-  if (ctx->INTLIT() != nullptr) o << ctx->INTLIT()->getText();\r
-}\r
-void toc(std::ostream & o, TocParser::SubscriptContext * ctx) {\r
-  toc(o, ctx->nonSubscriptExpr());\r
-  o << "[";\r
-  toc(o, ctx->expr());\r
-  o << "]";\r
-}\r
-void toc(std::ostream & o, TocParser::MemberAccessContext * ctx) {\r
-  toc(o, ctx->identifier(0));\r
-  o << ".";\r
-  toc(o, ctx->identifier(1));\r

 }\r
 }\r

-void toc(std::ostream & o, TocParser::ParenExprContext * ctx) {\r
-  o << "(";\r
-  toc(o, ctx->expr());\r
-  o << ")";\r
-}\r
-void toc(std::ostream & o, TocParser::BinaryOperatorContext * ctx) {\r
-  for (int i = 0; i < ctx->BINARY_OPERATOR().size(); i++) {\r
-    toc(o, ctx->nonOpExpr(i));\r
-    o\r
-      << " "\r
-      << ctx->BINARY_OPERATOR(i)->getText()\r
-      << " ";\r
-    toc(o, ctx->nonOpExpr(i + 1));\r
+void tocProgram (std::ostream & out, const Program & p)\r
+{\r
+  globalPrg = p;\r
+  for (auto n : p.namespaces)\r
+  {\r
+    tocNamespace(out, n, true);\r
+  }\r
+  for (auto s : p.structs)\r
+  {\r
+    tocStruct(out, s, true);\r
+  }\r
+  for (auto f : p.functions)\r
+  {\r
+    tocFunction(out, f, true);\r

   }\r
   }\r

-}\r

 \r
 \r

-void toc_stub(std::ostream & o, TocParser::FuncContext * ctx) {\r
-  o\r
-    << ctx->type()->getText()\r
-    << " "\r
-    << ctx->funcName()->getText()\r
-    << "(";\r
+  for (auto v : p.variables)\r
+  {\r
+    out << v << ";\n";\r
+  }\r
+  for (auto n : p.namespaces)\r
+  {\r
+    tocNamespace(out, n, false);\r
+  }\r
+  for (auto s : p.structs)\r
+  {\r
+    tocStruct(out, s, false);\r
+  }\r
+  for (auto f : p.functions)\r
+  {\r
+    tocFunction(out, f, false);\r
+  }\r
+}\r

 \r
 \r

-  if (ctx->parameter()->firstParameter() != nullptr) {\r
-    o\r
-      << ctx->parameter()->firstParameter()->var()->type()->getText()\r
-      << " "\r
-      << ctx->parameter()->firstParameter()->var()->varName()->getText();\r

 \r
 \r

-    for (auto * par : ctx->parameter()->additionalParameter()) {\r
-      o\r
-        << ", "\r
-        << par->var()->type()->getText()\r
-        << " "\r
-        << par->var()->varName()->getText();\r
+void tocNamespace  (std::ostream & out, const Namespace & n, bool stub)\r
+{\r
+  namespaces.push_back(n.name);\r
+  if (!stub)\r
+  {\r
+    for (auto v : n.variables)\r
+    {\r
+      out << v << ";\n";\r

     }\r
   }\r
     }\r
   }\r

-\r
-  o << ");\n";\r
-}\r
-void toc_stub(std::ostream & o, TocParser::StructDeclContext * ctx) {\r
-  o\r
-    << "struct "\r
-    << ctx->structName()->getText()\r
-    << ";\n";\r
-}\r
+  for (auto n : n.namespaces)\r
+  {\r
+    tocNamespace(out, n, stub);\r
+  }\r
+  for (auto s : n.structs)\r
+  {\r
+    tocStruct(out, s, stub);\r
+  }\r
+  for (auto f : n.functions)\r
+  {\r
+    tocFunction(out, f, stub);\r
+  }\r
+  namespaces.pop_back();\r
+}
\ No newline at end of file