FFmpeg coverage

Directory: ../../../ffmpeg/
File: src/libavcodec/snow.h
Date: 2025-07-28 20:30:09
Exec Total Coverage
Lines: 150 157 95.5%
Functions: 6 6 100.0%
Branches: 96 100 96.0%
Line Branch Exec Source
1 /*
2 * Copyright (C) 2004 Michael Niedermayer <michaelni@gmx.at>
3 * Copyright (C) 2006 Robert Edele <yartrebo@earthlink.net>
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 #ifndef AVCODEC_SNOW_H
23 #define AVCODEC_SNOW_H
24
25 #include "libavutil/motion_vector.h"
26
27 #include "avcodec.h"
28 #include "hpeldsp.h"
29 #include "snow_dwt.h"
30
31 #include "rangecoder.h"
32 #include "mathops.h"
33
34 #include "h264qpel.h"
35 #include "videodsp.h"
36
37 #define SNOW_MAX_PLANES 4
38
39 #define MID_STATE 128
40
41 #define MAX_PLANES 4
42 #define QSHIFT 5
43 #define QROOT (1<<QSHIFT)
44 #define LOSSLESS_QLOG -128
45 #define FRAC_BITS 4
46 #define MAX_REF_FRAMES 8
47
48 #define LOG2_OBMC_MAX 8
49 #define OBMC_MAX (1<<(LOG2_OBMC_MAX))
50 typedef struct BlockNode{
51 int16_t mx; ///< Motion vector component X, see mv_scale
52 int16_t my; ///< Motion vector component Y, see mv_scale
53 uint8_t ref; ///< Reference frame index
54 uint8_t color[3]; ///< Color for intra
55 uint8_t type; ///< Bitfield of BLOCK_*
56 //#define TYPE_SPLIT 1
57 #define BLOCK_INTRA 1 ///< Intra block, inter otherwise
58 #define BLOCK_OPT 2 ///< Block needs no checks in this round of iterative motion estiation
59 //#define TYPE_NOCOLOR 4
60 uint8_t level; //FIXME merge into type?
61 }BlockNode;
62
63 static const BlockNode null_block= { //FIXME add border maybe
64 .color= {128,128,128},
65 .mx= 0,
66 .my= 0,
67 .ref= 0,
68 .type= 0,
69 .level= 0,
70 };
71
72 #define LOG2_MB_SIZE 4
73 #define MB_SIZE (1<<LOG2_MB_SIZE)
74 #define ENCODER_EXTRA_BITS 4
75 #define HTAPS_MAX 8
76
77 typedef struct x_and_coeff{
78 int16_t x;
79 uint16_t coeff;
80 } x_and_coeff;
81
82 typedef struct SubBand{
83 int level;
84 int stride;
85 int width;
86 int height;
87 int qlog; ///< log(qscale)/log[2^(1/6)]
88 DWTELEM *buf;
89 IDWTELEM *ibuf;
90 int buf_x_offset;
91 int buf_y_offset;
92 int stride_line; ///< Stride measured in lines, not pixels.
93 x_and_coeff * x_coeff;
94 struct SubBand *parent;
95 uint8_t state[/*7*2*/ 7 + 512][32];
96 }SubBand;
97
98 typedef struct Plane{
99 int width;
100 int height;
101 SubBand band[MAX_DECOMPOSITIONS][4];
102
103 int htaps;
104 int8_t hcoeff[HTAPS_MAX/2];
105 int diag_mc;
106 int fast_mc;
107
108 int last_htaps;
109 int8_t last_hcoeff[HTAPS_MAX/2];
110 int last_diag_mc;
111 }Plane;
112
113 typedef struct SnowContext{
114 AVClass *class;
115 AVCodecContext *avctx;
116 RangeCoder c;
117 HpelDSPContext hdsp;
118 VideoDSPContext vdsp;
119 H264QpelContext h264qpel;
120 SnowDWTContext dwt;
121 AVFrame *input_picture; ///< new_picture with the internal linesizes
122 AVFrame *current_picture;
123 AVFrame *last_picture[MAX_REF_FRAMES];
124 AVFrame *mconly_picture;
125 // uint8_t q_context[16];
126 uint8_t header_state[32];
127 uint8_t block_state[128 + 32*128];
128 int keyframe;
129 int always_reset;
130 int version;
131 int spatial_decomposition_type;
132 int last_spatial_decomposition_type;
133 int temporal_decomposition_type;
134 int spatial_decomposition_count;
135 int last_spatial_decomposition_count;
136 int temporal_decomposition_count;
137 int max_ref_frames;
138 int ref_frames;
139 int16_t (*ref_mvs[MAX_REF_FRAMES])[2];
140 uint32_t *ref_scores[MAX_REF_FRAMES];
141 DWTELEM *spatial_dwt_buffer;
142 DWTELEM *temp_dwt_buffer;
143 IDWTELEM *spatial_idwt_buffer;
144 IDWTELEM *temp_idwt_buffer;
145 int *run_buffer;
146 int colorspace_type;
147 int chroma_h_shift;
148 int chroma_v_shift;
149 int spatial_scalability;
150 int qlog;
151 int last_qlog;
152 int mv_scale;
153 int last_mv_scale;
154 int qbias;
155 int last_qbias;
156 #define QBIAS_SHIFT 3
157 int b_width;
158 int b_height;
159 int block_max_depth;
160 int last_block_max_depth;
161 int nb_planes;
162 Plane plane[MAX_PLANES];
163 BlockNode *block;
164 slice_buffer sb;
165
166 uint8_t *scratchbuf;
167 uint8_t *emu_edge_buffer;
168
169 AVMotionVector *avmv;
170 unsigned avmv_size;
171 int avmv_index;
172 }SnowContext;
173
174 /* Tables */
175 extern const uint8_t * const ff_obmc_tab[4];
176 extern const uint8_t ff_qexp[QROOT];
177 extern int ff_scale_mv_ref[MAX_REF_FRAMES][MAX_REF_FRAMES];
178
179 /* common code */
180
181 int ff_snow_common_init(AVCodecContext *avctx);
182 int ff_snow_common_init_after_header(AVCodecContext *avctx);
183 void ff_snow_common_end(SnowContext *s);
184 void ff_snow_reset_contexts(SnowContext *s);
185 int ff_snow_alloc_blocks(SnowContext *s);
186 int ff_snow_frames_prepare(SnowContext *s);
187 void ff_snow_pred_block(SnowContext *s, uint8_t *dst, uint8_t *tmp, ptrdiff_t stride,
188 int sx, int sy, int b_w, int b_h, const BlockNode *block,
189 int plane_index, int w, int h);
190 /* common inline functions */
191 //XXX doublecheck all of them should stay inlined
192
193 1913171 static inline void pred_mv(SnowContext *s, int *mx, int *my, int ref,
194 const BlockNode *left, const BlockNode *top, const BlockNode *tr){
195
1/2
✓ Branch 0 taken 1913171 times.
✗ Branch 1 not taken.
 1913171 if(s->ref_frames == 1){
196 1913171 *mx = mid_pred(left->mx, top->mx, tr->mx);
197 1913171 *my = mid_pred(left->my, top->my, tr->my);
198 }else{
199 ✗ const int *scale = ff_scale_mv_ref[ref];
200 ✗ *mx = mid_pred((left->mx * scale[left->ref] + 128) >>8,
201 ✗ (top ->mx * scale[top ->ref] + 128) >>8,
202 ✗ (tr ->mx * scale[tr ->ref] + 128) >>8);
203 ✗ *my = mid_pred((left->my * scale[left->ref] + 128) >>8,
204 ✗ (top ->my * scale[top ->ref] + 128) >>8,
205 ✗ (tr ->my * scale[tr ->ref] + 128) >>8);
206 }
207 1913171 }
208
209 16265809 static av_always_inline int same_block(BlockNode *a, BlockNode *b){
210
4/4
✓ Branch 0 taken 1952427 times.
✓ Branch 1 taken 14313382 times.
✓ Branch 2 taken 1096516 times.
✓ Branch 3 taken 855911 times.
 16265809 if((a->type&BLOCK_INTRA) && (b->type&BLOCK_INTRA)){
211 1096516 return !((a->color[0] - b->color[0]) | (a->color[1] - b->color[1]) | (a->color[2] - b->color[2]));
212 }else{
213 15169293 return !((a->mx - b->mx) | (a->my - b->my) | (a->ref - b->ref) | ((a->type ^ b->type)&BLOCK_INTRA));
214 }
215 }
216
217 //FIXME name cleanup (b_w, block_w, b_width stuff)
218 //XXX should we really inline it?
219 3102231 static av_always_inline void add_yblock(SnowContext *s, int sliced, slice_buffer *sb, IDWTELEM *dst, uint8_t *dst8, const uint8_t *obmc, int src_x, int src_y, int b_w, int b_h, int w, int h, int dst_stride, int src_stride, int obmc_stride, int b_x, int b_y, int add, int offset_dst, int plane_index){
220 3102231 const int b_width = s->b_width << s->block_max_depth;
221 3102231 const int b_height= s->b_height << s->block_max_depth;
222 3102231 const int b_stride= b_width;
223 3102231 BlockNode *lt= &s->block[b_x + b_y*b_stride];
224 3102231 BlockNode *rt= lt+1;
225 3102231 BlockNode *lb= lt+b_stride;
226 3102231 BlockNode *rb= lb+1;
227 uint8_t *block[4];
228 // When src_stride is large enough, it is possible to interleave the blocks.
229 // Otherwise the blocks are written sequentially in the tmp buffer.
230
2/2
✓ Branch 0 taken 24300 times.
✓ Branch 1 taken 3077931 times.
 3102231 int tmp_step= src_stride >= 7*MB_SIZE ? MB_SIZE : MB_SIZE*src_stride;
231 3102231 uint8_t *tmp = s->scratchbuf;
232 uint8_t *ptmp;
233 int x,y;
234
235
2/2
✓ Branch 0 taken 80451 times.
✓ Branch 1 taken 3021780 times.
 3102231 if(b_x<0){
236 80451 lt= rt;
237 80451 lb= rb;
238
2/2
✓ Branch 0 taken 77583 times.
✓ Branch 1 taken 2944197 times.
 3021780 }else if(b_x + 1 >= b_width){
239 77583 rt= lt;
240 77583 rb= lb;
241 }
242
2/2
✓ Branch 0 taken 118527 times.
✓ Branch 1 taken 2983704 times.
 3102231 if(b_y<0){
243 118527 lt= lb;
244 118527 rt= rb;
245
2/2
✓ Branch 0 taken 109827 times.
✓ Branch 1 taken 2873877 times.
 2983704 }else if(b_y + 1 >= b_height){
246 109827 lb= lt;
247 109827 rb= rt;
248 }
249
250
2/2
✓ Branch 0 taken 80451 times.
✓ Branch 1 taken 3021780 times.
 3102231 if(src_x<0){ //FIXME merge with prev & always round internal width up to *16
251 80451 obmc -= src_x;
252 80451 b_w += src_x;
253
4/4
✓ Branch 0 taken 61416 times.
✓ Branch 1 taken 19035 times.
✓ Branch 2 taken 13446 times.
✓ Branch 3 taken 47970 times.
 80451 if(!sliced && !offset_dst)
254 13446 dst -= src_x;
255 80451 src_x=0;
256 }
257
2/2
✓ Branch 0 taken 77583 times.
✓ Branch 1 taken 3024648 times.
 3102231 if(src_x + b_w > w){
258 77583 b_w = w - src_x;
259 }
260
2/2
✓ Branch 0 taken 118527 times.
✓ Branch 1 taken 2983704 times.
 3102231 if(src_y<0){
261 118527 obmc -= src_y*obmc_stride;
262 118527 b_h += src_y;
263
4/4
✓ Branch 0 taken 93012 times.
✓ Branch 1 taken 25515 times.
✓ Branch 2 taken 29652 times.
✓ Branch 3 taken 63360 times.
 118527 if(!sliced && !offset_dst)
264 29652 dst -= src_y*dst_stride;
265 118527 src_y=0;
266 }
267
2/2
✓ Branch 0 taken 109827 times.
✓ Branch 1 taken 2992404 times.
 3102231 if(src_y + b_h> h){
268 109827 b_h = h - src_y;
269 }
270
271
2/4
✓ Branch 0 taken 3102231 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 3102231 times.
 3102231 if(b_w<=0 || b_h<=0) return;
272
273
4/4
✓ Branch 0 taken 2391456 times.
✓ Branch 1 taken 710775 times.
✓ Branch 2 taken 2172600 times.
✓ Branch 3 taken 218856 times.
 3102231 if(!sliced && offset_dst)
274 2172600 dst += src_x + src_y*dst_stride;
275
4/4
✓ Branch 0 taken 2391456 times.
✓ Branch 1 taken 710775 times.
✓ Branch 2 taken 1461825 times.
✓ Branch 3 taken 929631 times.
 3102231 if (sliced || add)
276 2172600 dst8+= src_x + src_y*src_stride;
277 // src += src_x + src_y*src_stride;
278
279 3102231 ptmp= tmp + 3*tmp_step;
280 3102231 block[0]= ptmp;
281 3102231 ptmp+=tmp_step;
282 3102231 ff_snow_pred_block(s, block[0], tmp, src_stride, src_x, src_y, b_w, b_h, lt, plane_index, w, h);
283
284
2/2
✓ Branch 1 taken 846930 times.
✓ Branch 2 taken 2255301 times.
 3102231 if(same_block(lt, rt)){
285 846930 block[1]= block[0];
286 }else{
287 2255301 block[1]= ptmp;
288 2255301 ptmp+=tmp_step;
289 2255301 ff_snow_pred_block(s, block[1], tmp, src_stride, src_x, src_y, b_w, b_h, rt, plane_index, w, h);
290 }
291
292
2/2
✓ Branch 1 taken 873846 times.
✓ Branch 2 taken 2228385 times.
 3102231 if(same_block(lt, lb)){
293 873846 block[2]= block[0];
294
2/2
✓ Branch 1 taken 208689 times.
✓ Branch 2 taken 2019696 times.
 2228385 }else if(same_block(rt, lb)){
295 208689 block[2]= block[1];
296 }else{
297 2019696 block[2]= ptmp;
298 2019696 ptmp+=tmp_step;
299 2019696 ff_snow_pred_block(s, block[2], tmp, src_stride, src_x, src_y, b_w, b_h, lb, plane_index, w, h);
300 }
301
302
2/2
✓ Branch 1 taken 482262 times.
✓ Branch 2 taken 2619969 times.
 3102231 if(same_block(lt, rb) ){
303 482262 block[3]= block[0];
304
2/2
✓ Branch 1 taken 527445 times.
✓ Branch 2 taken 2092524 times.
 2619969 }else if(same_block(rt, rb)){
305 527445 block[3]= block[1];
306
2/2
✓ Branch 1 taken 397221 times.
✓ Branch 2 taken 1695303 times.
 2092524 }else if(same_block(lb, rb)){
307 397221 block[3]= block[2];
308 }else{
309 1695303 block[3]= ptmp;
310 1695303 ff_snow_pred_block(s, block[3], tmp, src_stride, src_x, src_y, b_w, b_h, rb, plane_index, w, h);
311 }
312
2/2
✓ Branch 0 taken 710775 times.
✓ Branch 1 taken 2391456 times.
 3102231 if(sliced){
313 710775 s->dwt.inner_add_yblock(obmc, obmc_stride, block, b_w, b_h, src_x,src_y, src_stride, sb, add, dst8);
314 }else{
315
2/2
✓ Branch 0 taken 14419776 times.
✓ Branch 1 taken 2391456 times.
 16811232 for(y=0; y<b_h; y++){
316 //FIXME ugly misuse of obmc_stride
317 14419776 const uint8_t *obmc1= obmc + y*obmc_stride;
318 14419776 const uint8_t *obmc2= obmc1+ (obmc_stride>>1);
319 14419776 const uint8_t *obmc3= obmc1+ obmc_stride*(obmc_stride>>1);
320 14419776 const uint8_t *obmc4= obmc3+ (obmc_stride>>1);
321
2/2
✓ Branch 0 taken 108428832 times.
✓ Branch 1 taken 14419776 times.
 122848608 for(x=0; x<b_w; x++){
322 108428832 int v= obmc1[x] * block[3][x + y*src_stride]
323 108428832 +obmc2[x] * block[2][x + y*src_stride]
324 108428832 +obmc3[x] * block[1][x + y*src_stride]
325 108428832 +obmc4[x] * block[0][x + y*src_stride];
326
327 108428832 v <<= 8 - LOG2_OBMC_MAX;
328 if(FRAC_BITS != 8){
329 108428832 v >>= 8 - FRAC_BITS;
330 }
331
2/2
✓ Branch 0 taken 61171200 times.
✓ Branch 1 taken 47257632 times.
 108428832 if(add){
332 61171200 v += dst[x + y*dst_stride];
333 61171200 v = (v + (1<<(FRAC_BITS-1))) >> FRAC_BITS;
334
2/2
✓ Branch 0 taken 1126 times.
✓ Branch 1 taken 61170074 times.
 61171200 if(v&(~255)) v= ~(v>>31);
335 61171200 dst8[x + y*src_stride] = v;
336 }else{
337 47257632 dst[x + y*dst_stride] -= v;
338 }
339 }
340 }
341 }
342 }
343
344 52200 static av_always_inline void predict_slice(SnowContext *s, IDWTELEM *buf, int plane_index, int add, int mb_y){
345 52200 Plane *p= &s->plane[plane_index];
346 52200 const int mb_w= s->b_width << s->block_max_depth;
347 52200 const int mb_h= s->b_height << s->block_max_depth;
348 int x, y, mb_x;
349 52200 int block_size = MB_SIZE >> s->block_max_depth;
350
2/2
✓ Branch 0 taken 34800 times.
✓ Branch 1 taken 17400 times.
 52200 int block_w = plane_index ? block_size>>s->chroma_h_shift : block_size;
351
2/2
✓ Branch 0 taken 34800 times.
✓ Branch 1 taken 17400 times.
 52200 int block_h = plane_index ? block_size>>s->chroma_v_shift : block_size;
352
2/2
✓ Branch 0 taken 34800 times.
✓ Branch 1 taken 17400 times.
 52200 const uint8_t *obmc = plane_index ? ff_obmc_tab[s->block_max_depth+s->chroma_h_shift] : ff_obmc_tab[s->block_max_depth];
353
2/2
✓ Branch 0 taken 34800 times.
✓ Branch 1 taken 17400 times.
 52200 const int obmc_stride= plane_index ? (2*block_size)>>s->chroma_h_shift : 2*block_size;
354 52200 int ref_stride= s->current_picture->linesize[plane_index];
355 52200 uint8_t *dst8= s->current_picture->data[plane_index];
356 52200 int w= p->width;
357 52200 int h= p->height;
358 av_assert2(s->chroma_h_shift == s->chroma_v_shift); // obmc params assume squares
359
3/4
✓ Branch 0 taken 47970 times.
✓ Branch 1 taken 4230 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 47970 times.
 52200 if(s->keyframe || (s->avctx->debug&512)){
360
2/2
✓ Branch 0 taken 270 times.
✓ Branch 1 taken 3960 times.
 4230 if(mb_y==mb_h)
361 270 return;
362
363
2/2
✓ Branch 0 taken 1980 times.
✓ Branch 1 taken 1980 times.
 3960 if(add){
364
2/2
✓ Branch 0 taken 12480 times.
✓ Branch 1 taken 1980 times.
 14460 for(y=block_h*mb_y; y<FFMIN(h,block_h*(mb_y+1)); y++){
365
2/2
✓ Branch 0 taken 2649600 times.
✓ Branch 1 taken 12480 times.
 2662080 for(x=0; x<w; x++){
366 2649600 int v= buf[x + y*w] + (128<<FRAC_BITS) + (1<<(FRAC_BITS-1));
367 2649600 v >>= FRAC_BITS;
368
2/2
✓ Branch 0 taken 104 times.
✓ Branch 1 taken 2649496 times.
 2649600 if(v&(~255)) v= ~(v>>31);
369 2649600 dst8[x + y*ref_stride]= v;
370 }
371 }
372 }else{
373
2/2
✓ Branch 0 taken 12480 times.
✓ Branch 1 taken 1980 times.
 14460 for(y=block_h*mb_y; y<FFMIN(h,block_h*(mb_y+1)); y++){
374
2/2
✓ Branch 0 taken 2649600 times.
✓ Branch 1 taken 12480 times.
 2662080 for(x=0; x<w; x++){
375 2649600 buf[x + y*w]-= 128<<FRAC_BITS;
376 }
377 }
378 }
379
380 3960 return;
381 }
382
383
2/2
✓ Branch 0 taken 2172600 times.
✓ Branch 1 taken 47970 times.
 2220570 for(mb_x=0; mb_x<=mb_w; mb_x++){
384 2172600 add_yblock(s, 0, NULL, buf, dst8, obmc,
385 2172600 block_w*mb_x - block_w/2,
386 2172600 block_h*mb_y - block_h/2,
387 block_w, block_h,
388 w, h,
389 w, ref_stride, obmc_stride,
390 mb_x - 1, mb_y - 1,
391 add, 1, plane_index);
392 }
393 }
394
395 2880 static av_always_inline void predict_plane(SnowContext *s, IDWTELEM *buf, int plane_index, int add){
396 2880 const int mb_h= s->b_height << s->block_max_depth;
397 int mb_y;
398
2/2
✓ Branch 0 taken 52200 times.
✓ Branch 1 taken 2880 times.
 55080 for(mb_y=0; mb_y<=mb_h; mb_y++)
399 52200 predict_slice(s, buf, plane_index, add, mb_y);
400 2880 }
401
402 702004 static inline void set_blocks(SnowContext *s, int level, int x, int y, int l, int cb, int cr, int mx, int my, int ref, int type){
403 702004 const int w= s->b_width << s->block_max_depth;
404 702004 const int rem_depth= s->block_max_depth - level;
405 702004 const int index= (x + y*w) << rem_depth;
406 702004 const int block_w= 1<<rem_depth;
407 702004 const int block_h= 1<<rem_depth; //FIXME "w!=h"
408 BlockNode block;
409 int i,j;
410
411 702004 block.color[0]= l;
412 702004 block.color[1]= cb;
413 702004 block.color[2]= cr;
414 702004 block.mx= mx;
415 702004 block.my= my;
416 702004 block.ref= ref;
417 702004 block.type= type;
418 702004 block.level= level;
419
420
2/2
✓ Branch 0 taken 737432 times.
✓ Branch 1 taken 702004 times.
 1439436 for(j=0; j<block_h; j++){
421
2/2
✓ Branch 0 taken 808288 times.
✓ Branch 1 taken 737432 times.
 1545720 for(i=0; i<block_w; i++){
422 808288 s->block[index + i + j*w]= block;
423 }
424 }
425 702004 }
426
427 extern const int8_t ff_quant3bA[256];
428
429 #define QEXPSHIFT (7-FRAC_BITS+8) //FIXME try to change this to 0
430
431 #endif /* AVCODEC_SNOW_H */
432