|
@@ -0,0 +1,128 @@
|
|
|
+/*
|
|
|
+ * SPDX-License-Identifier: BSD-2-Clause
|
|
|
+ *
|
|
|
+ * Copyright (c) 2020 Western Digital Corporation or its affiliates.
|
|
|
+ *
|
|
|
+ * Authors:
|
|
|
+ * Anup Patel <anup.patel@wdc.com>
|
|
|
+ */
|
|
|
+
|
|
|
+#ifndef __SBI_BITMAP_H__
|
|
|
+#define __SBI_BITMAP_H__
|
|
|
+
|
|
|
+#include <sbi/sbi_bitops.h>
|
|
|
+
|
|
|
+#define BITMAP_FIRST_WORD_MASK(start) (~0UL << ((start) % BITS_PER_LONG))
|
|
|
+#define BITMAP_LAST_WORD_MASK(nbits) \
|
|
|
+( \
|
|
|
+ ((nbits) % BITS_PER_LONG) ? \
|
|
|
+ ((1UL << ((nbits) % BITS_PER_LONG)) - 1) : ~0UL \
|
|
|
+)
|
|
|
+
|
|
|
+#define small_const_nbits(nbits) \
|
|
|
+ (__builtin_constant_p(nbits) && (nbits) <= BITS_PER_LONG)
|
|
|
+
|
|
|
+#define DECLARE_BITMAP(name, nbits) unsigned long name[BITS_TO_LONGS(nbits)]
|
|
|
+#define DEFINE_BITMAP(name) extern unsigned long name[]
|
|
|
+
|
|
|
+static inline unsigned long bitmap_estimate_size(int nbits)
|
|
|
+{
|
|
|
+ return (BITS_TO_LONGS(nbits) * sizeof(unsigned long));
|
|
|
+}
|
|
|
+
|
|
|
+void __bitmap_and(unsigned long *dst, const unsigned long *bitmap1,
|
|
|
+ const unsigned long *bitmap2, int bits);
|
|
|
+void __bitmap_or(unsigned long *dst, const unsigned long *bitmap1,
|
|
|
+ const unsigned long *bitmap2, int bits);
|
|
|
+void __bitmap_xor(unsigned long *dst, const unsigned long *bitmap1,
|
|
|
+ const unsigned long *bitmap2, int bits);
|
|
|
+
|
|
|
+static inline void bitmap_set(unsigned long *bmap, int start, int len)
|
|
|
+{
|
|
|
+ int bit;
|
|
|
+ for (bit = start; bit < (start + len); bit++)
|
|
|
+ bmap[BIT_WORD(bit)] |= (0x1UL << BIT_WORD_OFFSET(bit));
|
|
|
+}
|
|
|
+
|
|
|
+static inline void bitmap_clear(unsigned long *bmap, int start, int len)
|
|
|
+{
|
|
|
+ int bit;
|
|
|
+ for (bit = start; bit < (start + len); bit++)
|
|
|
+ bmap[BIT_WORD(bit)] &= ~(0x1UL << BIT_WORD_OFFSET(bit));
|
|
|
+}
|
|
|
+
|
|
|
+static inline void bitmap_zero(unsigned long *dst, int nbits)
|
|
|
+{
|
|
|
+ if (small_const_nbits(nbits))
|
|
|
+ *dst = 0UL;
|
|
|
+ else {
|
|
|
+ size_t i, len = BITS_TO_LONGS(nbits);
|
|
|
+ for (i = 0; i < len; i++)
|
|
|
+ dst[i] = 0;
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+static inline void bitmap_zero_except(unsigned long *dst,
|
|
|
+ int exception, int nbits)
|
|
|
+{
|
|
|
+ if (small_const_nbits(nbits))
|
|
|
+ *dst = 0UL;
|
|
|
+ else {
|
|
|
+ size_t i, len = BITS_TO_LONGS(nbits);
|
|
|
+ for (i = 0; i < len; i++)
|
|
|
+ dst[i] = 0;
|
|
|
+ }
|
|
|
+ if (exception < nbits)
|
|
|
+ __set_bit(exception, dst);
|
|
|
+}
|
|
|
+
|
|
|
+static inline void bitmap_fill(unsigned long *dst, int nbits)
|
|
|
+{
|
|
|
+ size_t i, nlongs = BITS_TO_LONGS(nbits);
|
|
|
+ if (!small_const_nbits(nbits)) {
|
|
|
+ for (i = 0; i < (nlongs - 1); i++)
|
|
|
+ dst[i] = -1UL;
|
|
|
+ }
|
|
|
+ dst[nlongs - 1] = BITMAP_LAST_WORD_MASK(nbits);
|
|
|
+}
|
|
|
+
|
|
|
+static inline void bitmap_copy(unsigned long *dst,
|
|
|
+ const unsigned long *src, int nbits)
|
|
|
+{
|
|
|
+ if (small_const_nbits(nbits))
|
|
|
+ *dst = *src;
|
|
|
+ else {
|
|
|
+ size_t i, len = BITS_TO_LONGS(nbits);
|
|
|
+ for (i = 0; i < len; i++)
|
|
|
+ dst[i] = src[i];
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+static inline void bitmap_and(unsigned long *dst, const unsigned long *src1,
|
|
|
+ const unsigned long *src2, int nbits)
|
|
|
+{
|
|
|
+ if (small_const_nbits(nbits))
|
|
|
+ *dst = *src1 & *src2;
|
|
|
+ else
|
|
|
+ __bitmap_and(dst, src1, src2, nbits);
|
|
|
+}
|
|
|
+
|
|
|
+static inline void bitmap_or(unsigned long *dst, const unsigned long *src1,
|
|
|
+ const unsigned long *src2, int nbits)
|
|
|
+{
|
|
|
+ if (small_const_nbits(nbits))
|
|
|
+ *dst = *src1 | *src2;
|
|
|
+ else
|
|
|
+ __bitmap_or(dst, src1, src2, nbits);
|
|
|
+}
|
|
|
+
|
|
|
+static inline void bitmap_xor(unsigned long *dst, const unsigned long *src1,
|
|
|
+ const unsigned long *src2, int nbits)
|
|
|
+{
|
|
|
+ if (small_const_nbits(nbits))
|
|
|
+ *dst = *src1 ^ *src2;
|
|
|
+ else
|
|
|
+ __bitmap_xor(dst, src1, src2, nbits);
|
|
|
+}
|
|
|
+
|
|
|
+#endif
|