X-Git-Url: https://gitweb.ps.run/toc/blobdiff_plain/45409c781a9e35df68c43b1e2f028d30bf90c0a0..c4231c6faf4e1b4650b075c641b0bb8c053739e4:/src/toc.h diff --git a/src/toc.h b/src/toc.h index b36d878..e95ab3e 100644 --- a/src/toc.h +++ b/src/toc.h @@ -1,274 +1,507 @@ #pragma once #include +#include -#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()); +#include "repr.h" +#include "generic.h" +#include "typeInfo.h" + +template +std::string vectorStr (const std::vector & v, const std::string & separator, bool end = false) +{ + std::stringstream sstr; + + 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(); +} + +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, ' '); +} + +static std::vector namespaces; +static std::string namespacePrefix() { + std::stringstream sstr; + for (auto n : namespaces) + { + sstr << n << "_"; } - for (auto * decl : ctx->decl()) { - if (decl->varDecl() != nullptr) { - toc(o, decl->varDecl()); - o << ";\n"; + return sstr.str(); +} + +static std::map currentInstantiation; + +static Program globalPrg; +static std::shared_ptr globalCtx; + + + +// std::string getPrefix(std::shared_ptr ctx) +// { +// std::string result; +// for (auto it = ctx; it != nullptr; it = it->parent) +// { +// if (it->name.has_value()) +// { +// result = it->name.value() + "_" + result; +// } +// } +// return result; +// } + +std::ostream & operator<< (std::ostream & out, const Type & t) +{ + for (auto kv : currentInstantiation) + { + if (t.name == kv.first) + { + out << kv.second; + return out; } - else if (decl->structDecl() != nullptr) toc(o, decl->structDecl()); - else if (decl->funcDecl() != nullptr) toc(o, decl->funcDecl()->func()); } + TypeInfo ti = typeType(globalCtx, t); + if (ti.isStruct) + out << "struct "; + auto s = findStruct(t.name, t.namespacePrefixes, globalCtx); + if (s.has_value()) + out << vectorStr(std::get<1>(*s), "_", true) << t.name; + else + out << vectorStr(t.namespacePrefixes, "_", true) << t.name; + if (!t.genericInstantiation.empty()) + out << genericAppendix(t.genericInstantiation); + + return out; } -void toc(std::ostream & o, TocParser::VarDeclContext * ctx) { - o - << ctx->var()->type()->getText() - << " " - << ctx->var()->varName()->getText(); +std::ostream & operator<< (std::ostream & out, const Variable & v) +{ + out << v.type << " "; + + std::stringstream sstr; + std::string s = v.name; - if (ctx->var()->expr() != nullptr) { - o << " = "; - toc(o, ctx->var()->expr()); + auto var = findVariable(v.name, namespaces, globalCtx); + if (var.has_value()) + s = vectorStr(std::get<1>(*var), "_", true) + s; + + 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; + + return out; } -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(); +std::ostream & operator<< (std::ostream & out, const Body & b) +{ + b.ctx->parent = globalCtx; + globalCtx = b.ctx; + + indent(out); + out << "{\n"; + indentation += 2; + + for (auto v : b.ctx->variables) + { + indent(out); + out << v << ";\n"; + } + + out << "\n"; + + for (auto s : b.statements) + { + indent(out); + out << s << "\n"; + } + + indent(out, -2); + out << "}\n"; + + globalCtx = b.ctx->parent; + + return out; +} +std::ostream & operator<< (std::ostream & out, const Expr & e) +{ + switch (e.type) + { + case ExprType::Func: + { + auto f = findFunction(e._func.functionName, e._func.namespacePrefixes, globalCtx); + + if (std::get<0>(*f).defined) + out << vectorStr(std::get<1>(*f), "_", true); + + out << e._func.functionName; + if (!e._func.genericInstantiation.empty()) + out << genericAppendix(e._func.genericInstantiation); + out <<"(" << vectorStr(e._func.arguments, ", ") << ")"; break; + } + case ExprType::Method: + { + TypeInfo ti = typeExpr(globalCtx, *e._method.expr); + out << + vectorStr(ti.type.namespacePrefixes, "_", true) << + ti.type.name << genericAppendix(ti.type.genericInstantiation) << "_" << e._method.methodName; + if (!e._method.genericInstantiation.empty()) + out << genericAppendix(e._method.genericInstantiation); + out << "("; + if (e._method.expr->type == ExprType::Identifier) + out << "&"; + out << *e._method.expr << (e._method.arguments.empty() ? "" : ", ") << + 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.isPointer ? "->" : ".") << 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: + auto v = findVariable(e._identifier.identifier, e._identifier.namespacePrefixes, globalCtx); + if (v.has_value()) + out << vectorStr(std::get<1>(*v), "_", true); + else + out << vectorStr(e._identifier.namespacePrefixes, "_", true); + + out << e._identifier.identifier; break; + } + + 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; + } + + return out; +} + + +void tocFunction (std::ostream & out, const Function & f, bool stub) +{ + if (!stub && !f.defined) return; + + if (f.genericTypeNames.empty()) + { + out << f.returnType << " " << namespacePrefix() << f.name << " (" << vectorStr(f.parameters, ", ") << ")"; + + if (stub) + { + out << ";\n"; + } + else + { + out << "\n" << f.body; } } + else + { + for (auto instantiation : f.genericInstantiations) + { + for (int i = 0; i < f.genericTypeNames.size(); i++) + { + currentInstantiation[f.genericTypeNames[i]] = instantiation[i]; + } - o << ")\n{\n"; + out << f.returnType << " " << namespacePrefix() << f.name << genericAppendix(instantiation) << " (" << vectorStr(f.parameters, ", ") << ")"; - toc(o, ctx->body()); + if (stub) + { + out << ";\n"; + } + else + { + out << "\n" << f.body; + } - o << "}\n"; + currentInstantiation.clear(); + } + } } -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"; +void tocStruct (std::ostream & out, const Struct & s, bool stub) +{ + if (s.genericTypeNames.empty()) + { + out << "struct " << namespacePrefix() << s.name; + if (stub) + { + 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"; + } + return; + } + out << "\n{\n"; + indentation += 2; + + for (auto m : s.members) + { + indent(out); + out << m << ";\n"; + } + + 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 << "} " - << 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(); + else + { + for (auto instantiation : s.genericInstantiations) + { + for (int i = 0; i < s.genericTypeNames.size(); i++) + { + currentInstantiation[s.genericTypeNames[i]] = instantiation[i]; + } + + out << "struct " << namespacePrefix() << s.name << genericAppendix(instantiation); + if (stub) + { + out << ";\n"; + for (auto m : s.methods) + { + Function f = m; + + f.parameters.insert(f.parameters.begin(), + {"this", + { + namespaces, + s.name + genericAppendix(instantiation), + { + {TypeModifierType::Pointer, false, -1} + } + } + }); + out << f.returnType << " " << + namespacePrefix() << s.name << genericAppendix(instantiation) << "_" << f.name << + " (" << vectorStr(f.parameters, ", ") << ");\n"; } + return; } + out << "\n{\n"; + indentation += 2; - o << ")\n{\n"; + for (auto m : s.members) + { + indent(out); + out << m << ";\n"; + } - toc(o, member->structMethod()->func()->body()); + indent(out, -2); + out << "};\n"; + + for (auto m : s.methods) + { + Function f = m; + f.parameters.insert(f.parameters.begin(), + {"this", + { + namespaces, + s.name + genericAppendix(instantiation), + { + {TypeModifierType::Pointer, false, -1} + } + } + }); + out << f.returnType << " " << + namespacePrefix() << s.name << genericAppendix(instantiation) << "_" << f.name << + " (" << vectorStr(f.parameters, ", ") << ")\n" << f.body; + } - o << "}\n"; + currentInstantiation.clear(); } } } -void toc(std::ostream & o, TocParser::BodyContext * ctx) { - for (auto * stmt : ctx->stmt()) { - toc(o, stmt); - o << "\n"; +void tocProgram (std::ostream & out, const Program & p) +{ + globalCtx = p.ctx; + + globalPrg = p; + for (auto n : p.ctx->namespaces) + { + tocNamespace(out, n, true); } -} -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)); + out << "\n\n"; + for (auto s : p.ctx->structs) + { + tocStruct(out, s, true); } - 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)); + out << "\n\n"; + for (auto f : p.ctx->functions) + { + tocFunction(out, f, true); } + out << "\n\n"; + + for (auto v : p.ctx->variables) + { + out << v << ";\n"; + } + out << "\n\n"; + for (auto n : p.ctx->namespaces) + { + tocNamespace(out, n, false); + } + out << "\n\n"; + for (auto s : p.ctx->structs) + { + tocStruct(out, s, false); + } + out << "\n\n"; + for (auto f : p.ctx->functions) + { + tocFunction(out, f, false); + } + out << "\n\n"; } -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(); + +void tocNamespace (std::ostream & out, const Namespace & n, bool stub) +{ + n.ctx->parent = globalCtx; + globalCtx = n.ctx; + + namespaces.push_back(n.name); + if (!stub) + { + for (auto v : n.ctx->variables) + { + out << v << ";\n"; } + out << "\n\n"; } + for (auto n : n.ctx->namespaces) + { + tocNamespace(out, n, stub); + out << "\n\n"; + } + for (auto s : n.ctx->structs) + { + tocStruct(out, s, stub); + out << "\n\n"; + } + for (auto f : n.ctx->functions) + { + tocFunction(out, f, stub); + out << "\n\n"; + } + namespaces.pop_back(); - o << ");\n"; -} -void toc_stub(std::ostream & o, TocParser::StructDeclContext * ctx) { - o - << "struct " - << ctx->structName()->getText() - << ";\n"; -} + globalCtx = n.ctx->parent; +} \ No newline at end of file