Igor Sysoev | d90282d | 2004-09-28 08:34:51 +0000 | [diff] [blame] | 1 | |
| 2 | /* |
Igor Sysoev | ff8da91 | 2004-09-29 16:00:49 +0000 | [diff] [blame] | 3 | * Copyright (C) Igor Sysoev |
Igor Sysoev | d90282d | 2004-09-28 08:34:51 +0000 | [diff] [blame] | 4 | */ |
| 5 | |
| 6 | |
Igor Sysoev | f5003d8 | 2003-12-04 14:53:00 +0000 | [diff] [blame] | 7 | #ifndef _NGX_RBTREE_H_INCLUDED_ |
| 8 | #define _NGX_RBTREE_H_INCLUDED_ |
| 9 | |
| 10 | |
| 11 | #include <ngx_config.h> |
| 12 | #include <ngx_core.h> |
| 13 | |
| 14 | |
Igor Sysoev | 208eed2 | 2005-10-07 13:30:52 +0000 | [diff] [blame] | 15 | typedef ngx_uint_t ngx_rbtree_key_t; |
| 16 | typedef ngx_int_t ngx_rbtree_key_int_t; |
| 17 | |
| 18 | |
Igor Sysoev | 1bfa7bc | 2005-10-10 12:59:41 +0000 | [diff] [blame] | 19 | typedef struct ngx_rbtree_node_s ngx_rbtree_node_t; |
Igor Sysoev | f5003d8 | 2003-12-04 14:53:00 +0000 | [diff] [blame] | 20 | |
Igor Sysoev | 1bfa7bc | 2005-10-10 12:59:41 +0000 | [diff] [blame] | 21 | struct ngx_rbtree_node_s { |
| 22 | ngx_rbtree_key_t key; |
| 23 | ngx_rbtree_node_t *left; |
| 24 | ngx_rbtree_node_t *right; |
| 25 | ngx_rbtree_node_t *parent; |
Igor Sysoev | 8c5f37e | 2006-11-16 15:34:52 +0000 | [diff] [blame] | 26 | u_char color; |
| 27 | u_char data; |
Igor Sysoev | f5003d8 | 2003-12-04 14:53:00 +0000 | [diff] [blame] | 28 | }; |
| 29 | |
Igor Sysoev | 62260f2 | 2003-12-05 17:07:27 +0000 | [diff] [blame] | 30 | |
Igor Sysoev | 1bfa7bc | 2005-10-10 12:59:41 +0000 | [diff] [blame] | 31 | typedef struct ngx_rbtree_s ngx_rbtree_t; |
| 32 | |
Igor Sysoev | 8c5f37e | 2006-11-16 15:34:52 +0000 | [diff] [blame] | 33 | typedef void (*ngx_rbtree_insert_pt) (ngx_rbtree_node_t *root, |
Igor Sysoev | 1bfa7bc | 2005-10-10 12:59:41 +0000 | [diff] [blame] | 34 | ngx_rbtree_node_t *node, ngx_rbtree_node_t *sentinel); |
| 35 | |
| 36 | struct ngx_rbtree_s { |
| 37 | ngx_rbtree_node_t *root; |
| 38 | ngx_rbtree_node_t *sentinel; |
Igor Sysoev | 8c5f37e | 2006-11-16 15:34:52 +0000 | [diff] [blame] | 39 | ngx_rbtree_insert_pt insert; |
Igor Sysoev | 1bfa7bc | 2005-10-10 12:59:41 +0000 | [diff] [blame] | 40 | }; |
Igor Sysoev | f5003d8 | 2003-12-04 14:53:00 +0000 | [diff] [blame] | 41 | |
| 42 | |
Igor Sysoev | 1bfa7bc | 2005-10-10 12:59:41 +0000 | [diff] [blame] | 43 | void ngx_rbtree_insert(ngx_thread_volatile ngx_rbtree_t *tree, |
| 44 | ngx_rbtree_node_t *node); |
| 45 | void ngx_rbtree_delete(ngx_thread_volatile ngx_rbtree_t *tree, |
| 46 | ngx_rbtree_node_t *node); |
Igor Sysoev | f4eb017 | 2006-11-20 17:13:21 +0000 | [diff] [blame] | 47 | void ngx_rbtree_insert_value(ngx_rbtree_node_t *root, ngx_rbtree_node_t *node, |
| 48 | ngx_rbtree_node_t *sentinel); |
Igor Sysoev | 8c5f37e | 2006-11-16 15:34:52 +0000 | [diff] [blame] | 49 | void ngx_rbtree_insert_timer_value(ngx_rbtree_node_t *root, |
| 50 | ngx_rbtree_node_t *node, ngx_rbtree_node_t *sentinel); |
Igor Sysoev | 1bfa7bc | 2005-10-10 12:59:41 +0000 | [diff] [blame] | 51 | |
| 52 | |
Igor Sysoev | 618dc75 | 2007-01-12 19:48:30 +0000 | [diff] [blame] | 53 | #define ngx_rbt_red(node) ((node)->color = 1) |
| 54 | #define ngx_rbt_black(node) ((node)->color = 0) |
| 55 | #define ngx_rbt_is_red(node) ((node)->color) |
| 56 | #define ngx_rbt_is_black(node) (!ngx_rbt_is_red(node)) |
| 57 | #define ngx_rbt_copy_color(n1, n2) (n1->color = n2->color) |
| 58 | |
| 59 | |
Igor Sysoev | dcec2fa | 2007-01-02 23:04:54 +0000 | [diff] [blame] | 60 | /* a sentinel must be black */ |
| 61 | |
Igor Sysoev | 618dc75 | 2007-01-12 19:48:30 +0000 | [diff] [blame] | 62 | #define ngx_rbtree_sentinel_init(node) ngx_rbt_black(node) |
Igor Sysoev | dcec2fa | 2007-01-02 23:04:54 +0000 | [diff] [blame] | 63 | |
| 64 | |
Igor Sysoev | 1bfa7bc | 2005-10-10 12:59:41 +0000 | [diff] [blame] | 65 | static ngx_inline ngx_rbtree_node_t * |
| 66 | ngx_rbtree_min(ngx_rbtree_node_t *node, ngx_rbtree_node_t *sentinel) |
Igor Sysoev | f5003d8 | 2003-12-04 14:53:00 +0000 | [diff] [blame] | 67 | { |
Igor Sysoev | 208eed2 | 2005-10-07 13:30:52 +0000 | [diff] [blame] | 68 | while (node->left != sentinel) { |
| 69 | node = node->left; |
| 70 | } |
Igor Sysoev | f5003d8 | 2003-12-04 14:53:00 +0000 | [diff] [blame] | 71 | |
Igor Sysoev | 208eed2 | 2005-10-07 13:30:52 +0000 | [diff] [blame] | 72 | return node; |
Igor Sysoev | f5003d8 | 2003-12-04 14:53:00 +0000 | [diff] [blame] | 73 | } |
| 74 | |
| 75 | |
| 76 | #endif /* _NGX_RBTREE_H_INCLUDED_ */ |