feat(pretty-print): Just casually implemented the graphviz representation of parsed AST

This commit is contained in:
Jean Herail 2026-01-14 20:42:31 +01:00
parent ae2ddf0771
commit f6573d81f0
3 changed files with 95 additions and 2 deletions

1
.gitignore vendored
View file

@ -175,3 +175,4 @@ m4/ltversion.m4
m4/lt~obsolete.m4 m4/lt~obsolete.m4
Makefile Makefile
*.svg

View file

@ -1,9 +1,96 @@
#include "ast.h"
#include <assert.h> #include <assert.h>
#include <stdbool.h> #include <stdbool.h>
#include <stdio.h>
#include <stdlib.h> #include <stdlib.h>
#include "ast.h"
static void ast_print_dot_recursive(struct ast *node, FILE *out)
{
if (!node)
return;
switch (node->type)
{
case AST_IF:
{
struct ast_if *if_data = ast_get_if(node);
fprintf(out, " node%p [label=\"IF\"];\n", (void *)node);
if (if_data->condition)
{
fprintf(out, " node%p -> node%p;\n", (void *)node,
(void *)if_data->condition);
fprintf(out, " node%p [fillcolor=\"lightyellow\", style=\"filled\"];\n",
(void *)if_data->condition);
ast_print_dot_recursive(if_data->condition, out);
if (if_data->then_clause)
{
fprintf(out, " node%p -> node%p [label=\"true\"];\n",
(void *)if_data->condition, (void *)if_data->then_clause);
ast_print_dot_recursive(if_data->then_clause, out);
}
if (if_data->else_clause)
{
fprintf(out, " node%p -> node%p [label=\"false\"];\n",
(void *)if_data->condition, (void *)if_data->else_clause);
ast_print_dot_recursive(if_data->else_clause, out);
}
}
break;
}
case AST_CMD:
{
struct ast_cmd *cmd_data = ast_get_cmd(node);
fprintf(out, " node%p [label=\"", (void *)node);
struct list *l = cmd_data->cmd;
while (l)
{
fprintf(out, "%s", (char *)l->data);
if (l->next)
fprintf(out, " ");
l = l->next;
}
fprintf(out, "\"];\n");
break;
}
case AST_END:
fprintf(out, " node%p [label=\"END\"];\n", (void *)node);
break;
default:
break;
}
}
void ast_print_dot(struct ast *ast)
{
// Always print to stdout so the user can see it in the terminal
printf("digraph AST {\n");
if (ast)
{
ast_print_dot_recursive(ast, stdout);
}
printf("}\n");
if (!ast)
return;
FILE *dot_pipe = popen("dot -Tsvg -o ast.svg", "w");
if (!dot_pipe)
{
return;
}
fprintf(dot_pipe, "digraph AST {\n");
ast_print_dot_recursive(ast, dot_pipe);
fprintf(dot_pipe, "}\n");
pclose(dot_pipe);
system("open ast.svg");
}
bool ast_is_if(struct ast *node) bool ast_is_if(struct ast *node)
{ {
assert(node != NULL); assert(node != NULL);

View file

@ -70,4 +70,9 @@ struct ast *ast_create_if(struct ast *condition, struct ast *then_clause,
*/ */
struct ast *ast_create_cmd(struct list *cmd); struct ast *ast_create_cmd(struct list *cmd);
/**
* Prints the Graphviz DOT representation of the given AST to stdout.
*/
void ast_print_dot(struct ast *ast);
#endif /* ! AST_H */ #endif /* ! AST_H */