[media] vcodec: mediatek: Add Mediatek V4L2 Video Encoder Driver

	    To compile this driver as a module, choose M here: the

	    module will be called mtk-vpu.

config VIDEO_MEDIATEK_VCODEC

	tristate "Mediatek Video Codec driver"

	depends on MTK_IOMMU

	depends on VIDEO_DEV && VIDEO_V4L2

	depends on ARCH_MEDIATEK || COMPILE_TEST

	select VIDEOBUF2_DMA_CONTIG

	select V4L2_MEM2MEM_DEV

	select VIDEO_MEDIATEK_VPU

	default n

	---help---

	    Mediatek video codec driver provides HW capability to

	    encode and decode in a range of video formats

	    This driver rely on VPU driver to communicate with VPU.

	    To compile this driver as a module, choose M here: the

	    module will be called mtk-vcodec

config VIDEO_MEM2MEM_DEINTERLACE

	tristate "Deinterlace support"

	depends on VIDEO_DEV && VIDEO_V4L2 && DMA_ENGINE

ccflags-y += -I$(srctree)/drivers/media/i2c

obj-$(CONFIG_VIDEO_MEDIATEK_VPU)	+= mtk-vpu/

obj-$(CONFIG_VIDEO_MEDIATEK_VCODEC)	+= mtk-vcodec/

obj-$(CONFIG_VIDEO_MEDIATEK_VCODEC) += mtk-vcodec-enc.o mtk-vcodec-common.o

mtk-vcodec-enc-y := mtk_vcodec_enc.o \

		mtk_vcodec_enc_drv.o \

		mtk_vcodec_enc_pm.o \

		venc_drv_if.o \

mtk-vcodec-common-y := mtk_vcodec_intr.o \

		mtk_vcodec_util.o\

ccflags-y += -I$(srctree)/drivers/media/platform/mtk-vpu

/*

* Copyright (c) 2016 MediaTek Inc.

* Author: PC Chen <pc.chen@mediatek.com>

*         Tiffany Lin <tiffany.lin@mediatek.com>

*

* This program is free software; you can redistribute it and/or modify

* it under the terms of the GNU General Public License version 2 as

* published by the Free Software Foundation.

*

* This program 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 General Public License for more details.

*/

#ifndef _MTK_VCODEC_DRV_H_

#define _MTK_VCODEC_DRV_H_

#include <linux/platform_device.h>

#include <linux/videodev2.h>

#include <media/v4l2-ctrls.h>

#include <media/v4l2-device.h>

#include <media/v4l2-ioctl.h>

#include <media/videobuf2-core.h>

#include "mtk_vcodec_util.h"

#define MTK_VCODEC_DRV_NAME	"mtk_vcodec_drv"

#define MTK_VCODEC_ENC_NAME	"mtk-vcodec-enc"

#define MTK_PLATFORM_STR	"platform:mt8173"

#define MTK_VCODEC_MAX_PLANES	3

#define MTK_V4L2_BENCHMARK	0

#define WAIT_INTR_TIMEOUT_MS	1000

/**

 * enum mtk_hw_reg_idx - MTK hw register base index

 */

enum mtk_hw_reg_idx {

	VDEC_SYS,

	VDEC_MISC,

	VDEC_LD,

	VDEC_TOP,

	VDEC_CM,

	VDEC_AD,

	VDEC_AV,

	VDEC_PP,

	VDEC_HWD,

	VDEC_HWQ,

	VDEC_HWB,

	VDEC_HWG,

	NUM_MAX_VDEC_REG_BASE,

	/* h264 encoder */

	VENC_SYS = NUM_MAX_VDEC_REG_BASE,

	/* vp8 encoder */

	VENC_LT_SYS,

	NUM_MAX_VCODEC_REG_BASE

};

/**

 * enum mtk_instance_type - The type of an MTK Vcodec instance.

 */

enum mtk_instance_type {

	MTK_INST_DECODER		= 0,

	MTK_INST_ENCODER		= 1,

};

/**

 * enum mtk_instance_state - The state of an MTK Vcodec instance.

 * @MTK_STATE_FREE - default state when instance is created

 * @MTK_STATE_INIT - vcodec instance is initialized

 * @MTK_STATE_HEADER - vdec had sps/pps header parsed or venc

 *			had sps/pps header encoded

 * @MTK_STATE_FLUSH - vdec is flushing. Only used by decoder

 * @MTK_STATE_ABORT - vcodec should be aborted

 */

enum mtk_instance_state {

	MTK_STATE_FREE = 0,

	MTK_STATE_INIT = 1,

	MTK_STATE_HEADER = 2,

	MTK_STATE_FLUSH = 3,

	MTK_STATE_ABORT = 4,

};

/**

 * struct mtk_encode_param - General encoding parameters type

 */

enum mtk_encode_param {

	MTK_ENCODE_PARAM_NONE = 0,

	MTK_ENCODE_PARAM_BITRATE = (1 << 0),

	MTK_ENCODE_PARAM_FRAMERATE = (1 << 1),

	MTK_ENCODE_PARAM_INTRA_PERIOD = (1 << 2),

	MTK_ENCODE_PARAM_FORCE_INTRA = (1 << 3),

	MTK_ENCODE_PARAM_GOP_SIZE = (1 << 4),

};

enum mtk_fmt_type {

	MTK_FMT_DEC = 0,

	MTK_FMT_ENC = 1,

	MTK_FMT_FRAME = 2,

};

/**

 * struct mtk_video_fmt - Structure used to store information about pixelformats

 */

struct mtk_video_fmt {

	u32	fourcc;

	enum mtk_fmt_type	type;

	u32	num_planes;

};

/**

 * struct mtk_codec_framesizes - Structure used to store information about

 *							framesizes

 */

struct mtk_codec_framesizes {

	u32	fourcc;

	struct	v4l2_frmsize_stepwise	stepwise;

};

/**

 * struct mtk_q_type - Type of queue

 */

enum mtk_q_type {

	MTK_Q_DATA_SRC = 0,

	MTK_Q_DATA_DST = 1,

};

/**

 * struct mtk_q_data - Structure used to store information about queue

 */

struct mtk_q_data {

	unsigned int	visible_width;

	unsigned int	visible_height;

	unsigned int	coded_width;

	unsigned int	coded_height;

	enum v4l2_field	field;

	unsigned int	bytesperline[MTK_VCODEC_MAX_PLANES];

	unsigned int	sizeimage[MTK_VCODEC_MAX_PLANES];

	struct mtk_video_fmt	*fmt;

};

/**

 * struct mtk_enc_params - General encoding parameters

 * @bitrate: target bitrate in bits per second

 * @num_b_frame: number of b frames between p-frame

 * @rc_frame: frame based rate control

 * @rc_mb: macroblock based rate control

 * @seq_hdr_mode: H.264 sequence header is encoded separately or joined

 *		  with the first frame

 * @intra_period: I frame period

 * @gop_size: group of picture size, it's used as the intra frame period

 * @framerate_num: frame rate numerator. ex: framerate_num=30 and

 *		   framerate_denom=1 menas FPS is 30

 * @framerate_denom: frame rate denominator. ex: framerate_num=30 and

 *		     framerate_denom=1 menas FPS is 30

 * @h264_max_qp: Max value for H.264 quantization parameter

 * @h264_profile: V4L2 defined H.264 profile

 * @h264_level: V4L2 defined H.264 level

 * @force_intra: force/insert intra frame

 */

struct mtk_enc_params {

	unsigned int	bitrate;

	unsigned int	num_b_frame;

	unsigned int	rc_frame;

	unsigned int	rc_mb;

	unsigned int	seq_hdr_mode;

	unsigned int	intra_period;

	unsigned int	gop_size;

	unsigned int	framerate_num;

	unsigned int	framerate_denom;

	unsigned int	h264_max_qp;

	unsigned int	h264_profile;

	unsigned int	h264_level;

	unsigned int	force_intra;

};

/**

 * struct mtk_vcodec_pm - Power management data structure

 */

struct mtk_vcodec_pm {

	struct clk	*vcodecpll;

	struct clk	*univpll_d2;

	struct clk	*clk_cci400_sel;

	struct clk	*vdecpll;

	struct clk	*vdec_sel;

	struct clk	*vencpll_d2;

	struct clk	*venc_sel;

	struct clk	*univpll1_d2;

	struct clk	*venc_lt_sel;

	struct device	*larbvdec;

	struct device	*larbvenc;

	struct device	*larbvenclt;

	struct device	*dev;

	struct mtk_vcodec_dev	*mtkdev;

};

/**

 * struct mtk_vcodec_ctx - Context (instance) private data.

 *

 * @type: type of the instance - decoder or encoder

 * @dev: pointer to the mtk_vcodec_dev of the device

 * @list: link to ctx_list of mtk_vcodec_dev

 * @fh: struct v4l2_fh

 * @m2m_ctx: pointer to the v4l2_m2m_ctx of the context

 * @q_data: store information of input and output queue

 *	    of the context

 * @id: index of the context that this structure describes

 * @state: state of the context

 * @param_change: indicate encode parameter type

 * @enc_params: encoding parameters

 * @enc_if: hoooked encoder driver interface

 * @drv_handle: driver handle for specific decode/encode instance

 *

 * @int_cond: variable used by the waitqueue

 * @int_type: type of the last interrupt

 * @queue: waitqueue that can be used to wait for this context to

 *	   finish

 * @irq_status: irq status

 *

 * @ctrl_hdl: handler for v4l2 framework

 * @encode_work: worker for the encoding

 *

 * @colorspace: enum v4l2_colorspace; supplemental to pixelformat

 * @ycbcr_enc: enum v4l2_ycbcr_encoding, Y'CbCr encoding

 * @quantization: enum v4l2_quantization, colorspace quantization

 * @xfer_func: enum v4l2_xfer_func, colorspace transfer function

 */

struct mtk_vcodec_ctx {

	enum mtk_instance_type type;

	struct mtk_vcodec_dev *dev;

	struct list_head list;

	struct v4l2_fh fh;

	struct v4l2_m2m_ctx *m2m_ctx;

	struct mtk_q_data q_data[2];

	int id;

	enum mtk_instance_state state;

	enum mtk_encode_param param_change;

	struct mtk_enc_params enc_params;

	struct venc_common_if *enc_if;

	unsigned long drv_handle;

	int int_cond;

	int int_type;

	wait_queue_head_t queue;

	unsigned int irq_status;

	struct v4l2_ctrl_handler ctrl_hdl;

	struct work_struct encode_work;

	enum v4l2_colorspace colorspace;

	enum v4l2_ycbcr_encoding ycbcr_enc;

	enum v4l2_quantization quantization;

	enum v4l2_xfer_func xfer_func;

};

/**

 * struct mtk_vcodec_dev - driver data

 * @v4l2_dev: V4L2 device to register video devices for.

 * @vfd_enc: Video device for encoder.

 *

 * @m2m_dev_enc: m2m device for encoder.

 * @plat_dev: platform device

 * @vpu_plat_dev: mtk vpu platform device

 * @alloc_ctx: VB2 allocator context

 *	       (for allocations without kernel mapping).

 * @ctx_list: list of struct mtk_vcodec_ctx

 * @irqlock: protect data access by irq handler and work thread

 * @curr_ctx: The context that is waiting for codec hardware

 *

 * @reg_base: Mapped address of MTK Vcodec registers.

 *

 * @id_counter: used to identify current opened instance

 * @num_instances: counter of active MTK Vcodec instances

 *

 * @encode_workqueue: encode work queue

 *

 * @int_cond: used to identify interrupt condition happen

 * @int_type: used to identify what kind of interrupt condition happen

 * @dev_mutex: video_device lock

 * @queue: waitqueue for waiting for completion of device commands

 *

 * @enc_irq: h264 encoder irq resource

 * @enc_lt_irq: vp8 encoder irq resource

 *

 * @enc_mutex: encoder hardware lock.

 *

 * @pm: power management control

 * @dec_capability: used to identify decode capability, ex: 4k

 * @enc_capability: used to identify encode capability

 */

struct mtk_vcodec_dev {

	struct v4l2_device v4l2_dev;

	struct video_device *vfd_enc;

	struct v4l2_m2m_dev *m2m_dev_enc;

	struct platform_device *plat_dev;

	struct platform_device *vpu_plat_dev;

	struct vb2_alloc_ctx *alloc_ctx;

	struct list_head ctx_list;

	spinlock_t irqlock;

	struct mtk_vcodec_ctx *curr_ctx;

	void __iomem *reg_base[NUM_MAX_VCODEC_REG_BASE];

	unsigned long id_counter;

	int num_instances;

	struct workqueue_struct *encode_workqueue;

	int int_cond;

	int int_type;

	struct mutex dev_mutex;

	wait_queue_head_t queue;

	int enc_irq;

	int enc_lt_irq;

	struct mutex enc_mutex;

	struct mtk_vcodec_pm pm;

	unsigned int dec_capability;

	unsigned int enc_capability;

};

static inline struct mtk_vcodec_ctx *fh_to_ctx(struct v4l2_fh *fh)

{

	return container_of(fh, struct mtk_vcodec_ctx, fh);

}

static inline struct mtk_vcodec_ctx *ctrl_to_ctx(struct v4l2_ctrl *ctrl)

{

	return container_of(ctrl->handler, struct mtk_vcodec_ctx, ctrl_hdl);

}

#endif /* _MTK_VCODEC_DRV_H_ */