Implement Array__concat and ++=

2 files changed, 28 insertions, 0 deletions

diff --git a/builtins/array.c b/builtins/array.c
index a615a806..4f9070fe 100644
--- a/builtins/array.c
+++ b/builtins/array.c
@@ -228,6 +228,33 @@ public array_t Array__slice(array_t *array, int64_t first, int64_t stride, int64
     };
 }
 
+array_t Array__concat(array_t *x, array_t *y, const TypeInfo *type)
+{
+    int64_t item_size = get_item_size(type);
+    void *data = x->atomic ? GC_MALLOC_ATOMIC(item_size*(x->length + y->length)) : GC_MALLOC(item_size*(x->length + y->length));
+    if (x->stride == item_size) {
+        memcpy(data, x->data, item_size*x->length);
+    } else {
+        for (int64_t i = 0; i < x->length; i++)
+            memcpy(data + i*item_size, x->data + i*item_size, item_size);
+    }
+
+    if (y->stride == item_size) {
+        memcpy(data + item_size*x->length, y->data, item_size*y->length);
+    } else {
+        for (int64_t i = 0; i < x->length; i++)
+            memcpy(data + (x->length + i)*item_size, y->data + i*item_size, item_size);
+    }
+
+    return (array_t){
+        .data=data,
+        .length=x->length + y->length,
+        .stride=item_size,
+        .copy_on_write=0,
+        .atomic=x->atomic,
+    };
+}
+
 public bool Array__contains(array_t array, void *item, const TypeInfo *type)
 {
     TypeInfo *item_type = type->ArrayInfo.item;
diff --git a/builtins/array.h b/builtins/array.h
index 65b7028c..5a4b29ab 100644
--- a/builtins/array.h
+++ b/builtins/array.h
@@ -32,6 +32,7 @@ void Array__clear(array_t *array, const TypeInfo *type);
 void Array__compact(array_t *arr, const TypeInfo *type);
 bool Array__contains(array_t array, void *item, const TypeInfo *type);
 array_t Array__slice(array_t *array, int64_t first, int64_t stride, int64_t length, bool readonly, const TypeInfo *type);
+array_t Array__concat(array_t *x, array_t *y, const TypeInfo *type);
 uint32_t Array__hash(const array_t *arr, const TypeInfo *type);
 int32_t Array__compare(const array_t *x, const array_t *y, const TypeInfo *type);
 bool Array__equal(const array_t *x, const array_t *y, const TypeInfo *type);