Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (c) 2015 Cisco and/or its affiliates. |
| 3 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 4 | * you may not use this file except in compliance with the License. |
| 5 | * You may obtain a copy of the License at: |
| 6 | * |
| 7 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 8 | * |
| 9 | * Unless required by applicable law or agreed to in writing, software |
| 10 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 11 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 12 | * See the License for the specific language governing permissions and |
| 13 | * limitations under the License. |
| 14 | */ |
| 15 | /* |
| 16 | Copyright (c) 2001, 2002, 2003, 2005 Eliot Dresselhaus |
| 17 | |
| 18 | Permission is hereby granted, free of charge, to any person obtaining |
| 19 | a copy of this software and associated documentation files (the |
| 20 | "Software"), to deal in the Software without restriction, including |
| 21 | without limitation the rights to use, copy, modify, merge, publish, |
| 22 | distribute, sublicense, and/or sell copies of the Software, and to |
| 23 | permit persons to whom the Software is furnished to do so, subject to |
| 24 | the following conditions: |
| 25 | |
| 26 | The above copyright notice and this permission notice shall be |
| 27 | included in all copies or substantial portions of the Software. |
| 28 | |
| 29 | THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, |
| 30 | EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF |
| 31 | MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND |
| 32 | NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE |
| 33 | LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION |
| 34 | OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION |
| 35 | WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. |
| 36 | */ |
| 37 | |
| 38 | #ifndef included_clib_bitmap_h |
| 39 | #define included_clib_bitmap_h |
| 40 | |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 41 | /** \file |
| 42 | Bitmaps built as vectors of machine words |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 43 | */ |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 44 | |
| 45 | #include <vppinfra/vec.h> |
| 46 | #include <vppinfra/random.h> |
| 47 | #include <vppinfra/error.h> |
| 48 | #include <vppinfra/bitops.h> /* for count_set_bits */ |
| 49 | |
Damjan Marion | 0b14072 | 2016-06-14 00:36:09 +0200 | [diff] [blame] | 50 | typedef uword clib_bitmap_t; |
| 51 | |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 52 | /** predicate function; is an entire bitmap empty? |
| 53 | @param ai - pointer to a bitmap |
| 54 | @returns 1 if the entire bitmap is zero, 0 otherwise |
| 55 | */ |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 56 | always_inline uword |
| 57 | clib_bitmap_is_zero (uword * ai) |
| 58 | { |
| 59 | uword i; |
| 60 | for (i = 0; i < vec_len (ai); i++) |
| 61 | if (ai[i] != 0) |
| 62 | return 0; |
| 63 | return 1; |
| 64 | } |
| 65 | |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 66 | /** predicate function; are two bitmaps equal? |
| 67 | @param a - pointer to a bitmap |
| 68 | @param b - pointer to a bitmap |
| 69 | @returns 1 if the bitmaps are equal, 0 otherwise |
| 70 | */ |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 71 | always_inline uword |
| 72 | clib_bitmap_is_equal (uword * a, uword * b) |
| 73 | { |
| 74 | uword i; |
| 75 | if (vec_len (a) != vec_len (b)) |
| 76 | return 0; |
| 77 | for (i = 0; i < vec_len (a); i++) |
| 78 | if (a[i] != b[i]) |
| 79 | return 0; |
| 80 | return 1; |
| 81 | } |
| 82 | |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 83 | /** Duplicate a bitmap |
Chris Luke | d4024f5 | 2016-09-06 09:32:36 -0400 | [diff] [blame] | 84 | @param v - pointer to a bitmap |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 85 | @returns a duplicate of the bitmap |
| 86 | */ |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 87 | #define clib_bitmap_dup(v) vec_dup(v) |
| 88 | |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 89 | /** Free a bitmap |
| 90 | @param v - pointer to the bitmap to free |
| 91 | */ |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 92 | #define clib_bitmap_free(v) vec_free(v) |
| 93 | |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 94 | /** Number of bytes in a bitmap |
| 95 | @param v - pointer to the bitmap |
| 96 | */ |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 97 | #define clib_bitmap_bytes(v) vec_bytes(v) |
| 98 | |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 99 | /** Clear a bitmap |
| 100 | @param v - pointer to the bitmap to clear |
| 101 | */ |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 102 | #define clib_bitmap_zero(v) vec_zero(v) |
| 103 | |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 104 | /** Allocate a bitmap with the supplied number of bits |
| 105 | @param [out] v - the resulting bitmap |
| 106 | @param n_bits - the required number of bits |
| 107 | */ |
| 108 | |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 109 | #define clib_bitmap_alloc(v,n_bits) \ |
| 110 | v = vec_new (uword, ((n_bits) + BITS (uword) - 1) / BITS (uword)) |
| 111 | |
| 112 | #define clib_bitmap_vec_validate(v,i) vec_validate_aligned((v),(i),sizeof(uword)) |
| 113 | |
| 114 | /* Make sure that a bitmap is at least n_bits in size */ |
| 115 | #define clib_bitmap_validate(v,n_bits) \ |
| 116 | clib_bitmap_vec_validate ((v), ((n_bits) - 1) / BITS (uword)) |
| 117 | |
| 118 | /* low-level routine to remove trailing zeros from a bitmap */ |
| 119 | always_inline uword * |
| 120 | _clib_bitmap_remove_trailing_zeros (uword * a) |
| 121 | { |
| 122 | word i; |
| 123 | if (a) |
| 124 | { |
| 125 | for (i = _vec_len (a) - 1; i >= 0; i--) |
| 126 | if (a[i] != 0) |
| 127 | break; |
| 128 | _vec_len (a) = i + 1; |
| 129 | } |
| 130 | return a; |
| 131 | } |
| 132 | |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 133 | /** Sets the ith bit of a bitmap to new_value. |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 134 | No sanity checking. Be careful. |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 135 | @param a - pointer to the bitmap |
| 136 | @param i - the bit position to interrogate |
| 137 | @param new_value - new value for the bit |
| 138 | @returns the old value of the bit |
| 139 | */ |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 140 | always_inline uword |
| 141 | clib_bitmap_set_no_check (uword * a, uword i, uword new_value) |
| 142 | { |
| 143 | uword i0 = i / BITS (a[0]); |
| 144 | uword i1 = i % BITS (a[0]); |
| 145 | uword bit = (uword) 1 << i1; |
| 146 | uword ai, old_value; |
| 147 | |
| 148 | /* Removed ASSERT since uword * a may not be a vector. */ |
| 149 | /* ASSERT (i0 < vec_len (a)); */ |
| 150 | |
| 151 | ai = a[i0]; |
| 152 | old_value = (ai & bit) != 0; |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 153 | ai &= ~bit; |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 154 | ai |= ((uword) (new_value != 0)) << i1; |
| 155 | a[i0] = ai; |
| 156 | return old_value; |
| 157 | } |
| 158 | |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 159 | /** Sets the ith bit of a bitmap to new_value |
| 160 | Removes trailing zeros from the bitmap |
Chris Luke | d4024f5 | 2016-09-06 09:32:36 -0400 | [diff] [blame] | 161 | @param ai - pointer to the bitmap |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 162 | @param i - the bit position to interrogate |
Chris Luke | d4024f5 | 2016-09-06 09:32:36 -0400 | [diff] [blame] | 163 | @param value - new value for the bit |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 164 | @returns the old value of the bit |
| 165 | */ |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 166 | always_inline uword * |
| 167 | clib_bitmap_set (uword * ai, uword i, uword value) |
| 168 | { |
| 169 | uword i0 = i / BITS (ai[0]); |
| 170 | uword i1 = i % BITS (ai[0]); |
| 171 | uword a; |
| 172 | |
| 173 | /* Check for writing a zero to beyond end of bitmap. */ |
| 174 | if (value == 0 && i0 >= vec_len (ai)) |
| 175 | return ai; /* Implied trailing zeros. */ |
| 176 | |
| 177 | clib_bitmap_vec_validate (ai, i0); |
| 178 | |
| 179 | a = ai[i0]; |
| 180 | a &= ~((uword) 1 << i1); |
| 181 | a |= ((uword) (value != 0)) << i1; |
| 182 | ai[i0] = a; |
| 183 | |
| 184 | /* If bits have been cleared, test for zero. */ |
| 185 | if (a == 0) |
| 186 | ai = _clib_bitmap_remove_trailing_zeros (ai); |
| 187 | |
| 188 | return ai; |
| 189 | } |
| 190 | |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 191 | /** Gets the ith bit value from a bitmap |
| 192 | @param ai - pointer to the bitmap |
| 193 | @param i - the bit position to interrogate |
| 194 | @returns the indicated bit value |
| 195 | */ |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 196 | always_inline uword |
| 197 | clib_bitmap_get (uword * ai, uword i) |
| 198 | { |
| 199 | uword i0 = i / BITS (ai[0]); |
| 200 | uword i1 = i % BITS (ai[0]); |
| 201 | return i0 < vec_len (ai) && 0 != ((ai[i0] >> i1) & 1); |
| 202 | } |
| 203 | |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 204 | /** Gets the ith bit value from a bitmap |
| 205 | Does not sanity-check the bit position. Be careful. |
| 206 | @param ai - pointer to the bitmap |
| 207 | @param i - the bit position to interrogate |
| 208 | @returns the indicated bit value, or garbage if the bit position is |
| 209 | out of range. |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 210 | */ |
| 211 | always_inline uword |
| 212 | clib_bitmap_get_no_check (uword * ai, uword i) |
| 213 | { |
| 214 | uword i0 = i / BITS (ai[0]); |
| 215 | uword i1 = i % BITS (ai[0]); |
| 216 | return 0 != ((ai[i0] >> i1) & 1); |
| 217 | } |
| 218 | |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 219 | always_inline uword |
| 220 | clib_bitmap_get_multiple_no_check (uword * ai, uword i, uword n_bits) |
| 221 | { |
| 222 | uword i0 = i / BITS (ai[0]); |
| 223 | uword i1 = i % BITS (ai[0]); |
| 224 | ASSERT (i1 + n_bits <= BITS (uword)); |
| 225 | return 0 != ((ai[i0] >> i1) & pow2_mask (n_bits)); |
| 226 | } |
| 227 | |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 228 | /** Gets the ith through ith + n_bits bit values from a bitmap |
| 229 | @param bitmap - pointer to the bitmap |
| 230 | @param i - the first bit position to retrieve |
| 231 | @param n_bits - the number of bit positions to retrieve |
| 232 | @returns the indicated range of bits |
| 233 | */ |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 234 | always_inline uword |
| 235 | clib_bitmap_get_multiple (uword * bitmap, uword i, uword n_bits) |
| 236 | { |
| 237 | uword i0, i1, result; |
| 238 | uword l = vec_len (bitmap); |
| 239 | |
Dave Barach | f9c231e | 2016-08-05 10:10:18 -0400 | [diff] [blame] | 240 | ASSERT (n_bits <= BITS (result)); |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 241 | |
| 242 | i0 = i / BITS (bitmap[0]); |
| 243 | i1 = i % BITS (bitmap[0]); |
| 244 | |
| 245 | /* Check first word. */ |
| 246 | result = 0; |
| 247 | if (i0 < l) |
| 248 | { |
| 249 | result |= (bitmap[i0] >> i1); |
| 250 | if (n_bits < BITS (bitmap[0])) |
| 251 | result &= (((uword) 1 << n_bits) - 1); |
| 252 | } |
| 253 | |
| 254 | /* Check for overlap into next word. */ |
| 255 | i0++; |
| 256 | if (i1 + n_bits > BITS (bitmap[0]) && i0 < l) |
| 257 | { |
| 258 | n_bits -= BITS (bitmap[0]) - i1; |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 259 | result |= |
| 260 | (bitmap[i0] & (((uword) 1 << n_bits) - 1)) << (BITS (bitmap[0]) - i1); |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 261 | } |
| 262 | |
| 263 | return result; |
| 264 | } |
| 265 | |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 266 | /** sets the ith through ith + n_bits bits in a bitmap |
| 267 | @param bitmap - pointer to the bitmap |
| 268 | @param i - the first bit position to retrieve |
| 269 | @param value - the values to set |
| 270 | @param n_bits - the number of bit positions to set |
| 271 | @returns a pointer to the updated bitmap, which may expand and move |
| 272 | */ |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 273 | |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 274 | always_inline uword * |
| 275 | clib_bitmap_set_multiple (uword * bitmap, uword i, uword value, uword n_bits) |
| 276 | { |
| 277 | uword i0, i1, l, t, m; |
| 278 | |
Dave Barach | f9c231e | 2016-08-05 10:10:18 -0400 | [diff] [blame] | 279 | ASSERT (n_bits <= BITS (value)); |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 280 | |
| 281 | i0 = i / BITS (bitmap[0]); |
| 282 | i1 = i % BITS (bitmap[0]); |
| 283 | |
| 284 | /* Allocate bitmap. */ |
| 285 | clib_bitmap_vec_validate (bitmap, (i + n_bits) / BITS (bitmap[0])); |
| 286 | l = vec_len (bitmap); |
| 287 | |
| 288 | m = ~0; |
| 289 | if (n_bits < BITS (value)) |
| 290 | m = (((uword) 1 << n_bits) - 1); |
| 291 | value &= m; |
| 292 | |
| 293 | /* Insert into first word. */ |
| 294 | t = bitmap[i0]; |
| 295 | t &= ~(m << i1); |
| 296 | t |= value << i1; |
| 297 | bitmap[i0] = t; |
| 298 | |
| 299 | /* Insert into second word. */ |
| 300 | i0++; |
| 301 | if (i1 + n_bits > BITS (bitmap[0]) && i0 < l) |
| 302 | { |
| 303 | t = BITS (bitmap[0]) - i1; |
| 304 | value >>= t; |
| 305 | n_bits -= t; |
| 306 | t = bitmap[i0]; |
| 307 | m = ((uword) 1 << n_bits) - 1; |
| 308 | t &= ~m; |
| 309 | t |= value; |
| 310 | bitmap[i0] = t; |
| 311 | } |
| 312 | |
| 313 | return bitmap; |
| 314 | } |
| 315 | |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 316 | always_inline uword * |
Korian Edeline | 40e6bdf | 2018-08-08 11:30:30 +0200 | [diff] [blame] | 317 | clib_bitmap_set_region (uword * bitmap, uword i, uword value, uword n_bits) |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 318 | { |
| 319 | uword a0, a1, b0; |
| 320 | uword i_end, mask; |
| 321 | |
| 322 | a0 = i / BITS (bitmap[0]); |
| 323 | a1 = i % BITS (bitmap[0]); |
| 324 | |
| 325 | i_end = i + n_bits; |
| 326 | b0 = i_end / BITS (bitmap[0]); |
| 327 | |
| 328 | clib_bitmap_vec_validate (bitmap, b0); |
| 329 | |
| 330 | /* First word. */ |
| 331 | mask = n_bits < BITS (bitmap[0]) ? pow2_mask (n_bits) : ~0; |
| 332 | mask <<= a1; |
| 333 | |
| 334 | if (value) |
| 335 | bitmap[a0] |= mask; |
| 336 | else |
| 337 | bitmap[a0] &= ~mask; |
| 338 | |
| 339 | for (a0++; a0 < b0; a0++) |
| 340 | bitmap[a0] = value ? ~0 : 0; |
| 341 | |
| 342 | if (a0 == b0) |
| 343 | { |
| 344 | word n_bits_left = n_bits - (BITS (bitmap[0]) - a1); |
| 345 | mask = pow2_mask (n_bits_left); |
| 346 | if (value) |
| 347 | bitmap[a0] |= mask; |
| 348 | else |
| 349 | bitmap[a0] &= ~mask; |
| 350 | } |
| 351 | |
| 352 | return bitmap; |
| 353 | } |
| 354 | |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 355 | /** Macro to iterate across set bits in a bitmap |
| 356 | |
| 357 | @param i - the current set bit |
| 358 | @param ai - the bitmap |
| 359 | @param body - the expression to evaluate for each set bit |
| 360 | */ |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 361 | #define clib_bitmap_foreach(i,ai,body) \ |
| 362 | do { \ |
| 363 | uword __bitmap_i, __bitmap_ai, __bitmap_len, __bitmap_first_set; \ |
| 364 | __bitmap_len = vec_len ((ai)); \ |
| 365 | for (__bitmap_i = 0; __bitmap_i < __bitmap_len; __bitmap_i++) \ |
| 366 | { \ |
| 367 | __bitmap_ai = (ai)[__bitmap_i]; \ |
| 368 | while (__bitmap_ai != 0) \ |
| 369 | { \ |
| 370 | __bitmap_first_set = first_set (__bitmap_ai); \ |
| 371 | (i) = (__bitmap_i * BITS ((ai)[0]) \ |
| 372 | + min_log2 (__bitmap_first_set)); \ |
| 373 | do { body; } while (0); \ |
| 374 | __bitmap_ai ^= __bitmap_first_set; \ |
| 375 | } \ |
| 376 | } \ |
| 377 | } while (0) |
| 378 | |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 379 | |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 380 | /** Return the lowest numbered set bit in a bitmap |
| 381 | @param ai - pointer to the bitmap |
| 382 | @returns lowest numbered set bit, or ~0 if the entire bitmap is zero |
| 383 | */ |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 384 | always_inline uword |
| 385 | clib_bitmap_first_set (uword * ai) |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 386 | { |
Damjan Marion | 32ab954 | 2018-06-26 01:29:55 +0200 | [diff] [blame] | 387 | uword i = 0; |
| 388 | #if uword_bits == 64 |
Damjan Marion | 13c98a2 | 2018-06-26 16:05:43 +0200 | [diff] [blame] | 389 | #if defined(CLIB_HAVE_VEC256) |
Damjan Marion | 32ab954 | 2018-06-26 01:29:55 +0200 | [diff] [blame] | 390 | while (i + 7 < vec_len (ai)) |
| 391 | { |
| 392 | u64x4 v; |
| 393 | v = u64x4_load_unaligned (ai + i) | u64x4_load_unaligned (ai + i + 4); |
| 394 | if (!u64x4_is_all_zero (v)) |
| 395 | break; |
| 396 | i += 8; |
| 397 | } |
Damjan Marion | 13c98a2 | 2018-06-26 16:05:43 +0200 | [diff] [blame] | 398 | #elif defined(CLIB_HAVE_VEC128) && defined(CLIB_HAVE_VEC128_UNALIGNED_LOAD_STORE) |
Damjan Marion | 32ab954 | 2018-06-26 01:29:55 +0200 | [diff] [blame] | 399 | while (i + 3 < vec_len (ai)) |
| 400 | { |
| 401 | u64x2 v; |
| 402 | v = u64x2_load_unaligned (ai + i) | u64x2_load_unaligned (ai + i + 2); |
| 403 | if (!u64x2_is_all_zero (v)) |
| 404 | break; |
| 405 | i += 4; |
| 406 | } |
| 407 | #endif |
| 408 | #endif |
| 409 | for (; i < vec_len (ai); i++) |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 410 | { |
| 411 | uword x = ai[i]; |
| 412 | if (x != 0) |
| 413 | return i * BITS (ai[0]) + log2_first_set (x); |
| 414 | } |
| 415 | return ~0; |
| 416 | } |
| 417 | |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 418 | /** Return the higest numbered set bit in a bitmap |
| 419 | @param ai - pointer to the bitmap |
| 420 | @returns lowest numbered set bit, or ~0 if the entire bitmap is zero |
| 421 | */ |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 422 | always_inline uword |
| 423 | clib_bitmap_last_set (uword * ai) |
Damjan Marion | 0247b46 | 2016-06-08 01:37:11 +0200 | [diff] [blame] | 424 | { |
| 425 | uword i; |
| 426 | |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 427 | for (i = vec_len (ai); i > 0; i--) |
Damjan Marion | 0247b46 | 2016-06-08 01:37:11 +0200 | [diff] [blame] | 428 | { |
Damjan Marion | ec17510 | 2016-06-30 01:02:17 +0200 | [diff] [blame] | 429 | uword x = ai[i - 1]; |
Damjan Marion | 0247b46 | 2016-06-08 01:37:11 +0200 | [diff] [blame] | 430 | if (x != 0) |
| 431 | { |
| 432 | uword first_bit; |
Damjan Marion | 1105600 | 2018-05-10 13:40:44 +0200 | [diff] [blame] | 433 | first_bit = count_leading_zeros (x); |
Damjan Marion | ec17510 | 2016-06-30 01:02:17 +0200 | [diff] [blame] | 434 | return (i) * BITS (ai[0]) - first_bit - 1; |
Damjan Marion | 0247b46 | 2016-06-08 01:37:11 +0200 | [diff] [blame] | 435 | } |
| 436 | } |
| 437 | return ~0; |
| 438 | } |
| 439 | |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 440 | /** Return the lowest numbered clear bit in a bitmap |
| 441 | @param ai - pointer to the bitmap |
| 442 | @returns lowest numbered clear bit |
| 443 | */ |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 444 | always_inline uword |
| 445 | clib_bitmap_first_clear (uword * ai) |
| 446 | { |
| 447 | uword i; |
| 448 | for (i = 0; i < vec_len (ai); i++) |
| 449 | { |
| 450 | uword x = ~ai[i]; |
| 451 | if (x != 0) |
| 452 | return i * BITS (ai[0]) + log2_first_set (x); |
| 453 | } |
| 454 | return i * BITS (ai[0]); |
| 455 | } |
| 456 | |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 457 | /** Return the number of set bits in a bitmap |
| 458 | @param ai - pointer to the bitmap |
| 459 | @returns the number of set bits in the bitmap |
| 460 | */ |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 461 | always_inline uword |
| 462 | clib_bitmap_count_set_bits (uword * ai) |
| 463 | { |
| 464 | uword i; |
| 465 | uword n_set = 0; |
| 466 | for (i = 0; i < vec_len (ai); i++) |
| 467 | n_set += count_set_bits (ai[i]); |
| 468 | return n_set; |
| 469 | } |
| 470 | |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 471 | /** Logical operator across two bitmaps |
| 472 | |
| 473 | @param ai - pointer to the destination bitmap |
| 474 | @param bi - pointer to the source bitmap |
| 475 | @returns ai = ai and bi. ai is modified, bi is not modified |
| 476 | */ |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 477 | always_inline uword *clib_bitmap_and (uword * ai, uword * bi); |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 478 | |
| 479 | /** Logical operator across two bitmaps |
| 480 | |
| 481 | @param ai - pointer to the destination bitmap |
| 482 | @param bi - pointer to the source bitmap |
| 483 | @returns ai = ai & ~bi. ai is modified, bi is not modified |
| 484 | */ |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 485 | always_inline uword *clib_bitmap_andnot (uword * ai, uword * bi); |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 486 | |
| 487 | /** Logical operator across two bitmaps |
| 488 | |
| 489 | @param ai - pointer to the destination bitmap |
| 490 | @param bi - pointer to the source bitmap |
| 491 | @returns ai = ai & ~bi. ai is modified, bi is not modified |
| 492 | */ |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 493 | always_inline uword *clib_bitmap_or (uword * ai, uword * bi); |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 494 | /** Logical operator across two bitmaps |
| 495 | |
| 496 | @param ai - pointer to the destination bitmap |
| 497 | @param bi - pointer to the source bitmap |
| 498 | @returns ai = ai or bi. ai is modified, bi is not modified |
| 499 | */ |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 500 | always_inline uword *clib_bitmap_or (uword * ai, uword * bi); |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 501 | |
| 502 | /** Logical operator across two bitmaps |
| 503 | |
| 504 | @param ai - pointer to the destination bitmap |
| 505 | @param bi - pointer to the source bitmap |
| 506 | @returns ai = ai xor bi. ai is modified, bi is not modified |
| 507 | */ |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 508 | always_inline uword *clib_bitmap_xor (uword * ai, uword * bi); |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 509 | |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 510 | /* ALU function definition macro for functions taking two bitmaps. */ |
| 511 | #define _(name, body, check_zero) \ |
| 512 | always_inline uword * \ |
| 513 | clib_bitmap_##name (uword * ai, uword * bi) \ |
| 514 | { \ |
| 515 | uword i, a, b, bi_len, n_trailing_zeros; \ |
| 516 | \ |
| 517 | n_trailing_zeros = 0; \ |
| 518 | bi_len = vec_len (bi); \ |
| 519 | if (bi_len > 0) \ |
| 520 | clib_bitmap_vec_validate (ai, bi_len - 1); \ |
| 521 | for (i = 0; i < vec_len (ai); i++) \ |
| 522 | { \ |
| 523 | a = ai[i]; \ |
| 524 | b = i < bi_len ? bi[i] : 0; \ |
| 525 | do { body; } while (0); \ |
| 526 | ai[i] = a; \ |
| 527 | if (check_zero) \ |
| 528 | n_trailing_zeros = a ? 0 : (n_trailing_zeros + 1); \ |
| 529 | } \ |
| 530 | if (check_zero) \ |
| 531 | _vec_len (ai) -= n_trailing_zeros; \ |
| 532 | return ai; \ |
| 533 | } |
| 534 | |
| 535 | /* ALU functions: */ |
Florin Coras | fcd2368 | 2018-06-29 03:22:44 -0700 | [diff] [blame] | 536 | /* *INDENT-OFF* */ |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 537 | _(and, a = a & b, 1) |
Florin Coras | fcd2368 | 2018-06-29 03:22:44 -0700 | [diff] [blame] | 538 | _(andnot, a = a & ~b, 1) |
| 539 | _(or, a = a | b, 0) |
| 540 | _(xor, a = a ^ b, 1) |
| 541 | /* *INDENT-ON* */ |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 542 | #undef _ |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 543 | /** Logical operator across two bitmaps which duplicates the first bitmap |
| 544 | |
| 545 | @param ai - pointer to the destination bitmap |
| 546 | @param bi - pointer to the source bitmap |
| 547 | @returns aiDup = ai and bi. Neither ai nor bi are modified |
| 548 | */ |
Florin Coras | fcd2368 | 2018-06-29 03:22:44 -0700 | [diff] [blame] | 549 | always_inline uword *clib_bitmap_dup_and (uword * ai, uword * bi); |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 550 | |
| 551 | /** Logical operator across two bitmaps which duplicates the first bitmap |
| 552 | |
| 553 | @param ai - pointer to the destination bitmap |
| 554 | @param bi - pointer to the source bitmap |
| 555 | @returns aiDup = ai & ~bi. Neither ai nor bi are modified |
| 556 | */ |
Florin Coras | fcd2368 | 2018-06-29 03:22:44 -0700 | [diff] [blame] | 557 | always_inline uword *clib_bitmap_dup_andnot (uword * ai, uword * bi); |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 558 | |
| 559 | /** Logical operator across two bitmaps which duplicates the first bitmap |
| 560 | |
| 561 | @param ai - pointer to the destination bitmap |
| 562 | @param bi - pointer to the source bitmap |
| 563 | @returns aiDup = ai or bi. Neither ai nor bi are modified |
| 564 | */ |
Florin Coras | fcd2368 | 2018-06-29 03:22:44 -0700 | [diff] [blame] | 565 | always_inline uword *clib_bitmap_dup_or (uword * ai, uword * bi); |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 566 | |
| 567 | /** Logical operator across two bitmaps which duplicates the first bitmap |
| 568 | |
| 569 | @param ai - pointer to the destination bitmap |
| 570 | @param bi - pointer to the source bitmap |
| 571 | @returns aiDup = ai xor bi. Neither ai nor bi are modified |
| 572 | */ |
Florin Coras | fcd2368 | 2018-06-29 03:22:44 -0700 | [diff] [blame] | 573 | always_inline uword *clib_bitmap_dup_xor (uword * ai, uword * bi); |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 574 | |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 575 | #define _(name) \ |
| 576 | always_inline uword * \ |
| 577 | clib_bitmap_dup_##name (uword * ai, uword * bi) \ |
| 578 | { return clib_bitmap_##name (clib_bitmap_dup (ai), bi); } |
| 579 | |
Florin Coras | fcd2368 | 2018-06-29 03:22:44 -0700 | [diff] [blame] | 580 | /* *INDENT-OFF* */ |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 581 | _(and); |
| 582 | _(andnot); |
| 583 | _(or); |
| 584 | _(xor); |
Florin Coras | fcd2368 | 2018-06-29 03:22:44 -0700 | [diff] [blame] | 585 | /* *INDENT-ON* */ |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 586 | #undef _ |
| 587 | |
Florin Coras | fcd2368 | 2018-06-29 03:22:44 -0700 | [diff] [blame] | 588 | /* ALU function definition macro for functions taking one bitmap and an |
| 589 | * immediate. */ |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 590 | #define _(name, body, check_zero) \ |
| 591 | always_inline uword * \ |
| 592 | clib_bitmap_##name (uword * ai, uword i) \ |
| 593 | { \ |
| 594 | uword i0 = i / BITS (ai[0]); \ |
| 595 | uword i1 = i % BITS (ai[0]); \ |
| 596 | uword a, b; \ |
| 597 | clib_bitmap_vec_validate (ai, i0); \ |
| 598 | a = ai[i0]; \ |
| 599 | b = (uword) 1 << i1; \ |
| 600 | do { body; } while (0); \ |
| 601 | ai[i0] = a; \ |
| 602 | if (check_zero && a == 0) \ |
| 603 | ai = _clib_bitmap_remove_trailing_zeros (ai); \ |
| 604 | return ai; \ |
| 605 | } |
| 606 | |
| 607 | /* ALU functions immediate: */ |
Florin Coras | fcd2368 | 2018-06-29 03:22:44 -0700 | [diff] [blame] | 608 | /* *INDENT-OFF* */ |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 609 | _(andi, a = a & b, 1) |
Florin Coras | fcd2368 | 2018-06-29 03:22:44 -0700 | [diff] [blame] | 610 | _(andnoti, a = a & ~b, 1) |
| 611 | _(ori, a = a | b, 0) |
| 612 | _(xori, a = a ^ b, 1) |
| 613 | /* *INDENT-ON* */ |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 614 | #undef _ |
Florin Coras | fcd2368 | 2018-06-29 03:22:44 -0700 | [diff] [blame] | 615 | |
| 616 | /* ALU function definition macro for functions taking one bitmap and an |
| 617 | * immediate. No tail trimming */ |
| 618 | #define _(name, body) \ |
| 619 | always_inline uword * \ |
| 620 | clib_bitmap_##name##_notrim (uword * ai, uword i) \ |
| 621 | { \ |
| 622 | uword i0 = i / BITS (ai[0]); \ |
| 623 | uword i1 = i % BITS (ai[0]); \ |
| 624 | uword a, b; \ |
| 625 | clib_bitmap_vec_validate (ai, i0); \ |
| 626 | a = ai[i0]; \ |
| 627 | b = (uword) 1 << i1; \ |
| 628 | do { body; } while (0); \ |
| 629 | ai[i0] = a; \ |
| 630 | return ai; \ |
| 631 | } |
| 632 | |
| 633 | /* ALU functions immediate: */ |
| 634 | /* *INDENT-OFF* */ |
| 635 | _(andi, a = a & b) |
| 636 | _(andnoti, a = a & ~b) |
| 637 | _(ori, a = a | b) |
| 638 | _(xori, a = a ^ b) |
| 639 | #undef _ |
| 640 | /* *INDENT-ON* */ |
| 641 | |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 642 | /** Return a random bitmap of the requested length |
| 643 | @param ai - pointer to the destination bitmap |
| 644 | @param n_bits - number of bits to allocate |
Chris Luke | d4024f5 | 2016-09-06 09:32:36 -0400 | [diff] [blame] | 645 | @param [in,out] seed - pointer to the random number seed |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 646 | @returns a reasonably random bitmap based. See random.h. |
| 647 | */ |
Florin Coras | fcd2368 | 2018-06-29 03:22:44 -0700 | [diff] [blame] | 648 | always_inline uword * |
| 649 | clib_bitmap_random (uword * ai, uword n_bits, u32 * seed) |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 650 | { |
| 651 | vec_reset_length (ai); |
| 652 | |
| 653 | if (n_bits > 0) |
| 654 | { |
| 655 | uword i = n_bits - 1; |
| 656 | uword i0, i1; |
| 657 | uword log2_rand_max; |
| 658 | |
| 659 | log2_rand_max = min_log2 (random_u32_max ()); |
| 660 | |
| 661 | i0 = i / BITS (ai[0]); |
| 662 | i1 = i % BITS (ai[0]); |
| 663 | |
| 664 | clib_bitmap_vec_validate (ai, i0); |
| 665 | for (i = 0; i <= i0; i++) |
| 666 | { |
| 667 | uword n; |
| 668 | for (n = 0; n < BITS (ai[i]); n += log2_rand_max) |
| 669 | ai[i] |= random_u32 (seed) << n; |
| 670 | } |
| 671 | if (i1 + 1 < BITS (ai[0])) |
| 672 | ai[i0] &= (((uword) 1 << (i1 + 1)) - 1); |
| 673 | } |
| 674 | return ai; |
| 675 | } |
| 676 | |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 677 | /** Return the next set bit in a bitmap starting at bit i |
| 678 | @param ai - pointer to the bitmap |
| 679 | @param i - first bit position to test |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 680 | @returns first set bit position at or after i, |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 681 | ~0 if no further set bits are found |
| 682 | */ |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 683 | always_inline uword |
| 684 | clib_bitmap_next_set (uword * ai, uword i) |
| 685 | { |
| 686 | uword i0 = i / BITS (ai[0]); |
| 687 | uword i1 = i % BITS (ai[0]); |
| 688 | uword t; |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 689 | |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 690 | if (i0 < vec_len (ai)) |
| 691 | { |
| 692 | t = (ai[i0] >> i1) << i1; |
| 693 | if (t) |
| 694 | return log2_first_set (t) + i0 * BITS (ai[0]); |
| 695 | |
| 696 | for (i0++; i0 < vec_len (ai); i0++) |
| 697 | { |
| 698 | t = ai[i0]; |
| 699 | if (t) |
| 700 | return log2_first_set (t) + i0 * BITS (ai[0]); |
| 701 | } |
| 702 | } |
| 703 | |
| 704 | return ~0; |
| 705 | } |
| 706 | |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 707 | /** Return the next clear bit in a bitmap starting at bit i |
| 708 | @param ai - pointer to the bitmap |
| 709 | @param i - first bit position to test |
| 710 | @returns first clear bit position at or after i |
| 711 | */ |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 712 | always_inline uword |
| 713 | clib_bitmap_next_clear (uword * ai, uword i) |
| 714 | { |
| 715 | uword i0 = i / BITS (ai[0]); |
| 716 | uword i1 = i % BITS (ai[0]); |
| 717 | uword t; |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 718 | |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 719 | if (i0 < vec_len (ai)) |
| 720 | { |
| 721 | t = (~ai[i0] >> i1) << i1; |
| 722 | if (t) |
| 723 | return log2_first_set (t) + i0 * BITS (ai[0]); |
| 724 | |
| 725 | for (i0++; i0 < vec_len (ai); i0++) |
| 726 | { |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 727 | t = ~ai[i0]; |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 728 | if (t) |
| 729 | return log2_first_set (t) + i0 * BITS (ai[0]); |
| 730 | } |
John Lo | ef8db36 | 2018-07-04 16:27:59 -0400 | [diff] [blame] | 731 | |
| 732 | /* no clear bit left in bitmap, return bit just beyond bitmap */ |
Dave Barach | 37b4454 | 2020-04-27 18:38:36 -0400 | [diff] [blame] | 733 | return (i0 * BITS (ai[0])) + 1; |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 734 | } |
| 735 | return i; |
| 736 | } |
| 737 | |
Yi He | e4a9eb7 | 2018-07-17 14:18:41 +0800 | [diff] [blame] | 738 | /** unformat an any sized hexadecimal bitmask into a bitmap |
| 739 | |
| 740 | uword * bitmap; |
| 741 | rv = unformat ("%U", unformat_bitmap_mask, &bitmap); |
| 742 | |
| 743 | Standard unformat_function_t arguments |
| 744 | |
| 745 | @param input - pointer an unformat_input_t |
| 746 | @param va - varargs list comprising a single uword ** |
| 747 | @returns 1 on success, 0 on failure |
| 748 | */ |
| 749 | static inline uword |
| 750 | unformat_bitmap_mask (unformat_input_t * input, va_list * va) |
| 751 | { |
| 752 | u8 *v = 0; /* hexadecimal vector */ |
| 753 | uword **bitmap_return = va_arg (*va, uword **); |
| 754 | uword *bitmap = 0; |
| 755 | |
| 756 | if (unformat (input, "%U", unformat_hex_string, &v)) |
| 757 | { |
| 758 | int i, s = vec_len (v) - 1; /* 's' for significance or shift */ |
| 759 | |
| 760 | /* v[0] holds the most significant byte */ |
| 761 | for (i = 0; s >= 0; i++, s--) |
| 762 | bitmap = clib_bitmap_set_multiple (bitmap, |
| 763 | s * BITS (v[i]), v[i], |
| 764 | BITS (v[i])); |
| 765 | |
| 766 | vec_free (v); |
| 767 | *bitmap_return = bitmap; |
| 768 | return 1; |
| 769 | } |
| 770 | |
| 771 | return 0; |
| 772 | } |
| 773 | |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 774 | /** unformat a list of bit ranges into a bitmap (eg "0-3,5-7,11" ) |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 775 | |
| 776 | uword * bitmap; |
| 777 | rv = unformat ("%U", unformat_bitmap_list, &bitmap); |
| 778 | |
| 779 | Standard unformat_function_t arguments |
| 780 | |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 781 | @param input - pointer an unformat_input_t |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 782 | @param va - varargs list comprising a single uword ** |
| 783 | @returns 1 on success, 0 on failure |
| 784 | */ |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 785 | static inline uword |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 786 | unformat_bitmap_list (unformat_input_t * input, va_list * va) |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 787 | { |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 788 | uword **bitmap_return = va_arg (*va, uword **); |
| 789 | uword *bitmap = 0; |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 790 | |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 791 | u32 a, b; |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 792 | |
| 793 | while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT) |
| 794 | { |
| 795 | int i; |
| 796 | if (unformat (input, "%u-%u,", &a, &b)) |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 797 | ; |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 798 | else if (unformat (input, "%u,", &a)) |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 799 | b = a; |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 800 | else if (unformat (input, "%u-%u", &a, &b)) |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 801 | ; |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 802 | else if (unformat (input, "%u", &a)) |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 803 | b = a; |
Damjan Marion | 75d1f24 | 2016-01-19 14:06:32 +0100 | [diff] [blame] | 804 | else if (bitmap) |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 805 | { |
| 806 | unformat_put_input (input); |
Damjan Marion | 75d1f24 | 2016-01-19 14:06:32 +0100 | [diff] [blame] | 807 | break; |
| 808 | } |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 809 | else |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 810 | goto error; |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 811 | |
Damjan Marion | 14a44d3 | 2016-02-05 23:33:21 +0100 | [diff] [blame] | 812 | if (b < a) |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 813 | goto error; |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 814 | |
| 815 | for (i = a; i <= b; i++) |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 816 | bitmap = clib_bitmap_set (bitmap, i, 1); |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 817 | } |
| 818 | *bitmap_return = bitmap; |
| 819 | return 1; |
| 820 | error: |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 821 | clib_bitmap_free (bitmap); |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 822 | return 0; |
| 823 | } |
| 824 | |
Dave Barach | 310f7fe | 2016-08-05 14:36:27 -0400 | [diff] [blame] | 825 | /** Format a bitmap as a string of hex bytes |
| 826 | |
| 827 | uword * bitmap; |
| 828 | s = format ("%U", format_bitmap_hex, bitmap); |
| 829 | |
| 830 | Standard format_function_t arguments |
| 831 | |
| 832 | @param s - string under construction |
| 833 | @param args - varargs list comprising a single uword * |
| 834 | @returns string under construction |
| 835 | */ |
Damjan Marion | 14a44d3 | 2016-02-05 23:33:21 +0100 | [diff] [blame] | 836 | static inline u8 * |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 837 | format_bitmap_hex (u8 * s, va_list * args) |
Damjan Marion | 14a44d3 | 2016-02-05 23:33:21 +0100 | [diff] [blame] | 838 | { |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 839 | uword *bitmap = va_arg (*args, uword *); |
Damjan Marion | 14a44d3 | 2016-02-05 23:33:21 +0100 | [diff] [blame] | 840 | int i, is_trailing_zero = 1; |
| 841 | |
| 842 | if (!bitmap) |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 843 | return format (s, "0"); |
Damjan Marion | 14a44d3 | 2016-02-05 23:33:21 +0100 | [diff] [blame] | 844 | |
| 845 | i = vec_bytes (bitmap) * 2; |
| 846 | |
| 847 | while (i > 0) |
| 848 | { |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 849 | u8 x = clib_bitmap_get_multiple (bitmap, --i * 4, 4); |
Damjan Marion | 14a44d3 | 2016-02-05 23:33:21 +0100 | [diff] [blame] | 850 | |
| 851 | if (x && is_trailing_zero) |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 852 | is_trailing_zero = 0; |
Damjan Marion | 14a44d3 | 2016-02-05 23:33:21 +0100 | [diff] [blame] | 853 | |
| 854 | if (x || !is_trailing_zero) |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 855 | s = format (s, "%x", x); |
Damjan Marion | 14a44d3 | 2016-02-05 23:33:21 +0100 | [diff] [blame] | 856 | } |
| 857 | return s; |
| 858 | } |
Ed Warnicke | cb9cada | 2015-12-08 15:45:58 -0700 | [diff] [blame] | 859 | #endif /* included_clib_bitmap_h */ |
Dave Barach | c379999 | 2016-08-15 11:12:27 -0400 | [diff] [blame] | 860 | |
| 861 | /* |
| 862 | * fd.io coding-style-patch-verification: ON |
| 863 | * |
| 864 | * Local Variables: |
| 865 | * eval: (c-set-style "gnu") |
| 866 | * End: |
| 867 | */ |