275 lines
15 KiB
C++
275 lines
15 KiB
C++
/* ************************************************************************** */
|
|
/* */
|
|
/* ::: :::::::: */
|
|
/* array.h :+: :+: :+: */
|
|
/* +:+ +:+ +:+ */
|
|
/* By: maiboyer <maiboyer@student.42.fr> +#+ +:+ +#+ */
|
|
/* +#+#+#+#+#+ +#+ */
|
|
/* Created: 2024/08/31 12:03:22 by maiboyer #+# #+# */
|
|
/* Updated: 2024/09/02 21:15:05 by maiboyer ### ########.fr */
|
|
/* */
|
|
/* ************************************************************************** */
|
|
|
|
#ifndef TREE_SITTER_ARRAY_H_
|
|
#define TREE_SITTER_ARRAY_H_
|
|
|
|
#include "me/types.h"
|
|
#include <assert.h>
|
|
#include <stdbool.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#include "me/mem/mem.h"
|
|
|
|
#define Array(T) \
|
|
struct \
|
|
{ \
|
|
T *contents; \
|
|
t_u32 size; \
|
|
t_u32 capacity; \
|
|
}
|
|
|
|
/// Initialize an array.
|
|
#define array_init(self) ((self)->size = 0, (self)->capacity = 0, (self)->contents = NULL)
|
|
|
|
/// Create an empty array.
|
|
#define array_new() {NULL, 0, 0}
|
|
|
|
/// Get a pointer to the element at a given `index` in the array.
|
|
#define array_get(self, _index) (assert((t_u32)(_index) < (self)->size), &(self)->contents[_index])
|
|
|
|
/// Get a pointer to the first element in the array.
|
|
#define array_front(self) array_get(self, 0)
|
|
|
|
/// Get a pointer to the last element in the array.
|
|
#define array_back(self) array_get(self, (self)->size - 1)
|
|
|
|
/// Clear the array, setting its size to zero. Note that this does not free any
|
|
/// memory allocated for the array's contents.
|
|
#define array_clear(self) ((self)->size = 0)
|
|
|
|
/// Reserve `new_capacity` elements of space in the array. If `new_capacity` is
|
|
/// less than the array's current capacity, this function has no effect.
|
|
#define array_reserve(self, new_capacity) _array__reserve((Array *)(self), array_elem_size(self), new_capacity)
|
|
|
|
/// Free any memory allocated for this array. Note that this does not free any
|
|
/// memory allocated for the array's contents.
|
|
#define array_delete(self) _array__delete((Array *)(self))
|
|
|
|
/// Push a new `element` onto the end of the array.
|
|
#define array_push(self, element) (_array__grow((Array *)(self), 1, array_elem_size(self)), (self)->contents[(self)->size++] = (element))
|
|
|
|
/// Increase the array's size by `count` elements.
|
|
/// New elements are zero-initialized.
|
|
#define array_grow_by(self, count) \
|
|
do \
|
|
{ \
|
|
if ((count) == 0) \
|
|
break; \
|
|
_array__grow((Array *)(self), count, array_elem_size(self)); \
|
|
memset((self)->contents + (self)->size, 0, (count) * array_elem_size(self)); \
|
|
(self)->size += (count); \
|
|
} while (0)
|
|
|
|
/// Append all elements from one array to the end of another.
|
|
#define array_push_all(self, other) array_extend((self), (other)->size, (other)->contents)
|
|
|
|
/// Append `count` elements to the end of the array, reading their values from the
|
|
/// `contents` pointer.
|
|
#define array_extend(self, count, contents) _array__splice((Array *)(self), array_elem_size(self), (self)->size, 0, count, contents)
|
|
|
|
/// Remove `old_count` elements from the array starting at the given `index`. At
|
|
/// the same index, insert `new_count` new elements, reading their values from the
|
|
/// `new_contents` pointer.
|
|
#define array_splice(self, _index, old_count, new_count, new_contents) \
|
|
_array__splice((Array *)(self), array_elem_size(self), _index, old_count, new_count, new_contents)
|
|
|
|
/// Insert one `element` into the array at the given `index`.
|
|
#define array_insert(self, _index, element) _array__splice((Array *)(self), array_elem_size(self), _index, 0, 1, &(element))
|
|
|
|
/// Remove one element from the array at the given `index`.
|
|
#define array_erase(self, _index) _array__erase((Array *)(self), array_elem_size(self), _index)
|
|
|
|
/// Pop the last element off the array, returning the element by value.
|
|
#define array_pop(self) ((self)->contents[--(self)->size])
|
|
|
|
/// Assign the contents of one array to another, reallocating if necessary.
|
|
#define array_assign(self, other) _array__assign((Array *)(self), (const Array *)(other), array_elem_size(self))
|
|
|
|
/// Swap one array with another
|
|
#define array_swap(self, other) _array__swap((Array *)(self), (Array *)(other))
|
|
|
|
/// Get the size of the array contents
|
|
#define array_elem_size(self) (sizeof *(self)->contents)
|
|
|
|
/// Search a sorted array for a given `needle` value, using the given `compare`
|
|
/// callback to determine the order.
|
|
///
|
|
/// If an existing element is found to be equal to `needle`, then the `index`
|
|
/// out-parameter is set to the existing value's index, and the `exists`
|
|
/// out-parameter is set to true. Otherwise, `index` is set to an index where
|
|
/// `needle` should be inserted in order to preserve the sorting, and `exists`
|
|
/// is set to false.
|
|
#define array_search_sorted_with(self, compare, needle, _index, _exists) _array__search_sorted(self, 0, compare, , needle, _index, _exists)
|
|
|
|
/// Search a sorted array for a given `needle` value, using integer comparisons
|
|
/// of a given struct field (specified with a leading dot) to determine the order.
|
|
///
|
|
/// See also `array_search_sorted_with`.
|
|
#define array_search_sorted_by(self, field, needle, _index, _exists) \
|
|
_array__search_sorted(self, 0, _compare_int, field, needle, _index, _exists)
|
|
|
|
/// Insert a given `value` into a sorted array, using the given `compare`
|
|
/// callback to determine the order.
|
|
#define array_insert_sorted_with(self, compare, value) \
|
|
do \
|
|
{ \
|
|
t_u32 _index, _exists; \
|
|
array_search_sorted_with(self, compare, &(value), &_index, &_exists); \
|
|
if (!_exists) \
|
|
array_insert(self, _index, value); \
|
|
} while (0)
|
|
|
|
/// Insert a given `value` into a sorted array, using integer comparisons of
|
|
/// a given struct field (specified with a leading dot) to determine the order.
|
|
///
|
|
/// See also `array_search_sorted_by`.
|
|
#define array_insert_sorted_by(self, field, value) \
|
|
do \
|
|
{ \
|
|
t_u32 _index, _exists; \
|
|
array_search_sorted_by(self, field, (value)field, &_index, &_exists); \
|
|
if (!_exists) \
|
|
array_insert(self, _index, value); \
|
|
} while (0)
|
|
|
|
// Private
|
|
|
|
typedef Array(void) Array;
|
|
|
|
/// This is not what you're looking for, see `array_delete`.
|
|
static inline void _array__delete(Array *self)
|
|
{
|
|
if (self->contents)
|
|
{
|
|
mem_free(self->contents);
|
|
self->contents = NULL;
|
|
self->size = 0;
|
|
self->capacity = 0;
|
|
}
|
|
}
|
|
|
|
/// This is not what you're looking for, see `array_erase`.
|
|
static inline void _array__erase(Array *self, size_t element_size, t_u32 index)
|
|
{
|
|
assert(index < self->size);
|
|
char *contents = (char *)self->contents;
|
|
memmove(contents + index * element_size, contents + (index + 1) * element_size, (self->size - index - 1) * element_size);
|
|
self->size--;
|
|
}
|
|
|
|
/// This is not what you're looking for, see `array_reserve`.
|
|
static inline void _array__reserve(Array *self, size_t element_size, t_u32 new_capacity)
|
|
{
|
|
if (new_capacity > self->capacity)
|
|
{
|
|
if (self->contents)
|
|
{
|
|
self->contents = mem_realloc(self->contents, new_capacity * element_size);
|
|
}
|
|
else
|
|
{
|
|
self->contents = mem_alloc(new_capacity * element_size);
|
|
}
|
|
self->capacity = new_capacity;
|
|
}
|
|
}
|
|
|
|
/// This is not what you're looking for, see `array_assign`.
|
|
static inline void _array__assign(Array *self, const Array *other, size_t element_size)
|
|
{
|
|
_array__reserve(self, element_size, other->size);
|
|
self->size = other->size;
|
|
mem_copy(self->contents, other->contents, self->size * element_size);
|
|
}
|
|
|
|
/// This is not what you're looking for, see `array_swap`.
|
|
static inline void _array__swap(Array *self, Array *other)
|
|
{
|
|
Array swap = *other;
|
|
*other = *self;
|
|
*self = swap;
|
|
}
|
|
|
|
/// This is not what you're looking for, see `array_push` or `array_grow_by`.
|
|
static inline void _array__grow(Array *self, t_u32 count, size_t element_size)
|
|
{
|
|
t_u32 new_size = self->size + count;
|
|
if (new_size > self->capacity)
|
|
{
|
|
t_u32 new_capacity = self->capacity * 2;
|
|
if (new_capacity < 8)
|
|
new_capacity = 8;
|
|
if (new_capacity < new_size)
|
|
new_capacity = new_size;
|
|
_array__reserve(self, element_size, new_capacity);
|
|
}
|
|
}
|
|
|
|
/// This is not what you're looking for, see `array_splice`.
|
|
static inline void _array__splice(Array *self, size_t element_size, t_u32 index, t_u32 old_count, t_u32 new_count, const void *elements)
|
|
{
|
|
char *contents;
|
|
t_u32 new_size;
|
|
t_u32 old_end;
|
|
t_u32 new_end;
|
|
|
|
new_size = self->size + new_count - old_count;
|
|
old_end = index + old_count;
|
|
new_end = index + new_count;
|
|
_array__reserve(self, element_size, new_size);
|
|
contents = (char *)self->contents;
|
|
if (self->size > old_end)
|
|
mem_move(contents + new_end * element_size, contents + old_end * element_size, (self->size - old_end) * element_size);
|
|
if (new_count > 0)
|
|
{
|
|
if (elements)
|
|
mem_copy((contents + index * element_size), elements, new_count * element_size);
|
|
else
|
|
mem_set_zero((contents + index * element_size), new_count * element_size);
|
|
}
|
|
self->size += new_count - old_count;
|
|
}
|
|
|
|
/// A binary search routine, based on Rust's `std::slice::binary_search_by`.
|
|
/// This is not what you're looking for, see `array_search_sorted_with` or `array_search_sorted_by`.
|
|
#define _array__search_sorted(self, start, compare, suffix, needle, _index, _exists) \
|
|
do \
|
|
{ \
|
|
*(_index) = start; \
|
|
*(_exists) = false; \
|
|
t_u32 size = (self)->size - *(_index); \
|
|
if (size == 0) \
|
|
break; \
|
|
int comparison; \
|
|
while (size > 1) \
|
|
{ \
|
|
t_u32 half_size = size / 2; \
|
|
t_u32 mid_index = *(_index) + half_size; \
|
|
comparison = compare(&((self)->contents[mid_index] suffix), (needle)); \
|
|
if (comparison <= 0) \
|
|
*(_index) = mid_index; \
|
|
size -= half_size; \
|
|
} \
|
|
comparison = compare(&((self)->contents[*(_index)] suffix), (needle)); \
|
|
if (comparison == 0) \
|
|
*(_exists) = true; \
|
|
else if (comparison < 0) \
|
|
*(_index) += 1; \
|
|
} while (0)
|
|
|
|
/// Helper macro for the `_sorted_by` routines below. This takes the left (existing)
|
|
/// parameter by reference in order to work with the generic sorting function above.
|
|
#define _compare_int(a, b) ((int)*(a) - (int)(b))
|
|
|
|
#endif // TREE_SITTER_ARRAY_H_
|