| 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/emms.h" | ||
| 22 | #include "libavutil/intmath.h" | ||
| 23 | #include "libavutil/log.h" | ||
| 24 | #include "libavutil/mem.h" | ||
| 25 | #include "avcodec.h" | ||
| 26 | #include "codec_internal.h" | ||
| 27 | #include "decode.h" | ||
| 28 | #include "snow_dwt.h" | ||
| 29 | #include "snow.h" | ||
| 30 | |||
| 31 | #include "rangecoder.h" | ||
| 32 | #include "mathops.h" | ||
| 33 | |||
| 34 | 398674 | static inline int get_symbol(RangeCoder *c, uint8_t *state, int is_signed) | |
| 35 | { | ||
| 36 |
2/2✓ Branch 1 taken 181336 times.
✓ Branch 2 taken 217338 times.
|
398674 | if (get_rac(c, state + 0)) |
| 37 | 181336 | return 0; | |
| 38 | else { | ||
| 39 | int e; | ||
| 40 | unsigned a; | ||
| 41 | 217338 | e = 0; | |
| 42 |
3/4✓ Branch 0 taken 396567 times.
✗ Branch 1 not taken.
✓ Branch 3 taken 179229 times.
✓ Branch 4 taken 217338 times.
|
396567 | while (get_rac(c, state + 1 + FFMIN(e, 9))) { //1..10 |
| 43 | 179229 | e++; | |
| 44 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 179229 times.
|
179229 | if (e > 31) |
| 45 | ✗ | return AVERROR_INVALIDDATA; | |
| 46 | } | ||
| 47 | |||
| 48 | 217338 | a = 1; | |
| 49 |
2/2✓ Branch 0 taken 179229 times.
✓ Branch 1 taken 217338 times.
|
396567 | for (int i = e - 1; i >= 0; i--) |
| 50 |
1/2✓ Branch 0 taken 179229 times.
✗ Branch 1 not taken.
|
179229 | a += a + get_rac(c, state + 22 + FFMIN(i, 9)); //22..31 |
| 51 | |||
| 52 |
5/6✓ Branch 0 taken 216900 times.
✓ Branch 1 taken 438 times.
✓ Branch 2 taken 216900 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 84748 times.
✓ Branch 6 taken 132152 times.
|
217338 | e = -(is_signed && get_rac(c, state + 11 + FFMIN(e, 10))); //11..21 |
| 53 | 217338 | return (a ^ e) - e; | |
| 54 | } | ||
| 55 | } | ||
| 56 | |||
| 57 | 16813713 | static inline int get_symbol2(RangeCoder *c, uint8_t *state, int log2) | |
| 58 | { | ||
| 59 |
2/2✓ Branch 0 taken 7066249 times.
✓ Branch 1 taken 9747464 times.
|
16813713 | int r = log2 >= 0 ? 1 << log2 : 1; |
| 60 | 16813713 | int v = 0; | |
| 61 | |||
| 62 | av_assert2(log2 >= -4); | ||
| 63 | |||
| 64 |
3/4✓ Branch 0 taken 29173192 times.
✗ Branch 1 not taken.
✓ Branch 3 taken 12359479 times.
✓ Branch 4 taken 16813713 times.
|
29173192 | while (log2 < 28 && get_rac(c, state + 4 + log2)) { |
| 65 | 12359479 | v += r; | |
| 66 | 12359479 | log2++; | |
| 67 |
2/2✓ Branch 0 taken 9002801 times.
✓ Branch 1 taken 3356678 times.
|
12359479 | if (log2 > 0) r += r; |
| 68 | } | ||
| 69 | |||
| 70 |
2/2✓ Branch 0 taken 17512479 times.
✓ Branch 1 taken 16813713 times.
|
34326192 | for (int i = log2 - 1; i >= 0; i--) |
| 71 | 17512479 | v += get_rac(c, state + 31 - i) << i; | |
| 72 | |||
| 73 | 16813713 | return v; | |
| 74 | } | ||
| 75 | |||
| 76 | 22128 | static void unpack_coeffs(SnowContext *s, SubBand *b, SubBand * parent, int orientation) | |
| 77 | { | ||
| 78 | 22128 | const int w = b->width; | |
| 79 | 22128 | const int h = b->height; | |
| 80 | |||
| 81 | int run, runs; | ||
| 82 | 22128 | x_and_coeff *xc = b->x_coeff; | |
| 83 | 22128 | x_and_coeff *prev_xc = NULL; | |
| 84 | 22128 | x_and_coeff *prev2_xc = xc; | |
| 85 |
2/2✓ Branch 0 taken 16596 times.
✓ Branch 1 taken 5532 times.
|
22128 | x_and_coeff *parent_xc = parent ? parent->x_coeff : NULL; |
| 86 | 22128 | x_and_coeff *prev_parent_xc = parent_xc; | |
| 87 | |||
| 88 | 22128 | runs = get_symbol2(&s->c, b->state[30], 0); | |
| 89 |
2/2✓ Branch 0 taken 17295 times.
✓ Branch 1 taken 4833 times.
|
22128 | if (runs-- > 0) run = get_symbol2(&s->c, b->state[1], 3); |
| 90 | 4833 | else run = INT_MAX; | |
| 91 | |||
| 92 |
2/2✓ Branch 0 taken 376000 times.
✓ Branch 1 taken 22128 times.
|
398128 | for (int y = 0; y < h; y++) { |
| 93 | 376000 | int v = 0; | |
| 94 | 376000 | int lt = 0, t = 0, rt = 0; | |
| 95 | |||
| 96 |
4/4✓ Branch 0 taken 353872 times.
✓ Branch 1 taken 22128 times.
✓ Branch 2 taken 214455 times.
✓ Branch 3 taken 139417 times.
|
376000 | if (y && prev_xc->x == 0) |
| 97 | 214455 | rt = prev_xc->coeff; | |
| 98 | |||
| 99 |
2/2✓ Branch 0 taken 25206764 times.
✓ Branch 1 taken 376000 times.
|
25582764 | for (int x = 0; x < w; x++) { |
| 100 | 25206764 | int p = 0; | |
| 101 | 25206764 | const int l = v; | |
| 102 | |||
| 103 | 25206764 | lt= t; t= rt; | |
| 104 | |||
| 105 |
2/2✓ Branch 0 taken 24754930 times.
✓ Branch 1 taken 451834 times.
|
25206764 | if (y) { |
| 106 |
2/2✓ Branch 0 taken 16207985 times.
✓ Branch 1 taken 8546945 times.
|
24754930 | if (prev_xc->x <= x) |
| 107 | 16207985 | prev_xc++; | |
| 108 |
2/2✓ Branch 0 taken 15993530 times.
✓ Branch 1 taken 8761400 times.
|
24754930 | if (prev_xc->x == x + 1) |
| 109 | 15993530 | rt = prev_xc->coeff; | |
| 110 | else | ||
| 111 | 8761400 | rt = 0; | |
| 112 | } | ||
| 113 |
2/2✓ Branch 0 taken 25102735 times.
✓ Branch 1 taken 104029 times.
|
25206764 | if (parent_xc) { |
| 114 |
2/2✓ Branch 0 taken 8437658 times.
✓ Branch 1 taken 16665077 times.
|
25102735 | if (x>>1 > parent_xc->x) |
| 115 | 8437658 | parent_xc++; | |
| 116 |
2/2✓ Branch 0 taken 17312584 times.
✓ Branch 1 taken 7790151 times.
|
25102735 | if (x>>1 == parent_xc->x) |
| 117 | 17312584 | p = parent_xc->coeff; | |
| 118 | } | ||
| 119 |
2/2✓ Branch 0 taken 24346735 times.
✓ Branch 1 taken 860029 times.
|
25206764 | if (/*ll|*/l|lt|t|rt|p) { |
| 120 | 24346735 | int context = av_log2(/*FFABS(ll) + */3*(l>>1) + (lt>>1) + (t&~1) + (rt>>1) + (p>>1)); | |
| 121 | |||
| 122 | 24346735 | v = get_rac(&s->c, &b->state[0][context]); | |
| 123 |
2/2✓ Branch 0 taken 16203821 times.
✓ Branch 1 taken 8142914 times.
|
24346735 | if (v) { |
| 124 | 16203821 | v = 2*(get_symbol2(&s->c, b->state[context + 2], context-4) + 1); | |
| 125 | 16203821 | v += get_rac(&s->c, &b->state[0][16 + 1 + 3 + ff_quant3bA[l&0xFF] + 3 * ff_quant3bA[t&0xFF]]); | |
| 126 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 16203821 times.
|
16203821 | if ((uint16_t)v != v) { |
| 127 | ✗ | av_log(s->avctx, AV_LOG_ERROR, "Coefficient damaged\n"); | |
| 128 | ✗ | v = 1; | |
| 129 | } | ||
| 130 | 16203821 | xc->x = x; | |
| 131 | 16203821 | (xc++)->coeff = v; | |
| 132 | } | ||
| 133 | } else { | ||
| 134 |
2/2✓ Branch 0 taken 293882 times.
✓ Branch 1 taken 566147 times.
|
860029 | if (!run) { |
| 135 |
2/2✓ Branch 0 taken 276587 times.
✓ Branch 1 taken 17295 times.
|
293882 | if (runs-- > 0) run = get_symbol2(&s->c, b->state[1], 3); |
| 136 | 17295 | else run = INT_MAX; | |
| 137 | 293882 | v = 2 * (get_symbol2(&s->c, b->state[0 + 2], 0-4) + 1); | |
| 138 | 293882 | v += get_rac(&s->c, &b->state[0][16 + 1 + 3]); | |
| 139 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 293882 times.
|
293882 | if ((uint16_t)v != v) { |
| 140 | ✗ | av_log(s->avctx, AV_LOG_ERROR, "Coefficient damaged\n"); | |
| 141 | ✗ | v = 1; | |
| 142 | } | ||
| 143 | |||
| 144 | 293882 | xc->x = x; | |
| 145 | 293882 | (xc++)->coeff = v; | |
| 146 | } else { | ||
| 147 | int max_run; | ||
| 148 | 566147 | run--; | |
| 149 | 566147 | v = 0; | |
| 150 | av_assert2(run >= 0); | ||
| 151 |
4/4✓ Branch 0 taken 534471 times.
✓ Branch 1 taken 31676 times.
✓ Branch 2 taken 314287 times.
✓ Branch 3 taken 220184 times.
|
566147 | if (y) max_run = FFMIN(run, prev_xc->x - x - 2); |
| 152 |
2/2✓ Branch 0 taken 8112 times.
✓ Branch 1 taken 23564 times.
|
31676 | else max_run = FFMIN(run, w-x-1); |
| 153 |
2/2✓ Branch 0 taken 562248 times.
✓ Branch 1 taken 3899 times.
|
566147 | if (parent_xc) |
| 154 |
2/2✓ Branch 0 taken 118292 times.
✓ Branch 1 taken 443956 times.
|
562248 | max_run = FFMIN(max_run, 2*parent_xc->x - x - 1); |
| 155 | av_assert2(max_run >= 0 && max_run <= run); | ||
| 156 | |||
| 157 | 566147 | x += max_run; | |
| 158 | 566147 | run -= max_run; | |
| 159 | } | ||
| 160 | } | ||
| 161 | } | ||
| 162 | 376000 | (xc++)->x = w+1; //end marker | |
| 163 | 376000 | prev_xc = prev2_xc; | |
| 164 | 376000 | prev2_xc = xc; | |
| 165 | |||
| 166 |
2/2✓ Branch 0 taken 360000 times.
✓ Branch 1 taken 16000 times.
|
376000 | if (parent_xc) { |
| 167 |
2/2✓ Branch 0 taken 179538 times.
✓ Branch 1 taken 180462 times.
|
360000 | if (y & 1) { |
| 168 |
2/2✓ Branch 0 taken 109725 times.
✓ Branch 1 taken 179538 times.
|
289263 | while (parent_xc->x != parent->width+1) |
| 169 | 109725 | parent_xc++; | |
| 170 | 179538 | parent_xc++; | |
| 171 | 179538 | prev_parent_xc= parent_xc; | |
| 172 | } else { | ||
| 173 | 180462 | parent_xc= prev_parent_xc; | |
| 174 | } | ||
| 175 | } | ||
| 176 | } | ||
| 177 | |||
| 178 | 22128 | (xc++)->x = w + 1; //end marker | |
| 179 | 22128 | } | |
| 180 | |||
| 181 | 21699 | static av_always_inline void predict_slice_buffered(SnowContext *s, slice_buffer * sb, IDWTELEM * old_buffer, int plane_index, int add, int mb_y){ | |
| 182 | 21699 | Plane *p= &s->plane[plane_index]; | |
| 183 | 21699 | const int mb_w= s->b_width << s->block_max_depth; | |
| 184 | 21699 | const int mb_h= s->b_height << s->block_max_depth; | |
| 185 | int x, y, mb_x; | ||
| 186 | 21699 | int block_size = MB_SIZE >> s->block_max_depth; | |
| 187 |
2/2✓ Branch 0 taken 14466 times.
✓ Branch 1 taken 7233 times.
|
21699 | int block_w = plane_index ? block_size>>s->chroma_h_shift : block_size; |
| 188 |
2/2✓ Branch 0 taken 14466 times.
✓ Branch 1 taken 7233 times.
|
21699 | int block_h = plane_index ? block_size>>s->chroma_v_shift : block_size; |
| 189 |
2/2✓ Branch 0 taken 14466 times.
✓ Branch 1 taken 7233 times.
|
21699 | const uint8_t *obmc = plane_index ? ff_obmc_tab[s->block_max_depth+s->chroma_h_shift] : ff_obmc_tab[s->block_max_depth]; |
| 190 |
2/2✓ Branch 0 taken 14466 times.
✓ Branch 1 taken 7233 times.
|
21699 | int obmc_stride= plane_index ? (2*block_size)>>s->chroma_h_shift : 2*block_size; |
| 191 | 21699 | int ref_stride= s->current_picture->linesize[plane_index]; | |
| 192 | 21699 | uint8_t *dst8= s->current_picture->data[plane_index]; | |
| 193 | 21699 | int w= p->width; | |
| 194 | 21699 | int h= p->height; | |
| 195 | |||
| 196 |
3/4✓ Branch 0 taken 19035 times.
✓ Branch 1 taken 2664 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 19035 times.
|
21699 | if(s->keyframe || (s->avctx->debug&512)){ |
| 197 |
2/2✓ Branch 0 taken 168 times.
✓ Branch 1 taken 2496 times.
|
2664 | if(mb_y==mb_h) |
| 198 | 168 | return; | |
| 199 | |||
| 200 |
1/2✓ Branch 0 taken 2496 times.
✗ Branch 1 not taken.
|
2496 | if(add){ |
| 201 |
2/2✓ Branch 0 taken 15680 times.
✓ Branch 1 taken 2496 times.
|
18176 | for(y=block_h*mb_y; y<FFMIN(h,block_h*(mb_y+1)); y++){ |
| 202 | // DWTELEM * line = slice_buffer_get_line(sb, y); | ||
| 203 | 15680 | IDWTELEM * line = sb->line[y]; | |
| 204 |
2/2✓ Branch 0 taken 3343872 times.
✓ Branch 1 taken 15680 times.
|
3359552 | for(x=0; x<w; x++){ |
| 205 | // int v= buf[x + y*w] + (128<<FRAC_BITS) + (1<<(FRAC_BITS-1)); | ||
| 206 | 3343872 | int v= line[x] + (128<<FRAC_BITS) + (1<<(FRAC_BITS-1)); | |
| 207 | 3343872 | v >>= FRAC_BITS; | |
| 208 |
2/2✓ Branch 0 taken 126 times.
✓ Branch 1 taken 3343746 times.
|
3343872 | if(v&(~255)) v= ~(v>>31); |
| 209 | 3343872 | dst8[x + y*ref_stride]= v; | |
| 210 | } | ||
| 211 | } | ||
| 212 | }else{ | ||
| 213 | ✗ | for(y=block_h*mb_y; y<FFMIN(h,block_h*(mb_y+1)); y++){ | |
| 214 | // DWTELEM * line = slice_buffer_get_line(sb, y); | ||
| 215 | ✗ | IDWTELEM * line = sb->line[y]; | |
| 216 | ✗ | for(x=0; x<w; x++){ | |
| 217 | ✗ | line[x] -= 128 << FRAC_BITS; | |
| 218 | // buf[x + y*w]-= 128<<FRAC_BITS; | ||
| 219 | } | ||
| 220 | } | ||
| 221 | } | ||
| 222 | |||
| 223 | 2496 | return; | |
| 224 | } | ||
| 225 | |||
| 226 |
2/2✓ Branch 0 taken 710775 times.
✓ Branch 1 taken 19035 times.
|
729810 | for(mb_x=0; mb_x<=mb_w; mb_x++){ |
| 227 | 710775 | add_yblock(s, 1, sb, old_buffer, dst8, obmc, | |
| 228 | 710775 | block_w*mb_x - block_w/2, | |
| 229 | 710775 | block_h*mb_y - block_h/2, | |
| 230 | block_w, block_h, | ||
| 231 | w, h, | ||
| 232 | w, ref_stride, obmc_stride, | ||
| 233 | mb_x - 1, mb_y - 1, | ||
| 234 | add, 0, plane_index); | ||
| 235 | } | ||
| 236 | |||
| 237 |
1/6✗ Branch 0 not taken.
✓ Branch 1 taken 19035 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
|
19035 | if(s->avmv && mb_y < mb_h && plane_index == 0) |
| 238 | ✗ | for(mb_x=0; mb_x<mb_w; mb_x++){ | |
| 239 | ✗ | AVMotionVector *avmv = s->avmv + s->avmv_index; | |
| 240 | ✗ | const int b_width = s->b_width << s->block_max_depth; | |
| 241 | ✗ | const int b_stride= b_width; | |
| 242 | ✗ | BlockNode *bn= &s->block[mb_x + mb_y*b_stride]; | |
| 243 | |||
| 244 | ✗ | if (bn->type) | |
| 245 | ✗ | continue; | |
| 246 | |||
| 247 | ✗ | s->avmv_index++; | |
| 248 | |||
| 249 | ✗ | avmv->w = block_w; | |
| 250 | ✗ | avmv->h = block_h; | |
| 251 | ✗ | avmv->dst_x = block_w*mb_x - block_w/2; | |
| 252 | ✗ | avmv->dst_y = block_h*mb_y - block_h/2; | |
| 253 | ✗ | avmv->motion_scale = 8; | |
| 254 | ✗ | avmv->motion_x = bn->mx * s->mv_scale; | |
| 255 | ✗ | avmv->motion_y = bn->my * s->mv_scale; | |
| 256 | ✗ | avmv->src_x = avmv->dst_x + avmv->motion_x / 8; | |
| 257 | ✗ | avmv->src_y = avmv->dst_y + avmv->motion_y / 8; | |
| 258 | ✗ | avmv->source= -1 - bn->ref; | |
| 259 | ✗ | avmv->flags = 0; | |
| 260 | } | ||
| 261 | } | ||
| 262 | |||
| 263 | 157912 | static inline void decode_subband_slice_buffered(SnowContext *s, SubBand *b, slice_buffer * sb, int start_y, int h, int save_state[1]){ | |
| 264 | 157912 | const int w= b->width; | |
| 265 | int y; | ||
| 266 | 157912 | const int qlog= av_clip(s->qlog + (int64_t)b->qlog, 0, QROOT*16); | |
| 267 | 157912 | int qmul= ff_qexp[qlog&(QROOT-1)]<<(qlog>>QSHIFT); | |
| 268 | 157912 | int qadd= (s->qbias*qmul)>>QBIAS_SHIFT; | |
| 269 | 157912 | int new_index = 0; | |
| 270 | |||
| 271 |
4/4✓ Branch 0 taken 156375 times.
✓ Branch 1 taken 1537 times.
✓ Branch 2 taken 129189 times.
✓ Branch 3 taken 27186 times.
|
157912 | if(b->ibuf == s->spatial_idwt_buffer || s->qlog == LOSSLESS_QLOG){ |
| 272 | 130726 | qadd= 0; | |
| 273 | 130726 | qmul= 1<<QEXPSHIFT; | |
| 274 | } | ||
| 275 | |||
| 276 | /* If we are on the second or later slice, restore our index. */ | ||
| 277 |
2/2✓ Branch 0 taken 135784 times.
✓ Branch 1 taken 22128 times.
|
157912 | if (start_y != 0) |
| 278 | 135784 | new_index = save_state[0]; | |
| 279 | |||
| 280 | |||
| 281 |
2/2✓ Branch 0 taken 376000 times.
✓ Branch 1 taken 157912 times.
|
533912 | for(y=start_y; y<h; y++){ |
| 282 | 376000 | int x = 0; | |
| 283 | int v; | ||
| 284 |
2/2✓ Branch 0 taken 260289 times.
✓ Branch 1 taken 115711 times.
|
376000 | IDWTELEM * line = slice_buffer_get_line(sb, y * b->stride_line + b->buf_y_offset) + b->buf_x_offset; |
| 285 | 376000 | memset(line, 0, b->width*sizeof(IDWTELEM)); | |
| 286 | 376000 | v = b->x_coeff[new_index].coeff; | |
| 287 | 376000 | x = b->x_coeff[new_index++].x; | |
| 288 |
2/2✓ Branch 0 taken 16497703 times.
✓ Branch 1 taken 376000 times.
|
16873703 | while(x < w){ |
| 289 | 16497703 | register int t= (int)( (v>>1)*(unsigned)qmul + qadd)>>QEXPSHIFT; | |
| 290 | 16497703 | register int u= -(v&1); | |
| 291 | 16497703 | line[x] = (t^u) - u; | |
| 292 | |||
| 293 | 16497703 | v = b->x_coeff[new_index].coeff; | |
| 294 | 16497703 | x = b->x_coeff[new_index++].x; | |
| 295 | } | ||
| 296 | } | ||
| 297 | |||
| 298 | /* Save our variables for the next slice. */ | ||
| 299 | 157912 | save_state[0] = new_index; | |
| 300 | |||
| 301 | 157912 | return; | |
| 302 | } | ||
| 303 | |||
| 304 | 250860 | static int decode_q_branch(SnowContext *s, int level, int x, int y){ | |
| 305 | 250860 | const int w= s->b_width << s->block_max_depth; | |
| 306 | 250860 | const int rem_depth= s->block_max_depth - level; | |
| 307 | 250860 | const int index= (x + y*w) << rem_depth; | |
| 308 | 250860 | int trx= (x+1)<<rem_depth; | |
| 309 |
2/2✓ Branch 0 taken 242678 times.
✓ Branch 1 taken 8182 times.
|
250860 | const BlockNode *left = x ? &s->block[index-1] : &null_block; |
| 310 |
2/2✓ Branch 0 taken 239944 times.
✓ Branch 1 taken 10916 times.
|
250860 | const BlockNode *top = y ? &s->block[index-w] : &null_block; |
| 311 |
4/4✓ Branch 0 taken 239944 times.
✓ Branch 1 taken 10916 times.
✓ Branch 2 taken 232338 times.
✓ Branch 3 taken 7606 times.
|
250860 | const BlockNode *tl = y && x ? &s->block[index-w-1] : left; |
| 312 |
8/8✓ Branch 0 taken 239944 times.
✓ Branch 1 taken 10916 times.
✓ Branch 2 taken 232222 times.
✓ Branch 3 taken 7722 times.
✓ Branch 4 taken 112250 times.
✓ Branch 5 taken 119972 times.
✓ Branch 6 taken 28943 times.
✓ Branch 7 taken 83307 times.
|
250860 | const BlockNode *tr = y && trx<w && ((x&1)==0 || level==0) ? &s->block[index-w+(1<<rem_depth)] : tl; //FIXME use lt |
| 313 | 250860 | int s_context= 2*left->level + 2*top->level + tl->level + tr->level; | |
| 314 | int res; | ||
| 315 | |||
| 316 |
2/2✓ Branch 0 taken 8708 times.
✓ Branch 1 taken 242152 times.
|
250860 | if(s->keyframe){ |
| 317 | 8708 | set_blocks(s, level, x, y, null_block.color[0], null_block.color[1], null_block.color[2], null_block.mx, null_block.my, null_block.ref, BLOCK_INTRA); | |
| 318 | 8708 | return 0; | |
| 319 | } | ||
| 320 | |||
| 321 |
4/4✓ Branch 0 taken 53460 times.
✓ Branch 1 taken 188692 times.
✓ Branch 3 taken 8447 times.
✓ Branch 4 taken 45013 times.
|
439291 | if(level==s->block_max_depth || get_rac(&s->c, &s->block_state[4 + s_context])){ |
| 322 | int type, mx, my; | ||
| 323 | 197139 | int l = left->color[0]; | |
| 324 | 197139 | int cb= left->color[1]; | |
| 325 | 197139 | int cr= left->color[2]; | |
| 326 | 197139 | unsigned ref = 0; | |
| 327 | 197139 | int ref_context= av_log2(2*left->ref) + av_log2(2*top->ref); | |
| 328 | 197139 | int mx_context= av_log2(2*FFABS(left->mx - top->mx)) + 0*av_log2(2*FFABS(tr->mx - top->mx)); | |
| 329 | 197139 | int my_context= av_log2(2*FFABS(left->my - top->my)) + 0*av_log2(2*FFABS(tr->my - top->my)); | |
| 330 | |||
| 331 | 197139 | type= get_rac(&s->c, &s->block_state[1 + left->type + top->type]) ? BLOCK_INTRA : 0; | |
| 332 |
2/2✓ Branch 0 taken 51 times.
✓ Branch 1 taken 197088 times.
|
197139 | if(type){ |
| 333 | int ld, cbd, crd; | ||
| 334 | 51 | pred_mv(s, &mx, &my, 0, left, top, tr); | |
| 335 | 51 | ld = get_symbol(&s->c, &s->block_state[32], 1); | |
| 336 |
2/4✓ Branch 0 taken 51 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 51 times.
|
51 | if (ld < -255 || ld > 255) { |
| 337 | ✗ | return AVERROR_INVALIDDATA; | |
| 338 | } | ||
| 339 | 51 | l += ld; | |
| 340 |
1/2✓ Branch 0 taken 51 times.
✗ Branch 1 not taken.
|
51 | if (s->nb_planes > 2) { |
| 341 | 51 | cbd = get_symbol(&s->c, &s->block_state[64], 1); | |
| 342 | 51 | crd = get_symbol(&s->c, &s->block_state[96], 1); | |
| 343 |
4/8✓ Branch 0 taken 51 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 51 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 51 times.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✓ Branch 7 taken 51 times.
|
51 | if (cbd < -255 || cbd > 255 || crd < -255 || crd > 255) { |
| 344 | ✗ | return AVERROR_INVALIDDATA; | |
| 345 | } | ||
| 346 | 51 | cb += cbd; | |
| 347 | 51 | cr += crd; | |
| 348 | } | ||
| 349 | }else{ | ||
| 350 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 197088 times.
|
197088 | if(s->ref_frames > 1) |
| 351 | ✗ | ref= get_symbol(&s->c, &s->block_state[128 + 1024 + 32*ref_context], 0); | |
| 352 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 197088 times.
|
197088 | if (ref >= s->ref_frames) { |
| 353 | ✗ | av_log(s->avctx, AV_LOG_ERROR, "Invalid ref\n"); | |
| 354 | ✗ | return AVERROR_INVALIDDATA; | |
| 355 | } | ||
| 356 | 197088 | pred_mv(s, &mx, &my, ref, left, top, tr); | |
| 357 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 197088 times.
|
197088 | mx+= (unsigned)get_symbol(&s->c, &s->block_state[128 + 32*(mx_context + 16*!!ref)], 1); |
| 358 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 197088 times.
|
197088 | my+= (unsigned)get_symbol(&s->c, &s->block_state[128 + 32*(my_context + 16*!!ref)], 1); |
| 359 | } | ||
| 360 | 197139 | set_blocks(s, level, x, y, l, cb, cr, mx, my, ref, type); | |
| 361 | }else{ | ||
| 362 |
2/4✓ Branch 1 taken 45013 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 45013 times.
✗ Branch 4 not taken.
|
90026 | if ((res = decode_q_branch(s, level+1, 2*x+0, 2*y+0)) < 0 || |
| 363 |
1/2✓ Branch 1 taken 45013 times.
✗ Branch 2 not taken.
|
90026 | (res = decode_q_branch(s, level+1, 2*x+1, 2*y+0)) < 0 || |
| 364 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 45013 times.
|
90026 | (res = decode_q_branch(s, level+1, 2*x+0, 2*y+1)) < 0 || |
| 365 | 45013 | (res = decode_q_branch(s, level+1, 2*x+1, 2*y+1)) < 0) | |
| 366 | ✗ | return res; | |
| 367 | } | ||
| 368 | 242152 | return 0; | |
| 369 | } | ||
| 370 | |||
| 371 | 1537 | static void dequantize_slice_buffered(SnowContext *s, slice_buffer * sb, SubBand *b, IDWTELEM *src, int stride, int start_y, int end_y){ | |
| 372 | 1537 | const int w= b->width; | |
| 373 | 1537 | const int qlog= av_clip(s->qlog + (int64_t)b->qlog, 0, QROOT*16); | |
| 374 | 1537 | const int qmul= ff_qexp[qlog&(QROOT-1)]<<(qlog>>QSHIFT); | |
| 375 | 1537 | const int qadd= (s->qbias*qmul)>>QBIAS_SHIFT; | |
| 376 | int x,y; | ||
| 377 | |||
| 378 |
2/2✓ Branch 0 taken 616 times.
✓ Branch 1 taken 921 times.
|
1537 | if(s->qlog == LOSSLESS_QLOG) return; |
| 379 | |||
| 380 |
2/2✓ Branch 0 taken 1228 times.
✓ Branch 1 taken 921 times.
|
2149 | for(y=start_y; y<end_y; y++){ |
| 381 | // DWTELEM * line = slice_buffer_get_line_from_address(sb, src + (y * stride)); | ||
| 382 |
1/2✓ Branch 0 taken 1228 times.
✗ Branch 1 not taken.
|
1228 | IDWTELEM * line = slice_buffer_get_line(sb, (y * b->stride_line) + b->buf_y_offset) + b->buf_x_offset; |
| 383 |
2/2✓ Branch 0 taken 3684 times.
✓ Branch 1 taken 1228 times.
|
4912 | for(x=0; x<w; x++){ |
| 384 | 3684 | int i= line[x]; | |
| 385 |
2/2✓ Branch 0 taken 1595 times.
✓ Branch 1 taken 2089 times.
|
3684 | if(i<0){ |
| 386 | 1595 | line[x]= -((-i*(unsigned)qmul + qadd)>>(QEXPSHIFT)); //FIXME try different bias | |
| 387 |
2/2✓ Branch 0 taken 980 times.
✓ Branch 1 taken 1109 times.
|
2089 | }else if(i>0){ |
| 388 | 980 | line[x]= (( i*(unsigned)qmul + qadd)>>(QEXPSHIFT)); | |
| 389 | } | ||
| 390 | } | ||
| 391 | } | ||
| 392 | } | ||
| 393 | |||
| 394 | 1537 | static void correlate_slice_buffered(SnowContext *s, slice_buffer * sb, SubBand *b, IDWTELEM *src, int stride, int inverse, int use_median, int start_y, int end_y){ | |
| 395 | 1537 | const int w= b->width; | |
| 396 | int x,y; | ||
| 397 | |||
| 398 | 1537 | IDWTELEM * line=0; // silence silly "could be used without having been initialized" warning | |
| 399 | IDWTELEM * prev; | ||
| 400 | |||
| 401 |
2/2✓ Branch 0 taken 154 times.
✓ Branch 1 taken 1383 times.
|
1537 | if (start_y != 0) |
| 402 |
1/2✓ Branch 0 taken 154 times.
✗ Branch 1 not taken.
|
154 | line = slice_buffer_get_line(sb, ((start_y - 1) * b->stride_line) + b->buf_y_offset) + b->buf_x_offset; |
| 403 | |||
| 404 |
2/2✓ Branch 0 taken 4154 times.
✓ Branch 1 taken 1537 times.
|
5691 | for(y=start_y; y<end_y; y++){ |
| 405 | 4154 | prev = line; | |
| 406 | // line = slice_buffer_get_line_from_address(sb, src + (y * stride)); | ||
| 407 |
1/2✓ Branch 0 taken 4154 times.
✗ Branch 1 not taken.
|
4154 | line = slice_buffer_get_line(sb, (y * b->stride_line) + b->buf_y_offset) + b->buf_x_offset; |
| 408 |
2/2✓ Branch 0 taken 28170 times.
✓ Branch 1 taken 4154 times.
|
32324 | for(x=0; x<w; x++){ |
| 409 |
2/2✓ Branch 0 taken 24016 times.
✓ Branch 1 taken 4154 times.
|
28170 | if(x){ |
| 410 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 24016 times.
|
24016 | if(use_median){ |
| 411 | ✗ | if(y && x+1<w) line[x] += mid_pred(line[x - 1], prev[x], prev[x + 1]); | |
| 412 | ✗ | else line[x] += line[x - 1]; | |
| 413 | }else{ | ||
| 414 |
2/2✓ Branch 0 taken 19401 times.
✓ Branch 1 taken 4615 times.
|
24016 | if(y) line[x] += mid_pred(line[x - 1], prev[x], line[x - 1] + prev[x] - prev[x - 1]); |
| 415 | 4615 | else line[x] += line[x - 1]; | |
| 416 | } | ||
| 417 | }else{ | ||
| 418 |
2/2✓ Branch 0 taken 2771 times.
✓ Branch 1 taken 1383 times.
|
4154 | if(y) line[x] += prev[x]; |
| 419 | } | ||
| 420 | } | ||
| 421 | } | ||
| 422 | 1537 | } | |
| 423 | |||
| 424 | 56 | static void decode_qlogs(SnowContext *s){ | |
| 425 | int plane_index, level, orientation; | ||
| 426 | |||
| 427 |
2/2✓ Branch 0 taken 168 times.
✓ Branch 1 taken 56 times.
|
224 | for(plane_index=0; plane_index < s->nb_planes; plane_index++){ |
| 428 |
2/2✓ Branch 0 taken 840 times.
✓ Branch 1 taken 168 times.
|
1008 | for(level=0; level<s->spatial_decomposition_count; level++){ |
| 429 |
2/2✓ Branch 0 taken 2688 times.
✓ Branch 1 taken 840 times.
|
3528 | for(orientation=level ? 1:0; orientation<4; orientation++){ |
| 430 | int q; | ||
| 431 |
2/2✓ Branch 0 taken 896 times.
✓ Branch 1 taken 1792 times.
|
2688 | if (plane_index==2) q= s->plane[1].band[level][orientation].qlog; |
| 432 |
2/2✓ Branch 0 taken 560 times.
✓ Branch 1 taken 1232 times.
|
1792 | else if(orientation==2) q= s->plane[plane_index].band[level][1].qlog; |
| 433 | 1232 | else q= get_symbol(&s->c, s->header_state, 1); | |
| 434 | 2688 | s->plane[plane_index].band[level][orientation].qlog= q; | |
| 435 | } | ||
| 436 | } | ||
| 437 | } | ||
| 438 | 56 | } | |
| 439 | |||
| 440 | #define GET_S(dst, check) \ | ||
| 441 | tmp= get_symbol(&s->c, s->header_state, 0);\ | ||
| 442 | if(!(check)){\ | ||
| 443 | av_log(s->avctx, AV_LOG_ERROR, "Error " #dst " is %d\n", tmp);\ | ||
| 444 | return AVERROR_INVALIDDATA;\ | ||
| 445 | }\ | ||
| 446 | dst= tmp; | ||
| 447 | |||
| 448 | 461 | static int decode_header(SnowContext *s){ | |
| 449 | int plane_index, tmp; | ||
| 450 | uint8_t kstate[32]; | ||
| 451 | |||
| 452 | 461 | memset(kstate, MID_STATE, sizeof(kstate)); | |
| 453 | |||
| 454 | 461 | s->keyframe= get_rac(&s->c, kstate); | |
| 455 |
3/4✓ Branch 0 taken 405 times.
✓ Branch 1 taken 56 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 405 times.
|
461 | if(s->keyframe || s->always_reset){ |
| 456 | 56 | ff_snow_reset_contexts(s); | |
| 457 | 56 | s->spatial_decomposition_type= | |
| 458 | 56 | s->qlog= | |
| 459 | 56 | s->qbias= | |
| 460 | 56 | s->mv_scale= | |
| 461 | 56 | s->block_max_depth= 0; | |
| 462 | } | ||
| 463 |
2/2✓ Branch 0 taken 56 times.
✓ Branch 1 taken 405 times.
|
461 | if(s->keyframe){ |
| 464 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 56 times.
|
56 | GET_S(s->version, tmp <= 0U) |
| 465 | 56 | s->always_reset= get_rac(&s->c, s->header_state); | |
| 466 | 56 | s->temporal_decomposition_type= get_symbol(&s->c, s->header_state, 0); | |
| 467 | 56 | s->temporal_decomposition_count= get_symbol(&s->c, s->header_state, 0); | |
| 468 |
2/4✓ Branch 1 taken 56 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 56 times.
|
56 | GET_S(s->spatial_decomposition_count, 0 < tmp && tmp <= MAX_DECOMPOSITIONS) |
| 469 | 56 | s->colorspace_type= get_symbol(&s->c, s->header_state, 0); | |
| 470 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 56 times.
|
56 | if (s->colorspace_type == 1) { |
| 471 | ✗ | s->avctx->pix_fmt= AV_PIX_FMT_GRAY8; | |
| 472 | ✗ | s->nb_planes = 1; | |
| 473 |
1/2✓ Branch 0 taken 56 times.
✗ Branch 1 not taken.
|
56 | } else if(s->colorspace_type == 0) { |
| 474 | 56 | s->chroma_h_shift= get_symbol(&s->c, s->header_state, 0); | |
| 475 | 56 | s->chroma_v_shift= get_symbol(&s->c, s->header_state, 0); | |
| 476 | |||
| 477 |
2/4✓ Branch 0 taken 56 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 56 times.
✗ Branch 3 not taken.
|
56 | if(s->chroma_h_shift == 1 && s->chroma_v_shift==1){ |
| 478 | 56 | s->avctx->pix_fmt= AV_PIX_FMT_YUV420P; | |
| 479 | ✗ | }else if(s->chroma_h_shift == 0 && s->chroma_v_shift==0){ | |
| 480 | ✗ | s->avctx->pix_fmt= AV_PIX_FMT_YUV444P; | |
| 481 | ✗ | }else if(s->chroma_h_shift == 2 && s->chroma_v_shift==2){ | |
| 482 | ✗ | s->avctx->pix_fmt= AV_PIX_FMT_YUV410P; | |
| 483 | } else { | ||
| 484 | ✗ | av_log(s->avctx, AV_LOG_ERROR, | |
| 485 | "unsupported color subsample mode %d %d\n", | ||
| 486 | s->chroma_h_shift, s->chroma_v_shift); | ||
| 487 | ✗ | s->chroma_h_shift = s->chroma_v_shift = 1; | |
| 488 | ✗ | s->avctx->pix_fmt= AV_PIX_FMT_YUV420P; | |
| 489 | ✗ | return AVERROR_INVALIDDATA; | |
| 490 | } | ||
| 491 | 56 | s->nb_planes = 3; | |
| 492 | } else { | ||
| 493 | ✗ | av_log(s->avctx, AV_LOG_ERROR, "unsupported color space\n"); | |
| 494 | ✗ | s->chroma_h_shift = s->chroma_v_shift = 1; | |
| 495 | ✗ | s->avctx->pix_fmt= AV_PIX_FMT_YUV420P; | |
| 496 | ✗ | return AVERROR_INVALIDDATA; | |
| 497 | } | ||
| 498 | |||
| 499 | |||
| 500 | 56 | s->spatial_scalability= get_rac(&s->c, s->header_state); | |
| 501 | // s->rate_scalability= get_rac(&s->c, s->header_state); | ||
| 502 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 56 times.
|
56 | GET_S(s->max_ref_frames, tmp < (unsigned)MAX_REF_FRAMES) |
| 503 | 56 | s->max_ref_frames++; | |
| 504 | |||
| 505 | 56 | decode_qlogs(s); | |
| 506 | } | ||
| 507 | |||
| 508 |
2/2✓ Branch 0 taken 405 times.
✓ Branch 1 taken 56 times.
|
461 | if(!s->keyframe){ |
| 509 |
2/2✓ Branch 1 taken 45 times.
✓ Branch 2 taken 360 times.
|
405 | if(get_rac(&s->c, s->header_state)){ |
| 510 |
2/2✓ Branch 0 taken 90 times.
✓ Branch 1 taken 45 times.
|
135 | for(plane_index=0; plane_index<FFMIN(s->nb_planes, 2); plane_index++){ |
| 511 | 90 | int htaps, i, sum=0; | |
| 512 | 90 | Plane *p= &s->plane[plane_index]; | |
| 513 | 90 | p->diag_mc= get_rac(&s->c, s->header_state); | |
| 514 | 90 | htaps= get_symbol(&s->c, s->header_state, 0); | |
| 515 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 90 times.
|
90 | if((unsigned)htaps >= HTAPS_MAX/2 - 1) |
| 516 | ✗ | return AVERROR_INVALIDDATA; | |
| 517 | 90 | htaps = htaps*2 + 2; | |
| 518 | 90 | p->htaps= htaps; | |
| 519 |
2/2✓ Branch 0 taken 270 times.
✓ Branch 1 taken 90 times.
|
360 | for(i= htaps/2; i; i--){ |
| 520 | 270 | unsigned hcoeff = get_symbol(&s->c, s->header_state, 0); | |
| 521 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 270 times.
|
270 | if (hcoeff > 127) |
| 522 | ✗ | return AVERROR_INVALIDDATA; | |
| 523 | 270 | p->hcoeff[i]= hcoeff * (1-2*(i&1)); | |
| 524 | 270 | sum += p->hcoeff[i]; | |
| 525 | } | ||
| 526 | 90 | p->hcoeff[0]= 32-sum; | |
| 527 | } | ||
| 528 | 45 | s->plane[2].diag_mc= s->plane[1].diag_mc; | |
| 529 | 45 | s->plane[2].htaps = s->plane[1].htaps; | |
| 530 | 45 | memcpy(s->plane[2].hcoeff, s->plane[1].hcoeff, sizeof(s->plane[1].hcoeff)); | |
| 531 | } | ||
| 532 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 405 times.
|
405 | if(get_rac(&s->c, s->header_state)){ |
| 533 | ✗ | GET_S(s->spatial_decomposition_count, 0 < tmp && tmp <= MAX_DECOMPOSITIONS) | |
| 534 | ✗ | decode_qlogs(s); | |
| 535 | } | ||
| 536 | } | ||
| 537 | |||
| 538 | 461 | s->spatial_decomposition_type+= (unsigned)get_symbol(&s->c, s->header_state, 1); | |
| 539 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 461 times.
|
461 | if(s->spatial_decomposition_type > 1U){ |
| 540 | ✗ | av_log(s->avctx, AV_LOG_ERROR, "spatial_decomposition_type %d not supported\n", s->spatial_decomposition_type); | |
| 541 | ✗ | return AVERROR_INVALIDDATA; | |
| 542 | } | ||
| 543 | 461 | if(FFMIN(s->avctx-> width>>s->chroma_h_shift, | |
| 544 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 461 times.
|
461 | s->avctx->height>>s->chroma_v_shift) >> (s->spatial_decomposition_count-1) <= 1){ |
| 545 | ✗ | av_log(s->avctx, AV_LOG_ERROR, "spatial_decomposition_count %d too large for size\n", s->spatial_decomposition_count); | |
| 546 | ✗ | return AVERROR_INVALIDDATA; | |
| 547 | } | ||
| 548 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 461 times.
|
461 | if (s->avctx->width > 65536-4) { |
| 549 | ✗ | av_log(s->avctx, AV_LOG_ERROR, "Width %d is too large\n", s->avctx->width); | |
| 550 | ✗ | return AVERROR_INVALIDDATA; | |
| 551 | } | ||
| 552 | |||
| 553 | |||
| 554 | 461 | s->qlog += (unsigned)get_symbol(&s->c, s->header_state, 1); | |
| 555 | 461 | s->mv_scale += (unsigned)get_symbol(&s->c, s->header_state, 1); | |
| 556 | 461 | s->qbias += (unsigned)get_symbol(&s->c, s->header_state, 1); | |
| 557 | 461 | s->block_max_depth+= (unsigned)get_symbol(&s->c, s->header_state, 1); | |
| 558 |
3/6✓ Branch 0 taken 461 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 461 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 461 times.
|
461 | if(s->block_max_depth > 1 || s->block_max_depth < 0 || s->mv_scale > 256U){ |
| 559 | ✗ | av_log(s->avctx, AV_LOG_ERROR, "block_max_depth= %d is too large\n", s->block_max_depth); | |
| 560 | ✗ | s->block_max_depth= 0; | |
| 561 | ✗ | s->mv_scale = 0; | |
| 562 | ✗ | return AVERROR_INVALIDDATA; | |
| 563 | } | ||
| 564 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 461 times.
|
461 | if (FFABS(s->qbias) > 127) { |
| 565 | ✗ | av_log(s->avctx, AV_LOG_ERROR, "qbias %d is too large\n", s->qbias); | |
| 566 | ✗ | s->qbias = 0; | |
| 567 | ✗ | return AVERROR_INVALIDDATA; | |
| 568 | } | ||
| 569 | |||
| 570 | 461 | return 0; | |
| 571 | } | ||
| 572 | |||
| 573 | 461 | static int decode_blocks(SnowContext *s){ | |
| 574 | int x, y; | ||
| 575 | 461 | int w= s->b_width; | |
| 576 | 461 | int h= s->b_height; | |
| 577 | int res; | ||
| 578 | |||
| 579 |
2/2✓ Branch 0 taken 4000 times.
✓ Branch 1 taken 461 times.
|
4461 | for(y=0; y<h; y++){ |
| 580 |
2/2✓ Branch 0 taken 70808 times.
✓ Branch 1 taken 4000 times.
|
74808 | for(x=0; x<w; x++){ |
| 581 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 70808 times.
|
70808 | if (s->c.bytestream >= s->c.bytestream_end) |
| 582 | ✗ | return AVERROR_INVALIDDATA; | |
| 583 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 70808 times.
|
70808 | if ((res = decode_q_branch(s, 0, x, y)) < 0) |
| 584 | ✗ | return res; | |
| 585 | } | ||
| 586 | } | ||
| 587 | 461 | return 0; | |
| 588 | } | ||
| 589 | |||
| 590 | 461 | static int decode_frame(AVCodecContext *avctx, AVFrame *picture, | |
| 591 | int *got_frame, AVPacket *avpkt) | ||
| 592 | { | ||
| 593 | 461 | const uint8_t *buf = avpkt->data; | |
| 594 | 461 | int buf_size = avpkt->size; | |
| 595 | 461 | SnowContext *s = avctx->priv_data; | |
| 596 | 461 | RangeCoder * const c= &s->c; | |
| 597 | int bytes_read; | ||
| 598 | int level, orientation, plane_index; | ||
| 599 | int res; | ||
| 600 | |||
| 601 | 461 | ff_init_range_decoder(c, buf, buf_size); | |
| 602 | 461 | ff_build_rac_states(c, 0.05*(1LL<<32), 256-8); | |
| 603 | |||
| 604 | 461 | s->current_picture->pict_type= AV_PICTURE_TYPE_I; //FIXME I vs. P | |
| 605 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 461 times.
|
461 | if ((res = decode_header(s)) < 0) |
| 606 | ✗ | return res; | |
| 607 | |||
| 608 |
2/2✓ Branch 0 taken 20 times.
✓ Branch 1 taken 441 times.
|
461 | if (!s->mconly_picture->data[0]) { |
| 609 | 20 | res = ff_get_buffer(avctx, s->mconly_picture, AV_GET_BUFFER_FLAG_REF); | |
| 610 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 20 times.
|
20 | if (res < 0) |
| 611 | ✗ | return res; | |
| 612 | } | ||
| 613 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 461 times.
|
461 | if (s->mconly_picture->format != avctx->pix_fmt) { |
| 614 | ✗ | av_log(avctx, AV_LOG_ERROR, "pixel format changed\n"); | |
| 615 | ✗ | return AVERROR_INVALIDDATA; | |
| 616 | } | ||
| 617 | |||
| 618 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 461 times.
|
461 | if ((res=ff_snow_common_init_after_header(avctx)) < 0) |
| 619 | ✗ | return res; | |
| 620 | |||
| 621 | // realloc slice buffer for the case that spatial_decomposition_count changed | ||
| 622 | 461 | ff_slice_buffer_destroy(&s->sb); | |
| 623 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 461 times.
|
461 | if ((res = ff_slice_buffer_init(&s->sb, s->plane[0].height, |
| 624 | 461 | (MB_SIZE >> s->block_max_depth) + | |
| 625 | 461 | s->spatial_decomposition_count * 11 + 1, | |
| 626 | s->plane[0].width, | ||
| 627 | s->spatial_idwt_buffer)) < 0) | ||
| 628 | ✗ | return res; | |
| 629 | |||
| 630 |
2/2✓ Branch 0 taken 1383 times.
✓ Branch 1 taken 461 times.
|
1844 | for(plane_index=0; plane_index < s->nb_planes; plane_index++){ |
| 631 | 1383 | Plane *p= &s->plane[plane_index]; | |
| 632 |
2/4✓ Branch 0 taken 1323 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 1323 times.
✗ Branch 3 not taken.
|
2706 | p->fast_mc= p->diag_mc && p->htaps==6 && p->hcoeff[0]==40 |
| 633 |
1/2✓ Branch 0 taken 1323 times.
✗ Branch 1 not taken.
|
1323 | && p->hcoeff[1]==-10 |
| 634 |
3/4✓ Branch 0 taken 1323 times.
✓ Branch 1 taken 60 times.
✓ Branch 2 taken 1323 times.
✗ Branch 3 not taken.
|
2706 | && p->hcoeff[2]==2; |
| 635 | } | ||
| 636 | |||
| 637 | 461 | ff_snow_alloc_blocks(s); | |
| 638 | |||
| 639 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 461 times.
|
461 | if ((res = ff_snow_frames_prepare(s)) < 0) |
| 640 | ✗ | return res; | |
| 641 | |||
| 642 | 461 | s->current_picture->width = s->avctx->width; | |
| 643 | 461 | s->current_picture->height = s->avctx->height; | |
| 644 | 461 | res = ff_get_buffer(s->avctx, s->current_picture, AV_GET_BUFFER_FLAG_REF); | |
| 645 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 461 times.
|
461 | if (res < 0) |
| 646 | ✗ | return res; | |
| 647 | |||
| 648 |
2/2✓ Branch 0 taken 56 times.
✓ Branch 1 taken 405 times.
|
461 | s->current_picture->pict_type = s->keyframe ? AV_PICTURE_TYPE_I : AV_PICTURE_TYPE_P; |
| 649 | |||
| 650 | //keyframe flag duplication mess FIXME | ||
| 651 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 461 times.
|
461 | if(avctx->debug&FF_DEBUG_PICT_INFO) |
| 652 | ✗ | av_log(avctx, AV_LOG_ERROR, | |
| 653 | "keyframe:%d qlog:%d qbias: %d mvscale: %d " | ||
| 654 | "decomposition_type:%d decomposition_count:%d\n", | ||
| 655 | s->keyframe, s->qlog, s->qbias, s->mv_scale, | ||
| 656 | s->spatial_decomposition_type, | ||
| 657 | s->spatial_decomposition_count | ||
| 658 | ); | ||
| 659 | |||
| 660 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 461 times.
|
461 | if (s->avctx->export_side_data & AV_CODEC_EXPORT_DATA_MVS) { |
| 661 | size_t size; | ||
| 662 | ✗ | res = av_size_mult(s->b_width * s->b_height, sizeof(AVMotionVector) << (s->block_max_depth*2), &size); | |
| 663 | ✗ | if (res) | |
| 664 | ✗ | return res; | |
| 665 | ✗ | av_fast_malloc(&s->avmv, &s->avmv_size, size); | |
| 666 | ✗ | if (!s->avmv) | |
| 667 | ✗ | return AVERROR(ENOMEM); | |
| 668 | } else { | ||
| 669 | 461 | s->avmv_size = 0; | |
| 670 | 461 | av_freep(&s->avmv); | |
| 671 | } | ||
| 672 | 461 | s->avmv_index = 0; | |
| 673 | |||
| 674 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 461 times.
|
461 | if ((res = decode_blocks(s)) < 0) |
| 675 | ✗ | return res; | |
| 676 | |||
| 677 |
2/2✓ Branch 0 taken 1383 times.
✓ Branch 1 taken 461 times.
|
1844 | for(plane_index=0; plane_index < s->nb_planes; plane_index++){ |
| 678 | 1383 | Plane *p= &s->plane[plane_index]; | |
| 679 | 1383 | int w= p->width; | |
| 680 | 1383 | int h= p->height; | |
| 681 | int x, y; | ||
| 682 | int decode_state[MAX_DECOMPOSITIONS][4][1]; /* Stored state info for unpack_coeffs. 1 variable per instance. */ | ||
| 683 | |||
| 684 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1383 times.
|
1383 | if(s->avctx->debug&2048){ |
| 685 | ✗ | memset(s->spatial_dwt_buffer, 0, sizeof(DWTELEM)*w*h); | |
| 686 | ✗ | predict_plane(s, s->spatial_idwt_buffer, plane_index, 1); | |
| 687 | |||
| 688 | ✗ | for(y=0; y<h; y++){ | |
| 689 | ✗ | for(x=0; x<w; x++){ | |
| 690 | ✗ | int v= s->current_picture->data[plane_index][y*s->current_picture->linesize[plane_index] + x]; | |
| 691 | ✗ | s->mconly_picture->data[plane_index][y*s->mconly_picture->linesize[plane_index] + x]= v; | |
| 692 | } | ||
| 693 | } | ||
| 694 | } | ||
| 695 | |||
| 696 |
2/2✓ Branch 0 taken 6915 times.
✓ Branch 1 taken 1383 times.
|
8298 | for(level=0; level<s->spatial_decomposition_count; level++){ |
| 697 |
2/2✓ Branch 0 taken 22128 times.
✓ Branch 1 taken 6915 times.
|
29043 | for(orientation=level ? 1 : 0; orientation<4; orientation++){ |
| 698 | 22128 | SubBand *b= &p->band[level][orientation]; | |
| 699 | 22128 | unpack_coeffs(s, b, b->parent, orientation); | |
| 700 | } | ||
| 701 | } | ||
| 702 | |||
| 703 | { | ||
| 704 | 1383 | const int mb_h= s->b_height << s->block_max_depth; | |
| 705 | 1383 | const int block_size = MB_SIZE >> s->block_max_depth; | |
| 706 |
2/2✓ Branch 0 taken 922 times.
✓ Branch 1 taken 461 times.
|
1383 | const int block_h = plane_index ? block_size>>s->chroma_v_shift : block_size; |
| 707 | int mb_y; | ||
| 708 | DWTCompose cs[MAX_DECOMPOSITIONS]; | ||
| 709 | 1383 | int yd=0, yq=0; | |
| 710 | int y; | ||
| 711 | int end_y; | ||
| 712 | |||
| 713 | 1383 | ff_spatial_idwt_buffered_init(cs, &s->sb, w, h, 1, s->spatial_decomposition_type, s->spatial_decomposition_count); | |
| 714 |
2/2✓ Branch 0 taken 21699 times.
✓ Branch 1 taken 1383 times.
|
23082 | for(mb_y=0; mb_y<=mb_h; mb_y++){ |
| 715 | |||
| 716 | 21699 | int slice_starty = block_h*mb_y; | |
| 717 | 21699 | int slice_h = block_h*(mb_y+1); | |
| 718 | |||
| 719 |
3/4✓ Branch 0 taken 19035 times.
✓ Branch 1 taken 2664 times.
✓ Branch 2 taken 19035 times.
✗ Branch 3 not taken.
|
21699 | if (!(s->keyframe || s->avctx->debug&512)){ |
| 720 | 19035 | slice_starty = FFMAX(0, slice_starty - (block_h >> 1)); | |
| 721 | 19035 | slice_h -= (block_h >> 1); | |
| 722 | } | ||
| 723 | |||
| 724 |
2/2✓ Branch 0 taken 108495 times.
✓ Branch 1 taken 21699 times.
|
130194 | for(level=0; level<s->spatial_decomposition_count; level++){ |
| 725 |
2/2✓ Branch 0 taken 347184 times.
✓ Branch 1 taken 108495 times.
|
455679 | for(orientation=level ? 1 : 0; orientation<4; orientation++){ |
| 726 | 347184 | SubBand *b= &p->band[level][orientation]; | |
| 727 | int start_y; | ||
| 728 | int end_y; | ||
| 729 | 347184 | int our_mb_start = mb_y; | |
| 730 | 347184 | int our_mb_end = (mb_y + 1); | |
| 731 | 347184 | const int extra= 3; | |
| 732 |
2/2✓ Branch 0 taken 325056 times.
✓ Branch 1 taken 22128 times.
|
347184 | start_y = (mb_y ? ((block_h * our_mb_start) >> (s->spatial_decomposition_count - level)) + s->spatial_decomposition_count - level + extra: 0); |
| 733 | 347184 | end_y = (((block_h * our_mb_end) >> (s->spatial_decomposition_count - level)) + s->spatial_decomposition_count - level + extra); | |
| 734 |
3/4✓ Branch 0 taken 304560 times.
✓ Branch 1 taken 42624 times.
✓ Branch 2 taken 304560 times.
✗ Branch 3 not taken.
|
347184 | if (!(s->keyframe || s->avctx->debug&512)){ |
| 735 | 304560 | start_y = FFMAX(0, start_y - (block_h >> (1+s->spatial_decomposition_count - level))); | |
| 736 | 304560 | end_y = FFMAX(0, end_y - (block_h >> (1+s->spatial_decomposition_count - level))); | |
| 737 | } | ||
| 738 | 347184 | start_y = FFMIN(b->height, start_y); | |
| 739 | 347184 | end_y = FFMIN(b->height, end_y); | |
| 740 | |||
| 741 |
2/2✓ Branch 0 taken 157912 times.
✓ Branch 1 taken 189272 times.
|
347184 | if (start_y != end_y){ |
| 742 |
2/2✓ Branch 0 taken 1537 times.
✓ Branch 1 taken 156375 times.
|
157912 | if (orientation == 0){ |
| 743 | 1537 | SubBand * correlate_band = &p->band[0][0]; | |
| 744 | 1537 | int correlate_end_y = FFMIN(b->height, end_y + 1); | |
| 745 |
2/2✓ Branch 0 taken 154 times.
✓ Branch 1 taken 1383 times.
|
1537 | int correlate_start_y = FFMIN(b->height, (start_y ? start_y + 1 : 0)); |
| 746 | 1537 | decode_subband_slice_buffered(s, correlate_band, &s->sb, correlate_start_y, correlate_end_y, decode_state[0][0]); | |
| 747 | 1537 | correlate_slice_buffered(s, &s->sb, correlate_band, correlate_band->ibuf, correlate_band->stride, 1, 0, correlate_start_y, correlate_end_y); | |
| 748 | 1537 | dequantize_slice_buffered(s, &s->sb, correlate_band, correlate_band->ibuf, correlate_band->stride, start_y, end_y); | |
| 749 | } | ||
| 750 | else | ||
| 751 | 156375 | decode_subband_slice_buffered(s, b, &s->sb, start_y, end_y, decode_state[level][orientation]); | |
| 752 | } | ||
| 753 | } | ||
| 754 | } | ||
| 755 | |||
| 756 |
2/2✓ Branch 0 taken 33857 times.
✓ Branch 1 taken 21699 times.
|
55556 | for(; yd<slice_h; yd+=4){ |
| 757 | 33857 | ff_spatial_idwt_buffered_slice(&s->dwt, cs, &s->sb, s->temp_idwt_buffer, w, h, 1, s->spatial_decomposition_type, s->spatial_decomposition_count, yd); | |
| 758 | } | ||
| 759 | |||
| 760 |
2/2✓ Branch 0 taken 17094 times.
✓ Branch 1 taken 4605 times.
|
21699 | if(s->qlog == LOSSLESS_QLOG){ |
| 761 |
4/4✓ Branch 0 taken 89166 times.
✓ Branch 1 taken 16632 times.
✓ Branch 2 taken 88704 times.
✓ Branch 3 taken 462 times.
|
105798 | for(; yq<slice_h && yq<h; yq++){ |
| 762 |
1/2✓ Branch 0 taken 88704 times.
✗ Branch 1 not taken.
|
88704 | IDWTELEM * line = slice_buffer_get_line(&s->sb, yq); |
| 763 |
2/2✓ Branch 0 taken 23417856 times.
✓ Branch 1 taken 88704 times.
|
23506560 | for(x=0; x<w; x++){ |
| 764 | 23417856 | line[x] *= 1<<FRAC_BITS; | |
| 765 | } | ||
| 766 | } | ||
| 767 | } | ||
| 768 | |||
| 769 | 21699 | predict_slice_buffered(s, &s->sb, s->spatial_idwt_buffer, plane_index, 1, mb_y); | |
| 770 | |||
| 771 | 21699 | y = FFMIN(p->height, slice_starty); | |
| 772 | 21699 | end_y = FFMIN(p->height, slice_h); | |
| 773 |
2/2✓ Branch 0 taken 128000 times.
✓ Branch 1 taken 21699 times.
|
149699 | while(y < end_y) |
| 774 | 128000 | ff_slice_buffer_release(&s->sb, y++); | |
| 775 | } | ||
| 776 | |||
| 777 | 1383 | ff_slice_buffer_flush(&s->sb); | |
| 778 | } | ||
| 779 | |||
| 780 | } | ||
| 781 | |||
| 782 | 461 | emms_c(); | |
| 783 | |||
| 784 | 461 | av_frame_unref(s->last_picture[s->max_ref_frames - 1]); | |
| 785 | |||
| 786 |
1/2✓ Branch 0 taken 461 times.
✗ Branch 1 not taken.
|
461 | if(!(s->avctx->debug&2048)) |
| 787 | 461 | res = av_frame_ref(picture, s->current_picture); | |
| 788 | else | ||
| 789 | ✗ | res = av_frame_ref(picture, s->mconly_picture); | |
| 790 |
2/4✓ Branch 0 taken 461 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 461 times.
|
461 | if (res >= 0 && s->avmv_index) { |
| 791 | AVFrameSideData *sd; | ||
| 792 | |||
| 793 | ✗ | sd = av_frame_new_side_data(picture, AV_FRAME_DATA_MOTION_VECTORS, s->avmv_index * sizeof(AVMotionVector)); | |
| 794 | ✗ | if (!sd) | |
| 795 | ✗ | return AVERROR(ENOMEM); | |
| 796 | ✗ | memcpy(sd->data, s->avmv, s->avmv_index * sizeof(AVMotionVector)); | |
| 797 | } | ||
| 798 | |||
| 799 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 461 times.
|
461 | if (res < 0) |
| 800 | ✗ | return res; | |
| 801 | |||
| 802 | 461 | *got_frame = 1; | |
| 803 | |||
| 804 | 461 | bytes_read= c->bytestream - c->bytestream_start; | |
| 805 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 461 times.
|
461 | if(bytes_read ==0) av_log(s->avctx, AV_LOG_ERROR, "error at end of frame\n"); //FIXME |
| 806 | |||
| 807 | 461 | return bytes_read; | |
| 808 | } | ||
| 809 | |||
| 810 | 20 | static av_cold int decode_end(AVCodecContext *avctx) | |
| 811 | { | ||
| 812 | 20 | SnowContext *s = avctx->priv_data; | |
| 813 | |||
| 814 | 20 | ff_slice_buffer_destroy(&s->sb); | |
| 815 | |||
| 816 | 20 | ff_snow_common_end(s); | |
| 817 | |||
| 818 | 20 | s->avmv_size = 0; | |
| 819 | 20 | av_freep(&s->avmv); | |
| 820 | |||
| 821 | 20 | return 0; | |
| 822 | } | ||
| 823 | |||
| 824 | const FFCodec ff_snow_decoder = { | ||
| 825 | .p.name = "snow", | ||
| 826 | CODEC_LONG_NAME("Snow"), | ||
| 827 | .p.type = AVMEDIA_TYPE_VIDEO, | ||
| 828 | .p.id = AV_CODEC_ID_SNOW, | ||
| 829 | .priv_data_size = sizeof(SnowContext), | ||
| 830 | .init = ff_snow_common_init, | ||
| 831 | .close = decode_end, | ||
| 832 | FF_CODEC_DECODE_CB(decode_frame), | ||
| 833 | .p.capabilities = AV_CODEC_CAP_DR1, | ||
| 834 | .caps_internal = FF_CODEC_CAP_INIT_CLEANUP, | ||
| 835 | }; | ||
| 836 |