xref: /third_party/ffmpeg/libavcodec/indeo2.c (revision cabdff1a) 
1/*
2 * Intel Indeo 2 codec
3 * Copyright (c) 2005 Konstantin Shishkov
4 *
5 * This file is part of FFmpeg.
6 *
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
11 *
12 * FFmpeg is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 */
21
22/**
23 * @file
24 * Intel Indeo 2 decoder.
25 */
26
27#include "libavutil/attributes.h"
28#include "libavutil/thread.h"
29
30#define BITSTREAM_READER_LE
31#include "avcodec.h"
32#include "codec_internal.h"
33#include "get_bits.h"
34#include "indeo2data.h"
35#include "internal.h"
36#include "mathops.h"
37
38typedef struct Ir2Context{
39    AVCodecContext *avctx;
40    AVFrame *picture;
41    GetBitContext gb;
42    int decode_delta;
43} Ir2Context;
44
45#define CODE_VLC_BITS 14
46static VLC ir2_vlc;
47
48/* Indeo 2 codes are in range 0x01..0x7F and 0x81..0x90 */
49static inline int ir2_get_code(GetBitContext *gb)
50{
51    return get_vlc2(gb, ir2_vlc.table, CODE_VLC_BITS, 1);
52}
53
54static int ir2_decode_plane(Ir2Context *ctx, int width, int height, uint8_t *dst,
55                            int pitch, const uint8_t *table)
56{
57    int i;
58    int j;
59    int out = 0;
60
61    if ((width & 1) || width * height / (2*(IR2_CODES - 0x7F)) > get_bits_left(&ctx->gb))
62        return AVERROR_INVALIDDATA;
63
64    /* first line contain absolute values, other lines contain deltas */
65    while (out < width) {
66        int c = ir2_get_code(&ctx->gb);
67        if (c >= 0x80) { /* we have a run */
68            c -= 0x7F;
69            if (out + c*2 > width)
70                return AVERROR_INVALIDDATA;
71            for (i = 0; i < c * 2; i++)
72                dst[out++] = 0x80;
73        } else { /* copy two values from table */
74            if (c <= 0)
75                return AVERROR_INVALIDDATA;
76            dst[out++] = table[c * 2];
77            dst[out++] = table[(c * 2) + 1];
78        }
79    }
80    dst += pitch;
81
82    for (j = 1; j < height; j++) {
83        out = 0;
84        while (out < width) {
85            int c;
86            if (get_bits_left(&ctx->gb) <= 0)
87                return AVERROR_INVALIDDATA;
88            c = ir2_get_code(&ctx->gb);
89            if (c >= 0x80) { /* we have a skip */
90                c -= 0x7F;
91                if (out + c*2 > width)
92                    return AVERROR_INVALIDDATA;
93                for (i = 0; i < c * 2; i++) {
94                    dst[out] = dst[out - pitch];
95                    out++;
96                }
97            } else { /* add two deltas from table */
98                int t;
99                if (c <= 0)
100                    return AVERROR_INVALIDDATA;
101                t        = dst[out - pitch] + (table[c * 2] - 128);
102                t        = av_clip_uint8(t);
103                dst[out] = t;
104                out++;
105                t        = dst[out - pitch] + (table[(c * 2) + 1] - 128);
106                t        = av_clip_uint8(t);
107                dst[out] = t;
108                out++;
109            }
110        }
111        dst += pitch;
112    }
113    return 0;
114}
115
116static int ir2_decode_plane_inter(Ir2Context *ctx, int width, int height, uint8_t *dst,
117                                  int pitch, const uint8_t *table)
118{
119    int j;
120    int out = 0;
121    int c;
122    int t;
123
124    if (width & 1)
125        return AVERROR_INVALIDDATA;
126
127    for (j = 0; j < height; j++) {
128        out = 0;
129        while (out < width) {
130            if (get_bits_left(&ctx->gb) <= 0)
131                return AVERROR_INVALIDDATA;
132            c = ir2_get_code(&ctx->gb);
133            if (c >= 0x80) { /* we have a skip */
134                c   -= 0x7F;
135                out += c * 2;
136            } else { /* add two deltas from table */
137                if (c <= 0)
138                    return AVERROR_INVALIDDATA;
139                t        = dst[out] + (((table[c * 2] - 128)*3) >> 2);
140                t        = av_clip_uint8(t);
141                dst[out] = t;
142                out++;
143                t        = dst[out] + (((table[(c * 2) + 1] - 128)*3) >> 2);
144                t        = av_clip_uint8(t);
145                dst[out] = t;
146                out++;
147            }
148        }
149        dst += pitch;
150    }
151    return 0;
152}
153
154static int ir2_decode_frame(AVCodecContext *avctx, AVFrame *picture,
155                            int *got_frame, AVPacket *avpkt)
156{
157    Ir2Context * const s = avctx->priv_data;
158    const uint8_t *buf   = avpkt->data;
159    int buf_size         = avpkt->size;
160    AVFrame * const p    = s->picture;
161    int start, ret;
162    int ltab, ctab;
163
164    if ((ret = ff_reget_buffer(avctx, p, 0)) < 0)
165        return ret;
166
167    start = 48; /* hardcoded for now */
168
169    if (start >= buf_size) {
170        av_log(s->avctx, AV_LOG_ERROR, "input buffer size too small (%d)\n", buf_size);
171        return AVERROR_INVALIDDATA;
172    }
173
174    s->decode_delta = buf[18];
175
176    /* decide whether frame uses deltas or not */
177
178    if ((ret = init_get_bits8(&s->gb, buf + start, buf_size - start)) < 0)
179        return ret;
180
181    ltab = buf[0x22] & 3;
182    ctab = buf[0x22] >> 2;
183
184    if (ctab > 3) {
185        av_log(avctx, AV_LOG_ERROR, "ctab %d is invalid\n", ctab);
186        return AVERROR_INVALIDDATA;
187    }
188
189    if (s->decode_delta) { /* intraframe */
190        if ((ret = ir2_decode_plane(s, avctx->width, avctx->height,
191                                    p->data[0], p->linesize[0],
192                                    ir2_delta_table[ltab])) < 0)
193            return ret;
194
195        /* swapped U and V */
196        if ((ret = ir2_decode_plane(s, avctx->width >> 2, avctx->height >> 2,
197                                    p->data[2], p->linesize[2],
198                                    ir2_delta_table[ctab])) < 0)
199            return ret;
200        if ((ret = ir2_decode_plane(s, avctx->width >> 2, avctx->height >> 2,
201                                    p->data[1], p->linesize[1],
202                                    ir2_delta_table[ctab])) < 0)
203            return ret;
204    } else { /* interframe */
205        if ((ret = ir2_decode_plane_inter(s, avctx->width, avctx->height,
206                                          p->data[0], p->linesize[0],
207                                          ir2_delta_table[ltab])) < 0)
208            return ret;
209        /* swapped U and V */
210        if ((ret = ir2_decode_plane_inter(s, avctx->width >> 2, avctx->height >> 2,
211                                          p->data[2], p->linesize[2],
212                                          ir2_delta_table[ctab])) < 0)
213            return ret;
214        if ((ret = ir2_decode_plane_inter(s, avctx->width >> 2, avctx->height >> 2,
215                                          p->data[1], p->linesize[1],
216                                          ir2_delta_table[ctab])) < 0)
217            return ret;
218    }
219
220    if ((ret = av_frame_ref(picture, p)) < 0)
221        return ret;
222
223    *got_frame = 1;
224
225    return buf_size;
226}
227
228static av_cold void ir2_init_static(void)
229{
230    INIT_VLC_STATIC_FROM_LENGTHS(&ir2_vlc, CODE_VLC_BITS, IR2_CODES,
231                                 &ir2_tab[0][1], 2, &ir2_tab[0][0], 2, 1,
232                                 0, INIT_VLC_OUTPUT_LE, 1 << CODE_VLC_BITS);
233}
234
235static av_cold int ir2_decode_init(AVCodecContext *avctx)
236{
237    static AVOnce init_static_once = AV_ONCE_INIT;
238    Ir2Context * const ic = avctx->priv_data;
239
240    ic->avctx = avctx;
241
242    avctx->pix_fmt= AV_PIX_FMT_YUV410P;
243
244    ic->picture = av_frame_alloc();
245    if (!ic->picture)
246        return AVERROR(ENOMEM);
247
248    ff_thread_once(&init_static_once, ir2_init_static);
249
250    return 0;
251}
252
253static av_cold int ir2_decode_end(AVCodecContext *avctx)
254{
255    Ir2Context * const ic = avctx->priv_data;
256
257    av_frame_free(&ic->picture);
258
259    return 0;
260}
261
262const FFCodec ff_indeo2_decoder = {
263    .p.name         = "indeo2",
264    .p.long_name    = NULL_IF_CONFIG_SMALL("Intel Indeo 2"),
265    .p.type         = AVMEDIA_TYPE_VIDEO,
266    .p.id           = AV_CODEC_ID_INDEO2,
267    .priv_data_size = sizeof(Ir2Context),
268    .init           = ir2_decode_init,
269    .close          = ir2_decode_end,
270    FF_CODEC_DECODE_CB(ir2_decode_frame),
271    .p.capabilities = AV_CODEC_CAP_DR1,
272    .caps_internal  = FF_CODEC_CAP_INIT_THREADSAFE,
273};
274