Add arena string builder

1 files changed, 26 insertions, 12 deletions

diff --git a/lib/arena.h b/lib/arena.h
index 7535f8a..1f53355 100644
--- a/lib/arena.h
+++ b/lib/arena.h
@@ -34,18 +34,32 @@ void *arena_realloc(arena *a, void *oldptr, size_t oldsz, size_t newsz);
 #define ARENA_REGION_DEFAULT_CAPACITY (8*1024)
 #define ARENA_DA_INIT_CAP (2)
 
-#define arena_da_append(a, da, item)												\
-    do {																			\
-        if ((da)->len >= (da)->cap) {												\
-            size_t new_cap = (da)->cap == 0 ? ARENA_DA_INIT_CAP : (da)->cap*2;		\
-            (da)->data = arena_realloc(												\
-                (a), (da)->data,													\
-                (da)->cap*sizeof(*(da)->data),										\
-                new_cap*sizeof(*(da)->data));										\
-            (da)->cap = new_cap;													\
-        }																			\
-																					\
-        (da)->data[(da)->len++] = (item);											\
+#define arena_da_reserve(a, da, expected_size)								\
+    do {																	\
+        if ((expected_size) >= (da)->cap) {									\
+        size_t new_cap = (da)->cap == 0 ? ARENA_DA_INIT_CAP : (da)->cap*2;	\
+        while (new_cap < expected_size) {									\
+            new_cap *= 2;													\
+        }																	\
+        (da)->data = arena_realloc(											\
+             (a), (da)->data,												\
+             (da)->cap*sizeof(*(da)->data),									\
+             new_cap*sizeof(*(da)->data));									\
+        (da)->cap = new_cap;												\
+        }																	\
+    } while (0)
+
+#define arena_da_append(a, da, item)				\
+    do {											\
+        arena_da_reserve((a), (da), (da)->len + 1);	\
+        (da)->data[(da)->len++] = (item);			\
+    } while (0)
+
+#define arena_da_append_many(a, da, items, items_len)							\
+    do {																		\
+        arena_da_reserve((a), (da), (da)->len+len);								\
+        memcpy((da)->data+(da)->len, (items), sizeof(*(da)->data)*items_len);	\
+        (da)->len += len;														\
     } while (0)
 
 #endif //ARENA_H_