FFmpeg coverage


Directory: ../../../ffmpeg/
File: src/libavcodec/snow.c
Date: 2024-04-19 07:31:02
Exec Total Coverage
Lines: 329 352 93.5%
Functions: 14 19 73.7%
Branches: 177 212 83.5%

Line Branch Exec Source
1 /*
2 * Copyright (C) 2004 Michael Niedermayer <michaelni@gmx.at>
3 *
4 * This file is part of FFmpeg.
5 *
6 * FFmpeg is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * FFmpeg is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with FFmpeg; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19 */
20
21 #include "libavutil/log.h"
22 #include "libavutil/mem.h"
23 #include "libavutil/thread.h"
24 #include "avcodec.h"
25 #include "snow_dwt.h"
26 #include "snow.h"
27 #include "snowdata.h"
28
29
30 710775 void ff_snow_inner_add_yblock(const uint8_t *obmc, const int obmc_stride, uint8_t * * block, int b_w, int b_h,
31 int src_x, int src_y, int src_stride, slice_buffer * sb, int add, uint8_t * dst8){
32 int y, x;
33 IDWTELEM * dst;
34
2/2
✓ Branch 0 taken 3810240 times.
✓ Branch 1 taken 710775 times.
4521015 for(y=0; y<b_h; y++){
35 //FIXME ugly misuse of obmc_stride
36 3810240 const uint8_t *obmc1= obmc + y*obmc_stride;
37 3810240 const uint8_t *obmc2= obmc1+ (obmc_stride>>1);
38 3810240 const uint8_t *obmc3= obmc1+ obmc_stride*(obmc_stride>>1);
39 3810240 const uint8_t *obmc4= obmc3+ (obmc_stride>>1);
40
1/2
✓ Branch 0 taken 3810240 times.
✗ Branch 1 not taken.
3810240 dst = slice_buffer_get_line(sb, src_y + y);
41
2/2
✓ Branch 0 taken 23846400 times.
✓ Branch 1 taken 3810240 times.
27656640 for(x=0; x<b_w; x++){
42 23846400 int v= obmc1[x] * block[3][x + y*src_stride]
43 23846400 +obmc2[x] * block[2][x + y*src_stride]
44 23846400 +obmc3[x] * block[1][x + y*src_stride]
45 23846400 +obmc4[x] * block[0][x + y*src_stride];
46
47 23846400 v <<= 8 - LOG2_OBMC_MAX;
48 if(FRAC_BITS != 8){
49 23846400 v >>= 8 - FRAC_BITS;
50 }
51
1/2
✓ Branch 0 taken 23846400 times.
✗ Branch 1 not taken.
23846400 if(add){
52 23846400 v += dst[x + src_x];
53 23846400 v = (v + (1<<(FRAC_BITS-1))) >> FRAC_BITS;
54
2/2
✓ Branch 0 taken 1126 times.
✓ Branch 1 taken 23845274 times.
23846400 if(v&(~255)) v= ~(v>>31);
55 23846400 dst8[x + y*src_stride] = v;
56 }else{
57 dst[x + src_x] -= v;
58 }
59 }
60 }
61 710775 }
62
63 103 void ff_snow_reset_contexts(SnowContext *s){ //FIXME better initial contexts
64 int plane_index, level, orientation;
65
66
2/2
✓ Branch 0 taken 309 times.
✓ Branch 1 taken 103 times.
412 for(plane_index=0; plane_index<3; plane_index++){
67
2/2
✓ Branch 0 taken 2472 times.
✓ Branch 1 taken 309 times.
2781 for(level=0; level<MAX_DECOMPOSITIONS; level++){
68
2/2
✓ Branch 0 taken 7725 times.
✓ Branch 1 taken 2472 times.
10197 for(orientation=level ? 1:0; orientation<4; orientation++){
69 7725 memset(s->plane[plane_index].band[level][orientation].state, MID_STATE, sizeof(s->plane[plane_index].band[level][orientation].state));
70 }
71 }
72 }
73 103 memset(s->header_state, MID_STATE, sizeof(s->header_state));
74 103 memset(s->block_state, MID_STATE, sizeof(s->block_state));
75 103 }
76
77 472 int ff_snow_alloc_blocks(SnowContext *s){
78 472 int w= AV_CEIL_RSHIFT(s->avctx->width, LOG2_MB_SIZE);
79 472 int h= AV_CEIL_RSHIFT(s->avctx->height, LOG2_MB_SIZE);
80
81 472 s->b_width = w;
82 472 s->b_height= h;
83
84 472 av_free(s->block);
85 472 s->block = av_calloc(w * h, sizeof(*s->block) << (s->block_max_depth*2));
86
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 472 times.
472 if (!s->block)
87 return AVERROR(ENOMEM);
88
89 472 return 0;
90 }
91
92 3226960 static void mc_block(Plane *p, uint8_t *dst, const uint8_t *src, int stride, int b_w, int b_h, int dx, int dy){
93 static const uint8_t weight[64]={
94 8,7,6,5,4,3,2,1,
95 7,7,0,0,0,0,0,1,
96 6,0,6,0,0,0,2,0,
97 5,0,0,5,0,3,0,0,
98 4,0,0,0,4,0,0,0,
99 3,0,0,5,0,3,0,0,
100 2,0,6,0,0,0,2,0,
101 1,7,0,0,0,0,0,1,
102 };
103
104 static const uint8_t brane[256]={
105 0x00,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x11,0x12,0x12,0x12,0x12,0x12,0x12,0x12,
106 0x04,0x05,0xcc,0xcc,0xcc,0xcc,0xcc,0x41,0x15,0x16,0xcc,0xcc,0xcc,0xcc,0xcc,0x52,
107 0x04,0xcc,0x05,0xcc,0xcc,0xcc,0x41,0xcc,0x15,0xcc,0x16,0xcc,0xcc,0xcc,0x52,0xcc,
108 0x04,0xcc,0xcc,0x05,0xcc,0x41,0xcc,0xcc,0x15,0xcc,0xcc,0x16,0xcc,0x52,0xcc,0xcc,
109 0x04,0xcc,0xcc,0xcc,0x41,0xcc,0xcc,0xcc,0x15,0xcc,0xcc,0xcc,0x16,0xcc,0xcc,0xcc,
110 0x04,0xcc,0xcc,0x41,0xcc,0x05,0xcc,0xcc,0x15,0xcc,0xcc,0x52,0xcc,0x16,0xcc,0xcc,
111 0x04,0xcc,0x41,0xcc,0xcc,0xcc,0x05,0xcc,0x15,0xcc,0x52,0xcc,0xcc,0xcc,0x16,0xcc,
112 0x04,0x41,0xcc,0xcc,0xcc,0xcc,0xcc,0x05,0x15,0x52,0xcc,0xcc,0xcc,0xcc,0xcc,0x16,
113 0x44,0x45,0x45,0x45,0x45,0x45,0x45,0x45,0x55,0x56,0x56,0x56,0x56,0x56,0x56,0x56,
114 0x48,0x49,0xcc,0xcc,0xcc,0xcc,0xcc,0x85,0x59,0x5A,0xcc,0xcc,0xcc,0xcc,0xcc,0x96,
115 0x48,0xcc,0x49,0xcc,0xcc,0xcc,0x85,0xcc,0x59,0xcc,0x5A,0xcc,0xcc,0xcc,0x96,0xcc,
116 0x48,0xcc,0xcc,0x49,0xcc,0x85,0xcc,0xcc,0x59,0xcc,0xcc,0x5A,0xcc,0x96,0xcc,0xcc,
117 0x48,0xcc,0xcc,0xcc,0x49,0xcc,0xcc,0xcc,0x59,0xcc,0xcc,0xcc,0x96,0xcc,0xcc,0xcc,
118 0x48,0xcc,0xcc,0x85,0xcc,0x49,0xcc,0xcc,0x59,0xcc,0xcc,0x96,0xcc,0x5A,0xcc,0xcc,
119 0x48,0xcc,0x85,0xcc,0xcc,0xcc,0x49,0xcc,0x59,0xcc,0x96,0xcc,0xcc,0xcc,0x5A,0xcc,
120 0x48,0x85,0xcc,0xcc,0xcc,0xcc,0xcc,0x49,0x59,0x96,0xcc,0xcc,0xcc,0xcc,0xcc,0x5A,
121 };
122
123 static const uint8_t needs[16]={
124 0,1,0,0,
125 2,4,2,0,
126 0,1,0,0,
127 15
128 };
129
130 int x, y, b, r, l;
131 int16_t tmpIt [64*(32+HTAPS_MAX)];
132 uint8_t tmp2t[3][64*(32+HTAPS_MAX)];
133 3226960 int16_t *tmpI= tmpIt;
134 3226960 uint8_t *tmp2= tmp2t[0];
135 const uint8_t *hpel[11];
136 av_assert2(dx<16 && dy<16);
137 3226960 r= brane[dx + 16*dy]&15;
138 3226960 l= brane[dx + 16*dy]>>4;
139
140 3226960 b= needs[l] | needs[r];
141
3/4
✓ Branch 0 taken 3206706 times.
✓ Branch 1 taken 20254 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 3206706 times.
3226960 if(p && !p->diag_mc)
142 b= 15;
143
144
2/2
✓ Branch 0 taken 2890899 times.
✓ Branch 1 taken 336061 times.
3226960 if(b&5){
145
2/2
✓ Branch 0 taken 32853085 times.
✓ Branch 1 taken 2890899 times.
35743984 for(y=0; y < b_h+HTAPS_MAX-1; y++){
146
2/2
✓ Branch 0 taken 150497304 times.
✓ Branch 1 taken 32853085 times.
183350389 for(x=0; x < b_w; x++){
147 150497304 int a_1=src[x + HTAPS_MAX/2-4];
148 150497304 int a0= src[x + HTAPS_MAX/2-3];
149 150497304 int a1= src[x + HTAPS_MAX/2-2];
150 150497304 int a2= src[x + HTAPS_MAX/2-1];
151 150497304 int a3= src[x + HTAPS_MAX/2+0];
152 150497304 int a4= src[x + HTAPS_MAX/2+1];
153 150497304 int a5= src[x + HTAPS_MAX/2+2];
154 150497304 int a6= src[x + HTAPS_MAX/2+3];
155 150497304 int am=0;
156
3/4
✓ Branch 0 taken 144872424 times.
✓ Branch 1 taken 5624880 times.
✓ Branch 2 taken 144872424 times.
✗ Branch 3 not taken.
150497304 if(!p || p->fast_mc){
157 150497304 am= 20*(a2+a3) - 5*(a1+a4) + (a0+a5);
158 150497304 tmpI[x]= am;
159 150497304 am= (am+16)>>5;
160 }else{
161 am= p->hcoeff[0]*(a2+a3) + p->hcoeff[1]*(a1+a4) + p->hcoeff[2]*(a0+a5) + p->hcoeff[3]*(a_1+a6);
162 tmpI[x]= am;
163 am= (am+32)>>6;
164 }
165
166
2/2
✓ Branch 0 taken 6243 times.
✓ Branch 1 taken 150491061 times.
150497304 if(am&(~255)) am= ~(am>>31);
167 150497304 tmp2[x]= am;
168 }
169 32853085 tmpI+= 64;
170 32853085 tmp2+= 64;
171 32853085 src += stride;
172 }
173 2890899 src -= stride*y;
174 }
175 3226960 src += HTAPS_MAX/2 - 1;
176 3226960 tmp2= tmp2t[1];
177
178
2/2
✓ Branch 0 taken 1986073 times.
✓ Branch 1 taken 1240887 times.
3226960 if(b&2){
179
2/2
✓ Branch 0 taken 8629972 times.
✓ Branch 1 taken 1986073 times.
10616045 for(y=0; y < b_h; y++){
180
2/2
✓ Branch 0 taken 49598708 times.
✓ Branch 1 taken 8629972 times.
58228680 for(x=0; x < b_w+1; x++){
181 49598708 int a_1=src[x + (HTAPS_MAX/2-4)*stride];
182 49598708 int a0= src[x + (HTAPS_MAX/2-3)*stride];
183 49598708 int a1= src[x + (HTAPS_MAX/2-2)*stride];
184 49598708 int a2= src[x + (HTAPS_MAX/2-1)*stride];
185 49598708 int a3= src[x + (HTAPS_MAX/2+0)*stride];
186 49598708 int a4= src[x + (HTAPS_MAX/2+1)*stride];
187 49598708 int a5= src[x + (HTAPS_MAX/2+2)*stride];
188 49598708 int a6= src[x + (HTAPS_MAX/2+3)*stride];
189 49598708 int am=0;
190
3/4
✓ Branch 0 taken 48247140 times.
✓ Branch 1 taken 1351568 times.
✓ Branch 2 taken 48247140 times.
✗ Branch 3 not taken.
49598708 if(!p || p->fast_mc)
191 49598708 am= (20*(a2+a3) - 5*(a1+a4) + (a0+a5) + 16)>>5;
192 else
193 am= (p->hcoeff[0]*(a2+a3) + p->hcoeff[1]*(a1+a4) + p->hcoeff[2]*(a0+a5) + p->hcoeff[3]*(a_1+a6) + 32)>>6;
194
195
2/2
✓ Branch 0 taken 3684 times.
✓ Branch 1 taken 49595024 times.
49598708 if(am&(~255)) am= ~(am>>31);
196 49598708 tmp2[x]= am;
197 }
198 8629972 src += stride;
199 8629972 tmp2+= 64;
200 }
201 1986073 src -= stride*y;
202 }
203 3226960 src += stride*(HTAPS_MAX/2 - 1);
204 3226960 tmp2= tmp2t[2];
205 3226960 tmpI= tmpIt;
206
2/2
✓ Branch 0 taken 1948890 times.
✓ Branch 1 taken 1278070 times.
3226960 if(b&4){
207
2/2
✓ Branch 0 taken 8451144 times.
✓ Branch 1 taken 1948890 times.
10400034 for(y=0; y < b_h; y++){
208
2/2
✓ Branch 0 taken 40641424 times.
✓ Branch 1 taken 8451144 times.
49092568 for(x=0; x < b_w; x++){
209 40641424 int a_1=tmpI[x + (HTAPS_MAX/2-4)*64];
210 40641424 int a0= tmpI[x + (HTAPS_MAX/2-3)*64];
211 40641424 int a1= tmpI[x + (HTAPS_MAX/2-2)*64];
212 40641424 int a2= tmpI[x + (HTAPS_MAX/2-1)*64];
213 40641424 int a3= tmpI[x + (HTAPS_MAX/2+0)*64];
214 40641424 int a4= tmpI[x + (HTAPS_MAX/2+1)*64];
215 40641424 int a5= tmpI[x + (HTAPS_MAX/2+2)*64];
216 40641424 int a6= tmpI[x + (HTAPS_MAX/2+3)*64];
217 40641424 int am=0;
218
3/4
✓ Branch 0 taken 38144400 times.
✓ Branch 1 taken 2497024 times.
✓ Branch 2 taken 38144400 times.
✗ Branch 3 not taken.
40641424 if(!p || p->fast_mc)
219 40641424 am= (20*(a2+a3) - 5*(a1+a4) + (a0+a5) + 512)>>10;
220 else
221 am= (p->hcoeff[0]*(a2+a3) + p->hcoeff[1]*(a1+a4) + p->hcoeff[2]*(a0+a5) + p->hcoeff[3]*(a_1+a6) + 2048)>>12;
222
2/2
✓ Branch 0 taken 6658 times.
✓ Branch 1 taken 40634766 times.
40641424 if(am&(~255)) am= ~(am>>31);
223 40641424 tmp2[x]= am;
224 }
225 8451144 tmpI+= 64;
226 8451144 tmp2+= 64;
227 }
228 }
229
230 3226960 hpel[ 0]= src;
231 3226960 hpel[ 1]= tmp2t[0] + 64*(HTAPS_MAX/2-1);
232 3226960 hpel[ 2]= src + 1;
233
234 3226960 hpel[ 4]= tmp2t[1];
235 3226960 hpel[ 5]= tmp2t[2];
236 3226960 hpel[ 6]= tmp2t[1] + 1;
237
238 3226960 hpel[ 8]= src + stride;
239 3226960 hpel[ 9]= hpel[1] + 64;
240 3226960 hpel[10]= hpel[8] + 1;
241
242 #define MC_STRIDE(x) (needs[x] ? 64 : stride)
243
244
2/2
✓ Branch 0 taken 786570 times.
✓ Branch 1 taken 2440390 times.
3226960 if(b==15){
245 786570 int dxy = dx / 8 + dy / 8 * 4;
246 786570 const uint8_t *src1 = hpel[dxy ];
247 786570 const uint8_t *src2 = hpel[dxy + 1];
248 786570 const uint8_t *src3 = hpel[dxy + 4];
249 786570 const uint8_t *src4 = hpel[dxy + 5];
250
2/2
✓ Branch 0 taken 197148 times.
✓ Branch 1 taken 589422 times.
786570 int stride1 = MC_STRIDE(dxy);
251
2/2
✓ Branch 0 taken 201514 times.
✓ Branch 1 taken 585056 times.
786570 int stride2 = MC_STRIDE(dxy + 1);
252
2/2
✓ Branch 0 taken 190394 times.
✓ Branch 1 taken 596176 times.
786570 int stride3 = MC_STRIDE(dxy + 4);
253
2/2
✓ Branch 0 taken 197514 times.
✓ Branch 1 taken 589056 times.
786570 int stride4 = MC_STRIDE(dxy + 5);
254 786570 dx&=7;
255 786570 dy&=7;
256
2/2
✓ Branch 0 taken 3452584 times.
✓ Branch 1 taken 786570 times.
4239154 for(y=0; y < b_h; y++){
257
2/2
✓ Branch 0 taken 16499096 times.
✓ Branch 1 taken 3452584 times.
19951680 for(x=0; x < b_w; x++){
258 16499096 dst[x]= ((8-dx)*(8-dy)*src1[x] + dx*(8-dy)*src2[x]+
259 16499096 (8-dx)* dy *src3[x] + dx* dy *src4[x]+32)>>6;
260 }
261 3452584 src1+=stride1;
262 3452584 src2+=stride2;
263 3452584 src3+=stride3;
264 3452584 src4+=stride4;
265 3452584 dst +=stride;
266 }
267 }else{
268 2440390 const uint8_t *src1= hpel[l];
269 2440390 const uint8_t *src2= hpel[r];
270
2/2
✓ Branch 0 taken 598440 times.
✓ Branch 1 taken 1841950 times.
2440390 int stride1 = MC_STRIDE(l);
271
2/2
✓ Branch 0 taken 652892 times.
✓ Branch 1 taken 1787498 times.
2440390 int stride2 = MC_STRIDE(r);
272 2440390 int a= weight[((dx&7) + (8*(dy&7)))];
273 2440390 int b= 8-a;
274
2/2
✓ Branch 0 taken 10676392 times.
✓ Branch 1 taken 2440390 times.
13116782 for(y=0; y < b_h; y++){
275
2/2
✓ Branch 0 taken 52670328 times.
✓ Branch 1 taken 10676392 times.
63346720 for(x=0; x < b_w; x++){
276 52670328 dst[x]= (a*src1[x] + b*src2[x] + 4)>>3;
277 }
278 10676392 src1+=stride1;
279 10676392 src2+=stride2;
280 10676392 dst +=stride;
281 }
282 }
283 3226960 }
284
285 9284876 void ff_snow_pred_block(SnowContext *s, uint8_t *dst, uint8_t *tmp, ptrdiff_t stride, int sx, int sy, int b_w, int b_h, const BlockNode *block, int plane_index, int w, int h){
286
2/2
✓ Branch 0 taken 1201940 times.
✓ Branch 1 taken 8082936 times.
9284876 if(block->type & BLOCK_INTRA){
287 int x, y;
288 1201940 const unsigned color = block->color[plane_index];
289 1201940 const unsigned color4 = color*0x01010101;
290
2/2
✓ Branch 0 taken 18314 times.
✓ Branch 1 taken 1183626 times.
1201940 if(b_w==32){
291
2/2
✓ Branch 0 taken 586048 times.
✓ Branch 1 taken 18314 times.
604362 for(y=0; y < b_h; y++){
292 586048 *(uint32_t*)&dst[0 + y*stride]= color4;
293 586048 *(uint32_t*)&dst[4 + y*stride]= color4;
294 586048 *(uint32_t*)&dst[8 + y*stride]= color4;
295 586048 *(uint32_t*)&dst[12+ y*stride]= color4;
296 586048 *(uint32_t*)&dst[16+ y*stride]= color4;
297 586048 *(uint32_t*)&dst[20+ y*stride]= color4;
298 586048 *(uint32_t*)&dst[24+ y*stride]= color4;
299 586048 *(uint32_t*)&dst[28+ y*stride]= color4;
300 }
301
2/2
✓ Branch 0 taken 127575 times.
✓ Branch 1 taken 1056051 times.
1183626 }else if(b_w==16){
302
2/2
✓ Branch 0 taken 1911240 times.
✓ Branch 1 taken 127575 times.
2038815 for(y=0; y < b_h; y++){
303 1911240 *(uint32_t*)&dst[0 + y*stride]= color4;
304 1911240 *(uint32_t*)&dst[4 + y*stride]= color4;
305 1911240 *(uint32_t*)&dst[8 + y*stride]= color4;
306 1911240 *(uint32_t*)&dst[12+ y*stride]= color4;
307 }
308
2/2
✓ Branch 0 taken 521653 times.
✓ Branch 1 taken 534398 times.
1056051 }else if(b_w==8){
309
2/2
✓ Branch 0 taken 4084224 times.
✓ Branch 1 taken 521653 times.
4605877 for(y=0; y < b_h; y++){
310 4084224 *(uint32_t*)&dst[0 + y*stride]= color4;
311 4084224 *(uint32_t*)&dst[4 + y*stride]= color4;
312 }
313
2/2
✓ Branch 0 taken 533470 times.
✓ Branch 1 taken 928 times.
534398 }else if(b_w==4){
314
2/2
✓ Branch 0 taken 2176640 times.
✓ Branch 1 taken 533470 times.
2710110 for(y=0; y < b_h; y++){
315 2176640 *(uint32_t*)&dst[0 + y*stride]= color4;
316 }
317 }else{
318
2/2
✓ Branch 0 taken 3656 times.
✓ Branch 1 taken 928 times.
4584 for(y=0; y < b_h; y++){
319
2/2
✓ Branch 0 taken 7312 times.
✓ Branch 1 taken 3656 times.
10968 for(x=0; x < b_w; x++){
320 7312 dst[x + y*stride]= color;
321 }
322 }
323 }
324 }else{
325 8082936 const uint8_t *src = s->last_picture[block->ref]->data[plane_index];
326
2/2
✓ Branch 0 taken 5196540 times.
✓ Branch 1 taken 2886396 times.
8082936 const int scale= plane_index ? (2*s->mv_scale)>>s->chroma_h_shift : 2*s->mv_scale;
327 8082936 int mx= block->mx*scale;
328 8082936 int my= block->my*scale;
329 8082936 const int dx= mx&15;
330 8082936 const int dy= my&15;
331 8082936 const int tab_index= 3 - (b_w>>2) + (b_w>>4);
332 8082936 sx += (mx>>4) - (HTAPS_MAX/2-1);
333 8082936 sy += (my>>4) - (HTAPS_MAX/2-1);
334 8082936 src += sx + sy*stride;
335
3/4
✓ Branch 0 taken 8082936 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 7574313 times.
✓ Branch 3 taken 508623 times.
8082936 if( (unsigned)sx >= FFMAX(w - b_w - (HTAPS_MAX-2), 0)
336
3/4
✓ Branch 0 taken 7574313 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 640831 times.
✓ Branch 3 taken 6933482 times.
7574313 || (unsigned)sy >= FFMAX(h - b_h - (HTAPS_MAX-2), 0)){
337 1149454 s->vdsp.emulated_edge_mc(tmp + MB_SIZE, src,
338 stride, stride,
339 b_w+HTAPS_MAX-1, b_h+HTAPS_MAX-1,
340 sx, sy, w, h);
341 1149454 src= tmp + MB_SIZE;
342 }
343
344 av_assert2(s->chroma_h_shift == s->chroma_v_shift); // only one mv_scale
345
346 av_assert2((tab_index>=0 && tab_index<4) || b_w==32);
347
4/4
✓ Branch 0 taken 5927358 times.
✓ Branch 1 taken 2155578 times.
✓ Branch 2 taken 4876230 times.
✓ Branch 3 taken 1051128 times.
8082936 if( (dx&3) || (dy&3)
348
5/6
✓ Branch 0 taken 347714 times.
✓ Branch 1 taken 4528516 times.
✓ Branch 2 taken 203675 times.
✓ Branch 3 taken 144039 times.
✓ Branch 4 taken 203675 times.
✗ Branch 5 not taken.
4876230 || !(b_w == b_h || 2*b_w == b_h || b_w == 2*b_h)
349
1/2
✓ Branch 0 taken 4876230 times.
✗ Branch 1 not taken.
4876230 || (b_w&(b_w-1))
350
1/2
✓ Branch 0 taken 4876230 times.
✗ Branch 1 not taken.
4876230 || b_w == 1
351
1/2
✓ Branch 0 taken 4876230 times.
✗ Branch 1 not taken.
4876230 || b_h == 1
352
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4876230 times.
4876230 || !s->plane[plane_index].fast_mc )
353 3206706 mc_block(&s->plane[plane_index], dst, src, stride, b_w, b_h, dx, dy);
354
2/2
✓ Branch 0 taken 288126 times.
✓ Branch 1 taken 4588104 times.
4876230 else if(b_w==32){
355 int y;
356
2/2
✓ Branch 0 taken 576252 times.
✓ Branch 1 taken 288126 times.
864378 for(y=0; y<b_h; y+=16){
357 576252 s->h264qpel.put_h264_qpel_pixels_tab[0][dy+(dx>>2)](dst + y*stride, src + 3 + (y+3)*stride,stride);
358 576252 s->h264qpel.put_h264_qpel_pixels_tab[0][dy+(dx>>2)](dst + 16 + y*stride, src + 19 + (y+3)*stride,stride);
359 }
360
2/2
✓ Branch 0 taken 4240390 times.
✓ Branch 1 taken 347714 times.
4588104 }else if(b_w==b_h)
361 4240390 s->h264qpel.put_h264_qpel_pixels_tab[tab_index ][dy+(dx>>2)](dst,src + 3 + 3*stride,stride);
362
2/2
✓ Branch 0 taken 203675 times.
✓ Branch 1 taken 144039 times.
347714 else if(b_w==2*b_h){
363 203675 s->h264qpel.put_h264_qpel_pixels_tab[tab_index+1][dy+(dx>>2)](dst ,src + 3 + 3*stride,stride);
364 203675 s->h264qpel.put_h264_qpel_pixels_tab[tab_index+1][dy+(dx>>2)](dst+b_h,src + 3 + b_h + 3*stride,stride);
365 }else{
366 av_assert2(2*b_w==b_h);
367 144039 s->h264qpel.put_h264_qpel_pixels_tab[tab_index ][dy+(dx>>2)](dst ,src + 3 + 3*stride ,stride);
368 144039 s->h264qpel.put_h264_qpel_pixels_tab[tab_index ][dy+(dx>>2)](dst+b_w*stride,src + 3 + 3*stride+b_w*stride,stride);
369 }
370 }
371 9284876 }
372
373 #define mca(dx,dy,b_w)\
374 static void mc_block_hpel ## dx ## dy ## b_w(uint8_t *dst, const uint8_t *src, ptrdiff_t stride, int h){\
375 av_assert2(h==b_w);\
376 mc_block(NULL, dst, src-(HTAPS_MAX/2-1)-(HTAPS_MAX/2-1)*stride, stride, b_w, b_w, dx, dy);\
377 }
378
379 mca( 0, 0,16)
380 5531 mca( 8, 0,16)
381 4969 mca( 0, 8,16)
382 9754 mca( 8, 8,16)
383 mca( 0, 0,8)
384 mca( 8, 0,8)
385 mca( 0, 8,8)
386 mca( 8, 8,8)
387
388 22 static av_cold void snow_static_init(void)
389 {
390
2/2
✓ Branch 0 taken 176 times.
✓ Branch 1 taken 22 times.
198 for (int i = 0; i < MAX_REF_FRAMES; i++)
391
2/2
✓ Branch 0 taken 1408 times.
✓ Branch 1 taken 176 times.
1584 for (int j = 0; j < MAX_REF_FRAMES; j++)
392 1408 ff_scale_mv_ref[i][j] = 256 * (i + 1) / (j + 1);
393 22 }
394
395 31 av_cold int ff_snow_common_init(AVCodecContext *avctx){
396 static AVOnce init_static_once = AV_ONCE_INIT;
397 31 SnowContext *s = avctx->priv_data;
398 int width, height;
399 int i;
400
401 31 s->avctx= avctx;
402 31 s->max_ref_frames=1; //just make sure it's not an invalid value in case of no initial keyframe
403 31 s->spatial_decomposition_count = 1;
404
405 31 ff_videodsp_init(&s->vdsp, 8);
406 31 ff_dwt_init(&s->dwt);
407 31 ff_h264qpel_init(&s->h264qpel, 8);
408
409 #define mcfh(dx,dy)\
410 s->hdsp.put_pixels_tab [0][dy/4+dx/8]=\
411 s->hdsp.put_no_rnd_pixels_tab[0][dy/4+dx/8]=\
412 mc_block_hpel ## dx ## dy ## 16;\
413 s->hdsp.put_pixels_tab [1][dy/4+dx/8]=\
414 s->hdsp.put_no_rnd_pixels_tab[1][dy/4+dx/8]=\
415 mc_block_hpel ## dx ## dy ## 8;
416
417 31 mcfh(0, 0)
418 31 mcfh(8, 0)
419 31 mcfh(0, 8)
420 31 mcfh(8, 8)
421
422 // dec += FFMAX(s->chroma_h_shift, s->chroma_v_shift);
423
424 31 width= s->avctx->width;
425 31 height= s->avctx->height;
426
427
1/2
✓ Branch 1 taken 31 times.
✗ Branch 2 not taken.
31 if (!FF_ALLOCZ_TYPED_ARRAY(s->spatial_idwt_buffer, width * height) ||
428
1/2
✓ Branch 1 taken 31 times.
✗ Branch 2 not taken.
31 !FF_ALLOCZ_TYPED_ARRAY(s->spatial_dwt_buffer, width * height) || //FIXME this does not belong here
429
1/2
✓ Branch 1 taken 31 times.
✗ Branch 2 not taken.
31 !FF_ALLOCZ_TYPED_ARRAY(s->temp_dwt_buffer, width) ||
430
1/2
✓ Branch 1 taken 31 times.
✗ Branch 2 not taken.
31 !FF_ALLOCZ_TYPED_ARRAY(s->temp_idwt_buffer, width) ||
431
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 31 times.
31 !FF_ALLOCZ_TYPED_ARRAY(s->run_buffer, ((width + 1) >> 1) * ((height + 1) >> 1)))
432 return AVERROR(ENOMEM);
433
434
2/2
✓ Branch 0 taken 248 times.
✓ Branch 1 taken 31 times.
279 for(i=0; i<MAX_REF_FRAMES; i++) {
435 248 s->last_picture[i] = av_frame_alloc();
436
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 248 times.
248 if (!s->last_picture[i])
437 return AVERROR(ENOMEM);
438 }
439
440 31 s->mconly_picture = av_frame_alloc();
441 31 s->current_picture = av_frame_alloc();
442
2/4
✓ Branch 0 taken 31 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 31 times.
31 if (!s->mconly_picture || !s->current_picture)
443 return AVERROR(ENOMEM);
444
445 31 ff_thread_once(&init_static_once, snow_static_init);
446
447 31 return 0;
448 }
449
450 971 int ff_snow_common_init_after_header(AVCodecContext *avctx) {
451 971 SnowContext *s = avctx->priv_data;
452 int plane_index, level, orientation;
453
454
2/2
✓ Branch 0 taken 31 times.
✓ Branch 1 taken 940 times.
971 if(!s->scratchbuf) {
455 int emu_buf_size;
456 31 emu_buf_size = FFMAX(s->mconly_picture->linesize[0], 2*avctx->width+256) * (2 * MB_SIZE + HTAPS_MAX - 1);
457
1/2
✓ Branch 1 taken 31 times.
✗ Branch 2 not taken.
31 if (!FF_ALLOCZ_TYPED_ARRAY(s->scratchbuf, FFMAX(s->mconly_picture->linesize[0], 2*avctx->width+256) * 7 * MB_SIZE) ||
458
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 31 times.
31 !FF_ALLOCZ_TYPED_ARRAY(s->emu_edge_buffer, emu_buf_size))
459 return AVERROR(ENOMEM);
460 }
461
462
2/2
✓ Branch 0 taken 2913 times.
✓ Branch 1 taken 971 times.
3884 for(plane_index=0; plane_index < s->nb_planes; plane_index++){
463 2913 int w= s->avctx->width;
464 2913 int h= s->avctx->height;
465
466
2/2
✓ Branch 0 taken 1942 times.
✓ Branch 1 taken 971 times.
2913 if(plane_index){
467 1942 w = AV_CEIL_RSHIFT(w, s->chroma_h_shift);
468 1942 h = AV_CEIL_RSHIFT(h, s->chroma_v_shift);
469 }
470 2913 s->plane[plane_index].width = w;
471 2913 s->plane[plane_index].height= h;
472
473
2/2
✓ Branch 0 taken 14565 times.
✓ Branch 1 taken 2913 times.
17478 for(level=s->spatial_decomposition_count-1; level>=0; level--){
474
2/2
✓ Branch 0 taken 46608 times.
✓ Branch 1 taken 14565 times.
61173 for(orientation=level ? 1 : 0; orientation<4; orientation++){
475 46608 SubBand *b= &s->plane[plane_index].band[level][orientation];
476
477 46608 b->buf= s->spatial_dwt_buffer;
478 46608 b->level= level;
479 46608 b->stride= s->plane[plane_index].width << (s->spatial_decomposition_count - level);
480 46608 b->width = (w + !(orientation&1))>>1;
481 46608 b->height= (h + !(orientation>1))>>1;
482
483 46608 b->stride_line = 1 << (s->spatial_decomposition_count - level);
484 46608 b->buf_x_offset = 0;
485 46608 b->buf_y_offset = 0;
486
487
2/2
✓ Branch 0 taken 29130 times.
✓ Branch 1 taken 17478 times.
46608 if(orientation&1){
488 29130 b->buf += (w+1)>>1;
489 29130 b->buf_x_offset = (w+1)>>1;
490 }
491
2/2
✓ Branch 0 taken 29130 times.
✓ Branch 1 taken 17478 times.
46608 if(orientation>1){
492 29130 b->buf += b->stride>>1;
493 29130 b->buf_y_offset = b->stride_line >> 1;
494 }
495 46608 b->ibuf= s->spatial_idwt_buffer + (b->buf - s->spatial_dwt_buffer);
496
497
2/2
✓ Branch 0 taken 34956 times.
✓ Branch 1 taken 11652 times.
46608 if(level)
498 34956 b->parent= &s->plane[plane_index].band[level-1][orientation];
499 //FIXME avoid this realloc
500 46608 av_freep(&b->x_coeff);
501 46608 b->x_coeff = av_calloc((b->width + 1) * b->height + 1,
502 sizeof(*b->x_coeff));
503
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 46608 times.
46608 if (!b->x_coeff)
504 return AVERROR(ENOMEM);
505 }
506 14565 w= (w+1)>>1;
507 14565 h= (h+1)>>1;
508 }
509 }
510
511 971 return 0;
512 }
513
514 1942 void ff_snow_release_buffer(AVCodecContext *avctx)
515 {
516 1942 SnowContext *s = avctx->priv_data;
517
518
2/2
✓ Branch 0 taken 940 times.
✓ Branch 1 taken 1002 times.
1942 if(s->last_picture[s->max_ref_frames-1]->data[0]){
519 940 av_frame_unref(s->last_picture[s->max_ref_frames-1]);
520 }
521 1942 }
522
523 971 int ff_snow_frames_prepare(SnowContext *s)
524 {
525 AVFrame *tmp;
526
527 971 ff_snow_release_buffer(s->avctx);
528
529 971 tmp= s->last_picture[s->max_ref_frames-1];
530
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 971 times.
971 for (int i = s->max_ref_frames - 1; i > 0; i--)
531 s->last_picture[i] = s->last_picture[i-1];
532 971 s->last_picture[0] = s->current_picture;
533 971 s->current_picture = tmp;
534
535
2/2
✓ Branch 0 taken 103 times.
✓ Branch 1 taken 868 times.
971 if(s->keyframe){
536 103 s->ref_frames= 0;
537 103 s->current_picture->flags |= AV_FRAME_FLAG_KEY;
538 }else{
539 int i;
540
3/4
✓ Branch 0 taken 868 times.
✓ Branch 1 taken 868 times.
✓ Branch 2 taken 868 times.
✗ Branch 3 not taken.
1736 for(i=0; i<s->max_ref_frames && s->last_picture[i]->data[0]; i++)
541
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 868 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
868 if(i && (s->last_picture[i-1]->flags & AV_FRAME_FLAG_KEY))
542 break;
543 868 s->ref_frames= i;
544
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 868 times.
868 if(s->ref_frames==0){
545 av_log(s->avctx,AV_LOG_ERROR, "No reference frames\n");
546 return AVERROR_INVALIDDATA;
547 }
548 868 s->current_picture->flags &= ~AV_FRAME_FLAG_KEY;
549 }
550
551 971 return 0;
552 }
553
554 31 av_cold void ff_snow_common_end(SnowContext *s)
555 {
556 int plane_index, level, orientation, i;
557
558 31 av_freep(&s->spatial_dwt_buffer);
559 31 av_freep(&s->temp_dwt_buffer);
560 31 av_freep(&s->spatial_idwt_buffer);
561 31 av_freep(&s->temp_idwt_buffer);
562 31 av_freep(&s->run_buffer);
563
564 31 av_freep(&s->block);
565 31 av_freep(&s->scratchbuf);
566 31 av_freep(&s->emu_edge_buffer);
567
568
2/2
✓ Branch 0 taken 248 times.
✓ Branch 1 taken 31 times.
279 for(i=0; i<MAX_REF_FRAMES; i++){
569
2/4
✓ Branch 0 taken 248 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 248 times.
248 if(s->last_picture[i] && s->last_picture[i]->data[0]) {
570 av_assert0(s->last_picture[i]->data[0] != s->current_picture->data[0]);
571 }
572 248 av_frame_free(&s->last_picture[i]);
573 }
574
575
2/2
✓ Branch 0 taken 124 times.
✓ Branch 1 taken 31 times.
155 for(plane_index=0; plane_index < MAX_PLANES; plane_index++){
576
2/2
✓ Branch 0 taken 992 times.
✓ Branch 1 taken 124 times.
1116 for(level=MAX_DECOMPOSITIONS-1; level>=0; level--){
577
2/2
✓ Branch 0 taken 3100 times.
✓ Branch 1 taken 992 times.
4092 for(orientation=level ? 1 : 0; orientation<4; orientation++){
578 3100 SubBand *b= &s->plane[plane_index].band[level][orientation];
579
580 3100 av_freep(&b->x_coeff);
581 }
582 }
583 }
584 31 av_frame_free(&s->mconly_picture);
585 31 av_frame_free(&s->current_picture);
586 31 }
587