FFmpeg coverage

Directory: ../../../ffmpeg/
File: src/libavcodec/h264dsp_template.c
Date: 2025-01-20 09:27:23
Exec Total Coverage
Lines: 162 162 100.0%
Functions: 74 140 52.9%
Branches: 70 70 100.0%
Line Branch Exec Source
1 /*
2 * H.26L/H.264/AVC/JVT/14496-10/... encoder/decoder
3 * Copyright (c) 2003-2011 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 DSP functions.
25 * @author Michael Niedermayer <michaelni@gmx.at>
26 */
27
28 #include "bit_depth_template.c"
29
30 #define op_scale1(x) block[x] = av_clip_pixel( (block[x]*weight + offset) >> log2_denom )
31 #define op_scale2(x) dst[x] = av_clip_pixel( (src[x]*weights + dst[x]*weightd + offset) >> (log2_denom+1))
32 #define H264_WEIGHT(W) \
33 static void FUNCC(weight_h264_pixels ## W)(uint8_t *_block, ptrdiff_t stride, int height, \
34 int log2_denom, int weight, int offset) \
35 { \
36 int y; \
37 pixel *block = (pixel*)_block; \
38 stride >>= sizeof(pixel)-1; \
39 offset = (unsigned)offset << (log2_denom + (BIT_DEPTH-8)); \
40 if(log2_denom) offset += 1<<(log2_denom-1); \
41 for (y = 0; y < height; y++, block += stride) { \
42 op_scale1(0); \
43 op_scale1(1); \
44 if(W==2) continue; \
45 op_scale1(2); \
46 op_scale1(3); \
47 if(W==4) continue; \
48 op_scale1(4); \
49 op_scale1(5); \
50 op_scale1(6); \
51 op_scale1(7); \
52 if(W==8) continue; \
53 op_scale1(8); \
54 op_scale1(9); \
55 op_scale1(10); \
56 op_scale1(11); \
57 op_scale1(12); \
58 op_scale1(13); \
59 op_scale1(14); \
60 op_scale1(15); \
61 } \
62 } \
63 static void FUNCC(biweight_h264_pixels ## W)(uint8_t *_dst, uint8_t *_src, ptrdiff_t stride, int height, \
64 int log2_denom, int weightd, int weights, int offset) \
65 { \
66 int y; \
67 pixel *dst = (pixel*)_dst; \
68 pixel *src = (pixel*)_src; \
69 stride >>= sizeof(pixel)-1; \
70 offset = (unsigned)offset << (BIT_DEPTH-8); \
71 offset = (unsigned)((offset + 1) | 1) << log2_denom; \
72 for (y = 0; y < height; y++, dst += stride, src += stride) { \
73 op_scale2(0); \
74 op_scale2(1); \
75 if(W==2) continue; \
76 op_scale2(2); \
77 op_scale2(3); \
78 if(W==4) continue; \
79 op_scale2(4); \
80 op_scale2(5); \
81 op_scale2(6); \
82 op_scale2(7); \
83 if(W==8) continue; \
84 op_scale2(8); \
85 op_scale2(9); \
86 op_scale2(10); \
87 op_scale2(11); \
88 op_scale2(12); \
89 op_scale2(13); \
90 op_scale2(14); \
91 op_scale2(15); \
92 } \
93 }
94
95
4/4
✓ Branch 0 taken 3592355 times.
✓ Branch 1 taken 672212 times.
✓ Branch 2 taken 9614472 times.
✓ Branch 3 taken 635068 times.
 27758078 H264_WEIGHT(16)
96
4/4
✓ Branch 0 taken 4237432 times.
✓ Branch 1 taken 531375 times.
✓ Branch 2 taken 3562912 times.
✓ Branch 3 taken 475486 times.
 16663438 H264_WEIGHT(8)
97
4/4
✓ Branch 0 taken 1273110 times.
✓ Branch 1 taken 151816 times.
✓ Branch 2 taken 3111408 times.
✓ Branch 3 taken 697530 times.
 9072668 H264_WEIGHT(4)
98
4/4
✓ Branch 0 taken 642584 times.
✓ Branch 1 taken 98228 times.
✓ Branch 2 taken 1261776 times.
✓ Branch 3 taken 447120 times.
 4005176 H264_WEIGHT(2)
99
100 #undef op_scale1
101 #undef op_scale2
102 #undef H264_WEIGHT
103
104 75246130 static av_always_inline av_flatten void FUNCC(h264_loop_filter_luma)(uint8_t *p_pix, ptrdiff_t xstride, ptrdiff_t ystride, int inner_iters, int alpha, int beta, int8_t *tc0)
105 {
106 75246130 pixel *pix = (pixel*)p_pix;
107 int i, d;
108 75246130 xstride >>= sizeof(pixel)-1;
109 75246130 ystride >>= sizeof(pixel)-1;
110 75246130 alpha <<= BIT_DEPTH - 8;
111 75246130 beta <<= BIT_DEPTH - 8;
112
2/2
✓ Branch 0 taken 150492260 times.
✓ Branch 1 taken 37623065 times.
 376230650 for( i = 0; i < 4; i++ ) {
113 300984520 const int tc_orig = tc0[i] * (1 << (BIT_DEPTH - 8));
114
2/2
✓ Branch 0 taken 25249971 times.
✓ Branch 1 taken 125242289 times.
 300984520 if( tc_orig < 0 ) {
115 50499942 pix += inner_iters*ystride;
116 50499942 continue;
117 }
118
2/2
✓ Branch 0 taken 498564508 times.
✓ Branch 1 taken 125242289 times.
 1247613594 for( d = 0; d < inner_iters; d++ ) {
119 997129016 const int p0 = pix[-1*xstride];
120 997129016 const int p1 = pix[-2*xstride];
121 997129016 const int p2 = pix[-3*xstride];
122 997129016 const int q0 = pix[0];
123 997129016 const int q1 = pix[1*xstride];
124 997129016 const int q2 = pix[2*xstride];
125
126
2/2
✓ Branch 0 taken 427338042 times.
✓ Branch 1 taken 71226466 times.
 997129016 if( FFABS( p0 - q0 ) < alpha &&
127
2/2
✓ Branch 0 taken 321257967 times.
✓ Branch 1 taken 106080075 times.
 854676084 FFABS( p1 - p0 ) < beta &&
128
2/2
✓ Branch 0 taken 268951564 times.
✓ Branch 1 taken 52306403 times.
 642515934 FFABS( q1 - q0 ) < beta ) {
129
130 537903128 int tc = tc_orig;
131 int i_delta;
132
133
2/2
✓ Branch 0 taken 227971735 times.
✓ Branch 1 taken 40979829 times.
 537903128 if( FFABS( p2 - p0 ) < beta ) {
134
2/2
✓ Branch 0 taken 213176554 times.
✓ Branch 1 taken 14795181 times.
 455943470 if(tc_orig)
135 426353108 pix[-2*xstride] = p1 + av_clip( (( p2 + ( ( p0 + q0 + 1 ) >> 1 ) ) >> 1) - p1, -tc_orig, tc_orig );
136 455943470 tc++;
137 }
138
2/2
✓ Branch 0 taken 226809812 times.
✓ Branch 1 taken 42141752 times.
 537903128 if( FFABS( q2 - q0 ) < beta ) {
139
2/2
✓ Branch 0 taken 212041746 times.
✓ Branch 1 taken 14768066 times.
 453619624 if(tc_orig)
140 424083492 pix[ xstride] = q1 + av_clip( (( q2 + ( ( p0 + q0 + 1 ) >> 1 ) ) >> 1) - q1, -tc_orig, tc_orig );
141 453619624 tc++;
142 }
143
144 537903128 i_delta = av_clip( (((q0 - p0 ) * 4) + (p1 - q1) + 4) >> 3, -tc, tc );
145 537903128 pix[-xstride] = av_clip_pixel( p0 + i_delta ); /* p0' */
146 537903128 pix[0] = av_clip_pixel( q0 - i_delta ); /* q0' */
147 }
148 997129016 pix += ystride;
149 }
150 }
151 75246130 }
152 38185250 static void FUNCC(h264_v_loop_filter_luma)(uint8_t *pix, ptrdiff_t stride, int alpha, int beta, int8_t *tc0)
153 {
154 38185250 FUNCC(h264_loop_filter_luma)(pix, stride, sizeof(pixel), 4, alpha, beta, tc0);
155 38185250 }
156 36459718 static void FUNCC(h264_h_loop_filter_luma)(uint8_t *pix, ptrdiff_t stride, int alpha, int beta, int8_t *tc0)
157 {
158 36459718 FUNCC(h264_loop_filter_luma)(pix, sizeof(pixel), stride, 4, alpha, beta, tc0);
159 36459718 }
160 601162 static void FUNCC(h264_h_loop_filter_luma_mbaff)(uint8_t *pix, ptrdiff_t stride, int alpha, int beta, int8_t *tc0)
161 {
162 601162 FUNCC(h264_loop_filter_luma)(pix, sizeof(pixel), stride, 2, alpha, beta, tc0);
163 601162 }
164
165 11272660 static av_always_inline av_flatten void FUNCC(h264_loop_filter_luma_intra)(uint8_t *p_pix, ptrdiff_t xstride, ptrdiff_t ystride, int inner_iters, int alpha, int beta)
166 {
167 11272660 pixel *pix = (pixel*)p_pix;
168 int d;
169 11272660 xstride >>= sizeof(pixel)-1;
170 11272660 ystride >>= sizeof(pixel)-1;
171 11272660 alpha <<= BIT_DEPTH - 8;
172 11272660 beta <<= BIT_DEPTH - 8;
173
2/2
✓ Branch 0 taken 87174472 times.
✓ Branch 1 taken 5636330 times.
 185621604 for( d = 0; d < 4 * inner_iters; d++ ) {
174 174348944 const int p2 = pix[-3*xstride];
175 174348944 const int p1 = pix[-2*xstride];
176 174348944 const int p0 = pix[-1*xstride];
177
178 174348944 const int q0 = pix[ 0*xstride];
179 174348944 const int q1 = pix[ 1*xstride];
180 174348944 const int q2 = pix[ 2*xstride];
181
182
2/2
✓ Branch 0 taken 76672761 times.
✓ Branch 1 taken 10501711 times.
 174348944 if( FFABS( p0 - q0 ) < alpha &&
183
2/2
✓ Branch 0 taken 61467335 times.
✓ Branch 1 taken 15205426 times.
 153345522 FFABS( p1 - p0 ) < beta &&
184
2/2
✓ Branch 0 taken 52937720 times.
✓ Branch 1 taken 8529615 times.
 122934670 FFABS( q1 - q0 ) < beta ) {
185
186
2/2
✓ Branch 0 taken 43988452 times.
✓ Branch 1 taken 8949268 times.
 105875440 if(FFABS( p0 - q0 ) < (( alpha >> 2 ) + 2 )){
187
2/2
✓ Branch 0 taken 39606022 times.
✓ Branch 1 taken 4382430 times.
 87976904 if( FFABS( p2 - p0 ) < beta)
188 {
189 79212044 const int p3 = pix[-4*xstride];
190 /* p0', p1', p2' */
191 79212044 pix[-1*xstride] = ( p2 + 2*p1 + 2*p0 + 2*q0 + q1 + 4 ) >> 3;
192 79212044 pix[-2*xstride] = ( p2 + p1 + p0 + q0 + 2 ) >> 2;
193 79212044 pix[-3*xstride] = ( 2*p3 + 3*p2 + p1 + p0 + q0 + 4 ) >> 3;
194 } else {
195 /* p0' */
196 8764860 pix[-1*xstride] = ( 2*p1 + p0 + q1 + 2 ) >> 2;
197 }
198
2/2
✓ Branch 0 taken 39376708 times.
✓ Branch 1 taken 4611744 times.
 87976904 if( FFABS( q2 - q0 ) < beta)
199 {
200 78753416 const int q3 = pix[3*xstride];
201 /* q0', q1', q2' */
202 78753416 pix[0*xstride] = ( p1 + 2*p0 + 2*q0 + 2*q1 + q2 + 4 ) >> 3;
203 78753416 pix[1*xstride] = ( p0 + q0 + q1 + q2 + 2 ) >> 2;
204 78753416 pix[2*xstride] = ( 2*q3 + 3*q2 + q1 + q0 + p0 + 4 ) >> 3;
205 } else {
206 /* q0' */
207 9223488 pix[0*xstride] = ( 2*q1 + q0 + p1 + 2 ) >> 2;
208 }
209 }else{
210 /* p0', q0' */
211 17898536 pix[-1*xstride] = ( 2*p1 + p0 + q1 + 2 ) >> 2;
212 17898536 pix[ 0*xstride] = ( 2*q1 + q0 + p1 + 2 ) >> 2;
213 }
214 }
215 174348944 pix += ystride;
216 }
217 11272660 }
218 4657142 static void FUNCC(h264_v_loop_filter_luma_intra)(uint8_t *pix, ptrdiff_t stride, int alpha, int beta)
219 {
220 4657142 FUNCC(h264_loop_filter_luma_intra)(pix, stride, sizeof(pixel), 4, alpha, beta);
221 4657142 }
222 5863816 static void FUNCC(h264_h_loop_filter_luma_intra)(uint8_t *pix, ptrdiff_t stride, int alpha, int beta)
223 {
224 5863816 FUNCC(h264_loop_filter_luma_intra)(pix, sizeof(pixel), stride, 4, alpha, beta);
225 5863816 }
226 751702 static void FUNCC(h264_h_loop_filter_luma_mbaff_intra)(uint8_t *pix, ptrdiff_t stride, int alpha, int beta)
227 {
228 751702 FUNCC(h264_loop_filter_luma_intra)(pix, sizeof(pixel), stride, 2, alpha, beta);
229 751702 }
230
231 89451538 static av_always_inline av_flatten void FUNCC(h264_loop_filter_chroma)(uint8_t *p_pix, ptrdiff_t xstride, ptrdiff_t ystride, int inner_iters, int alpha, int beta, int8_t *tc0)
232 {
233 89451538 pixel *pix = (pixel*)p_pix;
234 int i, d;
235 89451538 alpha <<= BIT_DEPTH - 8;
236 89451538 beta <<= BIT_DEPTH - 8;
237 89451538 xstride >>= sizeof(pixel)-1;
238 89451538 ystride >>= sizeof(pixel)-1;
239
2/2
✓ Branch 0 taken 178903076 times.
✓ Branch 1 taken 44725769 times.
 447257690 for( i = 0; i < 4; i++ ) {
240 357806152 const int tc = ((tc0[i] - 1U) << (BIT_DEPTH - 8)) + 1;
241
2/2
✓ Branch 0 taken 23871588 times.
✓ Branch 1 taken 155031488 times.
 357806152 if( tc <= 0 ) {
242 47743176 pix += inner_iters*ystride;
243 47743176 continue;
244 }
245
2/2
✓ Branch 0 taken 318529434 times.
✓ Branch 1 taken 155031488 times.
 947121844 for( d = 0; d < inner_iters; d++ ) {
246 637058868 const int p0 = pix[-1*xstride];
247 637058868 const int p1 = pix[-2*xstride];
248 637058868 const int q0 = pix[0];
249 637058868 const int q1 = pix[1*xstride];
250
251
2/2
✓ Branch 0 taken 311962201 times.
✓ Branch 1 taken 6567233 times.
 637058868 if( FFABS( p0 - q0 ) < alpha &&
252
2/2
✓ Branch 0 taken 293346691 times.
✓ Branch 1 taken 18615510 times.
 623924402 FFABS( p1 - p0 ) < beta &&
253
2/2
✓ Branch 0 taken 280894573 times.
✓ Branch 1 taken 12452118 times.
 586693382 FFABS( q1 - q0 ) < beta ) {
254
255 561789146 int delta = av_clip( ((q0 - p0) * 4 + (p1 - q1) + 4) >> 3, -tc, tc );
256
257 561789146 pix[-xstride] = av_clip_pixel( p0 + delta ); /* p0' */
258 561789146 pix[0] = av_clip_pixel( q0 - delta ); /* q0' */
259 }
260 637058868 pix += ystride;
261 }
262 }
263 89451538 }
264 48557272 static void FUNCC(h264_v_loop_filter_chroma)(uint8_t *pix, ptrdiff_t stride, int alpha, int beta, int8_t *tc0)
265 {
266 48557272 FUNCC(h264_loop_filter_chroma)(pix, stride, sizeof(pixel), 2, alpha, beta, tc0);
267 48557272 }
268 36819738 static void FUNCC(h264_h_loop_filter_chroma)(uint8_t *pix, ptrdiff_t stride, int alpha, int beta, int8_t *tc0)
269 {
270 36819738 FUNCC(h264_loop_filter_chroma)(pix, sizeof(pixel), stride, 2, alpha, beta, tc0);
271 36819738 }
272 1300824 static void FUNCC(h264_h_loop_filter_chroma_mbaff)(uint8_t *pix, ptrdiff_t stride, int alpha, int beta, int8_t *tc0)
273 {
274 1300824 FUNCC(h264_loop_filter_chroma)(pix, sizeof(pixel), stride, 1, alpha, beta, tc0);
275 1300824 }
276 2773272 static void FUNCC(h264_h_loop_filter_chroma422)(uint8_t *pix, ptrdiff_t stride, int alpha, int beta, int8_t *tc0)
277 {
278 2773272 FUNCC(h264_loop_filter_chroma)(pix, sizeof(pixel), stride, 4, alpha, beta, tc0);
279 2773272 }
280 432 static void FUNCC(h264_h_loop_filter_chroma422_mbaff)(uint8_t *pix, ptrdiff_t stride, int alpha, int beta, int8_t *tc0)
281 {
282 432 FUNCC(h264_loop_filter_chroma)(pix, sizeof(pixel), stride, 2, alpha, beta, tc0);
283 432 }
284
285 21986478 static av_always_inline av_flatten void FUNCC(h264_loop_filter_chroma_intra)(uint8_t *p_pix, ptrdiff_t xstride, ptrdiff_t ystride, int inner_iters, int alpha, int beta)
286 {
287 21986478 pixel *pix = (pixel*)p_pix;
288 int d;
289 21986478 xstride >>= sizeof(pixel)-1;
290 21986478 ystride >>= sizeof(pixel)-1;
291 21986478 alpha <<= BIT_DEPTH - 8;
292 21986478 beta <<= BIT_DEPTH - 8;
293
2/2
✓ Branch 0 taken 95815496 times.
✓ Branch 1 taken 10993239 times.
 213617470 for( d = 0; d < 4 * inner_iters; d++ ) {
294 191630992 const int p0 = pix[-1*xstride];
295 191630992 const int p1 = pix[-2*xstride];
296 191630992 const int q0 = pix[0];
297 191630992 const int q1 = pix[1*xstride];
298
299
2/2
✓ Branch 0 taken 92935886 times.
✓ Branch 1 taken 2879610 times.
 191630992 if( FFABS( p0 - q0 ) < alpha &&
300
2/2
✓ Branch 0 taken 87898759 times.
✓ Branch 1 taken 5037127 times.
 185871772 FFABS( p1 - p0 ) < beta &&
301
2/2
✓ Branch 0 taken 84352935 times.
✓ Branch 1 taken 3545824 times.
 175797518 FFABS( q1 - q0 ) < beta ) {
302
303 168705870 pix[-xstride] = ( 2*p1 + p0 + q1 + 2 ) >> 2; /* p0' */
304 168705870 pix[0] = ( 2*q1 + q0 + p1 + 2 ) >> 2; /* q0' */
305 }
306 191630992 pix += ystride;
307 }
308 21986478 }
309 9011836 static void FUNCC(h264_v_loop_filter_chroma_intra)(uint8_t *pix, ptrdiff_t stride, int alpha, int beta)
310 {
311 9011836 FUNCC(h264_loop_filter_chroma_intra)(pix, stride, sizeof(pixel), 2, alpha, beta);
312 9011836 }
313 8972342 static void FUNCC(h264_h_loop_filter_chroma_intra)(uint8_t *pix, ptrdiff_t stride, int alpha, int beta)
314 {
315 8972342 FUNCC(h264_loop_filter_chroma_intra)(pix, sizeof(pixel), stride, 2, alpha, beta);
316 8972342 }
317 1063536 static void FUNCC(h264_h_loop_filter_chroma_mbaff_intra)(uint8_t *pix, ptrdiff_t stride, int alpha, int beta)
318 {
319 1063536 FUNCC(h264_loop_filter_chroma_intra)(pix, sizeof(pixel), stride, 1, alpha, beta);
320 1063536 }
321 2499164 static void FUNCC(h264_h_loop_filter_chroma422_intra)(uint8_t *pix, ptrdiff_t stride, int alpha, int beta)
322 {
323 2499164 FUNCC(h264_loop_filter_chroma_intra)(pix, sizeof(pixel), stride, 4, alpha, beta);
324 2499164 }
325 439600 static void FUNCC(h264_h_loop_filter_chroma422_mbaff_intra)(uint8_t *pix, ptrdiff_t stride, int alpha, int beta)
326 {
327 439600 FUNCC(h264_loop_filter_chroma_intra)(pix, sizeof(pixel), stride, 2, alpha, beta);
328 439600 }
329