C-Math-Interpreter/nodes.c at master · jackprogramsjp/C-Math-Interpreter · GitHub

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
#include <stdio.h>
#include <stdlib.h>

#include "tokens.h"
#include "lexer.h"
#include "nodes.h"

void print_node(Node *node) {
    switch (node->node_type) {
        case NumberNode:
            printf("%lf", node->value);
            break;
        case PlusNode:
            if (node->node_a != NULL) {
                printf("(+");
                print_node(node->node_a);
                putchar(')');
            }
            break;
        case MinusNode:
            if (node->node_a != NULL) {
                printf("(-");
                print_node(node->node_a);
                putchar(')');
            }
            break;
        case AddNode:
            if (node->node_a != NULL && node->node_b != NULL) {
                putchar('(');
                print_node(node->node_a);
                printf(" + ");
                print_node(node->node_b);
                putchar(')');
            }
            break;
        case SubtractNode:
            if (node->node_a != NULL && node->node_b != NULL) {
                putchar('(');
                print_node(node->node_a);
                printf(" - ");
                print_node(node->node_b);
                putchar(')');
            }
            break;
        case MultiplyNode:
            if (node->node_a != NULL && node->node_b != NULL) {
                putchar('(');
                print_node(node->node_a);
                printf(" * ");
                print_node(node->node_b);
                putchar(')');
            }
            break;
        case DivideNode:
            if (node->node_a != NULL && node->node_b != NULL) {
                putchar('(');
                print_node(node->node_a);
                printf(" / ");
                print_node(node->node_b);
                putchar(')');
            }
            break;
        case PowerNode:
            if (node->node_a != NULL && node->node_b != NULL) {
                putchar('(');
                print_node(node->node_a);
                printf(" ** ");
                print_node(node->node_b);
                putchar(')');
            }
            break;
        default:
            printf("()");
    }
}

void free_node(Node *node) {
    if (node->node_a != NULL) {
        free_node(node->node_a);
        node->node_a = NULL;
    }

    if (node->node_b != NULL) {
        free_node(node->node_b);
        node->node_b = NULL;
    }

    free(node);
}

Node *node_init(NodeType node_type, double value, Node *node_a, Node *node_b) {
    Node *node = malloc(sizeof(Node));
    node->node_type = node_type;
    node->value = value;
    node->node_a = node_a;
    node->node_b = node_b;
    return node;
}