1
Fork 0
mirror of https://github.com/RGBCube/serenity synced 2025-07-25 21:07:35 +00:00

JSSpecCompiler: Add infrastructure to run compiler passes on AST

This commit is contained in:
Dan Klishch 2023-08-18 13:12:20 -04:00 committed by Andrew Kaster
parent cd8f4aaa7d
commit 198591cc20
6 changed files with 243 additions and 0 deletions

View file

@ -0,0 +1,31 @@
/*
* Copyright (c) 2023, Dan Klishch <danilklishch@gmail.com>
*
* SPDX-License-Identifier: BSD-2-Clause
*/
#pragma once
#include <AK/NonnullRefPtr.h>
#include <AK/RecursionDecision.h>
#include "Forward.h"
namespace JSSpecCompiler {
class CompilerPass {
public:
CompilerPass(FunctionRef function)
: m_function(function)
{
}
virtual ~CompilerPass() = default;
virtual void run() = 0;
protected:
FunctionRef m_function;
};
}

View file

@ -0,0 +1,49 @@
/*
* Copyright (c) 2023, Dan Klishch <danilklishch@gmail.com>
*
* SPDX-License-Identifier: BSD-2-Clause
*/
#include "Compiler/GenericASTPass.h"
#include "AST/AST.h"
#include "Function.h"
namespace JSSpecCompiler {
void RecursiveASTVisitor::run_in_subtree(Tree& tree)
{
NodeSubtreePointer pointer { &tree };
recurse(tree, pointer);
}
void RecursiveASTVisitor::replace_current_node_with(Tree tree)
{
m_current_subtree_pointer->replace_subtree(move(tree));
}
RecursionDecision RecursiveASTVisitor::recurse(Tree root, NodeSubtreePointer& pointer)
{
RecursionDecision decision;
m_current_subtree_pointer = &pointer;
decision = on_entry(root);
if (decision == RecursionDecision::Recurse) {
for (auto& child : root->subtrees()) {
if (recurse(child.get(), child) == RecursionDecision::Break)
return RecursionDecision::Break;
}
}
m_current_subtree_pointer = &pointer;
on_leave(root);
return RecursionDecision::Continue;
}
void GenericASTPass::run()
{
run_in_subtree(m_function->m_ast);
}
}

View file

@ -0,0 +1,46 @@
/*
* Copyright (c) 2023, Dan Klishch <danilklishch@gmail.com>
*
* SPDX-License-Identifier: BSD-2-Clause
*/
#pragma once
#include <AK/NonnullRefPtr.h>
#include <AK/RecursionDecision.h>
#include "Compiler/CompilerPass.h"
namespace JSSpecCompiler {
class RecursiveASTVisitor {
public:
virtual ~RecursiveASTVisitor() = default;
void run_in_subtree(Tree& tree);
protected:
virtual RecursionDecision on_entry(Tree) { return RecursionDecision::Recurse; }
virtual void on_leave(Tree) { }
void replace_current_node_with(Tree tree);
private:
RecursionDecision recurse(Tree root, NodeSubtreePointer& pointer);
NodeSubtreePointer* m_current_subtree_pointer = nullptr;
};
class GenericASTPass
: public CompilerPass
, protected RecursiveASTVisitor {
public:
GenericASTPass(FunctionRef function)
: CompilerPass(function)
{
}
void run() override;
};
}