annotate h264.h @ 10893:2aafcafbe1f0 libavcodec

Replace cabac checks in inline functions from h264.h with constants. No benchmark because its just replacing variables with litteral constants (so no risk for slowdown outside gcc silliness) and i need sleep.
author michael
date Sat, 16 Jan 2010 05:41:33 +0000
parents 4701b9355d93
children f112b4d030fa
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
1 /*
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
2 * H.26L/H.264/AVC/JVT/14496-10/... encoder/decoder
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
3 * Copyright (c) 2003 Michael Niedermayer <michaelni@gmx.at>
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
4 *
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
5 * This file is part of FFmpeg.
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
6 *
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
7 * FFmpeg is free software; you can redistribute it and/or
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
8 * modify it under the terms of the GNU Lesser General Public
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
9 * License as published by the Free Software Foundation; either
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
10 * version 2.1 of the License, or (at your option) any later version.
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
11 *
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
12 * FFmpeg is distributed in the hope that it will be useful,
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
15 * Lesser General Public License for more details.
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
16 *
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
17 * You should have received a copy of the GNU Lesser General Public
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
18 * License along with FFmpeg; if not, write to the Free Software
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
20 */
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
21
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
22 /**
8718
e9d9d946f213 Use full internal pathname in doxygen @file directives.
diego
parents: 8596
diff changeset
23 * @file libavcodec/h264.h
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
24 * H.264 / AVC / MPEG4 part10 codec.
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
25 * @author Michael Niedermayer <michaelni@gmx.at>
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
26 */
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
27
7760
c4a4495715dd Globally rename the header inclusion guard names.
stefano
parents: 7526
diff changeset
28 #ifndef AVCODEC_H264_H
c4a4495715dd Globally rename the header inclusion guard names.
stefano
parents: 7526
diff changeset
29 #define AVCODEC_H264_H
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
30
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
31 #include "dsputil.h"
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
32 #include "cabac.h"
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
33 #include "mpegvideo.h"
5638
4a26dc4ca11d Move H.264 intra prediction functions into their own context
kostya
parents: 5231
diff changeset
34 #include "h264pred.h"
10866
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
35 #include "rectangle.h"
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
36
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
37 #define interlaced_dct interlaced_dct_is_a_bad_name
5129
0244bba24b43 misc typo fixes
diego
parents: 5127
diff changeset
38 #define mb_intra mb_intra_is_not_initialized_see_mb_type
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
39
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
40 #define LUMA_DC_BLOCK_INDEX 25
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
41 #define CHROMA_DC_BLOCK_INDEX 26
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
42
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
43 #define CHROMA_DC_COEFF_TOKEN_VLC_BITS 8
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
44 #define COEFF_TOKEN_VLC_BITS 8
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
45 #define TOTAL_ZEROS_VLC_BITS 9
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
46 #define CHROMA_DC_TOTAL_ZEROS_VLC_BITS 3
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
47 #define RUN_VLC_BITS 3
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
48 #define RUN7_VLC_BITS 6
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
49
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
50 #define MAX_SPS_COUNT 32
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
51 #define MAX_PPS_COUNT 256
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
52
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
53 #define MAX_MMCO_COUNT 66
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
54
7283
39e81fdd8d23 Move the size of the H264Context delayed_pic array into a #define.
astrange
parents: 7178
diff changeset
55 #define MAX_DELAYED_PIC_COUNT 16
39e81fdd8d23 Move the size of the H264Context delayed_pic array into a #define.
astrange
parents: 7178
diff changeset
56
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
57 /* Compiling in interlaced support reduces the speed
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
58 * of progressive decoding by about 2%. */
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
59 #define ALLOW_INTERLACE
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
60
7407
6288bc103b9f chroma_format_idc=0 aka grayscale support.
michael
parents: 7347
diff changeset
61 #define ALLOW_NOCHROMA
6288bc103b9f chroma_format_idc=0 aka grayscale support.
michael
parents: 7347
diff changeset
62
8079
2d3c7cd7d143 Replace literally hardcoded max slice number by named constant.
michael
parents: 8025
diff changeset
63 /**
2d3c7cd7d143 Replace literally hardcoded max slice number by named constant.
michael
parents: 8025
diff changeset
64 * The maximum number of slices supported by the decoder.
2d3c7cd7d143 Replace literally hardcoded max slice number by named constant.
michael
parents: 8025
diff changeset
65 * must be a power of 2
2d3c7cd7d143 Replace literally hardcoded max slice number by named constant.
michael
parents: 8025
diff changeset
66 */
2d3c7cd7d143 Replace literally hardcoded max slice number by named constant.
michael
parents: 8025
diff changeset
67 #define MAX_SLICES 16
2d3c7cd7d143 Replace literally hardcoded max slice number by named constant.
michael
parents: 8025
diff changeset
68
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
69 #ifdef ALLOW_INTERLACE
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
70 #define MB_MBAFF h->mb_mbaff
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
71 #define MB_FIELD h->mb_field_decoding_flag
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
72 #define FRAME_MBAFF h->mb_aff_frame
5801
0be16259b7d8 Enable PAFF decoding
cehoyos
parents: 5781
diff changeset
73 #define FIELD_PICTURE (s->picture_structure != PICT_FRAME)
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
74 #else
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
75 #define MB_MBAFF 0
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
76 #define MB_FIELD 0
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
77 #define FRAME_MBAFF 0
5767
32b404ec4c19 Partial PAFF implementation at macroblock level.
andoma
parents: 5756
diff changeset
78 #define FIELD_PICTURE 0
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
79 #undef IS_INTERLACED
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
80 #define IS_INTERLACED(mb_type) 0
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
81 #endif
5781
0b3aa6f4c313 Modifies macroblock addressing and current macroblock y-position for field decoding.
andoma
parents: 5772
diff changeset
82 #define FIELD_OR_MBAFF_PICTURE (FRAME_MBAFF || FIELD_PICTURE)
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
83
7407
6288bc103b9f chroma_format_idc=0 aka grayscale support.
michael
parents: 7347
diff changeset
84 #ifdef ALLOW_NOCHROMA
6288bc103b9f chroma_format_idc=0 aka grayscale support.
michael
parents: 7347
diff changeset
85 #define CHROMA h->sps.chroma_format_idc
6288bc103b9f chroma_format_idc=0 aka grayscale support.
michael
parents: 7347
diff changeset
86 #else
6288bc103b9f chroma_format_idc=0 aka grayscale support.
michael
parents: 7347
diff changeset
87 #define CHROMA 1
6288bc103b9f chroma_format_idc=0 aka grayscale support.
michael
parents: 7347
diff changeset
88 #endif
6288bc103b9f chroma_format_idc=0 aka grayscale support.
michael
parents: 7347
diff changeset
89
10893
2aafcafbe1f0 Replace cabac checks in inline functions from h264.h with constants.
michael
parents: 10883
diff changeset
90 #ifndef CABAC
2aafcafbe1f0 Replace cabac checks in inline functions from h264.h with constants.
michael
parents: 10883
diff changeset
91 #define CABAC h->pps.cabac
2aafcafbe1f0 Replace cabac checks in inline functions from h264.h with constants.
michael
parents: 10883
diff changeset
92 #endif
2aafcafbe1f0 Replace cabac checks in inline functions from h264.h with constants.
michael
parents: 10883
diff changeset
93
7988
a7dfe657968d Move nal unit types enum and EXTENDED_SAR #define from h264data.h to h264.h.
michael
parents: 7906
diff changeset
94 #define EXTENDED_SAR 255
a7dfe657968d Move nal unit types enum and EXTENDED_SAR #define from h264data.h to h264.h.
michael
parents: 7906
diff changeset
95
8207
2f35f9781c31 Move #defines that are mostly used in h264.c out of h264data.h and into h264.h.
diego
parents: 8107
diff changeset
96 #define MB_TYPE_REF0 MB_TYPE_ACPRED //dirty but it fits in 16 bit
2f35f9781c31 Move #defines that are mostly used in h264.c out of h264data.h and into h264.h.
diego
parents: 8107
diff changeset
97 #define MB_TYPE_8x8DCT 0x01000000
2f35f9781c31 Move #defines that are mostly used in h264.c out of h264data.h and into h264.h.
diego
parents: 8107
diff changeset
98 #define IS_REF0(a) ((a) & MB_TYPE_REF0)
2f35f9781c31 Move #defines that are mostly used in h264.c out of h264data.h and into h264.h.
diego
parents: 8107
diff changeset
99 #define IS_8x8DCT(a) ((a) & MB_TYPE_8x8DCT)
2f35f9781c31 Move #defines that are mostly used in h264.c out of h264data.h and into h264.h.
diego
parents: 8107
diff changeset
100
10862
d9c084a0c22b Split all the reference picture handling off h264.c.
michael
parents: 10859
diff changeset
101 /**
d9c084a0c22b Split all the reference picture handling off h264.c.
michael
parents: 10859
diff changeset
102 * Value of Picture.reference when Picture is not a reference picture, but
d9c084a0c22b Split all the reference picture handling off h264.c.
michael
parents: 10859
diff changeset
103 * is held for delayed output.
d9c084a0c22b Split all the reference picture handling off h264.c.
michael
parents: 10859
diff changeset
104 */
d9c084a0c22b Split all the reference picture handling off h264.c.
michael
parents: 10859
diff changeset
105 #define DELAYED_PIC_REF 4
d9c084a0c22b Split all the reference picture handling off h264.c.
michael
parents: 10859
diff changeset
106
d9c084a0c22b Split all the reference picture handling off h264.c.
michael
parents: 10859
diff changeset
107
7988
a7dfe657968d Move nal unit types enum and EXTENDED_SAR #define from h264data.h to h264.h.
michael
parents: 7906
diff changeset
108 /* NAL unit types */
a7dfe657968d Move nal unit types enum and EXTENDED_SAR #define from h264data.h to h264.h.
michael
parents: 7906
diff changeset
109 enum {
7990
58955234a2bd Fix indention of NAL type enum.
michael
parents: 7988
diff changeset
110 NAL_SLICE=1,
58955234a2bd Fix indention of NAL type enum.
michael
parents: 7988
diff changeset
111 NAL_DPA,
58955234a2bd Fix indention of NAL type enum.
michael
parents: 7988
diff changeset
112 NAL_DPB,
58955234a2bd Fix indention of NAL type enum.
michael
parents: 7988
diff changeset
113 NAL_DPC,
58955234a2bd Fix indention of NAL type enum.
michael
parents: 7988
diff changeset
114 NAL_IDR_SLICE,
58955234a2bd Fix indention of NAL type enum.
michael
parents: 7988
diff changeset
115 NAL_SEI,
58955234a2bd Fix indention of NAL type enum.
michael
parents: 7988
diff changeset
116 NAL_SPS,
58955234a2bd Fix indention of NAL type enum.
michael
parents: 7988
diff changeset
117 NAL_PPS,
58955234a2bd Fix indention of NAL type enum.
michael
parents: 7988
diff changeset
118 NAL_AUD,
58955234a2bd Fix indention of NAL type enum.
michael
parents: 7988
diff changeset
119 NAL_END_SEQUENCE,
58955234a2bd Fix indention of NAL type enum.
michael
parents: 7988
diff changeset
120 NAL_END_STREAM,
58955234a2bd Fix indention of NAL type enum.
michael
parents: 7988
diff changeset
121 NAL_FILLER_DATA,
58955234a2bd Fix indention of NAL type enum.
michael
parents: 7988
diff changeset
122 NAL_SPS_EXT,
58955234a2bd Fix indention of NAL type enum.
michael
parents: 7988
diff changeset
123 NAL_AUXILIARY_SLICE=19
7988
a7dfe657968d Move nal unit types enum and EXTENDED_SAR #define from h264data.h to h264.h.
michael
parents: 7906
diff changeset
124 };
a7dfe657968d Move nal unit types enum and EXTENDED_SAR #define from h264data.h to h264.h.
michael
parents: 7906
diff changeset
125
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
126 /**
8762
4f1567ce75c4 Replace hard-coded SEI type constants with symbolic names
superdump
parents: 8740
diff changeset
127 * SEI message types
4f1567ce75c4 Replace hard-coded SEI type constants with symbolic names
superdump
parents: 8740
diff changeset
128 */
4f1567ce75c4 Replace hard-coded SEI type constants with symbolic names
superdump
parents: 8740
diff changeset
129 typedef enum {
8972
eaf39f9b25df Parse buffering period (H.264, D.1.1).
cehoyos
parents: 8970
diff changeset
130 SEI_BUFFERING_PERIOD = 0, ///< buffering period (H.264, D.1.1)
8762
4f1567ce75c4 Replace hard-coded SEI type constants with symbolic names
superdump
parents: 8740
diff changeset
131 SEI_TYPE_PIC_TIMING = 1, ///< picture timing
4f1567ce75c4 Replace hard-coded SEI type constants with symbolic names
superdump
parents: 8740
diff changeset
132 SEI_TYPE_USER_DATA_UNREGISTERED = 5, ///< unregistered user data
4f1567ce75c4 Replace hard-coded SEI type constants with symbolic names
superdump
parents: 8740
diff changeset
133 SEI_TYPE_RECOVERY_POINT = 6 ///< recovery point (frame # to decoder sync)
4f1567ce75c4 Replace hard-coded SEI type constants with symbolic names
superdump
parents: 8740
diff changeset
134 } SEI_Type;
4f1567ce75c4 Replace hard-coded SEI type constants with symbolic names
superdump
parents: 8740
diff changeset
135
4f1567ce75c4 Replace hard-coded SEI type constants with symbolic names
superdump
parents: 8740
diff changeset
136 /**
8107
e61f76efc9f3 h264: Implement decoding of picture timing SEI message.
andoma
parents: 8079
diff changeset
137 * pic_struct in picture timing SEI message
e61f76efc9f3 h264: Implement decoding of picture timing SEI message.
andoma
parents: 8079
diff changeset
138 */
e61f76efc9f3 h264: Implement decoding of picture timing SEI message.
andoma
parents: 8079
diff changeset
139 typedef enum {
e61f76efc9f3 h264: Implement decoding of picture timing SEI message.
andoma
parents: 8079
diff changeset
140 SEI_PIC_STRUCT_FRAME = 0, ///< 0: %frame
e61f76efc9f3 h264: Implement decoding of picture timing SEI message.
andoma
parents: 8079
diff changeset
141 SEI_PIC_STRUCT_TOP_FIELD = 1, ///< 1: top field
e61f76efc9f3 h264: Implement decoding of picture timing SEI message.
andoma
parents: 8079
diff changeset
142 SEI_PIC_STRUCT_BOTTOM_FIELD = 2, ///< 2: bottom field
e61f76efc9f3 h264: Implement decoding of picture timing SEI message.
andoma
parents: 8079
diff changeset
143 SEI_PIC_STRUCT_TOP_BOTTOM = 3, ///< 3: top field, bottom field, in that order
e61f76efc9f3 h264: Implement decoding of picture timing SEI message.
andoma
parents: 8079
diff changeset
144 SEI_PIC_STRUCT_BOTTOM_TOP = 4, ///< 4: bottom field, top field, in that order
e61f76efc9f3 h264: Implement decoding of picture timing SEI message.
andoma
parents: 8079
diff changeset
145 SEI_PIC_STRUCT_TOP_BOTTOM_TOP = 5, ///< 5: top field, bottom field, top field repeated, in that order
e61f76efc9f3 h264: Implement decoding of picture timing SEI message.
andoma
parents: 8079
diff changeset
146 SEI_PIC_STRUCT_BOTTOM_TOP_BOTTOM = 6, ///< 6: bottom field, top field, bottom field repeated, in that order
e61f76efc9f3 h264: Implement decoding of picture timing SEI message.
andoma
parents: 8079
diff changeset
147 SEI_PIC_STRUCT_FRAME_DOUBLING = 7, ///< 7: %frame doubling
e61f76efc9f3 h264: Implement decoding of picture timing SEI message.
andoma
parents: 8079
diff changeset
148 SEI_PIC_STRUCT_FRAME_TRIPLING = 8 ///< 8: %frame tripling
e61f76efc9f3 h264: Implement decoding of picture timing SEI message.
andoma
parents: 8079
diff changeset
149 } SEI_PicStructType;
e61f76efc9f3 h264: Implement decoding of picture timing SEI message.
andoma
parents: 8079
diff changeset
150
e61f76efc9f3 h264: Implement decoding of picture timing SEI message.
andoma
parents: 8079
diff changeset
151 /**
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
152 * Sequence parameter set
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
153 */
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
154 typedef struct SPS{
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
155
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
156 int profile_idc;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
157 int level_idc;
7407
6288bc103b9f chroma_format_idc=0 aka grayscale support.
michael
parents: 7347
diff changeset
158 int chroma_format_idc;
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
159 int transform_bypass; ///< qpprime_y_zero_transform_bypass_flag
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
160 int log2_max_frame_num; ///< log2_max_frame_num_minus4 + 4
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
161 int poc_type; ///< pic_order_cnt_type
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
162 int log2_max_poc_lsb; ///< log2_max_pic_order_cnt_lsb_minus4
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
163 int delta_pic_order_always_zero_flag;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
164 int offset_for_non_ref_pic;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
165 int offset_for_top_to_bottom_field;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
166 int poc_cycle_length; ///< num_ref_frames_in_pic_order_cnt_cycle
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
167 int ref_frame_count; ///< num_ref_frames
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
168 int gaps_in_frame_num_allowed_flag;
5755
b45894d869da Cosmetic preparations for h264/PAFF implementation.
andoma
parents: 5642
diff changeset
169 int mb_width; ///< pic_width_in_mbs_minus1 + 1
b45894d869da Cosmetic preparations for h264/PAFF implementation.
andoma
parents: 5642
diff changeset
170 int mb_height; ///< pic_height_in_map_units_minus1 + 1
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
171 int frame_mbs_only_flag;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
172 int mb_aff; ///<mb_adaptive_frame_field_flag
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
173 int direct_8x8_inference_flag;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
174 int crop; ///< frame_cropping_flag
6582
7d40da88f0c7 Crop parameters are unsigned, having them negative could be bad and lead
michael
parents: 6488
diff changeset
175 unsigned int crop_left; ///< frame_cropping_rect_left_offset
7d40da88f0c7 Crop parameters are unsigned, having them negative could be bad and lead
michael
parents: 6488
diff changeset
176 unsigned int crop_right; ///< frame_cropping_rect_right_offset
7d40da88f0c7 Crop parameters are unsigned, having them negative could be bad and lead
michael
parents: 6488
diff changeset
177 unsigned int crop_top; ///< frame_cropping_rect_top_offset
7d40da88f0c7 Crop parameters are unsigned, having them negative could be bad and lead
michael
parents: 6488
diff changeset
178 unsigned int crop_bottom; ///< frame_cropping_rect_bottom_offset
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
179 int vui_parameters_present_flag;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
180 AVRational sar;
10837
e5905bfa625d Export fullrange flag and color information for h.264
conrad
parents: 10829
diff changeset
181 int video_signal_type_present_flag;
e5905bfa625d Export fullrange flag and color information for h.264
conrad
parents: 10829
diff changeset
182 int full_range;
e5905bfa625d Export fullrange flag and color information for h.264
conrad
parents: 10829
diff changeset
183 int colour_description_present_flag;
e5905bfa625d Export fullrange flag and color information for h.264
conrad
parents: 10829
diff changeset
184 enum AVColorPrimaries color_primaries;
e5905bfa625d Export fullrange flag and color information for h.264
conrad
parents: 10829
diff changeset
185 enum AVColorTransferCharacteristic color_trc;
e5905bfa625d Export fullrange flag and color information for h.264
conrad
parents: 10829
diff changeset
186 enum AVColorSpace colorspace;
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
187 int timing_info_present_flag;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
188 uint32_t num_units_in_tick;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
189 uint32_t time_scale;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
190 int fixed_frame_rate_flag;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
191 short offset_for_ref_frame[256]; //FIXME dyn aloc?
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
192 int bitstream_restriction_flag;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
193 int num_reorder_frames;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
194 int scaling_matrix_present;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
195 uint8_t scaling_matrix4[6][16];
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
196 uint8_t scaling_matrix8[2][64];
8107
e61f76efc9f3 h264: Implement decoding of picture timing SEI message.
andoma
parents: 8079
diff changeset
197 int nal_hrd_parameters_present_flag;
e61f76efc9f3 h264: Implement decoding of picture timing SEI message.
andoma
parents: 8079
diff changeset
198 int vcl_hrd_parameters_present_flag;
e61f76efc9f3 h264: Implement decoding of picture timing SEI message.
andoma
parents: 8079
diff changeset
199 int pic_struct_present_flag;
e61f76efc9f3 h264: Implement decoding of picture timing SEI message.
andoma
parents: 8079
diff changeset
200 int time_offset_length;
8970
9757b2f87d83 Store CPB count in the context.
cehoyos
parents: 8967
diff changeset
201 int cpb_cnt; ///< See H.264 E.1.2
8965
a5dce344904a Parse initial_cpb_removal_delay_length.
cehoyos
parents: 8783
diff changeset
202 int initial_cpb_removal_delay_length; ///< initial_cpb_removal_delay_length_minus1 +1
8107
e61f76efc9f3 h264: Implement decoding of picture timing SEI message.
andoma
parents: 8079
diff changeset
203 int cpb_removal_delay_length; ///< cpb_removal_delay_length_minus1 + 1
e61f76efc9f3 h264: Implement decoding of picture timing SEI message.
andoma
parents: 8079
diff changeset
204 int dpb_output_delay_length; ///< dpb_output_delay_length_minus1 + 1
8735
5d7ebbb7e91b Add fields to H264Context and SPS for upcoming VA API support.
cehoyos
parents: 8718
diff changeset
205 int bit_depth_luma; ///< bit_depth_luma_minus8 + 8
5d7ebbb7e91b Add fields to H264Context and SPS for upcoming VA API support.
cehoyos
parents: 8718
diff changeset
206 int bit_depth_chroma; ///< bit_depth_chroma_minus8 + 8
5d7ebbb7e91b Add fields to H264Context and SPS for upcoming VA API support.
cehoyos
parents: 8718
diff changeset
207 int residual_color_transform_flag; ///< residual_colour_transform_flag
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
208 }SPS;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
209
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
210 /**
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
211 * Picture parameter set
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
212 */
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
213 typedef struct PPS{
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
214 unsigned int sps_id;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
215 int cabac; ///< entropy_coding_mode_flag
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
216 int pic_order_present; ///< pic_order_present_flag
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
217 int slice_group_count; ///< num_slice_groups_minus1 + 1
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
218 int mb_slice_group_map_type;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
219 unsigned int ref_count[2]; ///< num_ref_idx_l0/1_active_minus1 + 1
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
220 int weighted_pred; ///< weighted_pred_flag
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
221 int weighted_bipred_idc;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
222 int init_qp; ///< pic_init_qp_minus26 + 26
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
223 int init_qs; ///< pic_init_qs_minus26 + 26
5231
07a97575d0c4 Add support for streams with different chroma_qp_index_offset
gpoirier
parents: 5226
diff changeset
224 int chroma_qp_index_offset[2];
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
225 int deblocking_filter_parameters_present; ///< deblocking_filter_parameters_present_flag
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
226 int constrained_intra_pred; ///< constrained_intra_pred_flag
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
227 int redundant_pic_cnt_present; ///< redundant_pic_cnt_present_flag
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
228 int transform_8x8_mode; ///< transform_8x8_mode_flag
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
229 uint8_t scaling_matrix4[6][16];
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
230 uint8_t scaling_matrix8[2][64];
7347
612a78c3b128 qscale has a range of 0..51 we thus do not need a 256 entry table and neither need
michael
parents: 7343
diff changeset
231 uint8_t chroma_qp_table[2][64]; ///< pre-scaled (with chroma_qp_index_offset) version of qp_table
5231
07a97575d0c4 Add support for streams with different chroma_qp_index_offset
gpoirier
parents: 5226
diff changeset
232 int chroma_qp_diff;
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
233 }PPS;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
234
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
235 /**
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
236 * Memory management control operation opcode.
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
237 */
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
238 typedef enum MMCOOpcode{
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
239 MMCO_END=0,
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
240 MMCO_SHORT2UNUSED,
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
241 MMCO_LONG2UNUSED,
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
242 MMCO_SHORT2LONG,
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
243 MMCO_SET_MAX_LONG,
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
244 MMCO_RESET,
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
245 MMCO_LONG,
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
246 } MMCOOpcode;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
247
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
248 /**
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
249 * Memory management control operation.
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
250 */
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
251 typedef struct MMCO{
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
252 MMCOOpcode opcode;
5756
db5a041fd77c Rename MMCO stuff to prepare for h264/PAFF implementation.
andoma
parents: 5755
diff changeset
253 int short_pic_num; ///< pic_num without wrapping (pic_num & max_pic_num)
db5a041fd77c Rename MMCO stuff to prepare for h264/PAFF implementation.
andoma
parents: 5755
diff changeset
254 int long_arg; ///< index, pic_num, or num long refs depending on opcode
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
255 } MMCO;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
256
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
257 /**
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
258 * H264Context
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
259 */
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
260 typedef struct H264Context{
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
261 MpegEncContext s;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
262 int nal_ref_idc;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
263 int nal_unit_type;
5174
2e038003cdfb Do not reuse the rbsp de-escape buffer if both
benoit
parents: 5129
diff changeset
264 uint8_t *rbsp_buffer[2];
2e038003cdfb Do not reuse the rbsp de-escape buffer if both
benoit
parents: 5129
diff changeset
265 unsigned int rbsp_buffer_size[2];
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
266
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
267 /**
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
268 * Used to parse AVC variant of h264
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
269 */
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
270 int is_avc; ///< this flag is != 0 if codec is avc1
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
271 int got_avcC; ///< flag used to parse avcC data only once
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
272 int nal_length_size; ///< Number of bytes used for nal length (1, 2 or 4)
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
273
5231
07a97575d0c4 Add support for streams with different chroma_qp_index_offset
gpoirier
parents: 5226
diff changeset
274 int chroma_qp[2]; //QPc
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
275
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
276 int prev_mb_skipped;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
277 int next_mb_skipped;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
278
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
279 //prediction stuff
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
280 int chroma_pred_mode;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
281 int intra16x16_pred_mode;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
282
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
283 int top_mb_xy;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
284 int left_mb_xy[2];
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
285
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
286 int8_t intra4x4_pred_mode_cache[5*8];
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
287 int8_t (*intra4x4_pred_mode)[8];
5638
4a26dc4ca11d Move H.264 intra prediction functions into their own context
kostya
parents: 5231
diff changeset
288 H264PredContext hpc;
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
289 unsigned int topleft_samples_available;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
290 unsigned int top_samples_available;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
291 unsigned int topright_samples_available;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
292 unsigned int left_samples_available;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
293 uint8_t (*top_borders[2])[16+2*8];
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
294 uint8_t left_border[2*(17+2*9)];
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
295
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
296 /**
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
297 * non zero coeff count cache.
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
298 * is 64 if not available.
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
299 */
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
300 DECLARE_ALIGNED_8(uint8_t, non_zero_count_cache[6*8]);
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
301 uint8_t (*non_zero_count)[16];
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
302
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
303 /**
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
304 * Motion vector cache.
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
305 */
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
306 DECLARE_ALIGNED_8(int16_t, mv_cache[2][5*8][2]);
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
307 DECLARE_ALIGNED_8(int8_t, ref_cache[2][5*8]);
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
308 #define LIST_NOT_USED -1 //FIXME rename?
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
309 #define PART_NOT_AVAILABLE -2
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
310
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
311 /**
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
312 * is 1 if the specific list MV&references are set to 0,0,-2.
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
313 */
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
314 int mv_cache_clean[2];
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
315
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
316 /**
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
317 * number of neighbors (top and/or left) that used 8x8 dct
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
318 */
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
319 int neighbor_transform_size;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
320
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
321 /**
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
322 * block_offset[ 0..23] for frame macroblocks
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
323 * block_offset[24..47] for field macroblocks
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
324 */
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
325 int block_offset[2*(16+8)];
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
326
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
327 uint32_t *mb2b_xy; //FIXME are these 4 a good idea?
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
328 uint32_t *mb2b8_xy;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
329 int b_stride; //FIXME use s->b4_stride
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
330 int b8_stride;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
331
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
332 int mb_linesize; ///< may be equal to s->linesize or s->linesize*2, for mbaff
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
333 int mb_uvlinesize;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
334
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
335 int emu_edge_width;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
336 int emu_edge_height;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
337
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
338 int halfpel_flag;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
339 int thirdpel_flag;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
340
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
341 int unknown_svq3_flag;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
342 int next_slice_index;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
343
5079
4ff805f87391 allocate PPS and SPS dynamically
benoit
parents: 4975
diff changeset
344 SPS *sps_buffers[MAX_SPS_COUNT];
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
345 SPS sps; ///< current sps
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
346
5079
4ff805f87391 allocate PPS and SPS dynamically
benoit
parents: 4975
diff changeset
347 PPS *pps_buffers[MAX_PPS_COUNT];
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
348 /**
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
349 * current pps
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
350 */
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
351 PPS pps; //FIXME move to Picture perhaps? (->no) do we need that?
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
352
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
353 uint32_t dequant4_buffer[6][52][16];
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
354 uint32_t dequant8_buffer[2][52][64];
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
355 uint32_t (*dequant4_coeff[6])[16];
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
356 uint32_t (*dequant8_coeff[2])[64];
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
357 int dequant_coeff_pps; ///< reinit tables when pps changes
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
358
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
359 int slice_num;
8079
2d3c7cd7d143 Replace literally hardcoded max slice number by named constant.
michael
parents: 8025
diff changeset
360 uint16_t *slice_table_base;
2d3c7cd7d143 Replace literally hardcoded max slice number by named constant.
michael
parents: 8025
diff changeset
361 uint16_t *slice_table; ///< slice_table_base + 2*mb_stride + 1
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
362 int slice_type;
7338
b088035801d5 Simplify slice type checks.
michael
parents: 7332
diff changeset
363 int slice_type_nos; ///< S free slice type (SI/SP are remapped to I/P)
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
364 int slice_type_fixed;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
365
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
366 //interlacing specific flags
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
367 int mb_aff_frame;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
368 int mb_field_decoding_flag;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
369 int mb_mbaff; ///< mb_aff_frame && mb_field_decoding_flag
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
370
9071
24d0633a5248 sub_mb_type[] needs to be 8-byte aligned because it is referenced
stefang
parents: 8996
diff changeset
371 DECLARE_ALIGNED_8(uint16_t, sub_mb_type[4]);
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
372
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
373 //POC stuff
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
374 int poc_lsb;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
375 int poc_msb;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
376 int delta_poc_bottom;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
377 int delta_poc[2];
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
378 int frame_num;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
379 int prev_poc_msb; ///< poc_msb of the last reference pic for POC type 0
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
380 int prev_poc_lsb; ///< poc_lsb of the last reference pic for POC type 0
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
381 int frame_num_offset; ///< for POC type 2
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
382 int prev_frame_num_offset; ///< for POC type 2
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
383 int prev_frame_num; ///< frame_num of the last pic for POC type 1/2
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
384
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
385 /**
5772
65b71bd21a4d Fix h->curr_pic_num for field pictures. Necessary for proper PAFF support.
andoma
parents: 5767
diff changeset
386 * frame_num for frames or 2*frame_num+1 for field pics.
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
387 */
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
388 int curr_pic_num;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
389
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
390 /**
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
391 * max_frame_num or 2*max_frame_num for field pics.
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
392 */
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
393 int max_pic_num;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
394
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
395 //Weighted pred stuff
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
396 int use_weight;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
397 int use_weight_chroma;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
398 int luma_log2_weight_denom;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
399 int chroma_log2_weight_denom;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
400 int luma_weight[2][48];
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
401 int luma_offset[2][48];
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
402 int chroma_weight[2][48][2];
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
403 int chroma_offset[2][48][2];
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
404 int implicit_weight[48][48];
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
405
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
406 //deblock
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
407 int deblocking_filter; ///< disable_deblocking_filter_idc with 1<->0
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
408 int slice_alpha_c0_offset;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
409 int slice_beta_offset;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
410
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
411 int redundant_pic_count;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
412
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
413 int direct_spatial_mv_pred;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
414 int dist_scale_factor[16];
7898
a33287a39a55 Make MBAFF temporal direct mode closer to the spec.
michael
parents: 7760
diff changeset
415 int dist_scale_factor_field[2][32];
7906
5be944626072 Another try to fix temporal direct mode references.
michael
parents: 7898
diff changeset
416 int map_col_to_list0[2][16+32];
5be944626072 Another try to fix temporal direct mode references.
michael
parents: 7898
diff changeset
417 int map_col_to_list0_field[2][2][16+32];
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
418
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
419 /**
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
420 * num_ref_idx_l0/1_active_minus1 + 1
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
421 */
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
422 unsigned int ref_count[2]; ///< counts frames or fields, depending on current mb mode
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
423 unsigned int list_count;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
424 Picture *short_ref[32];
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
425 Picture *long_ref[32];
5755
b45894d869da Cosmetic preparations for h264/PAFF implementation.
andoma
parents: 5642
diff changeset
426 Picture default_ref_list[2][32]; ///< base reference list for all slices of a coded picture
b45894d869da Cosmetic preparations for h264/PAFF implementation.
andoma
parents: 5642
diff changeset
427 Picture ref_list[2][48]; /**< 0..15: frame refs, 16..47: mbaff field refs.
b45894d869da Cosmetic preparations for h264/PAFF implementation.
andoma
parents: 5642
diff changeset
428 Reordered version of default_ref_list
b45894d869da Cosmetic preparations for h264/PAFF implementation.
andoma
parents: 5642
diff changeset
429 according to picture reordering in slice header */
8079
2d3c7cd7d143 Replace literally hardcoded max slice number by named constant.
michael
parents: 8025
diff changeset
430 int ref2frm[MAX_SLICES][2][64]; ///< reference to frame number lists, used in the loop filter, the first 2 are for -2,-1
7283
39e81fdd8d23 Move the size of the H264Context delayed_pic array into a #define.
astrange
parents: 7178
diff changeset
431 Picture *delayed_pic[MAX_DELAYED_PIC_COUNT+2]; //FIXME size?
7310
033d10927d45 Remove delayed_output_pic, I do not understand what this variable was good for.
michael
parents: 7283
diff changeset
432 int outputed_poc;
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
433
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
434 /**
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
435 * memory management control operations buffer.
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
436 */
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
437 MMCO mmco[MAX_MMCO_COUNT];
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
438 int mmco_index;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
439
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
440 int long_ref_count; ///< number of actual long term references
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
441 int short_ref_count; ///< number of actual short term references
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
442
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
443 //data partitioning
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
444 GetBitContext intra_gb;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
445 GetBitContext inter_gb;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
446 GetBitContext *intra_gb_ptr;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
447 GetBitContext *inter_gb_ptr;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
448
6320
ffb2a7b80d6d ff_h264_idct8_add_sse2.
lorenm
parents: 5830
diff changeset
449 DECLARE_ALIGNED_16(DCTELEM, mb[16*24]);
6488
ecbe9565dcae typo fixes
diego
parents: 6320
diff changeset
450 DCTELEM mb_padding[256]; ///< as mb is addressed by scantable[i] and scantable is uint8_t we can either check that i is not too large or ensure that there is some unused stuff after mb
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
451
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
452 /**
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
453 * Cabac
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
454 */
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
455 CABACContext cabac;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
456 uint8_t cabac_state[460];
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
457 int cabac_init_idc;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
458
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
459 /* 0x100 -> non null luma_dc, 0x80/0x40 -> non null chroma_dc (cb/cr), 0x?0 -> chroma_cbp(0,1,2), 0x0? luma_cbp */
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
460 uint16_t *cbp_table;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
461 int cbp;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
462 int top_cbp;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
463 int left_cbp;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
464 /* chroma_pred_mode for i4x4 or i16x16, else 0 */
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
465 uint8_t *chroma_pred_mode_table;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
466 int last_qscale_diff;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
467 int16_t (*mvd_table[2])[2];
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
468 DECLARE_ALIGNED_8(int16_t, mvd_cache[2][5*8][2]);
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
469 uint8_t *direct_table;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
470 uint8_t direct_cache[5*8];
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
471
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
472 uint8_t zigzag_scan[16];
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
473 uint8_t zigzag_scan8x8[64];
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
474 uint8_t zigzag_scan8x8_cavlc[64];
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
475 uint8_t field_scan[16];
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
476 uint8_t field_scan8x8[64];
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
477 uint8_t field_scan8x8_cavlc[64];
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
478 const uint8_t *zigzag_scan_q0;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
479 const uint8_t *zigzag_scan8x8_q0;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
480 const uint8_t *zigzag_scan8x8_cavlc_q0;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
481 const uint8_t *field_scan_q0;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
482 const uint8_t *field_scan8x8_q0;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
483 const uint8_t *field_scan8x8_cavlc_q0;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
484
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
485 int x264_build;
5642
d2598034f2a9 Add slice-based parallel H.264 decoding
gpoirier
parents: 5638
diff changeset
486
d2598034f2a9 Add slice-based parallel H.264 decoding
gpoirier
parents: 5638
diff changeset
487 /**
d2598034f2a9 Add slice-based parallel H.264 decoding
gpoirier
parents: 5638
diff changeset
488 * @defgroup multithreading Members for slice based multithreading
d2598034f2a9 Add slice-based parallel H.264 decoding
gpoirier
parents: 5638
diff changeset
489 * @{
d2598034f2a9 Add slice-based parallel H.264 decoding
gpoirier
parents: 5638
diff changeset
490 */
d2598034f2a9 Add slice-based parallel H.264 decoding
gpoirier
parents: 5638
diff changeset
491 struct H264Context *thread_context[MAX_THREADS];
d2598034f2a9 Add slice-based parallel H.264 decoding
gpoirier
parents: 5638
diff changeset
492
d2598034f2a9 Add slice-based parallel H.264 decoding
gpoirier
parents: 5638
diff changeset
493 /**
d2598034f2a9 Add slice-based parallel H.264 decoding
gpoirier
parents: 5638
diff changeset
494 * current slice number, used to initalize slice_num of each thread/context
d2598034f2a9 Add slice-based parallel H.264 decoding
gpoirier
parents: 5638
diff changeset
495 */
d2598034f2a9 Add slice-based parallel H.264 decoding
gpoirier
parents: 5638
diff changeset
496 int current_slice;
d2598034f2a9 Add slice-based parallel H.264 decoding
gpoirier
parents: 5638
diff changeset
497
d2598034f2a9 Add slice-based parallel H.264 decoding
gpoirier
parents: 5638
diff changeset
498 /**
d2598034f2a9 Add slice-based parallel H.264 decoding
gpoirier
parents: 5638
diff changeset
499 * Max number of threads / contexts.
d2598034f2a9 Add slice-based parallel H.264 decoding
gpoirier
parents: 5638
diff changeset
500 * This is equal to AVCodecContext.thread_count unless
d2598034f2a9 Add slice-based parallel H.264 decoding
gpoirier
parents: 5638
diff changeset
501 * multithreaded decoding is impossible, in which case it is
d2598034f2a9 Add slice-based parallel H.264 decoding
gpoirier
parents: 5638
diff changeset
502 * reduced to 1.
d2598034f2a9 Add slice-based parallel H.264 decoding
gpoirier
parents: 5638
diff changeset
503 */
d2598034f2a9 Add slice-based parallel H.264 decoding
gpoirier
parents: 5638
diff changeset
504 int max_contexts;
d2598034f2a9 Add slice-based parallel H.264 decoding
gpoirier
parents: 5638
diff changeset
505
d2598034f2a9 Add slice-based parallel H.264 decoding
gpoirier
parents: 5638
diff changeset
506 /**
d2598034f2a9 Add slice-based parallel H.264 decoding
gpoirier
parents: 5638
diff changeset
507 * 1 if the single thread fallback warning has already been
d2598034f2a9 Add slice-based parallel H.264 decoding
gpoirier
parents: 5638
diff changeset
508 * displayed, 0 otherwise.
d2598034f2a9 Add slice-based parallel H.264 decoding
gpoirier
parents: 5638
diff changeset
509 */
d2598034f2a9 Add slice-based parallel H.264 decoding
gpoirier
parents: 5638
diff changeset
510 int single_decode_warning;
d2598034f2a9 Add slice-based parallel H.264 decoding
gpoirier
parents: 5638
diff changeset
511
d2598034f2a9 Add slice-based parallel H.264 decoding
gpoirier
parents: 5638
diff changeset
512 int last_slice_type;
d2598034f2a9 Add slice-based parallel H.264 decoding
gpoirier
parents: 5638
diff changeset
513 /** @} */
d2598034f2a9 Add slice-based parallel H.264 decoding
gpoirier
parents: 5638
diff changeset
514
6783
df0893f4fd86 Store mb_xy in H264Context and only calculate it once per MB.
astrange
parents: 6582
diff changeset
515 int mb_xy;
df0893f4fd86 Store mb_xy in H264Context and only calculate it once per MB.
astrange
parents: 6582
diff changeset
516
8025
58a16a83f4e2 Add variable to hold the svq3 watermarking key,
michael
parents: 7990
diff changeset
517 uint32_t svq3_watermark_key;
8107
e61f76efc9f3 h264: Implement decoding of picture timing SEI message.
andoma
parents: 8079
diff changeset
518
e61f76efc9f3 h264: Implement decoding of picture timing SEI message.
andoma
parents: 8079
diff changeset
519 /**
e61f76efc9f3 h264: Implement decoding of picture timing SEI message.
andoma
parents: 8079
diff changeset
520 * pic_struct in picture timing SEI message
e61f76efc9f3 h264: Implement decoding of picture timing SEI message.
andoma
parents: 8079
diff changeset
521 */
e61f76efc9f3 h264: Implement decoding of picture timing SEI message.
andoma
parents: 8079
diff changeset
522 SEI_PicStructType sei_pic_struct;
8353
add1a0d65370 Do not recalculate constant part of is_complex for every MB.
michael
parents: 8207
diff changeset
523
8783
e91ea98d868a Add SEI recovery point frame counter to H264Context and use it when
cehoyos
parents: 8762
diff changeset
524 /**
9831
febe1855b64a Add field prev_interlaced_frame to H264Context to be able to flag soft telecine
cehoyos
parents: 9384
diff changeset
525 * Complement sei_pic_struct
febe1855b64a Add field prev_interlaced_frame to H264Context to be able to flag soft telecine
cehoyos
parents: 9384
diff changeset
526 * SEI_PIC_STRUCT_TOP_BOTTOM and SEI_PIC_STRUCT_BOTTOM_TOP indicate interlaced frames.
febe1855b64a Add field prev_interlaced_frame to H264Context to be able to flag soft telecine
cehoyos
parents: 9384
diff changeset
527 * However, soft telecined frames may have these values.
febe1855b64a Add field prev_interlaced_frame to H264Context to be able to flag soft telecine
cehoyos
parents: 9384
diff changeset
528 * This is used in an attempt to flag soft telecine progressive.
febe1855b64a Add field prev_interlaced_frame to H264Context to be able to flag soft telecine
cehoyos
parents: 9384
diff changeset
529 */
febe1855b64a Add field prev_interlaced_frame to H264Context to be able to flag soft telecine
cehoyos
parents: 9384
diff changeset
530 int prev_interlaced_frame;
febe1855b64a Add field prev_interlaced_frame to H264Context to be able to flag soft telecine
cehoyos
parents: 9384
diff changeset
531
febe1855b64a Add field prev_interlaced_frame to H264Context to be able to flag soft telecine
cehoyos
parents: 9384
diff changeset
532 /**
9128
793cf8c68c4f Add support for ct_type to correctly detect interlaced flag
schreter
parents: 9071
diff changeset
533 * Bit set of clock types for fields/frames in picture timing SEI message.
793cf8c68c4f Add support for ct_type to correctly detect interlaced flag
schreter
parents: 9071
diff changeset
534 * For each found ct_type, appropriate bit is set (e.g., bit 1 for
793cf8c68c4f Add support for ct_type to correctly detect interlaced flag
schreter
parents: 9071
diff changeset
535 * interlaced).
793cf8c68c4f Add support for ct_type to correctly detect interlaced flag
schreter
parents: 9071
diff changeset
536 */
793cf8c68c4f Add support for ct_type to correctly detect interlaced flag
schreter
parents: 9071
diff changeset
537 int sei_ct_type;
793cf8c68c4f Add support for ct_type to correctly detect interlaced flag
schreter
parents: 9071
diff changeset
538
793cf8c68c4f Add support for ct_type to correctly detect interlaced flag
schreter
parents: 9071
diff changeset
539 /**
8966
328b7d545d0b Parse dpb_output_delay in SEI picture structure.
cehoyos
parents: 8965
diff changeset
540 * dpb_output_delay in picture timing SEI message, see H.264 C.2.2
328b7d545d0b Parse dpb_output_delay in SEI picture structure.
cehoyos
parents: 8965
diff changeset
541 */
328b7d545d0b Parse dpb_output_delay in SEI picture structure.
cehoyos
parents: 8965
diff changeset
542 int sei_dpb_output_delay;
328b7d545d0b Parse dpb_output_delay in SEI picture structure.
cehoyos
parents: 8965
diff changeset
543
328b7d545d0b Parse dpb_output_delay in SEI picture structure.
cehoyos
parents: 8965
diff changeset
544 /**
8967
f0abc5f9e726 Parse cbp_removal_delay in SEI picture structure.
cehoyos
parents: 8966
diff changeset
545 * cpb_removal_delay in picture timing SEI message, see H.264 C.1.2
f0abc5f9e726 Parse cbp_removal_delay in SEI picture structure.
cehoyos
parents: 8966
diff changeset
546 */
f0abc5f9e726 Parse cbp_removal_delay in SEI picture structure.
cehoyos
parents: 8966
diff changeset
547 int sei_cpb_removal_delay;
f0abc5f9e726 Parse cbp_removal_delay in SEI picture structure.
cehoyos
parents: 8966
diff changeset
548
f0abc5f9e726 Parse cbp_removal_delay in SEI picture structure.
cehoyos
parents: 8966
diff changeset
549 /**
8783
e91ea98d868a Add SEI recovery point frame counter to H264Context and use it when
cehoyos
parents: 8762
diff changeset
550 * recovery_frame_cnt from SEI message
e91ea98d868a Add SEI recovery point frame counter to H264Context and use it when
cehoyos
parents: 8762
diff changeset
551 *
e91ea98d868a Add SEI recovery point frame counter to H264Context and use it when
cehoyos
parents: 8762
diff changeset
552 * Set to -1 if no recovery point SEI message found or to number of frames
e91ea98d868a Add SEI recovery point frame counter to H264Context and use it when
cehoyos
parents: 8762
diff changeset
553 * before playback synchronizes. Frames having recovery point are key
e91ea98d868a Add SEI recovery point frame counter to H264Context and use it when
cehoyos
parents: 8762
diff changeset
554 * frames.
e91ea98d868a Add SEI recovery point frame counter to H264Context and use it when
cehoyos
parents: 8762
diff changeset
555 */
e91ea98d868a Add SEI recovery point frame counter to H264Context and use it when
cehoyos
parents: 8762
diff changeset
556 int sei_recovery_frame_cnt;
e91ea98d868a Add SEI recovery point frame counter to H264Context and use it when
cehoyos
parents: 8762
diff changeset
557
8353
add1a0d65370 Do not recalculate constant part of is_complex for every MB.
michael
parents: 8207
diff changeset
558 int is_complex;
8735
5d7ebbb7e91b Add fields to H264Context and SPS for upcoming VA API support.
cehoyos
parents: 8718
diff changeset
559
5d7ebbb7e91b Add fields to H264Context and SPS for upcoming VA API support.
cehoyos
parents: 8718
diff changeset
560 int luma_weight_flag[2]; ///< 7.4.3.2 luma_weight_lX_flag
5d7ebbb7e91b Add fields to H264Context and SPS for upcoming VA API support.
cehoyos
parents: 8718
diff changeset
561 int chroma_weight_flag[2]; ///< 7.4.3.2 chroma_weight_lX_flag
8972
eaf39f9b25df Parse buffering period (H.264, D.1.1).
cehoyos
parents: 8970
diff changeset
562
eaf39f9b25df Parse buffering period (H.264, D.1.1).
cehoyos
parents: 8970
diff changeset
563 // Timestamp stuff
eaf39f9b25df Parse buffering period (H.264, D.1.1).
cehoyos
parents: 8970
diff changeset
564 int sei_buffering_period_present; ///< Buffering period SEI flag
eaf39f9b25df Parse buffering period (H.264, D.1.1).
cehoyos
parents: 8970
diff changeset
565 int initial_cpb_removal_delay[32]; ///< Initial timestamps for CPBs
4975
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
566 }H264Context;
9a6a0818e93f split h264.c to move parser in its own file
aurel
parents:
diff changeset
567
10859
762e6bb0ba40 Split (picture|seq) parameter set decoding out of h264.c.
michael
parents: 10858
diff changeset
568
762e6bb0ba40 Split (picture|seq) parameter set decoding out of h264.c.
michael
parents: 10858
diff changeset
569 extern const uint8_t ff_h264_chroma_qp[52];
762e6bb0ba40 Split (picture|seq) parameter set decoding out of h264.c.
michael
parents: 10858
diff changeset
570
762e6bb0ba40 Split (picture|seq) parameter set decoding out of h264.c.
michael
parents: 10858
diff changeset
571
8996
e65778184ded Make the following H264 functions available to the parser:
cehoyos
parents: 8972
diff changeset
572 /**
e65778184ded Make the following H264 functions available to the parser:
cehoyos
parents: 8972
diff changeset
573 * Decode SEI
e65778184ded Make the following H264 functions available to the parser:
cehoyos
parents: 8972
diff changeset
574 */
e65778184ded Make the following H264 functions available to the parser:
cehoyos
parents: 8972
diff changeset
575 int ff_h264_decode_sei(H264Context *h);
e65778184ded Make the following H264 functions available to the parser:
cehoyos
parents: 8972
diff changeset
576
e65778184ded Make the following H264 functions available to the parser:
cehoyos
parents: 8972
diff changeset
577 /**
e65778184ded Make the following H264 functions available to the parser:
cehoyos
parents: 8972
diff changeset
578 * Decode SPS
e65778184ded Make the following H264 functions available to the parser:
cehoyos
parents: 8972
diff changeset
579 */
e65778184ded Make the following H264 functions available to the parser:
cehoyos
parents: 8972
diff changeset
580 int ff_h264_decode_seq_parameter_set(H264Context *h);
e65778184ded Make the following H264 functions available to the parser:
cehoyos
parents: 8972
diff changeset
581
e65778184ded Make the following H264 functions available to the parser:
cehoyos
parents: 8972
diff changeset
582 /**
e65778184ded Make the following H264 functions available to the parser:
cehoyos
parents: 8972
diff changeset
583 * Decode PPS
e65778184ded Make the following H264 functions available to the parser:
cehoyos
parents: 8972
diff changeset
584 */
e65778184ded Make the following H264 functions available to the parser:
cehoyos
parents: 8972
diff changeset
585 int ff_h264_decode_picture_parameter_set(H264Context *h, int bit_length);
e65778184ded Make the following H264 functions available to the parser:
cehoyos
parents: 8972
diff changeset
586
e65778184ded Make the following H264 functions available to the parser:
cehoyos
parents: 8972
diff changeset
587 /**
e65778184ded Make the following H264 functions available to the parser:
cehoyos
parents: 8972
diff changeset
588 * Decodes a network abstraction layer unit.
e65778184ded Make the following H264 functions available to the parser:
cehoyos
parents: 8972
diff changeset
589 * @param consumed is the number of bytes used as input
e65778184ded Make the following H264 functions available to the parser:
cehoyos
parents: 8972
diff changeset
590 * @param length is the length of the array
e65778184ded Make the following H264 functions available to the parser:
cehoyos
parents: 8972
diff changeset
591 * @param dst_length is the number of decoded bytes FIXME here or a decode rbsp tailing?
e65778184ded Make the following H264 functions available to the parser:
cehoyos
parents: 8972
diff changeset
592 * @returns decoded bytes, might be src+1 if no escapes
e65778184ded Make the following H264 functions available to the parser:
cehoyos
parents: 8972
diff changeset
593 */
e65778184ded Make the following H264 functions available to the parser:
cehoyos
parents: 8972
diff changeset
594 const uint8_t *ff_h264_decode_nal(H264Context *h, const uint8_t *src, int *dst_length, int *consumed, int length);
e65778184ded Make the following H264 functions available to the parser:
cehoyos
parents: 8972
diff changeset
595
e65778184ded Make the following H264 functions available to the parser:
cehoyos
parents: 8972
diff changeset
596 /**
e65778184ded Make the following H264 functions available to the parser:
cehoyos
parents: 8972
diff changeset
597 * identifies the exact end of the bitstream
e65778184ded Make the following H264 functions available to the parser:
cehoyos
parents: 8972
diff changeset
598 * @return the length of the trailing, or 0 if damaged
e65778184ded Make the following H264 functions available to the parser:
cehoyos
parents: 8972
diff changeset
599 */
e65778184ded Make the following H264 functions available to the parser:
cehoyos
parents: 8972
diff changeset
600 int ff_h264_decode_rbsp_trailing(H264Context *h, const uint8_t *src);
e65778184ded Make the following H264 functions available to the parser:
cehoyos
parents: 8972
diff changeset
601
9380
54e650136c87 Add a ff_h264_free_context function and call it from the H.264 parser.
reimar
parents: 9128
diff changeset
602 /**
54e650136c87 Add a ff_h264_free_context function and call it from the H.264 parser.
reimar
parents: 9128
diff changeset
603 * frees any data that may have been allocated in the H264 context like SPS, PPS etc.
54e650136c87 Add a ff_h264_free_context function and call it from the H.264 parser.
reimar
parents: 9128
diff changeset
604 */
9384
828d15d322ed Add forgotten void return type to ff_h264_free_context
reimar
parents: 9380
diff changeset
605 av_cold void ff_h264_free_context(H264Context *h);
9380
54e650136c87 Add a ff_h264_free_context function and call it from the H.264 parser.
reimar
parents: 9128
diff changeset
606
10829
46a247e54d6e Export get_slice_type() as ff_h264_get_slice_type().
ramiro
parents: 9831
diff changeset
607 /**
46a247e54d6e Export get_slice_type() as ff_h264_get_slice_type().
ramiro
parents: 9831
diff changeset
608 * reconstructs bitstream slice_type.
46a247e54d6e Export get_slice_type() as ff_h264_get_slice_type().
ramiro
parents: 9831
diff changeset
609 */
46a247e54d6e Export get_slice_type() as ff_h264_get_slice_type().
ramiro
parents: 9831
diff changeset
610 int ff_h264_get_slice_type(H264Context *h);
46a247e54d6e Export get_slice_type() as ff_h264_get_slice_type().
ramiro
parents: 9831
diff changeset
611
10852
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
612 /**
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
613 * allocates tables.
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
614 * needs width/height
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
615 */
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
616 int ff_h264_alloc_tables(H264Context *h);
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
617
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
618 /**
10862
d9c084a0c22b Split all the reference picture handling off h264.c.
michael
parents: 10859
diff changeset
619 * fills the default_ref_list.
d9c084a0c22b Split all the reference picture handling off h264.c.
michael
parents: 10859
diff changeset
620 */
d9c084a0c22b Split all the reference picture handling off h264.c.
michael
parents: 10859
diff changeset
621 int ff_h264_fill_default_ref_list(H264Context *h);
d9c084a0c22b Split all the reference picture handling off h264.c.
michael
parents: 10859
diff changeset
622
d9c084a0c22b Split all the reference picture handling off h264.c.
michael
parents: 10859
diff changeset
623 int ff_h264_decode_ref_pic_list_reordering(H264Context *h);
d9c084a0c22b Split all the reference picture handling off h264.c.
michael
parents: 10859
diff changeset
624 void ff_h264_fill_mbaff_ref_list(H264Context *h);
d9c084a0c22b Split all the reference picture handling off h264.c.
michael
parents: 10859
diff changeset
625 void ff_h264_remove_all_refs(H264Context *h);
d9c084a0c22b Split all the reference picture handling off h264.c.
michael
parents: 10859
diff changeset
626
d9c084a0c22b Split all the reference picture handling off h264.c.
michael
parents: 10859
diff changeset
627 /**
d9c084a0c22b Split all the reference picture handling off h264.c.
michael
parents: 10859
diff changeset
628 * Executes the reference picture marking (memory management control operations).
d9c084a0c22b Split all the reference picture handling off h264.c.
michael
parents: 10859
diff changeset
629 */
d9c084a0c22b Split all the reference picture handling off h264.c.
michael
parents: 10859
diff changeset
630 int ff_h264_execute_ref_pic_marking(H264Context *h, MMCO *mmco, int mmco_count);
d9c084a0c22b Split all the reference picture handling off h264.c.
michael
parents: 10859
diff changeset
631
d9c084a0c22b Split all the reference picture handling off h264.c.
michael
parents: 10859
diff changeset
632 int ff_h264_decode_ref_pic_marking(H264Context *h, GetBitContext *gb);
d9c084a0c22b Split all the reference picture handling off h264.c.
michael
parents: 10859
diff changeset
633
d9c084a0c22b Split all the reference picture handling off h264.c.
michael
parents: 10859
diff changeset
634
d9c084a0c22b Split all the reference picture handling off h264.c.
michael
parents: 10859
diff changeset
635 /**
10852
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
636 * checks if the top & left blocks are available if needed & changes the dc mode so it only uses the available blocks.
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
637 */
10863
974ac220c93a Move check_intra4x4_pred_mode() back from h264.h to h264.c, the function is just
michael
parents: 10862
diff changeset
638 int ff_h264_check_intra4x4_pred_mode(H264Context *h);
974ac220c93a Move check_intra4x4_pred_mode() back from h264.h to h264.c, the function is just
michael
parents: 10862
diff changeset
639
974ac220c93a Move check_intra4x4_pred_mode() back from h264.h to h264.c, the function is just
michael
parents: 10862
diff changeset
640 /**
974ac220c93a Move check_intra4x4_pred_mode() back from h264.h to h264.c, the function is just
michael
parents: 10862
diff changeset
641 * checks if the top & left blocks are available if needed & changes the dc mode so it only uses the available blocks.
974ac220c93a Move check_intra4x4_pred_mode() back from h264.h to h264.c, the function is just
michael
parents: 10862
diff changeset
642 */
10852
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
643 int ff_h264_check_intra_pred_mode(H264Context *h, int mode);
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
644
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
645 void ff_h264_write_back_intra_pred_mode(H264Context *h);
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
646 void ff_h264_hl_decode_mb(H264Context *h);
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
647 int ff_h264_frame_start(H264Context *h);
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
648 av_cold int ff_h264_decode_init(AVCodecContext *avctx);
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
649 av_cold int ff_h264_decode_end(AVCodecContext *avctx);
10866
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
650 av_cold void ff_h264_decode_init_vlc(void);
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
651
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
652 /**
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
653 * decodes a macroblock
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
654 * @returns 0 if OK, AC_ERROR / DC_ERROR / MV_ERROR if an error is noticed
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
655 */
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
656 int ff_h264_decode_mb_cavlc(H264Context *h);
10852
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
657
10869
7101061bfa0f Split cabac decoding code out of h264.c.
michael
parents: 10866
diff changeset
658 /**
7101061bfa0f Split cabac decoding code out of h264.c.
michael
parents: 10866
diff changeset
659 * decodes a CABAC coded macroblock
7101061bfa0f Split cabac decoding code out of h264.c.
michael
parents: 10866
diff changeset
660 * @returns 0 if OK, AC_ERROR / DC_ERROR / MV_ERROR if an error is noticed
7101061bfa0f Split cabac decoding code out of h264.c.
michael
parents: 10866
diff changeset
661 */
7101061bfa0f Split cabac decoding code out of h264.c.
michael
parents: 10866
diff changeset
662 int ff_h264_decode_mb_cabac(H264Context *h);
7101061bfa0f Split cabac decoding code out of h264.c.
michael
parents: 10866
diff changeset
663
7101061bfa0f Split cabac decoding code out of h264.c.
michael
parents: 10866
diff changeset
664 void ff_h264_init_cabac_states(H264Context *h);
7101061bfa0f Split cabac decoding code out of h264.c.
michael
parents: 10866
diff changeset
665
10857
b20434143fd5 Split direct mode (macro)block decoding off h264.c.
michael
parents: 10854
diff changeset
666 void ff_h264_direct_dist_scale_factor(H264Context * const h);
b20434143fd5 Split direct mode (macro)block decoding off h264.c.
michael
parents: 10854
diff changeset
667 void ff_h264_direct_ref_list_init(H264Context * const h);
b20434143fd5 Split direct mode (macro)block decoding off h264.c.
michael
parents: 10854
diff changeset
668 void ff_h264_pred_direct_motion(H264Context * const h, int *mb_type);
b20434143fd5 Split direct mode (macro)block decoding off h264.c.
michael
parents: 10854
diff changeset
669
10854
f6fc6ace95e3 Split h264 loop filter off h264.c.
michael
parents: 10852
diff changeset
670 void ff_h264_filter_mb_fast( H264Context *h, int mb_x, int mb_y, uint8_t *img_y, uint8_t *img_cb, uint8_t *img_cr, unsigned int linesize, unsigned int uvlinesize);
f6fc6ace95e3 Split h264 loop filter off h264.c.
michael
parents: 10852
diff changeset
671 void ff_h264_filter_mb( H264Context *h, int mb_x, int mb_y, uint8_t *img_y, uint8_t *img_cb, uint8_t *img_cr, unsigned int linesize, unsigned int uvlinesize);
f6fc6ace95e3 Split h264 loop filter off h264.c.
michael
parents: 10852
diff changeset
672
10858
31645cf81fcb Split SEI code off h264.c.
michael
parents: 10857
diff changeset
673 /**
31645cf81fcb Split SEI code off h264.c.
michael
parents: 10857
diff changeset
674 * Reset SEI values at the beginning of the frame.
31645cf81fcb Split SEI code off h264.c.
michael
parents: 10857
diff changeset
675 *
31645cf81fcb Split SEI code off h264.c.
michael
parents: 10857
diff changeset
676 * @param h H.264 context.
31645cf81fcb Split SEI code off h264.c.
michael
parents: 10857
diff changeset
677 */
31645cf81fcb Split SEI code off h264.c.
michael
parents: 10857
diff changeset
678 void ff_h264_reset_sei(H264Context *h);
31645cf81fcb Split SEI code off h264.c.
michael
parents: 10857
diff changeset
679
31645cf81fcb Split SEI code off h264.c.
michael
parents: 10857
diff changeset
680
10852
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
681 /*
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
682 o-o o-o
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
683 / / /
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
684 o-o o-o
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
685 ,---'
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
686 o-o o-o
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
687 / / /
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
688 o-o o-o
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
689 */
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
690 //This table must be here because scan8[constant] must be known at compiletime
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
691 static const uint8_t scan8[16 + 2*4]={
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
692 4+1*8, 5+1*8, 4+2*8, 5+2*8,
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
693 6+1*8, 7+1*8, 6+2*8, 7+2*8,
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
694 4+3*8, 5+3*8, 4+4*8, 5+4*8,
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
695 6+3*8, 7+3*8, 6+4*8, 7+4*8,
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
696 1+1*8, 2+1*8,
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
697 1+2*8, 2+2*8,
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
698 1+4*8, 2+4*8,
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
699 1+5*8, 2+5*8,
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
700 };
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
701
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
702 static av_always_inline uint32_t pack16to32(int a, int b){
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
703 #if HAVE_BIGENDIAN
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
704 return (b&0xFFFF) + (a<<16);
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
705 #else
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
706 return (a&0xFFFF) + (b<<16);
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
707 #endif
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
708 }
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
709
86d7ab878805 Get rid of #include "svq3.c"
michael
parents: 10837
diff changeset
710 /**
10854
f6fc6ace95e3 Split h264 loop filter off h264.c.
michael
parents: 10852
diff changeset
711 * gets the chroma qp.
f6fc6ace95e3 Split h264 loop filter off h264.c.
michael
parents: 10852
diff changeset
712 */
f6fc6ace95e3 Split h264 loop filter off h264.c.
michael
parents: 10852
diff changeset
713 static inline int get_chroma_qp(H264Context *h, int t, int qscale){
f6fc6ace95e3 Split h264 loop filter off h264.c.
michael
parents: 10852
diff changeset
714 return h->pps.chroma_qp_table[t][qscale];
f6fc6ace95e3 Split h264 loop filter off h264.c.
michael
parents: 10852
diff changeset
715 }
f6fc6ace95e3 Split h264 loop filter off h264.c.
michael
parents: 10852
diff changeset
716
10866
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
717 static inline void pred_pskip_motion(H264Context * const h, int * const mx, int * const my);
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
718
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
719 static void fill_caches(H264Context *h, int mb_type, int for_deblock){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
720 MpegEncContext * const s = &h->s;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
721 const int mb_xy= h->mb_xy;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
722 int topleft_xy, top_xy, topright_xy, left_xy[2];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
723 int topleft_type, top_type, topright_type, left_type[2];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
724 const uint8_t * left_block;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
725 int topleft_partition= -1;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
726 int i;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
727 static const uint8_t left_block_options[4][8]={
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
728 {0,1,2,3,7,10,8,11},
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
729 {2,2,3,3,8,11,8,11},
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
730 {0,0,1,1,7,10,7,10},
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
731 {0,2,0,2,7,10,7,10}
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
732 };
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
733
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
734 top_xy = mb_xy - (s->mb_stride << FIELD_PICTURE);
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
735
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
736 //FIXME deblocking could skip the intra and nnz parts.
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
737 if(for_deblock && (h->slice_num == 1 || h->slice_table[mb_xy] == h->slice_table[top_xy]) && !FRAME_MBAFF)
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
738 return;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
739
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
740 /* Wow, what a mess, why didn't they simplify the interlacing & intra
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
741 * stuff, I can't imagine that these complex rules are worth it. */
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
742
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
743 topleft_xy = top_xy - 1;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
744 topright_xy= top_xy + 1;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
745 left_xy[1] = left_xy[0] = mb_xy-1;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
746 left_block = left_block_options[0];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
747 if(FRAME_MBAFF){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
748 const int pair_xy = s->mb_x + (s->mb_y & ~1)*s->mb_stride;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
749 const int top_pair_xy = pair_xy - s->mb_stride;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
750 const int topleft_pair_xy = top_pair_xy - 1;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
751 const int topright_pair_xy = top_pair_xy + 1;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
752 const int topleft_mb_field_flag = IS_INTERLACED(s->current_picture.mb_type[topleft_pair_xy]);
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
753 const int top_mb_field_flag = IS_INTERLACED(s->current_picture.mb_type[top_pair_xy]);
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
754 const int topright_mb_field_flag = IS_INTERLACED(s->current_picture.mb_type[topright_pair_xy]);
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
755 const int left_mb_field_flag = IS_INTERLACED(s->current_picture.mb_type[pair_xy-1]);
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
756 const int curr_mb_field_flag = IS_INTERLACED(mb_type);
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
757 const int bottom = (s->mb_y & 1);
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
758 tprintf(s->avctx, "fill_caches: curr_mb_field_flag:%d, left_mb_field_flag:%d, topleft_mb_field_flag:%d, top_mb_field_flag:%d, topright_mb_field_flag:%d\n", curr_mb_field_flag, left_mb_field_flag, topleft_mb_field_flag, top_mb_field_flag, topright_mb_field_flag);
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
759
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
760 if (curr_mb_field_flag && (bottom || top_mb_field_flag)){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
761 top_xy -= s->mb_stride;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
762 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
763 if (curr_mb_field_flag && (bottom || topleft_mb_field_flag)){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
764 topleft_xy -= s->mb_stride;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
765 } else if(bottom && !curr_mb_field_flag && left_mb_field_flag) {
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
766 topleft_xy += s->mb_stride;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
767 // take top left mv from the middle of the mb, as opposed to all other modes which use the bottom right partition
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
768 topleft_partition = 0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
769 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
770 if (curr_mb_field_flag && (bottom || topright_mb_field_flag)){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
771 topright_xy -= s->mb_stride;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
772 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
773 if (left_mb_field_flag != curr_mb_field_flag) {
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
774 left_xy[1] = left_xy[0] = pair_xy - 1;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
775 if (curr_mb_field_flag) {
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
776 left_xy[1] += s->mb_stride;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
777 left_block = left_block_options[3];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
778 } else {
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
779 left_block= left_block_options[2 - bottom];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
780 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
781 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
782 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
783
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
784 h->top_mb_xy = top_xy;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
785 h->left_mb_xy[0] = left_xy[0];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
786 h->left_mb_xy[1] = left_xy[1];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
787 if(for_deblock){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
788 topleft_type = 0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
789 topright_type = 0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
790 top_type = h->slice_table[top_xy ] < 0xFFFF ? s->current_picture.mb_type[top_xy] : 0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
791 left_type[0] = h->slice_table[left_xy[0] ] < 0xFFFF ? s->current_picture.mb_type[left_xy[0]] : 0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
792 left_type[1] = h->slice_table[left_xy[1] ] < 0xFFFF ? s->current_picture.mb_type[left_xy[1]] : 0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
793
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
794 if(MB_MBAFF && !IS_INTRA(mb_type)){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
795 int list;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
796 for(list=0; list<h->list_count; list++){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
797 //These values where changed for ease of performing MC, we need to change them back
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
798 //FIXME maybe we can make MC and loop filter use the same values or prevent
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
799 //the MC code from changing ref_cache and rather use a temporary array.
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
800 if(USES_LIST(mb_type,list)){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
801 int8_t *ref = &s->current_picture.ref_index[list][h->mb2b8_xy[mb_xy]];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
802 *(uint32_t*)&h->ref_cache[list][scan8[ 0]] =
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
803 *(uint32_t*)&h->ref_cache[list][scan8[ 2]] = (pack16to32(ref[0],ref[1])&0x00FF00FF)*0x0101;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
804 ref += h->b8_stride;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
805 *(uint32_t*)&h->ref_cache[list][scan8[ 8]] =
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
806 *(uint32_t*)&h->ref_cache[list][scan8[10]] = (pack16to32(ref[0],ref[1])&0x00FF00FF)*0x0101;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
807 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
808 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
809 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
810 }else{
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
811 topleft_type = h->slice_table[topleft_xy ] == h->slice_num ? s->current_picture.mb_type[topleft_xy] : 0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
812 top_type = h->slice_table[top_xy ] == h->slice_num ? s->current_picture.mb_type[top_xy] : 0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
813 topright_type= h->slice_table[topright_xy] == h->slice_num ? s->current_picture.mb_type[topright_xy]: 0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
814 left_type[0] = h->slice_table[left_xy[0] ] == h->slice_num ? s->current_picture.mb_type[left_xy[0]] : 0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
815 left_type[1] = h->slice_table[left_xy[1] ] == h->slice_num ? s->current_picture.mb_type[left_xy[1]] : 0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
816
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
817 if(IS_INTRA(mb_type)){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
818 int type_mask= h->pps.constrained_intra_pred ? IS_INTRA(-1) : -1;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
819 h->topleft_samples_available=
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
820 h->top_samples_available=
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
821 h->left_samples_available= 0xFFFF;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
822 h->topright_samples_available= 0xEEEA;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
823
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
824 if(!(top_type & type_mask)){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
825 h->topleft_samples_available= 0xB3FF;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
826 h->top_samples_available= 0x33FF;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
827 h->topright_samples_available= 0x26EA;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
828 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
829 if(IS_INTERLACED(mb_type) != IS_INTERLACED(left_type[0])){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
830 if(IS_INTERLACED(mb_type)){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
831 if(!(left_type[0] & type_mask)){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
832 h->topleft_samples_available&= 0xDFFF;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
833 h->left_samples_available&= 0x5FFF;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
834 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
835 if(!(left_type[1] & type_mask)){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
836 h->topleft_samples_available&= 0xFF5F;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
837 h->left_samples_available&= 0xFF5F;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
838 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
839 }else{
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
840 int left_typei = h->slice_table[left_xy[0] + s->mb_stride ] == h->slice_num
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
841 ? s->current_picture.mb_type[left_xy[0] + s->mb_stride] : 0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
842 assert(left_xy[0] == left_xy[1]);
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
843 if(!((left_typei & type_mask) && (left_type[0] & type_mask))){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
844 h->topleft_samples_available&= 0xDF5F;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
845 h->left_samples_available&= 0x5F5F;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
846 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
847 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
848 }else{
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
849 if(!(left_type[0] & type_mask)){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
850 h->topleft_samples_available&= 0xDF5F;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
851 h->left_samples_available&= 0x5F5F;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
852 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
853 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
854
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
855 if(!(topleft_type & type_mask))
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
856 h->topleft_samples_available&= 0x7FFF;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
857
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
858 if(!(topright_type & type_mask))
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
859 h->topright_samples_available&= 0xFBFF;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
860
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
861 if(IS_INTRA4x4(mb_type)){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
862 if(IS_INTRA4x4(top_type)){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
863 h->intra4x4_pred_mode_cache[4+8*0]= h->intra4x4_pred_mode[top_xy][4];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
864 h->intra4x4_pred_mode_cache[5+8*0]= h->intra4x4_pred_mode[top_xy][5];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
865 h->intra4x4_pred_mode_cache[6+8*0]= h->intra4x4_pred_mode[top_xy][6];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
866 h->intra4x4_pred_mode_cache[7+8*0]= h->intra4x4_pred_mode[top_xy][3];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
867 }else{
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
868 int pred;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
869 if(!(top_type & type_mask))
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
870 pred= -1;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
871 else{
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
872 pred= 2;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
873 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
874 h->intra4x4_pred_mode_cache[4+8*0]=
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
875 h->intra4x4_pred_mode_cache[5+8*0]=
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
876 h->intra4x4_pred_mode_cache[6+8*0]=
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
877 h->intra4x4_pred_mode_cache[7+8*0]= pred;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
878 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
879 for(i=0; i<2; i++){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
880 if(IS_INTRA4x4(left_type[i])){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
881 h->intra4x4_pred_mode_cache[3+8*1 + 2*8*i]= h->intra4x4_pred_mode[left_xy[i]][left_block[0+2*i]];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
882 h->intra4x4_pred_mode_cache[3+8*2 + 2*8*i]= h->intra4x4_pred_mode[left_xy[i]][left_block[1+2*i]];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
883 }else{
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
884 int pred;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
885 if(!(left_type[i] & type_mask))
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
886 pred= -1;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
887 else{
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
888 pred= 2;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
889 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
890 h->intra4x4_pred_mode_cache[3+8*1 + 2*8*i]=
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
891 h->intra4x4_pred_mode_cache[3+8*2 + 2*8*i]= pred;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
892 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
893 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
894 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
895 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
896 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
897
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
898
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
899 /*
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
900 0 . T T. T T T T
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
901 1 L . .L . . . .
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
902 2 L . .L . . . .
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
903 3 . T TL . . . .
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
904 4 L . .L . . . .
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
905 5 L . .. . . . .
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
906 */
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
907 //FIXME constraint_intra_pred & partitioning & nnz (let us hope this is just a typo in the spec)
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
908 if(top_type){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
909 h->non_zero_count_cache[4+8*0]= h->non_zero_count[top_xy][4];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
910 h->non_zero_count_cache[5+8*0]= h->non_zero_count[top_xy][5];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
911 h->non_zero_count_cache[6+8*0]= h->non_zero_count[top_xy][6];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
912 h->non_zero_count_cache[7+8*0]= h->non_zero_count[top_xy][3];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
913
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
914 h->non_zero_count_cache[1+8*0]= h->non_zero_count[top_xy][9];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
915 h->non_zero_count_cache[2+8*0]= h->non_zero_count[top_xy][8];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
916
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
917 h->non_zero_count_cache[1+8*3]= h->non_zero_count[top_xy][12];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
918 h->non_zero_count_cache[2+8*3]= h->non_zero_count[top_xy][11];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
919
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
920 }else{
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
921 h->non_zero_count_cache[4+8*0]=
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
922 h->non_zero_count_cache[5+8*0]=
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
923 h->non_zero_count_cache[6+8*0]=
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
924 h->non_zero_count_cache[7+8*0]=
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
925
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
926 h->non_zero_count_cache[1+8*0]=
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
927 h->non_zero_count_cache[2+8*0]=
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
928
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
929 h->non_zero_count_cache[1+8*3]=
10893
2aafcafbe1f0 Replace cabac checks in inline functions from h264.h with constants.
michael
parents: 10883
diff changeset
930 h->non_zero_count_cache[2+8*3]= CABAC && !IS_INTRA(mb_type) ? 0 : 64;
10866
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
931
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
932 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
933
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
934 for (i=0; i<2; i++) {
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
935 if(left_type[i]){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
936 h->non_zero_count_cache[3+8*1 + 2*8*i]= h->non_zero_count[left_xy[i]][left_block[0+2*i]];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
937 h->non_zero_count_cache[3+8*2 + 2*8*i]= h->non_zero_count[left_xy[i]][left_block[1+2*i]];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
938 h->non_zero_count_cache[0+8*1 + 8*i]= h->non_zero_count[left_xy[i]][left_block[4+2*i]];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
939 h->non_zero_count_cache[0+8*4 + 8*i]= h->non_zero_count[left_xy[i]][left_block[5+2*i]];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
940 }else{
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
941 h->non_zero_count_cache[3+8*1 + 2*8*i]=
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
942 h->non_zero_count_cache[3+8*2 + 2*8*i]=
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
943 h->non_zero_count_cache[0+8*1 + 8*i]=
10893
2aafcafbe1f0 Replace cabac checks in inline functions from h264.h with constants.
michael
parents: 10883
diff changeset
944 h->non_zero_count_cache[0+8*4 + 8*i]= CABAC && !IS_INTRA(mb_type) ? 0 : 64;
10866
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
945 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
946 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
947
10893
2aafcafbe1f0 Replace cabac checks in inline functions from h264.h with constants.
michael
parents: 10883
diff changeset
948 if( CABAC ) {
10866
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
949 // top_cbp
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
950 if(top_type) {
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
951 h->top_cbp = h->cbp_table[top_xy];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
952 } else if(IS_INTRA(mb_type)) {
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
953 h->top_cbp = 0x1C0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
954 } else {
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
955 h->top_cbp = 0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
956 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
957 // left_cbp
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
958 if (left_type[0]) {
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
959 h->left_cbp = h->cbp_table[left_xy[0]] & 0x1f0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
960 } else if(IS_INTRA(mb_type)) {
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
961 h->left_cbp = 0x1C0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
962 } else {
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
963 h->left_cbp = 0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
964 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
965 if (left_type[0]) {
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
966 h->left_cbp |= ((h->cbp_table[left_xy[0]]>>((left_block[0]&(~1))+1))&0x1) << 1;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
967 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
968 if (left_type[1]) {
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
969 h->left_cbp |= ((h->cbp_table[left_xy[1]]>>((left_block[2]&(~1))+1))&0x1) << 3;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
970 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
971 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
972
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
973 #if 1
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
974 if(IS_INTER(mb_type) || IS_DIRECT(mb_type)){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
975 int list;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
976 for(list=0; list<h->list_count; list++){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
977 if(!USES_LIST(mb_type, list) && !IS_DIRECT(mb_type) && !h->deblocking_filter){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
978 /*if(!h->mv_cache_clean[list]){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
979 memset(h->mv_cache [list], 0, 8*5*2*sizeof(int16_t)); //FIXME clean only input? clean at all?
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
980 memset(h->ref_cache[list], PART_NOT_AVAILABLE, 8*5*sizeof(int8_t));
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
981 h->mv_cache_clean[list]= 1;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
982 }*/
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
983 continue;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
984 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
985 h->mv_cache_clean[list]= 0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
986
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
987 if(USES_LIST(top_type, list)){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
988 const int b_xy= h->mb2b_xy[top_xy] + 3*h->b_stride;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
989 const int b8_xy= h->mb2b8_xy[top_xy] + h->b8_stride;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
990 *(uint32_t*)h->mv_cache[list][scan8[0] + 0 - 1*8]= *(uint32_t*)s->current_picture.motion_val[list][b_xy + 0];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
991 *(uint32_t*)h->mv_cache[list][scan8[0] + 1 - 1*8]= *(uint32_t*)s->current_picture.motion_val[list][b_xy + 1];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
992 *(uint32_t*)h->mv_cache[list][scan8[0] + 2 - 1*8]= *(uint32_t*)s->current_picture.motion_val[list][b_xy + 2];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
993 *(uint32_t*)h->mv_cache[list][scan8[0] + 3 - 1*8]= *(uint32_t*)s->current_picture.motion_val[list][b_xy + 3];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
994 h->ref_cache[list][scan8[0] + 0 - 1*8]=
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
995 h->ref_cache[list][scan8[0] + 1 - 1*8]= s->current_picture.ref_index[list][b8_xy + 0];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
996 h->ref_cache[list][scan8[0] + 2 - 1*8]=
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
997 h->ref_cache[list][scan8[0] + 3 - 1*8]= s->current_picture.ref_index[list][b8_xy + 1];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
998 }else{
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
999 *(uint32_t*)h->mv_cache [list][scan8[0] + 0 - 1*8]=
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1000 *(uint32_t*)h->mv_cache [list][scan8[0] + 1 - 1*8]=
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1001 *(uint32_t*)h->mv_cache [list][scan8[0] + 2 - 1*8]=
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1002 *(uint32_t*)h->mv_cache [list][scan8[0] + 3 - 1*8]= 0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1003 *(uint32_t*)&h->ref_cache[list][scan8[0] + 0 - 1*8]= ((top_type ? LIST_NOT_USED : PART_NOT_AVAILABLE)&0xFF)*0x01010101;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1004 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1005
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1006 for(i=0; i<2; i++){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1007 int cache_idx = scan8[0] - 1 + i*2*8;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1008 if(USES_LIST(left_type[i], list)){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1009 const int b_xy= h->mb2b_xy[left_xy[i]] + 3;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1010 const int b8_xy= h->mb2b8_xy[left_xy[i]] + 1;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1011 *(uint32_t*)h->mv_cache[list][cache_idx ]= *(uint32_t*)s->current_picture.motion_val[list][b_xy + h->b_stride*left_block[0+i*2]];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1012 *(uint32_t*)h->mv_cache[list][cache_idx+8]= *(uint32_t*)s->current_picture.motion_val[list][b_xy + h->b_stride*left_block[1+i*2]];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1013 h->ref_cache[list][cache_idx ]= s->current_picture.ref_index[list][b8_xy + h->b8_stride*(left_block[0+i*2]>>1)];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1014 h->ref_cache[list][cache_idx+8]= s->current_picture.ref_index[list][b8_xy + h->b8_stride*(left_block[1+i*2]>>1)];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1015 }else{
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1016 *(uint32_t*)h->mv_cache [list][cache_idx ]=
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1017 *(uint32_t*)h->mv_cache [list][cache_idx+8]= 0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1018 h->ref_cache[list][cache_idx ]=
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1019 h->ref_cache[list][cache_idx+8]= left_type[i] ? LIST_NOT_USED : PART_NOT_AVAILABLE;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1020 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1021 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1022
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1023 if(for_deblock || ((IS_DIRECT(mb_type) && !h->direct_spatial_mv_pred) && !FRAME_MBAFF))
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1024 continue;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1025
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1026 if(USES_LIST(topleft_type, list)){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1027 const int b_xy = h->mb2b_xy[topleft_xy] + 3 + h->b_stride + (topleft_partition & 2*h->b_stride);
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1028 const int b8_xy= h->mb2b8_xy[topleft_xy] + 1 + (topleft_partition & h->b8_stride);
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1029 *(uint32_t*)h->mv_cache[list][scan8[0] - 1 - 1*8]= *(uint32_t*)s->current_picture.motion_val[list][b_xy];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1030 h->ref_cache[list][scan8[0] - 1 - 1*8]= s->current_picture.ref_index[list][b8_xy];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1031 }else{
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1032 *(uint32_t*)h->mv_cache[list][scan8[0] - 1 - 1*8]= 0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1033 h->ref_cache[list][scan8[0] - 1 - 1*8]= topleft_type ? LIST_NOT_USED : PART_NOT_AVAILABLE;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1034 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1035
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1036 if(USES_LIST(topright_type, list)){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1037 const int b_xy= h->mb2b_xy[topright_xy] + 3*h->b_stride;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1038 const int b8_xy= h->mb2b8_xy[topright_xy] + h->b8_stride;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1039 *(uint32_t*)h->mv_cache[list][scan8[0] + 4 - 1*8]= *(uint32_t*)s->current_picture.motion_val[list][b_xy];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1040 h->ref_cache[list][scan8[0] + 4 - 1*8]= s->current_picture.ref_index[list][b8_xy];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1041 }else{
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1042 *(uint32_t*)h->mv_cache [list][scan8[0] + 4 - 1*8]= 0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1043 h->ref_cache[list][scan8[0] + 4 - 1*8]= topright_type ? LIST_NOT_USED : PART_NOT_AVAILABLE;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1044 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1045
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1046 if((IS_SKIP(mb_type) || IS_DIRECT(mb_type)) && !FRAME_MBAFF)
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1047 continue;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1048
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1049 h->ref_cache[list][scan8[5 ]+1] =
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1050 h->ref_cache[list][scan8[7 ]+1] =
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1051 h->ref_cache[list][scan8[13]+1] = //FIXME remove past 3 (init somewhere else)
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1052 h->ref_cache[list][scan8[4 ]] =
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1053 h->ref_cache[list][scan8[12]] = PART_NOT_AVAILABLE;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1054 *(uint32_t*)h->mv_cache [list][scan8[5 ]+1]=
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1055 *(uint32_t*)h->mv_cache [list][scan8[7 ]+1]=
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1056 *(uint32_t*)h->mv_cache [list][scan8[13]+1]= //FIXME remove past 3 (init somewhere else)
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1057 *(uint32_t*)h->mv_cache [list][scan8[4 ]]=
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1058 *(uint32_t*)h->mv_cache [list][scan8[12]]= 0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1059
10893
2aafcafbe1f0 Replace cabac checks in inline functions from h264.h with constants.
michael
parents: 10883
diff changeset
1060 if( CABAC ) {
10866
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1061 /* XXX beurk, Load mvd */
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1062 if(USES_LIST(top_type, list)){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1063 const int b_xy= h->mb2b_xy[top_xy] + 3*h->b_stride;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1064 *(uint32_t*)h->mvd_cache[list][scan8[0] + 0 - 1*8]= *(uint32_t*)h->mvd_table[list][b_xy + 0];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1065 *(uint32_t*)h->mvd_cache[list][scan8[0] + 1 - 1*8]= *(uint32_t*)h->mvd_table[list][b_xy + 1];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1066 *(uint32_t*)h->mvd_cache[list][scan8[0] + 2 - 1*8]= *(uint32_t*)h->mvd_table[list][b_xy + 2];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1067 *(uint32_t*)h->mvd_cache[list][scan8[0] + 3 - 1*8]= *(uint32_t*)h->mvd_table[list][b_xy + 3];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1068 }else{
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1069 *(uint32_t*)h->mvd_cache [list][scan8[0] + 0 - 1*8]=
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1070 *(uint32_t*)h->mvd_cache [list][scan8[0] + 1 - 1*8]=
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1071 *(uint32_t*)h->mvd_cache [list][scan8[0] + 2 - 1*8]=
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1072 *(uint32_t*)h->mvd_cache [list][scan8[0] + 3 - 1*8]= 0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1073 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1074 if(USES_LIST(left_type[0], list)){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1075 const int b_xy= h->mb2b_xy[left_xy[0]] + 3;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1076 *(uint32_t*)h->mvd_cache[list][scan8[0] - 1 + 0*8]= *(uint32_t*)h->mvd_table[list][b_xy + h->b_stride*left_block[0]];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1077 *(uint32_t*)h->mvd_cache[list][scan8[0] - 1 + 1*8]= *(uint32_t*)h->mvd_table[list][b_xy + h->b_stride*left_block[1]];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1078 }else{
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1079 *(uint32_t*)h->mvd_cache [list][scan8[0] - 1 + 0*8]=
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1080 *(uint32_t*)h->mvd_cache [list][scan8[0] - 1 + 1*8]= 0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1081 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1082 if(USES_LIST(left_type[1], list)){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1083 const int b_xy= h->mb2b_xy[left_xy[1]] + 3;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1084 *(uint32_t*)h->mvd_cache[list][scan8[0] - 1 + 2*8]= *(uint32_t*)h->mvd_table[list][b_xy + h->b_stride*left_block[2]];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1085 *(uint32_t*)h->mvd_cache[list][scan8[0] - 1 + 3*8]= *(uint32_t*)h->mvd_table[list][b_xy + h->b_stride*left_block[3]];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1086 }else{
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1087 *(uint32_t*)h->mvd_cache [list][scan8[0] - 1 + 2*8]=
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1088 *(uint32_t*)h->mvd_cache [list][scan8[0] - 1 + 3*8]= 0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1089 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1090 *(uint32_t*)h->mvd_cache [list][scan8[5 ]+1]=
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1091 *(uint32_t*)h->mvd_cache [list][scan8[7 ]+1]=
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1092 *(uint32_t*)h->mvd_cache [list][scan8[13]+1]= //FIXME remove past 3 (init somewhere else)
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1093 *(uint32_t*)h->mvd_cache [list][scan8[4 ]]=
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1094 *(uint32_t*)h->mvd_cache [list][scan8[12]]= 0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1095
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1096 if(h->slice_type_nos == FF_B_TYPE){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1097 fill_rectangle(&h->direct_cache[scan8[0]], 4, 4, 8, 0, 1);
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1098
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1099 if(IS_DIRECT(top_type)){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1100 *(uint32_t*)&h->direct_cache[scan8[0] - 1*8]= 0x01010101;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1101 }else if(IS_8X8(top_type)){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1102 int b8_xy = h->mb2b8_xy[top_xy] + h->b8_stride;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1103 h->direct_cache[scan8[0] + 0 - 1*8]= h->direct_table[b8_xy];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1104 h->direct_cache[scan8[0] + 2 - 1*8]= h->direct_table[b8_xy + 1];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1105 }else{
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1106 *(uint32_t*)&h->direct_cache[scan8[0] - 1*8]= 0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1107 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1108
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1109 if(IS_DIRECT(left_type[0]))
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1110 h->direct_cache[scan8[0] - 1 + 0*8]= 1;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1111 else if(IS_8X8(left_type[0]))
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1112 h->direct_cache[scan8[0] - 1 + 0*8]= h->direct_table[h->mb2b8_xy[left_xy[0]] + 1 + h->b8_stride*(left_block[0]>>1)];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1113 else
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1114 h->direct_cache[scan8[0] - 1 + 0*8]= 0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1115
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1116 if(IS_DIRECT(left_type[1]))
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1117 h->direct_cache[scan8[0] - 1 + 2*8]= 1;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1118 else if(IS_8X8(left_type[1]))
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1119 h->direct_cache[scan8[0] - 1 + 2*8]= h->direct_table[h->mb2b8_xy[left_xy[1]] + 1 + h->b8_stride*(left_block[2]>>1)];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1120 else
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1121 h->direct_cache[scan8[0] - 1 + 2*8]= 0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1122 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1123 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1124
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1125 if(FRAME_MBAFF){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1126 #define MAP_MVS\
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1127 MAP_F2F(scan8[0] - 1 - 1*8, topleft_type)\
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1128 MAP_F2F(scan8[0] + 0 - 1*8, top_type)\
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1129 MAP_F2F(scan8[0] + 1 - 1*8, top_type)\
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1130 MAP_F2F(scan8[0] + 2 - 1*8, top_type)\
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1131 MAP_F2F(scan8[0] + 3 - 1*8, top_type)\
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1132 MAP_F2F(scan8[0] + 4 - 1*8, topright_type)\
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1133 MAP_F2F(scan8[0] - 1 + 0*8, left_type[0])\
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1134 MAP_F2F(scan8[0] - 1 + 1*8, left_type[0])\
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1135 MAP_F2F(scan8[0] - 1 + 2*8, left_type[1])\
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1136 MAP_F2F(scan8[0] - 1 + 3*8, left_type[1])
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1137 if(MB_FIELD){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1138 #define MAP_F2F(idx, mb_type)\
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1139 if(!IS_INTERLACED(mb_type) && h->ref_cache[list][idx] >= 0){\
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1140 h->ref_cache[list][idx] <<= 1;\
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1141 h->mv_cache[list][idx][1] /= 2;\
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1142 h->mvd_cache[list][idx][1] /= 2;\
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1143 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1144 MAP_MVS
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1145 #undef MAP_F2F
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1146 }else{
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1147 #define MAP_F2F(idx, mb_type)\
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1148 if(IS_INTERLACED(mb_type) && h->ref_cache[list][idx] >= 0){\
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1149 h->ref_cache[list][idx] >>= 1;\
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1150 h->mv_cache[list][idx][1] <<= 1;\
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1151 h->mvd_cache[list][idx][1] <<= 1;\
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1152 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1153 MAP_MVS
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1154 #undef MAP_F2F
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1155 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1156 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1157 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1158 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1159 #endif
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1160
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1161 h->neighbor_transform_size= !!IS_8x8DCT(top_type) + !!IS_8x8DCT(left_type[0]);
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1162 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1163
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1164 /**
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1165 * gets the predicted intra4x4 prediction mode.
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1166 */
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1167 static inline int pred_intra_mode(H264Context *h, int n){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1168 const int index8= scan8[n];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1169 const int left= h->intra4x4_pred_mode_cache[index8 - 1];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1170 const int top = h->intra4x4_pred_mode_cache[index8 - 8];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1171 const int min= FFMIN(left, top);
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1172
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1173 tprintf(h->s.avctx, "mode:%d %d min:%d\n", left ,top, min);
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1174
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1175 if(min<0) return DC_PRED;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1176 else return min;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1177 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1178
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1179 static inline void write_back_non_zero_count(H264Context *h){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1180 const int mb_xy= h->mb_xy;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1181
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1182 h->non_zero_count[mb_xy][0]= h->non_zero_count_cache[7+8*1];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1183 h->non_zero_count[mb_xy][1]= h->non_zero_count_cache[7+8*2];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1184 h->non_zero_count[mb_xy][2]= h->non_zero_count_cache[7+8*3];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1185 h->non_zero_count[mb_xy][3]= h->non_zero_count_cache[7+8*4];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1186 h->non_zero_count[mb_xy][4]= h->non_zero_count_cache[4+8*4];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1187 h->non_zero_count[mb_xy][5]= h->non_zero_count_cache[5+8*4];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1188 h->non_zero_count[mb_xy][6]= h->non_zero_count_cache[6+8*4];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1189
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1190 h->non_zero_count[mb_xy][9]= h->non_zero_count_cache[1+8*2];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1191 h->non_zero_count[mb_xy][8]= h->non_zero_count_cache[2+8*2];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1192 h->non_zero_count[mb_xy][7]= h->non_zero_count_cache[2+8*1];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1193
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1194 h->non_zero_count[mb_xy][12]=h->non_zero_count_cache[1+8*5];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1195 h->non_zero_count[mb_xy][11]=h->non_zero_count_cache[2+8*5];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1196 h->non_zero_count[mb_xy][10]=h->non_zero_count_cache[2+8*4];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1197 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1198
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1199 static inline void write_back_motion(H264Context *h, int mb_type){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1200 MpegEncContext * const s = &h->s;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1201 const int b_xy = 4*s->mb_x + 4*s->mb_y*h->b_stride;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1202 const int b8_xy= 2*s->mb_x + 2*s->mb_y*h->b8_stride;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1203 int list;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1204
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1205 if(!USES_LIST(mb_type, 0))
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1206 fill_rectangle(&s->current_picture.ref_index[0][b8_xy], 2, 2, h->b8_stride, (uint8_t)LIST_NOT_USED, 1);
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1207
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1208 for(list=0; list<h->list_count; list++){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1209 int y;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1210 if(!USES_LIST(mb_type, list))
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1211 continue;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1212
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1213 for(y=0; y<4; y++){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1214 *(uint64_t*)s->current_picture.motion_val[list][b_xy + 0 + y*h->b_stride]= *(uint64_t*)h->mv_cache[list][scan8[0]+0 + 8*y];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1215 *(uint64_t*)s->current_picture.motion_val[list][b_xy + 2 + y*h->b_stride]= *(uint64_t*)h->mv_cache[list][scan8[0]+2 + 8*y];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1216 }
10893
2aafcafbe1f0 Replace cabac checks in inline functions from h264.h with constants.
michael
parents: 10883
diff changeset
1217 if( CABAC ) {
10866
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1218 if(IS_SKIP(mb_type))
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1219 fill_rectangle(h->mvd_table[list][b_xy], 4, 4, h->b_stride, 0, 4);
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1220 else
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1221 for(y=0; y<4; y++){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1222 *(uint64_t*)h->mvd_table[list][b_xy + 0 + y*h->b_stride]= *(uint64_t*)h->mvd_cache[list][scan8[0]+0 + 8*y];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1223 *(uint64_t*)h->mvd_table[list][b_xy + 2 + y*h->b_stride]= *(uint64_t*)h->mvd_cache[list][scan8[0]+2 + 8*y];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1224 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1225 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1226
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1227 {
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1228 int8_t *ref_index = &s->current_picture.ref_index[list][b8_xy];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1229 ref_index[0+0*h->b8_stride]= h->ref_cache[list][scan8[0]];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1230 ref_index[1+0*h->b8_stride]= h->ref_cache[list][scan8[4]];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1231 ref_index[0+1*h->b8_stride]= h->ref_cache[list][scan8[8]];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1232 ref_index[1+1*h->b8_stride]= h->ref_cache[list][scan8[12]];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1233 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1234 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1235
10893
2aafcafbe1f0 Replace cabac checks in inline functions from h264.h with constants.
michael
parents: 10883
diff changeset
1236 if(h->slice_type_nos == FF_B_TYPE && CABAC){
10866
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1237 if(IS_8X8(mb_type)){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1238 uint8_t *direct_table = &h->direct_table[b8_xy];
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1239 direct_table[1+0*h->b8_stride] = IS_DIRECT(h->sub_mb_type[1]) ? 1 : 0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1240 direct_table[0+1*h->b8_stride] = IS_DIRECT(h->sub_mb_type[2]) ? 1 : 0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1241 direct_table[1+1*h->b8_stride] = IS_DIRECT(h->sub_mb_type[3]) ? 1 : 0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1242 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1243 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1244 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1245
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1246 static inline int get_dct8x8_allowed(H264Context *h){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1247 if(h->sps.direct_8x8_inference_flag)
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1248 return !(*(uint64_t*)h->sub_mb_type & ((MB_TYPE_16x8|MB_TYPE_8x16|MB_TYPE_8x8 )*0x0001000100010001ULL));
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1249 else
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1250 return !(*(uint64_t*)h->sub_mb_type & ((MB_TYPE_16x8|MB_TYPE_8x16|MB_TYPE_8x8|MB_TYPE_DIRECT2)*0x0001000100010001ULL));
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1251 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1252
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1253 static void predict_field_decoding_flag(H264Context *h){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1254 MpegEncContext * const s = &h->s;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1255 const int mb_xy= h->mb_xy;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1256 int mb_type = (h->slice_table[mb_xy-1] == h->slice_num)
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1257 ? s->current_picture.mb_type[mb_xy-1]
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1258 : (h->slice_table[mb_xy-s->mb_stride] == h->slice_num)
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1259 ? s->current_picture.mb_type[mb_xy-s->mb_stride]
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1260 : 0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1261 h->mb_mbaff = h->mb_field_decoding_flag = IS_INTERLACED(mb_type) ? 1 : 0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1262 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1263
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1264 /**
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1265 * decodes a P_SKIP or B_SKIP macroblock
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1266 */
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1267 static void decode_mb_skip(H264Context *h){
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1268 MpegEncContext * const s = &h->s;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1269 const int mb_xy= h->mb_xy;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1270 int mb_type=0;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1271
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1272 memset(h->non_zero_count[mb_xy], 0, 16);
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1273 memset(h->non_zero_count_cache + 8, 0, 8*5); //FIXME ugly, remove pfui
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1274
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1275 if(MB_FIELD)
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1276 mb_type|= MB_TYPE_INTERLACED;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1277
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1278 if( h->slice_type_nos == FF_B_TYPE )
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1279 {
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1280 // just for fill_caches. pred_direct_motion will set the real mb_type
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1281 mb_type|= MB_TYPE_P0L0|MB_TYPE_P0L1|MB_TYPE_DIRECT2|MB_TYPE_SKIP;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1282
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1283 fill_caches(h, mb_type, 0); //FIXME check what is needed and what not ...
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1284 ff_h264_pred_direct_motion(h, &mb_type);
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1285 mb_type|= MB_TYPE_SKIP;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1286 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1287 else
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1288 {
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1289 int mx, my;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1290 mb_type|= MB_TYPE_16x16|MB_TYPE_P0L0|MB_TYPE_P1L0|MB_TYPE_SKIP;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1291
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1292 fill_caches(h, mb_type, 0); //FIXME check what is needed and what not ...
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1293 pred_pskip_motion(h, &mx, &my);
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1294 fill_rectangle(&h->ref_cache[0][scan8[0]], 4, 4, 8, 0, 1);
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1295 fill_rectangle( h->mv_cache[0][scan8[0]], 4, 4, 8, pack16to32(mx,my), 4);
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1296 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1297
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1298 write_back_motion(h, mb_type);
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1299 s->current_picture.mb_type[mb_xy]= mb_type;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1300 s->current_picture.qscale_table[mb_xy]= s->qscale;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1301 h->slice_table[ mb_xy ]= h->slice_num;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1302 h->prev_mb_skipped= 1;
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1303 }
d26e9b4d2ca1 Split cavlc out of h264.c.
michael
parents: 10864
diff changeset
1304
10883
4701b9355d93 Add forgotten include of h264_mvpred.h to h264.h.
michael
parents: 10869
diff changeset
1305 #include "h264_mvpred.h" //For pred_pskip_motion()
4701b9355d93 Add forgotten include of h264_mvpred.h to h264.h.
michael
parents: 10869
diff changeset
1306
7760
c4a4495715dd Globally rename the header inclusion guard names.
stefano
parents: 7526
diff changeset
1307 #endif /* AVCODEC_H264_H */