]> git.meshlink.io Git - meshlink/blob - src/splay_tree.h
ca53dc37c89d4cda66c3488e8b78340b893bbcf2
[meshlink] / src / splay_tree.h
1 #ifndef MESHLINK_SPLAY_TREE_H
2 #define MESHLINK_SPLAY_TREE_H
3
4 /*
5     splay_tree.h -- header file for splay_tree.c
6     Copyright (C) 2014, 2017 Guus Sliepen <guus@meshlink.io>
7
8     This program is free software; you can redistribute it and/or modify
9     it under the terms of the GNU General Public License as published by
10     the Free Software Foundation; either version 2 of the License, or
11     (at your option) any later version.
12
13     This program is distributed in the hope that it will be useful,
14     but WITHOUT ANY WARRANTY; without even the implied warranty of
15     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16     GNU General Public License for more details.
17
18     You should have received a copy of the GNU General Public License along
19     with this program; if not, write to the Free Software Foundation, Inc.,
20     51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
21 */
22
23 typedef struct splay_node_t {
24
25         /* Linked list part */
26
27         struct splay_node_t *next;
28         struct splay_node_t *prev;
29
30         /* Tree part */
31
32         struct splay_node_t *parent;
33         struct splay_node_t *left;
34         struct splay_node_t *right;
35
36         /* Payload */
37
38         void *data;
39
40 } splay_node_t;
41
42 typedef int (*splay_compare_t)(const void *, const void *);
43 typedef void (*splay_action_t)(const void *);
44 typedef void (*splay_action_node_t)(const splay_node_t *);
45
46 typedef struct splay_tree_t {
47
48         /* Linked list part */
49
50         splay_node_t *head;
51         splay_node_t *tail;
52
53         /* Tree part */
54
55         splay_node_t *root;
56
57         splay_compare_t compare;
58         splay_action_t delete;
59
60         unsigned int count;
61
62 } splay_tree_t;
63
64 /* (De)constructors */
65
66 extern splay_tree_t *splay_alloc_tree(splay_compare_t, splay_action_t) __attribute__((__malloc__));
67 extern void splay_delete_tree(splay_tree_t *);
68
69 extern splay_node_t *splay_alloc_node(void) __attribute__((__malloc__));
70 extern void splay_free_node(splay_tree_t *tree, splay_node_t *);
71
72 /* Insertion and deletion */
73
74 extern splay_node_t *splay_insert(splay_tree_t *, void *);
75 extern splay_node_t *splay_insert_node(splay_tree_t *, splay_node_t *);
76
77 extern splay_node_t *splay_unlink(splay_tree_t *, void *);
78 extern void splay_unlink_node(splay_tree_t *tree, splay_node_t *);
79 extern void splay_delete(splay_tree_t *, void *);
80 extern void splay_delete_node(splay_tree_t *, splay_node_t *);
81
82 /* Searching */
83
84 extern void *splay_search(splay_tree_t *, const void *);
85 extern void *splay_search_closest(splay_tree_t *, const void *, int *);
86 extern void *splay_search_closest_smaller(splay_tree_t *, const void *);
87 extern void *splay_search_closest_greater(splay_tree_t *, const void *);
88
89 extern splay_node_t *splay_search_node(splay_tree_t *, const void *);
90 extern splay_node_t *splay_search_closest_node(splay_tree_t *, const void *, int *);
91 extern splay_node_t *splay_search_closest_node_nosplay(const splay_tree_t *, const void *, int *);
92 extern splay_node_t *splay_search_closest_smaller_node(splay_tree_t *, const void *);
93 extern splay_node_t *splay_search_closest_greater_node(splay_tree_t *, const void *);
94
95 /* Tree walking */
96
97 extern void splay_foreach(const splay_tree_t *, splay_action_t);
98 extern void splay_foreach_node(const splay_tree_t *, splay_action_t);
99
100 #define splay_each(type, item, tree) (type *item = (type *)1; item; item = NULL) for(splay_node_t *node = (tree)->head, *next; item = node ? node->data : NULL, next = node ? node->next : NULL, node; node = next)
101
102 #endif