8#include "internal/bits.h"
9#include "internal/gc.h"
23#define rb_darray(T) struct { rb_darray_meta_t meta; T data[]; } *
29#define rb_darray_get(ary, idx) ((ary)->data[(idx)])
35#define rb_darray_set(ary, idx, element) ((ary)->data[(idx)] = (element))
41#define rb_darray_ref(ary, idx) (&((ary)->data[(idx)]))
47#define rb_darray_append(ptr_to_ary, element) \
48 rb_darray_append_impl(ptr_to_ary, element, rb_xrealloc_mul_add)
50#define rb_darray_append_without_gc(ptr_to_ary, element) \
51 rb_darray_append_impl(ptr_to_ary, element, rb_darray_realloc_mul_add_without_gc)
53#define rb_darray_append_impl(ptr_to_ary, element, realloc_func) do { \
54 rb_darray_ensure_space((ptr_to_ary), \
55 sizeof(**(ptr_to_ary)), \
56 sizeof((*(ptr_to_ary))->data[0]), \
58 rb_darray_set(*(ptr_to_ary), \
59 (*(ptr_to_ary))->meta.size, \
61 (*(ptr_to_ary))->meta.size++; \
66#define rb_darray_foreach(ary, idx_name, elem_ptr_var) \
67 for (size_t idx_name = 0; idx_name < rb_darray_size(ary) && ((elem_ptr_var) = rb_darray_ref(ary, idx_name)); ++idx_name)
71#define rb_darray_for(ary, idx_name) \
72 for (size_t idx_name = 0; idx_name < rb_darray_size(ary); ++idx_name)
81#define rb_darray_make(ptr_to_ary, size) \
82 rb_darray_make_impl((ptr_to_ary), size, sizeof(**(ptr_to_ary)), \
83 sizeof((*(ptr_to_ary))->data[0]), rb_xcalloc_mul_add)
85#define rb_darray_make_without_gc(ptr_to_ary, size) \
86 rb_darray_make_impl((ptr_to_ary), size, sizeof(**(ptr_to_ary)), \
87 sizeof((*(ptr_to_ary))->data[0]), rb_darray_calloc_mul_add_without_gc)
94#define rb_darray_resize_capa_without_gc(ptr_to_ary, capa) \
95 rb_darray_resize_capa_impl((ptr_to_ary), rb_darray_next_power_of_two(capa), sizeof(**(ptr_to_ary)), \
96 sizeof((*(ptr_to_ary))->data[0]), rb_darray_realloc_mul_add_without_gc)
98#define rb_darray_data_ptr(ary) ((ary)->data)
108rb_darray_clear(
void *ary)
110 rb_darray_meta_t *meta = ary;
119rb_darray_size(
const void *ary)
121 const rb_darray_meta_t *meta = ary;
122 return meta ? meta->size : 0;
128rb_darray_capa(
const void *ary)
130 const rb_darray_meta_t *meta = ary;
131 return meta ? meta->capa : 0;
136rb_darray_free(
void *ary)
138 rb_darray_meta_t *meta = ary;
139 if (meta) ruby_sized_xfree(ary, meta->capa);
143rb_darray_free_without_gc(
void *ary)
151rb_darray_calloc_mul_add_without_gc(
size_t x,
size_t y,
size_t z)
153 size_t size = (x * y) + z;
155 void *ptr = calloc(1, size);
156 if (ptr == NULL) rb_bug(
"rb_darray_calloc_mul_add_without_gc: failed");
164rb_darray_realloc_mul_add_without_gc(
const void *orig_ptr,
size_t x,
size_t y,
size_t z)
166 size_t size = (x * y) + z;
168 void *ptr = realloc((
void *)orig_ptr, size);
169 if (ptr == NULL) rb_bug(
"rb_darray_realloc_mul_add_without_gc: failed");
177rb_darray_next_power_of_two(
size_t n)
179 return (
size_t)(1 << (64 - nlz_int64(n)));
185rb_darray_resize_capa_impl(
void *ptr_to_ary,
size_t new_capa,
size_t header_size,
size_t element_size,
186 void *(*realloc_mul_add_impl)(
const void *,
size_t,
size_t,
size_t))
188 rb_darray_meta_t **ptr_to_ptr_to_meta = ptr_to_ary;
189 rb_darray_meta_t *meta = *ptr_to_ptr_to_meta;
191 rb_darray_meta_t *new_ary = realloc_mul_add_impl(meta, new_capa, element_size, header_size);
199 assert(new_ary->size <= new_capa);
201 new_ary->capa = new_capa;
205 memcpy(ptr_to_ary, &new_ary,
sizeof(new_ary));
212rb_darray_ensure_space(
void *ptr_to_ary,
size_t header_size,
size_t element_size,
213 void *(*realloc_mul_add_impl)(
const void *,
size_t,
size_t,
size_t))
215 rb_darray_meta_t **ptr_to_ptr_to_meta = ptr_to_ary;
216 rb_darray_meta_t *meta = *ptr_to_ptr_to_meta;
217 size_t current_capa = rb_darray_capa(meta);
218 if (rb_darray_size(meta) < current_capa)
return;
221 size_t new_capa = current_capa == 0 ? 1 : current_capa * 2;
223 rb_darray_resize_capa_impl(ptr_to_ary, new_capa, header_size, element_size, realloc_mul_add_impl);
227rb_darray_make_impl(
void *ptr_to_ary,
size_t array_size,
size_t header_size,
size_t element_size,
228 void *(*calloc_mul_add_impl)(
size_t,
size_t,
size_t))
230 rb_darray_meta_t **ptr_to_ptr_to_meta = ptr_to_ary;
231 if (array_size == 0) {
232 *ptr_to_ptr_to_meta = NULL;
236 rb_darray_meta_t *meta = calloc_mul_add_impl(array_size, element_size, header_size);
238 meta->size = array_size;
239 meta->capa = array_size;
243 memcpy(ptr_to_ary, &meta,
sizeof(meta));