RGBCube/serenity
1
Fork 0
mirror of https://github.com/RGBCube/serenity synced 2025-07-25 17:07:34 +00:00
Code Issues Activity Actions

JSSpecCompiler: Add --dump-cfg option

Browse source
This commit is contained in:
Dan Klishch 2023-10-01 23:30:43 -04:00 • committed by Andrew Kaster
parent ff44aea917
commit ce6b987330
4 changed files with 53 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
Meta/Lagom/Tools/CodeGenerators/JSSpecCompiler/CMakeLists.txt
View file

@ -2,6 +2,7 @@ set(SOURCES
AST/AST.cpp AST/AST.cpp
AST/ASTPrinting.cpp AST/ASTPrinting.cpp
Compiler/CompilerPass.cpp Compiler/CompilerPass.cpp
Compiler/ControlFlowGraph.cpp
Compiler/GenericASTPass.cpp Compiler/GenericASTPass.cpp
Compiler/Passes/CFGBuildingPass.cpp Compiler/Passes/CFGBuildingPass.cpp
Compiler/Passes/FunctionCallCanonicalizationPass.cpp Compiler/Passes/FunctionCallCanonicalizationPass.cpp

29
Meta/Lagom/Tools/CodeGenerators/JSSpecCompiler/Compiler/ControlFlowGraph.cpp Normal file
View file

@ -0,0 +1,29 @@
/*
* Copyright (c) 2023, Dan Klishch <danilklishch@gmail.com>
*
* SPDX-License-Identifier: BSD-2-Clause
*/
#include <AK/StringBuilder.h>
#include "AST/AST.h"
#include "Compiler/ControlFlowGraph.h"
using namespace JSSpecCompiler;
ErrorOr<void> AK::Formatter<ControlFlowGraph>::format(FormatBuilder& format_builder, ControlFlowGraph const& control_flow_graph)
{
auto& builder = format_builder.builder();
for (auto const& block : control_flow_graph.blocks) {
builder.appendff("{}:\n", block->m_index);
for (auto const& expression : block->m_expressions)
builder.appendff("{}", expression);
builder.appendff("{}\n", Tree(block->m_continuation));
}
// Remove trailing \n
builder.trim(1);
return {};
}

9
Meta/Lagom/Tools/CodeGenerators/JSSpecCompiler/Compiler/ControlFlowGraph.h
View file

@ -39,3 +39,12 @@ public:
}; };
} }
namespace AK {
template<>
struct Formatter<JSSpecCompiler::ControlFlowGraph> : Formatter<StringView> {
ErrorOr<void> format(FormatBuilder& builder, JSSpecCompiler::ControlFlowGraph const& control_flow_graph);
};
}

14
Meta/Lagom/Tools/CodeGenerators/JSSpecCompiler/main.cpp
View file

@ -21,6 +21,7 @@ using namespace JSSpecCompiler;
struct CompilationStepWithDumpOptions { struct CompilationStepWithDumpOptions {
OwnPtr<CompilationStep> step; OwnPtr<CompilationStep> step;
bool dump_ast = false; bool dump_ast = false;
bool dump_cfg = false;
}; };
class CompilationPipeline { class CompilationPipeline {
@ -91,6 +92,9 @@ ErrorOr<int> serenity_main(Main::Arguments arguments)
StringView passes_to_dump_ast; StringView passes_to_dump_ast;
args_parser.add_option(passes_to_dump_ast, "Dump AST after specified passes.", "dump-ast", 0, "{all|last|<pass-name>|-<pass-name>[,...]}"); args_parser.add_option(passes_to_dump_ast, "Dump AST after specified passes.", "dump-ast", 0, "{all|last|<pass-name>|-<pass-name>[,...]}");
StringView passes_to_dump_cfg;
args_parser.add_option(passes_to_dump_cfg, "Dump CFG after specified passes.", "dump-cfg", 0, "{all|last|<pass-name>|-<pass-name>[,...]}");
args_parser.parse(arguments); args_parser.parse(arguments);
CompilationPipeline pipeline; CompilationPipeline pipeline;
@ -106,6 +110,9 @@ ErrorOr<int> serenity_main(Main::Arguments arguments)
pipeline.for_each_step_in(passes_to_dump_ast, [](CompilationStepWithDumpOptions& step) { pipeline.for_each_step_in(passes_to_dump_ast, [](CompilationStepWithDumpOptions& step) {
step.dump_ast = true; step.dump_ast = true;
}); });
pipeline.for_each_step_in(passes_to_dump_cfg, [](CompilationStepWithDumpOptions& step) {
step.dump_cfg = true;
});
TranslationUnit translation_unit; TranslationUnit translation_unit;
translation_unit.filename = filename; translation_unit.filename = filename;
@ -132,6 +139,13 @@ ErrorOr<int> serenity_main(Main::Arguments arguments)
outln(stderr, "{}", function->m_ast); outln(stderr, "{}", function->m_ast);
} }
} }
if (step.dump_cfg && translation_unit.function_definitions[0]->m_cfg != nullptr) {
outln(stderr, "===== CFG after {} =====", step.step->name());
for (auto const& function : translation_unit.function_definitions) {
outln(stderr, "{}():", function->m_name);
outln(stderr, "{}", *function->m_cfg);
}
}
} }
return 0; return 0;