// Copyright 2020-2021 Beken
//
// Licensed 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.

#include <os/os.h>
#include <components/log.h>

#include "media_core.h"

#include <driver/int.h>
#include <os/mem.h>

#include <driver/media_types.h>
#include <driver/psram.h>
#include <driver/video_common_driver.h>

#include "frame_buffer.h"

#include "psram_mem_slab.h"

#include "mlist.h"

#define TAG "frame_buffer"

#include "bk_list.h"

#define LOGI(...) BK_LOGI(TAG, ##__VA_ARGS__)
#define LOGW(...) BK_LOGW(TAG, ##__VA_ARGS__)
#define LOGE(...) BK_LOGE(TAG, ##__VA_ARGS__)
#define LOGD(...) BK_LOGD(TAG, ##__VA_ARGS__)

#define FRAME_BUFFER_READ_COMPLETE   (1 << 0)

#define list_for_each_safe_edge(pos, n, head) \
	for (pos = (head)->next, n = pos->next; (pos != (head)) && (pos->next != pos); \
	     pos = n, n = pos->next)

extern uint32_t  platform_is_in_interrupt_context(void);

fb_info_t *fb_info = NULL;
fb_mem_list_t fb_mem_list[FB_INDEX_MAX] = {0};
uint8_t fb_count[FB_INDEX_MAX] = {5, 4, 8, H265_GOP_FRAME_CNT * 2};

fb_mem_list_t *frame_buffer_list_get(pixel_format_t fmt)
{
	fb_mem_list_t *ret = NULL;

	switch (fmt)
	{
		case PIXEL_FMT_JPEG:
			ret = &fb_mem_list[FB_INDEX_JPEG];
			break;
		case PIXEL_FMT_RGB565:
		case PIXEL_FMT_RGB565_LE:
		case PIXEL_FMT_YUYV:
		case PIXEL_FMT_UYVY:
		case PIXEL_FMT_YYUV:
		case PIXEL_FMT_UVYY:
		case PIXEL_FMT_VUYY:
		case PIXEL_FMT_RGB888:
		case PIXEL_FMT_BGR888:
			ret = &fb_mem_list[FB_INDEX_DISPLAY];
			break;
		case PIXEL_FMT_H264:
            ret = &fb_mem_list[FB_INDEX_H264];
            break;
		case PIXEL_FMT_H265:
			ret = &fb_mem_list[FB_INDEX_H265];
			break;
		case PIXEL_FMT_UNKNOW:
		default:
			break;
	}

	return ret;
}

fb_type_t frame_buffer_type_get(pixel_format_t fmt)
{
	fb_type_t ret = FB_INDEX_MAX;

	switch (fmt)
	{
		case PIXEL_FMT_JPEG:
			ret = FB_INDEX_JPEG;
			break;
		case PIXEL_FMT_RGB565_LE:
		case PIXEL_FMT_RGB565:
		case PIXEL_FMT_YUYV:
		case PIXEL_FMT_UYVY:
		case PIXEL_FMT_YYUV:
		case PIXEL_FMT_UVYY:
		case PIXEL_FMT_VUYY:
		case PIXEL_FMT_RGB888:
		case PIXEL_FMT_BGR888:
			ret = FB_INDEX_DISPLAY;
			break;
		case PIXEL_FMT_H264:
            ret = FB_INDEX_H264;
            break;
		case PIXEL_FMT_H265:
			ret = FB_INDEX_H265;
			break;
		case PIXEL_FMT_UNKNOW:
		default:
			break;
	}

	return ret;
}

bk_err_t frame_buffer_list_remove(frame_buffer_t *frame, LIST_HEADER_T *list)
{
	frame_buffer_node_t *tmp = NULL;
	frame_buffer_node_t *node = list_entry(frame, frame_buffer_node_t, frame);
	LIST_HEADER_T *pos, *n;
	bk_err_t ret = BK_FAIL;


	list_for_each_safe(pos, n, list)
	{
		tmp = list_entry(pos, frame_buffer_node_t, list);
		if (tmp != NULL
		    && (tmp->frame.frame == node->frame.frame))
		{
			list_del(pos);
			ret = BK_OK;
			break;
		}
	}

	return ret;
}

int frame_buffer_fb_init(fb_type_t type)
{
	fb_mem_list_t *mem_list = &fb_mem_list[type];

	if (mem_list->enable == true)
	{
		LOGE("%s already init\n", __func__);
		return BK_FAIL;
	}

	if (type == FB_INDEX_DISPLAY)
	{
		mem_list->mode = FB_MEM_ALONE;
		mem_list->count = fb_count[type];
		mem_list->free_request = false;
		mem_list->ready_request = false;

		if (rtos_init_semaphore_ex(&mem_list->free_sem, mem_list->count, 0) != BK_OK)
		{
			LOGE("%s free_sem init failed\n", __func__);
		}

		if (rtos_init_semaphore_ex(&mem_list->ready_sem, mem_list->count, 0) != BK_OK)
		{
			LOGE("%s ready_sem init failed\n", __func__);
		}
	}
	else if (type == FB_INDEX_JPEG || type == FB_INDEX_H264 || type == FB_INDEX_H265)
	{
		mem_list->mode = FB_MEM_SHARED;
		mem_list->count = fb_count[type];
	}
	else
	{
		LOGE("%s unknow type: %d\n", __func__, type);
		return BK_FAIL;
	}

	for (int i = 0; i < fb_count[type]; i ++)
	{
		frame_buffer_node_t *node = (frame_buffer_node_t *)os_malloc(sizeof(frame_buffer_node_t));

		if (node == NULL)
		{
			LOGE("%s os_malloc node failed\n", __func__);
			return BK_FAIL;
		}

		os_memset(node, 0, sizeof(frame_buffer_node_t));

		list_add_tail(&node->list, &mem_list->free);
	}

	mem_list->enable = true;

	return BK_OK;
}

int frame_buffer_fb_deinit(fb_type_t type)
{
	int ret = BK_OK;
	fb_mem_list_t *mem_list = NULL;
	frame_buffer_node_t *tmp = NULL;
	LIST_HEADER_T *pos, *n;
	uint32_t isr_context = platform_is_in_interrupt_context();

	GLOBAL_INT_DECLARATION() = 0;

	mem_list = &fb_mem_list[type];

	if (mem_list->enable == false)
	{
		LOGE("%s already deinit\n", __func__);
		return ret;
	}

	if (!isr_context)
	{
		rtos_lock_mutex(&mem_list->lock);
		GLOBAL_INT_DISABLE();
	}

	if (type == FB_INDEX_DISPLAY)
	{
		LOGI("display mem deinit\n");

		if (!isr_context)
		{
			GLOBAL_INT_RESTORE();
		}

		if (rtos_deinit_semaphore(&mem_list->free_sem) != BK_OK)
		{
			LOGE("%s free_sem init failed\n", __func__);
		}

		if (rtos_deinit_semaphore(&mem_list->ready_sem) != BK_OK)
		{
			LOGE("%s ready_sem init failed\n", __func__);
		}

		if (!isr_context)
		{
			GLOBAL_INT_DISABLE();
		}
	}
	else if (type == FB_INDEX_JPEG)
	{
		LOGI("jpeg mem deinit\n");
	}
	else if (type == FB_INDEX_H264)
	{
		LOGI("h264 mem deinit\n");
	}
	else if (type == FB_INDEX_H265)
	{
		LOGI("h265 mem deinit\n");
	}
	else
	{
		LOGE("%s unknow type: %d\n", __func__, type);
		ret = BK_FAIL;
		goto out;
	}

	if (!list_empty(&mem_list->free))
	{
		list_for_each_safe(pos, n, &mem_list->free)
		{
			tmp = list_entry(pos, frame_buffer_node_t, list);
			LOGD("free list: %p\n", tmp);
			if (tmp != NULL)
			{
				if (tmp->frame.base_addr)
				{
					bk_psram_frame_buffer_free(tmp->frame.base_addr);
					tmp->frame.base_addr = tmp->frame.frame = NULL;
				}
				list_del(pos);
				os_free(tmp);
				tmp = NULL;
			}
		}

		INIT_LIST_HEAD(&mem_list->free);
	}

	if (!list_empty(&mem_list->ready))
	{
		list_for_each_safe(pos, n, &mem_list->ready)
		{
			LOGD("ready list: %p\n", tmp);
			tmp = list_entry(pos, frame_buffer_node_t, list);
			if (tmp != NULL)
			{
				if (tmp->frame.base_addr)
				{
					bk_psram_frame_buffer_free(tmp->frame.base_addr);
					tmp->frame.base_addr = tmp->frame.frame = NULL;
				}
				list_del(pos);
				os_free(tmp);
				tmp = NULL;
			}
		}

		INIT_LIST_HEAD(&mem_list->ready);
	}

	mem_list->enable = false;

out:

	if (!isr_context)
	{
		GLOBAL_INT_RESTORE();
		rtos_unlock_mutex(&mem_list->lock);
	}

	return ret;
}

void frame_buffer_fb_clear(fb_type_t type)
{
	fb_mem_list_t *mem_list = NULL;
	frame_buffer_node_t *tmp = NULL;
	LIST_HEADER_T *pos, *n;
	uint32_t isr_context = platform_is_in_interrupt_context();

	GLOBAL_INT_DECLARATION() = 0;

	mem_list = &fb_mem_list[type];

	if (mem_list->enable == false)
	{
		LOGE("%s already deinit\n", __func__);
		return;
	}

	if (!isr_context)
	{
		rtos_lock_mutex(&mem_list->lock);
		GLOBAL_INT_DISABLE();
	}

	if (!list_empty(&mem_list->ready))
	{
		list_for_each_safe(pos, n, &mem_list->ready)
		{
			tmp = list_entry(pos, frame_buffer_node_t, list);
			if (tmp != NULL)
			{
				tmp->frame.err_state = true;
			}
		}
	}

	if (!isr_context)
	{
		GLOBAL_INT_RESTORE();
		rtos_unlock_mutex(&mem_list->lock);
	}
}

void frame_buffer_fb_free(frame_buffer_t *frame, frame_module_t index)
{
	if (frame == NULL || index >= MODULE_MAX)
	{
		LOGE("%s %d, frame is null\r\n", __func__, index);
		return;
	}

	fb_type_t type = frame_buffer_type_get(frame->fmt);
	fb_mem_list_t *mem_list = &fb_mem_list[type];
	frame_buffer_node_t *node = list_entry(frame, frame_buffer_node_t, frame);
	uint32_t isr_context = platform_is_in_interrupt_context();

	if (mem_list == NULL)
	{
		LOGE("%s invalid mem_list: %p, %d\n", __func__, mem_list, index);
		if (fb_info->modules[index].enable)
		{
			xEventGroupSetBits(fb_info->modules[index].handle, FRAME_BUFFER_READ_COMPLETE);
		}

		return;
	}

	GLOBAL_INT_DECLARATION() = 0;

	if (!isr_context)
	{
		rtos_lock_mutex(&mem_list->lock);
		GLOBAL_INT_DISABLE();
	}

	if (mem_list->enable == false)
	{
		if (!isr_context)
		{
			GLOBAL_INT_RESTORE();
		}

		if (fb_info->modules[index].enable)
		{
			LOGE("%s fb_mem_list disable: %p, %d\n", __func__, mem_list, index);
			xEventGroupSetBits(fb_info->modules[index].handle, FRAME_BUFFER_READ_COMPLETE);
		}

		if (!isr_context)
		{
			GLOBAL_INT_DISABLE();
		}
		goto out;
	}

	if (node->free_mask & INDEX_MASK(index))
	{
		LOGE("%s refree: %d\n", __func__, index);
		goto out;
	}
	else
	{
		node->free_mask |= INDEX_MASK(index);
	}

	if (node->read_mask)
	{
		if ((node->read_mask == node->free_mask)
		    && ((node->free_mask & fb_info->register_mask[type]) == fb_info->register_mask[type]))
		{
			if (BK_OK != frame_buffer_list_remove(frame, &mem_list->ready))
			{
				LOGE("%s remove failed\n", __func__);
			}

			node->free_mask = 0;
			node->read_mask = 0;

			// fix frame_buffer address, not free
			/*if (node->frame.base_addr)
			{
				bk_psram_frame_buffer_free(node->frame.base_addr);
				node->frame.base_addr = node->frame.frame = NULL;
			}*/
			list_add_tail(&node->list, &mem_list->free);
		}

		if (!isr_context)
		{
			GLOBAL_INT_RESTORE();
		}

		if (fb_info->modules[index].enable)
		{
			xEventGroupSetBits(fb_info->modules[index].handle, FRAME_BUFFER_READ_COMPLETE);
		}

		if (!isr_context)
		{
			GLOBAL_INT_DISABLE();
		}
	}
	else
	{
		/* safte check */
		if (BK_OK != frame_buffer_list_remove(frame, &mem_list->ready))
		{
			LOGD("%s remove failed\n", __func__);
		}

		if (node->free_mask != 0)
		{
			node->free_mask = 0;
			node->read_mask = 0;
			list_add_tail(&node->list, &mem_list->free);
		}
	}

out:

	if (!isr_context)
	{
		GLOBAL_INT_RESTORE();
		rtos_unlock_mutex(&mem_list->lock);
	}
}

frame_buffer_t *frame_buffer_fb_dual_malloc(fb_type_t type, uint32_t size)
{
    BK_ASSERT(type == FB_INDEX_H264 || type == FB_INDEX_H265);
	frame_buffer_node_t *node = NULL, *tmp = NULL;
	fb_mem_list_t *mem_list = &fb_mem_list[type];
	LIST_HEADER_T *pos, *n;
	uint8_t i_frame = 0;
	uint32_t isr_context = platform_is_in_interrupt_context();

	if (mem_list == NULL)
	{
		LOGE("%s invalid mem_list: %p\n", __func__, mem_list);
		return NULL;
	}

	if (mem_list->lock == NULL)
	{
		LOGE("[%s]:mem_list->lock is NULL\r\n", __func__);
		return NULL;
	}

	GLOBAL_INT_DECLARATION() = 0;

	if (!isr_context)
	{
		rtos_lock_mutex(&mem_list->lock);
		GLOBAL_INT_DISABLE();
	}

	list_for_each_safe(pos, n, &mem_list->free)
	{
		tmp = list_entry(pos, frame_buffer_node_t, list);
		if (tmp != NULL && !tmp->error)
		{
			node = tmp;
			list_del(pos);
			break;
		}
	}


	// find a free node
	if (node)
	{
		// get frame free list, need malloc psram buffer again
		if (node->frame.base_addr == NULL)
		{
			node->frame.size = size;
#if (CONFIG_PSRAM_WRITE_THROUGH)
			node->frame.size += 32;// 1024
#endif
			// malloc psram mem to frame_buffer data
			if (type == FB_INDEX_DISPLAY)
			{
				// need align 64K
#if CONFIG_SOC_BK7256XX
				node->frame.size += 64 * 1024;
				node->frame.base_addr = (uint8_t *)bk_psram_frame_buffer_malloc(PSRAM_HEAP_YUV, node->frame.size);
				if (node->frame.base_addr != NULL)
				{
					if ((uint32_t)node->frame.base_addr & 0xFFFF)
					{
						node->frame.frame = (uint8_t *)((((uint32_t)node->frame.base_addr >> 16) + 1) << 16);
					}
				}
#else
				node->frame.base_addr = (uint8_t *)bk_psram_frame_buffer_malloc(PSRAM_HEAP_YUV, node->frame.size);
				node->frame.frame = node->frame.base_addr;
#endif
			}
			else
			{
				node->frame.base_addr = (uint8_t *)bk_psram_frame_buffer_malloc(PSRAM_HEAP_ENCODE, node->frame.size);
				node->frame.frame = node->frame.base_addr;
			}

			if (node->frame.base_addr == NULL)
			{
				// add this node to free_list, and from ready node_list to find a frame_buffer(direct free)
				node->error = true;
				list_add_tail(&node->list, &mem_list->free);
				node = NULL;
			}
#if (CONFIG_PSRAM_WRITE_THROUGH)
			else
			{
				// checkout node->frame->frame is aligned 32byte(8word)
				if ((uint32_t)node->frame.base_addr & 0x1F)
				{
					node->frame.frame = (uint8_t *)((((uint32_t)node->frame.base_addr >> 5) + 1) << 5);
				}

				LOGD("%s, %d, %p, size:%d\r\n", __func__, type, node->frame.frame, size);
			}
#endif
		}
	}

	if (node == NULL)
	{
		list_for_each_safe(pos, n, &mem_list->ready)
		{
			tmp = list_entry(pos, frame_buffer_node_t, list);
			if (tmp != NULL
			    && (tmp->read_mask == tmp->free_mask))
			{
				node = tmp;
				list_del(pos);
				break;
			}
		}
	}

	if (!isr_context)
	{
		GLOBAL_INT_RESTORE();
		rtos_unlock_mutex(&mem_list->lock);
	}

	if (node == NULL)
	{
		LOGE("%s %d i_frame:%d failed\n", __func__, type, i_frame);
		return NULL;
	}

	node->frame.size = size;
	node->frame.length = 0;
	node->frame.width = 0;
	node->frame.height = 0;
	node->frame.h264_type = 0;
	node->frame.fmt = 0;
	node->read_mask = 0;
	node->free_mask = 0;
	node->frame.mix = 0;
	node->frame.err_state = false;
	node->frame.cb = NULL;

	return &node->frame;
}

frame_buffer_t *frame_buffer_fb_malloc(fb_type_t type, uint32_t size)
{
	frame_buffer_node_t *node = NULL, *tmp = NULL;
	fb_mem_list_t *mem_list = &fb_mem_list[type];
	LIST_HEADER_T *pos, *n;
	uint32_t isr_context = platform_is_in_interrupt_context();

	if (mem_list == NULL)
	{
		LOGE("%s invalid mem_list: %p\n", __func__, mem_list);
		return NULL;
	}

	if (mem_list->lock == NULL)
	{
		LOGE("[%s]:mem_list->lock is NULL\r\n", __func__);
		return NULL;
	}

	GLOBAL_INT_DECLARATION() = 0;

	if (!isr_context)
	{
		rtos_lock_mutex(&mem_list->lock);
		GLOBAL_INT_DISABLE();
	}

	list_for_each_safe(pos, n, &mem_list->free)
	{
		tmp = list_entry(pos, frame_buffer_node_t, list);
		if (tmp != NULL && !tmp->error)
		{
			node = tmp;
			list_del(pos);
			break;
		}
	}

	// find a free node
	if (node)
	{
		// get frame free list, need malloc psram buffer again
		if (node->frame.base_addr == NULL)
		{
			node->frame.size = size;
#if (CONFIG_PSRAM_WRITE_THROUGH)
			node->frame.size += 32;// 1024
#endif
			// malloc psram mem to frame_buffer data
			if (type == FB_INDEX_DISPLAY)
			{
				// need align 64K
#if CONFIG_SOC_BK7256XX
				node->frame.size += 64 * 1024;
				node->frame.base_addr = (uint8_t *)bk_psram_frame_buffer_malloc(PSRAM_HEAP_YUV, node->frame.size);
				if (node->frame.base_addr != NULL)
				{
					if ((uint32_t)node->frame.base_addr & 0xFFFF)
					{
						node->frame.frame = (uint8_t *)((((uint32_t)node->frame.base_addr >> 16) + 1) << 16);
					}
				}
#else
				node->frame.base_addr = (uint8_t *)bk_psram_frame_buffer_malloc(PSRAM_HEAP_YUV, node->frame.size);
				node->frame.frame = node->frame.base_addr;
#endif
			}
			else
			{
				node->frame.base_addr = (uint8_t *)bk_psram_frame_buffer_malloc(PSRAM_HEAP_ENCODE, node->frame.size);
				node->frame.frame = node->frame.base_addr;
			}

			if (node->frame.base_addr == NULL)
			{
				// add this node to free_list, and from ready node_list to find a frame_buffer(direct free)
				node->error = true;
				list_add_tail(&node->list, &mem_list->free);
				node = NULL;
			}
#if (CONFIG_PSRAM_WRITE_THROUGH)
			else
			{
				// checkout node->frame->frame is aligned 32byte(8word)
				if ((uint32_t)node->frame.base_addr & 0x1F)
				{
					node->frame.frame = (uint8_t *)((((uint32_t)node->frame.base_addr >> 5) + 1) << 5);
				}

				LOGD("%s, %d, %p, size:%d\r\n", __func__, type, node->frame.frame, size);
			}
#endif
		}
	}

	if (node == NULL && type != FB_INDEX_H264)
	{
		list_for_each_safe(pos, n, &mem_list->ready)
		{
			tmp = list_entry(pos, frame_buffer_node_t, list);
			if (tmp != NULL
			    && (tmp->read_mask == tmp->free_mask))
			{
				node = tmp;
				list_del(pos);
				break;
			}
		}
	}

	if (!isr_context)
	{
		GLOBAL_INT_RESTORE();
		rtos_unlock_mutex(&mem_list->lock);
	}

	if (node == NULL)
	{
		LOGD("%s %d failed...\n", __func__, type);
		return NULL;
	}

	node->frame.size = size;
	node->frame.length = 0;
	node->frame.width = 0;
	node->frame.height = 0;
	node->frame.h264_type = 0;
	node->frame.fmt = 0;
	node->read_mask = 0;
	node->free_mask = 0;
	node->frame.mix = 0;
	node->frame.sequence = 0;
	node->frame.err_state = false;
	node->frame.cb = NULL;

	return &node->frame;
}

void frame_buffer_fb_push(frame_buffer_t *frame)
{
	fb_mem_list_t *mem_list = NULL;
	fb_type_t type = frame_buffer_type_get(frame->fmt);
	frame_buffer_node_t *tmp = NULL, *node = list_entry(frame, frame_buffer_node_t, frame);
	uint32_t isr_context = platform_is_in_interrupt_context();
	LIST_HEADER_T *pos, *n;
	uint32_t i = 0, length = 0;
	bk_err_t ret;
	GLOBAL_INT_DECLARATION() = 0;

	if (type >= FB_INDEX_MAX)
	{
		LOGE("%s invalid frame type\n", __func__);
		return;
	}

	mem_list = &fb_mem_list[type];

	if (!isr_context)
	{
		rtos_lock_mutex(&mem_list->lock);
		GLOBAL_INT_DISABLE();
	}

	node->read_mask = 0;
	node->free_mask = 0;

	list_add_tail(&node->list, &mem_list->ready);

	if (type == FB_INDEX_DISPLAY)
	{
		list_for_each_safe(pos, n, &mem_list->ready)
		{
			tmp = list_entry(pos, frame_buffer_node_t, list);
			if (tmp != NULL)
			{
				length++;
			}
		}

		if (length == 1)
		{
			if (!isr_context)
			{
				GLOBAL_INT_RESTORE();
			}

			ret = rtos_set_semaphore(&mem_list->ready_sem);

			if (ret != BK_OK)
			{
				LOGD("%s semaphore set failed: %d\n", __func__, ret);
			}

			if (!isr_context)
			{
				GLOBAL_INT_DISABLE();
			}
		}
	}
	else
	{
		for (i = 0; i < MODULE_MAX; i++)
		{
			if (!isr_context)
			{
				GLOBAL_INT_RESTORE();
			}

			if (fb_info->modules[i].enable == true
			    && fb_info->modules[i].type == type)
			{
				LOGD("cmp plugin\n");

				if (mem_list->mode == FB_MEM_SHARED
				    && fb_info->modules[i].plugin == false)
				{
					xEventGroupSetBits(fb_info->modules[i].handle, FRAME_BUFFER_READ_COMPLETE);

					fb_info->modules[i].plugin = true;
				}
			}

			if (!isr_context)
			{
				GLOBAL_INT_DISABLE();
			}
		}
	}

	if (!isr_context)
	{
		GLOBAL_INT_RESTORE();
		rtos_unlock_mutex(&mem_list->lock);
	}
}

frame_buffer_t *frame_buffer_fb_pop(frame_module_t index, fb_type_t type)
{
	fb_mem_list_t *mem_list = &fb_mem_list[type];
	LIST_HEADER_T *pos, *n;
	frame_buffer_node_t *node = NULL, *tmp = NULL;
	uint32_t isr_context = platform_is_in_interrupt_context();
	GLOBAL_INT_DECLARATION();

	if (mem_list == NULL)
	{
		LOGE("%s invalid mem_list: %p\n", __func__, mem_list);
		return NULL;
	}

	if (fb_info == NULL)
	{
		LOGE("%s fb_info was NULL\n", __func__);
		return NULL;
	}

	if (!isr_context)
	{
		rtos_lock_mutex(&mem_list->lock);
		GLOBAL_INT_DISABLE();
	}

	LOGD("type: %d, %p\n", type, mem_list);

	list_for_each_safe_edge(pos, n, &mem_list->ready)
	{
		tmp = list_entry(pos, frame_buffer_node_t, list);
		if (tmp != NULL
		    && ((tmp->read_mask & INDEX_MASK(index)) == 0))
		{
			//LOGI("GET %u, %x, %x\n", tmp->frame.sequence, tmp->read_mask, tmp->read_mask & INDEX_MASK(index));
			node = tmp;
			break;
		}
	}

	if (node != NULL)
	{
		node->read_mask |= INDEX_MASK(index);
	}

	if (!isr_context)
	{
		GLOBAL_INT_RESTORE();
		rtos_unlock_mutex(&mem_list->lock);
	}

	if (node == NULL)
	{
		LOGD("pop NULL\n");
		return NULL;
	}

	//LOGI("pop %u, %x, %x\n", node->frame.sequence, node->read_mask, tmp->read_mask);

	return &node->frame;
}

fb_type_t frame_buffer_available_index(void)
{
	uint32_t isr_context = platform_is_in_interrupt_context();
	fb_type_t index = FB_INDEX_JPEG;
	GLOBAL_INT_DECLARATION();
	uint32_t i = 0;

	if (!isr_context)
	{
		rtos_lock_mutex(&fb_info->lock);
		GLOBAL_INT_DISABLE();
	}

	for (i = 0; i < FB_INDEX_MAX; i++)
	{
		if (!list_empty(&fb_mem_list[i].ready))
		{
			index = i;
			break;
		}
	}

	if (!isr_context)
	{
		GLOBAL_INT_RESTORE();
		rtos_unlock_mutex(&fb_info->lock);
	}

	return index;
}

frame_buffer_t *frame_buffer_fb_read(frame_module_t index)
{
	frame_buffer_t *frame = NULL;
	bk_err_t ret = BK_FAIL;
	uint32_t isr_context = platform_is_in_interrupt_context();
	GLOBAL_INT_DECLARATION();
	fb_type_t type;

	if (index >= MODULE_MAX)
	{
		LOGE("%s invalid module: %d\n", __func__, index);
		goto out;
	}

	if (fb_info->modules[index].enable == false)
	{
		LOGE("%s module not register: %d\n", __func__, index);
		goto out;
	}

	do
	{
		rtos_lock_mutex(&fb_info->modules[index].lock);

		if (fb_info->modules[index].enable == true)
		{
			ret = xEventGroupWaitBits(fb_info->modules[index].handle, FRAME_BUFFER_READ_COMPLETE, true, true, 200);//BEKEN_NEVER_TIMEOUT
		}

		if (ret != FRAME_BUFFER_READ_COMPLETE)
		{
			LOGD("%s EventGroup get failed: %d, index:%d, plugin:%d\n", __func__, ret, index, fb_info->modules[index].plugin);
			rtos_unlock_mutex(&fb_info->modules[index].lock);
			goto out;
		}

		if (fb_info->modules[index].enable == false)
		{
			rtos_unlock_mutex(&fb_info->modules[index].lock);
			break;
		}

		rtos_unlock_mutex(&fb_info->modules[index].lock);

		type = fb_info->modules[index].type;

		frame = frame_buffer_fb_pop(index, type);

		if (frame)
		{
			if (frame->err_state)
			{
				frame_buffer_fb_free(frame, index);
				continue;
			}
		}

		if (frame == NULL)
		{
			LOGD("%s faild, plugin: %d\n", __func__, fb_info->modules[index].plugin);

			if (!isr_context)
			{
				rtos_lock_mutex(&fb_info->lock);
				GLOBAL_INT_DISABLE();
			}

			fb_info->modules[index].plugin = false;

			if (!isr_context)
			{
				GLOBAL_INT_RESTORE();
				rtos_unlock_mutex(&fb_info->lock);
			}
		}
		else
		{
			break;
		}
	}
	while (fb_info && fb_info->modules[index].enable);

out:

	return frame;
}

bk_err_t frame_buffer_fb_register(frame_module_t index, fb_type_t type)
{
	bk_err_t ret = BK_FAIL;
	uint32_t isr_context = platform_is_in_interrupt_context();
	GLOBAL_INT_DECLARATION();

	if (fb_info->modules[index].enable == true)
	{
		LOGE("%s frame_module index already register\n", __func__);
		return ret;
	}

	if (index >= MODULE_MAX)
	{
		LOGE("%s invalid module: %d\n", __func__, index);
		return ret;
	}

	if (fb_info == NULL)
	{
		LOGE("%s fb_info was NULL\n", __func__);
		return ret;
	}

	if (!isr_context)
	{
		rtos_lock_mutex(&fb_info->lock);
		GLOBAL_INT_DISABLE();
	}

	fb_info->modules[index].plugin = false;
	fb_info->modules[index].type = type;
	fb_info->register_mask[type] |= INDEX_MASK(index);
	xEventGroupClearBits(fb_info->modules[index].handle, FRAME_BUFFER_READ_COMPLETE);
	xEventGroupSetBits(fb_info->modules[index].handle, FRAME_BUFFER_READ_COMPLETE);
	fb_info->modules[index].enable = true;

	if (!isr_context)
	{
		GLOBAL_INT_RESTORE();
		rtos_unlock_mutex(&fb_info->lock);
	}
	return ret;
}

bk_err_t frame_buffer_fb_deregister(frame_module_t index, fb_type_t type)
{
	bk_err_t ret = BK_FAIL;
	uint32_t isr_context = platform_is_in_interrupt_context();
	GLOBAL_INT_DECLARATION();

	if (fb_info->modules[index].enable == false)
	{
		LOGE("%s frame_module index already degister\n", __func__);
		return ret;
	}

	if (index >= MODULE_MAX)
	{
		LOGE("%s invalid module: %d\n", __func__, index);
		return ret;
	}

	if (fb_info == NULL)
	{
		LOGE("%s fb_info was NULL\n", __func__);
		return ret;
	}

	if (!isr_context)
	{
		rtos_lock_mutex(&fb_info->lock);
		GLOBAL_INT_DISABLE();
	}

	fb_info->modules[index].enable = false;
	fb_info->register_mask[type] &= INDEX_UNMASK(index);

	if (!isr_context)
	{
		GLOBAL_INT_RESTORE();
		rtos_unlock_mutex(&fb_info->lock);
	}

	return ret;
}

void frame_buffer_fb_direct_free_without_lock(frame_buffer_t *frame)
{
	if (frame == NULL)
	{
		LOGE("%s, frame is null\r\n", __func__);
		return;
	}

	bk_err_t ret = BK_OK;
	LIST_HEADER_T *pos, *n;
	fb_type_t type = frame_buffer_type_get(frame->fmt);
	fb_mem_list_t *mem_list = &fb_mem_list[type];
	frame_buffer_node_t *tmp = NULL, *node = list_entry(frame, frame_buffer_node_t, frame);
	uint32_t length = 0;
	uint32_t isr_context = platform_is_in_interrupt_context();
	GLOBAL_INT_DECLARATION() = 0;

	if (mem_list == NULL || type >= FB_INDEX_MAX)
	{
		LOGE("%s invalid mem_list: %p, type:%d\n", __func__, mem_list, type);
		return;
	}

	if (!isr_context)
	{
		//rtos_lock_mutex(&mem_list->lock);
		GLOBAL_INT_DISABLE();
	}

	node->free_mask = 0;
	node->read_mask = 0;

	// fix frame_buffer address, not free
	/*if (node->frame.base_addr)
	{
		bk_psram_frame_buffer_free(node->frame.base_addr);
		node->frame.base_addr = node->frame.frame = NULL;
	}*/

	list_add_tail(&node->list, &mem_list->free);

	if (type == FB_INDEX_DISPLAY)
	{
		list_for_each_safe(pos, n, &mem_list->free)
		{
			tmp = list_entry(pos, frame_buffer_node_t, list);
			if (tmp != NULL)
			{
				length++;
			}
		}

		if (mem_list->free_request == true)
		{
			if (!isr_context)
			{
				GLOBAL_INT_RESTORE();
			}

			ret = rtos_set_semaphore(&mem_list->free_sem);

			if (ret != BK_OK)
			{
				LOGE("%s semaphore set failed: %d\n", __func__, ret);
			}

			if (!isr_context)
			{
				GLOBAL_INT_DISABLE();
			}

			mem_list->free_request = false;
		}
	}

	if (!isr_context)
	{
		GLOBAL_INT_RESTORE();
		//rtos_unlock_mutex(&mem_list->lock);
	}
}

void frame_buffer_fb_direct_free(frame_buffer_t *frame)
{
	if (frame == NULL)
	{
		LOGE("%s, frame is null\r\n", __func__);
		return;
	}

	bk_err_t ret = BK_OK;
	LIST_HEADER_T *pos, *n;
	fb_type_t type = frame_buffer_type_get(frame->fmt);
	fb_mem_list_t *mem_list = &fb_mem_list[type];
	frame_buffer_node_t *tmp = NULL, *node = list_entry(frame, frame_buffer_node_t, frame);
	uint32_t length = 0;
	uint32_t isr_context = platform_is_in_interrupt_context();
	GLOBAL_INT_DECLARATION() = 0;

	if (mem_list == NULL || type >= FB_INDEX_MAX)
	{
		LOGE("%s invalid mem_list: %p, type:%d\n", __func__, mem_list, type);
		return;
	}

	if (!isr_context)
	{
		rtos_lock_mutex(&mem_list->lock);
		GLOBAL_INT_DISABLE();
	}

	node->free_mask = 0;
	node->read_mask = 0;

	// fix frame_buffer address, not free
	/*if (node->frame.base_addr)
	{
		bk_psram_frame_buffer_free(node->frame.base_addr);
		node->frame.base_addr = node->frame.frame = NULL;
	}*/

	list_add_tail(&node->list, &mem_list->free);

	if (type == FB_INDEX_DISPLAY)
	{
		list_for_each_safe(pos, n, &mem_list->free)
		{
			tmp = list_entry(pos, frame_buffer_node_t, list);
			if (tmp != NULL)
			{
				length++;
			}
		}

		if (mem_list->free_request == true)
		{
			if (!isr_context)
			{
				GLOBAL_INT_RESTORE();
			}

			ret = rtos_set_semaphore(&mem_list->free_sem);

			if (ret != BK_OK)
			{
				LOGE("%s semaphore set failed: %d\n", __func__, ret);
			}

			if (!isr_context)
			{
				GLOBAL_INT_DISABLE();
			}

			mem_list->free_request = false;
		}
	}

	if (!isr_context)
	{
		GLOBAL_INT_RESTORE();
		rtos_unlock_mutex(&mem_list->lock);
	}
}


frame_buffer_t *frame_buffer_fb_display_malloc_wait(uint32_t size)
{
	frame_buffer_t *frame = NULL;
	fb_mem_list_t *mem_list = &fb_mem_list[FB_INDEX_DISPLAY];
	uint32_t isr_context = platform_is_in_interrupt_context();
	bk_err_t ret;
	GLOBAL_INT_DECLARATION();

	do
	{
		frame = frame_buffer_fb_malloc(FB_INDEX_DISPLAY, size);

		if (frame == NULL)
		{
			if (!isr_context)
			{
				rtos_lock_mutex(&mem_list->lock);
				GLOBAL_INT_DISABLE();
			}

			mem_list->free_request = true;

			if (!isr_context)
			{
				GLOBAL_INT_RESTORE();
				rtos_unlock_mutex(&mem_list->lock);
			}

			ret = rtos_get_semaphore(&mem_list->free_sem, BEKEN_NEVER_TIMEOUT);

			if (ret != BK_OK)
			{
				LOGD("%s semaphore get failed: %d\n", __func__, ret);
			}

			continue;
		}

	}
	while (0);

	return frame;
}

frame_buffer_t *frame_buffer_fb_display_pop(void)
{
	frame_buffer_node_t *node = NULL, *tmp = NULL;
	fb_mem_list_t *mem_list = &fb_mem_list[FB_INDEX_DISPLAY];
	LIST_HEADER_T *pos, *n;
	GLOBAL_INT_DECLARATION() = 0;

	rtos_lock_mutex(&mem_list->lock);
	GLOBAL_INT_DISABLE();

	list_for_each_safe(pos, n, &mem_list->ready)
	{
		tmp = list_entry(pos, frame_buffer_node_t, list);
		if (tmp != NULL)
		{
			node = tmp;
			list_del(pos);
			break;
		}
	}

	GLOBAL_INT_RESTORE();
	rtos_unlock_mutex(&mem_list->lock);

	if (node == NULL)
	{
		LOGD("%s failed\n", __func__);
		return NULL;
	}

	return &node->frame;
}


frame_buffer_t *frame_buffer_fb_display_pop_wait(void)
{
	frame_buffer_t *frame = NULL;
	fb_mem_list_t *mem_list = &fb_mem_list[FB_INDEX_DISPLAY];
	bk_err_t ret;

	do
	{
		frame = frame_buffer_fb_display_pop();

		if (frame != NULL)
		{
			break;
		}

		ret = rtos_get_semaphore(&mem_list->ready_sem, 1000);

		if (ret != BK_OK)
		{
			LOGD("%s semaphore get failed: %d\n", __func__, ret);
			break;
		}

	}
	while (true);

	return frame;
}

void frame_buffer_init(void)
{
	uint32_t i = 0;

	for (i = 0; i < FB_INDEX_MAX; i++)
	{
		fb_mem_list[i].free.next = &fb_mem_list[i].free;
		fb_mem_list[i].free.prev = &fb_mem_list[i].free;
		fb_mem_list[i].ready.next = &fb_mem_list[i].ready;
		fb_mem_list[i].ready.prev = &fb_mem_list[i].ready;
		fb_mem_list[i].enable = false;
		rtos_init_mutex(&fb_mem_list[i].lock);
	}

	if (fb_info == NULL)
	{
		fb_info = (fb_info_t *)os_malloc(sizeof(fb_info_t));
		os_memset((void *)fb_info, 0, sizeof(fb_info_t));
		rtos_init_mutex(&fb_info->lock);

		for (uint8_t index = 0; index < MODULE_MAX; index++)
		{
			rtos_init_mutex(&fb_info->modules[index].lock);
			fb_info->modules[index].handle = xEventGroupCreate();
		}
	}

	bk_psram_frame_buffer_init();
}


void frame_buffer_deinit(void)
{
	if (fb_info)
	{
		for (uint8_t index = 0; index < MODULE_MAX; index++)
		{
			rtos_deinit_mutex(&fb_info->modules[index].lock);
			vEventGroupDelete(fb_info->modules[index].handle);
		}

		os_free(fb_info);
		fb_info = NULL;
	}
}

frame_buffer_t *frame_buffer_display_malloc(uint32_t size)
{
	frame_buffer_t *frame = bk_psram_frame_buffer_malloc(PSRAM_HEAP_YUV, size + sizeof(frame_buffer_t) + 32);

	if (frame == NULL)
	{
		return NULL;
	}

	os_memset(frame, 0, sizeof(frame_buffer_t));
	frame->frame = (uint8_t *)((((uint32_t)(frame + 1) >> 5) + 1) << 5);
	frame->size = size;

	return frame;
}

void frame_buffer_display_free(frame_buffer_t *frame)
{
	if (frame->cb == NULL) {
		bk_psram_frame_buffer_free(frame);
	} else {
		if (frame->cb->free) {
			frame->cb->free(frame);
		} else {
			LOGE("frame buffer free callback is null\r\n");
		}
	}
}