// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright (c) 2021 MediaTek Inc.
 * Author: Yunfei Dong <yunfei.dong@mediatek.com>
 */

#ifndef _VDEC_MSG_QUEUE_H_
#define _VDEC_MSG_QUEUE_H_

#include <linux/sched.h>
#include <linux/semaphore.h>
#include <linux/slab.h>
#include <media/videobuf2-v4l2.h>

#include "mtk_vcodec_util.h"

#define NUM_BUFFER_COUNT 3
#define VDEC_LAT_SLICE_HEADER_SZ    (640 * 1024)
#define VDEC_ERR_MAP_SZ_AVC         ((8192 / 16) * (4352 / 16) / 8)

struct vdec_lat_buf;
struct mtk_vcodec_ctx;
struct mtk_vcodec_dev;
typedef int (*core_decode_t)(struct vdec_lat_buf *lat_buf);

struct vdec_lat_buf {
	struct mtk_vcodec_mem wdma_err_addr;
	struct mtk_vcodec_mem slice_bc_addr;
	struct vb2_v4l2_buffer ts_info;

	void *private_data;
	struct mtk_vcodec_ctx *ctx;
	core_decode_t core_decode;
	struct list_head lat_list;
	struct list_head core_list;
};

struct vdec_msg_queue {
	struct vdec_lat_buf lat_buf[NUM_BUFFER_COUNT];

	struct mtk_vcodec_mem wdma_addr;
	uint64_t wdma_rptr_addr;
	uint64_t wdma_wptr_addr;

	wait_queue_head_t lat_read;
	struct list_head lat_queue;
	spinlock_t lat_lock;
	int num_lat;
	bool init_done;
};

/**
 * vdec_msg_queue_init - init lat buffer information.
 * @ctx: v4l2 ctx
 * @msg_queue: used to store lat buffer information
 * @core_decode: core decoder callback for each codec
 * @private_size: the size used to share with core
 */
int vdec_msg_queue_init(
	struct mtk_vcodec_ctx *ctx,
	struct vdec_msg_queue *msg_queue,
	core_decode_t core_decode,
	int private_size);

/**
 * vdec_msg_queue_get_lat_buf - get used lat buffer for core decode.
 * @msg_queue: used to store lat buffer information
 */
struct vdec_lat_buf *vdec_msg_queue_get_lat_buf(
	struct vdec_msg_queue *msg_queue);

/**
 * vdec_msg_queue_get_core_buf - get used core buffer for lat decode.
 * @dev: mtk vcodec device
 */
struct vdec_lat_buf *vdec_msg_queue_get_core_buf(
	struct mtk_vcodec_dev *dev);

/**
 * vdec_msg_queue_buf_to_core - queue buf to core for core decode.
 * @dev: mtk vcodec device
 * @buf: current lat buffer
 */
void vdec_msg_queue_buf_to_core(struct mtk_vcodec_dev *dev,
	struct vdec_lat_buf *buf);

/**
 * vdec_msg_queue_buf_to_lat - queue buf to lat for lat decode.
 * @buf: current lat buffer
 */
void vdec_msg_queue_buf_to_lat(struct vdec_lat_buf *buf);

/**
 * vdec_msg_queue_update_ube_rptr - used to updata ube read point.
 * @msg_queue: used to store lat buffer information
 * @ube_rptr: current ube read point
 */
void vdec_msg_queue_update_ube_rptr(struct vdec_msg_queue *msg_queue,
	uint64_t ube_rptr);

/**
 * vdec_msg_queue_update_ube_wptr - used to updata ube write point.
 * @msg_queue: used to store lat buffer information
 * @ube_wptr: current ube write point
 */
void vdec_msg_queue_update_ube_wptr(struct vdec_msg_queue *msg_queue,
	uint64_t ube_wptr);

/**
 * vdec_msg_queue_wait_lat_buf_full - used to check whether all lat buffer
 *                                    in lat list.
 * @msg_queue: used to store lat buffer information
 */
bool vdec_msg_queue_wait_lat_buf_full(struct vdec_msg_queue *msg_queue);

/**
 * vdec_msg_queue_deinit - deinit lat buffer information.
 * @ctx: v4l2 ctx
 * @msg_queue: used to store lat buffer information
 */
void vdec_msg_queue_deinit(
	struct mtk_vcodec_ctx *ctx,
	struct vdec_msg_queue *msg_queue);

/**
 * vdec_msg_queue_core_thead - used for core decoder.
 * @data: private data used for each codec
 */
int vdec_msg_queue_core_thead(void *data);

#endif