83 lines
5.2 KiB
C
83 lines
5.2 KiB
C
#pragma once
|
|
|
|
// Functions that operate on arrays
|
|
|
|
#include <stdbool.h>
|
|
#include <gc/cord.h>
|
|
|
|
#include "datatypes.h"
|
|
#include "functions.h"
|
|
#include "types.h"
|
|
#include "util.h"
|
|
|
|
// Convert negative indices to back-indexed without branching: index0 = index + (index < 0)*(len+1)) - 1
|
|
#define Array_get(item_type, arr_expr, index_expr, filename, start, end) *({ \
|
|
const array_t arr = arr_expr; int64_t index = (int64_t)(index_expr); \
|
|
int64_t off = index + (index < 0) * (arr.length + 1) - 1; \
|
|
if (__builtin_expect(off < 0 || off >= arr.length, 0)) \
|
|
fail_source(filename, start, end, "Invalid array index: %r (array has length %ld)\n", Int$as_text(&index, no, NULL), arr.length); \
|
|
(item_type*)(arr.data + arr.stride * off);})
|
|
#define Array_lvalue(item_type, arr_expr, index_expr, typeinfo, filename, start, end) *({ \
|
|
array_t *arr = arr_expr; int64_t index = (int64_t)(index_expr); \
|
|
int64_t off = index + (index < 0) * (arr->length + 1) - 1; \
|
|
if (__builtin_expect(off < 0 || off >= arr->length, 0)) \
|
|
fail_source(filename, start, end, "Invalid array index: %r (array has length %ld)\n", Int$as_text(&index, no, NULL), arr->length); \
|
|
if (arr->data_refcount > 0) \
|
|
Array$compact(arr, typeinfo); \
|
|
(item_type*)(arr->data + arr->stride * off); })
|
|
#define Array_set(item_type, arr, index, value, typeinfo, filename, start, end) \
|
|
Array_lvalue(item_type, arr_expr, index, typeinfo, filename, start, end) = value
|
|
#define Array_get_unchecked(type, x, i) *({ const array_t arr = x; int64_t index = (int64_t)(i); \
|
|
int64_t off = index + (index < 0) * (arr.length + 1) - 1; \
|
|
(type*)(arr.data + arr.stride * off);})
|
|
#define is_atomic(x) _Generic(x, bool: true, int8_t: true, int16_t: true, int32_t: true, int64_t: true, float: true, double: true, default: false)
|
|
#define TypedArray(t, ...) ({ t items[] = {__VA_ARGS__}; \
|
|
(array_t){.length=sizeof(items)/sizeof(items[0]), \
|
|
.stride=(int64_t)&items[1] - (int64_t)&items[0], \
|
|
.data=memcpy(GC_MALLOC(sizeof(items)), items, sizeof(items)), \
|
|
.atomic=0, \
|
|
.data_refcount=1}; })
|
|
#define TypedArrayN(t, N, ...) ({ t items[N] = {__VA_ARGS__}; \
|
|
(array_t){.length=N, \
|
|
.stride=(int64_t)&items[1] - (int64_t)&items[0], \
|
|
.data=memcpy(GC_MALLOC(sizeof(items)), items, sizeof(items)), \
|
|
.atomic=0, \
|
|
.data_refcount=1}; })
|
|
#define Array(x, ...) ({ __typeof(x) items[] = {x, __VA_ARGS__}; \
|
|
(array_t){.length=sizeof(items)/sizeof(items[0]), \
|
|
.stride=(int64_t)&items[1] - (int64_t)&items[0], \
|
|
.data=memcpy(is_atomic(x) ? GC_MALLOC_ATOMIC(sizeof(items)) : GC_MALLOC(sizeof(items)), items, sizeof(items)), \
|
|
.atomic=is_atomic(x), \
|
|
.data_refcount=1}; })
|
|
#define ARRAY_INCREF(arr) (arr).data_refcount |= ((arr).data_refcount << 1) | 1
|
|
#define ARRAY_DECREF(arr) (arr).data_refcount &= 2
|
|
|
|
#define Array$insert_value(arr, item_expr, index, type) ({ __typeof(item_expr) item = item_expr; Array$insert(arr, &item, index, type); })
|
|
void Array$insert(array_t *arr, const void *item, int64_t index, const TypeInfo *type);
|
|
void Array$insert_all(array_t *arr, array_t to_insert, int64_t index, const TypeInfo *type);
|
|
void Array$remove(array_t *arr, int64_t index, int64_t count, const TypeInfo *type);
|
|
void Array$sort(array_t *arr, closure_t comparison, const TypeInfo *type);
|
|
array_t Array$sorted(array_t arr, closure_t comparison, const TypeInfo *type);
|
|
void Array$shuffle(array_t *arr, const TypeInfo *type);
|
|
void *Array$random(array_t arr);
|
|
array_t Array$sample(array_t arr, int64_t n, array_t weights, const TypeInfo *type);
|
|
void Array$clear(array_t *array);
|
|
void Array$compact(array_t *arr, const TypeInfo *type);
|
|
bool Array$contains(array_t array, void *item, const TypeInfo *type);
|
|
array_t Array$from(array_t *array, int64_t first);
|
|
array_t Array$to(array_t *array, int64_t last);
|
|
array_t Array$by(array_t *array, int64_t stride);
|
|
array_t Array$reversed(array_t array);
|
|
array_t Array$concat(array_t x, array_t y, const TypeInfo *type);
|
|
uint32_t Array$hash(const array_t *arr, const TypeInfo *type);
|
|
int32_t Array$compare(const array_t *x, const array_t *y, const TypeInfo *type);
|
|
bool Array$equal(const array_t *x, const array_t *y, const TypeInfo *type);
|
|
CORD Array$as_text(const array_t *arr, bool colorize, const TypeInfo *type);
|
|
void Array$heapify(array_t *heap, closure_t comparison, const TypeInfo *type);
|
|
void Array$heap_push(array_t *heap, const void *item, closure_t comparison, const TypeInfo *type);
|
|
#define Array$heap_push_value(heap, _value, comparison, typeinfo) ({ __typeof(_value) value = _value; Array$heap_push(heap, &value, comparison, typeinfo); })
|
|
void Array$heap_pop(array_t *heap, void *out, closure_t comparison, const TypeInfo *type);
|
|
#define Array$heap_pop_value(heap, comparison, typeinfo, type) ({ type value; Array$heap_pop(heap, &value, comparison, typeinfo); value; })
|
|
|
|
// vim: ts=4 sw=0 et cino=L2,l1,(0,W4,m1,\:0
|