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/display.h"
#include "libavutil/internal.h"
#include "libavutil/md5.h"
#include "libavutil/opt.h"
#include "libavutil/pixdesc.h"
#include "libavutil/stereo3d.h"
#include "bswapdsp.h"
#include "bytestream.h"
#include "cabac_functions.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 SUBDIVIDE(x, y, idx)
 
#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 get_buffer_sao (HEVCContext *s, AVFrame *frame, const HEVCSPS *sps)
 
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_cross_component_pred (HEVCContext *s, int idx)
 
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 blk_idx, int cbf_luma, int *cbf_cb, int *cbf_cr)
 
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, const int *base_cbf_cb, const int *base_cbf_cr)
 
static int hls_pcm_sample (HEVCContext *s, int x0, int y0, int log2_cb_size)
 
static void luma_mc_uni (HEVCContext *s, uint8_t *dst, ptrdiff_t dststride, AVFrame *ref, const Mv *mv, int x_off, int y_off, int block_w, int block_h, int luma_weight, int luma_offset)
 8.5.3.2.2.1 Luma sample unidirectional interpolation process
 
static void luma_mc_bi (HEVCContext *s, uint8_t *dst, ptrdiff_t dststride, AVFrame *ref0, const Mv *mv0, int x_off, int y_off, int block_w, int block_h, AVFrame *ref1, const Mv *mv1, struct MvField *current_mv)
 8.5.3.2.2.1 Luma sample bidirectional interpolation process
 
static void chroma_mc_uni (HEVCContext *s, uint8_t *dst0, ptrdiff_t dststride, uint8_t *src0, ptrdiff_t srcstride, int reflist, int x_off, int y_off, int block_w, int block_h, struct MvField *current_mv, int chroma_weight, int chroma_offset)
 8.5.3.2.2.2 Chroma sample uniprediction interpolation process
 
static void chroma_mc_bi (HEVCContext *s, uint8_t *dst0, ptrdiff_t dststride, AVFrame *ref0, AVFrame *ref1, int x_off, int y_off, int block_w, int block_h, struct MvField *current_mv, int cidx)
 8.5.3.2.2.2 Chroma sample bidirectional interpolation process
 
static void hevc_await_progress (HEVCContext *s, HEVCFrame *ref, const Mv *mv, int y0, int height)
 
static void hevc_luma_mv_mpv_mode (HEVCContext *s, int x0, int y0, int nPbW, int nPbH, int log2_cb_size, int part_idx, int merge_idx, MvField *mv)
 
static void hls_prediction_unit (HEVCContext *s, int x0, int y0, int nPbW, int nPbH, int log2_cb_size, int partIdx, int idx)
 
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 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_pel_weight [65] = { [2] = 0, [4] = 1, [6] = 2, [8] = 3, [12] = 4, [16] = 5, [24] = 6, [32] = 7, [48] = 8, [64] = 9 }
 
static const uint8_t tab_mode_idx []
 
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 761 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 763 of file hevc.c.

Referenced by hls_sao_param().

#define SUBDIVIDE (   x,
  y,
  idx 
)
Value:
do { \
ret = hls_transform_tree(s, x, y, x0, y0, cb_xBase, cb_yBase, log2_cb_size, \
log2_trafo_size - 1, trafo_depth + 1, idx, \
cbf_cb, cbf_cr); \
if (ret < 0) \
return ret; \
} while (0)

Referenced by hls_transform_tree().

#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 3404 of file hevc.c.

Definition at line 3405 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(), pic_arrays_init(), and set_sps().

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

Definition at line 82 of file hevc.c.

Referenced by set_sps().

static void pred_weight_table ( HEVCContext s,
GetBitContext gb 
)
static

Definition at line 138 of file hevc.c.

Referenced by hls_slice_header().

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

Definition at line 227 of file hevc.c.

Referenced by hls_slice_header().

static int get_buffer_sao ( HEVCContext s,
AVFrame frame,
const HEVCSPS sps 
)
static

Definition at line 279 of file hevc.c.

Referenced by set_sps().

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

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

Referenced by decode_nal_unit().

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

Definition at line 775 of file hevc.c.

Referenced by hls_decode_entry(), and hls_decode_entry_wpp().

static int hls_cross_component_pred ( HEVCContext s,
int  idx 
)
static

Definition at line 849 of file hevc.c.

Referenced by hls_transform_unit().

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  blk_idx,
int  cbf_luma,
int *  cbf_cb,
int *  cbf_cr 
)
static

Definition at line 865 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 1071 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,
const int *  base_cbf_cb,
const int *  base_cbf_cr 
)
static

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

Referenced by hls_coding_unit().

static void luma_mc_uni ( HEVCContext s,
uint8_t dst,
ptrdiff_t  dststride,
AVFrame ref,
const Mv mv,
int  x_off,
int  y_off,
int  block_w,
int  block_h,
int  luma_weight,
int  luma_offset 
)
static

8.5.3.2.2.1 Luma sample unidirectional 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
luma_weightweighting factor applied to the luma prediction
luma_offsetadditive offset applied to the luma prediction value

Definition at line 1263 of file hevc.c.

Referenced by hls_prediction_unit().

static void luma_mc_bi ( HEVCContext s,
uint8_t dst,
ptrdiff_t  dststride,
AVFrame ref0,
const Mv mv0,
int  x_off,
int  y_off,
int  block_w,
int  block_h,
AVFrame ref1,
const Mv mv1,
struct MvField current_mv 
)
static

8.5.3.2.2.1 Luma sample bidirectional interpolation process

Parameters
sHEVC decoding context
dsttarget buffer for block data at block position
dststridestride of the dst buffer
ref0reference picture0 buffer at origin (0, 0)
mv0motion vector0 (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
ref1reference picture1 buffer at origin (0, 0)
mv1motion vector1 (relative to block position) to get pixel data from
current_mvcurrent motion vector structure

Definition at line 1324 of file hevc.c.

Referenced by hls_prediction_unit().

static void chroma_mc_uni ( HEVCContext s,
uint8_t dst0,
ptrdiff_t  dststride,
uint8_t src0,
ptrdiff_t  srcstride,
int  reflist,
int  x_off,
int  y_off,
int  block_w,
int  block_h,
struct MvField current_mv,
int  chroma_weight,
int  chroma_offset 
)
static

8.5.3.2.2.2 Chroma sample uniprediction 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
chroma_weightweighting factor applied to the chroma prediction
chroma_offsetadditive offset applied to the chroma prediction value

Definition at line 1415 of file hevc.c.

Referenced by hls_prediction_unit().

static void chroma_mc_bi ( HEVCContext s,
uint8_t dst0,
ptrdiff_t  dststride,
AVFrame ref0,
AVFrame ref1,
int  x_off,
int  y_off,
int  block_w,
int  block_h,
struct MvField current_mv,
int  cidx 
)
static

8.5.3.2.2.2 Chroma sample bidirectional interpolation process

Parameters
sHEVC decoding context
dsttarget buffer for block data at block position
dststridestride of the dst buffer
ref0reference picture0 buffer at origin (0, 0)
mv0motion vector0 (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
ref1reference picture1 buffer at origin (0, 0)
mv1motion vector1 (relative to block position) to get pixel data from
current_mvcurrent motion vector structure
cidxchroma component(cb, cr)

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

Referenced by hls_prediction_unit().

static void hevc_luma_mv_mpv_mode ( HEVCContext s,
int  x0,
int  y0,
int  nPbW,
int  nPbH,
int  log2_cb_size,
int  part_idx,
int  merge_idx,
MvField mv 
)
static

Definition at line 1579 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,
int  idx 
)
static

Definition at line 1624 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 1744 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 1824 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 1841 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 1909 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 1932 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 2095 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 2180 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 2229 of file hevc.c.

Referenced by hls_slice_data().

static int hls_slice_data ( HEVCContext s)
static

Definition at line 2285 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 2296 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 2371 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 2452 of file hevc.c.

Referenced by decode_nal_unit(), and decode_nal_units().

static int set_side_data ( HEVCContext s)
static

Definition at line 2474 of file hevc.c.

Referenced by hevc_frame_start().

static int hevc_frame_start ( HEVCContext s)
static

Definition at line 2523 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 2579 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 2718 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 2825 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 2938 of file hevc.c.

Referenced by verify_md5().

static int verify_md5 ( HEVCContext s,
AVFrame frame 
)
static

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

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

Definition at line 3055 of file hevc.c.

Referenced by hevc_update_thread_context().

static av_cold int hevc_decode_free ( AVCodecContext avctx)
static

Definition at line 3089 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 3150 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 3200 of file hevc.c.

static int hevc_decode_extradata ( HEVCContext s)
static

Definition at line 3282 of file hevc.c.

Referenced by hevc_decode_init().

static av_cold int hevc_decode_init ( AVCodecContext avctx)
static

Definition at line 3346 of file hevc.c.

static av_cold int hevc_init_thread_copy ( AVCodecContext avctx)
static

Definition at line 3383 of file hevc.c.

static void hevc_decode_flush ( AVCodecContext avctx)
static

Definition at line 3397 of file hevc.c.

Variable Documentation

const uint8_t ff_hevc_pel_weight[65] = { [2] = 0, [4] = 1, [6] = 2, [8] = 3, [12] = 4, [16] = 5, [24] = 6, [32] = 7, [48] = 8, [64] = 9 }

Definition at line 42 of file hevc.c.

Referenced by chroma_mc_bi(), chroma_mc_uni(), luma_mc_bi(), and luma_mc_uni().

const uint8_t tab_mode_idx[]
static
Initial value:
= {
0, 1, 2, 2, 2, 2, 3, 5, 7, 8, 10, 12, 13, 15, 17, 18, 19, 20,
21, 22, 23, 23, 24, 24, 25, 25, 26, 27, 27, 28, 28, 29, 29, 30, 31}

Definition at line 1837 of file hevc.c.

Referenced by intra_prediction_unit().

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_HEVC_REXT, "Rext" },
}

Definition at line 3407 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 3415 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 3423 of file hevc.c.

AVCodec ff_hevc_decoder
Initial value:

Definition at line 3430 of file hevc.c.