Function.cpp 1.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263
  1. /*
  2. * Copyright (c) 2023, Dan Klishch <danilklishch@gmail.com>
  3. *
  4. * SPDX-License-Identifier: BSD-2-Clause
  5. */
  6. #include "Function.h"
  7. #include "AST/AST.h"
  8. #include "Compiler/ControlFlowGraph.h"
  9. namespace JSSpecCompiler {
  10. TranslationUnit::TranslationUnit(StringView filename)
  11. : m_filename(filename)
  12. {
  13. }
  14. TranslationUnit::~TranslationUnit() = default;
  15. void TranslationUnit::adopt_declaration(NonnullRefPtr<FunctionDeclaration>&& declaration)
  16. {
  17. declaration->m_translation_unit = this;
  18. m_function_index.set(declaration->m_name, declaration.ptr());
  19. m_declarations_owner.append(move(declaration));
  20. }
  21. FunctionDefinitionRef TranslationUnit::adopt_function(NonnullRefPtr<FunctionDefinition>&& definition)
  22. {
  23. FunctionDefinitionRef result = definition.ptr();
  24. m_functions_to_compile.append(result);
  25. adopt_declaration(definition);
  26. return result;
  27. }
  28. FunctionDeclarationRef TranslationUnit::find_declaration_by_name(StringView name) const
  29. {
  30. auto it = m_function_index.find(name);
  31. if (it == m_function_index.end())
  32. return nullptr;
  33. return it->value;
  34. }
  35. FunctionDeclaration::FunctionDeclaration(StringView name, Vector<FunctionArgument>&& arguments)
  36. : m_name(name)
  37. , m_arguments(arguments)
  38. {
  39. }
  40. FunctionDefinition::FunctionDefinition(StringView name, Tree ast, Vector<FunctionArgument>&& arguments)
  41. : FunctionDeclaration(name, move(arguments))
  42. , m_ast(move(ast))
  43. , m_named_return_value(make_ref_counted<NamedVariableDeclaration>("$return"sv))
  44. {
  45. }
  46. void FunctionDefinition::reindex_ssa_variables()
  47. {
  48. size_t index = 0;
  49. for (auto const& var : m_local_ssa_variables)
  50. var->m_index = index++;
  51. }
  52. }