From 20ab601ffbb5a42c1cca21c44211c58bb129076f Mon Sep 17 00:00:00 2001
From: chai <chaifix@163.com>
Date: Sun, 19 Jul 2020 17:10:15 +0800
Subject: +misc

---
 src/util/darray.c | 39 +++++++++++++++++++++++++++++++++++++++
 src/util/darray.h | 24 ++++++++++++++++++++++++
 2 files changed, 63 insertions(+)
 create mode 100644 src/util/darray.c
 create mode 100644 src/util/darray.h

(limited to 'src/util')

diff --git a/src/util/darray.c b/src/util/darray.c
new file mode 100644
index 0000000..f117ccf
--- /dev/null
+++ b/src/util/darray.c
@@ -0,0 +1,39 @@
+#include "assert.h"
+#include "darray.h"
+
+
+int darray_size(void *darray) {
+	return darray != NULL ? DARRAY_OCCUPIED(darray) : 0;
+}
+
+void darray_free(void *darray) {
+	if (darray != NULL) {
+		free(DARRAY_RAW_DATA(darray));
+	}
+}
+
+void *darray_hold(void *darray, int count, int itemsize) {
+	assert(count > 0 && itemsize > 0);
+	if (darray == NULL) {
+		int raw_size = sizeof(int) * 2 + itemsize * count;
+		int *base = (int*)malloc(raw_size);
+		base[0] = count;  /* capacity */
+		base[1] = count;  /* occupied */
+		return base + 2;
+	}
+	else if (DARRAY_OCCUPIED(darray) + count <= DARRAY_CAPACITY(darray)) {
+		DARRAY_OCCUPIED(darray) += count;
+		return darray;
+	}
+	else {
+		int needed_size = DARRAY_OCCUPIED(darray) + count;
+		int double_curr = DARRAY_CAPACITY(darray) * 2;
+		int capacity = needed_size > double_curr ? needed_size : double_curr;
+		int occupied = needed_size;
+		int raw_size = sizeof(int) * 2 + itemsize * capacity;
+		int *base = (int*)realloc(DARRAY_RAW_DATA(darray), raw_size);
+		base[0] = capacity;
+		base[1] = occupied;
+		return base + 2;
+	}
+}
diff --git a/src/util/darray.h b/src/util/darray.h
new file mode 100644
index 0000000..3347fd2
--- /dev/null
+++ b/src/util/darray.h
@@ -0,0 +1,24 @@
+#ifndef _SSR_DARRAY_H
+#define _SSR_DARRAY_H
+
+#ifndef NULL 
+#define NULL 0
+#endif
+
+#define darray_push(darray, value)                                          \
+    do {                                                                    \
+        (darray) = darray_hold((darray), 1, sizeof(*(darray)));             \
+        (darray)[darray_size(darray) - 1] = (value);                        \
+    } while (0)
+
+#define DARRAY_RAW_DATA(darray) ((int*)(darray) - 2)
+#define DARRAY_CAPACITY(darray) (DARRAY_RAW_DATA(darray)[0])
+#define DARRAY_OCCUPIED(darray) (DARRAY_RAW_DATA(darray)[1])
+
+int darray_size(void *darray);
+
+void darray_free(void *darray) ;
+
+void *darray_hold(void *darray, int count, int itemsize);
+
+#endif
\ No newline at end of file
-- 
cgit v1.1-26-g67d0