mirror of
https://github.com/RGBCube/serenity
synced 2025-05-14 12:14:58 +00:00
83 lines
2.5 KiB
C++
83 lines
2.5 KiB
C++
/*
|
|
* Copyright (c) 2023, Dan Klishch <danilklishch@gmail.com>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <AK/HashMap.h>
|
|
#include <AK/RefCounted.h>
|
|
#include <AK/RefPtr.h>
|
|
#include <AK/StringView.h>
|
|
|
|
#include "DiagnosticEngine.h"
|
|
#include "Forward.h"
|
|
|
|
namespace JSSpecCompiler {
|
|
|
|
class TranslationUnit {
|
|
public:
|
|
TranslationUnit(StringView filename);
|
|
~TranslationUnit();
|
|
|
|
void adopt_declaration(NonnullRefPtr<FunctionDeclaration>&& declaration);
|
|
FunctionDefinitionRef adopt_function(NonnullRefPtr<FunctionDefinition>&& definition);
|
|
|
|
FunctionDeclarationRef find_declaration_by_name(StringView name) const;
|
|
|
|
StringView filename() const { return m_filename; }
|
|
DiagnosticEngine& diag() { return m_diagnostic_engine; }
|
|
Vector<FunctionDefinitionRef> functions_to_compile() const { return m_functions_to_compile; }
|
|
|
|
EnumeratorRef get_node_for_enumerator_value(StringView value);
|
|
|
|
private:
|
|
StringView m_filename;
|
|
DiagnosticEngine m_diagnostic_engine;
|
|
Vector<FunctionDefinitionRef> m_functions_to_compile;
|
|
Vector<NonnullRefPtr<FunctionDeclaration>> m_declarations_owner;
|
|
HashMap<StringView, FunctionDeclarationRef> m_function_index;
|
|
HashMap<StringView, EnumeratorRef> m_enumerator_nodes;
|
|
};
|
|
|
|
struct FunctionArgument {
|
|
StringView name;
|
|
size_t optional_arguments_group;
|
|
};
|
|
|
|
class FunctionDeclaration : public RefCounted<FunctionDeclaration> {
|
|
public:
|
|
FunctionDeclaration(StringView name, Vector<FunctionArgument>&& arguments);
|
|
|
|
virtual ~FunctionDeclaration() = default;
|
|
|
|
TranslationUnitRef m_translation_unit = nullptr;
|
|
StringView m_name;
|
|
Vector<FunctionArgument> m_arguments;
|
|
};
|
|
|
|
class FunctionDefinition : public FunctionDeclaration {
|
|
public:
|
|
FunctionDefinition(StringView name, Tree ast, Vector<FunctionArgument>&& arguments);
|
|
|
|
void reindex_ssa_variables();
|
|
|
|
Tree m_ast;
|
|
|
|
// Populates during reference resolving
|
|
// NOTE: The hash map here is ordered since we do not want random hash changes to break our test
|
|
// expectations (looking at you, SipHash).
|
|
OrderedHashMap<StringView, NamedVariableDeclarationRef> m_local_variables;
|
|
|
|
// Fields populate during CFG building
|
|
NamedVariableDeclarationRef m_named_return_value;
|
|
RefPtr<ControlFlowGraph> m_cfg;
|
|
|
|
// Fields populate during SSA building
|
|
Vector<SSAVariableDeclarationRef> m_ssa_arguments;
|
|
SSAVariableDeclarationRef m_return_value;
|
|
Vector<SSAVariableDeclarationRef> m_local_ssa_variables;
|
|
};
|
|
|
|
}
|