Browse code

Ut Video decoder

Signed-off-by: Anton Khirnov <anton@khirnov.net>

Kostya Shishkov authored on 2011/10/17 00:20:17
Showing 8 changed files
... ...
@@ -52,6 +52,7 @@ easier to use. The changes are:
52 52
 - VC-1 interlaced decoding
53 53
 - lut, lutrgb, and lutyuv filters
54 54
 - boxblur filter
55
+- Ut Video decoder
55 56
 
56 57
 
57 58
 version 0.7:
... ...
@@ -516,6 +516,7 @@ following image formats are supported:
516 516
     @tab encoding supported through external library libtheora
517 517
 @item Tiertex Limited SEQ video  @tab     @tab  X
518 518
     @tab Codec used in DOS CD-ROM FlashBack game.
519
+@item Ut Video               @tab     @tab  X
519 520
 @item V210 Quicktime Uncompressed 4:2:2 10-bit     @tab  X  @tab  X
520 521
 @item VMware Screen Codec / VMware Video  @tab     @tab  X
521 522
     @tab Codec used in videos captured by VMware.
... ...
@@ -378,6 +378,7 @@ OBJS-$(CONFIG_TTA_DECODER)             += tta.o
378 378
 OBJS-$(CONFIG_TWINVQ_DECODER)          += twinvq.o celp_math.o
379 379
 OBJS-$(CONFIG_TXD_DECODER)             += txd.o s3tc.o
380 380
 OBJS-$(CONFIG_ULTI_DECODER)            += ulti.o
381
+OBJS-$(CONFIG_UTVIDEO_DECODER)         += utvideo.o
381 382
 OBJS-$(CONFIG_V210_DECODER)            += v210dec.o
382 383
 OBJS-$(CONFIG_V210_ENCODER)            += v210enc.o
383 384
 OBJS-$(CONFIG_V210X_DECODER)           += v210x.o
... ...
@@ -199,6 +199,7 @@ void avcodec_register_all(void)
199 199
     REGISTER_DECODER (TSCC, tscc);
200 200
     REGISTER_DECODER (TXD, txd);
201 201
     REGISTER_DECODER (ULTI, ulti);
202
+    REGISTER_DECODER (UTVIDEO, utvideo);
202 203
     REGISTER_ENCDEC  (V210,  v210);
203 204
     REGISTER_DECODER (V210X, v210x);
204 205
     REGISTER_DECODER (VB, vb);
... ...
@@ -212,6 +212,7 @@ enum CodecID {
212 212
     CODEC_ID_VC1IMAGE,
213 213
     CODEC_ID_G723_1,
214 214
     CODEC_ID_G729,
215
+    CODEC_ID_UTVIDEO,
215 216
 
216 217
     /* various PCM "codecs" */
217 218
     CODEC_ID_FIRST_AUDIO = 0x10000,     ///< A dummy id pointing at the start of audio codecs
218 219
new file mode 100644
... ...
@@ -0,0 +1,460 @@
0
+/*
1
+ * Ut Video decoder
2
+ * Copyright (c) 2011 Konstantin Shishkov
3
+ *
4
+ * This file is part of Libav.
5
+ *
6
+ * Libav 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
+ * Libav 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 Libav; if not, write to the Free Software
18
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19
+ */
20
+
21
+/**
22
+ * @file
23
+ * Ut Video decoder
24
+ */
25
+
26
+#include <stdlib.h>
27
+
28
+#include "libavutil/intreadwrite.h"
29
+#include "avcodec.h"
30
+#include "bytestream.h"
31
+#include "get_bits.h"
32
+#include "dsputil.h"
33
+
34
+enum {
35
+    PRED_NONE = 0,
36
+    PRED_LEFT,
37
+    PRED_GRADIENT,
38
+    PRED_MEDIAN,
39
+};
40
+
41
+typedef struct UtvideoContext {
42
+    AVCodecContext *avctx;
43
+    AVFrame pic;
44
+    DSPContext dsp;
45
+
46
+    uint32_t frame_info_size, flags, frame_info;
47
+    int planes;
48
+    int slices;
49
+    int compression;
50
+    int interlaced;
51
+    int frame_pred;
52
+
53
+    uint8_t *slice_bits;
54
+    int slice_bits_size;
55
+} UtvideoContext;
56
+
57
+typedef struct HuffEntry {
58
+    uint8_t sym;
59
+    uint8_t len;
60
+} HuffEntry;
61
+
62
+static int huff_cmp(const void *a, const void *b)
63
+{
64
+    const HuffEntry *aa = a, *bb = b;
65
+    return (aa->len - bb->len)*256 + aa->sym - bb->sym;
66
+}
67
+
68
+static int build_huff(const uint8_t *src, VLC *vlc)
69
+{
70
+    int i;
71
+    HuffEntry he[256];
72
+    int last;
73
+    uint32_t codes[256];
74
+    uint8_t bits[256];
75
+    uint8_t syms[256];
76
+    uint32_t code;
77
+
78
+    for (i = 0; i < 256; i++) {
79
+        he[i].sym = i;
80
+        he[i].len = *src++;
81
+    }
82
+    qsort(he, 256, sizeof(*he), huff_cmp);
83
+
84
+    if (!he[0].len || he[0].len > 32)
85
+        return -1;
86
+
87
+    last = 255;
88
+    while (he[last].len == 255 && last)
89
+        last--;
90
+
91
+    code = 1;
92
+    for (i = last; i >= 0; i--) {
93
+        codes[i] = code >> (32 - he[i].len);
94
+        bits[i]  = he[i].len;
95
+        syms[i]  = he[i].sym;
96
+        code += 0x80000000u >> (he[i].len - 1);
97
+    }
98
+
99
+    return init_vlc_sparse(vlc, FFMIN(he[last].len, 9), last + 1,
100
+                           bits,  sizeof(*bits),  sizeof(*bits),
101
+                           codes, sizeof(*codes), sizeof(*codes),
102
+                           syms,  sizeof(*syms),  sizeof(*syms), 0);
103
+}
104
+
105
+static int decode_plane(UtvideoContext *c, int plane_no,
106
+                        uint8_t *dst, int step, int stride,
107
+                        int width, int height,
108
+                        const uint8_t *src, int src_size, int use_pred)
109
+{
110
+    int i, j, slice, pix;
111
+    int sstart, send;
112
+    VLC vlc;
113
+    GetBitContext gb;
114
+    int prev;
115
+
116
+    if (build_huff(src, &vlc)) {
117
+        av_log(c->avctx, AV_LOG_ERROR, "Cannot build Huffman codes\n");
118
+        return AVERROR_INVALIDDATA;
119
+    }
120
+
121
+    src      += 256;
122
+    src_size -= 256;
123
+
124
+    send = 0;
125
+    for (slice = 0; slice < c->slices; slice++) {
126
+        uint8_t *dest;
127
+        int slice_data_start, slice_data_end, slice_size;
128
+
129
+        sstart = send;
130
+        send   = height * (slice + 1) / c->slices;
131
+        dest   = dst + sstart * stride;
132
+
133
+        // slice offset and size validation was done earlier
134
+        slice_data_start = slice ? AV_RL32(src + slice * 4 - 4) : 0;
135
+        slice_data_end   = AV_RL32(src + slice * 4);
136
+        slice_size       = slice_data_end - slice_data_start;
137
+
138
+        if (!slice_size) {
139
+            for (j = sstart; j < send; j++) {
140
+                for (i = 0; i < width * step; i += step)
141
+                    dest[i] = 0x80;
142
+                dest += stride;
143
+            }
144
+            continue;
145
+        }
146
+
147
+        memcpy(c->slice_bits, src + slice_data_start + c->slices * 4, slice_size);
148
+        memset(c->slice_bits + slice_size, 0, FF_INPUT_BUFFER_PADDING_SIZE);
149
+        c->dsp.bswap_buf((uint32_t*)c->slice_bits, (uint32_t*)c->slice_bits,
150
+                         (slice_data_end - slice_data_start + 3) >> 2);
151
+        init_get_bits(&gb, c->slice_bits, slice_size * 8);
152
+
153
+        prev = 0x80;
154
+        for (j = sstart; j < send; j++) {
155
+            for (i = 0; i < width * step; i += step) {
156
+                if (get_bits_left(&gb) <= 0) {
157
+                    av_log(c->avctx, AV_LOG_ERROR, "Slice decoding ran out of bits\n");
158
+                    goto fail;
159
+                }
160
+                pix = get_vlc2(&gb, vlc.table, vlc.bits, 4);
161
+                if (pix < 0) {
162
+                    av_log(c->avctx, AV_LOG_ERROR, "Decoding error\n");
163
+                    goto fail;
164
+                }
165
+                if (use_pred) {
166
+                    prev += pix;
167
+                    pix   = prev;
168
+                }
169
+                dest[i] = pix;
170
+            }
171
+            dest += stride;
172
+        }
173
+        if (get_bits_left(&gb) > 32)
174
+            av_log(c->avctx, AV_LOG_WARNING, "%d bits left after decoding slice\n",
175
+                   get_bits_left(&gb));
176
+    }
177
+
178
+    free_vlc(&vlc);
179
+
180
+    return 0;
181
+fail:
182
+    free_vlc(&vlc);
183
+    return AVERROR_INVALIDDATA;
184
+}
185
+
186
+static const int rgb_order[4] = { 1, 2, 0, 3 };
187
+
188
+static void restore_rgb_planes(uint8_t *src, int step, int stride, int width, int height)
189
+{
190
+    int i, j;
191
+    uint8_t r, g, b;
192
+
193
+    for (j = 0; j < height; j++) {
194
+        for (i = 0; i < width * step; i += step) {
195
+            r = src[i];
196
+            g = src[i + 1];
197
+            b = src[i + 2];
198
+            src[i]     = r + g - 0x80;
199
+            src[i + 2] = b + g - 0x80;
200
+        }
201
+        src += stride;
202
+    }
203
+}
204
+
205
+static void restore_median(uint8_t *src, int step, int stride,
206
+                           int width, int height, int slices)
207
+{
208
+    int i, j, slice;
209
+    int A, B, C;
210
+    uint8_t *bsrc;
211
+    int slice_start, slice_height;
212
+
213
+    for (slice = 0; slice < slices; slice++) {
214
+        slice_start = (slice * height) / slices;
215
+        slice_height = ((slice + 1) * height) / slices - slice_start;
216
+
217
+        bsrc = src + slice_start * stride;
218
+
219
+        // first line - left neighbour prediction
220
+        bsrc[0] += 0x80;
221
+        A = bsrc[0];
222
+        for (i = step; i < width * step; i += step) {
223
+            bsrc[i] += A;
224
+            A = bsrc[i];
225
+        }
226
+        bsrc += stride;
227
+        if (slice_height == 1)
228
+            continue;
229
+        // second line - first element has top predition, the rest uses median
230
+        C = bsrc[-stride];
231
+        bsrc[0] += C;
232
+        A = bsrc[0];
233
+        for (i = step; i < width * step; i += step) {
234
+            B = bsrc[i - stride];
235
+            bsrc[i] += mid_pred(A, B, (uint8_t)(A + B - C));
236
+            C = B;
237
+            A = bsrc[i];
238
+        }
239
+        bsrc += stride;
240
+        // the rest of lines use continuous median prediction
241
+        for (j = 2; j < slice_height; j++) {
242
+            for (i = 0; i < width * step; i += step) {
243
+                B = bsrc[i - stride];
244
+                bsrc[i] += mid_pred(A, B, (uint8_t)(A + B - C));
245
+                C = B;
246
+                A = bsrc[i];
247
+            }
248
+            bsrc += stride;
249
+        }
250
+    }
251
+}
252
+
253
+static int decode_frame(AVCodecContext *avctx, void *data, int *data_size, AVPacket *avpkt)
254
+{
255
+    const uint8_t *buf = avpkt->data;
256
+    int buf_size = avpkt->size;
257
+    const uint8_t *buf_end = buf + buf_size;
258
+    UtvideoContext *c = avctx->priv_data;
259
+    const uint8_t *ptr;
260
+    int i, j;
261
+    const uint8_t *plane_start[5];
262
+    int plane_size, max_slice_size = 0, slice_start, slice_end, slice_size;
263
+    int ret;
264
+
265
+    if (c->pic.data[0])
266
+        avctx->release_buffer(avctx, &c->pic);
267
+
268
+    c->pic.reference = 1;
269
+    c->pic.buffer_hints = FF_BUFFER_HINTS_VALID;
270
+    if ((ret = avctx->get_buffer(avctx, &c->pic)) < 0) {
271
+        av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n");
272
+        return ret;
273
+    }
274
+
275
+    /* parse plane structure to retrieve frame flags and validate slice offsets */
276
+    ptr = buf;
277
+    for (i = 0; i < c->planes; i++) {
278
+        plane_start[i] = ptr;
279
+        if (buf_end - ptr < 256 + 4 * c->slices) {
280
+            av_log(avctx, AV_LOG_ERROR, "Insufficient data for a plane\n");
281
+            return AVERROR_INVALIDDATA;
282
+        }
283
+        ptr += 256;
284
+        slice_start = 0;
285
+        slice_end   = 0;
286
+        for (j = 0; j < c->slices; j++) {
287
+            slice_end   = bytestream_get_le32(&ptr);
288
+            slice_size  = slice_end - slice_start;
289
+            if (slice_size < 0) {
290
+                av_log(avctx, AV_LOG_ERROR, "Incorrect slice size\n");
291
+                return AVERROR_INVALIDDATA;
292
+            }
293
+            slice_start = slice_end;
294
+            max_slice_size = FFMAX(max_slice_size, slice_size);
295
+        }
296
+        plane_size = slice_end;
297
+        if (buf_end - ptr < plane_size) {
298
+            av_log(avctx, AV_LOG_ERROR, "Plane size is bigger than available data\n");
299
+            return AVERROR_INVALIDDATA;
300
+        }
301
+        ptr += plane_size;
302
+    }
303
+    plane_start[c->planes] = ptr;
304
+    if (buf_end - ptr < c->frame_info_size) {
305
+        av_log(avctx, AV_LOG_ERROR, "Not enough data for frame information\n");
306
+        return AVERROR_INVALIDDATA;
307
+    }
308
+    c->frame_info = AV_RL32(ptr);
309
+    av_log(avctx, AV_LOG_DEBUG, "frame information flags %X\n", c->frame_info);
310
+
311
+    c->frame_pred = (c->frame_info >> 8) & 3;
312
+
313
+    if (c->frame_pred == PRED_GRADIENT) {
314
+        av_log_ask_for_sample(avctx, "Frame uses gradient prediction\n");
315
+        return AVERROR_PATCHWELCOME;
316
+    }
317
+
318
+    av_fast_malloc(&c->slice_bits, &c->slice_bits_size,
319
+                   max_slice_size + FF_INPUT_BUFFER_PADDING_SIZE);
320
+
321
+    if (!c->slice_bits) {
322
+        av_log(avctx, AV_LOG_ERROR, "Cannot allocate temporary buffer\n");
323
+        return AVERROR(ENOMEM);
324
+    }
325
+
326
+    switch (c->avctx->pix_fmt) {
327
+    case PIX_FMT_RGB24:
328
+    case PIX_FMT_RGBA:
329
+        for (i = 0; i < c->planes; i++) {
330
+            ret = decode_plane(c, i, c->pic.data[0] + rgb_order[i], c->planes,
331
+                               c->pic.linesize[0], avctx->width, avctx->height,
332
+                               plane_start[i], plane_start[i + 1] - plane_start[i],
333
+                               c->frame_pred == PRED_LEFT);
334
+            if (ret)
335
+                return ret;
336
+            if (c->frame_pred == PRED_MEDIAN)
337
+                restore_median(c->pic.data[0] + rgb_order[i], c->planes,
338
+                               c->pic.linesize[0], avctx->width, avctx->height,
339
+                               c->slices);
340
+        }
341
+        restore_rgb_planes(c->pic.data[0], c->planes, c->pic.linesize[0],
342
+                           avctx->width, avctx->height);
343
+        break;
344
+    case PIX_FMT_YUV420P:
345
+        for (i = 0; i < 3; i++) {
346
+            ret = decode_plane(c, i, c->pic.data[i], 1,
347
+                               c->pic.linesize[i], avctx->width >> !!i, avctx->height >> !!i,
348
+                               plane_start[i], plane_start[i + 1] - plane_start[i],
349
+                               c->frame_pred == PRED_LEFT);
350
+            if (ret)
351
+                return ret;
352
+            if (c->frame_pred == PRED_MEDIAN)
353
+                restore_median(c->pic.data[i], 1, c->pic.linesize[i],
354
+                               avctx->width >> !!i, avctx->height >> !!i,
355
+                               c->slices);
356
+        }
357
+        break;
358
+    case PIX_FMT_YUV422P:
359
+        for (i = 0; i < 3; i++) {
360
+            ret = decode_plane(c, i, c->pic.data[i], 1,
361
+                               c->pic.linesize[i], avctx->width >> !!i, avctx->height,
362
+                               plane_start[i], plane_start[i + 1] - plane_start[i],
363
+                               c->frame_pred == PRED_LEFT);
364
+            if (ret)
365
+                return ret;
366
+            if (c->frame_pred == PRED_MEDIAN)
367
+                restore_median(c->pic.data[i], 1, c->pic.linesize[i],
368
+                               avctx->width >> !!i, avctx->height, c->slices);
369
+        }
370
+        break;
371
+    }
372
+
373
+    *data_size = sizeof(AVFrame);
374
+    *(AVFrame*)data = c->pic;
375
+
376
+    /* always report that the buffer was completely consumed */
377
+    return buf_size;
378
+}
379
+
380
+static av_cold int decode_init(AVCodecContext *avctx)
381
+{
382
+    UtvideoContext * const c = avctx->priv_data;
383
+
384
+    c->avctx = avctx;
385
+
386
+    dsputil_init(&c->dsp, avctx);
387
+
388
+    if (avctx->extradata_size < 16) {
389
+        av_log(avctx, AV_LOG_ERROR, "Insufficient extradata size %d, should be at least 16\n",
390
+               avctx->extradata_size);
391
+        return AVERROR_INVALIDDATA;
392
+    }
393
+
394
+    av_log(avctx, AV_LOG_DEBUG, "Encoder version %d.%d.%d.%d\n",
395
+           avctx->extradata[3], avctx->extradata[2],
396
+           avctx->extradata[1], avctx->extradata[0]);
397
+    av_log(avctx, AV_LOG_DEBUG, "Original format %X\n", AV_RB32(avctx->extradata + 4));
398
+    c->frame_info_size = AV_RL32(avctx->extradata + 8);
399
+    c->flags           = AV_RL32(avctx->extradata + 12);
400
+
401
+    if (c->frame_info_size != 4)
402
+        av_log_ask_for_sample(avctx, "Frame info is not 4 bytes\n");
403
+    av_log(avctx, AV_LOG_DEBUG, "Encoding parameters %08X\n", c->flags);
404
+    c->slices      = (c->flags >> 24) + 1;
405
+    c->compression = c->flags & 1;
406
+    c->interlaced  = c->flags & 0x800;
407
+
408
+    c->slice_bits_size = 0;
409
+
410
+    switch (avctx->codec_tag) {
411
+    case MKTAG('U', 'L', 'R', 'G'):
412
+        c->planes      = 3;
413
+        avctx->pix_fmt = PIX_FMT_RGB24;
414
+        break;
415
+    case MKTAG('U', 'L', 'R', 'A'):
416
+        c->planes      = 4;
417
+        avctx->pix_fmt = PIX_FMT_RGBA;
418
+        break;
419
+    case MKTAG('U', 'L', 'Y', '0'):
420
+        c->planes      = 3;
421
+        avctx->pix_fmt = PIX_FMT_YUV420P;
422
+        break;
423
+    case MKTAG('U', 'L', 'Y', '2'):
424
+        c->planes      = 3;
425
+        avctx->pix_fmt = PIX_FMT_YUV422P;
426
+        break;
427
+    default:
428
+        av_log(avctx, AV_LOG_ERROR, "Unknown Ut Video FOURCC provided (%08X)\n",
429
+               avctx->codec_tag);
430
+        return AVERROR_INVALIDDATA;
431
+    }
432
+
433
+    return 0;
434
+}
435
+
436
+static av_cold int decode_end(AVCodecContext *avctx)
437
+{
438
+    UtvideoContext * const c = avctx->priv_data;
439
+
440
+    if (c->pic.data[0])
441
+        avctx->release_buffer(avctx, &c->pic);
442
+
443
+    av_freep(&c->slice_bits);
444
+
445
+    return 0;
446
+}
447
+
448
+AVCodec ff_utvideo_decoder = {
449
+    .name           = "utvideo",
450
+    .type           = AVMEDIA_TYPE_VIDEO,
451
+    .id             = CODEC_ID_UTVIDEO,
452
+    .priv_data_size = sizeof(UtvideoContext),
453
+    .init           = decode_init,
454
+    .close          = decode_end,
455
+    .decode         = decode_frame,
456
+    .capabilities   = CODEC_CAP_DR1,
457
+    .long_name      = NULL_IF_CONFIG_SMALL("Ut Video"),
458
+};
459
+
... ...
@@ -21,8 +21,8 @@
21 21
 #define AVCODEC_VERSION_H
22 22
 
23 23
 #define LIBAVCODEC_VERSION_MAJOR 53
24
-#define LIBAVCODEC_VERSION_MINOR  12
25
-#define LIBAVCODEC_VERSION_MICRO  2
24
+#define LIBAVCODEC_VERSION_MINOR  13
25
+#define LIBAVCODEC_VERSION_MICRO  0
26 26
 
27 27
 #define LIBAVCODEC_VERSION_INT  AV_VERSION_INT(LIBAVCODEC_VERSION_MAJOR, \
28 28
                                                LIBAVCODEC_VERSION_MINOR, \
... ...
@@ -275,6 +275,10 @@ const AVCodecTag ff_codec_bmp_tags[] = {
275 275
     { CODEC_ID_DPX,          MKTAG('d', 'p', 'x', ' ') },
276 276
     { CODEC_ID_KGV1,         MKTAG('K', 'G', 'V', '1') },
277 277
     { CODEC_ID_LAGARITH,     MKTAG('L', 'A', 'G', 'S') },
278
+    { CODEC_ID_UTVIDEO,      MKTAG('U', 'L', 'R', 'A') },
279
+    { CODEC_ID_UTVIDEO,      MKTAG('U', 'L', 'R', 'G') },
280
+    { CODEC_ID_UTVIDEO,      MKTAG('U', 'L', 'Y', '0') },
281
+    { CODEC_ID_UTVIDEO,      MKTAG('U', 'L', 'Y', '2') },
278 282
     { CODEC_ID_NONE,         0 }
279 283
 };
280 284