diff options
Diffstat (limited to 'lib/container/container.c')
| -rw-r--r-- | lib/container/container.c | 95 |
1 files changed, 95 insertions, 0 deletions
diff --git a/lib/container/container.c b/lib/container/container.c new file mode 100644 index 0000000..4a45d0e --- /dev/null +++ b/lib/container/container.c @@ -0,0 +1,95 @@ +#include "container.h" + +#include "../malloc/malloc.h" + +#define DEFAULT_SIZE 128 + +container_t *new_container(u64 element_size) +{ + container_t *container = malloc(sizeof(container_t)); + + container->data = malloc(element_size * DEFAULT_SIZE); + container->element_size = element_size; + container->size = 0; + container->data_size = DEFAULT_SIZE; + + return container; +} + + +void free_container(container_t *container) +{ + free(container->data); + free(container); +} + +void container_push(container_t *container, u64 value) +{ + ++container->size; + + + if (container->size > container->data_size) + container_resize(container, container->data_size + DEFAULT_SIZE); + + + container_set(container, container->size - 1, value); +} + + +void container_erase(container_t *container, u64 start, u64 end) +{ + u64 data_end = container->size; + container->size -= end - start; + + for (int i = 0; i < data_end - end; ++i) + container_set(container, start + i, container_get(container, end + i)); +} + +void container_resize(container_t *container, u64 new_size) +{ + container->data_size = new_size; + container->data = realloc(container->data, new_size * container->element_size); +} + + +u64 container_get(container_t *container, u64 index) +{ + switch (container->element_size) { + case sizeof(u8): return ((u8*)container->data)[index]; + case sizeof(u16): return ((u16*)container->data)[index]; + case sizeof(u32): return ((u32*)container->data)[index]; + case sizeof(u64): return ((u64*)container->data)[index]; + } + + return 0; +} + + +void container_set(container_t *container, u64 index, u64 data) +{ + switch (container->element_size) { + case sizeof(u8): ((u8*)container->data)[index] = (u8)data; break; + case sizeof(u16): ((u16*)container->data)[index] = (u16)data; break; + case sizeof(u32): ((u32*)container->data)[index] = (u32)data; break; + case sizeof(u64): ((u64*)container->data)[index] = (u64)data; break; + } +} + + +void container_merge(container_t *dest, container_t *src) +{ + container_append(dest, src->data, src->size); +} + + +void container_append(container_t *container, void *src, u64 size) +{ + for (int i = 0; i < size; ++i) { + switch (container->element_size) { + case sizeof(u8): container_push(container, (u64)((u8*)src)[i]); break; + case sizeof(u16): container_push(container, (u64)((u16*)src)[i]); break; + case sizeof(u32): container_push(container, (u64)((u32*)src)[i]); break; + case sizeof(u64): container_push(container, (u64)((u64*)src)[i]); break; + } + } +} |