From d26ef6386f8936886450c186dd22c30c6079a8a1 Mon Sep 17 00:00:00 2001 From: asterix Date: Tue, 31 Aug 2010 15:25:40 +0000 Subject: [PATCH] Refactor bitarray module. Add more test for new check funtcion. git-svn-id: https://src.develer.com/svnoss/bertos/trunk@4212 38d2e660-2303-0410-9eaa-f027e97ec537 --- bertos/struct/bitarray.h | 88 +++++++++++++++++++++++------------ bertos/struct/bitarray_test.c | 81 ++++++++++++++++++-------------- 2 files changed, 105 insertions(+), 64 deletions(-) diff --git a/bertos/struct/bitarray.h b/bertos/struct/bitarray.h index 805b9445..76899c50 100644 --- a/bertos/struct/bitarray.h +++ b/bertos/struct/bitarray.h @@ -45,72 +45,100 @@ #include -typedef struct BitArray -{ - size_t size; - uint8_t *array; - -} BitArray; - #define ALLOC_BITARRAY(name, size) uint8_t name[DIV_ROUNDUP((size), 8)] +#define BIT_ARRAY_SIZE(name) (sizeof((name)) * 8) - -INLINE void bitarray_set(BitArray *ctx, int idx) +INLINE void bitarray_set(int idx, uint8_t *bit_array, size_t len) { - ASSERT((size_t)idx <= ctx->size); + ASSERT((size_t)idx <= len); int page = idx / 8; uint8_t bit = idx % 8; - ctx->array[page] |= BV(bit); + bit_array[page] |= BV(bit); } -INLINE void bitarray_clear(BitArray *ctx, int idx) +INLINE void bitarray_clear(int idx, uint8_t *bit_array, size_t len) { - ASSERT((size_t)idx <= ctx->size); + ASSERT((size_t)idx <= len); int page = idx / 8; uint8_t bit = idx % 8; - ctx->array[page] &= ~BV(bit); + bit_array[page] &= ~BV(bit); } -INLINE bool bitarray_check(BitArray *ctx, int idx) +INLINE void bitarray_setOffset(int idx, int offset, uint8_t *bit_array, size_t len) { - ASSERT((size_t)idx <= ctx->size); - int page = idx / 8; - uint8_t bit = idx % 8; + ASSERT((size_t)idx <= len); - return (ctx->array[page] & BV(bit)); + for (int i = idx; i < offset + idx; i++) + bitarray_set(i, bit_array, len); } -INLINE void init_bitarray(BitArray *ctx, uint8_t *array, size_t size) + +INLINE void bitarray_clearOffset(int idx, int offset, uint8_t *bit_array, size_t len) { - ctx->size = size * 8; - ctx->array = array; + ASSERT((size_t)idx <= len); + + for (int i = idx; i < offset + idx; i++) + bitarray_clear(i, bit_array, len); } -INLINE size_t bitarray_size(BitArray *ctx) +INLINE bool bitarray_check(int idx, uint8_t *bit_array, size_t len) { - return ctx->size; + ASSERT((size_t)idx <= len); + int page = idx / 8; + uint8_t bit = idx % 8; + + return (bit_array[page] & BV(bit)); } -INLINE void bitarray_dump(BitArray *ctx) +INLINE bool bitarray_full(uint8_t *bit_array, size_t len) { + int count = len; + for (size_t page = 0; page <= len / 8; page++) + { + if (count < 8) + { + for (size_t i = page * 8; i <= len; i++) + if (!bitarray_check(i, bit_array, len)) + return 0; + count--; + } + else + { + if (!(bit_array[page] == 0xff)) + return 0; + } + count -= 8; + } + + return 1; +} + + +INLINE void bitarray_dump(uint8_t *bit_array, size_t len) +{ + kprintf("bitarray size[%zu]bits\n", len); + int i = 0; int j = 0; - size_t len = ctx->size; - kprintf("bitarray size[%zu]\n", ctx->size); - while (len--) + int count = len; + + while (count--) { - kprintf("%d", bitarray_check(ctx, i++)); + kprintf("%d", bitarray_check(i++, bit_array, len)); if (j == 7) { - kprintf("..%02x [%zu]\n", ctx->array[i / 8], len); + kprintf("..%02x [%d]\n", bit_array[(i / 8) - 1], i); j = 0; continue; } j++; } + + if (j != 0) + kprintf("..%02x [%d]\n", bit_array[i / 8], i); } int bitarray_testSetup(void); diff --git a/bertos/struct/bitarray_test.c b/bertos/struct/bitarray_test.c index 81bcd8ab..3bdc7140 100644 --- a/bertos/struct/bitarray_test.c +++ b/bertos/struct/bitarray_test.c @@ -43,68 +43,81 @@ #include -ALLOC_BITARRAY(test1, 31); -BitArray ctx; +#define TEST1_LEN 31 +#define TEST2_LEN 17 + +ALLOC_BITARRAY(test1, TEST1_LEN); +ALLOC_BITARRAY(test2, TEST2_LEN); + int bitarray_testSetup(void) { kdbg_init(); - init_bitarray(&ctx, test1, sizeof(test1)); return 0; } int bitarray_testRun(void) { memset(test1, 0xaa, sizeof(test1)); - bitarray_dump(&ctx); - for (size_t i = 0; i < bitarray_size(&ctx); i++) + bitarray_dump(test1, TEST1_LEN); + for (size_t i = 0; i < TEST1_LEN; i++) { - if (!((bool)(i % 2) == bitarray_check(&ctx, i))) - { - kprintf("Error!\n"); - return -1; - } + if (!((bool)(i % 2) == bitarray_check(i, test1, TEST1_LEN))) + goto error; } memset(test1, 0, sizeof(test1)); - for (size_t i = 0; i < bitarray_size(&ctx); i++) + for (size_t i = 0; i < TEST1_LEN; i++) { if ((i % 2) == 0) - bitarray_clear(&ctx, i); + bitarray_clear(i, test1, TEST1_LEN); else - bitarray_set(&ctx, i); + bitarray_set(i, test1, TEST1_LEN); } - bitarray_dump(&ctx); - for (size_t i = 0; i < bitarray_size(&ctx); i++) + bitarray_dump(test1, TEST1_LEN); + for (size_t i = 0; i < TEST1_LEN; i++) { - if (!((bool)(i % 2) == bitarray_check(&ctx, i))) - { - kprintf("Error!\n"); - return -1; - } + if (!((bool)(i % 2) == bitarray_check(i, test1, TEST1_LEN))) + goto error; } memset(test1, 0, sizeof(test1)); - bitarray_set(&ctx, 0); - bitarray_dump(&ctx); - if (!bitarray_check(&ctx, 0)) - { - kprintf("Error!\n"); - return -1; - } + bitarray_set(0, test1, TEST1_LEN); + bitarray_dump(test1, TEST1_LEN); + if (!bitarray_check(0, test1, TEST1_LEN)) + goto error; memset(test1, 0, sizeof(test1)); - bitarray_set(&ctx, bitarray_size(&ctx)); - bitarray_dump(&ctx); - if (!bitarray_check(&ctx, bitarray_size(&ctx))) - { - kprintf("Error!\n"); - return -1; - } + bitarray_set(TEST1_LEN, test1, TEST1_LEN); + bitarray_dump(test1, TEST1_LEN); + if (!bitarray_check(TEST1_LEN, test1, TEST1_LEN)) + goto error; + + kprintf("Test 2\n"); + memset(test2, 0xFF, sizeof(test2)); + bitarray_dump(test2, TEST2_LEN); + if (!bitarray_full(test2, TEST2_LEN)) + goto error; + + memset(test2, 0xFF, sizeof(test2)); + bitarray_clear(5, test2, TEST2_LEN); + bitarray_dump(test2, TEST2_LEN); + if (bitarray_full(test2, TEST2_LEN)) + goto error; + + memset(test2, 0xFF, sizeof(test2)); + bitarray_clear(13, test2, TEST2_LEN); + bitarray_dump(test2, TEST2_LEN); + if (bitarray_full(test2, TEST2_LEN)) + goto error; return 0; + +error: + kprintf("Error!\n"); + return -1; } int bitarray_testTearDown(void) -- 2.25.1