#pragma once

#include "types.h"
#include "util.h"
#include "assert.h"
#include "spinlock.h"
#include "kobject.h"
#define MAGIC_NUM 0xFEDCBA98 //!< mm的幻数
#define MEM_HEAP_NAME KOBJ_NAME_SIZE      //!< 名字长度

typedef struct mem_heap
{
    char name[MEM_HEAP_NAME];
    struct mem_heap *next;
    struct mem_heap *prev;
    umword_t size; // 可用的大小
    uint32_t used;
    uint32_t magic;
} mem_heap_t;

#define MEM_HEAP_STRUCT_SIZE ALIGN(sizeof(struct mem_heap), 4) //!< 对齐大小

typedef struct mem
{
    struct mem_heap *heap_start; //!< 开始位置
    struct mem_heap *heap_end;   //!< 结束位置
    struct mem_heap *l_heap;     //!< 空闲位置
    spinlock_t lock;
} mem_t;

void mem_init(mem_t *_this);
void mem_free(mem_t *_this, void *mem);
void *mem_alloc(mem_t *_this, uint32_t size);
int mem_heap_add(mem_t *_this, void *mem, uint32_t size);
void *mem_split(mem_t *_this, void *mem, uint32_t size);
void *mem_alloc_align(mem_t *_this, uint32_t size, uint32_t align);
void mem_free_align(mem_t *_this, void *f_mem);
size_t mem_get_free_size(mem_t *_this);
struct mem_heap *mem_get_free(mem_t *_this, struct mem_heap *next,
                              int32_t hope_size, uint32_t *ret_addr);

void mem_trace(mem_t *_this);
void mem_info(mem_t *_this, size_t *total, size_t *free);