/*
 * This software is licensed under the terms of the MIT License.
 * See COPYING for further information.
 * ---
 * Copyright (c) 2011-2024, Lukas Weber <laochailan@web.de>.
 * Copyright (c) 2012-2024, Andrei Alexeyev <akari@taisei-project.org>.
 */

#pragma once
#include "taisei.h"

#include "arena.h"

/*
 * WARNING: Unlike the memory.h API, the allocator API is currently not guaranteed to return
 * zero-initialized memory.
 */

typedef struct Allocator Allocator;

struct Allocator {
	void *userdata;
	struct {
		void *(*alloc)(Allocator *alloc, size_t size);
		void *(*alloc_aligned)(Allocator *alloc, size_t size, size_t align);
		void (*free)(Allocator *alloc, void *mem);
		void (*deinit)(Allocator *alloc);
	} procs;
};

extern Allocator default_allocator;

void *allocator_alloc(Allocator *alloc, size_t size)
	attr_alloc_size(2)
	attr_malloc
	attr_returns_allocated
	attr_nonnull_all;

void *allocator_alloc_array(Allocator *alloc, size_t num_members, size_t size)
	attr_alloc_size(2, 3)
	attr_malloc
	attr_returns_allocated
	attr_nonnull_all;

void *allocator_alloc_aligned(Allocator *alloc, size_t size, size_t alignment)
	attr_alloc_size(2)
	attr_alloc_align(3)
	attr_malloc
	attr_returns_allocated
	attr_nonnull_all;

void *allocator_alloc_array_aligned(Allocator *alloc, size_t num_members, size_t size, size_t alignment)
	attr_alloc_size(2, 3)
	attr_alloc_align(4)
	attr_malloc
	attr_returns_allocated
	attr_nonnull_all;

void allocator_free(Allocator *alloc, void *mem)
	attr_nonnull(1);

void allocator_deinit(Allocator *alloc)
	attr_nonnull_all;

void allocator_init_from_arena(Allocator *alloc, MemArena *arena)
	attr_nonnull_all;

/*
 * These are similar to the macros in memory.h, except they also take an Allocator pointer as the
 * first argument.
 */

#define ALLOC_VIA(_allocator, _type, ...)\
	MACROHAX_OVERLOAD_HASARGS(ALLOC_VIA_, __VA_ARGS__)(_allocator, _type, ##__VA_ARGS__)

#define ALLOC_VIA_0(_allocator, _type) \
	(_type *)__builtin_choose_expr( \
		alignof(_type) > alignof(max_align_t), \
		allocator_alloc_aligned(_allocator, sizeof(_type), alignof(_type)), \
		allocator_alloc(_allocator, sizeof(_type)))

#define ALLOC_VIA_1(_allocator, _type, ...) ({ \
		auto _alloc_ptr = ALLOC_VIA_0(_allocator, _type); \
		*_alloc_ptr = (_type) __VA_ARGS__; \
		_alloc_ptr; \
	})

#define ALLOC_ARRAY_VIA(_allocator, _nmemb, _type) \
	(_type *)__builtin_choose_expr( \
		alignof(_type) > alignof(max_align_t), \
		allocator_alloc_array_aligned(_allocator, _nmemb, sizeof(_type), alignof(_type)), \
		allocator_alloc_array(_allocator, _nmemb, sizeof(_type)))

#define ALLOC_FLEX_VIA(_allocator, _type, _extra_size) \
	(_type *)__builtin_choose_expr( \
		alignof(_type) > alignof(max_align_t), \
		allocator_alloc_aligned(_allocator, sizeof(_type) + (_extra_size), alignof(_type)), \
		allocator_alloc(_allocator, sizeof(_type) + (_extra_size)))
