123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128 |
- /*
- * 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
|