Line | Branch | Exec | Source |
---|---|---|---|
1 | /* | ||
2 | * H.26L/H.264/AVC/JVT/14496-10/... direct mb/block decoding | ||
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 | /** | ||
23 | * @file | ||
24 | * H.264 / AVC / MPEG-4 part10 direct mb/block decoding. | ||
25 | * @author Michael Niedermayer <michaelni@gmx.at> | ||
26 | */ | ||
27 | |||
28 | #include "avcodec.h" | ||
29 | #include "h264dec.h" | ||
30 | #include "h264_ps.h" | ||
31 | #include "mpegutils.h" | ||
32 | #include "rectangle.h" | ||
33 | #include "threadframe.h" | ||
34 | |||
35 | #include <assert.h> | ||
36 | |||
37 | 27546 | static int get_scale_factor(const H264SliceContext *sl, | |
38 | int poc, int poc1, int i) | ||
39 | { | ||
40 | 27546 | int poc0 = sl->ref_list[0][i].poc; | |
41 | 27546 | int64_t pocdiff = poc1 - (int64_t)poc0; | |
42 | 27546 | int td = av_clip_int8(pocdiff); | |
43 | |||
44 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 27546 times.
|
27546 | if (pocdiff != (int)pocdiff) |
45 | ✗ | avpriv_request_sample(sl->h264->avctx, "pocdiff overflow"); | |
46 | |||
47 |
4/4✓ Branch 0 taken 19940 times.
✓ Branch 1 taken 7606 times.
✓ Branch 2 taken 531 times.
✓ Branch 3 taken 19409 times.
|
27546 | if (td == 0 || sl->ref_list[0][i].parent->long_ref) { |
48 | 8137 | return 256; | |
49 | } else { | ||
50 | 19409 | int64_t pocdiff0 = poc - (int64_t)poc0; | |
51 | 19409 | int tb = av_clip_int8(pocdiff0); | |
52 | 19409 | int tx = (16384 + (FFABS(td) >> 1)) / td; | |
53 | |||
54 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 19409 times.
|
19409 | if (pocdiff0 != (int)pocdiff0) |
55 | ✗ | av_log(sl->h264->avctx, AV_LOG_DEBUG, "pocdiff0 overflow\n"); | |
56 | |||
57 | 19409 | return av_clip_intp2((tb * tx + 32) >> 6, 10); | |
58 | } | ||
59 | } | ||
60 | |||
61 | 7709 | void ff_h264_direct_dist_scale_factor(const H264Context *const h, | |
62 | H264SliceContext *sl) | ||
63 | { | ||
64 | 18806 | const int poc = FIELD_PICTURE(h) ? h->cur_pic_ptr->field_poc[h->picture_structure == PICT_BOTTOM_FIELD] | |
65 |
2/2✓ Branch 0 taken 3388 times.
✓ Branch 1 taken 4321 times.
|
7709 | : h->cur_pic_ptr->poc; |
66 | 7709 | const int poc1 = sl->ref_list[1][0].poc; | |
67 | int i, field; | ||
68 | |||
69 |
2/2✓ Branch 0 taken 515 times.
✓ Branch 1 taken 7194 times.
|
7709 | if (FRAME_MBAFF(h)) |
70 |
2/2✓ Branch 0 taken 1030 times.
✓ Branch 1 taken 515 times.
|
1545 | for (field = 0; field < 2; field++) { |
71 | 1030 | const int poc = h->cur_pic_ptr->field_poc[field]; | |
72 | 1030 | const int poc1 = sl->ref_list[1][0].parent->field_poc[field]; | |
73 |
2/2✓ Branch 0 taken 6592 times.
✓ Branch 1 taken 1030 times.
|
7622 | for (i = 0; i < 2 * sl->ref_count[0]; i++) |
74 | 6592 | sl->dist_scale_factor_field[field][i ^ field] = | |
75 | 6592 | get_scale_factor(sl, poc, poc1, i + 16); | |
76 | } | ||
77 | |||
78 |
2/2✓ Branch 0 taken 20954 times.
✓ Branch 1 taken 7709 times.
|
28663 | for (i = 0; i < sl->ref_count[0]; i++) |
79 | 20954 | sl->dist_scale_factor[i] = get_scale_factor(sl, poc, poc1, i); | |
80 | 7709 | } | |
81 | |||
82 | 17478 | static void fill_colmap(const H264Context *h, H264SliceContext *sl, | |
83 | int map[2][16 + 32], int list, | ||
84 | int field, int colfield, int mbafi) | ||
85 | { | ||
86 | 17478 | const H264Picture *const ref1 = sl->ref_list[1][0].parent; | |
87 | int j, old_ref, rfield; | ||
88 |
2/2✓ Branch 0 taken 2060 times.
✓ Branch 1 taken 15418 times.
|
17478 | int start = mbafi ? 16 : 0; |
89 |
2/2✓ Branch 0 taken 2060 times.
✓ Branch 1 taken 15418 times.
|
17478 | int end = mbafi ? 16 + 2 * sl->ref_count[0] : sl->ref_count[0]; |
90 |
4/4✓ Branch 0 taken 15418 times.
✓ Branch 1 taken 2060 times.
✓ Branch 2 taken 6776 times.
✓ Branch 3 taken 8642 times.
|
17478 | int interl = mbafi || h->picture_structure != PICT_FRAME; |
91 | |||
92 | /* bogus; fills in for missing frames */ | ||
93 | 17478 | memset(map[list], 0, sizeof(map[list])); | |
94 | |||
95 |
2/2✓ Branch 0 taken 34956 times.
✓ Branch 1 taken 17478 times.
|
52434 | for (rfield = 0; rfield < 2; rfield++) { |
96 |
2/2✓ Branch 0 taken 37236 times.
✓ Branch 1 taken 34956 times.
|
72192 | for (old_ref = 0; old_ref < ref1->ref_count[colfield][list]; old_ref++) { |
97 | 37236 | int poc = ref1->ref_poc[colfield][list][old_ref]; | |
98 | |||
99 |
2/2✓ Branch 0 taken 16808 times.
✓ Branch 1 taken 20428 times.
|
37236 | if (!interl) |
100 | 16808 | poc |= 3; | |
101 | // FIXME: store all MBAFF references so this is not needed | ||
102 |
3/4✓ Branch 0 taken 20428 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 5396 times.
✓ Branch 3 taken 15032 times.
|
20428 | else if (interl && (poc & 3) == 3) |
103 | 5396 | poc = (poc & ~3) + rfield + 1; | |
104 | |||
105 |
2/2✓ Branch 0 taken 101288 times.
✓ Branch 1 taken 7392 times.
|
108680 | for (j = start; j < end; j++) { |
106 | 101288 | if (4 * sl->ref_list[0][j].parent->frame_num + | |
107 |
2/2✓ Branch 0 taken 29844 times.
✓ Branch 1 taken 71444 times.
|
101288 | (sl->ref_list[0][j].reference & 3) == poc) { |
108 |
2/2✓ Branch 0 taken 4196 times.
✓ Branch 1 taken 25648 times.
|
29844 | int cur_ref = mbafi ? (j - 16) ^ field : j; |
109 |
2/2✓ Branch 0 taken 5736 times.
✓ Branch 1 taken 24108 times.
|
29844 | if (ref1->mbaff) |
110 | 5736 | map[list][2 * old_ref + (rfield ^ field) + 16] = cur_ref; | |
111 |
4/4✓ Branch 0 taken 14922 times.
✓ Branch 1 taken 14922 times.
✓ Branch 2 taken 6580 times.
✓ Branch 3 taken 8342 times.
|
29844 | if (rfield == field || !interl) |
112 | 21502 | map[list][old_ref] = cur_ref; | |
113 | 29844 | break; | |
114 | } | ||
115 | } | ||
116 | } | ||
117 | } | ||
118 | 17478 | } | |
119 | |||
120 | 37374 | void ff_h264_direct_ref_list_init(const H264Context *const h, H264SliceContext *sl) | |
121 | { | ||
122 | 37374 | H264Ref *const ref1 = &sl->ref_list[1][0]; | |
123 | 37374 | H264Picture *const cur = h->cur_pic_ptr; | |
124 | int list, j, field; | ||
125 | 37374 | int sidx = (h->picture_structure & 1) ^ 1; | |
126 | 37374 | int ref1sidx = (ref1->reference & 1) ^ 1; | |
127 | |||
128 |
2/2✓ Branch 0 taken 46016 times.
✓ Branch 1 taken 37374 times.
|
83390 | for (list = 0; list < sl->list_count; list++) { |
129 | 46016 | cur->ref_count[sidx][list] = sl->ref_count[list]; | |
130 |
2/2✓ Branch 0 taken 140558 times.
✓ Branch 1 taken 46016 times.
|
186574 | for (j = 0; j < sl->ref_count[list]; j++) |
131 | 140558 | cur->ref_poc[sidx][list][j] = 4 * sl->ref_list[list][j].parent->frame_num + | |
132 | 140558 | (sl->ref_list[list][j].reference & 3); | |
133 | } | ||
134 | |||
135 |
2/2✓ Branch 0 taken 28418 times.
✓ Branch 1 taken 8956 times.
|
37374 | if (h->picture_structure == PICT_FRAME) { |
136 | 28418 | memcpy(cur->ref_count[1], cur->ref_count[0], sizeof(cur->ref_count[0])); | |
137 | 28418 | memcpy(cur->ref_poc[1], cur->ref_poc[0], sizeof(cur->ref_poc[0])); | |
138 | } | ||
139 | |||
140 |
2/2✓ Branch 0 taken 28301 times.
✓ Branch 1 taken 9073 times.
|
37374 | if (h->current_slice == 0) { |
141 | 28301 | cur->mbaff = FRAME_MBAFF(h); | |
142 | } else { | ||
143 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 9073 times.
|
9073 | av_assert0(cur->mbaff == FRAME_MBAFF(h)); |
144 | } | ||
145 | |||
146 | 37374 | sl->col_fieldoff = 0; | |
147 | |||
148 |
3/4✓ Branch 0 taken 13110 times.
✓ Branch 1 taken 24264 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 13110 times.
|
37374 | if (sl->list_count != 2 || !sl->ref_count[1]) |
149 | 24264 | return; | |
150 | |||
151 |
2/2✓ Branch 0 taken 8297 times.
✓ Branch 1 taken 4813 times.
|
13110 | if (h->picture_structure == PICT_FRAME) { |
152 | 8297 | int cur_poc = h->cur_pic_ptr->poc; | |
153 | 8297 | const int *col_poc = sl->ref_list[1][0].parent->field_poc; | |
154 |
3/4✓ Branch 0 taken 10 times.
✓ Branch 1 taken 8287 times.
✓ Branch 2 taken 10 times.
✗ Branch 3 not taken.
|
8297 | if (col_poc[0] == INT_MAX && col_poc[1] == INT_MAX) { |
155 | 10 | av_log(h->avctx, AV_LOG_ERROR, "co located POCs unavailable\n"); | |
156 | 10 | sl->col_parity = 1; | |
157 | } else | ||
158 | 8287 | sl->col_parity = (FFABS(col_poc[0] - (int64_t)cur_poc) >= | |
159 | 8287 | FFABS(col_poc[1] - (int64_t)cur_poc)); | |
160 | 8297 | ref1sidx = | |
161 | 8297 | sidx = sl->col_parity; | |
162 | // FL -> FL & differ parity | ||
163 |
2/2✓ Branch 0 taken 1866 times.
✓ Branch 1 taken 2947 times.
|
4813 | } else if (!(h->picture_structure & sl->ref_list[1][0].reference) && |
164 |
2/2✓ Branch 0 taken 1728 times.
✓ Branch 1 taken 138 times.
|
1866 | !sl->ref_list[1][0].parent->mbaff) { |
165 | 1728 | sl->col_fieldoff = 2 * sl->ref_list[1][0].reference - 3; | |
166 | } | ||
167 | |||
168 |
3/4✓ Branch 0 taken 13110 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 5401 times.
✓ Branch 3 taken 7709 times.
|
13110 | if (sl->slice_type_nos != AV_PICTURE_TYPE_B || sl->direct_spatial_mv_pred) |
169 | 5401 | return; | |
170 | |||
171 |
2/2✓ Branch 0 taken 15418 times.
✓ Branch 1 taken 7709 times.
|
23127 | for (list = 0; list < 2; list++) { |
172 | 15418 | fill_colmap(h, sl, sl->map_col_to_list0, list, sidx, ref1sidx, 0); | |
173 |
2/2✓ Branch 0 taken 1030 times.
✓ Branch 1 taken 14388 times.
|
15418 | if (FRAME_MBAFF(h)) |
174 |
2/2✓ Branch 0 taken 2060 times.
✓ Branch 1 taken 1030 times.
|
3090 | for (field = 0; field < 2; field++) |
175 | 2060 | fill_colmap(h, sl, sl->map_col_to_list0_field[field], list, field, | |
176 | field, 1); | ||
177 | } | ||
178 | } | ||
179 | |||
180 | 4564564 | static void await_reference_mb_row(const H264Context *const h, H264Ref *ref, | |
181 | int mb_y) | ||
182 | { | ||
183 | 4564564 | int ref_field = ref->reference - 1; | |
184 | 4564564 | int ref_field_picture = ref->parent->field_picture; | |
185 | 4564564 | int ref_height = 16 * h->mb_height >> ref_field_picture; | |
186 | |||
187 |
2/2✓ Branch 0 taken 4535681 times.
✓ Branch 1 taken 28883 times.
|
4564564 | if (!HAVE_THREADS || !(h->avctx->active_thread_type & FF_THREAD_FRAME)) |
188 | 4535681 | return; | |
189 | |||
190 | /* FIXME: It can be safe to access mb stuff | ||
191 | * even if pixels aren't deblocked yet. */ | ||
192 | |||
193 |
1/4✗ Branch 0 not taken.
✓ Branch 1 taken 28883 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
|
57766 | ff_thread_await_progress(&ref->parent->tf, |
194 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 28883 times.
|
28883 | FFMIN(16 * mb_y >> ref_field_picture, |
195 | ref_height - 1), | ||
196 | ref_field_picture && ref_field); | ||
197 | } | ||
198 | |||
199 | 2976098 | static void pred_spatial_direct_motion(const H264Context *const h, H264SliceContext *sl, | |
200 | int *mb_type) | ||
201 | { | ||
202 | 2976098 | int b8_stride = 2; | |
203 | 2976098 | int b4_stride = h->b_stride; | |
204 | 2976098 | int mb_xy = sl->mb_xy, mb_y = sl->mb_y; | |
205 | int mb_type_col[2]; | ||
206 | const int16_t (*l1mv0)[2], (*l1mv1)[2]; | ||
207 | const int8_t *l1ref0, *l1ref1; | ||
208 | 2976098 | const int is_b8x8 = IS_8X8(*mb_type); | |
209 | 2976098 | unsigned int sub_mb_type = MB_TYPE_L0L1; | |
210 | int i8, i4; | ||
211 | int ref[2]; | ||
212 | int mv[2]; | ||
213 | int list; | ||
214 | |||
215 | assert(sl->ref_list[1][0].reference & 3); | ||
216 | |||
217 | 2976098 | await_reference_mb_row(h, &sl->ref_list[1][0], | |
218 | 2976098 | sl->mb_y + !!IS_INTERLACED(*mb_type)); | |
219 | |||
220 | #define MB_TYPE_16x16_OR_INTRA (MB_TYPE_16x16 | MB_TYPE_INTRA4x4 | \ | ||
221 | MB_TYPE_INTRA16x16 | MB_TYPE_INTRA_PCM) | ||
222 | |||
223 | /* ref = min(neighbors) */ | ||
224 |
2/2✓ Branch 0 taken 5952196 times.
✓ Branch 1 taken 2976098 times.
|
8928294 | for (list = 0; list < 2; list++) { |
225 | 5952196 | int left_ref = sl->ref_cache[list][scan8[0] - 1]; | |
226 | 5952196 | int top_ref = sl->ref_cache[list][scan8[0] - 8]; | |
227 | 5952196 | int refc = sl->ref_cache[list][scan8[0] - 8 + 4]; | |
228 | 5952196 | const int16_t *C = sl->mv_cache[list][scan8[0] - 8 + 4]; | |
229 |
2/2✓ Branch 0 taken 730810 times.
✓ Branch 1 taken 5221386 times.
|
5952196 | if (refc == PART_NOT_AVAILABLE) { |
230 | 730810 | refc = sl->ref_cache[list][scan8[0] - 8 - 1]; | |
231 | 730810 | C = sl->mv_cache[list][scan8[0] - 8 - 1]; | |
232 | } | ||
233 | 5952196 | ref[list] = FFMIN3((unsigned)left_ref, | |
234 | (unsigned)top_ref, | ||
235 | (unsigned)refc); | ||
236 |
2/2✓ Branch 0 taken 5452845 times.
✓ Branch 1 taken 499351 times.
|
5952196 | if (ref[list] >= 0) { |
237 | /* This is just pred_motion() but with the cases removed that | ||
238 | * cannot happen for direct blocks. */ | ||
239 | 5452845 | const int16_t *const A = sl->mv_cache[list][scan8[0] - 1]; | |
240 | 5452845 | const int16_t *const B = sl->mv_cache[list][scan8[0] - 8]; | |
241 | |||
242 | 5452845 | int match_count = (left_ref == ref[list]) + | |
243 | 5452845 | (top_ref == ref[list]) + | |
244 | 5452845 | (refc == ref[list]); | |
245 | |||
246 |
2/2✓ Branch 0 taken 4841611 times.
✓ Branch 1 taken 611234 times.
|
5452845 | if (match_count > 1) { // most common |
247 | 4841611 | mv[list] = pack16to32(mid_pred(A[0], B[0], C[0]), | |
248 | 4841611 | mid_pred(A[1], B[1], C[1])); | |
249 | } else { | ||
250 | assert(match_count == 1); | ||
251 |
2/2✓ Branch 0 taken 478970 times.
✓ Branch 1 taken 132264 times.
|
611234 | if (left_ref == ref[list]) |
252 | 478970 | mv[list] = AV_RN32A(A); | |
253 |
2/2✓ Branch 0 taken 62280 times.
✓ Branch 1 taken 69984 times.
|
132264 | else if (top_ref == ref[list]) |
254 | 62280 | mv[list] = AV_RN32A(B); | |
255 | else | ||
256 | 69984 | mv[list] = AV_RN32A(C); | |
257 | } | ||
258 | av_assert2(ref[list] < (sl->ref_count[list] << !!FRAME_MBAFF(h))); | ||
259 | } else { | ||
260 | 499351 | int mask = ~(MB_TYPE_L0 << (2 * list)); | |
261 | 499351 | mv[list] = 0; | |
262 | 499351 | ref[list] = -1; | |
263 |
2/2✓ Branch 0 taken 473427 times.
✓ Branch 1 taken 25924 times.
|
499351 | if (!is_b8x8) |
264 | 473427 | *mb_type &= mask; | |
265 | 499351 | sub_mb_type &= mask; | |
266 | } | ||
267 | } | ||
268 |
4/4✓ Branch 0 taken 247564 times.
✓ Branch 1 taken 2728534 times.
✓ Branch 2 taken 6565 times.
✓ Branch 3 taken 240999 times.
|
2976098 | if (ref[0] < 0 && ref[1] < 0) { |
269 | 6565 | ref[0] = ref[1] = 0; | |
270 |
2/2✓ Branch 0 taken 5674 times.
✓ Branch 1 taken 891 times.
|
6565 | if (!is_b8x8) |
271 | 5674 | *mb_type |= MB_TYPE_L0L1; | |
272 | 6565 | sub_mb_type |= MB_TYPE_L0L1; | |
273 | } | ||
274 | |||
275 |
2/2✓ Branch 0 taken 1892012 times.
✓ Branch 1 taken 1084086 times.
|
2976098 | if (!(is_b8x8 | mv[0] | mv[1])) { |
276 | 1892012 | fill_rectangle(&sl->ref_cache[0][scan8[0]], 4, 4, 8, (uint8_t)ref[0], 1); | |
277 | 1892012 | fill_rectangle(&sl->ref_cache[1][scan8[0]], 4, 4, 8, (uint8_t)ref[1], 1); | |
278 | 1892012 | fill_rectangle(&sl->mv_cache[0][scan8[0]], 4, 4, 8, 0, 4); | |
279 | 1892012 | fill_rectangle(&sl->mv_cache[1][scan8[0]], 4, 4, 8, 0, 4); | |
280 | 1892012 | *mb_type = (*mb_type & ~(MB_TYPE_8x8 | MB_TYPE_16x8 | MB_TYPE_8x16 | | |
281 | MB_TYPE_P1L0 | MB_TYPE_P1L1)) | | ||
282 | 1892012 | MB_TYPE_16x16 | MB_TYPE_DIRECT2; | |
283 | 1892012 | return; | |
284 | } | ||
285 | |||
286 |
2/2✓ Branch 0 taken 718188 times.
✓ Branch 1 taken 365898 times.
|
1084086 | if (IS_INTERLACED(sl->ref_list[1][0].parent->mb_type[mb_xy])) { // AFL/AFR/FR/FL -> AFL/FL |
287 |
2/2✓ Branch 0 taken 43571 times.
✓ Branch 1 taken 674617 times.
|
718188 | if (!IS_INTERLACED(*mb_type)) { // AFR/FR -> AFL/FL |
288 | 43571 | mb_y = (sl->mb_y & ~1) + sl->col_parity; | |
289 | 43571 | mb_xy = sl->mb_x + | |
290 | 43571 | ((sl->mb_y & ~1) + sl->col_parity) * h->mb_stride; | |
291 | 43571 | b8_stride = 0; | |
292 | } else { | ||
293 | 674617 | mb_y += sl->col_fieldoff; | |
294 | 674617 | mb_xy += h->mb_stride * sl->col_fieldoff; // non-zero for FL -> FL & differ parity | |
295 | } | ||
296 | 718188 | goto single_col; | |
297 | } else { // AFL/AFR/FR/FL -> AFR/FR | ||
298 |
2/2✓ Branch 0 taken 51346 times.
✓ Branch 1 taken 314552 times.
|
365898 | if (IS_INTERLACED(*mb_type)) { // AFL /FL -> AFR/FR |
299 | 51346 | mb_y = sl->mb_y & ~1; | |
300 | 51346 | mb_xy = (sl->mb_y & ~1) * h->mb_stride + sl->mb_x; | |
301 | 51346 | mb_type_col[0] = sl->ref_list[1][0].parent->mb_type[mb_xy]; | |
302 | 51346 | mb_type_col[1] = sl->ref_list[1][0].parent->mb_type[mb_xy + h->mb_stride]; | |
303 | 51346 | b8_stride = 2 + 4 * h->mb_stride; | |
304 | 51346 | b4_stride *= 6; | |
305 | 51346 | if (IS_INTERLACED(mb_type_col[0]) != | |
306 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 51346 times.
|
51346 | IS_INTERLACED(mb_type_col[1])) { |
307 | ✗ | mb_type_col[0] &= ~MB_TYPE_INTERLACED; | |
308 | ✗ | mb_type_col[1] &= ~MB_TYPE_INTERLACED; | |
309 | } | ||
310 | |||
311 | 51346 | sub_mb_type |= MB_TYPE_16x16 | MB_TYPE_DIRECT2; /* B_SUB_8x8 */ | |
312 |
2/2✓ Branch 0 taken 40827 times.
✓ Branch 1 taken 10519 times.
|
51346 | if ((mb_type_col[0] & MB_TYPE_16x16_OR_INTRA) && |
313 |
4/4✓ Branch 0 taken 35974 times.
✓ Branch 1 taken 4853 times.
✓ Branch 2 taken 32330 times.
✓ Branch 3 taken 3644 times.
|
40827 | (mb_type_col[1] & MB_TYPE_16x16_OR_INTRA) && |
314 | !is_b8x8) { | ||
315 | 32330 | *mb_type |= MB_TYPE_16x8 | MB_TYPE_DIRECT2; /* B_16x8 */ | |
316 | } else { | ||
317 | 19016 | *mb_type |= MB_TYPE_8x8; | |
318 | } | ||
319 | } else { // AFR/FR -> AFR/FR | ||
320 | 314552 | single_col: | |
321 | 1032740 | mb_type_col[0] = | |
322 | 1032740 | mb_type_col[1] = sl->ref_list[1][0].parent->mb_type[mb_xy]; | |
323 | |||
324 | 1032740 | sub_mb_type |= MB_TYPE_16x16 | MB_TYPE_DIRECT2; /* B_SUB_8x8 */ | |
325 |
4/4✓ Branch 0 taken 957953 times.
✓ Branch 1 taken 74787 times.
✓ Branch 2 taken 855880 times.
✓ Branch 3 taken 102073 times.
|
1032740 | if (!is_b8x8 && (mb_type_col[0] & MB_TYPE_16x16_OR_INTRA)) { |
326 | 855880 | *mb_type |= MB_TYPE_16x16 | MB_TYPE_DIRECT2; /* B_16x16 */ | |
327 |
2/2✓ Branch 0 taken 102073 times.
✓ Branch 1 taken 74787 times.
|
176860 | } else if (!is_b8x8 && |
328 |
2/2✓ Branch 0 taken 57259 times.
✓ Branch 1 taken 44814 times.
|
102073 | (mb_type_col[0] & (MB_TYPE_16x8 | MB_TYPE_8x16))) { |
329 | 57259 | *mb_type |= MB_TYPE_DIRECT2 | | |
330 | 57259 | (mb_type_col[0] & (MB_TYPE_16x8 | MB_TYPE_8x16)); | |
331 | } else { | ||
332 |
2/2✓ Branch 0 taken 21562 times.
✓ Branch 1 taken 98039 times.
|
119601 | if (!h->ps.sps->direct_8x8_inference_flag) { |
333 | /* FIXME: Save sub mb types from previous frames (or derive | ||
334 | * from MVs) so we know exactly what block size to use. */ | ||
335 | 21562 | sub_mb_type += (MB_TYPE_8x8 - MB_TYPE_16x16); /* B_SUB_4x4 */ | |
336 | } | ||
337 | 119601 | *mb_type |= MB_TYPE_8x8; | |
338 | } | ||
339 | } | ||
340 | } | ||
341 | |||
342 | 1084086 | await_reference_mb_row(h, &sl->ref_list[1][0], mb_y); | |
343 | |||
344 | 1084086 | l1mv0 = (void*)&sl->ref_list[1][0].parent->motion_val[0][h->mb2b_xy[mb_xy]]; | |
345 | 1084086 | l1mv1 = (void*)&sl->ref_list[1][0].parent->motion_val[1][h->mb2b_xy[mb_xy]]; | |
346 | 1084086 | l1ref0 = &sl->ref_list[1][0].parent->ref_index[0][4 * mb_xy]; | |
347 | 1084086 | l1ref1 = &sl->ref_list[1][0].parent->ref_index[1][4 * mb_xy]; | |
348 |
2/2✓ Branch 0 taken 43571 times.
✓ Branch 1 taken 1040515 times.
|
1084086 | if (!b8_stride) { |
349 |
2/2✓ Branch 0 taken 22047 times.
✓ Branch 1 taken 21524 times.
|
43571 | if (sl->mb_y & 1) { |
350 | 22047 | l1ref0 += 2; | |
351 | 22047 | l1ref1 += 2; | |
352 | 22047 | l1mv0 += 2 * b4_stride; | |
353 | 22047 | l1mv1 += 2 * b4_stride; | |
354 | } | ||
355 | } | ||
356 | |||
357 |
2/2✓ Branch 0 taken 94917 times.
✓ Branch 1 taken 989169 times.
|
1084086 | if (IS_INTERLACED(*mb_type) != IS_INTERLACED(mb_type_col[0])) { |
358 | 94917 | int n = 0; | |
359 |
2/2✓ Branch 0 taken 379668 times.
✓ Branch 1 taken 94917 times.
|
474585 | for (i8 = 0; i8 < 4; i8++) { |
360 | 379668 | int x8 = i8 & 1; | |
361 | 379668 | int y8 = i8 >> 1; | |
362 | 379668 | int xy8 = x8 + y8 * b8_stride; | |
363 | 379668 | int xy4 = x8 * 3 + y8 * b4_stride; | |
364 | int a, b; | ||
365 | |||
366 |
4/4✓ Branch 0 taken 66816 times.
✓ Branch 1 taken 312852 times.
✓ Branch 2 taken 27121 times.
✓ Branch 3 taken 39695 times.
|
379668 | if (is_b8x8 && !IS_DIRECT(sl->sub_mb_type[i8])) |
367 | 27121 | continue; | |
368 | 352547 | sl->sub_mb_type[i8] = sub_mb_type; | |
369 | |||
370 | 352547 | fill_rectangle(&sl->ref_cache[0][scan8[i8 * 4]], 2, 2, 8, | |
371 | 352547 | (uint8_t)ref[0], 1); | |
372 | 352547 | fill_rectangle(&sl->ref_cache[1][scan8[i8 * 4]], 2, 2, 8, | |
373 | 352547 | (uint8_t)ref[1], 1); | |
374 |
3/4✓ Branch 0 taken 197262 times.
✓ Branch 1 taken 155285 times.
✓ Branch 2 taken 197262 times.
✗ Branch 3 not taken.
|
352547 | if (!IS_INTRA(mb_type_col[y8]) && !sl->ref_list[1][0].parent->long_ref && |
375 |
2/2✓ Branch 0 taken 134513 times.
✓ Branch 1 taken 62749 times.
|
197262 | ((l1ref0[xy8] == 0 && |
376 |
2/2✓ Branch 0 taken 10493 times.
✓ Branch 1 taken 124020 times.
|
134513 | FFABS(l1mv0[xy4][0]) <= 1 && |
377 |
2/2✓ Branch 0 taken 4460 times.
✓ Branch 1 taken 6033 times.
|
10493 | FFABS(l1mv0[xy4][1]) <= 1) || |
378 |
2/2✓ Branch 0 taken 12234 times.
✓ Branch 1 taken 178995 times.
|
191229 | (l1ref0[xy8] < 0 && |
379 |
2/2✓ Branch 0 taken 9667 times.
✓ Branch 1 taken 2567 times.
|
12234 | l1ref1[xy8] == 0 && |
380 |
2/2✓ Branch 0 taken 826 times.
✓ Branch 1 taken 8841 times.
|
9667 | FFABS(l1mv1[xy4][0]) <= 1 && |
381 |
2/2✓ Branch 0 taken 502 times.
✓ Branch 1 taken 324 times.
|
826 | FFABS(l1mv1[xy4][1]) <= 1))) { |
382 | 6535 | a = | |
383 | 6535 | b = 0; | |
384 |
2/2✓ Branch 0 taken 515 times.
✓ Branch 1 taken 6020 times.
|
6535 | if (ref[0] > 0) |
385 | 515 | a = mv[0]; | |
386 |
2/2✓ Branch 0 taken 159 times.
✓ Branch 1 taken 6376 times.
|
6535 | if (ref[1] > 0) |
387 | 159 | b = mv[1]; | |
388 | 6535 | n++; | |
389 | } else { | ||
390 | 346012 | a = mv[0]; | |
391 | 346012 | b = mv[1]; | |
392 | } | ||
393 | 352547 | fill_rectangle(&sl->mv_cache[0][scan8[i8 * 4]], 2, 2, 8, a, 4); | |
394 | 352547 | fill_rectangle(&sl->mv_cache[1][scan8[i8 * 4]], 2, 2, 8, b, 4); | |
395 | } | ||
396 |
4/4✓ Branch 0 taken 78213 times.
✓ Branch 1 taken 16704 times.
✓ Branch 2 taken 77034 times.
✓ Branch 3 taken 1179 times.
|
94917 | if (!is_b8x8 && !(n & 3)) |
397 | 77034 | *mb_type = (*mb_type & ~(MB_TYPE_8x8 | MB_TYPE_16x8 | MB_TYPE_8x16 | | |
398 | MB_TYPE_P1L0 | MB_TYPE_P1L1)) | | ||
399 | 77034 | MB_TYPE_16x16 | MB_TYPE_DIRECT2; | |
400 |
2/2✓ Branch 0 taken 826966 times.
✓ Branch 1 taken 162203 times.
|
989169 | } else if (IS_16X16(*mb_type)) { |
401 | int a, b; | ||
402 | |||
403 | 826966 | fill_rectangle(&sl->ref_cache[0][scan8[0]], 4, 4, 8, (uint8_t)ref[0], 1); | |
404 | 826966 | fill_rectangle(&sl->ref_cache[1][scan8[0]], 4, 4, 8, (uint8_t)ref[1], 1); | |
405 |
4/4✓ Branch 0 taken 616533 times.
✓ Branch 1 taken 210433 times.
✓ Branch 2 taken 616522 times.
✓ Branch 3 taken 11 times.
|
826966 | if (!IS_INTRA(mb_type_col[0]) && !sl->ref_list[1][0].parent->long_ref && |
406 |
2/2✓ Branch 0 taken 343172 times.
✓ Branch 1 taken 273350 times.
|
616522 | ((l1ref0[0] == 0 && |
407 |
2/2✓ Branch 0 taken 66157 times.
✓ Branch 1 taken 277015 times.
|
343172 | FFABS(l1mv0[0][0]) <= 1 && |
408 |
2/2✓ Branch 0 taken 36149 times.
✓ Branch 1 taken 30008 times.
|
66157 | FFABS(l1mv0[0][1]) <= 1) || |
409 |
4/4✓ Branch 0 taken 18004 times.
✓ Branch 1 taken 568510 times.
✓ Branch 2 taken 17315 times.
✓ Branch 3 taken 689 times.
|
586514 | (l1ref0[0] < 0 && !l1ref1[0] && |
410 |
2/2✓ Branch 0 taken 4745 times.
✓ Branch 1 taken 12570 times.
|
17315 | FFABS(l1mv1[0][0]) <= 1 && |
411 |
2/2✓ Branch 0 taken 3908 times.
✓ Branch 1 taken 837 times.
|
4745 | FFABS(l1mv1[0][1]) <= 1 && |
412 |
1/2✓ Branch 0 taken 3908 times.
✗ Branch 1 not taken.
|
3908 | h->x264_build > 33U))) { |
413 | 33916 | a = b = 0; | |
414 |
2/2✓ Branch 0 taken 7029 times.
✓ Branch 1 taken 26887 times.
|
33916 | if (ref[0] > 0) |
415 | 7029 | a = mv[0]; | |
416 |
2/2✓ Branch 0 taken 4709 times.
✓ Branch 1 taken 29207 times.
|
33916 | if (ref[1] > 0) |
417 | 4709 | b = mv[1]; | |
418 | } else { | ||
419 | 793050 | a = mv[0]; | |
420 | 793050 | b = mv[1]; | |
421 | } | ||
422 | 826966 | fill_rectangle(&sl->mv_cache[0][scan8[0]], 4, 4, 8, a, 4); | |
423 | 826966 | fill_rectangle(&sl->mv_cache[1][scan8[0]], 4, 4, 8, b, 4); | |
424 | } else { | ||
425 | 162203 | int n = 0; | |
426 |
2/2✓ Branch 0 taken 648812 times.
✓ Branch 1 taken 162203 times.
|
811015 | for (i8 = 0; i8 < 4; i8++) { |
427 | 648812 | const int x8 = i8 & 1; | |
428 | 648812 | const int y8 = i8 >> 1; | |
429 | |||
430 |
4/4✓ Branch 0 taken 268860 times.
✓ Branch 1 taken 379952 times.
✓ Branch 2 taken 122589 times.
✓ Branch 3 taken 146271 times.
|
648812 | if (is_b8x8 && !IS_DIRECT(sl->sub_mb_type[i8])) |
431 | 122589 | continue; | |
432 | 526223 | sl->sub_mb_type[i8] = sub_mb_type; | |
433 | |||
434 | 526223 | fill_rectangle(&sl->mv_cache[0][scan8[i8 * 4]], 2, 2, 8, mv[0], 4); | |
435 | 526223 | fill_rectangle(&sl->mv_cache[1][scan8[i8 * 4]], 2, 2, 8, mv[1], 4); | |
436 | 526223 | fill_rectangle(&sl->ref_cache[0][scan8[i8 * 4]], 2, 2, 8, | |
437 | 526223 | (uint8_t)ref[0], 1); | |
438 | 526223 | fill_rectangle(&sl->ref_cache[1][scan8[i8 * 4]], 2, 2, 8, | |
439 | 526223 | (uint8_t)ref[1], 1); | |
440 | |||
441 | assert(b8_stride == 2); | ||
442 | /* col_zero_flag */ | ||
443 |
4/4✓ Branch 0 taken 486591 times.
✓ Branch 1 taken 39632 times.
✓ Branch 2 taken 486459 times.
✓ Branch 3 taken 132 times.
|
526223 | if (!IS_INTRA(mb_type_col[0]) && !sl->ref_list[1][0].parent->long_ref && |
444 |
2/2✓ Branch 0 taken 170383 times.
✓ Branch 1 taken 316076 times.
|
486459 | (l1ref0[i8] == 0 || |
445 |
2/2✓ Branch 0 taken 12349 times.
✓ Branch 1 taken 158034 times.
|
170383 | (l1ref0[i8] < 0 && |
446 |
2/2✓ Branch 0 taken 11245 times.
✓ Branch 1 taken 1104 times.
|
12349 | l1ref1[i8] == 0 && |
447 |
1/2✓ Branch 0 taken 11245 times.
✗ Branch 1 not taken.
|
11245 | h->x264_build > 33U))) { |
448 |
2/2✓ Branch 0 taken 316076 times.
✓ Branch 1 taken 11245 times.
|
327321 | const int16_t (*l1mv)[2] = l1ref0[i8] == 0 ? l1mv0 : l1mv1; |
449 |
2/2✓ Branch 0 taken 297018 times.
✓ Branch 1 taken 30303 times.
|
327321 | if (IS_SUB_8X8(sub_mb_type)) { |
450 | 297018 | const int16_t *mv_col = l1mv[x8 * 3 + y8 * 3 * b4_stride]; | |
451 |
4/4✓ Branch 0 taken 47180 times.
✓ Branch 1 taken 249838 times.
✓ Branch 2 taken 25062 times.
✓ Branch 3 taken 22118 times.
|
297018 | if (FFABS(mv_col[0]) <= 1 && FFABS(mv_col[1]) <= 1) { |
452 |
2/2✓ Branch 0 taken 21331 times.
✓ Branch 1 taken 3731 times.
|
25062 | if (ref[0] == 0) |
453 | 21331 | fill_rectangle(&sl->mv_cache[0][scan8[i8 * 4]], 2, 2, | |
454 | 8, 0, 4); | ||
455 |
2/2✓ Branch 0 taken 22821 times.
✓ Branch 1 taken 2241 times.
|
25062 | if (ref[1] == 0) |
456 | 22821 | fill_rectangle(&sl->mv_cache[1][scan8[i8 * 4]], 2, 2, | |
457 | 8, 0, 4); | ||
458 | 25062 | n += 4; | |
459 | } | ||
460 | } else { | ||
461 | 30303 | int m = 0; | |
462 |
2/2✓ Branch 0 taken 121212 times.
✓ Branch 1 taken 30303 times.
|
151515 | for (i4 = 0; i4 < 4; i4++) { |
463 | 121212 | const int16_t *mv_col = l1mv[x8 * 2 + (i4 & 1) + | |
464 | 121212 | (y8 * 2 + (i4 >> 1)) * b4_stride]; | |
465 |
4/4✓ Branch 0 taken 8548 times.
✓ Branch 1 taken 112664 times.
✓ Branch 2 taken 3036 times.
✓ Branch 3 taken 5512 times.
|
121212 | if (FFABS(mv_col[0]) <= 1 && FFABS(mv_col[1]) <= 1) { |
466 |
2/2✓ Branch 0 taken 2071 times.
✓ Branch 1 taken 965 times.
|
3036 | if (ref[0] == 0) |
467 | 2071 | AV_ZERO32(sl->mv_cache[0][scan8[i8 * 4 + i4]]); | |
468 |
2/2✓ Branch 0 taken 2790 times.
✓ Branch 1 taken 246 times.
|
3036 | if (ref[1] == 0) |
469 | 2790 | AV_ZERO32(sl->mv_cache[1][scan8[i8 * 4 + i4]]); | |
470 | 3036 | m++; | |
471 | } | ||
472 | } | ||
473 |
2/2✓ Branch 0 taken 29990 times.
✓ Branch 1 taken 313 times.
|
30303 | if (!(m & 3)) |
474 | 29990 | sl->sub_mb_type[i8] += MB_TYPE_16x16 - MB_TYPE_8x8; | |
475 | 30303 | n += m; | |
476 | } | ||
477 | } | ||
478 | } | ||
479 |
4/4✓ Branch 0 taken 94988 times.
✓ Branch 1 taken 67215 times.
✓ Branch 2 taken 86692 times.
✓ Branch 3 taken 8296 times.
|
162203 | if (!is_b8x8 && !(n & 15)) |
480 | 86692 | *mb_type = (*mb_type & ~(MB_TYPE_8x8 | MB_TYPE_16x8 | MB_TYPE_8x16 | | |
481 | MB_TYPE_P1L0 | MB_TYPE_P1L1)) | | ||
482 | 86692 | MB_TYPE_16x16 | MB_TYPE_DIRECT2; | |
483 | } | ||
484 | } | ||
485 | |||
486 | 252190 | static void pred_temp_direct_motion(const H264Context *const h, H264SliceContext *sl, | |
487 | int *mb_type) | ||
488 | { | ||
489 | 252190 | int b8_stride = 2; | |
490 | 252190 | int b4_stride = h->b_stride; | |
491 | 252190 | int mb_xy = sl->mb_xy, mb_y = sl->mb_y; | |
492 | int mb_type_col[2]; | ||
493 | const int16_t (*l1mv0)[2], (*l1mv1)[2]; | ||
494 | const int8_t *l1ref0, *l1ref1; | ||
495 | 252190 | const int is_b8x8 = IS_8X8(*mb_type); | |
496 | unsigned int sub_mb_type; | ||
497 | int i8, i4; | ||
498 | |||
499 | assert(sl->ref_list[1][0].reference & 3); | ||
500 | |||
501 | 252190 | await_reference_mb_row(h, &sl->ref_list[1][0], | |
502 | 252190 | sl->mb_y + !!IS_INTERLACED(*mb_type)); | |
503 | |||
504 |
2/2✓ Branch 0 taken 89151 times.
✓ Branch 1 taken 163039 times.
|
252190 | if (IS_INTERLACED(sl->ref_list[1][0].parent->mb_type[mb_xy])) { // AFL/AFR/FR/FL -> AFL/FL |
505 |
2/2✓ Branch 0 taken 15717 times.
✓ Branch 1 taken 73434 times.
|
89151 | if (!IS_INTERLACED(*mb_type)) { // AFR/FR -> AFL/FL |
506 | 15717 | mb_y = (sl->mb_y & ~1) + sl->col_parity; | |
507 | 15717 | mb_xy = sl->mb_x + | |
508 | 15717 | ((sl->mb_y & ~1) + sl->col_parity) * h->mb_stride; | |
509 | 15717 | b8_stride = 0; | |
510 | } else { | ||
511 | 73434 | mb_y += sl->col_fieldoff; | |
512 | 73434 | mb_xy += h->mb_stride * sl->col_fieldoff; // non-zero for FL -> FL & differ parity | |
513 | } | ||
514 | 89151 | goto single_col; | |
515 | } else { // AFL/AFR/FR/FL -> AFR/FR | ||
516 |
2/2✓ Branch 0 taken 16549 times.
✓ Branch 1 taken 146490 times.
|
163039 | if (IS_INTERLACED(*mb_type)) { // AFL /FL -> AFR/FR |
517 | 16549 | mb_y = sl->mb_y & ~1; | |
518 | 16549 | mb_xy = sl->mb_x + (sl->mb_y & ~1) * h->mb_stride; | |
519 | 16549 | mb_type_col[0] = sl->ref_list[1][0].parent->mb_type[mb_xy]; | |
520 | 16549 | mb_type_col[1] = sl->ref_list[1][0].parent->mb_type[mb_xy + h->mb_stride]; | |
521 | 16549 | b8_stride = 2 + 4 * h->mb_stride; | |
522 | 16549 | b4_stride *= 6; | |
523 | 16549 | if (IS_INTERLACED(mb_type_col[0]) != | |
524 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 16549 times.
|
16549 | IS_INTERLACED(mb_type_col[1])) { |
525 | ✗ | mb_type_col[0] &= ~MB_TYPE_INTERLACED; | |
526 | ✗ | mb_type_col[1] &= ~MB_TYPE_INTERLACED; | |
527 | } | ||
528 | |||
529 | 16549 | sub_mb_type = MB_TYPE_16x16 | MB_TYPE_P0L0 | MB_TYPE_P0L1 | | |
530 | MB_TYPE_DIRECT2; /* B_SUB_8x8 */ | ||
531 | |||
532 |
2/2✓ Branch 0 taken 9887 times.
✓ Branch 1 taken 6662 times.
|
16549 | if ((mb_type_col[0] & MB_TYPE_16x16_OR_INTRA) && |
533 |
4/4✓ Branch 0 taken 7209 times.
✓ Branch 1 taken 2678 times.
✓ Branch 2 taken 5873 times.
✓ Branch 3 taken 1336 times.
|
9887 | (mb_type_col[1] & MB_TYPE_16x16_OR_INTRA) && |
534 | !is_b8x8) { | ||
535 | 5873 | *mb_type |= MB_TYPE_16x8 | MB_TYPE_L0L1 | | |
536 | MB_TYPE_DIRECT2; /* B_16x8 */ | ||
537 | } else { | ||
538 | 10676 | *mb_type |= MB_TYPE_8x8 | MB_TYPE_L0L1; | |
539 | } | ||
540 | } else { // AFR/FR -> AFR/FR | ||
541 | 146490 | single_col: | |
542 | 235641 | mb_type_col[0] = | |
543 | 235641 | mb_type_col[1] = sl->ref_list[1][0].parent->mb_type[mb_xy]; | |
544 | |||
545 | 235641 | sub_mb_type = MB_TYPE_16x16 | MB_TYPE_P0L0 | MB_TYPE_P0L1 | | |
546 | MB_TYPE_DIRECT2; /* B_SUB_8x8 */ | ||
547 |
4/4✓ Branch 0 taken 190068 times.
✓ Branch 1 taken 45573 times.
✓ Branch 2 taken 135149 times.
✓ Branch 3 taken 54919 times.
|
235641 | if (!is_b8x8 && (mb_type_col[0] & MB_TYPE_16x16_OR_INTRA)) { |
548 | 135149 | *mb_type |= MB_TYPE_16x16 | MB_TYPE_P0L0 | MB_TYPE_P0L1 | | |
549 | MB_TYPE_DIRECT2; /* B_16x16 */ | ||
550 |
2/2✓ Branch 0 taken 54919 times.
✓ Branch 1 taken 45573 times.
|
100492 | } else if (!is_b8x8 && |
551 |
2/2✓ Branch 0 taken 35091 times.
✓ Branch 1 taken 19828 times.
|
54919 | (mb_type_col[0] & (MB_TYPE_16x8 | MB_TYPE_8x16))) { |
552 | 35091 | *mb_type |= MB_TYPE_L0L1 | MB_TYPE_DIRECT2 | | |
553 | 35091 | (mb_type_col[0] & (MB_TYPE_16x8 | MB_TYPE_8x16)); | |
554 | } else { | ||
555 |
2/2✓ Branch 0 taken 10593 times.
✓ Branch 1 taken 54808 times.
|
65401 | if (!h->ps.sps->direct_8x8_inference_flag) { |
556 | /* FIXME: save sub mb types from previous frames (or derive | ||
557 | * from MVs) so we know exactly what block size to use */ | ||
558 | 10593 | sub_mb_type = MB_TYPE_8x8 | MB_TYPE_P0L0 | MB_TYPE_P0L1 | | |
559 | MB_TYPE_DIRECT2; /* B_SUB_4x4 */ | ||
560 | } | ||
561 | 65401 | *mb_type |= MB_TYPE_8x8 | MB_TYPE_L0L1; | |
562 | } | ||
563 | } | ||
564 | } | ||
565 | |||
566 | 252190 | await_reference_mb_row(h, &sl->ref_list[1][0], mb_y); | |
567 | |||
568 | 252190 | l1mv0 = (void*)&sl->ref_list[1][0].parent->motion_val[0][h->mb2b_xy[mb_xy]]; | |
569 | 252190 | l1mv1 = (void*)&sl->ref_list[1][0].parent->motion_val[1][h->mb2b_xy[mb_xy]]; | |
570 | 252190 | l1ref0 = &sl->ref_list[1][0].parent->ref_index[0][4 * mb_xy]; | |
571 | 252190 | l1ref1 = &sl->ref_list[1][0].parent->ref_index[1][4 * mb_xy]; | |
572 |
2/2✓ Branch 0 taken 15717 times.
✓ Branch 1 taken 236473 times.
|
252190 | if (!b8_stride) { |
573 |
2/2✓ Branch 0 taken 7766 times.
✓ Branch 1 taken 7951 times.
|
15717 | if (sl->mb_y & 1) { |
574 | 7766 | l1ref0 += 2; | |
575 | 7766 | l1ref1 += 2; | |
576 | 7766 | l1mv0 += 2 * b4_stride; | |
577 | 7766 | l1mv1 += 2 * b4_stride; | |
578 | } | ||
579 | } | ||
580 | |||
581 | { | ||
582 | 252190 | const int *map_col_to_list0[2] = { sl->map_col_to_list0[0], | |
583 | 252190 | sl->map_col_to_list0[1] }; | |
584 | 252190 | const int *dist_scale_factor = sl->dist_scale_factor; | |
585 | int ref_offset; | ||
586 | |||
587 |
4/4✓ Branch 0 taken 42409 times.
✓ Branch 1 taken 209781 times.
✓ Branch 2 taken 20122 times.
✓ Branch 3 taken 22287 times.
|
252190 | if (FRAME_MBAFF(h) && IS_INTERLACED(*mb_type)) { |
588 | 20122 | map_col_to_list0[0] = sl->map_col_to_list0_field[sl->mb_y & 1][0]; | |
589 | 20122 | map_col_to_list0[1] = sl->map_col_to_list0_field[sl->mb_y & 1][1]; | |
590 | 20122 | dist_scale_factor = sl->dist_scale_factor_field[sl->mb_y & 1]; | |
591 | } | ||
592 | 252190 | ref_offset = (sl->ref_list[1][0].parent->mbaff << 4) & (mb_type_col[0] >> 3); | |
593 | |||
594 |
2/2✓ Branch 0 taken 32266 times.
✓ Branch 1 taken 219924 times.
|
252190 | if (IS_INTERLACED(*mb_type) != IS_INTERLACED(mb_type_col[0])) { |
595 |
2/2✓ Branch 0 taken 15717 times.
✓ Branch 1 taken 16549 times.
|
32266 | int y_shift = 2 * !IS_INTERLACED(*mb_type); |
596 | assert(h->ps.sps->direct_8x8_inference_flag); | ||
597 | |||
598 |
2/2✓ Branch 0 taken 129064 times.
✓ Branch 1 taken 32266 times.
|
161330 | for (i8 = 0; i8 < 4; i8++) { |
599 | 129064 | const int x8 = i8 & 1; | |
600 | 129064 | const int y8 = i8 >> 1; | |
601 | int ref0, scale; | ||
602 | 129064 | const int16_t (*l1mv)[2] = l1mv0; | |
603 | |||
604 |
4/4✓ Branch 0 taken 43260 times.
✓ Branch 1 taken 85804 times.
✓ Branch 2 taken 22674 times.
✓ Branch 3 taken 20586 times.
|
129064 | if (is_b8x8 && !IS_DIRECT(sl->sub_mb_type[i8])) |
605 | 22674 | continue; | |
606 | 106390 | sl->sub_mb_type[i8] = sub_mb_type; | |
607 | |||
608 | 106390 | fill_rectangle(&sl->ref_cache[1][scan8[i8 * 4]], 2, 2, 8, 0, 1); | |
609 |
2/2✓ Branch 0 taken 25592 times.
✓ Branch 1 taken 80798 times.
|
106390 | if (IS_INTRA(mb_type_col[y8])) { |
610 | 25592 | fill_rectangle(&sl->ref_cache[0][scan8[i8 * 4]], 2, 2, 8, 0, 1); | |
611 | 25592 | fill_rectangle(&sl->mv_cache[0][scan8[i8 * 4]], 2, 2, 8, 0, 4); | |
612 | 25592 | fill_rectangle(&sl->mv_cache[1][scan8[i8 * 4]], 2, 2, 8, 0, 4); | |
613 | 25592 | continue; | |
614 | } | ||
615 | |||
616 | 80798 | ref0 = l1ref0[x8 + y8 * b8_stride]; | |
617 |
2/2✓ Branch 0 taken 80605 times.
✓ Branch 1 taken 193 times.
|
80798 | if (ref0 >= 0) |
618 | 80605 | ref0 = map_col_to_list0[0][ref0 + ref_offset]; | |
619 | else { | ||
620 | 193 | ref0 = map_col_to_list0[1][l1ref1[x8 + y8 * b8_stride] + | |
621 | ref_offset]; | ||
622 | 193 | l1mv = l1mv1; | |
623 | } | ||
624 | 80798 | scale = dist_scale_factor[ref0]; | |
625 | 80798 | fill_rectangle(&sl->ref_cache[0][scan8[i8 * 4]], 2, 2, 8, | |
626 | ref0, 1); | ||
627 | |||
628 | { | ||
629 | 80798 | const int16_t *mv_col = l1mv[x8 * 3 + y8 * b4_stride]; | |
630 | 80798 | int my_col = (mv_col[1] * (1 << y_shift)) / 2; | |
631 | 80798 | int mx = (scale * mv_col[0] + 128) >> 8; | |
632 | 80798 | int my = (scale * my_col + 128) >> 8; | |
633 | 80798 | fill_rectangle(&sl->mv_cache[0][scan8[i8 * 4]], 2, 2, 8, | |
634 | pack16to32(mx, my), 4); | ||
635 | 80798 | fill_rectangle(&sl->mv_cache[1][scan8[i8 * 4]], 2, 2, 8, | |
636 | 80798 | pack16to32(mx - mv_col[0], my - my_col), 4); | |
637 | } | ||
638 | } | ||
639 | 32266 | return; | |
640 | } | ||
641 | |||
642 | /* one-to-one mv scaling */ | ||
643 | |||
644 |
2/2✓ Branch 0 taken 127488 times.
✓ Branch 1 taken 92436 times.
|
219924 | if (IS_16X16(*mb_type)) { |
645 | int ref, mv0, mv1; | ||
646 | |||
647 | 127488 | fill_rectangle(&sl->ref_cache[1][scan8[0]], 4, 4, 8, 0, 1); | |
648 |
2/2✓ Branch 0 taken 61077 times.
✓ Branch 1 taken 66411 times.
|
127488 | if (IS_INTRA(mb_type_col[0])) { |
649 | 61077 | ref = mv0 = mv1 = 0; | |
650 | } else { | ||
651 | 199182 | const int ref0 = l1ref0[0] >= 0 ? map_col_to_list0[0][l1ref0[0] + ref_offset] | |
652 |
2/2✓ Branch 0 taken 66360 times.
✓ Branch 1 taken 51 times.
|
66411 | : map_col_to_list0[1][l1ref1[0] + ref_offset]; |
653 | 66411 | const int scale = dist_scale_factor[ref0]; | |
654 |
2/2✓ Branch 0 taken 66360 times.
✓ Branch 1 taken 51 times.
|
66411 | const int16_t *mv_col = l1ref0[0] >= 0 ? l1mv0[0] : l1mv1[0]; |
655 | int mv_l0[2]; | ||
656 | 66411 | mv_l0[0] = (scale * mv_col[0] + 128) >> 8; | |
657 | 66411 | mv_l0[1] = (scale * mv_col[1] + 128) >> 8; | |
658 | 66411 | ref = ref0; | |
659 | 66411 | mv0 = pack16to32(mv_l0[0], mv_l0[1]); | |
660 | 66411 | mv1 = pack16to32(mv_l0[0] - mv_col[0], mv_l0[1] - mv_col[1]); | |
661 | } | ||
662 | 127488 | fill_rectangle(&sl->ref_cache[0][scan8[0]], 4, 4, 8, ref, 1); | |
663 | 127488 | fill_rectangle(&sl->mv_cache[0][scan8[0]], 4, 4, 8, mv0, 4); | |
664 | 127488 | fill_rectangle(&sl->mv_cache[1][scan8[0]], 4, 4, 8, mv1, 4); | |
665 | } else { | ||
666 |
2/2✓ Branch 0 taken 369744 times.
✓ Branch 1 taken 92436 times.
|
462180 | for (i8 = 0; i8 < 4; i8++) { |
667 | 369744 | const int x8 = i8 & 1; | |
668 | 369744 | const int y8 = i8 >> 1; | |
669 | int ref0, scale; | ||
670 | 369744 | const int16_t (*l1mv)[2] = l1mv0; | |
671 | |||
672 |
4/4✓ Branch 0 taken 163900 times.
✓ Branch 1 taken 205844 times.
✓ Branch 2 taken 92460 times.
✓ Branch 3 taken 71440 times.
|
369744 | if (is_b8x8 && !IS_DIRECT(sl->sub_mb_type[i8])) |
673 | 92460 | continue; | |
674 | 277284 | sl->sub_mb_type[i8] = sub_mb_type; | |
675 | 277284 | fill_rectangle(&sl->ref_cache[1][scan8[i8 * 4]], 2, 2, 8, 0, 1); | |
676 |
2/2✓ Branch 0 taken 22293 times.
✓ Branch 1 taken 254991 times.
|
277284 | if (IS_INTRA(mb_type_col[0])) { |
677 | 22293 | fill_rectangle(&sl->ref_cache[0][scan8[i8 * 4]], 2, 2, 8, 0, 1); | |
678 | 22293 | fill_rectangle(&sl->mv_cache[0][scan8[i8 * 4]], 2, 2, 8, 0, 4); | |
679 | 22293 | fill_rectangle(&sl->mv_cache[1][scan8[i8 * 4]], 2, 2, 8, 0, 4); | |
680 | 22293 | continue; | |
681 | } | ||
682 | |||
683 | assert(b8_stride == 2); | ||
684 | 254991 | ref0 = l1ref0[i8]; | |
685 |
2/2✓ Branch 0 taken 254971 times.
✓ Branch 1 taken 20 times.
|
254991 | if (ref0 >= 0) |
686 | 254971 | ref0 = map_col_to_list0[0][ref0 + ref_offset]; | |
687 | else { | ||
688 | 20 | ref0 = map_col_to_list0[1][l1ref1[i8] + ref_offset]; | |
689 | 20 | l1mv = l1mv1; | |
690 | } | ||
691 | 254991 | scale = dist_scale_factor[ref0]; | |
692 | |||
693 | 254991 | fill_rectangle(&sl->ref_cache[0][scan8[i8 * 4]], 2, 2, 8, | |
694 | ref0, 1); | ||
695 |
2/2✓ Branch 0 taken 229683 times.
✓ Branch 1 taken 25308 times.
|
254991 | if (IS_SUB_8X8(sub_mb_type)) { |
696 | 229683 | const int16_t *mv_col = l1mv[x8 * 3 + y8 * 3 * b4_stride]; | |
697 | 229683 | int mx = (scale * mv_col[0] + 128) >> 8; | |
698 | 229683 | int my = (scale * mv_col[1] + 128) >> 8; | |
699 | 229683 | fill_rectangle(&sl->mv_cache[0][scan8[i8 * 4]], 2, 2, 8, | |
700 | pack16to32(mx, my), 4); | ||
701 | 229683 | fill_rectangle(&sl->mv_cache[1][scan8[i8 * 4]], 2, 2, 8, | |
702 | 229683 | pack16to32(mx - mv_col[0], my - mv_col[1]), 4); | |
703 | } else { | ||
704 |
2/2✓ Branch 0 taken 101232 times.
✓ Branch 1 taken 25308 times.
|
126540 | for (i4 = 0; i4 < 4; i4++) { |
705 | 101232 | const int16_t *mv_col = l1mv[x8 * 2 + (i4 & 1) + | |
706 | 101232 | (y8 * 2 + (i4 >> 1)) * b4_stride]; | |
707 | 101232 | int16_t *mv_l0 = sl->mv_cache[0][scan8[i8 * 4 + i4]]; | |
708 | 101232 | mv_l0[0] = (scale * mv_col[0] + 128) >> 8; | |
709 | 101232 | mv_l0[1] = (scale * mv_col[1] + 128) >> 8; | |
710 | 101232 | AV_WN32A(sl->mv_cache[1][scan8[i8 * 4 + i4]], | |
711 | pack16to32(mv_l0[0] - mv_col[0], | ||
712 | mv_l0[1] - mv_col[1])); | ||
713 | } | ||
714 | } | ||
715 | } | ||
716 | } | ||
717 | } | ||
718 | } | ||
719 | |||
720 | 3228288 | void ff_h264_pred_direct_motion(const H264Context *const h, H264SliceContext *sl, | |
721 | int *mb_type) | ||
722 | { | ||
723 |
2/2✓ Branch 0 taken 2976098 times.
✓ Branch 1 taken 252190 times.
|
3228288 | if (sl->direct_spatial_mv_pred) |
724 | 2976098 | pred_spatial_direct_motion(h, sl, mb_type); | |
725 | else | ||
726 | 252190 | pred_temp_direct_motion(h, sl, mb_type); | |
727 | 3228288 | } | |
728 |