/*
 * Copyright (c) 2020 YuQing <384681@qq.com>
 *
 * This program is free software: you can use, redistribute, and/or modify
 * it under the terms of the GNU Affero General Public License, version 3
 * or later ("AGPL"), as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <https://www.gnu.org/licenses/>.
 */

#ifndef ARRAY_ALLOCATOR_H
#define ARRAY_ALLOCATOR_H

#include "fast_allocator.h"

typedef struct
{
    int alloc;
    int count;
    char elts[0];
} VoidArray;

typedef struct
{
    int alloc;
    int count;
    int64_t elts[0];
} I64Array;

typedef struct
{
    int alloc;
    int count;
    int32_t elts[0];
} I32Array;

typedef struct array_allocator_context
{
    struct fast_allocator_context allocator;
    int element_size;
    int min_count;
} ArrayAllocatorContext;

#ifdef __cplusplus
extern "C" {
#endif

    int array_allocator_init(ArrayAllocatorContext *ctx,
            const char *name_prefix, const int element_size,
            const int min_bits, const int max_bits);

    VoidArray *array_allocator_alloc(ArrayAllocatorContext *ctx,
            const int target_count);

    VoidArray *array_allocator_realloc(ArrayAllocatorContext *ctx,
            VoidArray *old_array, const int target_count);

    static inline void array_allocator_free(ArrayAllocatorContext *ctx,
            VoidArray *array)
    {
        array->count = 0;
        fast_allocator_free(&ctx->allocator, array);
    }

    int array_compare_element_int64(const int64_t *n1, const int64_t *n2);

    int array_compare_element_int32(const int32_t *n1, const int32_t *n2);

#define i64_array_allocator_init(ctx, min_bits, max_bits) \
    array_allocator_init(ctx, "i64", sizeof(int64_t), min_bits, max_bits)

#define i64_array_allocator_alloc(ctx, target_count) \
    (I64Array *)array_allocator_alloc(ctx, target_count)

#define i64_array_allocator_realloc(ctx, old_array, target_count) \
    (I64Array *)array_allocator_realloc(ctx, (VoidArray *)old_array, target_count)

#define i64_array_allocator_free(ctx, array) \
    array_allocator_free(ctx, (VoidArray *)array)


#ifdef __cplusplus
}
#endif

#endif