1*4882a593Smuzhiyun /* SPDX-License-Identifier: GPL-2.0 */ 2*4882a593Smuzhiyun #ifndef __PERF_RBLIST_H 3*4882a593Smuzhiyun #define __PERF_RBLIST_H 4*4882a593Smuzhiyun 5*4882a593Smuzhiyun #include <linux/rbtree.h> 6*4882a593Smuzhiyun #include <stdbool.h> 7*4882a593Smuzhiyun 8*4882a593Smuzhiyun /* 9*4882a593Smuzhiyun * create node structs of the form: 10*4882a593Smuzhiyun * struct my_node { 11*4882a593Smuzhiyun * struct rb_node rb_node; 12*4882a593Smuzhiyun * ... my data ... 13*4882a593Smuzhiyun * }; 14*4882a593Smuzhiyun * 15*4882a593Smuzhiyun * create list structs of the form: 16*4882a593Smuzhiyun * struct mylist { 17*4882a593Smuzhiyun * struct rblist rblist; 18*4882a593Smuzhiyun * ... my data ... 19*4882a593Smuzhiyun * }; 20*4882a593Smuzhiyun */ 21*4882a593Smuzhiyun 22*4882a593Smuzhiyun struct rblist { 23*4882a593Smuzhiyun struct rb_root_cached entries; 24*4882a593Smuzhiyun unsigned int nr_entries; 25*4882a593Smuzhiyun 26*4882a593Smuzhiyun int (*node_cmp)(struct rb_node *rbn, const void *entry); 27*4882a593Smuzhiyun struct rb_node *(*node_new)(struct rblist *rlist, const void *new_entry); 28*4882a593Smuzhiyun void (*node_delete)(struct rblist *rblist, struct rb_node *rb_node); 29*4882a593Smuzhiyun }; 30*4882a593Smuzhiyun 31*4882a593Smuzhiyun void rblist__init(struct rblist *rblist); 32*4882a593Smuzhiyun void rblist__exit(struct rblist *rblist); 33*4882a593Smuzhiyun void rblist__delete(struct rblist *rblist); 34*4882a593Smuzhiyun int rblist__add_node(struct rblist *rblist, const void *new_entry); 35*4882a593Smuzhiyun void rblist__remove_node(struct rblist *rblist, struct rb_node *rb_node); 36*4882a593Smuzhiyun struct rb_node *rblist__find(struct rblist *rblist, const void *entry); 37*4882a593Smuzhiyun struct rb_node *rblist__findnew(struct rblist *rblist, const void *entry); 38*4882a593Smuzhiyun struct rb_node *rblist__entry(const struct rblist *rblist, unsigned int idx); 39*4882a593Smuzhiyun rblist__empty(const struct rblist * rblist)40*4882a593Smuzhiyunstatic inline bool rblist__empty(const struct rblist *rblist) 41*4882a593Smuzhiyun { 42*4882a593Smuzhiyun return rblist->nr_entries == 0; 43*4882a593Smuzhiyun } 44*4882a593Smuzhiyun rblist__nr_entries(const struct rblist * rblist)45*4882a593Smuzhiyunstatic inline unsigned int rblist__nr_entries(const struct rblist *rblist) 46*4882a593Smuzhiyun { 47*4882a593Smuzhiyun return rblist->nr_entries; 48*4882a593Smuzhiyun } 49*4882a593Smuzhiyun 50*4882a593Smuzhiyun #endif /* __PERF_RBLIST_H */ 51