xref: /openbmc/linux/tools/perf/util/intlist.h (revision a2cce7a9)
1 #ifndef __PERF_INTLIST_H
2 #define __PERF_INTLIST_H
3 
4 #include <linux/rbtree.h>
5 #include <stdbool.h>
6 
7 #include "rblist.h"
8 
9 struct int_node {
10 	struct rb_node rb_node;
11 	int i;
12 	void *priv;
13 };
14 
15 struct intlist {
16 	struct rblist rblist;
17 };
18 
19 struct intlist *intlist__new(const char *slist);
20 void intlist__delete(struct intlist *ilist);
21 
22 void intlist__remove(struct intlist *ilist, struct int_node *in);
23 int intlist__add(struct intlist *ilist, int i);
24 
25 struct int_node *intlist__entry(const struct intlist *ilist, unsigned int idx);
26 struct int_node *intlist__find(struct intlist *ilist, int i);
27 struct int_node *intlist__findnew(struct intlist *ilist, int i);
28 
29 static inline bool intlist__has_entry(struct intlist *ilist, int i)
30 {
31 	return intlist__find(ilist, i) != NULL;
32 }
33 
34 static inline bool intlist__empty(const struct intlist *ilist)
35 {
36 	return rblist__empty(&ilist->rblist);
37 }
38 
39 static inline unsigned int intlist__nr_entries(const struct intlist *ilist)
40 {
41 	return rblist__nr_entries(&ilist->rblist);
42 }
43 
44 /* For intlist iteration */
45 static inline struct int_node *intlist__first(struct intlist *ilist)
46 {
47 	struct rb_node *rn = rb_first(&ilist->rblist.entries);
48 	return rn ? rb_entry(rn, struct int_node, rb_node) : NULL;
49 }
50 static inline struct int_node *intlist__next(struct int_node *in)
51 {
52 	struct rb_node *rn;
53 	if (!in)
54 		return NULL;
55 	rn = rb_next(&in->rb_node);
56 	return rn ? rb_entry(rn, struct int_node, rb_node) : NULL;
57 }
58 
59 /**
60  * intlist_for_each      - iterate over a intlist
61  * @pos:	the &struct int_node to use as a loop cursor.
62  * @ilist:	the &struct intlist for loop.
63  */
64 #define intlist__for_each(pos, ilist)	\
65 	for (pos = intlist__first(ilist); pos; pos = intlist__next(pos))
66 
67 /**
68  * intlist_for_each_safe - iterate over a intlist safe against removal of
69  *                         int_node
70  * @pos:	the &struct int_node to use as a loop cursor.
71  * @n:		another &struct int_node to use as temporary storage.
72  * @ilist:	the &struct intlist for loop.
73  */
74 #define intlist__for_each_safe(pos, n, ilist)	\
75 	for (pos = intlist__first(ilist), n = intlist__next(pos); pos;\
76 	     pos = n, n = intlist__next(n))
77 #endif /* __PERF_INTLIST_H */
78