utvideodec.c 18.6 KB
Newer Older
Kostya Shishkov's avatar
Kostya Shishkov committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26
/*
 * Ut Video decoder
 * Copyright (c) 2011 Konstantin Shishkov
 *
 * This file is part of Libav.
 *
 * Libav is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * Libav is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with Libav; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

/**
 * @file
 * Ut Video decoder
 */

27
#include <inttypes.h>
Kostya Shishkov's avatar
Kostya Shishkov committed
28 29 30 31
#include <stdlib.h>

#include "libavutil/intreadwrite.h"
#include "avcodec.h"
32
#include "bswapdsp.h"
Kostya Shishkov's avatar
Kostya Shishkov committed
33 34
#include "bytestream.h"
#include "get_bits.h"
35
#include "thread.h"
36
#include "utvideo.h"
Kostya Shishkov's avatar
Kostya Shishkov committed
37

38
static int build_huff(const uint8_t *src, VLC *vlc, int *fsym)
Kostya Shishkov's avatar
Kostya Shishkov committed
39 40 41 42 43 44 45 46 47
{
    int i;
    HuffEntry he[256];
    int last;
    uint32_t codes[256];
    uint8_t bits[256];
    uint8_t syms[256];
    uint32_t code;

48
    *fsym = -1;
Kostya Shishkov's avatar
Kostya Shishkov committed
49 50 51 52
    for (i = 0; i < 256; i++) {
        he[i].sym = i;
        he[i].len = *src++;
    }
53
    qsort(he, 256, sizeof(*he), ff_ut_huff_cmp_len);
Kostya Shishkov's avatar
Kostya Shishkov committed
54

55 56 57 58 59
    if (!he[0].len) {
        *fsym = he[0].sym;
        return 0;
    }
    if (he[0].len > 32)
Kostya Shishkov's avatar
Kostya Shishkov committed
60 61 62 63 64 65 66 67 68 69 70 71 72 73
        return -1;

    last = 255;
    while (he[last].len == 255 && last)
        last--;

    code = 1;
    for (i = last; i >= 0; i--) {
        codes[i] = code >> (32 - he[i].len);
        bits[i]  = he[i].len;
        syms[i]  = he[i].sym;
        code += 0x80000000u >> (he[i].len - 1);
    }

74 75 76 77
    return ff_init_vlc_sparse(vlc, FFMIN(he[last].len, 9), last + 1,
                              bits,  sizeof(*bits),  sizeof(*bits),
                              codes, sizeof(*codes), sizeof(*codes),
                              syms,  sizeof(*syms),  sizeof(*syms), 0);
Kostya Shishkov's avatar
Kostya Shishkov committed
78 79 80
}

static int decode_plane(UtvideoContext *c, int plane_no,
81
                        uint8_t *dst, int step, ptrdiff_t stride,
Kostya Shishkov's avatar
Kostya Shishkov committed
82
                        int width, int height,
83
                        const uint8_t *src, int use_pred)
Kostya Shishkov's avatar
Kostya Shishkov committed
84 85 86 87 88
{
    int i, j, slice, pix;
    int sstart, send;
    VLC vlc;
    GetBitContext gb;
89
    int prev, fsym;
90
    const int cmask = ~(!plane_no && c->avctx->pix_fmt == AV_PIX_FMT_YUV420P);
Kostya Shishkov's avatar
Kostya Shishkov committed
91

92
    if (build_huff(src, &vlc, &fsym)) {
Kostya Shishkov's avatar
Kostya Shishkov committed
93 94 95
        av_log(c->avctx, AV_LOG_ERROR, "Cannot build Huffman codes\n");
        return AVERROR_INVALIDDATA;
    }
96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119
    if (fsym >= 0) { // build_huff reported a symbol to fill slices with
        send = 0;
        for (slice = 0; slice < c->slices; slice++) {
            uint8_t *dest;

            sstart = send;
            send   = (height * (slice + 1) / c->slices) & cmask;
            dest   = dst + sstart * stride;

            prev = 0x80;
            for (j = sstart; j < send; j++) {
                for (i = 0; i < width * step; i += step) {
                    pix = fsym;
                    if (use_pred) {
                        prev += pix;
                        pix   = prev;
                    }
                    dest[i] = pix;
                }
                dest += stride;
            }
        }
        return 0;
    }
Kostya Shishkov's avatar
Kostya Shishkov committed
120 121 122 123 124 125 126 127 128

    src      += 256;

    send = 0;
    for (slice = 0; slice < c->slices; slice++) {
        uint8_t *dest;
        int slice_data_start, slice_data_end, slice_size;

        sstart = send;
129
        send   = (height * (slice + 1) / c->slices) & cmask;
Kostya Shishkov's avatar
Kostya Shishkov committed
130 131 132 133 134 135 136 137
        dest   = dst + sstart * stride;

        // slice offset and size validation was done earlier
        slice_data_start = slice ? AV_RL32(src + slice * 4 - 4) : 0;
        slice_data_end   = AV_RL32(src + slice * 4);
        slice_size       = slice_data_end - slice_data_start;

        if (!slice_size) {
138 139 140
            av_log(c->avctx, AV_LOG_ERROR, "Plane has more than one symbol "
                   "yet a slice has a length of zero.\n");
            goto fail;
Kostya Shishkov's avatar
Kostya Shishkov committed
141 142
        }

143 144
        memcpy(c->slice_bits, src + slice_data_start + c->slices * 4,
               slice_size);
145
        memset(c->slice_bits + slice_size, 0, AV_INPUT_BUFFER_PADDING_SIZE);
146 147 148
        c->bdsp.bswap_buf((uint32_t *) c->slice_bits,
                          (uint32_t *) c->slice_bits,
                          (slice_data_end - slice_data_start + 3) >> 2);
Kostya Shishkov's avatar
Kostya Shishkov committed
149 150 151 152 153 154
        init_get_bits(&gb, c->slice_bits, slice_size * 8);

        prev = 0x80;
        for (j = sstart; j < send; j++) {
            for (i = 0; i < width * step; i += step) {
                if (get_bits_left(&gb) <= 0) {
155 156
                    av_log(c->avctx, AV_LOG_ERROR,
                           "Slice decoding ran out of bits\n");
Kostya Shishkov's avatar
Kostya Shishkov committed
157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172
                    goto fail;
                }
                pix = get_vlc2(&gb, vlc.table, vlc.bits, 4);
                if (pix < 0) {
                    av_log(c->avctx, AV_LOG_ERROR, "Decoding error\n");
                    goto fail;
                }
                if (use_pred) {
                    prev += pix;
                    pix   = prev;
                }
                dest[i] = pix;
            }
            dest += stride;
        }
        if (get_bits_left(&gb) > 32)
173 174
            av_log(c->avctx, AV_LOG_WARNING,
                   "%d bits left after decoding slice\n", get_bits_left(&gb));
Kostya Shishkov's avatar
Kostya Shishkov committed
175 176
    }

177
    ff_free_vlc(&vlc);
Kostya Shishkov's avatar
Kostya Shishkov committed
178 179 180

    return 0;
fail:
181
    ff_free_vlc(&vlc);
Kostya Shishkov's avatar
Kostya Shishkov committed
182 183 184
    return AVERROR_INVALIDDATA;
}

185 186
static void restore_rgb_planes(uint8_t *src, int step, ptrdiff_t stride,
                               int width, int height)
Kostya Shishkov's avatar
Kostya Shishkov committed
187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202
{
    int i, j;
    uint8_t r, g, b;

    for (j = 0; j < height; j++) {
        for (i = 0; i < width * step; i += step) {
            r = src[i];
            g = src[i + 1];
            b = src[i + 2];
            src[i]     = r + g - 0x80;
            src[i + 2] = b + g - 0x80;
        }
        src += stride;
    }
}

203
static void restore_median(uint8_t *src, int step, ptrdiff_t stride,
204
                           int width, int height, int slices, int rmode)
Kostya Shishkov's avatar
Kostya Shishkov committed
205 206 207 208 209
{
    int i, j, slice;
    int A, B, C;
    uint8_t *bsrc;
    int slice_start, slice_height;
210
    const int cmask = ~rmode;
Kostya Shishkov's avatar
Kostya Shishkov committed
211 212

    for (slice = 0; slice < slices; slice++) {
213 214 215
        slice_start  = ((slice * height) / slices) & cmask;
        slice_height = ((((slice + 1) * height) / slices) & cmask) -
                       slice_start;
216 217
        if (!slice_height)
            continue;
Kostya Shishkov's avatar
Kostya Shishkov committed
218 219 220 221 222 223 224 225

        bsrc = src + slice_start * stride;

        // first line - left neighbour prediction
        bsrc[0] += 0x80;
        A = bsrc[0];
        for (i = step; i < width * step; i += step) {
            bsrc[i] += A;
226
            A        = bsrc[i];
Kostya Shishkov's avatar
Kostya Shishkov committed
227 228 229 230
        }
        bsrc += stride;
        if (slice_height == 1)
            continue;
231 232
        // second line - first element has top prediction, the rest uses median
        C        = bsrc[-stride];
Kostya Shishkov's avatar
Kostya Shishkov committed
233
        bsrc[0] += C;
234
        A        = bsrc[0];
Kostya Shishkov's avatar
Kostya Shishkov committed
235
        for (i = step; i < width * step; i += step) {
236
            B        = bsrc[i - stride];
Kostya Shishkov's avatar
Kostya Shishkov committed
237
            bsrc[i] += mid_pred(A, B, (uint8_t)(A + B - C));
238 239
            C        = B;
            A        = bsrc[i];
Kostya Shishkov's avatar
Kostya Shishkov committed
240 241 242 243 244
        }
        bsrc += stride;
        // the rest of lines use continuous median prediction
        for (j = 2; j < slice_height; j++) {
            for (i = 0; i < width * step; i += step) {
245
                B        = bsrc[i - stride];
Kostya Shishkov's avatar
Kostya Shishkov committed
246
                bsrc[i] += mid_pred(A, B, (uint8_t)(A + B - C));
247 248
                C        = B;
                A        = bsrc[i];
Kostya Shishkov's avatar
Kostya Shishkov committed
249 250 251 252 253 254
            }
            bsrc += stride;
        }
    }
}

255 256 257 258
/* UtVideo interlaced mode treats every two lines as a single one,
 * so restoring function should take care of possible padding between
 * two parts of the same "line".
 */
259
static void restore_median_il(uint8_t *src, int step, ptrdiff_t stride,
260 261 262 263 264 265
                              int width, int height, int slices, int rmode)
{
    int i, j, slice;
    int A, B, C;
    uint8_t *bsrc;
    int slice_start, slice_height;
266
    const int cmask   = ~(rmode ? 3 : 1);
267
    const ptrdiff_t stride2 = stride << 1;
268 269 270

    for (slice = 0; slice < slices; slice++) {
        slice_start    = ((slice * height) / slices) & cmask;
271 272
        slice_height   = ((((slice + 1) * height) / slices) & cmask) -
                         slice_start;
273
        slice_height >>= 1;
274 275
        if (!slice_height)
            continue;
276 277 278 279 280

        bsrc = src + slice_start * stride;

        // first line - left neighbour prediction
        bsrc[0] += 0x80;
281
        A        = bsrc[0];
282 283
        for (i = step; i < width * step; i += step) {
            bsrc[i] += A;
284
            A        = bsrc[i];
285 286 287
        }
        for (i = 0; i < width * step; i += step) {
            bsrc[stride + i] += A;
288
            A                 = bsrc[stride + i];
289 290 291 292
        }
        bsrc += stride2;
        if (slice_height == 1)
            continue;
293 294
        // second line - first element has top prediction, the rest uses median
        C        = bsrc[-stride2];
295
        bsrc[0] += C;
296
        A        = bsrc[0];
297
        for (i = step; i < width * step; i += step) {
298
            B        = bsrc[i - stride2];
299
            bsrc[i] += mid_pred(A, B, (uint8_t)(A + B - C));
300 301
            C        = B;
            A        = bsrc[i];
302 303
        }
        for (i = 0; i < width * step; i += step) {
304
            B                 = bsrc[i - stride];
305
            bsrc[stride + i] += mid_pred(A, B, (uint8_t)(A + B - C));
306 307
            C                 = B;
            A                 = bsrc[stride + i];
308 309 310 311 312
        }
        bsrc += stride2;
        // the rest of lines use continuous median prediction
        for (j = 2; j < slice_height; j++) {
            for (i = 0; i < width * step; i += step) {
313
                B        = bsrc[i - stride2];
314
                bsrc[i] += mid_pred(A, B, (uint8_t)(A + B - C));
315 316
                C        = B;
                A        = bsrc[i];
317 318
            }
            for (i = 0; i < width * step; i += step) {
319
                B                 = bsrc[i - stride];
320
                bsrc[i + stride] += mid_pred(A, B, (uint8_t)(A + B - C));
321 322
                C                 = B;
                A                 = bsrc[i + stride];
323 324 325 326 327 328
            }
            bsrc += stride2;
        }
    }
}

329
static int decode_frame(AVCodecContext *avctx, void *data, int *got_frame,
330
                        AVPacket *avpkt)
Kostya Shishkov's avatar
Kostya Shishkov committed
331 332 333 334 335 336 337 338
{
    const uint8_t *buf = avpkt->data;
    int buf_size = avpkt->size;
    UtvideoContext *c = avctx->priv_data;
    int i, j;
    const uint8_t *plane_start[5];
    int plane_size, max_slice_size = 0, slice_start, slice_end, slice_size;
    int ret;
339
    GetByteContext gb;
340
    ThreadFrame frame = { .f = data };
Kostya Shishkov's avatar
Kostya Shishkov committed
341

342
    if ((ret = ff_thread_get_buffer(avctx, &frame, 0)) < 0) {
Kostya Shishkov's avatar
Kostya Shishkov committed
343 344 345 346
        av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n");
        return ret;
    }

347 348
    ff_thread_finish_setup(avctx);

349
    /* parse plane structure to get frame flags and validate slice offsets */
350
    bytestream2_init(&gb, buf, buf_size);
Kostya Shishkov's avatar
Kostya Shishkov committed
351
    for (i = 0; i < c->planes; i++) {
352 353
        plane_start[i] = gb.buffer;
        if (bytestream2_get_bytes_left(&gb) < 256 + 4 * c->slices) {
Kostya Shishkov's avatar
Kostya Shishkov committed
354 355 356
            av_log(avctx, AV_LOG_ERROR, "Insufficient data for a plane\n");
            return AVERROR_INVALIDDATA;
        }
357
        bytestream2_skipu(&gb, 256);
Kostya Shishkov's avatar
Kostya Shishkov committed
358 359 360
        slice_start = 0;
        slice_end   = 0;
        for (j = 0; j < c->slices; j++) {
361
            slice_end   = bytestream2_get_le32u(&gb);
Kostya Shishkov's avatar
Kostya Shishkov committed
362
            slice_size  = slice_end - slice_start;
363
            if (slice_end < 0 || slice_size < 0 ||
364
                bytestream2_get_bytes_left(&gb) < slice_end) {
Kostya Shishkov's avatar
Kostya Shishkov committed
365 366 367 368 369 370 371
                av_log(avctx, AV_LOG_ERROR, "Incorrect slice size\n");
                return AVERROR_INVALIDDATA;
            }
            slice_start = slice_end;
            max_slice_size = FFMAX(max_slice_size, slice_size);
        }
        plane_size = slice_end;
372
        bytestream2_skipu(&gb, plane_size);
Kostya Shishkov's avatar
Kostya Shishkov committed
373
    }
374 375
    plane_start[c->planes] = gb.buffer;
    if (bytestream2_get_bytes_left(&gb) < c->frame_info_size) {
Kostya Shishkov's avatar
Kostya Shishkov committed
376 377 378
        av_log(avctx, AV_LOG_ERROR, "Not enough data for frame information\n");
        return AVERROR_INVALIDDATA;
    }
379
    c->frame_info = bytestream2_get_le32u(&gb);
380 381
    av_log(avctx, AV_LOG_DEBUG, "frame information flags %"PRIX32"\n",
           c->frame_info);
Kostya Shishkov's avatar
Kostya Shishkov committed
382 383 384 385

    c->frame_pred = (c->frame_info >> 8) & 3;

    if (c->frame_pred == PRED_GRADIENT) {
386
        avpriv_request_sample(avctx, "Frame with gradient prediction");
Kostya Shishkov's avatar
Kostya Shishkov committed
387 388 389 390
        return AVERROR_PATCHWELCOME;
    }

    av_fast_malloc(&c->slice_bits, &c->slice_bits_size,
391
                   max_slice_size + AV_INPUT_BUFFER_PADDING_SIZE);
Kostya Shishkov's avatar
Kostya Shishkov committed
392 393 394 395 396 397 398

    if (!c->slice_bits) {
        av_log(avctx, AV_LOG_ERROR, "Cannot allocate temporary buffer\n");
        return AVERROR(ENOMEM);
    }

    switch (c->avctx->pix_fmt) {
399 400
    case AV_PIX_FMT_RGB24:
    case AV_PIX_FMT_RGBA:
Kostya Shishkov's avatar
Kostya Shishkov committed
401
        for (i = 0; i < c->planes; i++) {
402 403
            ret = decode_plane(c, i, frame.f->data[0] + ff_ut_rgb_order[i],
                               c->planes, frame.f->linesize[0], avctx->width,
404 405
                               avctx->height, plane_start[i],
                               c->frame_pred == PRED_LEFT);
Kostya Shishkov's avatar
Kostya Shishkov committed
406 407
            if (ret)
                return ret;
408 409
            if (c->frame_pred == PRED_MEDIAN) {
                if (!c->interlaced) {
410 411
                    restore_median(frame.f->data[0] + ff_ut_rgb_order[i],
                                   c->planes, frame.f->linesize[0], avctx->width,
412 413
                                   avctx->height, c->slices, 0);
                } else {
414 415
                    restore_median_il(frame.f->data[0] + ff_ut_rgb_order[i],
                                      c->planes, frame.f->linesize[0],
416 417
                                      avctx->width, avctx->height, c->slices,
                                      0);
418 419
                }
            }
Kostya Shishkov's avatar
Kostya Shishkov committed
420
        }
421
        restore_rgb_planes(frame.f->data[0], c->planes, frame.f->linesize[0],
Kostya Shishkov's avatar
Kostya Shishkov committed
422 423
                           avctx->width, avctx->height);
        break;
424
    case AV_PIX_FMT_YUV420P:
Kostya Shishkov's avatar
Kostya Shishkov committed
425
        for (i = 0; i < 3; i++) {
426
            ret = decode_plane(c, i, frame.f->data[i], 1, frame.f->linesize[i],
427
                               avctx->width >> !!i, avctx->height >> !!i,
428
                               plane_start[i], c->frame_pred == PRED_LEFT);
Kostya Shishkov's avatar
Kostya Shishkov committed
429 430
            if (ret)
                return ret;
431 432
            if (c->frame_pred == PRED_MEDIAN) {
                if (!c->interlaced) {
433
                    restore_median(frame.f->data[i], 1, frame.f->linesize[i],
434 435 436
                                   avctx->width >> !!i, avctx->height >> !!i,
                                   c->slices, !i);
                } else {
437
                    restore_median_il(frame.f->data[i], 1, frame.f->linesize[i],
438 439 440 441 442
                                      avctx->width  >> !!i,
                                      avctx->height >> !!i,
                                      c->slices, !i);
                }
            }
Kostya Shishkov's avatar
Kostya Shishkov committed
443 444
        }
        break;
445
    case AV_PIX_FMT_YUV422P:
Kostya Shishkov's avatar
Kostya Shishkov committed
446
        for (i = 0; i < 3; i++) {
447
            ret = decode_plane(c, i, frame.f->data[i], 1, frame.f->linesize[i],
448
                               avctx->width >> !!i, avctx->height,
449
                               plane_start[i], c->frame_pred == PRED_LEFT);
Kostya Shishkov's avatar
Kostya Shishkov committed
450 451
            if (ret)
                return ret;
452 453
            if (c->frame_pred == PRED_MEDIAN) {
                if (!c->interlaced) {
454
                    restore_median(frame.f->data[i], 1, frame.f->linesize[i],
455 456 457
                                   avctx->width >> !!i, avctx->height,
                                   c->slices, 0);
                } else {
458
                    restore_median_il(frame.f->data[i], 1, frame.f->linesize[i],
459 460 461 462
                                      avctx->width >> !!i, avctx->height,
                                      c->slices, 0);
                }
            }
Kostya Shishkov's avatar
Kostya Shishkov committed
463 464 465 466
        }
        break;
    }

467 468 469
    frame.f->key_frame = 1;
    frame.f->pict_type = AV_PICTURE_TYPE_I;
    frame.f->interlaced_frame = !!c->interlaced;
470

471
    *got_frame = 1;
Kostya Shishkov's avatar
Kostya Shishkov committed
472 473 474 475 476 477 478 479 480 481 482

    /* always report that the buffer was completely consumed */
    return buf_size;
}

static av_cold int decode_init(AVCodecContext *avctx)
{
    UtvideoContext * const c = avctx->priv_data;

    c->avctx = avctx;

483
    ff_bswapdsp_init(&c->bdsp);
Kostya Shishkov's avatar
Kostya Shishkov committed
484 485

    if (avctx->extradata_size < 16) {
486 487
        av_log(avctx, AV_LOG_ERROR,
               "Insufficient extradata size %d, should be at least 16\n",
Kostya Shishkov's avatar
Kostya Shishkov committed
488 489 490 491 492 493 494
               avctx->extradata_size);
        return AVERROR_INVALIDDATA;
    }

    av_log(avctx, AV_LOG_DEBUG, "Encoder version %d.%d.%d.%d\n",
           avctx->extradata[3], avctx->extradata[2],
           avctx->extradata[1], avctx->extradata[0]);
495
    av_log(avctx, AV_LOG_DEBUG, "Original format %"PRIX32"\n",
496
           AV_RB32(avctx->extradata + 4));
Kostya Shishkov's avatar
Kostya Shishkov committed
497 498 499 500
    c->frame_info_size = AV_RL32(avctx->extradata + 8);
    c->flags           = AV_RL32(avctx->extradata + 12);

    if (c->frame_info_size != 4)
501
        avpriv_request_sample(avctx, "Frame info not 4 bytes");
502
    av_log(avctx, AV_LOG_DEBUG, "Encoding parameters %08"PRIX32"\n", c->flags);
Kostya Shishkov's avatar
Kostya Shishkov committed
503 504 505 506 507 508 509 510 511
    c->slices      = (c->flags >> 24) + 1;
    c->compression = c->flags & 1;
    c->interlaced  = c->flags & 0x800;

    c->slice_bits_size = 0;

    switch (avctx->codec_tag) {
    case MKTAG('U', 'L', 'R', 'G'):
        c->planes      = 3;
512
        avctx->pix_fmt = AV_PIX_FMT_RGB24;
Kostya Shishkov's avatar
Kostya Shishkov committed
513 514 515
        break;
    case MKTAG('U', 'L', 'R', 'A'):
        c->planes      = 4;
516
        avctx->pix_fmt = AV_PIX_FMT_RGBA;
Kostya Shishkov's avatar
Kostya Shishkov committed
517 518 519
        break;
    case MKTAG('U', 'L', 'Y', '0'):
        c->planes      = 3;
520
        avctx->pix_fmt = AV_PIX_FMT_YUV420P;
521
        avctx->colorspace = AVCOL_SPC_BT470BG;
Kostya Shishkov's avatar
Kostya Shishkov committed
522 523 524
        break;
    case MKTAG('U', 'L', 'Y', '2'):
        c->planes      = 3;
525
        avctx->pix_fmt = AV_PIX_FMT_YUV422P;
526
        avctx->colorspace = AVCOL_SPC_BT470BG;
Kostya Shishkov's avatar
Kostya Shishkov committed
527
        break;
528 529 530
    case MKTAG('U', 'L', 'H', '0'):
        c->planes      = 3;
        avctx->pix_fmt = AV_PIX_FMT_YUV420P;
531
        avctx->colorspace = AVCOL_SPC_BT709;
532 533 534 535
        break;
    case MKTAG('U', 'L', 'H', '2'):
        c->planes      = 3;
        avctx->pix_fmt = AV_PIX_FMT_YUV422P;
536
        avctx->colorspace = AVCOL_SPC_BT709;
537
        break;
Kostya Shishkov's avatar
Kostya Shishkov committed
538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557
    default:
        av_log(avctx, AV_LOG_ERROR, "Unknown Ut Video FOURCC provided (%08X)\n",
               avctx->codec_tag);
        return AVERROR_INVALIDDATA;
    }

    return 0;
}

static av_cold int decode_end(AVCodecContext *avctx)
{
    UtvideoContext * const c = avctx->priv_data;

    av_freep(&c->slice_bits);

    return 0;
}

AVCodec ff_utvideo_decoder = {
    .name           = "utvideo",
558
    .long_name      = NULL_IF_CONFIG_SMALL("Ut Video"),
Kostya Shishkov's avatar
Kostya Shishkov committed
559
    .type           = AVMEDIA_TYPE_VIDEO,
560
    .id             = AV_CODEC_ID_UTVIDEO,
Kostya Shishkov's avatar
Kostya Shishkov committed
561 562 563 564
    .priv_data_size = sizeof(UtvideoContext),
    .init           = decode_init,
    .close          = decode_end,
    .decode         = decode_frame,
565
    .capabilities   = AV_CODEC_CAP_DR1 | AV_CODEC_CAP_FRAME_THREADS,
Kostya Shishkov's avatar
Kostya Shishkov committed
566
};