Mercurial > libavcodec.hg
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 |
rev | line source |
---|---|
4975 | 1 /* |
2 * H.26L/H.264/AVC/JVT/14496-10/... encoder/decoder | |
3 * Copyright (c) 2003 Michael Niedermayer <michaelni@gmx.at> | |
4 * | |
5 * This file is part of FFmpeg. | |
6 * | |
7 * FFmpeg is free software; you can redistribute it and/or | |
8 * modify it under the terms of the GNU Lesser General Public | |
9 * License as published by the Free Software Foundation; either | |
10 * version 2.1 of the License, or (at your option) any later version. | |
11 * | |
12 * FFmpeg is distributed in the hope that it will be useful, | |
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
15 * Lesser General Public License for more details. | |
16 * | |
17 * You should have received a copy of the GNU Lesser General Public | |
18 * License along with FFmpeg; if not, write to the Free Software | |
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |
20 */ | |
21 | |
22 /** | |
8718
e9d9d946f213
Use full internal pathname in doxygen @file directives.
diego
parents:
8596
diff
changeset
|
23 * @file libavcodec/h264.h |
4975 | 24 * H.264 / AVC / MPEG4 part10 codec. |
25 * @author Michael Niedermayer <michaelni@gmx.at> | |
26 */ | |
27 | |
7760 | 28 #ifndef AVCODEC_H264_H |
29 #define AVCODEC_H264_H | |
4975 | 30 |
31 #include "dsputil.h" | |
32 #include "cabac.h" | |
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 | 35 #include "rectangle.h" |
4975 | 36 |
37 #define interlaced_dct interlaced_dct_is_a_bad_name | |
5129 | 38 #define mb_intra mb_intra_is_not_initialized_see_mb_type |
4975 | 39 |
40 #define LUMA_DC_BLOCK_INDEX 25 | |
41 #define CHROMA_DC_BLOCK_INDEX 26 | |
42 | |
43 #define CHROMA_DC_COEFF_TOKEN_VLC_BITS 8 | |
44 #define COEFF_TOKEN_VLC_BITS 8 | |
45 #define TOTAL_ZEROS_VLC_BITS 9 | |
46 #define CHROMA_DC_TOTAL_ZEROS_VLC_BITS 3 | |
47 #define RUN_VLC_BITS 3 | |
48 #define RUN7_VLC_BITS 6 | |
49 | |
50 #define MAX_SPS_COUNT 32 | |
51 #define MAX_PPS_COUNT 256 | |
52 | |
53 #define MAX_MMCO_COUNT 66 | |
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 | 57 /* Compiling in interlaced support reduces the speed |
58 * of progressive decoding by about 2%. */ | |
59 #define ALLOW_INTERLACE | |
60 | |
7407 | 61 #define ALLOW_NOCHROMA |
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 | 69 #ifdef ALLOW_INTERLACE |
70 #define MB_MBAFF h->mb_mbaff | |
71 #define MB_FIELD h->mb_field_decoding_flag | |
72 #define FRAME_MBAFF h->mb_aff_frame | |
5801 | 73 #define FIELD_PICTURE (s->picture_structure != PICT_FRAME) |
4975 | 74 #else |
75 #define MB_MBAFF 0 | |
76 #define MB_FIELD 0 | |
77 #define FRAME_MBAFF 0 | |
5767 | 78 #define FIELD_PICTURE 0 |
4975 | 79 #undef IS_INTERLACED |
80 #define IS_INTERLACED(mb_type) 0 | |
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 | 83 |
7407 | 84 #ifdef ALLOW_NOCHROMA |
85 #define CHROMA h->sps.chroma_format_idc | |
86 #else | |
87 #define CHROMA 1 | |
88 #endif | |
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 | 110 NAL_SLICE=1, |
111 NAL_DPA, | |
112 NAL_DPB, | |
113 NAL_DPC, | |
114 NAL_IDR_SLICE, | |
115 NAL_SEI, | |
116 NAL_SPS, | |
117 NAL_PPS, | |
118 NAL_AUD, | |
119 NAL_END_SEQUENCE, | |
120 NAL_END_STREAM, | |
121 NAL_FILLER_DATA, | |
122 NAL_SPS_EXT, | |
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 | 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 | 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 | 152 * Sequence parameter set |
153 */ | |
154 typedef struct SPS{ | |
155 | |
156 int profile_idc; | |
157 int level_idc; | |
7407 | 158 int chroma_format_idc; |
4975 | 159 int transform_bypass; ///< qpprime_y_zero_transform_bypass_flag |
160 int log2_max_frame_num; ///< log2_max_frame_num_minus4 + 4 | |
161 int poc_type; ///< pic_order_cnt_type | |
162 int log2_max_poc_lsb; ///< log2_max_pic_order_cnt_lsb_minus4 | |
163 int delta_pic_order_always_zero_flag; | |
164 int offset_for_non_ref_pic; | |
165 int offset_for_top_to_bottom_field; | |
166 int poc_cycle_length; ///< num_ref_frames_in_pic_order_cnt_cycle | |
167 int ref_frame_count; ///< num_ref_frames | |
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 | 171 int frame_mbs_only_flag; |
172 int mb_aff; ///<mb_adaptive_frame_field_flag | |
173 int direct_8x8_inference_flag; | |
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 | 179 int vui_parameters_present_flag; |
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 | 187 int timing_info_present_flag; |
188 uint32_t num_units_in_tick; | |
189 uint32_t time_scale; | |
190 int fixed_frame_rate_flag; | |
191 short offset_for_ref_frame[256]; //FIXME dyn aloc? | |
192 int bitstream_restriction_flag; | |
193 int num_reorder_frames; | |
194 int scaling_matrix_present; | |
195 uint8_t scaling_matrix4[6][16]; | |
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 | 201 int cpb_cnt; ///< See H.264 E.1.2 |
8965 | 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 | 208 }SPS; |
209 | |
210 /** | |
211 * Picture parameter set | |
212 */ | |
213 typedef struct PPS{ | |
214 unsigned int sps_id; | |
215 int cabac; ///< entropy_coding_mode_flag | |
216 int pic_order_present; ///< pic_order_present_flag | |
217 int slice_group_count; ///< num_slice_groups_minus1 + 1 | |
218 int mb_slice_group_map_type; | |
219 unsigned int ref_count[2]; ///< num_ref_idx_l0/1_active_minus1 + 1 | |
220 int weighted_pred; ///< weighted_pred_flag | |
221 int weighted_bipred_idc; | |
222 int init_qp; ///< pic_init_qp_minus26 + 26 | |
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 | 225 int deblocking_filter_parameters_present; ///< deblocking_filter_parameters_present_flag |
226 int constrained_intra_pred; ///< constrained_intra_pred_flag | |
227 int redundant_pic_cnt_present; ///< redundant_pic_cnt_present_flag | |
228 int transform_8x8_mode; ///< transform_8x8_mode_flag | |
229 uint8_t scaling_matrix4[6][16]; | |
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 | 233 }PPS; |
234 | |
235 /** | |
236 * Memory management control operation opcode. | |
237 */ | |
238 typedef enum MMCOOpcode{ | |
239 MMCO_END=0, | |
240 MMCO_SHORT2UNUSED, | |
241 MMCO_LONG2UNUSED, | |
242 MMCO_SHORT2LONG, | |
243 MMCO_SET_MAX_LONG, | |
244 MMCO_RESET, | |
245 MMCO_LONG, | |
246 } MMCOOpcode; | |
247 | |
248 /** | |
249 * Memory management control operation. | |
250 */ | |
251 typedef struct MMCO{ | |
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 | 255 } MMCO; |
256 | |
257 /** | |
258 * H264Context | |
259 */ | |
260 typedef struct H264Context{ | |
261 MpegEncContext s; | |
262 int nal_ref_idc; | |
263 int nal_unit_type; | |
5174 | 264 uint8_t *rbsp_buffer[2]; |
265 unsigned int rbsp_buffer_size[2]; | |
4975 | 266 |
267 /** | |
268 * Used to parse AVC variant of h264 | |
269 */ | |
270 int is_avc; ///< this flag is != 0 if codec is avc1 | |
271 int got_avcC; ///< flag used to parse avcC data only once | |
272 int nal_length_size; ///< Number of bytes used for nal length (1, 2 or 4) | |
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 | 275 |
276 int prev_mb_skipped; | |
277 int next_mb_skipped; | |
278 | |
279 //prediction stuff | |
280 int chroma_pred_mode; | |
281 int intra16x16_pred_mode; | |
282 | |
283 int top_mb_xy; | |
284 int left_mb_xy[2]; | |
285 | |
286 int8_t intra4x4_pred_mode_cache[5*8]; | |
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 | 289 unsigned int topleft_samples_available; |
290 unsigned int top_samples_available; | |
291 unsigned int topright_samples_available; | |
292 unsigned int left_samples_available; | |
293 uint8_t (*top_borders[2])[16+2*8]; | |
294 uint8_t left_border[2*(17+2*9)]; | |
295 | |
296 /** | |
297 * non zero coeff count cache. | |
298 * is 64 if not available. | |
299 */ | |
300 DECLARE_ALIGNED_8(uint8_t, non_zero_count_cache[6*8]); | |
301 uint8_t (*non_zero_count)[16]; | |
302 | |
303 /** | |
304 * Motion vector cache. | |
305 */ | |
306 DECLARE_ALIGNED_8(int16_t, mv_cache[2][5*8][2]); | |
307 DECLARE_ALIGNED_8(int8_t, ref_cache[2][5*8]); | |
308 #define LIST_NOT_USED -1 //FIXME rename? | |
309 #define PART_NOT_AVAILABLE -2 | |
310 | |
311 /** | |
312 * is 1 if the specific list MV&references are set to 0,0,-2. | |
313 */ | |
314 int mv_cache_clean[2]; | |
315 | |
316 /** | |
317 * number of neighbors (top and/or left) that used 8x8 dct | |
318 */ | |
319 int neighbor_transform_size; | |
320 | |
321 /** | |
322 * block_offset[ 0..23] for frame macroblocks | |
323 * block_offset[24..47] for field macroblocks | |
324 */ | |
325 int block_offset[2*(16+8)]; | |
326 | |
327 uint32_t *mb2b_xy; //FIXME are these 4 a good idea? | |
328 uint32_t *mb2b8_xy; | |
329 int b_stride; //FIXME use s->b4_stride | |
330 int b8_stride; | |
331 | |
332 int mb_linesize; ///< may be equal to s->linesize or s->linesize*2, for mbaff | |
333 int mb_uvlinesize; | |
334 | |
335 int emu_edge_width; | |
336 int emu_edge_height; | |
337 | |
338 int halfpel_flag; | |
339 int thirdpel_flag; | |
340 | |
341 int unknown_svq3_flag; | |
342 int next_slice_index; | |
343 | |
5079 | 344 SPS *sps_buffers[MAX_SPS_COUNT]; |
4975 | 345 SPS sps; ///< current sps |
346 | |
5079 | 347 PPS *pps_buffers[MAX_PPS_COUNT]; |
4975 | 348 /** |
349 * current pps | |
350 */ | |
351 PPS pps; //FIXME move to Picture perhaps? (->no) do we need that? | |
352 | |
353 uint32_t dequant4_buffer[6][52][16]; | |
354 uint32_t dequant8_buffer[2][52][64]; | |
355 uint32_t (*dequant4_coeff[6])[16]; | |
356 uint32_t (*dequant8_coeff[2])[64]; | |
357 int dequant_coeff_pps; ///< reinit tables when pps changes | |
358 | |
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 | 362 int slice_type; |
7338 | 363 int slice_type_nos; ///< S free slice type (SI/SP are remapped to I/P) |
4975 | 364 int slice_type_fixed; |
365 | |
366 //interlacing specific flags | |
367 int mb_aff_frame; | |
368 int mb_field_decoding_flag; | |
369 int mb_mbaff; ///< mb_aff_frame && mb_field_decoding_flag | |
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 | 372 |
373 //POC stuff | |
374 int poc_lsb; | |
375 int poc_msb; | |
376 int delta_poc_bottom; | |
377 int delta_poc[2]; | |
378 int frame_num; | |
379 int prev_poc_msb; ///< poc_msb of the last reference pic for POC type 0 | |
380 int prev_poc_lsb; ///< poc_lsb of the last reference pic for POC type 0 | |
381 int frame_num_offset; ///< for POC type 2 | |
382 int prev_frame_num_offset; ///< for POC type 2 | |
383 int prev_frame_num; ///< frame_num of the last pic for POC type 1/2 | |
384 | |
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 | 387 */ |
388 int curr_pic_num; | |
389 | |
390 /** | |
391 * max_frame_num or 2*max_frame_num for field pics. | |
392 */ | |
393 int max_pic_num; | |
394 | |
395 //Weighted pred stuff | |
396 int use_weight; | |
397 int use_weight_chroma; | |
398 int luma_log2_weight_denom; | |
399 int chroma_log2_weight_denom; | |
400 int luma_weight[2][48]; | |
401 int luma_offset[2][48]; | |
402 int chroma_weight[2][48][2]; | |
403 int chroma_offset[2][48][2]; | |
404 int implicit_weight[48][48]; | |
405 | |
406 //deblock | |
407 int deblocking_filter; ///< disable_deblocking_filter_idc with 1<->0 | |
408 int slice_alpha_c0_offset; | |
409 int slice_beta_offset; | |
410 | |
411 int redundant_pic_count; | |
412 | |
413 int direct_spatial_mv_pred; | |
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 | 418 |
419 /** | |
420 * num_ref_idx_l0/1_active_minus1 + 1 | |
421 */ | |
422 unsigned int ref_count[2]; ///< counts frames or fields, depending on current mb mode | |
423 unsigned int list_count; | |
424 Picture *short_ref[32]; | |
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 | 433 |
434 /** | |
435 * memory management control operations buffer. | |
436 */ | |
437 MMCO mmco[MAX_MMCO_COUNT]; | |
438 int mmco_index; | |
439 | |
440 int long_ref_count; ///< number of actual long term references | |
441 int short_ref_count; ///< number of actual short term references | |
442 | |
443 //data partitioning | |
444 GetBitContext intra_gb; | |
445 GetBitContext inter_gb; | |
446 GetBitContext *intra_gb_ptr; | |
447 GetBitContext *inter_gb_ptr; | |
448 | |
6320 | 449 DECLARE_ALIGNED_16(DCTELEM, mb[16*24]); |
6488 | 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 | 451 |
452 /** | |
453 * Cabac | |
454 */ | |
455 CABACContext cabac; | |
456 uint8_t cabac_state[460]; | |
457 int cabac_init_idc; | |
458 | |
459 /* 0x100 -> non null luma_dc, 0x80/0x40 -> non null chroma_dc (cb/cr), 0x?0 -> chroma_cbp(0,1,2), 0x0? luma_cbp */ | |
460 uint16_t *cbp_table; | |
461 int cbp; | |
462 int top_cbp; | |
463 int left_cbp; | |
464 /* chroma_pred_mode for i4x4 or i16x16, else 0 */ | |
465 uint8_t *chroma_pred_mode_table; | |
466 int last_qscale_diff; | |
467 int16_t (*mvd_table[2])[2]; | |
468 DECLARE_ALIGNED_8(int16_t, mvd_cache[2][5*8][2]); | |
469 uint8_t *direct_table; | |
470 uint8_t direct_cache[5*8]; | |
471 | |
472 uint8_t zigzag_scan[16]; | |
473 uint8_t zigzag_scan8x8[64]; | |
474 uint8_t zigzag_scan8x8_cavlc[64]; | |
475 uint8_t field_scan[16]; | |
476 uint8_t field_scan8x8[64]; | |
477 uint8_t field_scan8x8_cavlc[64]; | |
478 const uint8_t *zigzag_scan_q0; | |
479 const uint8_t *zigzag_scan8x8_q0; | |
480 const uint8_t *zigzag_scan8x8_cavlc_q0; | |
481 const uint8_t *field_scan_q0; | |
482 const uint8_t *field_scan8x8_q0; | |
483 const uint8_t *field_scan8x8_cavlc_q0; | |
484 | |
485 int x264_build; | |
5642 | 486 |
487 /** | |
488 * @defgroup multithreading Members for slice based multithreading | |
489 * @{ | |
490 */ | |
491 struct H264Context *thread_context[MAX_THREADS]; | |
492 | |
493 /** | |
494 * current slice number, used to initalize slice_num of each thread/context | |
495 */ | |
496 int current_slice; | |
497 | |
498 /** | |
499 * Max number of threads / contexts. | |
500 * This is equal to AVCodecContext.thread_count unless | |
501 * multithreaded decoding is impossible, in which case it is | |
502 * reduced to 1. | |
503 */ | |
504 int max_contexts; | |
505 | |
506 /** | |
507 * 1 if the single thread fallback warning has already been | |
508 * displayed, 0 otherwise. | |
509 */ | |
510 int single_decode_warning; | |
511 | |
512 int last_slice_type; | |
513 /** @} */ | |
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 | 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 | 540 * dpb_output_delay in picture timing SEI message, see H.264 C.2.2 |
541 */ | |
542 int sei_dpb_output_delay; | |
543 | |
544 /** | |
8967 | 545 * cpb_removal_delay in picture timing SEI message, see H.264 C.1.2 |
546 */ | |
547 int sei_cpb_removal_delay; | |
548 | |
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 | 562 |
563 // Timestamp stuff | |
564 int sei_buffering_period_present; ///< Buffering period SEI flag | |
565 int initial_cpb_removal_delay[32]; ///< Initial timestamps for CPBs | |
4975 | 566 }H264Context; |
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 | 612 /** |
613 * allocates tables. | |
614 * needs width/height | |
615 */ | |
616 int ff_h264_alloc_tables(H264Context *h); | |
617 | |
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 | 636 * checks if the top & left blocks are available if needed & changes the dc mode so it only uses the available blocks. |
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 | 643 int ff_h264_check_intra_pred_mode(H264Context *h, int mode); |
644 | |
645 void ff_h264_write_back_intra_pred_mode(H264Context *h); | |
646 void ff_h264_hl_decode_mb(H264Context *h); | |
647 int ff_h264_frame_start(H264Context *h); | |
648 av_cold int ff_h264_decode_init(AVCodecContext *avctx); | |
649 av_cold int ff_h264_decode_end(AVCodecContext *avctx); | |
10866 | 650 av_cold void ff_h264_decode_init_vlc(void); |
651 | |
652 /** | |
653 * decodes a macroblock | |
654 * @returns 0 if OK, AC_ERROR / DC_ERROR / MV_ERROR if an error is noticed | |
655 */ | |
656 int ff_h264_decode_mb_cavlc(H264Context *h); | |
10852 | 657 |
10869 | 658 /** |
659 * decodes a CABAC coded macroblock | |
660 * @returns 0 if OK, AC_ERROR / DC_ERROR / MV_ERROR if an error is noticed | |
661 */ | |
662 int ff_h264_decode_mb_cabac(H264Context *h); | |
663 | |
664 void ff_h264_init_cabac_states(H264Context *h); | |
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 | 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); |
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); | |
672 | |
10858 | 673 /** |
674 * Reset SEI values at the beginning of the frame. | |
675 * | |
676 * @param h H.264 context. | |
677 */ | |
678 void ff_h264_reset_sei(H264Context *h); | |
679 | |
680 | |
10852 | 681 /* |
682 o-o o-o | |
683 / / / | |
684 o-o o-o | |
685 ,---' | |
686 o-o o-o | |
687 / / / | |
688 o-o o-o | |
689 */ | |
690 //This table must be here because scan8[constant] must be known at compiletime | |
691 static const uint8_t scan8[16 + 2*4]={ | |
692 4+1*8, 5+1*8, 4+2*8, 5+2*8, | |
693 6+1*8, 7+1*8, 6+2*8, 7+2*8, | |
694 4+3*8, 5+3*8, 4+4*8, 5+4*8, | |
695 6+3*8, 7+3*8, 6+4*8, 7+4*8, | |
696 1+1*8, 2+1*8, | |
697 1+2*8, 2+2*8, | |
698 1+4*8, 2+4*8, | |
699 1+5*8, 2+5*8, | |
700 }; | |
701 | |
702 static av_always_inline uint32_t pack16to32(int a, int b){ | |
703 #if HAVE_BIGENDIAN | |
704 return (b&0xFFFF) + (a<<16); | |
705 #else | |
706 return (a&0xFFFF) + (b<<16); | |
707 #endif | |
708 } | |
709 | |
710 /** | |
10854 | 711 * gets the chroma qp. |
712 */ | |
713 static inline int get_chroma_qp(H264Context *h, int t, int qscale){ | |
714 return h->pps.chroma_qp_table[t][qscale]; | |
715 } | |
716 | |
10866 | 717 static inline void pred_pskip_motion(H264Context * const h, int * const mx, int * const my); |
718 | |
719 static void fill_caches(H264Context *h, int mb_type, int for_deblock){ | |
720 MpegEncContext * const s = &h->s; | |
721 const int mb_xy= h->mb_xy; | |
722 int topleft_xy, top_xy, topright_xy, left_xy[2]; | |
723 int topleft_type, top_type, topright_type, left_type[2]; | |
724 const uint8_t * left_block; | |
725 int topleft_partition= -1; | |
726 int i; | |
727 static const uint8_t left_block_options[4][8]={ | |
728 {0,1,2,3,7,10,8,11}, | |
729 {2,2,3,3,8,11,8,11}, | |
730 {0,0,1,1,7,10,7,10}, | |
731 {0,2,0,2,7,10,7,10} | |
732 }; | |
733 | |
734 top_xy = mb_xy - (s->mb_stride << FIELD_PICTURE); | |
735 | |
736 //FIXME deblocking could skip the intra and nnz parts. | |
737 if(for_deblock && (h->slice_num == 1 || h->slice_table[mb_xy] == h->slice_table[top_xy]) && !FRAME_MBAFF) | |
738 return; | |
739 | |
740 /* Wow, what a mess, why didn't they simplify the interlacing & intra | |
741 * stuff, I can't imagine that these complex rules are worth it. */ | |
742 | |
743 topleft_xy = top_xy - 1; | |
744 topright_xy= top_xy + 1; | |
745 left_xy[1] = left_xy[0] = mb_xy-1; | |
746 left_block = left_block_options[0]; | |
747 if(FRAME_MBAFF){ | |
748 const int pair_xy = s->mb_x + (s->mb_y & ~1)*s->mb_stride; | |
749 const int top_pair_xy = pair_xy - s->mb_stride; | |
750 const int topleft_pair_xy = top_pair_xy - 1; | |
751 const int topright_pair_xy = top_pair_xy + 1; | |
752 const int topleft_mb_field_flag = IS_INTERLACED(s->current_picture.mb_type[topleft_pair_xy]); | |
753 const int top_mb_field_flag = IS_INTERLACED(s->current_picture.mb_type[top_pair_xy]); | |
754 const int topright_mb_field_flag = IS_INTERLACED(s->current_picture.mb_type[topright_pair_xy]); | |
755 const int left_mb_field_flag = IS_INTERLACED(s->current_picture.mb_type[pair_xy-1]); | |
756 const int curr_mb_field_flag = IS_INTERLACED(mb_type); | |
757 const int bottom = (s->mb_y & 1); | |
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); | |
759 | |
760 if (curr_mb_field_flag && (bottom || top_mb_field_flag)){ | |
761 top_xy -= s->mb_stride; | |
762 } | |
763 if (curr_mb_field_flag && (bottom || topleft_mb_field_flag)){ | |
764 topleft_xy -= s->mb_stride; | |
765 } else if(bottom && !curr_mb_field_flag && left_mb_field_flag) { | |
766 topleft_xy += s->mb_stride; | |
767 // take top left mv from the middle of the mb, as opposed to all other modes which use the bottom right partition | |
768 topleft_partition = 0; | |
769 } | |
770 if (curr_mb_field_flag && (bottom || topright_mb_field_flag)){ | |
771 topright_xy -= s->mb_stride; | |
772 } | |
773 if (left_mb_field_flag != curr_mb_field_flag) { | |
774 left_xy[1] = left_xy[0] = pair_xy - 1; | |
775 if (curr_mb_field_flag) { | |
776 left_xy[1] += s->mb_stride; | |
777 left_block = left_block_options[3]; | |
778 } else { | |
779 left_block= left_block_options[2 - bottom]; | |
780 } | |
781 } | |
782 } | |
783 | |
784 h->top_mb_xy = top_xy; | |
785 h->left_mb_xy[0] = left_xy[0]; | |
786 h->left_mb_xy[1] = left_xy[1]; | |
787 if(for_deblock){ | |
788 topleft_type = 0; | |
789 topright_type = 0; | |
790 top_type = h->slice_table[top_xy ] < 0xFFFF ? s->current_picture.mb_type[top_xy] : 0; | |
791 left_type[0] = h->slice_table[left_xy[0] ] < 0xFFFF ? s->current_picture.mb_type[left_xy[0]] : 0; | |
792 left_type[1] = h->slice_table[left_xy[1] ] < 0xFFFF ? s->current_picture.mb_type[left_xy[1]] : 0; | |
793 | |
794 if(MB_MBAFF && !IS_INTRA(mb_type)){ | |
795 int list; | |
796 for(list=0; list<h->list_count; list++){ | |
797 //These values where changed for ease of performing MC, we need to change them back | |
798 //FIXME maybe we can make MC and loop filter use the same values or prevent | |
799 //the MC code from changing ref_cache and rather use a temporary array. | |
800 if(USES_LIST(mb_type,list)){ | |
801 int8_t *ref = &s->current_picture.ref_index[list][h->mb2b8_xy[mb_xy]]; | |
802 *(uint32_t*)&h->ref_cache[list][scan8[ 0]] = | |
803 *(uint32_t*)&h->ref_cache[list][scan8[ 2]] = (pack16to32(ref[0],ref[1])&0x00FF00FF)*0x0101; | |
804 ref += h->b8_stride; | |
805 *(uint32_t*)&h->ref_cache[list][scan8[ 8]] = | |
806 *(uint32_t*)&h->ref_cache[list][scan8[10]] = (pack16to32(ref[0],ref[1])&0x00FF00FF)*0x0101; | |
807 } | |
808 } | |
809 } | |
810 }else{ | |
811 topleft_type = h->slice_table[topleft_xy ] == h->slice_num ? s->current_picture.mb_type[topleft_xy] : 0; | |
812 top_type = h->slice_table[top_xy ] == h->slice_num ? s->current_picture.mb_type[top_xy] : 0; | |
813 topright_type= h->slice_table[topright_xy] == h->slice_num ? s->current_picture.mb_type[topright_xy]: 0; | |
814 left_type[0] = h->slice_table[left_xy[0] ] == h->slice_num ? s->current_picture.mb_type[left_xy[0]] : 0; | |
815 left_type[1] = h->slice_table[left_xy[1] ] == h->slice_num ? s->current_picture.mb_type[left_xy[1]] : 0; | |
816 | |
817 if(IS_INTRA(mb_type)){ | |
818 int type_mask= h->pps.constrained_intra_pred ? IS_INTRA(-1) : -1; | |
819 h->topleft_samples_available= | |
820 h->top_samples_available= | |
821 h->left_samples_available= 0xFFFF; | |
822 h->topright_samples_available= 0xEEEA; | |
823 | |
824 if(!(top_type & type_mask)){ | |
825 h->topleft_samples_available= 0xB3FF; | |
826 h->top_samples_available= 0x33FF; | |
827 h->topright_samples_available= 0x26EA; | |
828 } | |
829 if(IS_INTERLACED(mb_type) != IS_INTERLACED(left_type[0])){ | |
830 if(IS_INTERLACED(mb_type)){ | |
831 if(!(left_type[0] & type_mask)){ | |
832 h->topleft_samples_available&= 0xDFFF; | |
833 h->left_samples_available&= 0x5FFF; | |
834 } | |
835 if(!(left_type[1] & type_mask)){ | |
836 h->topleft_samples_available&= 0xFF5F; | |
837 h->left_samples_available&= 0xFF5F; | |
838 } | |
839 }else{ | |
840 int left_typei = h->slice_table[left_xy[0] + s->mb_stride ] == h->slice_num | |
841 ? s->current_picture.mb_type[left_xy[0] + s->mb_stride] : 0; | |
842 assert(left_xy[0] == left_xy[1]); | |
843 if(!((left_typei & type_mask) && (left_type[0] & type_mask))){ | |
844 h->topleft_samples_available&= 0xDF5F; | |
845 h->left_samples_available&= 0x5F5F; | |
846 } | |
847 } | |
848 }else{ | |
849 if(!(left_type[0] & type_mask)){ | |
850 h->topleft_samples_available&= 0xDF5F; | |
851 h->left_samples_available&= 0x5F5F; | |
852 } | |
853 } | |
854 | |
855 if(!(topleft_type & type_mask)) | |
856 h->topleft_samples_available&= 0x7FFF; | |
857 | |
858 if(!(topright_type & type_mask)) | |
859 h->topright_samples_available&= 0xFBFF; | |
860 | |
861 if(IS_INTRA4x4(mb_type)){ | |
862 if(IS_INTRA4x4(top_type)){ | |
863 h->intra4x4_pred_mode_cache[4+8*0]= h->intra4x4_pred_mode[top_xy][4]; | |
864 h->intra4x4_pred_mode_cache[5+8*0]= h->intra4x4_pred_mode[top_xy][5]; | |
865 h->intra4x4_pred_mode_cache[6+8*0]= h->intra4x4_pred_mode[top_xy][6]; | |
866 h->intra4x4_pred_mode_cache[7+8*0]= h->intra4x4_pred_mode[top_xy][3]; | |
867 }else{ | |
868 int pred; | |
869 if(!(top_type & type_mask)) | |
870 pred= -1; | |
871 else{ | |
872 pred= 2; | |
873 } | |
874 h->intra4x4_pred_mode_cache[4+8*0]= | |
875 h->intra4x4_pred_mode_cache[5+8*0]= | |
876 h->intra4x4_pred_mode_cache[6+8*0]= | |
877 h->intra4x4_pred_mode_cache[7+8*0]= pred; | |
878 } | |
879 for(i=0; i<2; i++){ | |
880 if(IS_INTRA4x4(left_type[i])){ | |
881 h->intra4x4_pred_mode_cache[3+8*1 + 2*8*i]= h->intra4x4_pred_mode[left_xy[i]][left_block[0+2*i]]; | |
882 h->intra4x4_pred_mode_cache[3+8*2 + 2*8*i]= h->intra4x4_pred_mode[left_xy[i]][left_block[1+2*i]]; | |
883 }else{ | |
884 int pred; | |
885 if(!(left_type[i] & type_mask)) | |
886 pred= -1; | |
887 else{ | |
888 pred= 2; | |
889 } | |
890 h->intra4x4_pred_mode_cache[3+8*1 + 2*8*i]= | |
891 h->intra4x4_pred_mode_cache[3+8*2 + 2*8*i]= pred; | |
892 } | |
893 } | |
894 } | |
895 } | |
896 } | |
897 | |
898 | |
899 /* | |
900 0 . T T. T T T T | |
901 1 L . .L . . . . | |
902 2 L . .L . . . . | |
903 3 . T TL . . . . | |
904 4 L . .L . . . . | |
905 5 L . .. . . . . | |
906 */ | |
907 //FIXME constraint_intra_pred & partitioning & nnz (let us hope this is just a typo in the spec) | |
908 if(top_type){ | |
909 h->non_zero_count_cache[4+8*0]= h->non_zero_count[top_xy][4]; | |
910 h->non_zero_count_cache[5+8*0]= h->non_zero_count[top_xy][5]; | |
911 h->non_zero_count_cache[6+8*0]= h->non_zero_count[top_xy][6]; | |
912 h->non_zero_count_cache[7+8*0]= h->non_zero_count[top_xy][3]; | |
913 | |
914 h->non_zero_count_cache[1+8*0]= h->non_zero_count[top_xy][9]; | |
915 h->non_zero_count_cache[2+8*0]= h->non_zero_count[top_xy][8]; | |
916 | |
917 h->non_zero_count_cache[1+8*3]= h->non_zero_count[top_xy][12]; | |
918 h->non_zero_count_cache[2+8*3]= h->non_zero_count[top_xy][11]; | |
919 | |
920 }else{ | |
921 h->non_zero_count_cache[4+8*0]= | |
922 h->non_zero_count_cache[5+8*0]= | |
923 h->non_zero_count_cache[6+8*0]= | |
924 h->non_zero_count_cache[7+8*0]= | |
925 | |
926 h->non_zero_count_cache[1+8*0]= | |
927 h->non_zero_count_cache[2+8*0]= | |
928 | |
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 | 931 |
932 } | |
933 | |
934 for (i=0; i<2; i++) { | |
935 if(left_type[i]){ | |
936 h->non_zero_count_cache[3+8*1 + 2*8*i]= h->non_zero_count[left_xy[i]][left_block[0+2*i]]; | |
937 h->non_zero_count_cache[3+8*2 + 2*8*i]= h->non_zero_count[left_xy[i]][left_block[1+2*i]]; | |
938 h->non_zero_count_cache[0+8*1 + 8*i]= h->non_zero_count[left_xy[i]][left_block[4+2*i]]; | |
939 h->non_zero_count_cache[0+8*4 + 8*i]= h->non_zero_count[left_xy[i]][left_block[5+2*i]]; | |
940 }else{ | |
941 h->non_zero_count_cache[3+8*1 + 2*8*i]= | |
942 h->non_zero_count_cache[3+8*2 + 2*8*i]= | |
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 | 945 } |
946 } | |
947 | |
10893
2aafcafbe1f0
Replace cabac checks in inline functions from h264.h with constants.
michael
parents:
10883
diff
changeset
|
948 if( CABAC ) { |
10866 | 949 // top_cbp |
950 if(top_type) { | |
951 h->top_cbp = h->cbp_table[top_xy]; | |
952 } else if(IS_INTRA(mb_type)) { | |
953 h->top_cbp = 0x1C0; | |
954 } else { | |
955 h->top_cbp = 0; | |
956 } | |
957 // left_cbp | |
958 if (left_type[0]) { | |
959 h->left_cbp = h->cbp_table[left_xy[0]] & 0x1f0; | |
960 } else if(IS_INTRA(mb_type)) { | |
961 h->left_cbp = 0x1C0; | |
962 } else { | |
963 h->left_cbp = 0; | |
964 } | |
965 if (left_type[0]) { | |
966 h->left_cbp |= ((h->cbp_table[left_xy[0]]>>((left_block[0]&(~1))+1))&0x1) << 1; | |
967 } | |
968 if (left_type[1]) { | |
969 h->left_cbp |= ((h->cbp_table[left_xy[1]]>>((left_block[2]&(~1))+1))&0x1) << 3; | |
970 } | |
971 } | |
972 | |
973 #if 1 | |
974 if(IS_INTER(mb_type) || IS_DIRECT(mb_type)){ | |
975 int list; | |
976 for(list=0; list<h->list_count; list++){ | |
977 if(!USES_LIST(mb_type, list) && !IS_DIRECT(mb_type) && !h->deblocking_filter){ | |
978 /*if(!h->mv_cache_clean[list]){ | |
979 memset(h->mv_cache [list], 0, 8*5*2*sizeof(int16_t)); //FIXME clean only input? clean at all? | |
980 memset(h->ref_cache[list], PART_NOT_AVAILABLE, 8*5*sizeof(int8_t)); | |
981 h->mv_cache_clean[list]= 1; | |
982 }*/ | |
983 continue; | |
984 } | |
985 h->mv_cache_clean[list]= 0; | |
986 | |
987 if(USES_LIST(top_type, list)){ | |
988 const int b_xy= h->mb2b_xy[top_xy] + 3*h->b_stride; | |
989 const int b8_xy= h->mb2b8_xy[top_xy] + h->b8_stride; | |
990 *(uint32_t*)h->mv_cache[list][scan8[0] + 0 - 1*8]= *(uint32_t*)s->current_picture.motion_val[list][b_xy + 0]; | |
991 *(uint32_t*)h->mv_cache[list][scan8[0] + 1 - 1*8]= *(uint32_t*)s->current_picture.motion_val[list][b_xy + 1]; | |
992 *(uint32_t*)h->mv_cache[list][scan8[0] + 2 - 1*8]= *(uint32_t*)s->current_picture.motion_val[list][b_xy + 2]; | |
993 *(uint32_t*)h->mv_cache[list][scan8[0] + 3 - 1*8]= *(uint32_t*)s->current_picture.motion_val[list][b_xy + 3]; | |
994 h->ref_cache[list][scan8[0] + 0 - 1*8]= | |
995 h->ref_cache[list][scan8[0] + 1 - 1*8]= s->current_picture.ref_index[list][b8_xy + 0]; | |
996 h->ref_cache[list][scan8[0] + 2 - 1*8]= | |
997 h->ref_cache[list][scan8[0] + 3 - 1*8]= s->current_picture.ref_index[list][b8_xy + 1]; | |
998 }else{ | |
999 *(uint32_t*)h->mv_cache [list][scan8[0] + 0 - 1*8]= | |
1000 *(uint32_t*)h->mv_cache [list][scan8[0] + 1 - 1*8]= | |
1001 *(uint32_t*)h->mv_cache [list][scan8[0] + 2 - 1*8]= | |
1002 *(uint32_t*)h->mv_cache [list][scan8[0] + 3 - 1*8]= 0; | |
1003 *(uint32_t*)&h->ref_cache[list][scan8[0] + 0 - 1*8]= ((top_type ? LIST_NOT_USED : PART_NOT_AVAILABLE)&0xFF)*0x01010101; | |
1004 } | |
1005 | |
1006 for(i=0; i<2; i++){ | |
1007 int cache_idx = scan8[0] - 1 + i*2*8; | |
1008 if(USES_LIST(left_type[i], list)){ | |
1009 const int b_xy= h->mb2b_xy[left_xy[i]] + 3; | |
1010 const int b8_xy= h->mb2b8_xy[left_xy[i]] + 1; | |
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]]; | |
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]]; | |
1013 h->ref_cache[list][cache_idx ]= s->current_picture.ref_index[list][b8_xy + h->b8_stride*(left_block[0+i*2]>>1)]; | |
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)]; | |
1015 }else{ | |
1016 *(uint32_t*)h->mv_cache [list][cache_idx ]= | |
1017 *(uint32_t*)h->mv_cache [list][cache_idx+8]= 0; | |
1018 h->ref_cache[list][cache_idx ]= | |
1019 h->ref_cache[list][cache_idx+8]= left_type[i] ? LIST_NOT_USED : PART_NOT_AVAILABLE; | |
1020 } | |
1021 } | |
1022 | |
1023 if(for_deblock || ((IS_DIRECT(mb_type) && !h->direct_spatial_mv_pred) && !FRAME_MBAFF)) | |
1024 continue; | |
1025 | |
1026 if(USES_LIST(topleft_type, list)){ | |
1027 const int b_xy = h->mb2b_xy[topleft_xy] + 3 + h->b_stride + (topleft_partition & 2*h->b_stride); | |
1028 const int b8_xy= h->mb2b8_xy[topleft_xy] + 1 + (topleft_partition & h->b8_stride); | |
1029 *(uint32_t*)h->mv_cache[list][scan8[0] - 1 - 1*8]= *(uint32_t*)s->current_picture.motion_val[list][b_xy]; | |
1030 h->ref_cache[list][scan8[0] - 1 - 1*8]= s->current_picture.ref_index[list][b8_xy]; | |
1031 }else{ | |
1032 *(uint32_t*)h->mv_cache[list][scan8[0] - 1 - 1*8]= 0; | |
1033 h->ref_cache[list][scan8[0] - 1 - 1*8]= topleft_type ? LIST_NOT_USED : PART_NOT_AVAILABLE; | |
1034 } | |
1035 | |
1036 if(USES_LIST(topright_type, list)){ | |
1037 const int b_xy= h->mb2b_xy[topright_xy] + 3*h->b_stride; | |
1038 const int b8_xy= h->mb2b8_xy[topright_xy] + h->b8_stride; | |
1039 *(uint32_t*)h->mv_cache[list][scan8[0] + 4 - 1*8]= *(uint32_t*)s->current_picture.motion_val[list][b_xy]; | |
1040 h->ref_cache[list][scan8[0] + 4 - 1*8]= s->current_picture.ref_index[list][b8_xy]; | |
1041 }else{ | |
1042 *(uint32_t*)h->mv_cache [list][scan8[0] + 4 - 1*8]= 0; | |
1043 h->ref_cache[list][scan8[0] + 4 - 1*8]= topright_type ? LIST_NOT_USED : PART_NOT_AVAILABLE; | |
1044 } | |
1045 | |
1046 if((IS_SKIP(mb_type) || IS_DIRECT(mb_type)) && !FRAME_MBAFF) | |
1047 continue; | |
1048 | |
1049 h->ref_cache[list][scan8[5 ]+1] = | |
1050 h->ref_cache[list][scan8[7 ]+1] = | |
1051 h->ref_cache[list][scan8[13]+1] = //FIXME remove past 3 (init somewhere else) | |
1052 h->ref_cache[list][scan8[4 ]] = | |
1053 h->ref_cache[list][scan8[12]] = PART_NOT_AVAILABLE; | |
1054 *(uint32_t*)h->mv_cache [list][scan8[5 ]+1]= | |
1055 *(uint32_t*)h->mv_cache [list][scan8[7 ]+1]= | |
1056 *(uint32_t*)h->mv_cache [list][scan8[13]+1]= //FIXME remove past 3 (init somewhere else) | |
1057 *(uint32_t*)h->mv_cache [list][scan8[4 ]]= | |
1058 *(uint32_t*)h->mv_cache [list][scan8[12]]= 0; | |
1059 | |
10893
2aafcafbe1f0
Replace cabac checks in inline functions from h264.h with constants.
michael
parents:
10883
diff
changeset
|
1060 if( CABAC ) { |
10866 | 1061 /* XXX beurk, Load mvd */ |
1062 if(USES_LIST(top_type, list)){ | |
1063 const int b_xy= h->mb2b_xy[top_xy] + 3*h->b_stride; | |
1064 *(uint32_t*)h->mvd_cache[list][scan8[0] + 0 - 1*8]= *(uint32_t*)h->mvd_table[list][b_xy + 0]; | |
1065 *(uint32_t*)h->mvd_cache[list][scan8[0] + 1 - 1*8]= *(uint32_t*)h->mvd_table[list][b_xy + 1]; | |
1066 *(uint32_t*)h->mvd_cache[list][scan8[0] + 2 - 1*8]= *(uint32_t*)h->mvd_table[list][b_xy + 2]; | |
1067 *(uint32_t*)h->mvd_cache[list][scan8[0] + 3 - 1*8]= *(uint32_t*)h->mvd_table[list][b_xy + 3]; | |
1068 }else{ | |
1069 *(uint32_t*)h->mvd_cache [list][scan8[0] + 0 - 1*8]= | |
1070 *(uint32_t*)h->mvd_cache [list][scan8[0] + 1 - 1*8]= | |
1071 *(uint32_t*)h->mvd_cache [list][scan8[0] + 2 - 1*8]= | |
1072 *(uint32_t*)h->mvd_cache [list][scan8[0] + 3 - 1*8]= 0; | |
1073 } | |
1074 if(USES_LIST(left_type[0], list)){ | |
1075 const int b_xy= h->mb2b_xy[left_xy[0]] + 3; | |
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]]; | |
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]]; | |
1078 }else{ | |
1079 *(uint32_t*)h->mvd_cache [list][scan8[0] - 1 + 0*8]= | |
1080 *(uint32_t*)h->mvd_cache [list][scan8[0] - 1 + 1*8]= 0; | |
1081 } | |
1082 if(USES_LIST(left_type[1], list)){ | |
1083 const int b_xy= h->mb2b_xy[left_xy[1]] + 3; | |
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]]; | |
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]]; | |
1086 }else{ | |
1087 *(uint32_t*)h->mvd_cache [list][scan8[0] - 1 + 2*8]= | |
1088 *(uint32_t*)h->mvd_cache [list][scan8[0] - 1 + 3*8]= 0; | |
1089 } | |
1090 *(uint32_t*)h->mvd_cache [list][scan8[5 ]+1]= | |
1091 *(uint32_t*)h->mvd_cache [list][scan8[7 ]+1]= | |
1092 *(uint32_t*)h->mvd_cache [list][scan8[13]+1]= //FIXME remove past 3 (init somewhere else) | |
1093 *(uint32_t*)h->mvd_cache [list][scan8[4 ]]= | |
1094 *(uint32_t*)h->mvd_cache [list][scan8[12]]= 0; | |
1095 | |
1096 if(h->slice_type_nos == FF_B_TYPE){ | |
1097 fill_rectangle(&h->direct_cache[scan8[0]], 4, 4, 8, 0, 1); | |
1098 | |
1099 if(IS_DIRECT(top_type)){ | |
1100 *(uint32_t*)&h->direct_cache[scan8[0] - 1*8]= 0x01010101; | |
1101 }else if(IS_8X8(top_type)){ | |
1102 int b8_xy = h->mb2b8_xy[top_xy] + h->b8_stride; | |
1103 h->direct_cache[scan8[0] + 0 - 1*8]= h->direct_table[b8_xy]; | |
1104 h->direct_cache[scan8[0] + 2 - 1*8]= h->direct_table[b8_xy + 1]; | |
1105 }else{ | |
1106 *(uint32_t*)&h->direct_cache[scan8[0] - 1*8]= 0; | |
1107 } | |
1108 | |
1109 if(IS_DIRECT(left_type[0])) | |
1110 h->direct_cache[scan8[0] - 1 + 0*8]= 1; | |
1111 else if(IS_8X8(left_type[0])) | |
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)]; | |
1113 else | |
1114 h->direct_cache[scan8[0] - 1 + 0*8]= 0; | |
1115 | |
1116 if(IS_DIRECT(left_type[1])) | |
1117 h->direct_cache[scan8[0] - 1 + 2*8]= 1; | |
1118 else if(IS_8X8(left_type[1])) | |
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)]; | |
1120 else | |
1121 h->direct_cache[scan8[0] - 1 + 2*8]= 0; | |
1122 } | |
1123 } | |
1124 | |
1125 if(FRAME_MBAFF){ | |
1126 #define MAP_MVS\ | |
1127 MAP_F2F(scan8[0] - 1 - 1*8, topleft_type)\ | |
1128 MAP_F2F(scan8[0] + 0 - 1*8, top_type)\ | |
1129 MAP_F2F(scan8[0] + 1 - 1*8, top_type)\ | |
1130 MAP_F2F(scan8[0] + 2 - 1*8, top_type)\ | |
1131 MAP_F2F(scan8[0] + 3 - 1*8, top_type)\ | |
1132 MAP_F2F(scan8[0] + 4 - 1*8, topright_type)\ | |
1133 MAP_F2F(scan8[0] - 1 + 0*8, left_type[0])\ | |
1134 MAP_F2F(scan8[0] - 1 + 1*8, left_type[0])\ | |
1135 MAP_F2F(scan8[0] - 1 + 2*8, left_type[1])\ | |
1136 MAP_F2F(scan8[0] - 1 + 3*8, left_type[1]) | |
1137 if(MB_FIELD){ | |
1138 #define MAP_F2F(idx, mb_type)\ | |
1139 if(!IS_INTERLACED(mb_type) && h->ref_cache[list][idx] >= 0){\ | |
1140 h->ref_cache[list][idx] <<= 1;\ | |
1141 h->mv_cache[list][idx][1] /= 2;\ | |
1142 h->mvd_cache[list][idx][1] /= 2;\ | |
1143 } | |
1144 MAP_MVS | |
1145 #undef MAP_F2F | |
1146 }else{ | |
1147 #define MAP_F2F(idx, mb_type)\ | |
1148 if(IS_INTERLACED(mb_type) && h->ref_cache[list][idx] >= 0){\ | |
1149 h->ref_cache[list][idx] >>= 1;\ | |
1150 h->mv_cache[list][idx][1] <<= 1;\ | |
1151 h->mvd_cache[list][idx][1] <<= 1;\ | |
1152 } | |
1153 MAP_MVS | |
1154 #undef MAP_F2F | |
1155 } | |
1156 } | |
1157 } | |
1158 } | |
1159 #endif | |
1160 | |
1161 h->neighbor_transform_size= !!IS_8x8DCT(top_type) + !!IS_8x8DCT(left_type[0]); | |
1162 } | |
1163 | |
1164 /** | |
1165 * gets the predicted intra4x4 prediction mode. | |
1166 */ | |
1167 static inline int pred_intra_mode(H264Context *h, int n){ | |
1168 const int index8= scan8[n]; | |
1169 const int left= h->intra4x4_pred_mode_cache[index8 - 1]; | |
1170 const int top = h->intra4x4_pred_mode_cache[index8 - 8]; | |
1171 const int min= FFMIN(left, top); | |
1172 | |
1173 tprintf(h->s.avctx, "mode:%d %d min:%d\n", left ,top, min); | |
1174 | |
1175 if(min<0) return DC_PRED; | |
1176 else return min; | |
1177 } | |
1178 | |
1179 static inline void write_back_non_zero_count(H264Context *h){ | |
1180 const int mb_xy= h->mb_xy; | |
1181 | |
1182 h->non_zero_count[mb_xy][0]= h->non_zero_count_cache[7+8*1]; | |
1183 h->non_zero_count[mb_xy][1]= h->non_zero_count_cache[7+8*2]; | |
1184 h->non_zero_count[mb_xy][2]= h->non_zero_count_cache[7+8*3]; | |
1185 h->non_zero_count[mb_xy][3]= h->non_zero_count_cache[7+8*4]; | |
1186 h->non_zero_count[mb_xy][4]= h->non_zero_count_cache[4+8*4]; | |
1187 h->non_zero_count[mb_xy][5]= h->non_zero_count_cache[5+8*4]; | |
1188 h->non_zero_count[mb_xy][6]= h->non_zero_count_cache[6+8*4]; | |
1189 | |
1190 h->non_zero_count[mb_xy][9]= h->non_zero_count_cache[1+8*2]; | |
1191 h->non_zero_count[mb_xy][8]= h->non_zero_count_cache[2+8*2]; | |
1192 h->non_zero_count[mb_xy][7]= h->non_zero_count_cache[2+8*1]; | |
1193 | |
1194 h->non_zero_count[mb_xy][12]=h->non_zero_count_cache[1+8*5]; | |
1195 h->non_zero_count[mb_xy][11]=h->non_zero_count_cache[2+8*5]; | |
1196 h->non_zero_count[mb_xy][10]=h->non_zero_count_cache[2+8*4]; | |
1197 } | |
1198 | |
1199 static inline void write_back_motion(H264Context *h, int mb_type){ | |
1200 MpegEncContext * const s = &h->s; | |
1201 const int b_xy = 4*s->mb_x + 4*s->mb_y*h->b_stride; | |
1202 const int b8_xy= 2*s->mb_x + 2*s->mb_y*h->b8_stride; | |
1203 int list; | |
1204 | |
1205 if(!USES_LIST(mb_type, 0)) | |
1206 fill_rectangle(&s->current_picture.ref_index[0][b8_xy], 2, 2, h->b8_stride, (uint8_t)LIST_NOT_USED, 1); | |
1207 | |
1208 for(list=0; list<h->list_count; list++){ | |
1209 int y; | |
1210 if(!USES_LIST(mb_type, list)) | |
1211 continue; | |
1212 | |
1213 for(y=0; y<4; y++){ | |
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]; | |
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]; | |
1216 } | |
10893
2aafcafbe1f0
Replace cabac checks in inline functions from h264.h with constants.
michael
parents:
10883
diff
changeset
|
1217 if( CABAC ) { |
10866 | 1218 if(IS_SKIP(mb_type)) |
1219 fill_rectangle(h->mvd_table[list][b_xy], 4, 4, h->b_stride, 0, 4); | |
1220 else | |
1221 for(y=0; y<4; y++){ | |
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]; | |
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]; | |
1224 } | |
1225 } | |
1226 | |
1227 { | |
1228 int8_t *ref_index = &s->current_picture.ref_index[list][b8_xy]; | |
1229 ref_index[0+0*h->b8_stride]= h->ref_cache[list][scan8[0]]; | |
1230 ref_index[1+0*h->b8_stride]= h->ref_cache[list][scan8[4]]; | |
1231 ref_index[0+1*h->b8_stride]= h->ref_cache[list][scan8[8]]; | |
1232 ref_index[1+1*h->b8_stride]= h->ref_cache[list][scan8[12]]; | |
1233 } | |
1234 } | |
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 | 1237 if(IS_8X8(mb_type)){ |
1238 uint8_t *direct_table = &h->direct_table[b8_xy]; | |
1239 direct_table[1+0*h->b8_stride] = IS_DIRECT(h->sub_mb_type[1]) ? 1 : 0; | |
1240 direct_table[0+1*h->b8_stride] = IS_DIRECT(h->sub_mb_type[2]) ? 1 : 0; | |
1241 direct_table[1+1*h->b8_stride] = IS_DIRECT(h->sub_mb_type[3]) ? 1 : 0; | |
1242 } | |
1243 } | |
1244 } | |
1245 | |
1246 static inline int get_dct8x8_allowed(H264Context *h){ | |
1247 if(h->sps.direct_8x8_inference_flag) | |
1248 return !(*(uint64_t*)h->sub_mb_type & ((MB_TYPE_16x8|MB_TYPE_8x16|MB_TYPE_8x8 )*0x0001000100010001ULL)); | |
1249 else | |
1250 return !(*(uint64_t*)h->sub_mb_type & ((MB_TYPE_16x8|MB_TYPE_8x16|MB_TYPE_8x8|MB_TYPE_DIRECT2)*0x0001000100010001ULL)); | |
1251 } | |
1252 | |
1253 static void predict_field_decoding_flag(H264Context *h){ | |
1254 MpegEncContext * const s = &h->s; | |
1255 const int mb_xy= h->mb_xy; | |
1256 int mb_type = (h->slice_table[mb_xy-1] == h->slice_num) | |
1257 ? s->current_picture.mb_type[mb_xy-1] | |
1258 : (h->slice_table[mb_xy-s->mb_stride] == h->slice_num) | |
1259 ? s->current_picture.mb_type[mb_xy-s->mb_stride] | |
1260 : 0; | |
1261 h->mb_mbaff = h->mb_field_decoding_flag = IS_INTERLACED(mb_type) ? 1 : 0; | |
1262 } | |
1263 | |
1264 /** | |
1265 * decodes a P_SKIP or B_SKIP macroblock | |
1266 */ | |
1267 static void decode_mb_skip(H264Context *h){ | |
1268 MpegEncContext * const s = &h->s; | |
1269 const int mb_xy= h->mb_xy; | |
1270 int mb_type=0; | |
1271 | |
1272 memset(h->non_zero_count[mb_xy], 0, 16); | |
1273 memset(h->non_zero_count_cache + 8, 0, 8*5); //FIXME ugly, remove pfui | |
1274 | |
1275 if(MB_FIELD) | |
1276 mb_type|= MB_TYPE_INTERLACED; | |
1277 | |
1278 if( h->slice_type_nos == FF_B_TYPE ) | |
1279 { | |
1280 // just for fill_caches. pred_direct_motion will set the real mb_type | |
1281 mb_type|= MB_TYPE_P0L0|MB_TYPE_P0L1|MB_TYPE_DIRECT2|MB_TYPE_SKIP; | |
1282 | |
1283 fill_caches(h, mb_type, 0); //FIXME check what is needed and what not ... | |
1284 ff_h264_pred_direct_motion(h, &mb_type); | |
1285 mb_type|= MB_TYPE_SKIP; | |
1286 } | |
1287 else | |
1288 { | |
1289 int mx, my; | |
1290 mb_type|= MB_TYPE_16x16|MB_TYPE_P0L0|MB_TYPE_P1L0|MB_TYPE_SKIP; | |
1291 | |
1292 fill_caches(h, mb_type, 0); //FIXME check what is needed and what not ... | |
1293 pred_pskip_motion(h, &mx, &my); | |
1294 fill_rectangle(&h->ref_cache[0][scan8[0]], 4, 4, 8, 0, 1); | |
1295 fill_rectangle( h->mv_cache[0][scan8[0]], 4, 4, 8, pack16to32(mx,my), 4); | |
1296 } | |
1297 | |
1298 write_back_motion(h, mb_type); | |
1299 s->current_picture.mb_type[mb_xy]= mb_type; | |
1300 s->current_picture.qscale_table[mb_xy]= s->qscale; | |
1301 h->slice_table[ mb_xy ]= h->slice_num; | |
1302 h->prev_mb_skipped= 1; | |
1303 } | |
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 | 1307 #endif /* AVCODEC_H264_H */ |