/****************************************************************************
 *
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.  The
 * ASF licenses this file to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance with the
 * License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  See the
 * License for the specific language governing permissions and limitations
 * under the License.
 *
 ****************************************************************************/

#ifndef __PORT_MM_MEM_H
#define __PORT_MM_MEM_H

/****************************************************************************
 * Included Files
 ****************************************************************************/

#include <stdint.h>
#include <stddef.h>
#include <stdio.h>
#include <string.h>

/****************************************************************************
 * Pre-processor Definitions
 ****************************************************************************/

#define MEM_ASSERT(x)                                                     \
    {                                                                     \
        if (!(x)) {                                                       \
            printf(#x " assert failed at function %s\r\n", __FUNCTION__); \
            save_and_disable_interrupt();                                              \
            while (1)                                                     \
                ;                                                         \
        }                                                                 \
    }

#define MEM_LOG(fmt, ...)  //printf(fmt, __VA_ARGS__)

#define MEM_IS_VALID(heap) ((heap) != NULL && (heap)->mem_impl != NULL)

#define KMEM_HEAP          &g_kmemheap
#if defined(CONFIG_PSRAM) && (!defined(CONFIG_HEAP_ALWAYS_ON_KMEM))
#define PMEM_HEAP &g_pmemheap
#else
#define PMEM_HEAP &g_kmemheap
#endif

/****************************************************************************
 * Public Types
 ****************************************************************************/

struct mem_heap_s {
    void *priv;
    void *heapstart;
    size_t heapsize;
};

struct meminfo {
    int total_size;    /* This is the total size of memory allocated
                        * for use by malloc in bytes. */
    int free_node;     /* This is the number of free (not in use) chunks */
    int used_node;     /* This is the number of allocated (in use) chunks */
    int max_free_size; /* Size of the largest free (not in use) chunk */
    int used_size;     /* This is the total size of memory occupied by
                        * chunks handed out by malloc. */
    int free_size;     /* This is the total size of memory occupied
                        * by free (not in use) chunks. */
};

/****************************************************************************
 * Public Data
 ****************************************************************************/

#undef EXTERN
#if defined(__cplusplus)
#define EXTERN extern "C"
extern "C" {
#else
#define EXTERN extern
#endif

EXTERN struct mem_heap_s g_kmemheap;
EXTERN struct mem_heap_s g_pmemheap;

/****************************************************************************
 * Public Function Prototypes
 ****************************************************************************/

void kmem_init(void *heapstart, size_t heapsize);
void *kmalloc(size_t size);
void *kcalloc(size_t size, size_t len);
void kfree(void *addr);
uint32_t kfree_size(void);

void pmem_init(void *heapstart, size_t heapsize);
uint32_t pfree_size(void);

/* private api for mm*/

void port_mem_init(struct mem_heap_s *heap, void *heapstart, size_t heapsize);
void *port_malloc(struct mem_heap_s *heap, size_t nbytes);
void port_free(struct mem_heap_s *heap, void *ptr);
void *port_realloc(struct mem_heap_s *heap, void *ptr, size_t nbytes);
void *port_calloc(struct mem_heap_s *heap, size_t count, size_t size);
void *port_malloc_align(struct mem_heap_s *heap, size_t align, size_t size);
void port_mem_usage(struct mem_heap_s *heap, struct meminfo *info);

#undef EXTERN
#ifdef __cplusplus
}
#endif

#endif /* __PORT_MM_MEM_H */