jbitset: new module.

[ccan] / ccan / jbitset / jbitset_type.h

diff --git a/ccan/jbitset/jbitset_type.h b/ccan/jbitset/jbitset_type.h
new file mode 100644 (file)
index 0000000..abd8c2b
--- /dev/null
+++ b/ccan/jbitset/jbitset_type.h
@@ -0,0 +1,84 @@
+#ifndef CCAN_JBITSET_TYPE_H
+#define CCAN_JBITSET_TYPE_H
+#include <ccan/jbitset/jbitset.h>
+
+/**
+ * JBIT_DEFINE_TYPE - create a set of jbit ops for a given pointer type
+ * @type: a type whose pointers will go into the bitset.
+ * @name: a name for all the functions to define (of form jbit_<name>_*)
+ *
+ * This macro defines a set of inline functions for typesafe and convenient
+ * usage of a Judy bitset for pointers.  It is assumed that a NULL pointer
+ * is never set in the bitset.
+ *
+ * Example:
+ *     JBIT_DEFINE_TYPE(char, char);
+ *     JBIT_DEFINE_TYPE(struct foo, foo);
+ *
+ *     static struct jbitset_char *jc;
+ *     struct jbitset_foo *jf;
+ *
+ *     static void add_to_bitsets(const char *p, const struct foo *f)
+ *     {
+ *             // Note, this adds the pointer, not the string!
+ *             jbit_char_set(jc, p);
+ *             jbit_foo_set(jf, f);
+ *     }
+ */
+#define JBIT_DEFINE_TYPE(type, name)                                   \
+struct jbitset_##name;                                                 \
+static inline struct jbitset_##name *jbit_##name##_new(void)           \
+{                                                                      \
+       return (struct jbitset_##name *)jbit_new();                     \
+}                                                                      \
+static inline void jbit_##name##_free(const struct jbitset_##name *set)        \
+{                                                                      \
+       jbit_free((const struct jbitset *)set);                         \
+}                                                                      \
+static inline const char *jbit_##name##_error(struct jbitset_##name *set) \
+{                                                                      \
+       return jbit_error((struct jbitset *)set);                       \
+}                                                                      \
+static inline bool jbit_##name##_test(const struct jbitset_##name *set,        \
+                                     const type *index)                \
+{                                                                      \
+       return jbit_test((const struct jbitset *)set, (size_t)index);   \
+}                                                                      \
+static inline bool jbit_##name##_set(struct jbitset_##name *set,       \
+                                    const type *index)                 \
+{                                                                      \
+       return jbit_set((struct jbitset *)set, (size_t)index);          \
+}                                                                      \
+static inline bool jbit_##name##_clear(struct jbitset_##name *set,     \
+                                      const type *index)               \
+{                                                                      \
+       return jbit_clear((struct jbitset *)set, (size_t)index);        \
+}                                                                      \
+static inline size_t jbit_##name##_count(struct jbitset_##name *set)   \
+{                                                                      \
+       return jbit_popcount((const struct jbitset *)set, 0, -1);       \
+}                                                                      \
+static inline type *jbit_##name##_nth(const struct jbitset_##name *set,        \
+                                           size_t n)                   \
+{                                                                      \
+       return (type *)jbit_nth((const struct jbitset *)set, n, 0);     \
+}                                                                      \
+static inline type *jbit_##name##_first(const struct jbitset_##name *set) \
+{                                                                      \
+       return (type *)jbit_first((const struct jbitset *)set, 0);      \
+}                                                                      \
+static inline type *jbit_##name##_next(struct jbitset_##name *set,     \
+                                      const type *prev)                \
+{                                                                      \
+       return (type *)jbit_next((const struct jbitset *)set, (size_t)prev, 0); \
+}                                                                      \
+static inline type *jbit_##name##_last(struct jbitset_##name *set)     \
+{                                                                      \
+       return (type *)jbit_last((const struct jbitset *)set, 0);       \
+}                                                                      \
+static inline type *jbit_##name##_prev(struct jbitset_##name *set,     \
+                                      const type *prev)                \
+{                                                                      \
+       return (type *)jbit_prev((const struct jbitset *)set, (size_t)prev, 0); \
+}
+#endif /* CCAN_JBITSET_TYPE_H */