feat(pretty-print): Just casually implemented the graphviz representation of parsed AST
This commit is contained in:
parent
ae2ddf0771
commit
f6573d81f0
3 changed files with 95 additions and 2 deletions
1
.gitignore
vendored
1
.gitignore
vendored
|
|
@ -175,3 +175,4 @@ m4/ltversion.m4
|
||||||
m4/lt~obsolete.m4
|
m4/lt~obsolete.m4
|
||||||
|
|
||||||
Makefile
|
Makefile
|
||||||
|
*.svg
|
||||||
|
|
|
||||||
|
|
@ -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);
|
||||||
|
|
|
||||||
|
|
@ -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 */
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue