| Line | Branch | Exec | Source | 
|---|---|---|---|
| 1 | /* | ||
| 2 | * FFV1 decoder | ||
| 3 | * | ||
| 4 | * Copyright (c) 2003-2013 Michael Niedermayer <michaelni@gmx.at> | ||
| 5 | * | ||
| 6 | * This file is part of FFmpeg. | ||
| 7 | * | ||
| 8 | * FFmpeg is free software; you can redistribute it and/or | ||
| 9 | * modify it under the terms of the GNU Lesser General Public | ||
| 10 | * License as published by the Free Software Foundation; either | ||
| 11 | * version 2.1 of the License, or (at your option) any later version. | ||
| 12 | * | ||
| 13 | * FFmpeg is distributed in the hope that it will be useful, | ||
| 14 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
| 15 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | ||
| 16 | * Lesser General Public License for more details. | ||
| 17 | * | ||
| 18 | * You should have received a copy of the GNU Lesser General Public | ||
| 19 | * License along with FFmpeg; if not, write to the Free Software | ||
| 20 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | ||
| 21 | */ | ||
| 22 | |||
| 23 | /** | ||
| 24 | * @file | ||
| 25 | * FF Video Codec 1 (a lossless codec) decoder | ||
| 26 | */ | ||
| 27 | |||
| 28 | #include "libavutil/avassert.h" | ||
| 29 | #include "libavutil/crc.h" | ||
| 30 | #include "libavutil/mem.h" | ||
| 31 | #include "libavutil/imgutils.h" | ||
| 32 | #include "libavutil/pixdesc.h" | ||
| 33 | #include "avcodec.h" | ||
| 34 | #include "codec_internal.h" | ||
| 35 | #include "get_bits.h" | ||
| 36 | #include "rangecoder.h" | ||
| 37 | #include "golomb.h" | ||
| 38 | #include "mathops.h" | ||
| 39 | #include "ffv1.h" | ||
| 40 | #include "progressframe.h" | ||
| 41 | #include "libavutil/refstruct.h" | ||
| 42 | #include "thread.h" | ||
| 43 | #include "decode.h" | ||
| 44 | #include "hwconfig.h" | ||
| 45 | #include "hwaccel_internal.h" | ||
| 46 | #include "config_components.h" | ||
| 47 | |||
| 48 | 163659991 | static inline int get_vlc_symbol(GetBitContext *gb, VlcState *const state, | |
| 49 | int bits) | ||
| 50 | { | ||
| 51 | int k, i, v, ret; | ||
| 52 | |||
| 53 | 163659991 | i = state->count; | |
| 54 | 163659991 | k = 0; | |
| 55 | 
        2/2✓ Branch 0 taken 272532174 times. 
          ✓ Branch 1 taken 163659991 times. 
         | 
      436192165 | while (i < state->error_sum) { // FIXME: optimize | 
| 56 | 272532174 | k++; | |
| 57 | 272532174 | i += i; | |
| 58 | } | ||
| 59 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 163659991 times. 
         | 
      163659991 | if (k > bits) { | 
| 60 | ff_dlog(NULL, "k-overflow bias:%d error:%d drift:%d count:%d k:%d", | ||
| 61 | state->bias, state->error_sum, state->drift, state->count, k); | ||
| 62 | ✗ | k = bits; | |
| 63 | } | ||
| 64 | |||
| 65 | 163659991 | v = get_sr_golomb(gb, k, 12, bits); | |
| 66 | ff_dlog(NULL, "v:%d bias:%d error:%d drift:%d count:%d k:%d", | ||
| 67 | v, state->bias, state->error_sum, state->drift, state->count, k); | ||
| 68 | |||
| 69 | 163659991 | v ^= ((2 * state->drift + state->count) >> 31); | |
| 70 | |||
| 71 | 163659991 | ret = fold(v + state->bias, bits); | |
| 72 | |||
| 73 | 163659991 | update_vlc_state(state, v); | |
| 74 | |||
| 75 | 163659991 | return ret; | |
| 76 | } | ||
| 77 | |||
| 78 | 4542400 | static int is_input_end(RangeCoder *c, GetBitContext *gb, int ac) | |
| 79 | { | ||
| 80 | 
        2/2✓ Branch 0 taken 2514800 times. 
          ✓ Branch 1 taken 2027600 times. 
         | 
      4542400 | if (ac != AC_GOLOMB_RICE) { | 
| 81 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 2514800 times. 
         | 
      2514800 | if (c->overread > MAX_OVERREAD) | 
| 82 | ✗ | return AVERROR_INVALIDDATA; | |
| 83 | } else { | ||
| 84 | 
        1/2✗ Branch 1 not taken. 
          ✓ Branch 2 taken 2027600 times. 
         | 
      2027600 | if (get_bits_left(gb) < 1) | 
| 85 | ✗ | return AVERROR_INVALIDDATA; | |
| 86 | } | ||
| 87 | 4542400 | return 0; | |
| 88 | } | ||
| 89 | |||
| 90 | #define TYPE int16_t | ||
| 91 | #define RENAME(name) name | ||
| 92 | #include "ffv1dec_template.c" | ||
| 93 | #undef TYPE | ||
| 94 | #undef RENAME | ||
| 95 | |||
| 96 | #define TYPE int32_t | ||
| 97 | #define RENAME(name) name ## 32 | ||
| 98 | #include "ffv1dec_template.c" | ||
| 99 | |||
| 100 | 18300 | static int decode_plane(FFV1Context *f, FFV1SliceContext *sc, | |
| 101 | GetBitContext *gb, | ||
| 102 | uint8_t *src, int w, int h, int stride, int plane_index, | ||
| 103 | int remap_index, int pixel_stride, int ac) | ||
| 104 | { | ||
| 105 | int x, y; | ||
| 106 | int16_t *sample[2]; | ||
| 107 | int bits; | ||
| 108 | unsigned mask; | ||
| 109 | |||
| 110 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 18300 times. 
         | 
      18300 | if (sc->remap) { | 
| 111 | ✗ | bits = av_ceil_log2(sc->remap_count[remap_index]); | |
| 112 | ✗ | mask = (1<<bits)-1; | |
| 113 | } else { | ||
| 114 | 18300 | bits = f->avctx->bits_per_raw_sample; | |
| 115 | } | ||
| 116 | |||
| 117 | 18300 | sample[0] = sc->sample_buffer + 3; | |
| 118 | 18300 | sample[1] = sc->sample_buffer + w + 6 + 3; | |
| 119 | |||
| 120 | 18300 | sc->run_index = 0; | |
| 121 | |||
| 122 | 18300 | memset(sc->sample_buffer, 0, 2 * (w + 6) * sizeof(*sc->sample_buffer)); | |
| 123 | |||
| 124 | 
        2/2✓ Branch 0 taken 1732400 times. 
          ✓ Branch 1 taken 18300 times. 
         | 
      1750700 | for (y = 0; y < h; y++) { | 
| 125 | 1732400 | int16_t *temp = sample[0]; // FIXME: try a normal buffer | |
| 126 | |||
| 127 | 1732400 | sample[0] = sample[1]; | |
| 128 | 1732400 | sample[1] = temp; | |
| 129 | |||
| 130 | 1732400 | sample[1][-1] = sample[0][0]; | |
| 131 | 1732400 | sample[0][w] = sample[0][w - 1]; | |
| 132 | |||
| 133 | 
        2/2✓ Branch 0 taken 924200 times. 
          ✓ Branch 1 taken 808200 times. 
         | 
      1732400 | if (f->avctx->bits_per_raw_sample <= 8) { | 
| 134 | 924200 | int ret = decode_line(f, sc, gb, w, sample, plane_index, 8, ac); | |
| 135 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 924200 times. 
         | 
      924200 | if (ret < 0) | 
| 136 | ✗ | return ret; | |
| 137 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 924200 times. 
         | 
      924200 | if (sc->remap) | 
| 138 | ✗ | for (x = 0; x < w; x++) | |
| 139 | ✗ | sample[1][x] = sc->fltmap[remap_index][sample[1][x]]; | |
| 140 | 
        2/2✓ Branch 0 taken 145253900 times. 
          ✓ Branch 1 taken 924200 times. 
         | 
      146178100 | for (x = 0; x < w; x++) | 
| 141 | 145253900 | src[x*pixel_stride + stride * y] = sample[1][x]; | |
| 142 | } else { | ||
| 143 | 808200 | int ret = decode_line(f, sc, gb, w, sample, plane_index, bits, ac); | |
| 144 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 808200 times. 
         | 
      808200 | if (ret < 0) | 
| 145 | ✗ | return ret; | |
| 146 | |||
| 147 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 808200 times. 
         | 
      808200 | if (sc->remap) { | 
| 148 | ✗ | if (f->packed_at_lsb || f->avctx->bits_per_raw_sample == 16) { | |
| 149 | ✗ | for (x = 0; x < w; x++) { | |
| 150 | ✗ | ((uint16_t*)(src + stride*y))[x*pixel_stride] = sc->fltmap[remap_index][sample[1][x] & mask]; | |
| 151 | } | ||
| 152 | } else { | ||
| 153 | ✗ | for (x = 0; x < w; x++) { | |
| 154 | ✗ | int v = sc->fltmap[remap_index][sample[1][x] & mask]; | |
| 155 | ✗ | ((uint16_t*)(src + stride*y))[x*pixel_stride] = v << (16 - f->avctx->bits_per_raw_sample) | v >> (2 * f->avctx->bits_per_raw_sample - 16); | |
| 156 | } | ||
| 157 | } | ||
| 158 | } else { | ||
| 159 | 
        1/4✗ Branch 0 not taken. 
          ✓ Branch 1 taken 808200 times. 
          ✗ Branch 2 not taken. 
          ✗ Branch 3 not taken. 
         | 
      808200 | if (f->packed_at_lsb || f->avctx->bits_per_raw_sample == 16) { | 
| 160 | 
        2/2✓ Branch 0 taken 106856200 times. 
          ✓ Branch 1 taken 808200 times. 
         | 
      107664400 | for (x = 0; x < w; x++) { | 
| 161 | 106856200 | ((uint16_t*)(src + stride*y))[x*pixel_stride] = sample[1][x]; | |
| 162 | } | ||
| 163 | } else { | ||
| 164 | ✗ | for (x = 0; x < w; x++) { | |
| 165 | ✗ | ((uint16_t*)(src + stride*y))[x*pixel_stride] = sample[1][x] << (16 - f->avctx->bits_per_raw_sample) | ((uint16_t **)sample)[1][x] >> (2 * f->avctx->bits_per_raw_sample - 16); | |
| 166 | } | ||
| 167 | } | ||
| 168 | } | ||
| 169 | } | ||
| 170 | } | ||
| 171 | 18300 | return 0; | |
| 172 | } | ||
| 173 | |||
| 174 | 6700 | static int decode_slice_header(const FFV1Context *f, | |
| 175 | FFV1SliceContext *sc, AVFrame *frame) | ||
| 176 | { | ||
| 177 | 6700 | RangeCoder *c = &sc->c; | |
| 178 | uint8_t state[CONTEXT_SIZE]; | ||
| 179 | unsigned ps, context_count; | ||
| 180 | int sx, sy, sw, sh; | ||
| 181 | |||
| 182 | 6700 | memset(state, 128, sizeof(state)); | |
| 183 | 6700 | sx = ff_ffv1_get_symbol(c, state, 0); | |
| 184 | 6700 | sy = ff_ffv1_get_symbol(c, state, 0); | |
| 185 | 6700 | sw = ff_ffv1_get_symbol(c, state, 0) + 1U; | |
| 186 | 6700 | sh = ff_ffv1_get_symbol(c, state, 0) + 1U; | |
| 187 | |||
| 188 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 6700 times. 
         | 
      6700 | av_assert0(f->version > 2); | 
| 189 | |||
| 190 | |||
| 191 | 
        4/8✓ Branch 0 taken 6700 times. 
          ✗ Branch 1 not taken. 
          ✓ Branch 2 taken 6700 times. 
          ✗ Branch 3 not taken. 
          ✓ Branch 4 taken 6700 times. 
          ✗ Branch 5 not taken. 
          ✗ Branch 6 not taken. 
          ✓ Branch 7 taken 6700 times. 
         | 
      6700 | if (sx < 0 || sy < 0 || sw <= 0 || sh <= 0) | 
| 192 | ✗ | return AVERROR_INVALIDDATA; | |
| 193 | 
        2/4✓ Branch 0 taken 6700 times. 
          ✗ Branch 1 not taken. 
          ✗ Branch 2 not taken. 
          ✓ Branch 3 taken 6700 times. 
         | 
      6700 | if (sx > f->num_h_slices - sw || sy > f->num_v_slices - sh) | 
| 194 | ✗ | return AVERROR_INVALIDDATA; | |
| 195 | |||
| 196 | 6700 | sc->slice_x = ff_slice_coord(f, f->width , sx , f->num_h_slices, f->chroma_h_shift); | |
| 197 | 6700 | sc->slice_y = ff_slice_coord(f, f->height, sy , f->num_v_slices, f->chroma_v_shift); | |
| 198 | 6700 | sc->slice_width = ff_slice_coord(f, f->width , sx + sw, f->num_h_slices, f->chroma_h_shift) - sc->slice_x; | |
| 199 | 6700 | sc->slice_height = ff_slice_coord(f, f->height, sy + sh, f->num_v_slices, f->chroma_v_shift) - sc->slice_y; | |
| 200 | |||
| 201 | 
        2/4✓ Branch 0 taken 6700 times. 
          ✗ Branch 1 not taken. 
          ✗ Branch 2 not taken. 
          ✓ Branch 3 taken 6700 times. 
         | 
      6700 | av_assert0((unsigned)sc->slice_width <= f->width && | 
| 202 | (unsigned)sc->slice_height <= f->height); | ||
| 203 | 
        2/4✓ Branch 0 taken 6700 times. 
          ✗ Branch 1 not taken. 
          ✗ Branch 2 not taken. 
          ✓ Branch 3 taken 6700 times. 
         | 
      6700 | av_assert0 ( (unsigned)sc->slice_x + (uint64_t)sc->slice_width <= f->width | 
| 204 | && (unsigned)sc->slice_y + (uint64_t)sc->slice_height <= f->height); | ||
| 205 | |||
| 206 | 
        3/4✓ Branch 0 taken 2700 times. 
          ✓ Branch 1 taken 4000 times. 
          ✗ Branch 2 not taken. 
          ✓ Branch 3 taken 2700 times. 
         | 
      6700 | if (f->ac == AC_GOLOMB_RICE && sc->slice_width >= (1<<23)) | 
| 207 | ✗ | return AVERROR_INVALIDDATA; | |
| 208 | |||
| 209 | 
        2/2✓ Branch 0 taken 13400 times. 
          ✓ Branch 1 taken 6700 times. 
         | 
      20100 | for (unsigned i = 0; i < f->plane_count; i++) { | 
| 210 | 13400 | PlaneContext * const p = &sc->plane[i]; | |
| 211 | 13400 | int idx = ff_ffv1_get_symbol(c, state, 0); | |
| 212 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 13400 times. 
         | 
      13400 | if (idx >= (unsigned)f->quant_table_count) { | 
| 213 | ✗ | av_log(f->avctx, AV_LOG_ERROR, "quant_table_index out of range\n"); | |
| 214 | ✗ | return -1; | |
| 215 | } | ||
| 216 | 13400 | p->quant_table_index = idx; | |
| 217 | 13400 | context_count = f->context_count[idx]; | |
| 218 | |||
| 219 | 
        2/2✓ Branch 0 taken 1344 times. 
          ✓ Branch 1 taken 12056 times. 
         | 
      13400 | if (p->context_count < context_count) { | 
| 220 | 1344 | av_freep(&p->state); | |
| 221 | 1344 | av_freep(&p->vlc_state); | |
| 222 | } | ||
| 223 | 13400 | p->context_count = context_count; | |
| 224 | } | ||
| 225 | |||
| 226 | 6700 | ps = ff_ffv1_get_symbol(c, state, 0); | |
| 227 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 6700 times. 
         | 
      6700 | if (ps == 1) { | 
| 228 | ✗ | frame->flags |= AV_FRAME_FLAG_INTERLACED; | |
| 229 | ✗ | frame->flags |= AV_FRAME_FLAG_TOP_FIELD_FIRST; | |
| 230 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 6700 times. 
         | 
      6700 | } else if (ps == 2) { | 
| 231 | ✗ | frame->flags |= AV_FRAME_FLAG_INTERLACED; | |
| 232 | ✗ | frame->flags &= ~AV_FRAME_FLAG_TOP_FIELD_FIRST; | |
| 233 | 
        1/2✓ Branch 0 taken 6700 times. 
          ✗ Branch 1 not taken. 
         | 
      6700 | } else if (ps == 3) { | 
| 234 | 6700 | frame->flags &= ~AV_FRAME_FLAG_INTERLACED; | |
| 235 | } | ||
| 236 | 6700 | frame->sample_aspect_ratio.num = ff_ffv1_get_symbol(c, state, 0); | |
| 237 | 6700 | frame->sample_aspect_ratio.den = ff_ffv1_get_symbol(c, state, 0); | |
| 238 | |||
| 239 | 
        1/2✗ Branch 1 not taken. 
          ✓ Branch 2 taken 6700 times. 
         | 
      6700 | if (av_image_check_sar(f->width, f->height, | 
| 240 | frame->sample_aspect_ratio) < 0) { | ||
| 241 | ✗ | av_log(f->avctx, AV_LOG_WARNING, "ignoring invalid SAR: %u/%u\n", | |
| 242 | frame->sample_aspect_ratio.num, | ||
| 243 | frame->sample_aspect_ratio.den); | ||
| 244 | ✗ | frame->sample_aspect_ratio = (AVRational){ 0, 1 }; | |
| 245 | } | ||
| 246 | |||
| 247 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 6700 times. 
         | 
      6700 | if (f->version > 3) { | 
| 248 | ✗ | sc->slice_reset_contexts = get_rac(c, state); | |
| 249 | ✗ | sc->slice_coding_mode = ff_ffv1_get_symbol(c, state, 0); | |
| 250 | ✗ | if (sc->slice_coding_mode != 1 && f->colorspace == 1) { | |
| 251 | ✗ | sc->slice_rct_by_coef = ff_ffv1_get_symbol(c, state, 0); | |
| 252 | ✗ | sc->slice_rct_ry_coef = ff_ffv1_get_symbol(c, state, 0); | |
| 253 | ✗ | if ((uint64_t)sc->slice_rct_by_coef + (uint64_t)sc->slice_rct_ry_coef > 4) { | |
| 254 | ✗ | av_log(f->avctx, AV_LOG_ERROR, "slice_rct_y_coef out of range\n"); | |
| 255 | ✗ | return AVERROR_INVALIDDATA; | |
| 256 | } | ||
| 257 | } | ||
| 258 | ✗ | if (f->combined_version >= 0x40004) { | |
| 259 | ✗ | sc->remap = ff_ffv1_get_symbol(c, state, 0); | |
| 260 | ✗ | if (sc->remap > 2U || | |
| 261 | ✗ | sc->remap && !f->flt) { | |
| 262 | ✗ | av_log(f->avctx, AV_LOG_ERROR, "unsupported remap %d\n", sc->remap); | |
| 263 | ✗ | return AVERROR_INVALIDDATA; | |
| 264 | } | ||
| 265 | } | ||
| 266 | } | ||
| 267 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 6700 times. 
         | 
      6700 | if (f->avctx->bits_per_raw_sample == 32) { | 
| 268 | ✗ | if (!sc->remap) { | |
| 269 | ✗ | av_log(f->avctx, AV_LOG_ERROR, "unsupported remap\n"); | |
| 270 | ✗ | return AVERROR_INVALIDDATA; | |
| 271 | } | ||
| 272 | } | ||
| 273 | |||
| 274 | 6700 | return 0; | |
| 275 | } | ||
| 276 | |||
| 277 | ✗ | static void slice_set_damaged(FFV1Context *f, FFV1SliceContext *sc) | |
| 278 | { | ||
| 279 | ✗ | sc->slice_damaged = 1; | |
| 280 | |||
| 281 | // only set this for frame threading, as for slice threading its value is | ||
| 282 | // not used and setting it would be a race | ||
| 283 | ✗ | if (f->avctx->active_thread_type & FF_THREAD_FRAME) | |
| 284 | ✗ | f->frame_damaged = 1; | |
| 285 | ✗ | } | |
| 286 | |||
| 287 | ✗ | static int decode_current_mul(RangeCoder *rc, uint8_t state[32], int *mul, int mul_count, int64_t i) | |
| 288 | { | ||
| 289 | ✗ | int ndx = (i * mul_count) >> 32; | |
| 290 | av_assert2(ndx <= 4096U); | ||
| 291 | |||
| 292 | ✗ | if (mul[ndx] < 0) | |
| 293 | ✗ | mul[ndx] = ff_ffv1_get_symbol(rc, state, 0) & 0x3FFFFFFF; | |
| 294 | |||
| 295 | ✗ | return mul[ndx]; | |
| 296 | } | ||
| 297 | |||
| 298 | ✗ | static int decode_remap(FFV1Context *f, FFV1SliceContext *sc) | |
| 299 | { | ||
| 300 | ✗ | unsigned int end = (1LL<<f->avctx->bits_per_raw_sample) - 1; | |
| 301 | ✗ | int flip = sc->remap == 2 ? (end>>1) : 0; | |
| 302 | ✗ | const int pixel_num = sc->slice_width * sc->slice_height; | |
| 303 | |||
| 304 | ✗ | for (int p= 0; p < 1 + 2*f->chroma_planes + f->transparency; p++) { | |
| 305 | ✗ | int j = 0; | |
| 306 | ✗ | int lu = 0; | |
| 307 | uint8_t state[2][3][32]; | ||
| 308 | int64_t i; | ||
| 309 | int mul[4096+1]; | ||
| 310 | int mul_count; | ||
| 311 | |||
| 312 | ✗ | memset(state, 128, sizeof(state)); | |
| 313 | ✗ | mul_count = ff_ffv1_get_symbol(&sc->c, state[0][0], 0); | |
| 314 | |||
| 315 | ✗ | if (mul_count > 4096U) | |
| 316 | ✗ | return AVERROR_INVALIDDATA; | |
| 317 | ✗ | for (int i = 0; i<mul_count; i++) { | |
| 318 | ✗ | mul[i] = -1; | |
| 319 | |||
| 320 | } | ||
| 321 | ✗ | mul[mul_count] = 1; | |
| 322 | |||
| 323 | ✗ | memset(state, 128, sizeof(state)); | |
| 324 | ✗ | int current_mul = 1; | |
| 325 | ✗ | for (i=0; i <= end ;) { | |
| 326 | ✗ | unsigned run = get_symbol_inline(&sc->c, state[lu][0], 0); | |
| 327 | ✗ | unsigned run0 = lu ? 0 : run; | |
| 328 | ✗ | unsigned run1 = lu ? run : 1; | |
| 329 | |||
| 330 | ✗ | i += run0 * current_mul; | |
| 331 | |||
| 332 | ✗ | while (run1--) { | |
| 333 | ✗ | if (current_mul > 1) { | |
| 334 | ✗ | int delta = get_symbol_inline(&sc->c, state[lu][1], 1); | |
| 335 | ✗ | if (delta <= -current_mul || delta > current_mul/2) | |
| 336 | ✗ | return AVERROR_INVALIDDATA; //not sure we should check this | |
| 337 | ✗ | i += current_mul - 1 + delta; | |
| 338 | } | ||
| 339 | ✗ | if (i - 1 >= end) | |
| 340 | ✗ | break; | |
| 341 | ✗ | if (j >= pixel_num) | |
| 342 | ✗ | return AVERROR_INVALIDDATA; | |
| 343 | ✗ | if (end <= 0xFFFF) { | |
| 344 | ✗ | sc->fltmap [p][j++] = i ^ ((i& 0x8000) ? 0 : flip); | |
| 345 | } else | ||
| 346 | ✗ | sc->fltmap32[p][j++] = i ^ ((i&0x80000000) ? 0 : flip); | |
| 347 | ✗ | i++; | |
| 348 | ✗ | current_mul = decode_current_mul(&sc->c, state[0][2], mul, mul_count, i); | |
| 349 | } | ||
| 350 | ✗ | if (lu) { | |
| 351 | ✗ | i += current_mul; | |
| 352 | } | ||
| 353 | ✗ | lu ^= !run; | |
| 354 | } | ||
| 355 | ✗ | sc->remap_count[p] = j; | |
| 356 | } | ||
| 357 | ✗ | return 0; | |
| 358 | } | ||
| 359 | |||
| 360 | 7700 | static int decode_slice(AVCodecContext *c, void *arg) | |
| 361 | { | ||
| 362 | 7700 | FFV1Context *f = c->priv_data; | |
| 363 | 7700 | FFV1SliceContext *sc = arg; | |
| 364 | int width, height, x, y, ret; | ||
| 365 | 7700 | const int ps = av_pix_fmt_desc_get(f->pix_fmt)->comp[0].step; | |
| 366 | 7700 | AVFrame * const p = f->picture.f; | |
| 367 | 7700 | const int si = sc - f->slices; | |
| 368 | GetBitContext gb; | ||
| 369 | 
        3/4✓ Branch 0 taken 3700 times. 
          ✓ Branch 1 taken 4000 times. 
          ✗ Branch 2 not taken. 
          ✓ Branch 3 taken 3700 times. 
         | 
      7700 | int ac = f->ac || sc->slice_coding_mode == 1; | 
| 370 | |||
| 371 | 
        3/4✓ Branch 0 taken 6928 times. 
          ✓ Branch 1 taken 772 times. 
          ✓ Branch 2 taken 6928 times. 
          ✗ Branch 3 not taken. 
         | 
      7700 | if (!(p->flags & AV_FRAME_FLAG_KEY) && f->last_picture.f) | 
| 372 | 6928 | ff_progress_frame_await(&f->last_picture, si); | |
| 373 | |||
| 374 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 7700 times. 
         | 
      7700 | if (f->slice_damaged[si]) | 
| 375 | ✗ | slice_set_damaged(f, sc); | |
| 376 | |||
| 377 | 7700 | sc->slice_rct_by_coef = 1; | |
| 378 | 7700 | sc->slice_rct_ry_coef = 1; | |
| 379 | |||
| 380 | 
        2/2✓ Branch 0 taken 6700 times. 
          ✓ Branch 1 taken 1000 times. 
         | 
      7700 | if (f->version > 2) { | 
| 381 | 
        1/2✗ Branch 1 not taken. 
          ✓ Branch 2 taken 6700 times. 
         | 
      6700 | if (ff_ffv1_init_slice_state(f, sc) < 0) | 
| 382 | ✗ | return AVERROR(ENOMEM); | |
| 383 | 
        1/2✗ Branch 1 not taken. 
          ✓ Branch 2 taken 6700 times. 
         | 
      6700 | if (decode_slice_header(f, sc, p) < 0) { | 
| 384 | ✗ | sc->slice_x = sc->slice_y = sc->slice_height = sc->slice_width = 0; | |
| 385 | ✗ | slice_set_damaged(f, sc); | |
| 386 | ✗ | return AVERROR_INVALIDDATA; | |
| 387 | } | ||
| 388 | } | ||
| 389 | 
        1/2✗ Branch 1 not taken. 
          ✓ Branch 2 taken 7700 times. 
         | 
      7700 | if ((ret = ff_ffv1_init_slice_state(f, sc)) < 0) | 
| 390 | ✗ | return ret; | |
| 391 | 
        3/4✓ Branch 0 taken 6928 times. 
          ✓ Branch 1 taken 772 times. 
          ✗ Branch 2 not taken. 
          ✓ Branch 3 taken 6928 times. 
         | 
      7700 | if ((p->flags & AV_FRAME_FLAG_KEY) || sc->slice_reset_contexts) { | 
| 392 | 772 | ff_ffv1_clear_slice_state(f, sc); | |
| 393 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 6928 times. 
         | 
      6928 | } else if (sc->slice_damaged) { | 
| 394 | ✗ | return AVERROR_INVALIDDATA; | |
| 395 | } | ||
| 396 | |||
| 397 | 7700 | width = sc->slice_width; | |
| 398 | 7700 | height = sc->slice_height; | |
| 399 | 7700 | x = sc->slice_x; | |
| 400 | 7700 | y = sc->slice_y; | |
| 401 | |||
| 402 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 7700 times. 
         | 
      7700 | if (sc->remap) { | 
| 403 | ✗ | const int pixel_num = sc->slice_width * sc->slice_height; | |
| 404 | |||
| 405 | ✗ | for(int p = 0; p < 1 + 2*f->chroma_planes + f->transparency ; p++) { | |
| 406 | ✗ | if (f->avctx->bits_per_raw_sample == 32) { | |
| 407 | ✗ | av_fast_malloc(&sc->fltmap32[p], &sc->fltmap32_size[p], pixel_num * sizeof(*sc->fltmap32[p])); | |
| 408 | ✗ | if (!sc->fltmap32[p]) | |
| 409 | ✗ | return AVERROR(ENOMEM); | |
| 410 | } else { | ||
| 411 | ✗ | av_fast_malloc(&sc->fltmap[p], &sc->fltmap_size[p], pixel_num * sizeof(*sc->fltmap[p])); | |
| 412 | ✗ | if (!sc->fltmap[p]) | |
| 413 | ✗ | return AVERROR(ENOMEM); | |
| 414 | } | ||
| 415 | } | ||
| 416 | |||
| 417 | ✗ | ret = decode_remap(f, sc); | |
| 418 | ✗ | if (ret < 0) | |
| 419 | ✗ | return ret; | |
| 420 | } | ||
| 421 | |||
| 422 | 
        2/2✓ Branch 0 taken 3700 times. 
          ✓ Branch 1 taken 4000 times. 
         | 
      7700 | if (ac == AC_GOLOMB_RICE) { | 
| 423 | 
        2/2✓ Branch 0 taken 2700 times. 
          ✓ Branch 1 taken 1000 times. 
         | 
      3700 | if (f->combined_version >= 0x30002) | 
| 424 | 2700 | get_rac(&sc->c, (uint8_t[]) { 129 }); | |
| 425 | 
        6/6✓ Branch 0 taken 1000 times. 
          ✓ Branch 1 taken 2700 times. 
          ✓ Branch 2 taken 600 times. 
          ✓ Branch 3 taken 400 times. 
          ✓ Branch 4 taken 400 times. 
          ✓ Branch 5 taken 200 times. 
         | 
      3700 | sc->ac_byte_count = f->version > 2 || (!x && !y) ? sc->c.bytestream - sc->c.bytestream_start - 1 : 0; | 
| 426 | 3700 | init_get_bits(&gb, | |
| 427 | 3700 | sc->c.bytestream_start + sc->ac_byte_count, | |
| 428 | 3700 | (sc->c.bytestream_end - sc->c.bytestream_start - sc->ac_byte_count) * 8); | |
| 429 | } | ||
| 430 | |||
| 431 | av_assert1(width && height); | ||
| 432 | 
        3/6✓ Branch 0 taken 6100 times. 
          ✓ Branch 1 taken 1600 times. 
          ✗ Branch 2 not taken. 
          ✓ Branch 3 taken 6100 times. 
          ✗ Branch 4 not taken. 
          ✗ Branch 5 not taken. 
         | 
      13800 | if (f->colorspace == 0 && (f->chroma_planes || !f->transparency)) { | 
| 433 | 6100 | const int chroma_width = AV_CEIL_RSHIFT(width, f->chroma_h_shift); | |
| 434 | 6100 | const int chroma_height = AV_CEIL_RSHIFT(height, f->chroma_v_shift); | |
| 435 | 6100 | const int cx = x >> f->chroma_h_shift; | |
| 436 | 6100 | const int cy = y >> f->chroma_v_shift; | |
| 437 | 6100 | decode_plane(f, sc, &gb, p->data[0] + ps*x + y*p->linesize[0], width, height, p->linesize[0], 0, 0, 1, ac); | |
| 438 | |||
| 439 | 
        1/2✓ Branch 0 taken 6100 times. 
          ✗ Branch 1 not taken. 
         | 
      6100 | if (f->chroma_planes) { | 
| 440 | 6100 | decode_plane(f, sc, &gb, p->data[1] + ps*cx+cy*p->linesize[1], chroma_width, chroma_height, p->linesize[1], 1, 1, 1, ac); | |
| 441 | 6100 | decode_plane(f, sc, &gb, p->data[2] + ps*cx+cy*p->linesize[2], chroma_width, chroma_height, p->linesize[2], 1, 2, 1, ac); | |
| 442 | } | ||
| 443 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 6100 times. 
         | 
      6100 | if (f->transparency) | 
| 444 | ✗ | decode_plane(f, sc, &gb, p->data[3] + ps*x + y*p->linesize[3], width, height, p->linesize[3], (f->version >= 4 && !f->chroma_planes) ? 1 : 2, | |
| 445 | ✗ | (f->version >= 4 && !f->chroma_planes) ? 1 : 3, 1, ac); | |
| 446 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 1600 times. 
         | 
      1600 | } else if (f->colorspace == 0) { | 
| 447 | ✗ | decode_plane(f, sc, &gb, p->data[0] + ps*x + y*p->linesize[0] , width, height, p->linesize[0], 0, 0, 2, ac); | |
| 448 | ✗ | decode_plane(f, sc, &gb, p->data[0] + ps*x + y*p->linesize[0] + (ps>>1), width, height, p->linesize[0], 1, 1, 2, ac); | |
| 449 | 
        2/2✓ Branch 0 taken 800 times. 
          ✓ Branch 1 taken 800 times. 
         | 
      1600 | } else if (f->use32bit) { | 
| 450 | 800 | uint8_t *planes[4] = { p->data[0] + ps * x + y * p->linesize[0], | |
| 451 | 800 | p->data[1] + ps * x + y * p->linesize[1], | |
| 452 | 800 | p->data[2] + ps * x + y * p->linesize[2] }; | |
| 453 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 800 times. 
         | 
      800 | if (f->transparency) | 
| 454 | ✗ | planes[3] = p->data[3] + ps * x + y * p->linesize[3]; | |
| 455 | 800 | decode_rgb_frame32(f, sc, &gb, planes, width, height, p->linesize); | |
| 456 | } else { | ||
| 457 | 800 | uint8_t *planes[4] = { p->data[0] + ps * x + y * p->linesize[0] }; | |
| 458 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 800 times. 
         | 
      800 | if (f->avctx->bits_per_raw_sample > 8) { | 
| 459 | ✗ | planes[1] = p->data[1] + ps * x + y * p->linesize[1]; | |
| 460 | ✗ | planes[2] = p->data[2] + ps * x + y * p->linesize[2]; | |
| 461 | ✗ | if (f->transparency) | |
| 462 | ✗ | planes[3] = p->data[3] + ps * x + y * p->linesize[3]; | |
| 463 | } | ||
| 464 | 800 | decode_rgb_frame(f, sc, &gb, planes, width, height, p->linesize); | |
| 465 | } | ||
| 466 | 
        3/4✓ Branch 0 taken 4000 times. 
          ✓ Branch 1 taken 3700 times. 
          ✓ Branch 2 taken 4000 times. 
          ✗ Branch 3 not taken. 
         | 
      7700 | if (ac != AC_GOLOMB_RICE && f->version > 2) { | 
| 467 | int v; | ||
| 468 | 4000 | get_rac(&sc->c, (uint8_t[]) { 129 }); | |
| 469 | 
        1/2✓ Branch 0 taken 4000 times. 
          ✗ Branch 1 not taken. 
         | 
      4000 | v = sc->c.bytestream_end - sc->c.bytestream - 2 - 5*!!f->ec; | 
| 470 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 4000 times. 
         | 
      4000 | if (v) { | 
| 471 | ✗ | av_log(f->avctx, AV_LOG_ERROR, "bytestream end mismatching by %d\n", v); | |
| 472 | ✗ | slice_set_damaged(f, sc); | |
| 473 | } | ||
| 474 | } | ||
| 475 | |||
| 476 | 
        1/4✗ Branch 0 not taken. 
          ✓ Branch 1 taken 7700 times. 
          ✗ Branch 2 not taken. 
          ✗ Branch 3 not taken. 
         | 
      7700 | if (sc->slice_damaged && (f->avctx->err_recognition & AV_EF_EXPLODE)) | 
| 477 | ✗ | return AVERROR_INVALIDDATA; | |
| 478 | |||
| 479 | 
        1/4✗ Branch 0 not taken. 
          ✓ Branch 1 taken 7700 times. 
          ✗ Branch 2 not taken. 
          ✗ Branch 3 not taken. 
         | 
      7700 | if ((c->active_thread_type & FF_THREAD_FRAME) && !f->frame_damaged) | 
| 480 | ✗ | ff_progress_frame_report(&f->picture, si); | |
| 481 | |||
| 482 | 7700 | return 0; | |
| 483 | } | ||
| 484 | |||
| 485 | 87 | static enum AVPixelFormat get_pixel_format(FFV1Context *f) | |
| 486 | { | ||
| 487 | 87 | enum AVPixelFormat pix_fmts[] = { | |
| 488 | #if CONFIG_FFV1_VULKAN_HWACCEL | ||
| 489 | AV_PIX_FMT_VULKAN, | ||
| 490 | #endif | ||
| 491 | 87 | f->pix_fmt, | |
| 492 | AV_PIX_FMT_NONE, | ||
| 493 | }; | ||
| 494 | |||
| 495 | 87 | return ff_get_format(f->avctx, pix_fmts); | |
| 496 | } | ||
| 497 | |||
| 498 | 253 | static int read_header(FFV1Context *f, RangeCoder *c) | |
| 499 | { | ||
| 500 | uint8_t state[CONTEXT_SIZE]; | ||
| 501 | 253 | int context_count = -1; //-1 to avoid warning | |
| 502 | int ret; | ||
| 503 | |||
| 504 | 253 | memset(state, 128, sizeof(state)); | |
| 505 | |||
| 506 | 253 | ret = ff_ffv1_parse_header(f, c, state); | |
| 507 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 253 times. 
         | 
      253 | if (ret < 0) | 
| 508 | ✗ | return ret; | |
| 509 | |||
| 510 | 
        2/2✓ Branch 0 taken 87 times. 
          ✓ Branch 1 taken 166 times. 
         | 
      253 | if (f->configured_pix_fmt != f->pix_fmt) { | 
| 511 | 87 | f->avctx->pix_fmt = get_pixel_format(f); | |
| 512 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 87 times. 
         | 
      87 | if (f->avctx->pix_fmt < 0) | 
| 513 | ✗ | return AVERROR(EINVAL); | |
| 514 | 87 | f->configured_pix_fmt = f->pix_fmt; | |
| 515 | } | ||
| 516 | |||
| 517 | ff_dlog(f->avctx, "%d %d %d\n", | ||
| 518 | f->chroma_h_shift, f->chroma_v_shift, f->pix_fmt); | ||
| 519 | 
        2/2✓ Branch 0 taken 24 times. 
          ✓ Branch 1 taken 229 times. 
         | 
      253 | if (f->version < 2) { | 
| 520 | 24 | context_count = ff_ffv1_read_quant_tables(c, f->quant_tables[0]); | |
| 521 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 24 times. 
         | 
      24 | if (context_count < 0) { | 
| 522 | ✗ | av_log(f->avctx, AV_LOG_ERROR, "read_quant_table error\n"); | |
| 523 | ✗ | return AVERROR_INVALIDDATA; | |
| 524 | } | ||
| 525 | 24 | f->slice_count = f->max_slice_count; | |
| 526 | 
        2/2✓ Branch 0 taken 24 times. 
          ✓ Branch 1 taken 205 times. 
         | 
      229 | } else if (f->version < 3) { | 
| 527 | 24 | f->slice_count = ff_ffv1_get_symbol(c, state, 0); | |
| 528 | } else { | ||
| 529 | 205 | const uint8_t *p = c->bytestream_end; | |
| 530 | 205 | for (f->slice_count = 0; | |
| 531 | 
        4/6✓ Branch 0 taken 1035 times. 
          ✗ Branch 1 not taken. 
          ✓ Branch 2 taken 1035 times. 
          ✗ Branch 3 not taken. 
          ✓ Branch 4 taken 830 times. 
          ✓ Branch 5 taken 205 times. 
         | 
      1035 | f->slice_count < MAX_SLICES && 3 + 5*!!f->ec < p - c->bytestream_start; | 
| 532 | 830 | f->slice_count++) { | |
| 533 | 
        1/2✓ Branch 0 taken 830 times. 
          ✗ Branch 1 not taken. 
         | 
      830 | int trailer = 3 + 5*!!f->ec; | 
| 534 | 830 | int size = AV_RB24(p-trailer); | |
| 535 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 830 times. 
         | 
      830 | if (size + trailer > p - c->bytestream_start) | 
| 536 | ✗ | break; | |
| 537 | 830 | p -= size + trailer; | |
| 538 | } | ||
| 539 | } | ||
| 540 | 
        3/6✓ Branch 0 taken 253 times. 
          ✗ Branch 1 not taken. 
          ✓ Branch 2 taken 253 times. 
          ✗ Branch 3 not taken. 
          ✗ Branch 4 not taken. 
          ✓ Branch 5 taken 253 times. 
         | 
      253 | if (f->slice_count > (unsigned)MAX_SLICES || f->slice_count <= 0 || f->slice_count > f->max_slice_count) { | 
| 541 | ✗ | av_log(f->avctx, AV_LOG_ERROR, "slice count %d is invalid (max=%d)\n", f->slice_count, f->max_slice_count); | |
| 542 | ✗ | return AVERROR_INVALIDDATA; | |
| 543 | } | ||
| 544 | |||
| 545 | 253 | av_refstruct_unref(&f->slice_damaged); | |
| 546 | 253 | f->slice_damaged = av_refstruct_allocz(f->slice_count * sizeof(*f->slice_damaged)); | |
| 547 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 253 times. 
         | 
      253 | if (!f->slice_damaged) | 
| 548 | ✗ | return AVERROR(ENOMEM); | |
| 549 | |||
| 550 | 
        2/2✓ Branch 0 taken 950 times. 
          ✓ Branch 1 taken 253 times. 
         | 
      1203 | for (int j = 0; j < f->slice_count; j++) { | 
| 551 | 950 | FFV1SliceContext *sc = &f->slices[j]; | |
| 552 | |||
| 553 | 
        2/2✓ Branch 0 taken 96 times. 
          ✓ Branch 1 taken 854 times. 
         | 
      950 | if (f->version == 2) { | 
| 554 | 96 | int sx = ff_ffv1_get_symbol(c, state, 0); | |
| 555 | 96 | int sy = ff_ffv1_get_symbol(c, state, 0); | |
| 556 | 96 | int sw = ff_ffv1_get_symbol(c, state, 0) + 1U; | |
| 557 | 96 | int sh = ff_ffv1_get_symbol(c, state, 0) + 1U; | |
| 558 | |||
| 559 | 
        4/8✓ Branch 0 taken 96 times. 
          ✗ Branch 1 not taken. 
          ✓ Branch 2 taken 96 times. 
          ✗ Branch 3 not taken. 
          ✓ Branch 4 taken 96 times. 
          ✗ Branch 5 not taken. 
          ✗ Branch 6 not taken. 
          ✓ Branch 7 taken 96 times. 
         | 
      96 | if (sx < 0 || sy < 0 || sw <= 0 || sh <= 0) | 
| 560 | ✗ | return AVERROR_INVALIDDATA; | |
| 561 | 
        2/4✓ Branch 0 taken 96 times. 
          ✗ Branch 1 not taken. 
          ✗ Branch 2 not taken. 
          ✓ Branch 3 taken 96 times. 
         | 
      96 | if (sx > f->num_h_slices - sw || sy > f->num_v_slices - sh) | 
| 562 | ✗ | return AVERROR_INVALIDDATA; | |
| 563 | |||
| 564 | 96 | sc->slice_x = sx * (int64_t)f->width / f->num_h_slices; | |
| 565 | 96 | sc->slice_y = sy * (int64_t)f->height / f->num_v_slices; | |
| 566 | 96 | sc->slice_width = (sx + sw) * (int64_t)f->width / f->num_h_slices - sc->slice_x; | |
| 567 | 96 | sc->slice_height = (sy + sh) * (int64_t)f->height / f->num_v_slices - sc->slice_y; | |
| 568 | |||
| 569 | 
        2/4✓ Branch 0 taken 96 times. 
          ✗ Branch 1 not taken. 
          ✗ Branch 2 not taken. 
          ✓ Branch 3 taken 96 times. 
         | 
      96 | av_assert0((unsigned)sc->slice_width <= f->width && | 
| 570 | (unsigned)sc->slice_height <= f->height); | ||
| 571 | 
        2/4✓ Branch 0 taken 96 times. 
          ✗ Branch 1 not taken. 
          ✗ Branch 2 not taken. 
          ✓ Branch 3 taken 96 times. 
         | 
      96 | av_assert0 ( (unsigned)sc->slice_x + (uint64_t)sc->slice_width <= f->width | 
| 572 | && (unsigned)sc->slice_y + (uint64_t)sc->slice_height <= f->height); | ||
| 573 | } | ||
| 574 | |||
| 575 | 950 | av_refstruct_unref(&sc->plane); | |
| 576 | 950 | sc->plane = ff_ffv1_planes_alloc(); | |
| 577 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 950 times. 
         | 
      950 | if (!sc->plane) | 
| 578 | ✗ | return AVERROR(ENOMEM); | |
| 579 | |||
| 580 | 
        2/2✓ Branch 0 taken 1900 times. 
          ✓ Branch 1 taken 950 times. 
         | 
      2850 | for (int i = 0; i < f->plane_count; i++) { | 
| 581 | 1900 | PlaneContext *const p = &sc->plane[i]; | |
| 582 | |||
| 583 | 
        2/2✓ Branch 0 taken 192 times. 
          ✓ Branch 1 taken 1708 times. 
         | 
      1900 | if (f->version == 2) { | 
| 584 | 192 | int idx = ff_ffv1_get_symbol(c, state, 0); | |
| 585 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 192 times. 
         | 
      192 | if (idx >= (unsigned)f->quant_table_count) { | 
| 586 | ✗ | av_log(f->avctx, AV_LOG_ERROR, | |
| 587 | "quant_table_index out of range\n"); | ||
| 588 | ✗ | return AVERROR_INVALIDDATA; | |
| 589 | } | ||
| 590 | 192 | p->quant_table_index = idx; | |
| 591 | 192 | context_count = f->context_count[idx]; | |
| 592 | } | ||
| 593 | |||
| 594 | 
        2/2✓ Branch 0 taken 240 times. 
          ✓ Branch 1 taken 1660 times. 
         | 
      1900 | if (f->version <= 2) { | 
| 595 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 240 times. 
         | 
      240 | av_assert0(context_count >= 0); | 
| 596 | 240 | p->context_count = context_count; | |
| 597 | } | ||
| 598 | } | ||
| 599 | } | ||
| 600 | 253 | return 0; | |
| 601 | } | ||
| 602 | |||
| 603 | 88 | static av_cold int decode_init(AVCodecContext *avctx) | |
| 604 | { | ||
| 605 | 88 | FFV1Context *f = avctx->priv_data; | |
| 606 | int ret; | ||
| 607 | |||
| 608 | 88 | f->pix_fmt = AV_PIX_FMT_NONE; | |
| 609 | 88 | f->configured_pix_fmt = AV_PIX_FMT_NONE; | |
| 610 | |||
| 611 | 
        1/2✗ Branch 1 not taken. 
          ✓ Branch 2 taken 88 times. 
         | 
      88 | if ((ret = ff_ffv1_common_init(avctx, f)) < 0) | 
| 612 | ✗ | return ret; | |
| 613 | |||
| 614 | 
        3/4✓ Branch 0 taken 80 times. 
          ✓ Branch 1 taken 8 times. 
          ✗ Branch 3 not taken. 
          ✓ Branch 4 taken 80 times. 
         | 
      88 | if (avctx->extradata_size > 0 && (ret = ff_ffv1_read_extra_header(f)) < 0) | 
| 615 | ✗ | return ret; | |
| 616 | |||
| 617 | 
        1/2✗ Branch 1 not taken. 
          ✓ Branch 2 taken 88 times. 
         | 
      88 | if ((ret = ff_ffv1_init_slice_contexts(f)) < 0) | 
| 618 | ✗ | return ret; | |
| 619 | |||
| 620 | 88 | return 0; | |
| 621 | } | ||
| 622 | |||
| 623 | 7700 | static int find_next_slice(AVCodecContext *avctx, | |
| 624 | uint8_t *buf, uint8_t *buf_end, int idx, | ||
| 625 | uint8_t **pos, uint32_t *len) | ||
| 626 | { | ||
| 627 | 7700 | FFV1Context *f = avctx->priv_data; | |
| 628 | |||
| 629 | /* Length field */ | ||
| 630 | 7700 | uint32_t v = buf_end - buf; | |
| 631 | 
        4/4✓ Branch 0 taken 2075 times. 
          ✓ Branch 1 taken 5625 times. 
          ✓ Branch 2 taken 1675 times. 
          ✓ Branch 3 taken 400 times. 
         | 
      7700 | if (idx || f->version > 2) { | 
| 632 | /* Three bytes of length, plus flush bit + CRC */ | ||
| 633 | 
        2/2✓ Branch 0 taken 6700 times. 
          ✓ Branch 1 taken 600 times. 
         | 
      7300 | uint32_t trailer = 3 + 5*!!f->ec; | 
| 634 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 7300 times. 
         | 
      7300 | if (trailer > buf_end - buf) | 
| 635 | ✗ | v = INT_MAX; | |
| 636 | else | ||
| 637 | 7300 | v = AV_RB24(buf_end - trailer) + trailer; | |
| 638 | } | ||
| 639 | |||
| 640 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 7700 times. 
         | 
      7700 | if (buf_end - buf < v) { | 
| 641 | ✗ | av_log(avctx, AV_LOG_ERROR, "Slice pointer chain broken\n"); | |
| 642 | ✗ | ff_progress_frame_report(&f->picture, INT_MAX); | |
| 643 | ✗ | return AVERROR_INVALIDDATA; | |
| 644 | } | ||
| 645 | |||
| 646 | 7700 | *len = v; | |
| 647 | 
        2/2✓ Branch 0 taken 5625 times. 
          ✓ Branch 1 taken 2075 times. 
         | 
      7700 | if (idx) | 
| 648 | 5625 | *pos = buf_end - v; | |
| 649 | else | ||
| 650 | 2075 | *pos = buf; | |
| 651 | |||
| 652 | 7700 | return 0; | |
| 653 | } | ||
| 654 | |||
| 655 | 2120 | static int decode_header(AVCodecContext *avctx, RangeCoder *c, | |
| 656 | uint8_t *buf, size_t buf_size) | ||
| 657 | { | ||
| 658 | int ret; | ||
| 659 | 2120 | FFV1Context *f = avctx->priv_data; | |
| 660 | |||
| 661 | 2120 | uint8_t keystate = 128; | |
| 662 | 2120 | ff_init_range_decoder(c, buf, buf_size); | |
| 663 | 2120 | ff_build_rac_states(c, 0.05 * (1LL << 32), 256 - 8); | |
| 664 | |||
| 665 | 
        2/2✓ Branch 1 taken 253 times. 
          ✓ Branch 2 taken 1867 times. 
         | 
      2120 | if (get_rac(c, &keystate)) { | 
| 666 | 253 | f->key_frame = AV_FRAME_FLAG_KEY; | |
| 667 | 253 | f->key_frame_ok = 0; | |
| 668 | 
        1/2✗ Branch 1 not taken. 
          ✓ Branch 2 taken 253 times. 
         | 
      253 | if ((ret = read_header(f, c)) < 0) | 
| 669 | ✗ | return ret; | |
| 670 | 253 | f->key_frame_ok = 1; | |
| 671 | } else { | ||
| 672 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 1867 times. 
         | 
      1867 | if (!f->key_frame_ok) { | 
| 673 | ✗ | av_log(avctx, AV_LOG_ERROR, | |
| 674 | "Cannot decode non-keyframe without valid keyframe\n"); | ||
| 675 | ✗ | return AVERROR_INVALIDDATA; | |
| 676 | } | ||
| 677 | 1867 | f->key_frame = 0; | |
| 678 | } | ||
| 679 | |||
| 680 | 
        2/2✓ Branch 0 taken 1022 times. 
          ✓ Branch 1 taken 1098 times. 
         | 
      2120 | if (f->ac != AC_GOLOMB_RICE) { | 
| 681 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 1022 times. 
         | 
      1022 | if (buf_size < avctx->width * avctx->height / (128*8)) | 
| 682 | ✗ | return AVERROR_INVALIDDATA; | |
| 683 | } else { | ||
| 684 | 1098 | int w = avctx->width; | |
| 685 | 1098 | int s = 1 + w / (1<<23); | |
| 686 | int i; | ||
| 687 | |||
| 688 | 1098 | w /= s; | |
| 689 | |||
| 690 | 
        2/2✓ Branch 0 taken 25232 times. 
          ✓ Branch 1 taken 1098 times. 
         | 
      26330 | for (i = 0; w > (1<<ff_log2_run[i]); i++) | 
| 691 | 25232 | w -= ff_log2_run[i]; | |
| 692 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 1098 times. 
         | 
      1098 | if (buf_size < (avctx->height + i + 6) / 8 * s) | 
| 693 | ✗ | return AVERROR_INVALIDDATA; | |
| 694 | } | ||
| 695 | |||
| 696 | 2120 | return 0; | |
| 697 | } | ||
| 698 | |||
| 699 | 2075 | static int decode_slices(AVCodecContext *avctx, RangeCoder c, | |
| 700 | AVPacket *avpkt) | ||
| 701 | { | ||
| 702 | 2075 | FFV1Context *f = avctx->priv_data; | |
| 703 | 2075 | AVFrame *p = f->picture.f; | |
| 704 | |||
| 705 | 2075 | uint8_t *buf = avpkt->data; | |
| 706 | 2075 | size_t buf_size = avpkt->size; | |
| 707 | 2075 | uint8_t *buf_end = buf + buf_size; | |
| 708 | |||
| 709 | 
        2/2✓ Branch 0 taken 7700 times. 
          ✓ Branch 1 taken 2075 times. 
         | 
      9775 | for (int i = f->slice_count - 1; i >= 0; i--) { | 
| 710 | 7700 | FFV1SliceContext *sc = &f->slices[i]; | |
| 711 | |||
| 712 | uint8_t *pos; | ||
| 713 | uint32_t len; | ||
| 714 | 7700 | int err = find_next_slice(avctx, buf, buf_end, i, | |
| 715 | &pos, &len); | ||
| 716 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 7700 times. 
         | 
      7700 | if (err < 0) | 
| 717 | ✗ | return err; | |
| 718 | |||
| 719 | 7700 | buf_end -= len; | |
| 720 | |||
| 721 | 7700 | sc->slice_damaged = 0; | |
| 722 | |||
| 723 | 
        2/2✓ Branch 0 taken 6700 times. 
          ✓ Branch 1 taken 1000 times. 
         | 
      7700 | if (f->ec) { | 
| 724 | 6700 | unsigned crc = av_crc(av_crc_get_table(AV_CRC_32_IEEE), f->crcref, pos, len); | |
| 725 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 6700 times. 
         | 
      6700 | if (crc != f->crcref) { | 
| 726 | ✗ | int64_t ts = avpkt->pts != AV_NOPTS_VALUE ? avpkt->pts : avpkt->dts; | |
| 727 | ✗ | av_log(f->avctx, AV_LOG_ERROR, "slice CRC mismatch %X!", crc); | |
| 728 | ✗ | if (ts != AV_NOPTS_VALUE && avctx->pkt_timebase.num) { | |
| 729 | ✗ | av_log(f->avctx, AV_LOG_ERROR, "at %f seconds\n", ts*av_q2d(avctx->pkt_timebase)); | |
| 730 | ✗ | } else if (ts != AV_NOPTS_VALUE) { | |
| 731 | ✗ | av_log(f->avctx, AV_LOG_ERROR, "at %"PRId64"\n", ts); | |
| 732 | } else { | ||
| 733 | ✗ | av_log(f->avctx, AV_LOG_ERROR, "\n"); | |
| 734 | } | ||
| 735 | ✗ | slice_set_damaged(f, sc); | |
| 736 | } | ||
| 737 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 6700 times. 
         | 
      6700 | if (avctx->debug & FF_DEBUG_PICT_INFO) { | 
| 738 | ✗ | av_log(avctx, AV_LOG_DEBUG, "slice %d, CRC: 0x%08"PRIX32"\n", i, AV_RB32(pos + len - 4)); | |
| 739 | } | ||
| 740 | } | ||
| 741 | |||
| 742 | 
        2/2✓ Branch 0 taken 5625 times. 
          ✓ Branch 1 taken 2075 times. 
         | 
      7700 | if (i) { | 
| 743 | 5625 | ff_init_range_decoder(&sc->c, pos, len); | |
| 744 | 5625 | ff_build_rac_states(&sc->c, 0.05 * (1LL << 32), 256 - 8); | |
| 745 | } else { | ||
| 746 | 2075 | sc->c = c; | |
| 747 | 2075 | sc->c.bytestream_end = pos + len; | |
| 748 | } | ||
| 749 | } | ||
| 750 | |||
| 751 | 2075 | avctx->execute(avctx, | |
| 752 | decode_slice, | ||
| 753 | 2075 | f->slices, | |
| 754 | NULL, | ||
| 755 | f->slice_count, | ||
| 756 | sizeof(*f->slices)); | ||
| 757 | |||
| 758 | 
        2/2✓ Branch 0 taken 7700 times. 
          ✓ Branch 1 taken 2075 times. 
         | 
      9775 | for (int i = f->slice_count - 1; i >= 0; i--) { | 
| 759 | 7700 | FFV1SliceContext *sc = &f->slices[i]; | |
| 760 | 
        1/4✗ Branch 0 not taken. 
          ✓ Branch 1 taken 7700 times. 
          ✗ Branch 2 not taken. 
          ✗ Branch 3 not taken. 
         | 
      7700 | if (sc->slice_damaged && f->last_picture.f) { | 
| 761 | ✗ | const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(f->pix_fmt); | |
| 762 | const uint8_t *src[4]; | ||
| 763 | uint8_t *dst[4]; | ||
| 764 | ✗ | ff_progress_frame_await(&f->last_picture, INT_MAX); | |
| 765 | ✗ | for (int j = 0; j < desc->nb_components; j++) { | |
| 766 | ✗ | int pixshift = desc->comp[j].depth > 8; | |
| 767 | ✗ | int sh = (j == 1 || j == 2) ? f->chroma_h_shift : 0; | |
| 768 | ✗ | int sv = (j == 1 || j == 2) ? f->chroma_v_shift : 0; | |
| 769 | ✗ | dst[j] = p->data[j] + p->linesize[j] * | |
| 770 | ✗ | (sc->slice_y >> sv) + ((sc->slice_x >> sh) << pixshift); | |
| 771 | ✗ | src[j] = f->last_picture.f->data[j] + f->last_picture.f->linesize[j] * | |
| 772 | ✗ | (sc->slice_y >> sv) + ((sc->slice_x >> sh) << pixshift); | |
| 773 | |||
| 774 | } | ||
| 775 | |||
| 776 | ✗ | av_image_copy(dst, p->linesize, src, | |
| 777 | ✗ | f->last_picture.f->linesize, | |
| 778 | f->pix_fmt, | ||
| 779 | sc->slice_width, | ||
| 780 | sc->slice_height); | ||
| 781 | |||
| 782 | ✗ | f->slice_damaged[i] = 1; | |
| 783 | } | ||
| 784 | } | ||
| 785 | |||
| 786 | 2075 | return 0; | |
| 787 | } | ||
| 788 | |||
| 789 | 2120 | static int decode_frame(AVCodecContext *avctx, AVFrame *rframe, | |
| 790 | int *got_frame, AVPacket *avpkt) | ||
| 791 | { | ||
| 792 | 2120 | FFV1Context *f = avctx->priv_data; | |
| 793 | int ret; | ||
| 794 | AVFrame *p; | ||
| 795 | 2120 | const FFHWAccel *hwaccel = NULL; | |
| 796 | |||
| 797 | /* This is copied onto the first slice's range coder context */ | ||
| 798 | RangeCoder c; | ||
| 799 | |||
| 800 | 2120 | ff_progress_frame_unref(&f->last_picture); | |
| 801 | 2120 | av_refstruct_unref(&f->hwaccel_last_picture_private); | |
| 802 | |||
| 803 | 2120 | FFSWAP(ProgressFrame, f->picture, f->last_picture); | |
| 804 | 2120 | FFSWAP(void *, f->hwaccel_picture_private, f->hwaccel_last_picture_private); | |
| 805 | |||
| 806 | 2120 | f->avctx = avctx; | |
| 807 | 2120 | f->frame_damaged = 0; | |
| 808 | |||
| 809 | 2120 | ret = decode_header(avctx, &c, avpkt->data, avpkt->size); | |
| 810 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 2120 times. 
         | 
      2120 | if (ret < 0) | 
| 811 | ✗ | return ret; | |
| 812 | |||
| 813 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 2120 times. 
         | 
      2120 | if (avctx->debug & FF_DEBUG_PICT_INFO) | 
| 814 | ✗ | av_log(avctx, AV_LOG_DEBUG, "ver:%d keyframe:%d coder:%d ec:%d slices:%d bps:%d\n", | |
| 815 | ✗ | f->version, !!f->key_frame, f->ac, f->ec, f->slice_count, f->avctx->bits_per_raw_sample); | |
| 816 | |||
| 817 | 
        2/2✓ Branch 0 taken 45 times. 
          ✓ Branch 1 taken 2075 times. 
         | 
      2120 | if (avctx->skip_frame >= AVDISCARD_ALL) | 
| 818 | 45 | return avpkt->size; | |
| 819 | |||
| 820 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 2075 times. 
         | 
      2075 | if (avctx->hwaccel) | 
| 821 | ✗ | hwaccel = ffhwaccel(avctx->hwaccel); | |
| 822 | |||
| 823 | 2075 | ret = ff_progress_frame_get_buffer(avctx, &f->picture, | |
| 824 | AV_GET_BUFFER_FLAG_REF); | ||
| 825 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 2075 times. 
         | 
      2075 | if (ret < 0) | 
| 826 | ✗ | return ret; | |
| 827 | |||
| 828 | 2075 | ret = ff_hwaccel_frame_priv_alloc(avctx, &f->hwaccel_picture_private); | |
| 829 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 2075 times. 
         | 
      2075 | if (ret < 0) | 
| 830 | ✗ | return ret; | |
| 831 | |||
| 832 | 2075 | p = f->picture.f; | |
| 833 | |||
| 834 | 2075 | p->pict_type = AV_PICTURE_TYPE_I; //FIXME I vs. P | |
| 835 | 2075 | p->flags = (p->flags & ~AV_FRAME_FLAG_KEY) | f->key_frame; | |
| 836 | |||
| 837 | 
        3/4✓ Branch 0 taken 400 times. 
          ✓ Branch 1 taken 1675 times. 
          ✗ Branch 2 not taken. 
          ✓ Branch 3 taken 400 times. 
         | 
      2075 | if (f->version < 3 && avctx->field_order > AV_FIELD_PROGRESSIVE) { | 
| 838 | /* we have interlaced material flagged in container */ | ||
| 839 | ✗ | p->flags |= AV_FRAME_FLAG_INTERLACED; | |
| 840 | ✗ | if (avctx->field_order == AV_FIELD_TT || avctx->field_order == AV_FIELD_TB) | |
| 841 | ✗ | p->flags |= AV_FRAME_FLAG_TOP_FIELD_FIRST; | |
| 842 | } | ||
| 843 | |||
| 844 | /* Start */ | ||
| 845 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 2075 times. 
         | 
      2075 | if (hwaccel) { | 
| 846 | ✗ | ret = hwaccel->start_frame(avctx, avpkt->buf, avpkt->data, avpkt->size); | |
| 847 | ✗ | if (ret < 0) | |
| 848 | ✗ | return ret; | |
| 849 | } | ||
| 850 | |||
| 851 | 2075 | ff_thread_finish_setup(avctx); | |
| 852 | |||
| 853 | /* Decode slices */ | ||
| 854 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 2075 times. 
         | 
      2075 | if (hwaccel) { | 
| 855 | ✗ | uint8_t *buf_end = avpkt->data + avpkt->size; | |
| 856 | |||
| 857 | ✗ | if (!(p->flags & AV_FRAME_FLAG_KEY) && f->last_picture.f) | |
| 858 | ✗ | ff_progress_frame_await(&f->last_picture, f->slice_count - 1); | |
| 859 | |||
| 860 | ✗ | for (int i = f->slice_count - 1; i >= 0; i--) { | |
| 861 | uint8_t *pos; | ||
| 862 | uint32_t len; | ||
| 863 | ✗ | ret = find_next_slice(avctx, avpkt->data, buf_end, i, | |
| 864 | &pos, &len); | ||
| 865 | ✗ | if (ret < 0) | |
| 866 | ✗ | return ret; | |
| 867 | |||
| 868 | ✗ | buf_end -= len; | |
| 869 | |||
| 870 | ✗ | ret = hwaccel->decode_slice(avctx, pos, len); | |
| 871 | ✗ | if (ret < 0) | |
| 872 | ✗ | return ret; | |
| 873 | } | ||
| 874 | } else { | ||
| 875 | 2075 | ret = decode_slices(avctx, c, avpkt); | |
| 876 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 2075 times. 
         | 
      2075 | if (ret < 0) | 
| 877 | ✗ | return ret; | |
| 878 | } | ||
| 879 | |||
| 880 | /* Finalize */ | ||
| 881 | 
        1/2✗ Branch 0 not taken. 
          ✓ Branch 1 taken 2075 times. 
         | 
      2075 | if (hwaccel) { | 
| 882 | ✗ | ret = hwaccel->end_frame(avctx); | |
| 883 | ✗ | if (ret < 0) | |
| 884 | ✗ | return ret; | |
| 885 | } | ||
| 886 | |||
| 887 | 2075 | ff_progress_frame_report(&f->picture, INT_MAX); | |
| 888 | |||
| 889 | 2075 | ff_progress_frame_unref(&f->last_picture); | |
| 890 | 2075 | av_refstruct_unref(&f->hwaccel_last_picture_private); | |
| 891 | 
        1/2✗ Branch 1 not taken. 
          ✓ Branch 2 taken 2075 times. 
         | 
      2075 | if ((ret = av_frame_ref(rframe, f->picture.f)) < 0) | 
| 892 | ✗ | return ret; | |
| 893 | |||
| 894 | 2075 | *got_frame = 1; | |
| 895 | |||
| 896 | 2075 | return avpkt->size; | |
| 897 | } | ||
| 898 | |||
| 899 | #if HAVE_THREADS | ||
| 900 | ✗ | static int update_thread_context(AVCodecContext *dst, const AVCodecContext *src) | |
| 901 | { | ||
| 902 | ✗ | FFV1Context *fsrc = src->priv_data; | |
| 903 | ✗ | FFV1Context *fdst = dst->priv_data; | |
| 904 | |||
| 905 | ✗ | if (dst == src) | |
| 906 | ✗ | return 0; | |
| 907 | |||
| 908 | ✗ | fdst->version = fsrc->version; | |
| 909 | ✗ | fdst->micro_version = fsrc->micro_version; | |
| 910 | ✗ | fdst->combined_version = fsrc->combined_version; | |
| 911 | ✗ | fdst->chroma_planes = fsrc->chroma_planes; | |
| 912 | ✗ | fdst->chroma_h_shift = fsrc->chroma_h_shift; | |
| 913 | ✗ | fdst->chroma_v_shift = fsrc->chroma_v_shift; | |
| 914 | ✗ | fdst->transparency = fsrc->transparency; | |
| 915 | ✗ | fdst->plane_count = fsrc->plane_count; | |
| 916 | ✗ | fdst->ac = fsrc->ac; | |
| 917 | ✗ | fdst->colorspace = fsrc->colorspace; | |
| 918 | ✗ | fdst->pix_fmt = fsrc->pix_fmt; | |
| 919 | ✗ | fdst->configured_pix_fmt = fsrc->configured_pix_fmt; | |
| 920 | |||
| 921 | ✗ | fdst->ec = fsrc->ec; | |
| 922 | ✗ | fdst->intra = fsrc->intra; | |
| 923 | ✗ | fdst->key_frame_ok = fsrc->key_frame_ok; | |
| 924 | |||
| 925 | ✗ | fdst->packed_at_lsb = fsrc->packed_at_lsb; | |
| 926 | ✗ | fdst->slice_count = fsrc->slice_count; | |
| 927 | ✗ | fdst->use32bit = fsrc->use32bit; | |
| 928 | ✗ | memcpy(fdst->state_transition, fsrc->state_transition, | |
| 929 | sizeof(fdst->state_transition)); | ||
| 930 | |||
| 931 | // in version 1 there is a single per-keyframe quant table, so | ||
| 932 | // we need to propagate it between threads | ||
| 933 | ✗ | if (fsrc->version < 2) | |
| 934 | ✗ | memcpy(fdst->quant_tables[0], fsrc->quant_tables[0], sizeof(fsrc->quant_tables[0])); | |
| 935 | |||
| 936 | ✗ | for (int i = 0; i < fdst->num_h_slices * fdst->num_v_slices; i++) { | |
| 937 | ✗ | FFV1SliceContext *sc = &fdst->slices[i]; | |
| 938 | ✗ | const FFV1SliceContext *sc0 = &fsrc->slices[i]; | |
| 939 | |||
| 940 | ✗ | av_refstruct_replace(&sc->plane, sc0->plane); | |
| 941 | |||
| 942 | ✗ | if (fsrc->version < 3) { | |
| 943 | ✗ | sc->slice_x = sc0->slice_x; | |
| 944 | ✗ | sc->slice_y = sc0->slice_y; | |
| 945 | ✗ | sc->slice_width = sc0->slice_width; | |
| 946 | ✗ | sc->slice_height = sc0->slice_height; | |
| 947 | } | ||
| 948 | } | ||
| 949 | |||
| 950 | ✗ | av_refstruct_replace(&fdst->slice_damaged, fsrc->slice_damaged); | |
| 951 | |||
| 952 | av_assert1(fdst->max_slice_count == fsrc->max_slice_count); | ||
| 953 | |||
| 954 | ✗ | ff_progress_frame_replace(&fdst->picture, &fsrc->picture); | |
| 955 | ✗ | av_refstruct_replace(&fdst->hwaccel_picture_private, | |
| 956 | ✗ | fsrc->hwaccel_picture_private); | |
| 957 | |||
| 958 | ✗ | return 0; | |
| 959 | } | ||
| 960 | #endif | ||
| 961 | |||
| 962 | 88 | static av_cold int ffv1_decode_close(AVCodecContext *avctx) | |
| 963 | { | ||
| 964 | 88 | FFV1Context *const s = avctx->priv_data; | |
| 965 | |||
| 966 | 88 | ff_progress_frame_unref(&s->picture); | |
| 967 | 88 | av_refstruct_unref(&s->hwaccel_picture_private); | |
| 968 | |||
| 969 | 88 | ff_progress_frame_unref(&s->last_picture); | |
| 970 | 88 | av_refstruct_unref(&s->hwaccel_last_picture_private); | |
| 971 | |||
| 972 | 88 | ff_ffv1_close(s); | |
| 973 | |||
| 974 | 88 | return 0; | |
| 975 | } | ||
| 976 | |||
| 977 | const FFCodec ff_ffv1_decoder = { | ||
| 978 | .p.name = "ffv1", | ||
| 979 | CODEC_LONG_NAME("FFmpeg video codec #1"), | ||
| 980 | .p.type = AVMEDIA_TYPE_VIDEO, | ||
| 981 | .p.id = AV_CODEC_ID_FFV1, | ||
| 982 | .priv_data_size = sizeof(FFV1Context), | ||
| 983 | .init = decode_init, | ||
| 984 | .close = ffv1_decode_close, | ||
| 985 | FF_CODEC_DECODE_CB(decode_frame), | ||
| 986 | UPDATE_THREAD_CONTEXT(update_thread_context), | ||
| 987 | .p.capabilities = AV_CODEC_CAP_DR1 | | ||
| 988 | AV_CODEC_CAP_FRAME_THREADS | AV_CODEC_CAP_SLICE_THREADS, | ||
| 989 | .caps_internal = FF_CODEC_CAP_INIT_CLEANUP | | ||
| 990 | FF_CODEC_CAP_SKIP_FRAME_FILL_PARAM | | ||
| 991 | FF_CODEC_CAP_USES_PROGRESSFRAMES, | ||
| 992 | .hw_configs = (const AVCodecHWConfigInternal *const []) { | ||
| 993 | #if CONFIG_FFV1_VULKAN_HWACCEL | ||
| 994 | HWACCEL_VULKAN(ffv1), | ||
| 995 | #endif | ||
| 996 | NULL | ||
| 997 | }, | ||
| 998 | }; | ||
| 999 |