/*
 * Copyright (c) 2018 Intel Corporation
 *
 * SPDX-License-Identifier: Apache-2.0
 */

#ifndef ZEPHYR_INCLUDE_SYS_MEMPOOL_BASE_H_
#define ZEPHYR_INCLUDE_SYS_MEMPOOL_BASE_H_

#include <zephyr/types.h>
#include <stddef.h>

/*
 * Definitions and macros used by both the IRQ-safe k_mem_pool and user-mode
 * compatible sys_mem_pool implementations
 */

struct sys_mem_pool_lvl {
	union {
		uint32_t *bits_p;
		uint32_t bits[sizeof(uint32_t *)/4];
	};
	sys_dlist_t free_list;
};

#define SYS_MEM_POOL_KERNEL	BIT(0)
#define SYS_MEM_POOL_USER	BIT(1)

struct sys_mem_pool_base {
	void *buf;
	size_t max_sz;
	uint16_t n_max;
	uint8_t n_levels;
	int8_t max_inline_level;
	struct sys_mem_pool_lvl *levels;
	uint8_t flags;
};

#define _MPOOL_MINBLK sizeof(sys_dnode_t)

#define Z_MPOOL_HAVE_LVL(maxsz, minsz, l) \
	(((maxsz) >> (2*(l))) >= MAX((minsz), _MPOOL_MINBLK) ? 1 : 0)

#define Z_MPOOL_LVLS(maxsz, minsz)		\
	(Z_MPOOL_HAVE_LVL((maxsz), (minsz), 0) +	\
	Z_MPOOL_HAVE_LVL((maxsz), (minsz), 1) +	\
	Z_MPOOL_HAVE_LVL((maxsz), (minsz), 2) +	\
	Z_MPOOL_HAVE_LVL((maxsz), (minsz), 3) +	\
	Z_MPOOL_HAVE_LVL((maxsz), (minsz), 4) +	\
	Z_MPOOL_HAVE_LVL((maxsz), (minsz), 5) +	\
	Z_MPOOL_HAVE_LVL((maxsz), (minsz), 6) +	\
	Z_MPOOL_HAVE_LVL((maxsz), (minsz), 7) +	\
	Z_MPOOL_HAVE_LVL((maxsz), (minsz), 8) +	\
	Z_MPOOL_HAVE_LVL((maxsz), (minsz), 9) +	\
	Z_MPOOL_HAVE_LVL((maxsz), (minsz), 10) +	\
	Z_MPOOL_HAVE_LVL((maxsz), (minsz), 11) +	\
	Z_MPOOL_HAVE_LVL((maxsz), (minsz), 12) +	\
	Z_MPOOL_HAVE_LVL((maxsz), (minsz), 13) +	\
	Z_MPOOL_HAVE_LVL((maxsz), (minsz), 14) +	\
	Z_MPOOL_HAVE_LVL((maxsz), (minsz), 15))

/* Rounds the needed bits up to integer multiples of uint32_t */
#define Z_MPOOL_LBIT_WORDS_UNCLAMPED(n_max, l) \
	((((n_max) << (2*(l))) + 31) / 32)

/* One or two 32-bit words gets stored free unioned with the pointer,
 * otherwise the calculated unclamped value
 */
#define Z_MPOOL_LBIT_WORDS(n_max, l)					 \
	(Z_MPOOL_LBIT_WORDS_UNCLAMPED(n_max, l) <= sizeof(uint32_t *)/4 ? 0 \
	 : Z_MPOOL_LBIT_WORDS_UNCLAMPED(n_max, l))

/* How many bytes for the bitfields of a single level? */
#define Z_MPOOL_LBIT_BYTES(maxsz, minsz, l, n_max)	\
	(Z_MPOOL_HAVE_LVL((maxsz), (minsz), (l)) ?	\
	 4 * Z_MPOOL_LBIT_WORDS((n_max), l) : 0)

/* Size of the bitmap array that follows the buffer in allocated memory */
#define _MPOOL_BITS_SIZE(maxsz, minsz, n_max) \
	(Z_MPOOL_LBIT_BYTES(maxsz, minsz, 0, n_max) +	\
	Z_MPOOL_LBIT_BYTES(maxsz, minsz, 1, n_max) +	\
	Z_MPOOL_LBIT_BYTES(maxsz, minsz, 2, n_max) +	\
	Z_MPOOL_LBIT_BYTES(maxsz, minsz, 3, n_max) +	\
	Z_MPOOL_LBIT_BYTES(maxsz, minsz, 4, n_max) +	\
	Z_MPOOL_LBIT_BYTES(maxsz, minsz, 5, n_max) +	\
	Z_MPOOL_LBIT_BYTES(maxsz, minsz, 6, n_max) +	\
	Z_MPOOL_LBIT_BYTES(maxsz, minsz, 7, n_max) +	\
	Z_MPOOL_LBIT_BYTES(maxsz, minsz, 8, n_max) +	\
	Z_MPOOL_LBIT_BYTES(maxsz, minsz, 9, n_max) +	\
	Z_MPOOL_LBIT_BYTES(maxsz, minsz, 10, n_max) +	\
	Z_MPOOL_LBIT_BYTES(maxsz, minsz, 11, n_max) +	\
	Z_MPOOL_LBIT_BYTES(maxsz, minsz, 12, n_max) +	\
	Z_MPOOL_LBIT_BYTES(maxsz, minsz, 13, n_max) +	\
	Z_MPOOL_LBIT_BYTES(maxsz, minsz, 14, n_max) +	\
	Z_MPOOL_LBIT_BYTES(maxsz, minsz, 15, n_max))


void z_sys_mem_pool_base_init(struct sys_mem_pool_base *p);

int z_sys_mem_pool_block_alloc(struct sys_mem_pool_base *p, size_t size,
			      uint32_t *level_p, uint32_t *block_p, void **data_p);

void z_sys_mem_pool_block_free(struct sys_mem_pool_base *p, uint32_t level,
			      uint32_t block);

#endif /* ZEPHYR_INCLUDE_SYS_MEMPOOL_BASE_H_ */