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

diff --git a/bertos/struct/bitarray.h b/bertos/struct/bitarray.h
index 805b94459080e4685df0393db90d2d6052e1ab67..76899c50ba3843987ed48c4179f8a8dec8a7857c 100644 (file)
--- a/bertos/struct/bitarray.h
+++ b/bertos/struct/bitarray.h
@@ -45,72 +45,100 @@
 
 #include <cpu/types.h>
 
-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 81bcd8abbb80d5429e9f35211ee5cdb24ac33379..3bdc7140c9ef86eab8df7fb5c178038eb90c7fe7 100644 (file)
--- a/bertos/struct/bitarray_test.c
+++ b/bertos/struct/bitarray_test.c
@@ -43,68 +43,81 @@
 
 #include <string.h>
 
-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)