arch/s390/numa/toptree.h

   1 /* SPDX-License-Identifier: GPL-2.0 */
   2 /*
   3  * NUMA support for s390
   4  *
   5  * A tree structure used for machine topology mangling
   6  *
   7  * Copyright IBM Corp. 2015
   8  */
   9 #ifndef S390_TOPTREE_H
  10 #define S390_TOPTREE_H
  11
  12 #include <linux/cpumask.h>
  13 #include <linux/list.h>
  14
  15 struct toptree {
  16         int level;
  17         int id;
  18         cpumask_t mask;
  19         struct toptree *parent;
  20         struct list_head sibling;
  21         struct list_head children;
  22 };
  23
  24 struct toptree *toptree_alloc(int level, int id);
  25 void toptree_free(struct toptree *cand);
  26 void toptree_update_mask(struct toptree *cand);
  27 void toptree_unify(struct toptree *cand);
  28 struct toptree *toptree_get_child(struct toptree *cand, int id);
  29 void toptree_move(struct toptree *cand, struct toptree *target);
  30 int toptree_count(struct toptree *context, int level);
  31
  32 struct toptree *toptree_first(struct toptree *context, int level);
  33 struct toptree *toptree_next(struct toptree *cur, struct toptree *context,
  34                              int level);
  35
  36 #define toptree_for_each_child(child, ptree)                            \
  37         list_for_each_entry(child,  &ptree->children, sibling)
  38
  39 #define toptree_for_each_child_safe(child, ptmp, ptree)                 \
  40         list_for_each_entry_safe(child, ptmp, &ptree->children, sibling)
  41
  42 #define toptree_is_last(ptree)                                  \
  43         ((ptree->parent == NULL) ||                             \
  44          (ptree->parent->children.prev == &ptree->sibling))
  45
  46 #define toptree_for_each(ptree, cont, ttype)            \
  47         for (ptree = toptree_first(cont, ttype);        \
  48              ptree != NULL;                             \
  49              ptree = toptree_next(ptree, cont, ttype))
  50
  51 #define toptree_for_each_safe(ptree, tmp, cont, ttype)          \
  52         for (ptree = toptree_first(cont, ttype),                \
  53                      tmp = toptree_next(ptree, cont, ttype);    \
  54              ptree != NULL;                                     \
  55              ptree = tmp,                                       \
  56                      tmp = toptree_next(ptree, cont, ttype))
  57
  58 #define toptree_for_each_sibling(ptree, start)                  \
  59         toptree_for_each(ptree, start->parent, start->level)
  60
  61 #endif /* S390_TOPTREE_H */