summaryrefslogtreecommitdiff
path: root/source/kit/dynamic_array.c
blob: 1829c45d3079fd858534399c7d9d39919e5bdff4 (plain)
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
61
62
#include "dynamic_array.h"

#include <assert.h>
#include <string.h>

void kit_da_init(kit_da_void_t *array, ptrdiff_t element_size,
                 ptrdiff_t size, kit_allocator_t alloc) {
  assert(array != NULL);
  assert(element_size > 0);
  assert(size >= 0);
  assert(alloc.allocate != NULL);
  assert(alloc.deallocate != NULL);

  memset(array, 0, sizeof(kit_da_void_t));

  if (size > 0)
    array->values = alloc.allocate(alloc.state, element_size * size);

  if (array->values != NULL) {
    array->capacity = size;
    array->size     = size;
  }

  array->alloc = alloc;
}

static ptrdiff_t eval_capacity(ptrdiff_t current_cap,
                               ptrdiff_t required_cap) {
  if (current_cap == 0)
    return required_cap;
  ptrdiff_t cap = current_cap;
  while (cap < required_cap) cap *= 2;
  return cap;
}

void kit_da_resize(kit_da_void_t *array, ptrdiff_t element_size,
                   ptrdiff_t size) {
  assert(array != NULL);
  assert(element_size > 0);
  assert(size >= 0);

  if (size <= array->capacity) {
    array->size = size;
  } else {
    ptrdiff_t capacity = eval_capacity(array->capacity, size);

    assert(array->alloc.allocate != NULL);
    assert(array->alloc.deallocate != NULL);

    void *bytes = array->alloc.allocate(array->alloc.state,
                                        element_size * capacity);
    if (bytes != NULL) {
      if (array->size > 0)
        memcpy(bytes, array->values, element_size * array->size);
      if (array->values != NULL)
        array->alloc.deallocate(array->alloc.state, array->values);
      array->capacity = capacity;
      array->size     = size;
      array->values   = bytes;
    }
  }
}