/*
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * FFmpeg is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with FFmpeg; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

#ifndef AVCODEC_V4L2_REQUEST_H
#define AVCODEC_V4L2_REQUEST_H

#include <stdbool.h>
#include <stdint.h>
#include <linux/videodev2.h>

#include "libavutil/buffer.h"
#include "libavutil/log.h"
#include "libavutil/thread.h"
#include "avcodec.h"

typedef struct AVV4L2RequestFramesContextInternal AVV4L2RequestFramesContextInternal;

// Below defines are custom patch from Collabora, as these are not included in the current AOSP, So better just
// here rather than messing with the headers in the AOSP build system. by - Venkata Atchuta Bheemeswara Sarma Darbha
#define V4L2_CID_STATELESS_HEVC_EXT_SPS_ST_RPS  (V4L2_CID_CODEC_STATELESS_BASE + 408)
#define V4L2_CID_STATELESS_HEVC_EXT_SPS_LT_RPS  (V4L2_CID_CODEC_STATELESS_BASE + 409)

#define V4L2_HEVC_EXT_SPS_ST_RPS_FLAG_INTER_REF_PIC_SET_PRED	0x1

#define V4L2_HEVC_EXT_SPS_LT_RPS_FLAG_USED_LT		0x1

#define	V4L2_CTRL_TYPE_HEVC_EXT_SPS_ST_RPS	= 0x0275,
#define	V4L2_CTRL_TYPE_HEVC_EXT_SPS_LT_RPS	= 0x0276,



struct v4l2_ctrl_hevc_ext_sps_st_rps {
	__u8	delta_idx_minus1;
	__u8	delta_rps_sign;
	__u16	abs_delta_rps_minus1;
	__u8	num_negative_pics;
	__u8	num_positive_pics;
	__u32	used_by_curr_pic;
	__u32	use_delta_flag;
	__u16	delta_poc_s0_minus1[16];
	__u16	delta_poc_s1_minus1[16];
	__u8	flags;
};

/*
 * struct v4l2_ctrl_hevc_ext_sps_lt_rps - HEVC long term RPS parameters
 *
 * Dynamic size 1-dimension array for long term RPS. The number of elements
 * is v4l2_ctrl_hevc_sps::num_long_term_ref_pics_sps. It can contain up to 65 elements.
 *
 * @lt_ref_pic_poc_lsb_sps: picture order count modulo MaxPicOrderCntLsb of the i-th candidate
 *                          long-term reference picture.
 * @flags: See V4L2_HEVC_EXT_SPS_LT_RPS_FLAG_{}
 */
struct v4l2_ctrl_hevc_ext_sps_lt_rps {
	__u16	lt_ref_pic_poc_lsb_sps;
	__u8	flags;
};

/**
 * v4l2_timeval_to_ns - Convert timeval to nanoseconds
 * @tv:		pointer to the timeval variable to be converted
 *
 * Returns the scalar nanosecond representation of the timeval
 * parameter.
 */
static __inline__ __u64 v4l2_timeval_to_ns(const struct timeval *tv)
{
	return (__u64)tv->tv_sec * 1000000000ULL + tv->tv_usec * 1000;
}

typedef struct V4L2RequestOutputBuffer {
    AVBufferRef *ref;
    uint32_t index;
    int request_fd;
    uint8_t *addr;
    uint32_t size;
    uint32_t bytesused;
    struct timeval timestamp;
} V4L2RequestOutputBuffer;

typedef struct V4L2RequestContext {
    const AVClass *av_class;
    AVBufferRef *frames_ref;
    AVV4L2RequestFramesContextInternal *fctxi;
    AVMutex mutex;
    V4L2RequestOutputBuffer output[4];
    uint8_t next_output;
    uint32_t queued_output;
    uint32_t queued_request;
    uint64_t queued_capture;
    struct v4l2_ext_control *init_controls;
    int nb_init_controls;
} V4L2RequestContext;

typedef struct V4L2RequestPictureContext {
    V4L2RequestOutputBuffer *output;
    uint32_t capture_index;
} V4L2RequestPictureContext;

uint64_t ff_v4l2_request_get_capture_timestamp(AVFrame *frame);

int ff_v4l2_request_query_control(AVCodecContext *avctx,
                                  struct v4l2_query_ext_ctrl *control);

int ff_v4l2_request_query_control_default_value(AVCodecContext *avctx,
                                                uint32_t id);

int ff_v4l2_request_set_controls(AVCodecContext *avctx,
                                 struct v4l2_ext_control *control, int count);

int ff_v4l2_request_append_output(AVCodecContext *avctx,
                                  V4L2RequestPictureContext *pic,
                                  const uint8_t *data, uint32_t size);

int ff_v4l2_request_decode_slice(AVCodecContext *avctx,
                                 V4L2RequestPictureContext *pic,
                                 struct v4l2_ext_control *control, int count,
                                 bool first_slice, bool last_slice);

int ff_v4l2_request_decode_frame(AVCodecContext *avctx,
                                 V4L2RequestPictureContext *pic,
                                 struct v4l2_ext_control *control, int count);

int ff_v4l2_request_reset_picture(AVCodecContext *avctx,
                                  V4L2RequestPictureContext *pic);

int ff_v4l2_request_start_frame(AVCodecContext *avctx,
                                V4L2RequestPictureContext *pic, AVFrame *frame);

void ff_v4l2_request_flush(AVCodecContext *avctx);

int ff_v4l2_request_frame_params(AVCodecContext *avctx,
                                 AVBufferRef *hw_frames_ctx,
                                 uint32_t pixelformat,
                                 uint8_t bit_depth);

int ff_v4l2_request_uninit(AVCodecContext *avctx);

int ff_v4l2_request_init(AVCodecContext *avctx,
                         struct v4l2_ext_control *control, int count,
                         int (*post_frames_ctx)(AVCodecContext *avctx));

#endif /* AVCODEC_V4L2_REQUEST_H */