/*
 * Copyright (c) 2022 Jeppe Odgaard
 *
 * SPDX-License-Identifier: Apache-2.0
 */

#include <zephyr/kernel.h>
#include <zephyr/sys/sys_heap.h>

#define HEAP_SIZE	256

K_HEAP_DEFINE(my_kernel_heap, HEAP_SIZE);

static char heap_mem[HEAP_SIZE];
static struct sys_heap heap;

static void print_sys_memory_stats(struct sys_heap *);
static void print_all_heaps(void);

int main(void)
{
	void *p;

	printk("System heap sample\n\n");

	sys_heap_init(&heap, heap_mem, HEAP_SIZE);
	print_sys_memory_stats(&heap);

	p = sys_heap_alloc(&heap, 150);
	print_sys_memory_stats(&heap);

	p = sys_heap_realloc(&heap, p, 100);
	print_sys_memory_stats(&heap);

	sys_heap_free(&heap, p);
	print_sys_memory_stats(&heap);

	print_all_heaps();
	return 0;
}

static void print_sys_memory_stats(struct sys_heap *hp)
{
	struct sys_memory_stats stats;

	sys_heap_runtime_stats_get(hp, &stats);

	printk("allocated %zu, free %zu, max allocated %zu, heap size %u\n",
		stats.allocated_bytes, stats.free_bytes,
		stats.max_allocated_bytes, HEAP_SIZE);
}

static void print_all_heaps(void)
{
	struct sys_heap **ha;
	size_t i, n;

	n = sys_heap_array_get(&ha);
	printk("There are %zu heaps allocated:\n", n);

	for (i = 0; i < n; i++) {
		printk("\t%zu - address %p ", i, ha[i]);
		print_sys_memory_stats(ha[i]);
	}
}