Merge pull request #84 from davidscholberg/vec-append

added append function to vec

1 files changed, 21 insertions, 0 deletions

diff --git a/std/vec.zc b/std/vec.zc
index 2205cca..d310535 100644
--- a/std/vec.zc
+++ b/std/vec.zc
@@ -90,6 +90,19 @@ impl Vec<T> {
         self.data = (T*)realloc(self.data, self.cap * sizeof(T));
     }
 
+    fn grow_to_fit(self, new_len: usize) {
+        if self.cap >= new_len {
+            return;
+        }
+
+        if (self.cap == 0) { self.cap = 8; }
+        while self.cap < new_len {
+            self.cap = self.cap * 2;
+        }
+
+        self.data = (T*)realloc(self.data, self.cap * sizeof(T));
+    }
+
     fn iterator(self) -> VecIter<T> {
         return VecIter<T> {
             data: self.data,
@@ -161,6 +174,14 @@ impl Vec<T> {
         return item;
     }
 
+    fn append(self, other: Vec<T>) {
+        var new_len = self.len + other.len;
+        self.grow_to_fit(new_len);
+
+        memcpy(self.data + self.len, other.data, other.len * sizeof(T));
+        self.len = new_len;
+    }
+
     fn get(self, idx: usize) -> T {
         if (idx >= self.len) {
             !"Panic: Index out of bounds";