forked from vieter-v/libvieter
95 lines
2.6 KiB
C
95 lines
2.6 KiB
C
#ifndef VIETER_TREE_NODE
|
|
#define VIETER_TREE_NODE
|
|
|
|
#include "vieter_tree.h"
|
|
#include <stdbool.h>
|
|
#include <stdint.h>
|
|
|
|
typedef enum vieter_tree_node_flag {
|
|
vieter_tree_node_black = ((uint8_t)1) << 0,
|
|
// This flag is used to index the children array. If this flag isn't set,
|
|
// it'll evaluate to false, or index 0 in the children array.
|
|
vieter_tree_node_right = ((uint8_t)1) << 1
|
|
} vieter_tree_node_flag;
|
|
|
|
typedef struct vieter_tree_node {
|
|
uint64_t key;
|
|
void *data;
|
|
struct vieter_tree_node *parent;
|
|
struct vieter_tree_node *children[2];
|
|
uint8_t flags;
|
|
} vieter_tree_node;
|
|
|
|
/*
|
|
* Allocate a new node.
|
|
*/
|
|
vieter_tree_node *vieter_tree_node_init();
|
|
|
|
/*
|
|
* Free an allocated node.
|
|
*/
|
|
void vieter_tree_node_free(vieter_tree_node *node);
|
|
|
|
/*
|
|
* Insert a new key into the given tree.
|
|
*/
|
|
vieter_tree_error vieter_tree_node_insert(vieter_tree_node *root, uint64_t key,
|
|
void *data);
|
|
|
|
/*
|
|
* Return the node representing the requested value.
|
|
*/
|
|
vieter_tree_error vieter_tree_node_search_node(vieter_tree_node **out,
|
|
vieter_tree_node *root,
|
|
uint64_t key);
|
|
|
|
/*
|
|
* Search for the data represented by the given key.
|
|
*/
|
|
vieter_tree_error vieter_tree_node_search(void **out, vieter_tree_node *root,
|
|
uint64_t key);
|
|
|
|
/*
|
|
* Remove the data associated with the given key.
|
|
*/
|
|
vieter_tree_error
|
|
vieter_tree_node_remove(void **out, vieter_tree_node **root_ptr, uint64_t key);
|
|
|
|
/*
|
|
* Set a node's bit flag to the given value.
|
|
*/
|
|
void vieter_tree_node_set(vieter_tree_node *node, vieter_tree_node_flag flag,
|
|
bool value);
|
|
|
|
/*
|
|
* Get whether a node's bit flag is set.
|
|
*/
|
|
bool vieter_tree_node_get(vieter_tree_node *node, vieter_tree_node_flag flag);
|
|
|
|
/*
|
|
* Add a new child to the parent, replacing either its left or right child,
|
|
* depending on the key values.
|
|
*/
|
|
void vieter_tree_node_add_child(vieter_tree_node *parent, uint64_t key,
|
|
vieter_tree_node *child);
|
|
|
|
/*
|
|
* Replace a node's children array.
|
|
*/
|
|
void vieter_tree_node_set_children(vieter_tree_node *parent,
|
|
vieter_tree_node **children);
|
|
|
|
/*
|
|
* Set a node's left or right child to the given node.
|
|
*/
|
|
void vieter_tree_node_set_child(vieter_tree_node *parent,
|
|
vieter_tree_node *child, bool right);
|
|
|
|
/*
|
|
* Return the in-order successor of the given node, or NULL if it's the last
|
|
* node in the tree.
|
|
*/
|
|
vieter_tree_node *vieter_tree_node_next(vieter_tree_node *node);
|
|
|
|
#endif
|