libcfs/include/libcfs/bitmap.h

   1 /*
   2  * GPL HEADER START
   3  *
   4  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License version 2 only,
   8  * as published by the Free Software Foundation.
   9  *
  10  * This program is distributed in the hope that it will be useful, but
  11  * WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * General Public License version 2 for more details (a copy is included
  14  * in the LICENSE file that accompanied this code).
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * version 2 along with this program; If not, see
  18  * http://www.gnu.org/licenses/gpl-2.0.html
  19  *
  20  * GPL HEADER END
  21  */
  22 /*
  23  * Copyright (c) 2008, 2010, Oracle and/or its affiliates. All rights reserved.
  24  * Use is subject to license terms.
  25  *
  26  * Copyright (c) 2012, 2014, Intel Corporation.
  27  */
  28 /*
  29  * This file is part of Lustre, http://www.lustre.org/
  30  * Lustre is a trademark of Sun Microsystems, Inc.
  31  */
  32 #ifndef _LIBCFS_BITMAP_H_
  33 #define _LIBCFS_BITMAP_H_
  34
  35 #include <linux/interrupt.h>
  36 #include <libcfs/libcfs_private.h>
  37
  38 struct cfs_bitmap {
  39         unsigned int size;
  40         unsigned long data[0];
  41 };
  42
  43 #define CFS_BITMAP_SIZE(nbits) \
  44         (BITS_TO_LONGS(nbits) * sizeof(long) + sizeof(struct cfs_bitmap))
  45
  46 static inline
  47 struct cfs_bitmap *CFS_ALLOCATE_BITMAP(int size)
  48 {
  49         struct cfs_bitmap *ptr;
  50
  51         LIBCFS_ALLOC(ptr, CFS_BITMAP_SIZE(size));
  52         if (ptr == NULL)
  53                 RETURN(ptr);
  54
  55         ptr->size = size;
  56
  57         RETURN(ptr);
  58 }
  59
  60 static inline void CFS_RESET_BITMAP(struct cfs_bitmap *bitmap)
  61 {
  62         if (bitmap->size > 0) {
  63                 int nbits = bitmap->size;
  64
  65                 memset(bitmap, 0, CFS_BITMAP_SIZE(nbits));
  66                 bitmap->size = nbits;
  67         }
  68 }
  69
  70 #define CFS_FREE_BITMAP(ptr)    LIBCFS_FREE(ptr, CFS_BITMAP_SIZE(ptr->size))
  71
  72 static inline
  73 void cfs_bitmap_set(struct cfs_bitmap *bitmap, int nbit)
  74 {
  75         set_bit(nbit, bitmap->data);
  76 }
  77
  78 static inline
  79 void cfs_bitmap_clear(struct cfs_bitmap *bitmap, int nbit)
  80 {
  81         test_and_clear_bit(nbit, bitmap->data);
  82 }
  83
  84 static inline
  85 int cfs_bitmap_check(struct cfs_bitmap *bitmap, int nbit)
  86 {
  87         return test_bit(nbit, bitmap->data);
  88 }
  89
  90 static inline
  91 int cfs_bitmap_test_and_clear(struct cfs_bitmap *bitmap, int nbit)
  92 {
  93         return test_and_clear_bit(nbit, bitmap->data);
  94 }
  95
  96 /* return 0 is bitmap has none set bits */
  97 static inline
  98 int cfs_bitmap_check_empty(struct cfs_bitmap *bitmap)
  99 {
 100         return find_first_bit(bitmap->data, bitmap->size) == bitmap->size;
 101 }
 102
 103 static inline
 104 void cfs_bitmap_copy(struct cfs_bitmap *new, struct cfs_bitmap *old)
 105 {
 106         size_t newsize;
 107
 108         LASSERT(new->size >= old->size);
 109         newsize = new->size;
 110         memcpy(new, old, CFS_BITMAP_SIZE(old->size));
 111         new->size = newsize;
 112 }
 113
 114 #define cfs_foreach_bit(bitmap, pos)                                    \
 115         for ((pos) = find_first_bit((bitmap)->data, bitmap->size);      \
 116              (pos) < (bitmap)->size;                                    \
 117              (pos) = find_next_bit((bitmap)->data, (bitmap)->size, (pos) + 1))
 118
 119 #endif