aboutsummaryrefslogtreecommitdiff
path: root/lib
diff options
context:
space:
mode:
Diffstat (limited to 'lib')
-rw-r--r--lib/container/container.c95
-rw-r--r--lib/container/container.h25
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