28 #ifndef AVCODEC_H264DEC_H
29 #define AVCODEC_H264DEC_H
53 #define H264_MAX_PICTURE_COUNT 36
55 #define MAX_MMCO_COUNT 66
57 #define MAX_DELAYED_PIC_COUNT 16
61 #define ALLOW_INTERLACE
71 #ifdef ALLOW_INTERLACE
72 #define MB_MBAFF(h) (h)->mb_mbaff
73 #define MB_FIELD(sl) (sl)->mb_field_decoding_flag
74 #define FRAME_MBAFF(h) (h)->mb_aff_frame
75 #define FIELD_PICTURE(h) ((h)->picture_structure != PICT_FRAME)
82 #define MB_FIELD(sl) 0
83 #define FRAME_MBAFF(h) 0
84 #define FIELD_PICTURE(h) 0
86 #define IS_INTERLACED(mb_type) 0
92 #define FIELD_OR_MBAFF_PICTURE(h) (FRAME_MBAFF(h) || FIELD_PICTURE(h))
95 #define CABAC(h) (h)->ps.pps->cabac
98 #define CHROMA(h) ((h)->ps.sps->chroma_format_idc)
99 #define CHROMA422(h) ((h)->ps.sps->chroma_format_idc == 2)
100 #define CHROMA444(h) ((h)->ps.sps->chroma_format_idc == 3)
102 #define MB_TYPE_REF0 MB_TYPE_ACPRED
103 #define MB_TYPE_8x8DCT 0x01000000
104 #define IS_REF0(a) ((a) & MB_TYPE_REF0)
105 #define IS_8x8DCT(a) ((a) & MB_TYPE_8x8DCT)
397 #define LIST_NOT_USED -1
398 #define PART_NOT_AVAILABLE -2
524 #define FRAME_RECOVERED_IDR (1 << 0)
529 #define FRAME_RECOVERED_SEI (1 << 1)
612 unsigned int linesize,
unsigned int uvlinesize);
615 unsigned int linesize,
unsigned int uvlinesize);
647 #define LUMA_DC_BLOCK_INDEX 48
648 #define CHROMA_DC_BLOCK_INDEX 49
652 4 + 1 * 8, 5 + 1 * 8, 4 + 2 * 8, 5 + 2 * 8,
653 6 + 1 * 8, 7 + 1 * 8, 6 + 2 * 8, 7 + 2 * 8,
654 4 + 3 * 8, 5 + 3 * 8, 4 + 4 * 8, 5 + 4 * 8,
655 6 + 3 * 8, 7 + 3 * 8, 6 + 4 * 8, 7 + 4 * 8,
656 4 + 6 * 8, 5 + 6 * 8, 4 + 7 * 8, 5 + 7 * 8,
657 6 + 6 * 8, 7 + 6 * 8, 6 + 7 * 8, 7 + 7 * 8,
658 4 + 8 * 8, 5 + 8 * 8, 4 + 9 * 8, 5 + 9 * 8,
659 6 + 8 * 8, 7 + 8 * 8, 6 + 9 * 8, 7 + 9 * 8,
660 4 + 11 * 8, 5 + 11 * 8, 4 + 12 * 8, 5 + 12 * 8,
661 6 + 11 * 8, 7 + 11 * 8, 6 + 12 * 8, 7 + 12 * 8,
662 4 + 13 * 8, 5 + 13 * 8, 4 + 14 * 8, 5 + 14 * 8,
663 6 + 13 * 8, 7 + 13 * 8, 6 + 14 * 8, 7 + 14 * 8,
664 0 + 0 * 8, 0 + 5 * 8, 0 + 10 * 8
670 return (
b & 0xFFFF) + (
a << 16);
672 return (
a & 0xFFFF) + (
b << 16);
679 return (
b & 0xFF) + (
a << 8);
681 return (
a & 0xFF) + (
b << 8);
690 return pps->chroma_qp_table[t][qscale];
699 const int index8 =
scan8[n];
704 ff_tlog(
h->avctx,
"mode:%d %d min:%d\n", left, top,
min);
719 i4x4[4] = i4x4_cache[7 + 8 * 3];
720 i4x4[5] = i4x4_cache[7 + 8 * 2];
721 i4x4[6] = i4x4_cache[7 + 8 * 1];
727 const int mb_xy = sl->
mb_xy;
728 uint8_t *nnz =
h->non_zero_count[mb_xy];
731 AV_COPY32(&nnz[ 0], &nnz_cache[4 + 8 * 1]);
732 AV_COPY32(&nnz[ 4], &nnz_cache[4 + 8 * 2]);
733 AV_COPY32(&nnz[ 8], &nnz_cache[4 + 8 * 3]);
734 AV_COPY32(&nnz[12], &nnz_cache[4 + 8 * 4]);
735 AV_COPY32(&nnz[16], &nnz_cache[4 + 8 * 6]);
736 AV_COPY32(&nnz[20], &nnz_cache[4 + 8 * 7]);
737 AV_COPY32(&nnz[32], &nnz_cache[4 + 8 * 11]);
738 AV_COPY32(&nnz[36], &nnz_cache[4 + 8 * 12]);
740 if (!
h->chroma_y_shift) {
741 AV_COPY32(&nnz[24], &nnz_cache[4 + 8 * 8]);
742 AV_COPY32(&nnz[28], &nnz_cache[4 + 8 * 9]);
743 AV_COPY32(&nnz[40], &nnz_cache[4 + 8 * 13]);
744 AV_COPY32(&nnz[44], &nnz_cache[4 + 8 * 14]);
752 int mb_type,
int list)
754 int16_t(*mv_dst)[2] = &
h->cur_pic.motion_val[list][b_xy];
756 AV_COPY128(mv_dst + 0 * b_stride, mv_src + 8 * 0);
757 AV_COPY128(mv_dst + 1 * b_stride, mv_src + 8 * 1);
758 AV_COPY128(mv_dst + 2 * b_stride, mv_src + 8 * 2);
759 AV_COPY128(mv_dst + 3 * b_stride, mv_src + 8 * 3);
762 :
h->mb2br_xy[sl->
mb_xy]];
768 AV_COPY16(mvd_dst + 3 + 3, mvd_src + 3 + 8 * 0);
769 AV_COPY16(mvd_dst + 3 + 2, mvd_src + 3 + 8 * 1);
770 AV_COPY16(mvd_dst + 3 + 1, mvd_src + 3 + 8 * 2);
775 int8_t *ref_index = &
h->cur_pic.ref_index[list][b8_xy];
777 ref_index[0 + 0 * 2] = ref_cache[
scan8[0]];
778 ref_index[1 + 0 * 2] = ref_cache[
scan8[4]];
779 ref_index[0 + 1 * 2] = ref_cache[
scan8[8]];
780 ref_index[1 + 1 * 2] = ref_cache[
scan8[12]];
788 const int b_stride =
h->b_stride;
789 const int b_xy = 4 * sl->
mb_x + 4 * sl->
mb_y *
h->b_stride;
790 const int b8_xy = 4 * sl->
mb_xy;
813 if (
h->ps.sps->direct_8x8_inference_flag)
816 0x0001000100010001ULL));
820 0x0001000100010001ULL));
824 int buf_index,
int next_avc)
830 return FFMIN(buf_index, buf_size);
refcounted data buffer API
static int FUNC() pps(CodedBitstreamContext *ctx, RWContext *rw, H264RawPPS *current)
static void fill_rectangle(int x, int y, int w, int h)
#define DECLARE_ALIGNED(n, t, v)
Declare a variable that is aligned in memory.
@ AV_PICTURE_TYPE_B
Bi-dir predicted.
H.264 decoder/parser shared code.
H.264 parameter set handling.
void ff_h264_direct_ref_list_init(const H264Context *const h, H264SliceContext *sl)
int ff_h264_decode_ref_pic_list_reordering(H264SliceContext *sl, void *logctx)
int ff_h264_decode_ref_pic_marking(H264SliceContext *sl, GetBitContext *gb, const H2645NAL *nal, void *logctx)
void ff_h264_hl_decode_mb(const H264Context *h, H264SliceContext *sl)
int ff_h264_update_thread_context(AVCodecContext *dst, const AVCodecContext *src)
static const uint8_t scan8[16 *3+3]
const uint16_t ff_h264_mb_sizes[4]
int ff_h264_alloc_tables(H264Context *h)
Allocate tables.
static av_always_inline void write_back_motion(const H264Context *h, H264SliceContext *sl, int mb_type)
int ff_h264_slice_context_init(H264Context *h, H264SliceContext *sl)
Init context Allocate buffers which are not shared amongst multiple threads.
static av_always_inline int pred_intra_mode(const H264Context *h, H264SliceContext *sl, int n)
Get the predicted intra4x4 prediction mode.
int ff_h264_decode_mb_cavlc(const H264Context *h, H264SliceContext *sl)
Decode a macroblock.
#define H264_MAX_PICTURE_COUNT
int ff_h264_execute_decode_slices(H264Context *h)
Call decode_slice() for each context.
void ff_h264_remove_all_refs(H264Context *h)
void ff_h264_free_tables(H264Context *h)
static av_always_inline void write_back_motion_list(const H264Context *h, H264SliceContext *sl, int b_stride, int b_xy, int b8_xy, int mb_type, int list)
static av_always_inline int get_dct8x8_allowed(const H264Context *h, H264SliceContext *sl)
int ff_h264_queue_decode_slice(H264Context *h, const H2645NAL *nal)
Submit a slice for decoding.
void ff_h264_draw_horiz_band(const H264Context *h, H264SliceContext *sl, int y, int height)
static av_always_inline void write_back_non_zero_count(const H264Context *h, H264SliceContext *sl)
static av_always_inline uint32_t pack16to32(unsigned a, unsigned b)
static av_always_inline int get_chroma_qp(const PPS *pps, int t, int qscale)
Get the chroma qp.
void ff_h264_filter_mb(const H264Context *h, H264SliceContext *sl, int mb_x, int mb_y, uint8_t *img_y, uint8_t *img_cb, uint8_t *img_cr, unsigned int linesize, unsigned int uvlinesize)
void ff_h264_direct_dist_scale_factor(const H264Context *const h, H264SliceContext *sl)
static av_always_inline uint16_t pack8to16(unsigned a, unsigned b)
int ff_h264_execute_ref_pic_marking(H264Context *h)
Execute the reference picture marking (memory management control operations).
#define MAX_DELAYED_PIC_COUNT
MMCOOpcode
Memory management control operation opcode.
void ff_h264_decode_init_vlc(void)
int ff_h264_decode_mb_cabac(const H264Context *h, H264SliceContext *sl)
Decode a CABAC coded macroblock.
int ff_h264_get_slice_type(const H264SliceContext *sl)
Reconstruct bitstream slice_type.
int ff_h264_build_ref_list(H264Context *h, H264SliceContext *sl)
static int find_start_code(const uint8_t *buf, int buf_size, int buf_index, int next_avc)
void ff_h264_pred_direct_motion(const H264Context *const h, H264SliceContext *sl, int *mb_type)
#define MAX_SLICES
The maximum number of slices supported by the decoder.
static av_always_inline void write_back_intra_pred_mode(const H264Context *h, H264SliceContext *sl)
int ff_h264_field_end(H264Context *h, H264SliceContext *sl, int in_setup)
void ff_h264_unref_picture(H264Context *h, H264Picture *pic)
void ff_h264_init_cabac_states(const H264Context *h, H264SliceContext *sl)
void ff_h264_flush_change(H264Context *h)
void ff_h264_filter_mb_fast(const H264Context *h, H264SliceContext *sl, int mb_x, int mb_y, uint8_t *img_y, uint8_t *img_cb, uint8_t *img_cr, unsigned int linesize, unsigned int uvlinesize)
void ff_h264_set_erpic(ERPicture *dst, H264Picture *src)
int ff_h264_ref_picture(H264Context *h, H264Picture *dst, H264Picture *src)
H.264 / AVC / MPEG-4 prediction functions.
const uint8_t * avpriv_find_start_code(const uint8_t *p, const uint8_t *end, uint32_t *state)
common internal API header
#define USES_LIST(a, list)
useful rectangle filling function
A reference to a data buffer.
Describe the class of an AVClass context structure.
main external API structure.
This structure describes decoded (raw) audio or video data.
Decoded Picture Buffer (DPB).
H264Picture * delayed_pic[MAX_DELAYED_PIC_COUNT+2]
H264ChromaContext h264chroma
int short_ref_count
number of actual short term references
uint8_t field_scan_q0[16]
MMCO mmco[MAX_MMCO_COUNT]
memory management control operations buffer.
H264Picture * long_ref[32]
uint8_t * list_counts
Array of list_count per MB specifying the slice type.
uint8_t zigzag_scan8x8_cavlc_q0[64]
int pixel_shift
0 for 8-bit H.264, 1 for high-bit-depth H.264
H264Picture last_pic_for_ec
H264Picture * next_output_pic
uint8_t field_scan8x8[64]
H264Picture * cur_pic_ptr
uint8_t(*[2] mvd_table)[2]
int ref2frm[MAX_SLICES][2][64]
reference to frame number lists, used in the loop filter, the first 2 are for -2,-1
uint8_t zigzag_scan8x8_q0[64]
uint16_t * slice_table
slice_table_base + 2*mb_stride + 1
AVBufferPool * mb_type_pool
int last_pocs[MAX_DELAYED_PIC_COUNT]
int cur_chroma_format_idc
int8_t * intra4x4_pred_mode
uint8_t * chroma_pred_mode_table
uint8_t field_scan8x8_cavlc[64]
uint16_t * slice_table_base
int frame_recovered
Initial frame has been completely recovered.
int long_ref_count
number of actual long term references
uint8_t(* non_zero_count)[48]
AVBufferPool * ref_index_pool
int recovery_frame
recovery_frame is the frame_num at which the next frame should be fully constructed.
int has_slice
slice NAL is found in the packet, set by decode_nal_units, its state does not need to be preserved ou...
int bit_depth_luma
luma bit depth from sps to detect changes
int current_slice
current slice number, used to initialize slice_num of each thread/context
int prev_interlaced_frame
Complement sei_pic_struct SEI_PIC_STRUCT_TOP_BOTTOM and SEI_PIC_STRUCT_BOTTOM_TOP indicate interlaced...
int chroma_format_idc
chroma format from sps to detect changes
int valid_recovery_point
Are the SEI recovery points looking valid.
uint8_t field_scan8x8_cavlc_q0[64]
uint8_t zigzag_scan8x8_cavlc[64]
int block_offset[2 *(16 *3)]
block_offset[ 0..23] for frame macroblocks block_offset[24..47] for field macroblocks
H264Picture * short_ref[32]
int is_avc
Used to parse AVC variant of H.264.
uint8_t field_scan8x8_q0[64]
uint8_t zigzag_scan8x8[64]
int nal_length_size
Number of bytes used for nal length (1, 2 or 4)
uint8_t zigzag_scan_q0[16]
AVBufferPool * qscale_table_pool
int16_t slice_row[MAX_SLICES]
to detect when MAX_SLICES is too low
AVBufferPool * motion_val_pool
H264SliceContext * slice_ctx
Context for storing H.264 DSP functions.
int ref_count[2][2]
number of entries in ref_poc (FIXME need per slice)
int recovered
picture at IDR or recovery point + recovery count
AVBufferRef * motion_val_buf[2]
int16_t(*[2] motion_val)[2]
int sei_recovery_frame_cnt
int field_picture
whether or not picture was encoded in separate fields
int pic_id
pic_num (short -> no wrap version of pic_num, pic_num & max_pic_num; long -> long_pic_num)
int frame_num
frame_num (raw frame_num from slice header)
int long_ref
1->long term reference 0->short term reference
int ref_poc[2][2][32]
POCs of the frames/fields used as reference (FIXME need per slice)
void * hwaccel_picture_private
hardware accelerator private data
AVBufferRef * mb_type_buf
AVBufferRef * ref_index_buf[2]
int mmco_reset
MMCO_RESET set this 1.
int field_poc[2]
top/bottom POC
AVBufferRef * hwaccel_priv_buf
int mbaff
1 -> MBAFF frame 0-> not MBAFF
AVBufferRef * qscale_table_buf
Context for storing H.264 prediction functions.
ptrdiff_t mb_linesize
may be equal to s->linesize or s->linesize * 2, for mbaff
int mb_field_decoding_flag
int8_t ref_cache[2][5 *8]
int dist_scale_factor_field[2][32]
unsigned int first_mb_addr
uint16_t sub_mb_type[4]
as a DCT coefficient is int32_t in high depth, we need to reserve twice the space.
int dist_scale_factor[32]
int bipred_scratchpad_allocated
struct H264SliceContext::@64 ref_modifications[2][32]
const uint8_t * intra_pcm_ptr
int16_t mv_cache[2][5 *8][2]
Motion vector cache.
int nb_ref_modifications[2]
uint8_t * edge_emu_buffer
struct H264Context * h264
int deblocking_filter
disable_deblocking_filter_idc with 1 <-> 0
int top_borders_allocated[2]
uint8_t * bipred_scratchpad
unsigned int left_samples_available
int map_col_to_list0[2][16+32]
int qp_thresh
QP threshold to skip loopfilter.
uint8_t mvd_cache[2][5 *8][2]
int slice_type_nos
S free slice type (SI/SP are remapped to I/P)
unsigned int top_samples_available
uint8_t(*[2] top_borders)[(16 *3) *2]
unsigned int topright_samples_available
H264Ref ref_list[2][48]
0..15: frame refs, 16..47: mbaff field refs.
int direct_spatial_mv_pred
int mb_mbaff
mb_aff_frame && mb_field_decoding_flag
uint8_t(*[2] mvd_table)[2]
const uint8_t * left_block
int16_t mb_padding[256 *2]
int map_col_to_list0_field[2][2][16+32]
int8_t * intra4x4_pred_mode
int edge_emu_buffer_allocated
uint8_t direct_cache[5 *8]
int8_t intra4x4_pred_mode_cache[5 *8]
unsigned int topleft_samples_available
int slice_alpha_c0_offset
uint8_t non_zero_count_cache[15 *8]
non zero coeff count cache.
uint8_t cabac_state[1024]
unsigned int ref_count[2]
num_ref_idx_l0/1_active_minus1 + 1
int16_t mb_luma_dc[3][16 *2]
as mb is addressed by scantable[i] and scantable is uint8_t we can either check that i is not too lar...
MMCO mmco[MAX_MMCO_COUNT]
int neighbor_transform_size
number of neighbors (top and/or left) that used 8x8 dct
Memory management control operation.
int long_arg
index, pic_num, or num long refs depending on opcode
int short_pic_num
pic_num without wrapping (pic_num & max_pic_num)
Core video DSP helper functions.