diff options
1 files changed, 344 insertions, 0 deletions
| -rw-r--r-- | JSImpl/src/ASTPrinter.cpp | 344 |
diff --git a/JSImpl/src/ASTPrinter.cpp b/JSImpl/src/ASTPrinter.cpp new file mode 100644 index 0000000..4a05510 --- /dev/null +++ b/JSImpl/src/ASTPrinter.cpp @@ -0,0 +1,344 @@ +#include "ASTPrinter.h" +#include "Expression.h" + +#include <iostream> +#include <iosfwd> + +std::ostream& operator<<(std::ostream& os, const TokenType& t) +{ + switch (t) + { + case TokenType::LeftParen: os << "\"LeftParen\""; return os; + case TokenType::RightParen: os << "\"RightParen\""; return os; + case TokenType::LeftBrace: os << "\"LeftBrace\""; return os; + case TokenType::RightBrace: os << "\"RightBrace\""; return os; + case TokenType::Comma: os << "\"Comma\""; return os; + case TokenType::Dot: os << "\"Dot\""; return os; + case TokenType::Minus: os << "\"Minus\""; return os; + case TokenType::Plus: os << "\"Plus\""; return os; + case TokenType::Semicolon: os << "\"Semicolon\""; return os; + case TokenType::Star: os << "\"Star\""; return os; + case TokenType::Division: os << "\"Division\""; return os; + case TokenType::Modulo: os << "\"Modulo\""; return os; + case TokenType::BitwiseNot: os << "\"BitwiseNot\""; return os; + case TokenType::BitwiseAnd: os << "\"BitwiseAnd\""; return os; + case TokenType::BitwiseXor: os << "\"BitwiseXor\""; return os; + case TokenType::BitwiseOr: os << "\"BitwiseOr\""; return os; + case TokenType::QuestionMark: os << "\"QuestionMark\""; return os; + case TokenType::Colon: os << "\"Colon\""; return os; + case TokenType::LeftSquareBracket: os << "\"LeftSquareBracket\""; return os; + case TokenType::RightSquareBracket: os << "\"RightSquareBracket\""; return os; + case TokenType::Bang: os << "\"Bang\""; return os; + case TokenType::BangEqual: os << "\"BangEqual\""; return os; + case TokenType::Equal: os << "\"Equal\""; return os; + case TokenType::EqualEqual: os << "\"EqualEqual\""; return os; + case TokenType::StrictEqual: os << "\"StrictEqual\""; return os; + case TokenType::StrictNotEqual: os << "\"StrictNotEqual\""; return os; + case TokenType::Greater: os << "\"Greater\""; return os; + case TokenType::GreaterEqual: os << "\"GreaterEqual\""; return os; + case TokenType::Less: os << "\"Less\""; return os; + case TokenType::LessEqual: os << "\"LessEqual\""; return os; + case TokenType::MinusMinus: os << "\"MinusMinus\""; return os; + case TokenType::PlusPlus: os << "\"PlusPlus\""; return os; + case TokenType::LeftShift: os << "\"LeftShift\""; return os; + case TokenType::RightShift: os << "\"RightShift\""; return os; + case TokenType::LogicalAnd: os << "\"LogicalAnd\""; return os; + case TokenType::LogicalOr: os << "\"LogicalOr\""; return os; + case TokenType::StarEqual: os << "\"StarEqual\""; return os; + case TokenType::DivideEqual: os << "\"DivideEqual\""; return os; + case TokenType::ModuloEqual: os << "\"ModuloEqual\""; return os; + case TokenType::PlusEqual: os << "\"PlusEqual\""; return os; + case TokenType::MinusEqual: os << "\"MinusEqual\""; return os; + case TokenType::LeftShiftEqual: os << "\"LeftShiftEqual\""; return os; + case TokenType::RightShiftEqual: os << "\"RightShiftEqual\""; return os; + case TokenType::BitwiseAndEqual: os << "\"BitwiseAndEqual\""; return os; + case TokenType::BitwiseXorEqual: os << "\"BitwiseXorEqual\""; return os; + case TokenType::BitwiseOrEqual: os << "\"BitwiseOrEqual\""; return os; + case TokenType::Identifier: os << "\"Identifier\""; return os; + case TokenType::String: os << "\"String\""; return os; + case TokenType::Number: os << "\"Number\""; return os; + case TokenType::Break: os << "\"Break\""; return os; + case TokenType::Case: os << "\"Case\""; return os; + case TokenType::Catch: os << "\"Catch\""; return os; + case TokenType::Class: os << "\"Class\""; return os; + case TokenType::Const: os << "\"Const\""; return os; + case TokenType::Continue: os << "\"Continue\""; return os; + case TokenType::Debugger: os << "\"Debugger\""; return os; + case TokenType::Default: os << "\"Default\""; return os; + case TokenType::Delete: os << "\"Delete\""; return os; + case TokenType::Do: os << "\"Do\""; return os; + case TokenType::Else: os << "\"Else\""; return os; + case TokenType::Export: os << "\"Export\""; return os; + case TokenType::Extends: os << "\"Extends\""; return os; + case TokenType::Finally: os << "\"Finally\""; return os; + case TokenType::For: os << "\"For\""; return os; + case TokenType::Function: os << "\"Function\""; return os; + case TokenType::If: os << "\"If\""; return os; + case TokenType::Import: os << "\"Import\""; return os; + case TokenType::In: os << "\"In\""; return os; + case TokenType::Instanceof: os << "\"Instanceof\""; return os; + case TokenType::New: os << "\"New\""; return os; + case TokenType::Return: os << "\"Return\""; return os; + case TokenType::Super: os << "\"Super\""; return os; + case TokenType::Switch: os << "\"Switch\""; return os; + case TokenType::This: os << "\"This\""; return os; + case TokenType::Throw: os << "\"Throw\""; return os; + case TokenType::Try: os << "\"Try\""; return os; + case TokenType::Typeof: os << "\"Typeof\""; return os; + case TokenType::Var: os << "\"Var\""; return os; + case TokenType::Void: os << "\"Void\""; return os; + case TokenType::While: os << "\"While\""; return os; + case TokenType::With: os << "\"With\""; return os; + case TokenType::Yield: os << "\"Yield\""; return os; + case TokenType::Null: os << "\"Null\""; return os; + case TokenType::Undefined: os << "\"Undefined\""; return os; + case TokenType::True: os << "\"True\""; return os; + case TokenType::False: os << "\"False\""; return os; + case TokenType::Eof: os << "\"Eof\""; return os; + case TokenType::Invalid: os << "\"Invalid\""; return os; + default: + break; + } + return os; +} + +class ASTToJsonPrinter : public ExpressionVisitor +{ +public: + ASTToJsonPrinter(std::ostream& os); + virtual ~ASTToJsonPrinter() {} + virtual void Visit(LiteralNull* e) override; + virtual void Visit(LiteralUndefined* e) override; + virtual void Visit(LiteralString* e) override; + virtual void Visit(LiteralNumber* e) override; + virtual void Visit(LiteralBoolean* e) override; + virtual void Visit(LiteralField* e) override; + virtual void Visit(LiteralObject* e) override; + virtual void Visit(BinaryExpression* e) override; + virtual void Visit(UnaryExpression* e) override; + virtual void Visit(IdentifierExpression* e) override; + virtual void Visit(ListExpression* e) override; + virtual void Visit(VariableDefinitionExpression* e) override; + virtual void Visit(BlockStatement* e) override; + virtual void Visit(LabeledStatement* e) override; + virtual void Visit(IfStatement* e) override; + virtual void Visit(SwitchStatement* e) override; + virtual void Visit(CaseStatement *) override; + virtual void Visit(WhileStatement* e) override; + virtual void Visit(ForStatement* e) override; + virtual void Visit(Break* e) override; + virtual void Visit(Continue* e) override; + virtual void Visit(FunctionDeclaration* e) override; + virtual void Visit(TopStatements* e) override; + virtual void Visit(EmptyExpression* e) override; + virtual void Visit(Call* e) override; + virtual void Visit(MemberAccess* e) override; + + void Print( const char* name, const ExpressionPtr& expr, bool isLast); + void Print( const char* name, const IPLVector<ExpressionPtr>& list, bool isLast); + void Print( const char* name, const IPLVector<IPLString>& list, bool isLast); + void Print( const char* name, const IPLString& member, bool isLast); + void Print( const char* name, bool member, bool isLast); + void Print( const char* name, const TokenType& member, bool isLast); + void Print( const char* name, const double& member, bool isLast); + + inline void Enter(); + inline void Exit(); + inline void NoIndentNextAccept(); + inline void IndentNextAccept(); + inline void InsertIndent(); +private: + std::ostream& m_Output; + unsigned m_Level; + bool m_IndentNext; +}; + +ASTToJsonPrinter::ASTToJsonPrinter(std::ostream& os) + : m_Output(os) + , m_Level(0) + , m_IndentNext(true) +{} + +void PrepNextElement(std::ostream& os, bool isLast) +{ + if (!isLast) + { + os << ","; + } + os << "\n"; +} + +void ASTToJsonPrinter::Print(const char* name, const ExpressionPtr& expr, bool isLast) +{ + if (!expr) + { + return; + } + InsertIndent(); + m_Output << "\"" << name << "\": "; + NoIndentNextAccept(); + expr->Accept(*this); + PrepNextElement(m_Output, isLast); +} + +void ASTToJsonPrinter::Print(const char* name, const IPLVector<ExpressionPtr>& list, bool isLast) +{ + InsertIndent(); + m_Output << "\"" << name << "\": ["; + if (list.empty()) + { + m_Output << "]"; + PrepNextElement(m_Output, isLast); + return; + } + + Enter(); + for (auto i = 0u; i < list.size(); ++i) + { + IndentNextAccept(); + list[i]->Accept(*this); + if (i + 1 != list.size()) + { + m_Output << ",\n"; + } + } + Exit(); + m_Output << "\n"; + InsertIndent(); + m_Output << "]"; + PrepNextElement(m_Output, isLast); +} + +void ASTToJsonPrinter::Print(const char* name, const IPLVector<IPLString>& list, bool isLast) +{ + InsertIndent(); + m_Output << "\"" << name << "\": ["; + if (list.empty()) + { + m_Output << "]"; + PrepNextElement(m_Output, isLast); + return; + } + + Enter(); + for (auto i = 0u; i < list.size(); ++i) + { + InsertIndent(); + m_Output << "\"" << list[i] << "\""; + if (i + 1 != list.size()) + { + m_Output << ",\n"; + } + } + Exit(); + m_Output << "\n"; + InsertIndent(); + m_Output << "]"; + PrepNextElement(m_Output, isLast); +} + +void ASTToJsonPrinter::Print(const char* name, bool member, bool isLast) +{ + InsertIndent(); + m_Output << "\"" << name << "\": " << member; + PrepNextElement(m_Output, isLast); +} + +void ASTToJsonPrinter::Print(const char* name, const TokenType& member, bool isLast) +{ + InsertIndent(); + m_Output << "\"" << name << "\": " << member; + PrepNextElement(m_Output, isLast); +} + + +IPLString StripString(const IPLString& str) +{ + if (str.front() == '\"') + { + return str.substr(1, str.length() - 2); + } + return str; +} + +void ASTToJsonPrinter::Print(const char* name, const IPLString& member, bool isLast) +{ + InsertIndent(); + m_Output << "\"" << name << "\": \"" << StripString(member) << "\""; + PrepNextElement(m_Output, isLast); +} + +void ASTToJsonPrinter::Print(const char* name, const double& member, bool isLast) +{ + InsertIndent(); + m_Output << "\"" << name << "\": " << member; + PrepNextElement(m_Output, isLast); +} + + + +#define MEMBERS_COUNT(type, name, def) ++membersCount; +#define VISIT_MEMBER(type, name, def) ++currentVisitCount; Print(#name, e->Get##name(), currentVisitCount == membersCount); + +#define GENERATE_AST_PRINTER_FUNCTION(ClassName, MEMBERS_ITERATOR) \ + void ASTToJsonPrinter::Visit(ClassName* e) \ + { \ + int membersCount = 0, currentVisitCount = 0; \ + (void)currentVisitCount;(void)e; \ + MEMBERS_ITERATOR(MEMBERS_COUNT); \ + if (m_Level) \ + { \ + InsertIndent(); \ + } \ + m_Output << "{"; \ + Enter(); \ + InsertIndent(); \ + m_Output << "\"ExpressionType\": \"" << #ClassName "\""; \ + if (membersCount) \ + { \ + m_Output << ","; \ + } \ + m_Output << "\n"; \ + MEMBERS_ITERATOR(VISIT_MEMBER); \ + Exit(); \ + InsertIndent(); \ + m_Output << "}"; \ + } + +EXPRESSION_DEFINITION_ITERATOR(GENERATE_AST_PRINTER_FUNCTION); + +void ASTToJsonPrinter::Enter() +{ + m_Output << "\n"; + ++m_Level; +} + +void ASTToJsonPrinter::Exit() +{ + --m_Level; +} + +void ASTToJsonPrinter::IndentNextAccept() +{ + m_IndentNext = true; +} + +void ASTToJsonPrinter::InsertIndent() +{ + for (unsigned i = 0; i < m_Level; ++i) + { + m_Output << " "; + } +} + +void ASTToJsonPrinter::NoIndentNextAccept() +{ + m_IndentNext = false; +} + + +void PrintAST(const ExpressionPtr& ast, std::ostream& where) +{
+ ASTToJsonPrinter p(where); + ast->Accept(p); +}
\ No newline at end of file |