#include "libavutil/atomic.h"
#include "libavutil/attributes.h"
#include "libavutil/common.h"
#include "libavutil/internal.h"
#include "libavutil/md5.h"
#include "libavutil/opt.h"
#include "libavutil/pixdesc.h"
#include "libavutil/stereo3d.h"
#include "bytestream.h"
#include "cabac_functions.h"
#include "dsputil.h"
#include "golomb.h"
#include "hevc.h"

Macros
#define	CTB(tab, x, y) ((tab)[(y) * s->sps->ctb_width + (x)])

#define	SET_SAO(elem, value)

#define	POS(c_idx, x, y)

#define	STARTCODE_TEST

#define	OFFSET(x) offsetof(HEVCContext, x)

#define	PAR (AV_OPT_FLAG_DECODING_PARAM \| AV_OPT_FLAG_VIDEO_PARAM)

Functions
static void	pic_arrays_free (HEVCContext *s)
	NOTE: Each function hls_foo correspond to the function foo in the specification (HLS stands for High Level Syntax).

static int	pic_arrays_init (HEVCContext s, const HEVCSPS sps)

static void	pred_weight_table (HEVCContext s, GetBitContext gb)

static int	decode_lt_rps (HEVCContext s, LongTermRPS rps, GetBitContext *gb)

static int	set_sps (HEVCContext s, const HEVCSPS sps)

static int	hls_slice_header (HEVCContext *s)

static void	hls_sao_param (HEVCContext *s, int rx, int ry)

static int	hls_transform_unit (HEVCContext *s, int x0, int y0, int xBase, int yBase, int cb_xBase, int cb_yBase, int log2_cb_size, int log2_trafo_size, int trafo_depth, int blk_idx)

static void	set_deblocking_bypass (HEVCContext *s, int x0, int y0, int log2_cb_size)

static int	hls_transform_tree (HEVCContext *s, int x0, int y0, int xBase, int yBase, int cb_xBase, int cb_yBase, int log2_cb_size, int log2_trafo_size, int trafo_depth, int blk_idx)

static int	hls_pcm_sample (HEVCContext *s, int x0, int y0, int log2_cb_size)

static void	luma_mc (HEVCContext s, int16_t dst, ptrdiff_t dststride, AVFrame ref, const Mv mv, int x_off, int y_off, int block_w, int block_h)
	8.5.3.2.2.1 Luma sample interpolation process

static void	chroma_mc (HEVCContext s, int16_t dst1, int16_t dst2, ptrdiff_t dststride, AVFrame ref, const Mv *mv, int x_off, int y_off, int block_w, int block_h)
	8.5.3.2.2.2 Chroma sample interpolation process

static void	hevc_await_progress (HEVCContext s, HEVCFrame ref, const Mv *mv, int y0, int height)

static void	hls_prediction_unit (HEVCContext *s, int x0, int y0, int nPbW, int nPbH, int log2_cb_size, int partIdx)

static int	luma_intra_pred_mode (HEVCContext *s, int x0, int y0, int pu_size, int prev_intra_luma_pred_flag)
	8.4.1

static av_always_inline void	set_ct_depth (HEVCContext *s, int x0, int y0, int log2_cb_size, int ct_depth)

static void	intra_prediction_unit (HEVCContext *s, int x0, int y0, int log2_cb_size)

static void	intra_prediction_unit_default_value (HEVCContext *s, int x0, int y0, int log2_cb_size)

static int	hls_coding_unit (HEVCContext *s, int x0, int y0, int log2_cb_size)

static int	hls_coding_quadtree (HEVCContext *s, int x0, int y0, int log2_cb_size, int cb_depth)

static void	hls_decode_neighbour (HEVCContext *s, int x_ctb, int y_ctb, int ctb_addr_ts)

static int	hls_decode_entry (AVCodecContext avctxt, void isFilterThread)

static int	hls_slice_data (HEVCContext *s)

static int	hls_decode_entry_wpp (AVCodecContext avctxt, void input_ctb_row, int job, int self_id)

static int	hls_slice_data_wpp (HEVCContext s, const uint8_t nal, int length)

static int	hls_nal_unit (HEVCContext *s)

static void	restore_tqb_pixels (HEVCContext *s)

static int	set_side_data (HEVCContext *s)

static int	hevc_frame_start (HEVCContext *s)

static int	decode_nal_unit (HEVCContext s, const uint8_t nal, int length)

int	ff_hevc_extract_rbsp (HEVCContext s, const uint8_t src, int length, HEVCNAL *nal)

static int	decode_nal_units (HEVCContext s, const uint8_t buf, int length)

static void	print_md5 (void *log_ctx, int level, uint8_t md5[16])

static int	verify_md5 (HEVCContext s, AVFrame frame)

static int	hevc_decode_frame (AVCodecContext avctx, void data, int got_output, AVPacket avpkt)

static int	hevc_ref_frame (HEVCContext s, HEVCFrame dst, HEVCFrame *src)

static av_cold int	hevc_decode_free (AVCodecContext *avctx)

static av_cold int	hevc_init_context (AVCodecContext *avctx)

static int	hevc_update_thread_context (AVCodecContext dst, const AVCodecContext src)

static int	hevc_decode_extradata (HEVCContext *s)

static av_cold int	hevc_decode_init (AVCodecContext *avctx)

static av_cold int	hevc_init_thread_copy (AVCodecContext *avctx)

static void	hevc_decode_flush (AVCodecContext *avctx)

Variables
const uint8_t	ff_hevc_qpel_extra_before [4] = { 0, 3, 3, 2 }

const uint8_t	ff_hevc_qpel_extra_after [4] = { 0, 3, 4, 4 }

const uint8_t	ff_hevc_qpel_extra [4] = { 0, 6, 7, 6 }

static const AVProfile	profiles []

static const AVOption	options []

static const AVClass	hevc_decoder_class

AVCodec	ff_hevc_decoder

Macro Definition Documentation

#define CTB	(	tab,
		x,
		y
	)	((tab)[(y) * s->sps->ctb_width + (x)])

Definition at line 719 of file hevc.c.

Referenced by hls_sao_param().

#define SET_SAO	(	elem,
		value
	)

Value:

do {                                                    \
    if (!sao_merge_up_flag && !sao_merge_left_flag)     \
        sao->elem = value;                              \
    else if (sao_merge_left_flag)                       \
        sao->elem = CTB(s->sao, rx-1, ry).elem;         \
    else if (sao_merge_up_flag)                         \
        sao->elem = CTB(s->sao, rx, ry-1).elem;         \
    else                                                \
        sao->elem = 0;                                  \
} while (0)

Definition at line 721 of file hevc.c.

Referenced by hls_sao_param().

#define POS	(	c_idx,
		x,
		y
	)

Value:

&s->frame->data[c_idx][((y) >> s->sps->vshift[c_idx]) * s->frame->linesize[c_idx] + \

(((x) >> s->sps->hshift[c_idx]) << s->sps->pixel_shift)]

Referenced by hls_prediction_unit().

#define STARTCODE_TEST

Value:

if (i + 2 < length && src[i + 1] == 0 && src[i + 2] <= 3) {     \
            if (src[i + 2] != 3) {                                      \
                /* startcode, so we must be past the end */             \
                length = i;                                             \
            }                                                           \
            break;                                                      \
        }

#define OFFSET ( x ) offsetof(HEVCContext, x)

Definition at line 3086 of file hevc.c.

#define PAR (AV_OPT_FLAG_DECODING_PARAM | AV_OPT_FLAG_VIDEO_PARAM)

Definition at line 3087 of file hevc.c.

Function Documentation

static void pic_arrays_free ( HEVCContext * s )

static

NOTE: Each function hls_foo correspond to the function foo in the specification (HLS stands for High Level Syntax).

Section 5.7

Definition at line 55 of file hevc.c.

Referenced by hevc_decode_free(), pic_arrays_init(), and set_sps().

static int pic_arrays_init	(	HEVCContext *	s,
		const HEVCSPS *	sps
	)

static

Definition at line 84 of file hevc.c.

Referenced by set_sps().

static void pred_weight_table	(	HEVCContext *	s,
		GetBitContext *	gb
	)

static

Definition at line 142 of file hevc.c.

Referenced by hls_slice_header().

static int decode_lt_rps	(	HEVCContext *	s,
		LongTermRPS *	rps,
		GetBitContext *	gb
	)

static

Definition at line 231 of file hevc.c.

Referenced by hls_slice_header().

static int set_sps	(	HEVCContext *	s,
		const HEVCSPS *	sps
	)

static

Definition at line 283 of file hevc.c.

Referenced by hevc_update_thread_context(), and hls_slice_header().

static int hls_slice_header ( HEVCContext * s )

static

Definition at line 352 of file hevc.c.

Referenced by decode_nal_unit().

static void hls_sao_param	(	HEVCContext *	s,
		int	rx,
		int	ry
	)

static

Definition at line 733 of file hevc.c.

Referenced by hls_decode_entry(), and hls_decode_entry_wpp().

static int hls_transform_unit	(	HEVCContext *	s,
		int	x0,
		int	y0,
		int	xBase,
		int	yBase,
		int	cb_xBase,
		int	cb_yBase,
		int	log2_cb_size,
		int	log2_trafo_size,
		int	trafo_depth,
		int	blk_idx
	)

static

Definition at line 804 of file hevc.c.

Referenced by hls_transform_tree().

static void set_deblocking_bypass	(	HEVCContext *	s,
		int	x0,
		int	y0,
		int	log2_cb_size
	)

static

Definition at line 892 of file hevc.c.

Referenced by hls_coding_unit(), and hls_transform_tree().

static int hls_transform_tree	(	HEVCContext *	s,
		int	x0,
		int	y0,
		int	xBase,
		int	yBase,
		int	cb_xBase,
		int	cb_yBase,
		int	log2_cb_size,
		int	log2_trafo_size,
		int	trafo_depth,
		int	blk_idx
	)

static

Definition at line 907 of file hevc.c.

Referenced by hls_coding_unit().

static int hls_pcm_sample	(	HEVCContext *	s,
		int	x0,
		int	y0,
		int	log2_cb_size
	)

static

Definition at line 1027 of file hevc.c.

Referenced by hls_coding_unit().

static void luma_mc	(	HEVCContext *	s,
		int16_t *	dst,
		ptrdiff_t	dststride,
		AVFrame *	ref,
		const Mv *	mv,
		int	x_off,
		int	y_off,
		int	block_w,
		int	block_h
	)

static

8.5.3.2.2.1 Luma sample interpolation process

Parameters

s	HEVC decoding context
dst	target buffer for block data at block position
dststride	stride of the dst buffer
ref	reference picture buffer at origin (0, 0)
mv	motion vector (relative to block position) to get pixel data from
x_off	horizontal position of block from origin (0, 0)
y_off	vertical position of block from origin (0, 0)
block_w	width of block
block_h	height of block

Definition at line 1071 of file hevc.c.

Referenced by hls_prediction_unit().

static void chroma_mc	(	HEVCContext *	s,
		int16_t *	dst1,
		int16_t *	dst2,
		ptrdiff_t	dststride,
		AVFrame *	ref,
		const Mv *	mv,
		int	x_off,
		int	y_off,
		int	block_w,
		int	block_h
	)

static

8.5.3.2.2.2 Chroma sample interpolation process

Parameters

s	HEVC decoding context
dst1	target buffer for block data at block position (U plane)
dst2	target buffer for block data at block position (V plane)
dststride	stride of the dst1 and dst2 buffers
ref	reference picture buffer at origin (0, 0)
mv	motion vector (relative to block position) to get pixel data from
x_off	horizontal position of block from origin (0, 0)
y_off	vertical position of block from origin (0, 0)
block_w	width of block
block_h	height of block

Definition at line 1125 of file hevc.c.

Referenced by hls_prediction_unit().

static void hevc_await_progress	(	HEVCContext *	s,
		HEVCFrame *	ref,
		const Mv *	mv,
		int	y0,
		int	height
	)

static

Definition at line 1190 of file hevc.c.

Referenced by hls_prediction_unit().

static void hls_prediction_unit	(	HEVCContext *	s,
		int	x0,
		int	y0,
		int	nPbW,
		int	nPbH,
		int	log2_cb_size,
		int	partIdx
	)

static

Definition at line 1199 of file hevc.c.

Referenced by hls_coding_unit().

static int luma_intra_pred_mode	(	HEVCContext *	s,
		int	x0,
		int	y0,
		int	pu_size,
		int	prev_intra_luma_pred_flag
	)

static

8.4.1

Definition at line 1463 of file hevc.c.

Referenced by intra_prediction_unit().

static av_always_inline void set_ct_depth	(	HEVCContext *	s,
		int	x0,
		int	y0,
		int	log2_cb_size,
		int	ct_depth
	)

static

Definition at line 1551 of file hevc.c.

Referenced by hls_coding_unit().

static void intra_prediction_unit	(	HEVCContext *	s,
		int	x0,
		int	y0,
		int	log2_cb_size
	)

static

Definition at line 1564 of file hevc.c.

Referenced by hls_coding_unit().

static void intra_prediction_unit_default_value	(	HEVCContext *	s,
		int	x0,
		int	y0,
		int	log2_cb_size
	)

static

Definition at line 1604 of file hevc.c.

Referenced by hls_coding_unit().

static int hls_coding_unit	(	HEVCContext *	s,
		int	x0,
		int	y0,
		int	log2_cb_size
	)

static

Definition at line 1626 of file hevc.c.

Referenced by hls_coding_quadtree().

static int hls_coding_quadtree	(	HEVCContext *	s,
		int	x0,
		int	y0,
		int	log2_cb_size,
		int	cb_depth
	)

static

Definition at line 1784 of file hevc.c.

Referenced by hls_decode_entry(), and hls_decode_entry_wpp().

static void hls_decode_neighbour	(	HEVCContext *	s,
		int	x_ctb,
		int	y_ctb,
		int	ctb_addr_ts
	)

static

Definition at line 1865 of file hevc.c.

Referenced by hls_decode_entry(), and hls_decode_entry_wpp().

static int hls_decode_entry	(	AVCodecContext *	avctxt,
		void *	isFilterThread
	)

static

Definition at line 1917 of file hevc.c.

Referenced by hls_slice_data().

static int hls_slice_data ( HEVCContext * s )

static

Definition at line 1973 of file hevc.c.

Referenced by decode_nal_unit().

static int hls_decode_entry_wpp	(	AVCodecContext *	avctxt,
		void *	input_ctb_row,
		int	job,
		int	self_id
	)

static

Definition at line 1984 of file hevc.c.

Referenced by hls_slice_data_wpp().

static int hls_slice_data_wpp	(	HEVCContext *	s,
		const uint8_t *	nal,
		int	length
	)

static

Definition at line 2059 of file hevc.c.

Referenced by decode_nal_unit().

static int hls_nal_unit ( HEVCContext * s )

static

Returns: AVERROR_INVALIDDATA if the packet is not a valid NAL unit, 0 if the unit should be skipped, 1 otherwise

Definition at line 2140 of file hevc.c.

Referenced by decode_nal_unit(), and decode_nal_units().

static void restore_tqb_pixels ( HEVCContext * s )

static

Definition at line 2162 of file hevc.c.

Referenced by decode_nal_unit().

static int set_side_data ( HEVCContext * s )

static

Definition at line 2189 of file hevc.c.

Referenced by hevc_frame_start().

static int hevc_frame_start ( HEVCContext * s )

static

Definition at line 2224 of file hevc.c.

Referenced by decode_nal_unit().

static int decode_nal_unit	(	HEVCContext *	s,
		const uint8_t *	nal,
		int	length
	)

static

Definition at line 2274 of file hevc.c.

Referenced by decode_nal_units().

int ff_hevc_extract_rbsp	(	HEVCContext *	s,
		const uint8_t *	src,
		int	length,
		HEVCNAL *	nal
	)

Definition at line 2417 of file hevc.c.

Referenced by decode_nal_units(), and parse_nal_units().

static int decode_nal_units	(	HEVCContext *	s,
		const uint8_t *	buf,
		int	length
	)

static

Definition at line 2524 of file hevc.c.

Referenced by hevc_decode_extradata(), and hevc_decode_frame().

static void print_md5	(	void *	log_ctx,
		int	level,
		uint8_t	md5[16]
	)

static

Definition at line 2636 of file hevc.c.

Referenced by verify_md5().

static int verify_md5	(	HEVCContext *	s,
		AVFrame *	frame
	)

static

Definition at line 2643 of file hevc.c.

Referenced by hevc_decode_frame().

static int hevc_decode_frame	(	AVCodecContext *	avctx,
		void *	data,
		int *	got_output,
		AVPacket *	avpkt
	)

static

Definition at line 2709 of file hevc.c.

static int hevc_ref_frame	(	HEVCContext *	s,
		HEVCFrame *	dst,
		HEVCFrame *	src
	)

static

Definition at line 2753 of file hevc.c.

Referenced by hevc_update_thread_context().

static av_cold int hevc_decode_free ( AVCodecContext * avctx )

static

Definition at line 2787 of file hevc.c.

Referenced by hevc_decode_init(), and hevc_init_context().

static av_cold int hevc_init_context ( AVCodecContext * avctx )

static

Definition at line 2844 of file hevc.c.

Referenced by hevc_decode_init(), hevc_init_thread_copy(), and hevc_update_thread_context().

static int hevc_update_thread_context	(	AVCodecContext *	dst,
		const AVCodecContext *	src
	)

static

Definition at line 2893 of file hevc.c.

static int hevc_decode_extradata ( HEVCContext * s )

static

Definition at line 2964 of file hevc.c.

Referenced by hevc_decode_init().

static av_cold int hevc_decode_init ( AVCodecContext * avctx )

static

Definition at line 3028 of file hevc.c.

static av_cold int hevc_init_thread_copy ( AVCodecContext * avctx )

static

Definition at line 3065 of file hevc.c.

static void hevc_decode_flush ( AVCodecContext * avctx )

static

Definition at line 3079 of file hevc.c.

Variable Documentation

const uint8_t ff_hevc_qpel_extra_before[4] = { 0, 3, 3, 2 }

Definition at line 41 of file hevc.c.

Referenced by luma_mc().

const uint8_t ff_hevc_qpel_extra_after[4] = { 0, 3, 4, 4 }

Definition at line 42 of file hevc.c.

Referenced by luma_mc().

const uint8_t ff_hevc_qpel_extra[4] = { 0, 6, 7, 6 }

Definition at line 43 of file hevc.c.

Referenced by luma_mc().

const AVProfile profiles[]

static

Initial value:

= {
    { FF_PROFILE_HEVC_MAIN,                 "Main"                },
    { FF_PROFILE_HEVC_MAIN_10,              "Main 10"             },
    { FF_PROFILE_HEVC_MAIN_STILL_PICTURE,   "Main Still Picture"  },
    { FF_PROFILE_UNKNOWN },
}

Definition at line 3089 of file hevc.c.

const AVOption options[]

static

Initial value:

= {
    { "apply_defdispwin", "Apply default display window from VUI", OFFSET(apply_defdispwin),
        AV_OPT_TYPE_INT, {.i64 = 0}, 0, 1, PAR },
    { "strict-displaywin", "stricly apply default display window size", OFFSET(apply_defdispwin),
        AV_OPT_TYPE_INT, {.i64 = 0}, 0, 1, PAR },
    { NULL },
}

Definition at line 3096 of file hevc.c.

const AVClass hevc_decoder_class

static

Initial value:

= {
    .class_name = "HEVC decoder",
    .item_name  = av_default_item_name,
    .option     = options,
    .version    = LIBAVUTIL_VERSION_INT,
}

Definition at line 3104 of file hevc.c.

AVCodec ff_hevc_decoder

Initial value:

= {
    .name                  = "hevc",
    .long_name             = NULL_IF_CONFIG_SMALL("HEVC (High Efficiency Video Coding)"),
    .type                  = AVMEDIA_TYPE_VIDEO,
    .id                    = AV_CODEC_ID_HEVC,
    .priv_data_size        = sizeof(HEVCContext),
    .priv_class            = &hevc_decoder_class,
    .init                  = hevc_decode_init,
    .close                 = hevc_decode_free,
    .decode                = hevc_decode_frame,
    .flush                 = hevc_decode_flush,
    .update_thread_context = hevc_update_thread_context,
    .init_thread_copy      = hevc_init_thread_copy,
    .capabilities          = CODEC_CAP_DR1 | CODEC_CAP_DELAY |
                             CODEC_CAP_SLICE_THREADS | CODEC_CAP_FRAME_THREADS,
    .profiles              = NULL_IF_CONFIG_SMALL(profiles),
}

Definition at line 3111 of file hevc.c.

Macros

Functions

Variables

Macro Definition Documentation

Function Documentation

Variable Documentation