ASTPrinting.cpp 4.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156
  1. /*
  2. * Copyright (c) 2023, Dan Klishch <danilklishch@gmail.com>
  3. *
  4. * SPDX-License-Identifier: BSD-2-Clause
  5. */
  6. #include <AK/StringBuilder.h>
  7. #include <AK/TemporaryChange.h>
  8. #include "AST/AST.h"
  9. #include "Function.h"
  10. namespace JSSpecCompiler {
  11. void Node::format_tree(StringBuilder& builder)
  12. {
  13. static int current_depth = -1;
  14. TemporaryChange<int> depth_change(current_depth, current_depth + 1);
  15. builder.append_repeated(' ', current_depth * 2);
  16. dump_tree(builder);
  17. }
  18. template<typename... Parameters>
  19. void Node::dump_node(StringBuilder& builder, AK::CheckedFormatString<Parameters...>&& fmtstr, Parameters const&... parameters)
  20. {
  21. builder.append("<"sv);
  22. AK::VariadicFormatParams<AK::AllowDebugOnlyFormatters::No, Parameters...> variadic_format_params { parameters... };
  23. MUST(AK::vformat(builder, fmtstr.view(), variadic_format_params));
  24. builder.append(">\n"sv);
  25. }
  26. void ErrorNode::dump_tree(StringBuilder& builder)
  27. {
  28. dump_node(builder, "Error \"{}\"", m_error);
  29. }
  30. void MathematicalConstant::dump_tree(StringBuilder& builder)
  31. {
  32. dump_node(builder, "MathematicalConstant {}", m_number);
  33. }
  34. void StringLiteral::dump_tree(StringBuilder& builder)
  35. {
  36. dump_node(builder, "StringLiteral {}", m_literal);
  37. }
  38. void BinaryOperation::dump_tree(StringBuilder& builder)
  39. {
  40. dump_node(builder, "BinaryOperation {}", binary_operator_names[to_underlying(m_operation)]);
  41. m_left->format_tree(builder);
  42. m_right->format_tree(builder);
  43. }
  44. void UnaryOperation::dump_tree(StringBuilder& builder)
  45. {
  46. dump_node(builder, "UnaryOperation {}", unary_operator_names[to_underlying(m_operation)]);
  47. m_operand->format_tree(builder);
  48. }
  49. void IsOneOfOperation::dump_tree(StringBuilder& builder)
  50. {
  51. dump_node(builder, "IsOneOf");
  52. m_operand->format_tree(builder);
  53. for (auto const& compare_value : m_compare_values)
  54. compare_value->format_tree(builder);
  55. }
  56. void UnresolvedReference::dump_tree(StringBuilder& builder)
  57. {
  58. dump_node(builder, "UnresolvedReference {}", m_name);
  59. }
  60. void ReturnExpression::dump_tree(StringBuilder& builder)
  61. {
  62. dump_node(builder, "ReturnExpression");
  63. m_return_value->format_tree(builder);
  64. }
  65. void AssertExpression::dump_tree(StringBuilder& builder)
  66. {
  67. dump_node(builder, "AssertExpression");
  68. m_condition->format_tree(builder);
  69. }
  70. void IfBranch::dump_tree(StringBuilder& builder)
  71. {
  72. dump_node(builder, "IfBranch");
  73. m_condition->format_tree(builder);
  74. m_branch->format_tree(builder);
  75. }
  76. void ElseIfBranch::dump_tree(StringBuilder& builder)
  77. {
  78. dump_node(builder, "ElseIfBranch {}", m_condition ? "ElseIf" : "Else");
  79. if (m_condition)
  80. m_condition->format_tree(builder);
  81. m_branch->format_tree(builder);
  82. }
  83. void IfElseIfChain::dump_tree(StringBuilder& builder)
  84. {
  85. dump_node(builder, "IfElseIfChain");
  86. for (size_t i = 0; i < branches_count(); ++i) {
  87. m_conditions[i]->format_tree(builder);
  88. m_branches[i]->format_tree(builder);
  89. }
  90. if (m_else_branch)
  91. m_else_branch->format_tree(builder);
  92. }
  93. void TreeList::dump_tree(StringBuilder& builder)
  94. {
  95. dump_node(builder, "TreeList");
  96. for (auto const& expression : m_expressions)
  97. expression->format_tree(builder);
  98. }
  99. void RecordDirectListInitialization::dump_tree(StringBuilder& builder)
  100. {
  101. dump_node(builder, "RecordDirectListInitialization");
  102. m_type_reference->format_tree(builder);
  103. for (auto const& argument : m_arguments)
  104. builder.appendff("{}{}", argument.name, argument.value);
  105. }
  106. void FunctionCall::dump_tree(StringBuilder& builder)
  107. {
  108. dump_node(builder, "FunctionCall");
  109. m_name->format_tree(builder);
  110. for (auto const& argument : m_arguments)
  111. argument->format_tree(builder);
  112. }
  113. void SlotName::dump_tree(StringBuilder& builder)
  114. {
  115. dump_node(builder, "Slot {}", m_member_name);
  116. }
  117. void Variable::dump_tree(StringBuilder& builder)
  118. {
  119. dump_node(builder, "Var {}", m_name);
  120. }
  121. void FunctionPointer::dump_tree(StringBuilder& builder)
  122. {
  123. m_function.visit(
  124. [&](StringView name) {
  125. dump_node(builder, "Func external \"{}\"", name);
  126. },
  127. [&](FunctionRef function) {
  128. dump_node(builder, "Func local \"{}\"", function->m_name);
  129. });
  130. }
  131. }