X-Git-Url: https://codewiz.org/gitweb?a=blobdiff_plain;f=bertos%2Fstruct%2Fbitarray.h;h=326617a34f0a1f7a8a6bfc1aff4716b42bb0a233;hb=2b74d29e47161c81ee4f055072a63a54310249c8;hp=8fbef32c0089bc9c20fc1bf8bf0dee0a3d192235;hpb=cf483f342779f154657abe668956d1c57db395ac;p=bertos.git

diff --git a/bertos/struct/bitarray.h b/bertos/struct/bitarray.h
index 8fbef32c..326617a3 100644
--- a/bertos/struct/bitarray.h
+++ b/bertos/struct/bitarray.h
@@ -45,72 +45,129 @@
 
 #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 <= len);
+
+	for (int i = idx; i < offset + idx; i++)
+		bitarray_set(i, bit_array, len);
+}
+
+
+INLINE void bitarray_clearOffset(int idx, int offset, uint8_t *bit_array, size_t len)
 {
-	ASSERT((size_t)idx <= ctx->size);
+	ASSERT((size_t)idx <= len);
+
+	for (int i = idx; i < offset + idx; i++)
+		bitarray_clear(i, bit_array, len);
+}
+
+INLINE bool bitarray_check(int idx, uint8_t *bit_array, size_t len)
+{
+	ASSERT((size_t)idx <= len);
 	int page = idx / 8;
 	uint8_t bit = idx % 8;
 
-	return (ctx->array[page] & BV(bit));
+	return (bit_array[page] & BV(bit));
+}
+
+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 init_bitarray(BitArray *ctx, uint8_t *array, size_t size)
+/*
+ * Ugly!.. reformat it.
+ */
+INLINE bool bitarray_blockFull(int idx, int offset, uint8_t *bit_array, size_t len)
 {
-	ctx->size = size * 8;
-	ctx->array = array;
+	ASSERT((size_t)(idx + offset) <= len);
+
+	for (int i = idx; i <= idx + offset; i++)
+		if (!bitarray_check(i, bit_array, len))
+			return 0;
+
+	return 1;
 }
 
-INLINE size_t bitarray_size(BitArray *ctx)
+
+/*
+ * Ugly!.. reformat it.
+ */
+INLINE bool bitarray_blockEmpty(int idx, int offset, uint8_t *bit_array, size_t len)
 {
-	return ctx->size;
+	ASSERT((size_t)(idx + offset) <= len);
+
+	for (int i = idx; i <= idx + offset; i++)
+		if (bitarray_check(i, bit_array, len))
+			return 0;
+
+	return 1;
 }
 
-INLINE void bitarray_dump(BitArray *ctx)
+
+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 [%d] %d\n", ctx->array[i / 8], len, i);
+			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);