X-Git-Url: https://codewiz.org/gitweb?a=blobdiff_plain;f=bertos%2Fstruct%2Fbitarray.h;h=3bbaa8d19cf277b96037f0da96ee4eb4fb76c33a;hb=f56cf80247f6fbe18ddd34357ee306107fffeca0;hp=60dd26112bf3aa6d0a2ce4ceb6cfe1ce75ccf8b6;hpb=adde3a44787554723d12e61e5c69fca10a514c59;p=bertos.git

diff --git a/bertos/struct/bitarray.h b/bertos/struct/bitarray.h
index 60dd2611..3bbaa8d1 100644
--- a/bertos/struct/bitarray.h
+++ b/bertos/struct/bitarray.h
@@ -48,45 +48,143 @@
 typedef struct BitArray
 {
 	size_t size;
+	size_t bitarray_len;
 	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 *bitx, int idx)
+{
+	ASSERT((size_t)idx <= bitx->bitarray_len);
 
-#define ALLOC_BITARRAY(name, size)   uint8_t name[DIV_ROUNDUP((size),8)]
+	int page = idx / 8;
+	uint8_t bit = idx % 8;
 
+	bitx->array[page] |= BV(bit);
+}
 
-INLINE void bitarray_set(BitArray *ctx, int idx)
+INLINE void bitarray_clear(BitArray *bitx, int idx)
 {
-	ASSERT((size_t)idx <= ctx->size);
+	ASSERT((size_t)idx <= bitx->bitarray_len);
+
 	int page = idx / 8;
-	int bit = idx % 8;
+	uint8_t bit = idx % 8;
 
-	ctx->array[page] |= BV(bit);
+	bitx->array[page] &= ~BV(bit);
 }
 
-INLINE void bitarray_clear(BitArray *ctx, int idx)
+INLINE void bitarray_setOffset(BitArray *bitx, int idx, int offset)
 {
-	ASSERT((size_t)idx <= ctx->size);
-	int page = idx / 8;
-	int bit = idx % 8;
+	ASSERT((size_t)idx <= bitx->bitarray_len);
+
+	for (int i = idx; i < offset + idx; i++)
+		bitarray_set(bitx, i);
+}
+
+
+INLINE void bitarray_clearOffset(BitArray *bitx, int idx, int offset)
+{
+	ASSERT((size_t)idx <= bitx->bitarray_len);
 
-	ctx->array[page] &= ~BV(bit);
+	for (int i = idx; i < offset + idx; i++)
+		bitarray_clear(bitx, i);
 }
 
-INLINE bool bitarray_check(BitArray *ctx, int idx)
+INLINE bool bitarray_check(BitArray *bitx, int idx)
 {
-	ASSERT((size_t)idx <= ctx->size);
+	ASSERT((size_t)idx <= bitx->bitarray_len);
 	int page = idx / 8;
-	int bit = idx % 8;
+	uint8_t bit = idx % 8;
+
+	return (bitx->array[page] & BV(bit));
+}
+
+/*
+ * Ugly!.. reformat it.
+ */
+INLINE bool bitarray_full(BitArray *bitx)
+{
+	int count = bitx->size;
+	for (size_t page = 0; page <= bitx->size / 8; page++)
+	{
+		if (count < 8)
+		{
+			for (size_t i = page * 8; i <= bitx->bitarray_len; i++)
+				if (!bitarray_check(bitx, i))
+					return 0;
+				count--;
+		}
+		else
+		{
+			if (!(bitx->array[page] == 0xff))
+				return 0;
+		}
+		count -= 8;
+	}
+
+	return 1;
+}
+
+/*
+ * Ugly!.. reformat it.
+ */
+INLINE bool bitarray_blockFull(BitArray *bitx, int idx, int offset)
+{
+	ASSERT((size_t)(idx + offset) <= bitx->bitarray_len);
+
+	for (int i = idx; i <= idx + offset; i++)
+		if (!bitarray_check(bitx, i))
+			return 0;
+
+	return 1;
+}
 
-	return (ctx->array[page] & BV(bit));
+/*
+ * Ugly!.. reformat it.
+ */
+INLINE bool bitarray_blockEmpty(BitArray *bitx, int idx, int offset)
+{
+	ASSERT((size_t)(idx + offset) <= bitx->bitarray_len);
+
+	for (int i = idx; i <= idx + offset; i++)
+		if (bitarray_check(bitx, i))
+			return 0;
+
+	return 1;
+}
+
+
+INLINE void bitarray_dump(BitArray *bitx)
+{
+	kprintf("bitarray size[%zu]bits on [%zu]bytes\n", bitx->bitarray_len, bitx->size);
+
+	int i = 0;
+	int j = 0;
+	int count = bitx->bitarray_len;
+
+	while (count--)
+	{
+		kprintf("%d", bitarray_check(bitx, i++));
+		if (j == 7)
+		{
+			kprintf("..%02x [%d]\n", bitx->array[(i / 8) - 1], i);
+			j = 0;
+			continue;
+		}
+		j++;
+	}
+
+	if (j != 0)
+		kprintf("..%02x [%d]\n", bitx->array[i / 8], i);
 }
 
-INLINE void init_bitarray(BitArray *ctx, uint8_t *array, size_t size)
+INLINE void init_bitarray(BitArray *bitx, size_t bitarray_len, uint8_t *array, size_t size)
 {
-	ctx->size = size;
-	ctx->array = array;
+	bitx->size = size;
+	bitx->array = array;
+	bitx->bitarray_len = bitarray_len;
 }