aboutsummaryrefslogtreecommitdiff
path: root/source/tree.c
blob: dcae48598ef17941b2541dd4292e7b7a69f5045a (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

#include "kpl.h"

extern inline void kpl_tree_init(kpl_tree *tree, kpl_tree_node_compare_fn node_compare_fn,
    kpl_tree_node_free_fn node_free_fn);

static void kpl_tree_free_helper(kpl_tree_node *node, kpl_tree_node_free_fn node_free_fn) {
    if (node->tree_left)
        kpl_tree_free_helper(node->tree_left, node_free_fn);
    if (node->tree_right)
        kpl_tree_free_helper(node->tree_right, node_free_fn);
    node_free_fn(node);
}

void kpl_tree_free(kpl_tree *tree) {
    kpl_tree_free_helper(tree->root_node, tree->node_free_fn);
}

static void *kpl_tree_find_helper(kpl_tree_node *root_node, const void *find_node,
    kpl_tree_node_compare_fn compare_fn) {
    if (!root_node)
        return NULL;
    int32_t compare = compare_fn(root_node, find_node);
    if (compare < 0)
        return kpl_tree_find_helper(root_node->tree_left, find_node, compare_fn);
    if (compare > 0)
        return kpl_tree_find_helper(root_node->tree_right, find_node, compare_fn);
    return root_node;
}

void *kpl_tree_find(kpl_tree *tree, const void *find_node) {
    return kpl_tree_find_helper(tree->root_node, find_node, tree->node_compare_fn);
}

void kpl_tree_insert(kpl_tree *tree, void *insert_node) {
    // TODO
}