From 7cd40ea0dd3659ee599dc842739b9e68286ea213 Mon Sep 17 00:00:00 2001
From: Jonathan Bradley <jcb@pikum.xyz>
Date: Sat, 12 Aug 2023 11:04:42 -0400
Subject: Add DynamicArray

---
 src/dynamic-array.cpp | 21 ++++++++++++++++++++
 src/dynamic-array.hpp | 55 +++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 76 insertions(+)
 create mode 100644 src/dynamic-array.cpp
 create mode 100644 src/dynamic-array.hpp

(limited to 'src')

diff --git a/src/dynamic-array.cpp b/src/dynamic-array.cpp
new file mode 100644
index 0000000..2ec35dd
--- /dev/null
+++ b/src/dynamic-array.cpp
@@ -0,0 +1,21 @@
+
+#include "dynamic-array.hpp"
+
+void inline DynArrayReserve(DynArrayBase *arr, int64_t count) {
+	if (arr->reservedSize >= count) return;
+	char *a = pke::PkeNew<char>(arr->elementSize * count);
+	if (arr->ptr != nullptr) {
+		std::memcpy(a, arr->ptr, arr->elementSize * arr->reservedSize);
+		pke::PkeDelete<char>(arr->ptr, arr->elementSize * arr->reservedSize);
+	}
+	arr->reservedSize = count;
+	arr->ptr = a;
+}
+
+void inline DynArrayDestroy(DynArrayBase *arr) {
+	if (arr->ptr == nullptr) return;
+	if (arr->ptr == reinterpret_cast<char *>(0xCAFEBABE)) return;
+	pke::PkeDelete<char>(arr->ptr, arr->elementSize * arr->reservedSize);
+	arr->ptr = reinterpret_cast<char *>(0xCAFEBABE);
+}
+
diff --git a/src/dynamic-array.hpp b/src/dynamic-array.hpp
new file mode 100644
index 0000000..29fb765
--- /dev/null
+++ b/src/dynamic-array.hpp
@@ -0,0 +1,55 @@
+#ifndef FOUR_ZED_ZED_DYNAMIC_ARRAY_HPP
+#define FOUR_ZED_ZED_DYNAMIC_ARRAY_HPP
+
+#include "memory.hpp"
+
+#include <cstdint>
+#include <cstring>
+#include <cassert>
+
+#define BAKE_DYN_ARRAY(T) template struct DynArray<T>;
+
+struct DynArrayBase {
+	char *ptr = nullptr;
+	int64_t elementSize = 0;
+	int64_t reservedSize = 0;
+};
+
+template <typename T>
+struct DynArray: DynArrayBase {
+	DynArray();
+	~DynArray();
+	T operator[](std::size_t index);
+	T *GetPtr();
+	void Reserve(int64_t count);
+	private:
+		using DynArrayBase::ptr;
+		using DynArrayBase::elementSize;
+		using DynArrayBase::reservedSize;
+};
+
+void inline DynArrayReserve(DynArrayBase *arr, int64_t count);
+void inline DynArrayDestroy(DynArrayBase *arr);
+
+template <typename T> inline DynArray<T>::DynArray() {
+	this->elementSize = sizeof(T);
+}
+
+template <typename T> inline DynArray<T>::~DynArray() {
+	DynArrayDestroy(this);
+}
+
+template <typename T> inline T DynArray<T>::operator[](std::size_t index) {
+	assert(index < this->reservedSize && "Invalid DynArray<T> index");
+	return *reinterpret_cast<T *>((this->ptr + (sizeof(T) * index)));
+}
+
+template <typename T> inline T *DynArray<T>::GetPtr() {
+	return reinterpret_cast<T *>(reinterpret_cast<void *>(this->ptr));
+}
+
+template <typename T> inline void DynArray<T>::Reserve(int64_t count) {
+	return DynArrayReserve(this, count);
+}
+
+#endif /* FOUR_ZED_ZED_DYNAMIC_ARRAY_HPP */
-- 
cgit v1.2.3