git.meshlink.io Git - meshlink/blob - 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