DynArray can specify MemBucket in constructor

1 files changed, 4 insertions, 3 deletions

diff --git a/src/dynamic-array.cpp b/src/dynamic-array.cpp
index ed181f7..7dd15eb 100644
--- a/src/dynamic-array.cpp
+++ b/src/dynamic-array.cpp
@@ -4,10 +4,11 @@
 void DynArrayReserve(DynArrayBase *arr, int64_t count) {
 	assert(count != 0);
 	if (arr->reservedCount >= count) return;
-	char *a = Pke_New<char>(arr->elementSize * count);
+	char *a = nullptr;
+	a = Pke_New<char>(arr->elementSize * count, arr->bkt);
 	if (arr->ptr != nullptr) {
 		std::memcpy(a, arr->ptr, arr->elementSize * arr->reservedCount);
-		Pke_Delete<char>(arr->ptr, arr->elementSize * arr->reservedCount);
+		Pke_Delete<char>(arr->ptr, arr->elementSize * arr->reservedCount, arr->bkt);
 	}
 	arr->reservedCount = count;
 	arr->ptr = a;
@@ -17,7 +18,7 @@ void DynArrayDestroy(DynArrayBase *arr) {
 	if (arr->ptr == nullptr) return;
 	if (arr->ptr == CAFE_BABE(char)) return;
 	if (arr->reservedCount > 0)
-		Pke_Delete<char>(arr->ptr, arr->elementSize * arr->reservedCount);
+		Pke_Delete<char>(arr->ptr, arr->elementSize * arr->reservedCount, arr->bkt);
 	arr->ptr = CAFE_BABE(char);
 }