obj_pool.h   [plain text]

/*
 * Licensed under a two-clause BSD-style license.
 * See LICENSE for details.
 */

#ifndef OBJ_POOL_H_
#define OBJ_POOL_H_

#include "git-compat-util.h"

#define MAYBE_UNUSED __attribute__((__unused__))

#define obj_pool_gen(pre, obj_t, initial_capacity) \
static struct { \
	uint32_t committed; \
	uint32_t size; \
	uint32_t capacity; \
	obj_t *base; \
} pre##_pool = {0, 0, 0, NULL}; \
static MAYBE_UNUSED uint32_t pre##_alloc(uint32_t count) \
{ \
	uint32_t offset; \
	if (pre##_pool.size + count > pre##_pool.capacity) { \
		while (pre##_pool.size + count > pre##_pool.capacity) \
			if (pre##_pool.capacity) \
				pre##_pool.capacity *= 2; \
			else \
				pre##_pool.capacity = initial_capacity; \
		pre##_pool.base = realloc(pre##_pool.base, \
					pre##_pool.capacity * sizeof(obj_t)); \
	} \
	offset = pre##_pool.size; \
	pre##_pool.size += count; \
	return offset; \
} \
static MAYBE_UNUSED void pre##_free(uint32_t count) \
{ \
	pre##_pool.size -= count; \
} \
static MAYBE_UNUSED uint32_t pre##_offset(obj_t *obj) \
{ \
	return obj == NULL ? ~0 : obj - pre##_pool.base; \
} \
static MAYBE_UNUSED obj_t *pre##_pointer(uint32_t offset) \
{ \
	return offset >= pre##_pool.size ? NULL : &pre##_pool.base[offset]; \
} \
static MAYBE_UNUSED void pre##_commit(void) \
{ \
	pre##_pool.committed = pre##_pool.size; \
} \
static MAYBE_UNUSED void pre##_reset(void) \
{ \
	free(pre##_pool.base); \
	pre##_pool.base = NULL; \
	pre##_pool.size = 0; \
	pre##_pool.capacity = 0; \
	pre##_pool.committed = 0; \
}

#endif