#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 "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)

static void	pred_weight_table (HEVCContext s, GetBitContext gb)

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

static int	hls_slice_header (HEVCContext *s)

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

static void	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 void	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	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 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 662 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 664 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 2887 of file hevc.c.

#define PAR (AV_OPT_FLAG_DECODING_PARAM | AV_OPT_FLAG_VIDEO_PARAM)

Definition at line 2888 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 54 of file hevc.c.

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

static int pic_arrays_init ( HEVCContext * s )

static

Definition at line 83 of file hevc.c.

Referenced by hls_slice_header().

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 hls_slice_header ( HEVCContext * s )

static

Definition at line 283 of file hevc.c.

Referenced by decode_nal_unit().

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

static

Definition at line 676 of file hevc.c.

Referenced by hls_decode_entry(), and hls_decode_entry_wpp().

static void 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 747 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 821 of file hevc.c.

Referenced by hls_coding_unit(), and hls_transform_tree().

static void 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 836 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 937 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 981 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 1029 of file hevc.c.

Referenced by hls_prediction_unit(), and rv34_mc().

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

static

Definition at line 1083 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 1092 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 1351 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 1439 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 1452 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 1492 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 1514 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 1664 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 1730 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 1784 of file hevc.c.

Referenced by hls_slice_data().

static int hls_slice_data ( HEVCContext * s )

static

Definition at line 1829 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 1840 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 1913 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 1995 of file hevc.c.

Referenced by decode_nal_unit(), and decode_nal_units().

static void restore_tqb_pixels ( HEVCContext * s )

static

Definition at line 2017 of file hevc.c.

Referenced by decode_nal_unit().

static int hevc_frame_start ( HEVCContext * s )

static

Definition at line 2044 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 2093 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 2228 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 2335 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 2447 of file hevc.c.

Referenced by verify_md5().

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

static

Definition at line 2454 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 2520 of file hevc.c.

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

static

Definition at line 2565 of file hevc.c.

Referenced by hevc_update_thread_context().

static av_cold int hevc_decode_free ( AVCodecContext * avctx )

static

Definition at line 2599 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 2660 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 2708 of file hevc.c.

static int hevc_decode_extradata ( HEVCContext * s )

static

Definition at line 2767 of file hevc.c.

Referenced by hevc_decode_init().

static av_cold int hevc_decode_init ( AVCodecContext * avctx )

static

Definition at line 2830 of file hevc.c.

static av_cold int hevc_init_thread_copy ( AVCodecContext * avctx )

static

Definition at line 2866 of file hevc.c.

static void hevc_decode_flush ( AVCodecContext * avctx )

static

Definition at line 2880 of file hevc.c.

Variable Documentation

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

Definition at line 40 of file hevc.c.

Referenced by luma_mc().

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

Definition at line 41 of file hevc.c.

Referenced by luma_mc().

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

Definition at line 42 of file hevc.c.

Referenced by luma_mc().

const AVOption options[]

static

Initial value:

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

Definition at line 2889 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 2895 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,
}

Definition at line 2902 of file hevc.c.

Macros

Functions

Variables

Macro Definition Documentation

Function Documentation

Variable Documentation