git.ozlabs.org Git - ccan/blob - ccan/bitmap/test/run-ffs.c

   1 #include <ccan/bitmap/bitmap.h>
   2 #include <ccan/tap/tap.h>
   3 #include <ccan/array_size/array_size.h>
   4 #include <ccan/foreach/foreach.h>
   5
   6 #include <ccan/bitmap/bitmap.c>
   7
   8 int bitmap_sizes[] = {
   9         1, 2, 3, 4, 5, 6, 7, 8,
  10         16, 17, 24, 32, 33,
  11         64, 65, 127, 128, 129,
  12         1023, 1024, 1025,
  13 };
  14 #define NSIZES ARRAY_SIZE(bitmap_sizes)
  15
  16 #define ok_eq(a, b) \
  17         ok((a) == (b), "%s [%u] == %s [%u]", \
  18            #a, (unsigned)(a), #b, (unsigned)(b))
  19
  20 static void test_size(int nbits)
  21 {
  22         BITMAP_DECLARE(bitmap, nbits);
  23         int i;
  24
  25         bitmap_zero(bitmap, nbits);
  26         ok_eq(bitmap_ffs(bitmap, 0, nbits), nbits);
  27
  28         for (i = 0; i < nbits; i++) {
  29                 bitmap_zero(bitmap, nbits);
  30                 bitmap_set_bit(bitmap, i);
  31
  32                 ok_eq(bitmap_ffs(bitmap, 0, nbits), i);
  33                 ok_eq(bitmap_ffs(bitmap, i, nbits), i);
  34                 ok_eq(bitmap_ffs(bitmap, i + 1, nbits), nbits);
  35
  36                 bitmap_zero(bitmap, nbits);
  37                 bitmap_fill_range(bitmap, i, nbits);
  38
  39                 ok_eq(bitmap_ffs(bitmap, 0, nbits), i);
  40                 ok_eq(bitmap_ffs(bitmap, i, nbits), i);
  41                 ok_eq(bitmap_ffs(bitmap, i + 1, nbits), (i + 1));
  42                 ok_eq(bitmap_ffs(bitmap, nbits - 1, nbits), (nbits - 1));
  43
  44                 if (i > 0) {
  45                         ok_eq(bitmap_ffs(bitmap, 0, i), i);
  46                         ok_eq(bitmap_ffs(bitmap, 0, i - 1), (i - 1));
  47                 }
  48
  49                 if (i > 0) {
  50                         bitmap_zero(bitmap, nbits);
  51                         bitmap_fill_range(bitmap, 0, i);
  52
  53                         ok_eq(bitmap_ffs(bitmap, 0, nbits), 0);
  54                         ok_eq(bitmap_ffs(bitmap, i - 1, nbits), (i - 1));
  55                         ok_eq(bitmap_ffs(bitmap, i, nbits), nbits);
  56                 }
  57         }
  58 }
  59
  60 int main(void)
  61 {
  62         int i;
  63
  64         /* Too complicated to work out the exact number */
  65         plan_no_plan();
  66
  67         for (i = 0; i < NSIZES; i++) {
  68                 diag("Testing %d-bit bitmap", bitmap_sizes[i]);
  69                 test_size(bitmap_sizes[i]);
  70         }
  71
  72         /* This exits depending on whether all tests passed */
  73         return exit_status();
  74 }