FFmpeg
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
Macros | Functions | Variables
hevc.c File Reference
#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"

Go to the source code of this file.

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.

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
sHEVC decoding context
dsttarget buffer for block data at block position
dststridestride of the dst buffer
refreference picture buffer at origin (0, 0)
mvmotion vector (relative to block position) to get pixel data from
x_offhorizontal position of block from origin (0, 0)
y_offvertical position of block from origin (0, 0)
block_wwidth of block
block_hheight 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
sHEVC decoding context
dst1target buffer for block data at block position (U plane)
dst2target buffer for block data at block position (V plane)
dststridestride of the dst1 and dst2 buffers
refreference picture buffer at origin (0, 0)
mvmotion vector (relative to block position) to get pixel data from
x_offhorizontal position of block from origin (0, 0)
y_offvertical position of block from origin (0, 0)
block_wwidth of block
block_hheight 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" },
}

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,
}

Definition at line 3104 of file hevc.c.

AVCodec ff_hevc_decoder
Initial value:

Definition at line 3111 of file hevc.c.