aboutsummaryrefslogtreecommitdiff
path: root/src/vec.hff
blob: acb5e0761b652d60c9b7761485d696ea656abc82 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
import "common.hff";
import "mem.hff";
import "util.hff";

struct Vec<T> {
   dat *T,
   len uint,
   cap uint,

   fn push(vec *Vec, x T) void {
      if vec.len >= vec.cap {
         vec.cap = (vec.len + 1) > 8 ? (vec.len + 1) * 2 : 8;
         if vec.dat == #null {
            vec.dat = xmalloc(vec.cap * sizeof T);
         } else {
            vec.dat = xrealloc(vec.dat, vec.cap * sizeof T);
         }
      }
      vec.dat[vec.len++] = x;
   }

   fn pop(vec *Vec) T {
      return vec.dat[--vec.len];
   }

   fn clear(vec *Vec) void {
      free(vec.dat);
      *vec = {};
   }

   fn compact(vec *Vec) [#]T {
      if vec.dat {
         vec.cap = vec.len;
         vec.dat = xrealloc(vec.dat, vec.cap * sizeof T);
      }
      return vec.dat[0::vec.len];
   }

   fn move(vec *Vec, alloc *Allocator) [#]T {
      let len = vec.len;
      let dat *T = alloc->alloc(len * sizeof T, alignof T);
      memcpy(dat, vec.dat, len * sizeof T);
      vec->clear();
      return dat[0::len];
   }

   fn last(vec *const Vec) T { return vec.dat[vec.len - 1]; }

   fn lastp(vec *Vec) *T { return &vec.dat[vec.len - 1]; }
}

defmacro vec_each(x, i, v, &body) [
   {
      let $v = v;
      for let i = 0u; i < $v.len; ++i {
         let x = $v.dat[i];
         { body }
      }
   }
]