libcfs/include/libcfs/user-bitops.h

   1 /* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
   2  * vim:expandtab:shiftwidth=8:tabstop=8:
   3  *
   4  * GPL HEADER START
   5  *
   6  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   7  *
   8  * This program is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License version 2 only,
  10  * as published by the Free Software Foundation.
  11  *
  12  * This program is distributed in the hope that it will be useful, but
  13  * WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * General Public License version 2 for more details (a copy is included
  16  * in the LICENSE file that accompanied this code).
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * version 2 along with this program; If not, see
  20  * http://www.sun.com/software/products/lustre/docs/GPLv2.pdf
  21  *
  22  * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
  23  * CA 95054 USA or visit www.sun.com if you need additional information or
  24  * have any questions.
  25  *
  26  * GPL HEADER END
  27  */
  28 /*
  29  * Copyright  2008 Sun Microsystems, Inc. All rights reserved
  30  * Use is subject to license terms.
  31  */
  32 /*
  33  * This file is part of Lustre, http://www.lustre.org/
  34  * Lustre is a trademark of Sun Microsystems, Inc.
  35  *
  36  * libcfs/include/libcfs/user-bitops.h
  37  *
  38  * Author: Nikita Danilov <nikita@clusterfs.com>
  39  */
  40
  41 #ifndef __LIBCFS_USER_BITOPS_H__
  42 #define __LIBCFS_USER_BITOPS_H__
  43
  44 /* test if bit nr is set in bitmap addr; returns previous value of bit nr */
  45 static __inline__ int set_bit(int nr, unsigned long *addr)
  46 {
  47         unsigned long mask;
  48
  49         addr += nr / BITS_PER_LONG;
  50         mask = 1UL << (nr & (BITS_PER_LONG - 1));
  51         nr = (mask & *addr) != 0;
  52         *addr |= mask;
  53         return nr;
  54 }
  55
  56 /* clear bit nr in bitmap addr; returns previous value of bit nr*/
  57 static __inline__ int clear_bit(int nr, unsigned long *addr)
  58 {
  59         unsigned long mask;
  60
  61         addr += nr / BITS_PER_LONG;
  62         mask = 1UL << (nr & (BITS_PER_LONG - 1));
  63         nr = (mask & *addr) != 0;
  64         *addr &= ~mask;
  65         return nr;
  66 }
  67
  68 static __inline__ int test_bit(int nr, const unsigned long *addr)
  69 {
  70         return ((1UL << (nr & (BITS_PER_LONG - 1))) &
  71                 ((addr)[nr / BITS_PER_LONG])) != 0;
  72 }
  73
  74 /* using binary seach */
  75 static __inline__ unsigned long __fls(long data)
  76 {
  77         int pos = 32;
  78
  79         if (!data)
  80                 return 0;
  81
  82 #if BITS_PER_LONG == 64
  83         pos += 32;
  84
  85         if ((data & 0xFFFFFFFF) == 0) {
  86                 data <<= 32;
  87                 pos -= 32;
  88         }
  89 #endif
  90
  91         if (!(data & 0xFFFF0000u)) {
  92                 data <<= 16;
  93                 pos -= 16;
  94         }
  95         if (!(data & 0xFF000000u)) {
  96                 data <<= 8;
  97                 pos -= 8;
  98         }
  99         if (!(data & 0xF0000000u)) {
 100                 data <<= 4;
 101                 pos -= 4;
 102         }
 103         if (!(data & 0xC0000000u)) {
 104                 data <<= 2;
 105                 pos -= 2;
 106         }
 107         if (!(data & 0x80000000u)) {
 108                 data <<= 1;
 109                 pos -= 1;
 110         }
 111         return pos;
 112 }
 113
 114 static __inline__ unsigned long __ffs(long data)
 115 {
 116         int pos = 0;
 117
 118 #if BITS_PER_LONG == 64
 119         if ((data & 0xFFFFFFFF) == 0) {
 120                 pos += 32;
 121                 data >>= 32;
 122         }
 123 #endif
 124         if ((data & 0xFFFF) == 0) {
 125                 pos += 16;
 126                 data >>= 16;
 127         }
 128         if ((data & 0xFF) == 0) {
 129                 pos += 8;
 130                 data >>= 8;
 131         }
 132         if ((data & 0xF) == 0) {
 133                 pos += 4;
 134                 data >>= 4;
 135         }
 136         if ((data & 0x3) == 0) {
 137                 pos += 2;
 138                 data >>= 2;
 139         }
 140         if ((data & 0x1) == 0)
 141                 pos += 1;
 142
 143         return pos;
 144 }
 145
 146 #define __ffz(x)        __ffs(~(x))
 147 #define __flz(x)        __fls(~(x))
 148
 149 unsigned long find_next_bit(unsigned long *addr,
 150                             unsigned long size, unsigned long offset);
 151
 152 unsigned long find_next_zero_bit(unsigned long *addr,
 153                                  unsigned long size, unsigned long offset);
 154
 155 #define find_first_bit(addr,size)       (find_next_bit((addr),(size),0))
 156 #define find_first_zero_bit(addr,size)  (find_next_zero_bit((addr),(size),0))
 157
 158 #endif