hachem.wtf
aboutsummaryrefslogtreecommitdiff
path: root/ast.c
blob: 58cbaf5b1f288a231e8a90934e20cb9f8f6aaea2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
#include "ast.h"
#include "lexer.h"

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

void name_init(struct Name* name)
{
    name->length = 0;
    name->text[0] = '\0';
}

bool name_push(struct Name* name, char character)
{
    if (name->length == NAME_LENGTH_MAX)
        return false;

    name->text[name->length++] = character;
    name->text[name->length] = '\0';
    return true;
}

static struct Node* node_new(enum NodeKind kind)
{
    struct Node* node = calloc(1, sizeof *node);
    if (node == NULL)
    {
        (void)fprintf(stderr, "out of memory\n");
        exit(1);
    }
    node->kind = kind;
    node->line = yylineno;
    return node;
}

struct Node* node_leaf(enum NodeKind kind)
{
    return node_new(kind);
}

struct Node* node_char(char character)
{
    struct Node* node = node_new(NODE_CHAR);
    node->character = character;
    return node;
}

struct Node* node_unary(enum NodeKind kind, struct Node* first)
{
    struct Node* node = node_new(kind);
    node->first = first;
    return node;
}

struct Node* node_binary(enum NodeKind kind, struct Node* first, struct Node* second)
{
    struct Node* node = node_new(kind);
    node->first = first;
    node->second = second;
    return node;
}

struct Node* node_ternary(enum NodeKind kind, struct Node* first, struct Node* second,
                          struct Node* third)
{
    struct Node* node = node_new(kind);
    node->first = first;
    node->second = second;
    node->third = third;
    return node;
}

struct Node* node_named(enum NodeKind kind, const struct Name* name, struct Node* first)
{
    struct Node* node = node_new(kind);
    (void)snprintf(node->name, sizeof node->name, "%s", name->text);
    node->first = first;
    return node;
}

struct Node* node_block(enum NodeKind kind, struct Node* first, struct Node* body)
{
    struct Node* node = node_new(kind);
    node->first = first;
    node->body = body;
    return node;
}

struct Node* node_append(struct Node* list, struct Node* node)
{
    struct Node* last = list;

    if (list == NULL)
        return node;
    while (last->next != NULL)
        last = last->next;
    last->next = node;
    return list;
}

void node_free(struct Node* node)
{
    if (node == NULL)
        return;
    node_free(node->first);
    node_free(node->second);
    node_free(node->third);
    node_free(node->body);
    node_free(node->next);
    free(node);
}