1b2441318SGreg Kroah-Hartman /* SPDX-License-Identifier: GPL-2.0 */
28b40f521SJohn Kacur #ifndef __PERF_STRLIST_H
38b40f521SJohn Kacur #define __PERF_STRLIST_H
425903407SArnaldo Carvalho de Melo
543cbcd8aSArnaldo Carvalho de Melo #include <linux/rbtree.h>
625903407SArnaldo Carvalho de Melo #include <stdbool.h>
725903407SArnaldo Carvalho de Melo
8ee8dd3caSDavid Ahern #include "rblist.h"
9ee8dd3caSDavid Ahern
1025903407SArnaldo Carvalho de Melo struct str_node {
1125903407SArnaldo Carvalho de Melo struct rb_node rb_node;
1225903407SArnaldo Carvalho de Melo const char *s;
1325903407SArnaldo Carvalho de Melo };
1425903407SArnaldo Carvalho de Melo
1525903407SArnaldo Carvalho de Melo struct strlist {
16ee8dd3caSDavid Ahern struct rblist rblist;
1725903407SArnaldo Carvalho de Melo bool dupstr;
18dd8232bcSNamhyung Kim bool file_only;
1925903407SArnaldo Carvalho de Melo };
2025903407SArnaldo Carvalho de Melo
21dd8232bcSNamhyung Kim /*
22dd8232bcSNamhyung Kim * @file_only: When dirname is present, only consider entries as filenames,
23dd8232bcSNamhyung Kim * that should not be added to the list if dirname/entry is not
24dd8232bcSNamhyung Kim * found
25dd8232bcSNamhyung Kim */
264a77e218SArnaldo Carvalho de Melo struct strlist_config {
274a77e218SArnaldo Carvalho de Melo bool dont_dupstr;
28dd8232bcSNamhyung Kim bool file_only;
298ff9daf3SArnaldo Carvalho de Melo const char *dirname;
304a77e218SArnaldo Carvalho de Melo };
314a77e218SArnaldo Carvalho de Melo
324a77e218SArnaldo Carvalho de Melo struct strlist *strlist__new(const char *slist, const struct strlist_config *config);
33d8639f06SArnaldo Carvalho de Melo void strlist__delete(struct strlist *slist);
3425903407SArnaldo Carvalho de Melo
35d8639f06SArnaldo Carvalho de Melo void strlist__remove(struct strlist *slist, struct str_node *sn);
36d8639f06SArnaldo Carvalho de Melo int strlist__load(struct strlist *slist, const char *filename);
37d8639f06SArnaldo Carvalho de Melo int strlist__add(struct strlist *slist, const char *str);
3825903407SArnaldo Carvalho de Melo
39d8639f06SArnaldo Carvalho de Melo struct str_node *strlist__entry(const struct strlist *slist, unsigned int idx);
40d8639f06SArnaldo Carvalho de Melo struct str_node *strlist__find(struct strlist *slist, const char *entry);
413e340590SMasami Hiramatsu
strlist__has_entry(struct strlist * slist,const char * entry)42d8639f06SArnaldo Carvalho de Melo static inline bool strlist__has_entry(struct strlist *slist, const char *entry)
433e340590SMasami Hiramatsu {
44d8639f06SArnaldo Carvalho de Melo return strlist__find(slist, entry) != NULL;
453e340590SMasami Hiramatsu }
4625903407SArnaldo Carvalho de Melo
strlist__empty(const struct strlist * slist)47d8639f06SArnaldo Carvalho de Melo static inline bool strlist__empty(const struct strlist *slist)
4825903407SArnaldo Carvalho de Melo {
49d8639f06SArnaldo Carvalho de Melo return rblist__empty(&slist->rblist);
5027d0fd41SArnaldo Carvalho de Melo }
5127d0fd41SArnaldo Carvalho de Melo
strlist__nr_entries(const struct strlist * slist)52d8639f06SArnaldo Carvalho de Melo static inline unsigned int strlist__nr_entries(const struct strlist *slist)
5327d0fd41SArnaldo Carvalho de Melo {
54d8639f06SArnaldo Carvalho de Melo return rblist__nr_entries(&slist->rblist);
5525903407SArnaldo Carvalho de Melo }
5625903407SArnaldo Carvalho de Melo
57abf5ef72SMasami Hiramatsu /* For strlist iteration */
strlist__first(struct strlist * slist)58d8639f06SArnaldo Carvalho de Melo static inline struct str_node *strlist__first(struct strlist *slist)
59abf5ef72SMasami Hiramatsu {
60ca227029SDavidlohr Bueso struct rb_node *rn = rb_first_cached(&slist->rblist.entries);
61abf5ef72SMasami Hiramatsu return rn ? rb_entry(rn, struct str_node, rb_node) : NULL;
62abf5ef72SMasami Hiramatsu }
strlist__next(struct str_node * sn)63abf5ef72SMasami Hiramatsu static inline struct str_node *strlist__next(struct str_node *sn)
64abf5ef72SMasami Hiramatsu {
65abf5ef72SMasami Hiramatsu struct rb_node *rn;
66abf5ef72SMasami Hiramatsu if (!sn)
67abf5ef72SMasami Hiramatsu return NULL;
68abf5ef72SMasami Hiramatsu rn = rb_next(&sn->rb_node);
69abf5ef72SMasami Hiramatsu return rn ? rb_entry(rn, struct str_node, rb_node) : NULL;
70abf5ef72SMasami Hiramatsu }
71abf5ef72SMasami Hiramatsu
72abf5ef72SMasami Hiramatsu /**
73abf5ef72SMasami Hiramatsu * strlist_for_each - iterate over a strlist
74abf5ef72SMasami Hiramatsu * @pos: the &struct str_node to use as a loop cursor.
75d8639f06SArnaldo Carvalho de Melo * @slist: the &struct strlist for loop.
76abf5ef72SMasami Hiramatsu */
77602a1f4dSArnaldo Carvalho de Melo #define strlist__for_each_entry(pos, slist) \
78d8639f06SArnaldo Carvalho de Melo for (pos = strlist__first(slist); pos; pos = strlist__next(pos))
79abf5ef72SMasami Hiramatsu
80abf5ef72SMasami Hiramatsu /**
81abf5ef72SMasami Hiramatsu * strlist_for_each_safe - iterate over a strlist safe against removal of
82abf5ef72SMasami Hiramatsu * str_node
83abf5ef72SMasami Hiramatsu * @pos: the &struct str_node to use as a loop cursor.
84abf5ef72SMasami Hiramatsu * @n: another &struct str_node to use as temporary storage.
85d8639f06SArnaldo Carvalho de Melo * @slist: the &struct strlist for loop.
86abf5ef72SMasami Hiramatsu */
87602a1f4dSArnaldo Carvalho de Melo #define strlist__for_each_entry_safe(pos, n, slist) \
88d8639f06SArnaldo Carvalho de Melo for (pos = strlist__first(slist), n = strlist__next(pos); pos;\
89abf5ef72SMasami Hiramatsu pos = n, n = strlist__next(n))
908b40f521SJohn Kacur #endif /* __PERF_STRLIST_H */
91