interplayvideo.c 31.7 KB
Newer Older
1 2 3 4
/*
 * Interplay MVE Video Decoder
 * Copyright (C) 2003 the ffmpeg project
 *
5 6 7
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
8 9
 * modify it under the terms of the GNU Lesser General Public
 * 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 14 15 16 17
 * 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
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
 * Interplay MVE Video Decoder by Mike Melanson (melanson@pcisys.net)
25
 * For more information about the Interplay MVE format, visit:
26 27 28 29 30 31 32 33 34
 *   http://www.pcisys.net/~melanson/codecs/interplay-mve.txt
 * This code is written in such a way that the identifiers match up
 * with the encoding descriptions in the document.
 *
 * This decoder presently only supports a PAL8 output colorspace.
 *
 * An Interplay video frame consists of 2 parts: The decoding map and
 * the video data. A demuxer must load these 2 parts together in a single
 * buffer before sending it through the stream to this decoder.
35 36 37 38 39 40 41
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "avcodec.h"
42
#include "bytestream.h"
43
#include "hpeldsp.h"
44
#define BITSTREAM_READER_LE
45
#include "get_bits.h"
46
#include "internal.h"
47

48 49
#define PALETTE_COUNT 256

50 51 52
typedef struct IpvideoContext {

    AVCodecContext *avctx;
53
    HpelDSPContext hdsp;
54 55
    AVFrame *second_last_frame;
    AVFrame *last_frame;
Michael Niedermayer's avatar
Michael Niedermayer committed
56
    const unsigned char *decoding_map;
57 58
    int decoding_map_size;

59
    int is_16bpp;
60
    GetByteContext stream_ptr, mv_ptr;
61 62 63 64 65
    unsigned char *pixel_ptr;
    int line_inc;
    int stride;
    int upper_motion_limit_offset;

66
    uint32_t pal[256];
67 68
} IpvideoContext;

69
static int copy_from(IpvideoContext *s, AVFrame *src, AVFrame *dst, int delta_x, int delta_y)
70
{
71 72
    int current_offset = s->pixel_ptr - dst->data[0];
    int motion_offset = current_offset + delta_y * dst->linesize[0]
73
                       + delta_x * (1 + s->is_16bpp);
74
    if (motion_offset < 0) {
75
        av_log(s->avctx, AV_LOG_ERROR, "motion offset < 0 (%d)\n", motion_offset);
76
        return AVERROR_INVALIDDATA;
77
    } else if (motion_offset > s->upper_motion_limit_offset) {
78
        av_log(s->avctx, AV_LOG_ERROR, "motion offset above limit (%d >= %d)\n",
79
            motion_offset, s->upper_motion_limit_offset);
80
        return AVERROR_INVALIDDATA;
81
    }
82 83 84 85
    if (src->data[0] == NULL) {
        av_log(s->avctx, AV_LOG_ERROR, "Invalid decode type, corrupted header?\n");
        return AVERROR(EINVAL);
    }
86 87
    s->hdsp.put_pixels_tab[!s->is_16bpp][0](s->pixel_ptr, src->data[0] + motion_offset,
                                            dst->linesize[0], 8);
88 89 90
    return 0;
}

91
static int ipvideo_decode_block_opcode_0x0(IpvideoContext *s, AVFrame *frame)
92
{
93
    return copy_from(s, s->last_frame, frame, 0, 0);
94
}
95

96
static int ipvideo_decode_block_opcode_0x1(IpvideoContext *s, AVFrame *frame)
97
{
98
    return copy_from(s, s->second_last_frame, frame, 0, 0);
99 100
}

101
static int ipvideo_decode_block_opcode_0x2(IpvideoContext *s, AVFrame *frame)
102 103 104 105
{
    unsigned char B;
    int x, y;

106
    /* copy block from 2 frames ago using a motion vector; need 1 more byte */
107
    if (!s->is_16bpp) {
108
        B = bytestream2_get_byte(&s->stream_ptr);
109
    } else {
110
        B = bytestream2_get_byte(&s->mv_ptr);
111
    }
112 113 114 115 116 117 118 119 120

    if (B < 56) {
        x = 8 + (B % 7);
        y = B / 7;
    } else {
        x = -14 + ((B - 56) % 29);
        y =   8 + ((B - 56) / 29);
    }

121
    av_dlog(s->avctx, "motion byte = %d, (x, y) = (%d, %d)\n", B, x, y);
122
    return copy_from(s, s->second_last_frame, frame, x, y);
123 124
}

125
static int ipvideo_decode_block_opcode_0x3(IpvideoContext *s, AVFrame *frame)
126 127 128 129 130 131 132
{
    unsigned char B;
    int x, y;

    /* copy 8x8 block from current frame from an up/left block */

    /* need 1 more byte for motion */
133
    if (!s->is_16bpp) {
134
        B = bytestream2_get_byte(&s->stream_ptr);
135
    } else {
136
        B = bytestream2_get_byte(&s->mv_ptr);
137
    }
138 139 140 141 142 143 144 145 146

    if (B < 56) {
        x = -(8 + (B % 7));
        y = -(B / 7);
    } else {
        x = -(-14 + ((B - 56) % 29));
        y = -(  8 + ((B - 56) / 29));
    }

147
    av_dlog(s->avctx, "motion byte = %d, (x, y) = (%d, %d)\n", B, x, y);
148
    return copy_from(s, frame, frame, x, y);
149 150
}

151
static int ipvideo_decode_block_opcode_0x4(IpvideoContext *s, AVFrame *frame)
152 153 154 155 156
{
    int x, y;
    unsigned char B, BL, BH;

    /* copy a block from the previous frame; need 1 more byte */
157
    if (!s->is_16bpp) {
158
        B = bytestream2_get_byte(&s->stream_ptr);
159
    } else {
160
        B = bytestream2_get_byte(&s->mv_ptr);
161
    }
162 163 164 165 166 167

    BL = B & 0x0F;
    BH = (B >> 4) & 0x0F;
    x = -8 + BL;
    y = -8 + BH;

168
    av_dlog(s->avctx, "motion byte = %d, (x, y) = (%d, %d)\n", B, x, y);
169
    return copy_from(s, s->last_frame, frame, x, y);
170 171
}

172
static int ipvideo_decode_block_opcode_0x5(IpvideoContext *s, AVFrame *frame)
173 174 175 176 177
{
    signed char x, y;

    /* copy a block from the previous frame using an expanded range;
     * need 2 more bytes */
178 179
    x = bytestream2_get_byte(&s->stream_ptr);
    y = bytestream2_get_byte(&s->stream_ptr);
180

181
    av_dlog(s->avctx, "motion bytes = %d, %d\n", x, y);
182
    return copy_from(s, s->last_frame, frame, x, y);
183 184
}

185
static int ipvideo_decode_block_opcode_0x6(IpvideoContext *s, AVFrame *frame)
186 187
{
    /* mystery opcode? skip multiple blocks? */
188
    av_log(s->avctx, AV_LOG_ERROR, "Help! Mystery opcode 0x6 seen\n");
189 190 191 192 193

    /* report success */
    return 0;
}

194
static int ipvideo_decode_block_opcode_0x7(IpvideoContext *s, AVFrame *frame)
195 196
{
    int x, y;
197
    unsigned char P[2];
198 199
    unsigned int flags;

200 201 202 203 204
    if (bytestream2_get_bytes_left(&s->stream_ptr) < 4) {
        av_log(s->avctx, AV_LOG_ERROR, "too little data for opcode 0x7\n");
        return AVERROR_INVALIDDATA;
    }

205
    /* 2-color encoding */
206 207
    P[0] = bytestream2_get_byte(&s->stream_ptr);
    P[1] = bytestream2_get_byte(&s->stream_ptr);
208

209
    if (P[0] <= P[1]) {
210 211 212

        /* need 8 more bytes from the stream */
        for (y = 0; y < 8; y++) {
213
            flags = bytestream2_get_byte(&s->stream_ptr) | 0x100;
214
            for (; flags != 1; flags >>= 1)
215
                *s->pixel_ptr++ = P[flags & 1];
216
            s->pixel_ptr += s->line_inc;
217 218 219 220 221
        }

    } else {

        /* need 2 more bytes from the stream */
222
        flags = bytestream2_get_le16(&s->stream_ptr);
223
        for (y = 0; y < 8; y += 2) {
224
            for (x = 0; x < 8; x += 2, flags >>= 1) {
Reimar Döffinger's avatar
Reimar Döffinger committed
225 226 227 228
                s->pixel_ptr[x                ] =
                s->pixel_ptr[x + 1            ] =
                s->pixel_ptr[x +     s->stride] =
                s->pixel_ptr[x + 1 + s->stride] = P[flags & 1];
229
            }
230
            s->pixel_ptr += s->stride * 2;
231 232 233 234 235 236 237
        }
    }

    /* report success */
    return 0;
}

238
static int ipvideo_decode_block_opcode_0x8(IpvideoContext *s, AVFrame *frame)
239 240
{
    int x, y;
241
    unsigned char P[4];
242 243
    unsigned int flags = 0;

244 245 246 247 248
    if (bytestream2_get_bytes_left(&s->stream_ptr) < 12) {
        av_log(s->avctx, AV_LOG_ERROR, "too little data for opcode 0x8\n");
        return AVERROR_INVALIDDATA;
    }

249 250
    /* 2-color encoding for each 4x4 quadrant, or 2-color encoding on
     * either top and bottom or left and right halves */
251 252
    P[0] = bytestream2_get_byte(&s->stream_ptr);
    P[1] = bytestream2_get_byte(&s->stream_ptr);
253 254

    if (P[0] <= P[1]) {
255 256 257
        for (y = 0; y < 16; y++) {
            // new values for each 4x4 block
            if (!(y & 3)) {
258 259 260 261 262
                if (y) {
                    P[0]  = bytestream2_get_byte(&s->stream_ptr);
                    P[1]  = bytestream2_get_byte(&s->stream_ptr);
                }
                flags = bytestream2_get_le16(&s->stream_ptr);
263 264
            }

265
            for (x = 0; x < 4; x++, flags >>= 1)
266 267 268 269
                *s->pixel_ptr++ = P[flags & 1];
            s->pixel_ptr += s->stride - 4;
            // switch to right half
            if (y == 7) s->pixel_ptr -= 8 * s->stride - 4;
270 271 272
        }

    } else {
273 274 275
        flags = bytestream2_get_le32(&s->stream_ptr);
        P[2] = bytestream2_get_byte(&s->stream_ptr);
        P[3] = bytestream2_get_byte(&s->stream_ptr);
276

277
        if (P[2] <= P[3]) {
278

279 280
            /* vertical split; left & right halves are 2-color encoded */

281
            for (y = 0; y < 16; y++) {
282
                for (x = 0; x < 4; x++, flags >>= 1)
283 284 285 286 287
                    *s->pixel_ptr++ = P[flags & 1];
                s->pixel_ptr += s->stride - 4;
                // switch to right half
                if (y == 7) {
                    s->pixel_ptr -= 8 * s->stride - 4;
288 289 290
                    P[0]  = P[2];
                    P[1]  = P[3];
                    flags = bytestream2_get_le32(&s->stream_ptr);
291 292 293 294 295 296 297 298
                }
            }

        } else {

            /* horizontal split; top & bottom halves are 2-color encoded */

            for (y = 0; y < 8; y++) {
299
                if (y == 4) {
300 301 302
                    P[0]  = P[2];
                    P[1]  = P[3];
                    flags = bytestream2_get_le32(&s->stream_ptr);
303 304
                }

305
                for (x = 0; x < 8; x++, flags >>= 1)
306
                    *s->pixel_ptr++ = P[flags & 1];
307
                s->pixel_ptr += s->line_inc;
308 309 310 311 312 313 314 315
            }
        }
    }

    /* report success */
    return 0;
}

316
static int ipvideo_decode_block_opcode_0x9(IpvideoContext *s, AVFrame *frame)
317 318 319 320
{
    int x, y;
    unsigned char P[4];

321 322 323 324 325
    if (bytestream2_get_bytes_left(&s->stream_ptr) < 8) {
        av_log(s->avctx, AV_LOG_ERROR, "too little data for opcode 0x9\n");
        return AVERROR_INVALIDDATA;
    }

326
    /* 4-color encoding */
327
    bytestream2_get_buffer(&s->stream_ptr, P, 4);
328

329 330
    if (P[0] <= P[1]) {
        if (P[2] <= P[3]) {
331

Reimar Döffinger's avatar
Reimar Döffinger committed
332 333 334
            /* 1 of 4 colors for each pixel, need 16 more bytes */
            for (y = 0; y < 8; y++) {
                /* get the next set of 8 2-bit flags */
335
                int flags = bytestream2_get_le16(&s->stream_ptr);
336
                for (x = 0; x < 8; x++, flags >>= 2)
Reimar Döffinger's avatar
Reimar Döffinger committed
337 338
                    *s->pixel_ptr++ = P[flags & 0x03];
                s->pixel_ptr += s->line_inc;
339 340
            }

341
        } else {
Reimar Döffinger's avatar
Reimar Döffinger committed
342
            uint32_t flags;
343

Reimar Döffinger's avatar
Reimar Döffinger committed
344
            /* 1 of 4 colors for each 2x2 block, need 4 more bytes */
345
            flags = bytestream2_get_le32(&s->stream_ptr);
346

Reimar Döffinger's avatar
Reimar Döffinger committed
347 348 349 350 351 352 353 354
            for (y = 0; y < 8; y += 2) {
                for (x = 0; x < 8; x += 2, flags >>= 2) {
                    s->pixel_ptr[x                ] =
                    s->pixel_ptr[x + 1            ] =
                    s->pixel_ptr[x +     s->stride] =
                    s->pixel_ptr[x + 1 + s->stride] = P[flags & 0x03];
                }
                s->pixel_ptr += s->stride * 2;
355 356
            }

357 358
        }
    } else {
359
        uint64_t flags;
360

361
        /* 1 of 4 colors for each 2x1 or 1x2 block, need 8 more bytes */
362
        flags = bytestream2_get_le64(&s->stream_ptr);
363
        if (P[2] <= P[3]) {
Reimar Döffinger's avatar
Reimar Döffinger committed
364 365 366 367 368 369
            for (y = 0; y < 8; y++) {
                for (x = 0; x < 8; x += 2, flags >>= 2) {
                    s->pixel_ptr[x    ] =
                    s->pixel_ptr[x + 1] = P[flags & 0x03];
                }
                s->pixel_ptr += s->stride;
370
            }
371
        } else {
Reimar Döffinger's avatar
Reimar Döffinger committed
372 373 374 375 376 377
            for (y = 0; y < 8; y += 2) {
                for (x = 0; x < 8; x++, flags >>= 2) {
                    s->pixel_ptr[x            ] =
                    s->pixel_ptr[x + s->stride] = P[flags & 0x03];
                }
                s->pixel_ptr += s->stride * 2;
378
            }
379
        }
380 381 382 383 384 385
    }

    /* report success */
    return 0;
}

386
static int ipvideo_decode_block_opcode_0xA(IpvideoContext *s, AVFrame *frame)
387 388
{
    int x, y;
389
    unsigned char P[8];
390 391
    int flags = 0;

392 393 394 395 396
    if (bytestream2_get_bytes_left(&s->stream_ptr) < 16) {
        av_log(s->avctx, AV_LOG_ERROR, "too little data for opcode 0xA\n");
        return AVERROR_INVALIDDATA;
    }

397 398
    bytestream2_get_buffer(&s->stream_ptr, P, 4);

399 400
    /* 4-color encoding for each 4x4 quadrant, or 4-color encoding on
     * either top and bottom or left and right halves */
401
    if (P[0] <= P[1]) {
402

403
        /* 4-color encoding for each quadrant; need 32 bytes */
404 405 406
        for (y = 0; y < 16; y++) {
            // new values for each 4x4 block
            if (!(y & 3)) {
407 408
                if (y) bytestream2_get_buffer(&s->stream_ptr, P, 4);
                flags = bytestream2_get_le32(&s->stream_ptr);
409
            }
410

411
            for (x = 0; x < 4; x++, flags >>= 2)
412
                *s->pixel_ptr++ = P[flags & 0x03];
413

414 415 416
            s->pixel_ptr += s->stride - 4;
            // switch to right half
            if (y == 7) s->pixel_ptr -= 8 * s->stride - 4;
417 418 419
        }

    } else {
420
        // vertical split?
421 422 423 424 425
        int vert;
        uint64_t flags = bytestream2_get_le64(&s->stream_ptr);

        bytestream2_get_buffer(&s->stream_ptr, P + 4, 4);
        vert = P[4] <= P[5];
426 427

        /* 4-color encoding for either left and right or top and bottom
428
         * halves */
429

430 431 432
        for (y = 0; y < 16; y++) {
            for (x = 0; x < 4; x++, flags >>= 2)
                *s->pixel_ptr++ = P[flags & 0x03];
433

434 435 436 437 438
            if (vert) {
                s->pixel_ptr += s->stride - 4;
                // switch to right half
                if (y == 7) s->pixel_ptr -= 8 * s->stride - 4;
            } else if (y & 1) s->pixel_ptr += s->line_inc;
439 440 441 442 443 444

            // load values for second half
            if (y == 7) {
                memcpy(P, P + 4, 4);
                flags = bytestream2_get_le64(&s->stream_ptr);
            }
445 446 447 448 449 450 451
        }
    }

    /* report success */
    return 0;
}

452
static int ipvideo_decode_block_opcode_0xB(IpvideoContext *s, AVFrame *frame)
453
{
454
    int y;
455 456 457

    /* 64-color encoding (each pixel in block is a different color) */
    for (y = 0; y < 8; y++) {
458
        bytestream2_get_buffer(&s->stream_ptr, s->pixel_ptr, 8);
459
        s->pixel_ptr  += s->stride;
460 461 462 463 464 465
    }

    /* report success */
    return 0;
}

466
static int ipvideo_decode_block_opcode_0xC(IpvideoContext *s, AVFrame *frame)
467 468 469 470 471 472
{
    int x, y;

    /* 16-color block encoding: each 2x2 block is a different color */
    for (y = 0; y < 8; y += 2) {
        for (x = 0; x < 8; x += 2) {
473 474 475
            s->pixel_ptr[x                ] =
            s->pixel_ptr[x + 1            ] =
            s->pixel_ptr[x +     s->stride] =
476
            s->pixel_ptr[x + 1 + s->stride] = bytestream2_get_byte(&s->stream_ptr);
477
        }
478
        s->pixel_ptr += s->stride * 2;
479 480 481 482 483 484
    }

    /* report success */
    return 0;
}

485
static int ipvideo_decode_block_opcode_0xD(IpvideoContext *s, AVFrame *frame)
486
{
487
    int y;
488
    unsigned char P[2];
489

490 491 492 493 494
    if (bytestream2_get_bytes_left(&s->stream_ptr) < 4) {
        av_log(s->avctx, AV_LOG_ERROR, "too little data for opcode 0xD\n");
        return AVERROR_INVALIDDATA;
    }

495
    /* 4-color block encoding: each 4x4 block is a different color */
496 497
    for (y = 0; y < 8; y++) {
        if (!(y & 3)) {
498 499
            P[0] = bytestream2_get_byte(&s->stream_ptr);
            P[1] = bytestream2_get_byte(&s->stream_ptr);
500 501 502
        }
        memset(s->pixel_ptr,     P[0], 4);
        memset(s->pixel_ptr + 4, P[1], 4);
503
        s->pixel_ptr += s->stride;
504 505 506 507 508 509
    }

    /* report success */
    return 0;
}

510
static int ipvideo_decode_block_opcode_0xE(IpvideoContext *s, AVFrame *frame)
511
{
512
    int y;
513 514 515
    unsigned char pix;

    /* 1-color encoding: the whole block is 1 solid color */
516
    pix = bytestream2_get_byte(&s->stream_ptr);
517 518

    for (y = 0; y < 8; y++) {
519 520
        memset(s->pixel_ptr, pix, 8);
        s->pixel_ptr += s->stride;
521 522 523 524 525 526
    }

    /* report success */
    return 0;
}

527
static int ipvideo_decode_block_opcode_0xF(IpvideoContext *s, AVFrame *frame)
528 529
{
    int x, y;
530
    unsigned char sample[2];
531 532

    /* dithered encoding */
533 534
    sample[0] = bytestream2_get_byte(&s->stream_ptr);
    sample[1] = bytestream2_get_byte(&s->stream_ptr);
535 536 537

    for (y = 0; y < 8; y++) {
        for (x = 0; x < 8; x += 2) {
538 539
            *s->pixel_ptr++ = sample[  y & 1 ];
            *s->pixel_ptr++ = sample[!(y & 1)];
540
        }
541
        s->pixel_ptr += s->line_inc;
542 543 544 545 546 547
    }

    /* report success */
    return 0;
}

548
static int ipvideo_decode_block_opcode_0x6_16(IpvideoContext *s, AVFrame *frame)
549 550 551 552
{
    signed char x, y;

    /* copy a block from the second last frame using an expanded range */
553 554
    x = bytestream2_get_byte(&s->stream_ptr);
    y = bytestream2_get_byte(&s->stream_ptr);
555

556
    av_dlog(s->avctx, "motion bytes = %d, %d\n", x, y);
557
    return copy_from(s, s->second_last_frame, frame, x, y);
558 559
}

560
static int ipvideo_decode_block_opcode_0x7_16(IpvideoContext *s, AVFrame *frame)
561 562 563 564 565 566 567
{
    int x, y;
    uint16_t P[2];
    unsigned int flags;
    uint16_t *pixel_ptr = (uint16_t*)s->pixel_ptr;

    /* 2-color encoding */
568 569
    P[0] = bytestream2_get_le16(&s->stream_ptr);
    P[1] = bytestream2_get_le16(&s->stream_ptr);
570 571 572 573

    if (!(P[0] & 0x8000)) {

        for (y = 0; y < 8; y++) {
574
            flags = bytestream2_get_byte(&s->stream_ptr) | 0x100;
575 576 577 578 579 580 581
            for (; flags != 1; flags >>= 1)
                *pixel_ptr++ = P[flags & 1];
            pixel_ptr += s->line_inc;
        }

    } else {

582
        flags = bytestream2_get_le16(&s->stream_ptr);
583 584 585 586 587 588 589 590 591 592 593 594 595 596
        for (y = 0; y < 8; y += 2) {
            for (x = 0; x < 8; x += 2, flags >>= 1) {
                pixel_ptr[x                ] =
                pixel_ptr[x + 1            ] =
                pixel_ptr[x +     s->stride] =
                pixel_ptr[x + 1 + s->stride] = P[flags & 1];
            }
            pixel_ptr += s->stride * 2;
        }
    }

    return 0;
}

597
static int ipvideo_decode_block_opcode_0x8_16(IpvideoContext *s, AVFrame *frame)
598 599
{
    int x, y;
600
    uint16_t P[4];
601 602 603 604 605
    unsigned int flags = 0;
    uint16_t *pixel_ptr = (uint16_t*)s->pixel_ptr;

    /* 2-color encoding for each 4x4 quadrant, or 2-color encoding on
     * either top and bottom or left and right halves */
606 607
    P[0] = bytestream2_get_le16(&s->stream_ptr);
    P[1] = bytestream2_get_le16(&s->stream_ptr);
608 609 610 611 612 613

    if (!(P[0] & 0x8000)) {

        for (y = 0; y < 16; y++) {
            // new values for each 4x4 block
            if (!(y & 3)) {
614 615 616 617 618
                if (y) {
                    P[0] = bytestream2_get_le16(&s->stream_ptr);
                    P[1] = bytestream2_get_le16(&s->stream_ptr);
                }
                flags = bytestream2_get_le16(&s->stream_ptr);
619 620 621 622 623 624 625 626 627 628 629
            }

            for (x = 0; x < 4; x++, flags >>= 1)
                *pixel_ptr++ = P[flags & 1];
            pixel_ptr += s->stride - 4;
            // switch to right half
            if (y == 7) pixel_ptr -= 8 * s->stride - 4;
        }

    } else {

630 631 632
        flags = bytestream2_get_le32(&s->stream_ptr);
        P[2]  = bytestream2_get_le16(&s->stream_ptr);
        P[3]  = bytestream2_get_le16(&s->stream_ptr);
633

634
        if (!(P[2] & 0x8000)) {
635 636 637 638 639 640 641 642 643 644

            /* vertical split; left & right halves are 2-color encoded */

            for (y = 0; y < 16; y++) {
                for (x = 0; x < 4; x++, flags >>= 1)
                    *pixel_ptr++ = P[flags & 1];
                pixel_ptr += s->stride - 4;
                // switch to right half
                if (y == 7) {
                    pixel_ptr -= 8 * s->stride - 4;
645 646 647
                    P[0]  = P[2];
                    P[1]  = P[3];
                    flags = bytestream2_get_le32(&s->stream_ptr);
648 649 650 651 652 653 654 655 656
                }
            }

        } else {

            /* horizontal split; top & bottom halves are 2-color encoded */

            for (y = 0; y < 8; y++) {
                if (y == 4) {
657 658 659
                    P[0]  = P[2];
                    P[1]  = P[3];
                    flags = bytestream2_get_le32(&s->stream_ptr);
660 661
                }

662
                for (x = 0; x < 8; x++, flags >>= 1)
663 664 665 666 667 668 669 670 671 672
                    *pixel_ptr++ = P[flags & 1];
                pixel_ptr += s->line_inc;
            }
        }
    }

    /* report success */
    return 0;
}

673
static int ipvideo_decode_block_opcode_0x9_16(IpvideoContext *s, AVFrame *frame)
674 675 676 677 678 679 680
{
    int x, y;
    uint16_t P[4];
    uint16_t *pixel_ptr = (uint16_t*)s->pixel_ptr;

    /* 4-color encoding */
    for (x = 0; x < 4; x++)
681
        P[x] = bytestream2_get_le16(&s->stream_ptr);
682 683 684 685 686 687 688

    if (!(P[0] & 0x8000)) {
        if (!(P[2] & 0x8000)) {

            /* 1 of 4 colors for each pixel */
            for (y = 0; y < 8; y++) {
                /* get the next set of 8 2-bit flags */
689
                int flags = bytestream2_get_le16(&s->stream_ptr);
690 691 692 693 694 695 696 697 698
                for (x = 0; x < 8; x++, flags >>= 2)
                    *pixel_ptr++ = P[flags & 0x03];
                pixel_ptr += s->line_inc;
            }

        } else {
            uint32_t flags;

            /* 1 of 4 colors for each 2x2 block */
699
            flags = bytestream2_get_le32(&s->stream_ptr);
700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715

            for (y = 0; y < 8; y += 2) {
                for (x = 0; x < 8; x += 2, flags >>= 2) {
                    pixel_ptr[x                ] =
                    pixel_ptr[x + 1            ] =
                    pixel_ptr[x +     s->stride] =
                    pixel_ptr[x + 1 + s->stride] = P[flags & 0x03];
                }
                pixel_ptr += s->stride * 2;
            }

        }
    } else {
        uint64_t flags;

        /* 1 of 4 colors for each 2x1 or 1x2 block */
716
        flags = bytestream2_get_le64(&s->stream_ptr);
717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739
        if (!(P[2] & 0x8000)) {
            for (y = 0; y < 8; y++) {
                for (x = 0; x < 8; x += 2, flags >>= 2) {
                    pixel_ptr[x    ] =
                    pixel_ptr[x + 1] = P[flags & 0x03];
                }
                pixel_ptr += s->stride;
            }
        } else {
            for (y = 0; y < 8; y += 2) {
                for (x = 0; x < 8; x++, flags >>= 2) {
                    pixel_ptr[x            ] =
                    pixel_ptr[x + s->stride] = P[flags & 0x03];
                }
                pixel_ptr += s->stride * 2;
            }
        }
    }

    /* report success */
    return 0;
}

740
static int ipvideo_decode_block_opcode_0xA_16(IpvideoContext *s, AVFrame *frame)
741 742
{
    int x, y;
743
    uint16_t P[8];
744 745 746
    int flags = 0;
    uint16_t *pixel_ptr = (uint16_t*)s->pixel_ptr;

747 748 749
    for (x = 0; x < 4; x++)
        P[x] = bytestream2_get_le16(&s->stream_ptr);

750 751
    /* 4-color encoding for each 4x4 quadrant, or 4-color encoding on
     * either top and bottom or left and right halves */
752
    if (!(P[0] & 0x8000)) {
753 754 755 756 757

        /* 4-color encoding for each quadrant */
        for (y = 0; y < 16; y++) {
            // new values for each 4x4 block
            if (!(y & 3)) {
758 759 760 761
                if (y)
                    for (x = 0; x < 4; x++)
                        P[x] = bytestream2_get_le16(&s->stream_ptr);
                flags = bytestream2_get_le32(&s->stream_ptr);
762 763 764 765 766 767 768 769 770 771 772 773
            }

            for (x = 0; x < 4; x++, flags >>= 2)
                *pixel_ptr++ = P[flags & 0x03];

            pixel_ptr += s->stride - 4;
            // switch to right half
            if (y == 7) pixel_ptr -= 8 * s->stride - 4;
        }

    } else {
        // vertical split?
774 775 776 777 778 779
        int vert;
        uint64_t flags = bytestream2_get_le64(&s->stream_ptr);

        for (x = 4; x < 8; x++)
            P[x] = bytestream2_get_le16(&s->stream_ptr);
        vert = !(P[4] & 0x8000);
780 781 782 783 784 785 786 787 788 789 790 791 792

        /* 4-color encoding for either left and right or top and bottom
         * halves */

        for (y = 0; y < 16; y++) {
            for (x = 0; x < 4; x++, flags >>= 2)
                *pixel_ptr++ = P[flags & 0x03];

            if (vert) {
                pixel_ptr += s->stride - 4;
                // switch to right half
                if (y == 7) pixel_ptr -= 8 * s->stride - 4;
            } else if (y & 1) pixel_ptr += s->line_inc;
793 794 795 796 797 798

            // load values for second half
            if (y == 7) {
                memcpy(P, P + 4, 8);
                flags = bytestream2_get_le64(&s->stream_ptr);
            }
799 800 801 802 803 804 805
        }
    }

    /* report success */
    return 0;
}

806
static int ipvideo_decode_block_opcode_0xB_16(IpvideoContext *s, AVFrame *frame)
807 808 809 810 811 812 813
{
    int x, y;
    uint16_t *pixel_ptr = (uint16_t*)s->pixel_ptr;

    /* 64-color encoding (each pixel in block is a different color) */
    for (y = 0; y < 8; y++) {
        for (x = 0; x < 8; x++)
814
            pixel_ptr[x] = bytestream2_get_le16(&s->stream_ptr);
815 816 817 818 819 820 821
        pixel_ptr  += s->stride;
    }

    /* report success */
    return 0;
}

822
static int ipvideo_decode_block_opcode_0xC_16(IpvideoContext *s, AVFrame *frame)
823 824 825 826 827 828 829 830 831 832
{
    int x, y;
    uint16_t *pixel_ptr = (uint16_t*)s->pixel_ptr;

    /* 16-color block encoding: each 2x2 block is a different color */
    for (y = 0; y < 8; y += 2) {
        for (x = 0; x < 8; x += 2) {
            pixel_ptr[x                ] =
            pixel_ptr[x + 1            ] =
            pixel_ptr[x +     s->stride] =
833
            pixel_ptr[x + 1 + s->stride] = bytestream2_get_le16(&s->stream_ptr);
834 835 836 837 838 839 840 841
        }
        pixel_ptr += s->stride * 2;
    }

    /* report success */
    return 0;
}

842
static int ipvideo_decode_block_opcode_0xD_16(IpvideoContext *s, AVFrame *frame)
843 844 845 846 847 848 849 850
{
    int x, y;
    uint16_t P[2];
    uint16_t *pixel_ptr = (uint16_t*)s->pixel_ptr;

    /* 4-color block encoding: each 4x4 block is a different color */
    for (y = 0; y < 8; y++) {
        if (!(y & 3)) {
851 852
            P[0] = bytestream2_get_le16(&s->stream_ptr);
            P[1] = bytestream2_get_le16(&s->stream_ptr);
853 854 855 856 857 858 859 860 861 862
        }
        for (x = 0; x < 8; x++)
            pixel_ptr[x] = P[x >> 2];
        pixel_ptr += s->stride;
    }

    /* report success */
    return 0;
}

863
static int ipvideo_decode_block_opcode_0xE_16(IpvideoContext *s, AVFrame *frame)
864 865 866 867 868 869
{
    int x, y;
    uint16_t pix;
    uint16_t *pixel_ptr = (uint16_t*)s->pixel_ptr;

    /* 1-color encoding: the whole block is 1 solid color */
870
    pix = bytestream2_get_le16(&s->stream_ptr);
871 872 873 874 875 876 877 878 879 880 881

    for (y = 0; y < 8; y++) {
        for (x = 0; x < 8; x++)
            pixel_ptr[x] = pix;
        pixel_ptr += s->stride;
    }

    /* report success */
    return 0;
}

882
static int (* const ipvideo_decode_block[])(IpvideoContext *s, AVFrame *frame) = {
883 884 885 886 887 888 889 890 891
    ipvideo_decode_block_opcode_0x0, ipvideo_decode_block_opcode_0x1,
    ipvideo_decode_block_opcode_0x2, ipvideo_decode_block_opcode_0x3,
    ipvideo_decode_block_opcode_0x4, ipvideo_decode_block_opcode_0x5,
    ipvideo_decode_block_opcode_0x6, ipvideo_decode_block_opcode_0x7,
    ipvideo_decode_block_opcode_0x8, ipvideo_decode_block_opcode_0x9,
    ipvideo_decode_block_opcode_0xA, ipvideo_decode_block_opcode_0xB,
    ipvideo_decode_block_opcode_0xC, ipvideo_decode_block_opcode_0xD,
    ipvideo_decode_block_opcode_0xE, ipvideo_decode_block_opcode_0xF,
};
892

893
static int (* const ipvideo_decode_block16[])(IpvideoContext *s, AVFrame *frame) = {
894 895 896 897 898 899 900 901 902 903
    ipvideo_decode_block_opcode_0x0,    ipvideo_decode_block_opcode_0x1,
    ipvideo_decode_block_opcode_0x2,    ipvideo_decode_block_opcode_0x3,
    ipvideo_decode_block_opcode_0x4,    ipvideo_decode_block_opcode_0x5,
    ipvideo_decode_block_opcode_0x6_16, ipvideo_decode_block_opcode_0x7_16,
    ipvideo_decode_block_opcode_0x8_16, ipvideo_decode_block_opcode_0x9_16,
    ipvideo_decode_block_opcode_0xA_16, ipvideo_decode_block_opcode_0xB_16,
    ipvideo_decode_block_opcode_0xC_16, ipvideo_decode_block_opcode_0xD_16,
    ipvideo_decode_block_opcode_0xE_16, ipvideo_decode_block_opcode_0x1,
};

904
static void ipvideo_decode_opcodes(IpvideoContext *s, AVFrame *frame)
905 906 907 908
{
    int x, y;
    unsigned char opcode;
    int ret;
909
    GetBitContext gb;
910

911
    bytestream2_skip(&s->stream_ptr, 14); /* data starts 14 bytes in */
912
    if (!s->is_16bpp) {
913
        /* this is PAL8, so make the palette available */
914
        memcpy(frame->data[1], s->pal, AVPALETTE_SIZE);
915

916
        s->stride = frame->linesize[0];
917
    } else {
918
        s->stride = frame->linesize[0] >> 1;
919 920
        s->mv_ptr = s->stream_ptr;
        bytestream2_skip(&s->mv_ptr, bytestream2_get_le16(&s->stream_ptr));
921
    }
922
    s->line_inc = s->stride - 8;
923
    s->upper_motion_limit_offset = (s->avctx->height - 8) * frame->linesize[0]
924
                                  + (s->avctx->width - 8) * (1 + s->is_16bpp);
925

926
    init_get_bits(&gb, s->decoding_map, s->decoding_map_size * 8);
927 928
    for (y = 0; y < s->avctx->height; y += 8) {
        for (x = 0; x < s->avctx->width; x += 8) {
929
            opcode = get_bits(&gb, 4);
930

931 932 933
            av_dlog(s->avctx,
                    "  block @ (%3d, %3d): encoding 0x%X, data ptr offset %d\n",
                    x, y, opcode, bytestream2_tell(&s->stream_ptr));
934

935
            if (!s->is_16bpp) {
936 937 938
                s->pixel_ptr = frame->data[0] + x
                              + y*frame->linesize[0];
                ret = ipvideo_decode_block[opcode](s, frame);
939
            } else {
940 941 942
                s->pixel_ptr = frame->data[0] + x*2
                              + y*frame->linesize[0];
                ret = ipvideo_decode_block16[opcode](s, frame);
943
            }
944
            if (ret != 0) {
945
                av_log(s->avctx, AV_LOG_ERROR, "decode problem on frame %d, @ block (%d, %d)\n",
946
                       s->avctx->frame_number, x, y);
947
                return;
948 949 950
            }
        }
    }
951 952
    if (bytestream2_get_bytes_left(&s->stream_ptr) > 1) {
        av_log(s->avctx, AV_LOG_ERROR,
953
               "decode finished with %d bytes left over\n",
954
               bytestream2_get_bytes_left(&s->stream_ptr));
955
    }
956 957
}

958
static av_cold int ipvideo_decode_init(AVCodecContext *avctx)
959 960 961 962
{
    IpvideoContext *s = avctx->priv_data;

    s->avctx = avctx;
963

964
    s->is_16bpp = avctx->bits_per_coded_sample == 16;
965
    avctx->pix_fmt = s->is_16bpp ? AV_PIX_FMT_RGB555 : AV_PIX_FMT_PAL8;
966

967
    ff_hpeldsp_init(&s->hdsp, avctx->flags);
968

969 970 971 972 973 974 975
    s->last_frame        = av_frame_alloc();
    s->second_last_frame = av_frame_alloc();
    if (!s->last_frame || !s->second_last_frame) {
        av_frame_free(&s->last_frame);
        av_frame_free(&s->second_last_frame);
        return AVERROR(ENOMEM);
    }
976

977 978 979 980
    return 0;
}

static int ipvideo_decode_frame(AVCodecContext *avctx,
981
                                void *data, int *got_frame,
982
                                AVPacket *avpkt)
983
{
984 985
    const uint8_t *buf = avpkt->data;
    int buf_size = avpkt->size;
986
    IpvideoContext *s = avctx->priv_data;
987
    AVFrame *frame = data;
988
    int ret;
989

990 991 992
    /* decoding map contains 4 bits of information per 8x8 block */
    s->decoding_map_size = avctx->width * avctx->height / (8 * 8 * 2);

993 994 995 996 997
    /* compressed buffer needs to be large enough to at least hold an entire
     * decoding map */
    if (buf_size < s->decoding_map_size)
        return buf_size;

998 999 1000
    if (av_packet_get_side_data(avpkt, AV_PKT_DATA_PARAM_CHANGE, NULL)) {
        av_frame_unref(s->last_frame);
        av_frame_unref(s->second_last_frame);
1001 1002
    }

1003
    s->decoding_map = buf;
1004 1005
    bytestream2_init(&s->stream_ptr, buf + s->decoding_map_size,
                     buf_size - s->decoding_map_size);
1006

1007
    if ((ret = ff_get_buffer(avctx, frame, AV_GET_BUFFER_FLAG_REF)) < 0)
1008
        return ret;
1009

1010 1011 1012
    if (!s->is_16bpp) {
        const uint8_t *pal = av_packet_get_side_data(avpkt, AV_PKT_DATA_PALETTE, NULL);
        if (pal) {
1013
            frame->palette_has_changed = 1;
1014 1015
            memcpy(s->pal, pal, AVPALETTE_SIZE);
        }
1016 1017
    }

1018
    ipvideo_decode_opcodes(s, frame);
1019

1020
    *got_frame = 1;
1021

1022
    /* shuffle frames */
1023 1024 1025 1026
    av_frame_unref(s->second_last_frame);
    FFSWAP(AVFrame*, s->second_last_frame, s->last_frame);
    if ((ret = av_frame_ref(s->last_frame, frame)) < 0)
        return ret;
1027

1028
    /* report that the buffer was completely consumed */
1029 1030 1031
    return buf_size;
}

1032
static av_cold int ipvideo_decode_end(AVCodecContext *avctx)
1033 1034 1035
{
    IpvideoContext *s = avctx->priv_data;

1036 1037
    av_frame_free(&s->last_frame);
    av_frame_free(&s->second_last_frame);
1038 1039 1040 1041

    return 0;
}

1042
AVCodec ff_interplay_video_decoder = {
1043
    .name           = "interplayvideo",
1044
    .long_name      = NULL_IF_CONFIG_SMALL("Interplay MVE video"),
1045
    .type           = AVMEDIA_TYPE_VIDEO,
1046
    .id             = AV_CODEC_ID_INTERPLAY_VIDEO,
1047 1048 1049 1050
    .priv_data_size = sizeof(IpvideoContext),
    .init           = ipvideo_decode_init,
    .close          = ipvideo_decode_end,
    .decode         = ipvideo_decode_frame,
1051
    .capabilities   = CODEC_CAP_DR1 | CODEC_CAP_PARAM_CHANGE,
1052
};