#include "bitset.h"

void
bitarena_init(bitarena_t *arena, int size)
{
	arena->size = size;
	size_t memSize = sizeof(char) * BITSET_BYTES * size;
	arena->arena = malloc(memSize);
	memset(arena->arena, 0, memSize);
}

bitset_t
bitset_create()
{
	bitset_t tmp = (bitset_t) malloc(sizeof(char) * BITSET_BYTES);
	memset(tmp, 0, sizeof(char) * BITSET_BYTES);
	return tmp;
}

bitset_t
bitarena_at(bitarena_t *arena, int idx)
{
	return (arena->arena) + (idx * BITSET_BYTES);
}

bitset_t
bitset_and(bitset_t b1, bitset_t b2)
{
	bitset_t bitAnd = bitset_create();

	for (int i = 0; i < BITSET_BYTES; ++i)
	{
		bitAnd[i] = b1[i] & b2[i];
	}

	return bitAnd;
}

int
bitset_isZero(bitset_t bitset)
{
	for (int i = 0; i < BITSET_BYTES; ++i)
	{
		if (bitset[i] != 0)
			return 0;
	}
	return 1;
}

void
bitset_set(bitset_t bitset, int position)
{
	int byte = position / (sizeof(char) * 8);
	int shift = position % (sizeof(char) * 8);
	bitset[byte] = bitset[byte] | (1 << shift);
}

void
bitset_unset(bitset_t bitset, int position)
{
	int byte = position / (sizeof(char) * 8);
	int shift = position % (sizeof(char) * 8);
	bitset[byte] = bitset[byte] &  ~(1 << shift);
}

int
bitset_equals(bitset_t b1, bitset_t b2)
{
	for (int i = 0; i < BITSET_BYTES; ++i)
	{
		if (b1[i] != b2[i])
			return 0;
	}
	return 1;
}

void
bitset_print(bitset_t bitset)
{
	for (int i = 3; i >= 0; --i)
	{
		printf("%i ", bitset[i]);
	}
	printf("\n");
}