add general purpose container

2 files changed, 120 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;
+		}
+	}
+}
diff --git a/lib/container/container.h b/lib/container/container.h
new file mode 100644
index 0000000..a7e6690
--- /dev/null
+++ b/lib/container/container.h
@@ -0,0 +1,25 @@
+#ifndef CONTAINER_H
+#define CONTAINER_H
+
+#include "../sys/sizes.h"
+
+typedef struct {
+	void *data;
+	u64 size;
+	u64 element_size;
+	u64 data_size;
+} container_t;
+
+
+container_t *new_container(u64 element_size);
+void    free_container(container_t *);
+void    container_push(container_t *, u64 value);
+void    container_erase(container_t *, u64 start, u64 end);
+void    container_resize(container_t *, u64 new_size);
+u64     container_get(container_t *, u64 index);
+void    container_set(container_t *, u64 index, u64 data);
+void    container_merge(container_t *, container_t *);
+void    container_append(container_t *, void *, u64 size);
+
+
+#endif