interplayvideo.c 32.5 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 "dsputil.h"
44 45
#define ALT_BITSTREAM_READER_LE
#include "get_bits.h"
46

47 48 49 50 51
#define PALETTE_COUNT 256

/* debugging support */
#define DEBUG_INTERPLAY 0
#if DEBUG_INTERPLAY
Alex Beregszaszi's avatar
Alex Beregszaszi committed
52
#define debug_interplay(x,...) av_log(NULL, AV_LOG_DEBUG, x, __VA_ARGS__)
53 54 55 56
#else
static inline void debug_interplay(const char *format, ...) { }
#endif

57 58 59 60
typedef struct IpvideoContext {

    AVCodecContext *avctx;
    DSPContext dsp;
61
    AVFrame second_last_frame;
62 63
    AVFrame last_frame;
    AVFrame current_frame;
Michael Niedermayer's avatar
Michael Niedermayer committed
64
    const unsigned char *decoding_map;
65 66
    int decoding_map_size;

Michael Niedermayer's avatar
Michael Niedermayer committed
67
    const unsigned char *buf;
68 69
    int size;

70
    int is_16bpp;
Michael Niedermayer's avatar
Michael Niedermayer committed
71 72
    const unsigned char *stream_ptr;
    const unsigned char *stream_end;
73 74
    const uint8_t *mv_ptr;
    const uint8_t *mv_end;
75 76 77 78 79
    unsigned char *pixel_ptr;
    int line_inc;
    int stride;
    int upper_motion_limit_offset;

80 81
} IpvideoContext;

82 83
#define CHECK_STREAM_PTR(stream_ptr, stream_end, n) \
    if (stream_end - stream_ptr < n) { \
Kostya Shishkov's avatar
Kostya Shishkov committed
84
        av_log(s->avctx, AV_LOG_ERROR, "Interplay video warning: stream_ptr out of bounds (%p >= %p)\n", \
85
               stream_ptr + n, stream_end); \
Kostya Shishkov's avatar
Kostya Shishkov committed
86 87
        return -1; \
    }
88

89
static int copy_from(IpvideoContext *s, AVFrame *src, int delta_x, int delta_y)
90 91
{
    int current_offset = s->pixel_ptr - s->current_frame.data[0];
92 93
    int motion_offset = current_offset + delta_y * s->current_frame.linesize[0]
                       + delta_x * (1 + s->is_16bpp);
94 95 96 97 98 99 100 101
    if (motion_offset < 0) {
        av_log(s->avctx, AV_LOG_ERROR, " Interplay video: motion offset < 0 (%d)\n", motion_offset);
        return -1;
    } else if (motion_offset > s->upper_motion_limit_offset) {
        av_log(s->avctx, AV_LOG_ERROR, " Interplay video: motion offset above limit (%d >= %d)\n",
            motion_offset, s->upper_motion_limit_offset);
        return -1;
    }
102 103
    s->dsp.put_pixels_tab[!s->is_16bpp][0](s->pixel_ptr, src->data[0] + motion_offset,
                                           s->current_frame.linesize[0], 8);
104 105 106
    return 0;
}

107
static int ipvideo_decode_block_opcode_0x0(IpvideoContext *s)
108
{
109 110
    return copy_from(s, &s->last_frame, 0, 0);
}
111

112 113 114
static int ipvideo_decode_block_opcode_0x1(IpvideoContext *s)
{
    return copy_from(s, &s->second_last_frame, 0, 0);
115 116 117
}

static int ipvideo_decode_block_opcode_0x2(IpvideoContext *s)
118 119 120 121
{
    unsigned char B;
    int x, y;

122
    /* copy block from 2 frames ago using a motion vector; need 1 more byte */
123 124 125 126 127 128 129
    if (!s->is_16bpp) {
        CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 1);
        B = *s->stream_ptr++;
    } else {
        CHECK_STREAM_PTR(s->mv_ptr, s->mv_end, 1);
        B = *s->mv_ptr++;
    }
130 131 132 133 134 135 136 137 138 139

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

    debug_interplay ("    motion byte = %d, (x, y) = (%d, %d)\n", B, x, y);
140
    return copy_from(s, &s->second_last_frame, x, y);
141 142
}

143
static int ipvideo_decode_block_opcode_0x3(IpvideoContext *s)
144 145 146 147 148 149 150
{
    unsigned char B;
    int x, y;

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

    /* need 1 more byte for motion */
151 152 153 154 155 156 157
    if (!s->is_16bpp) {
        CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 1);
        B = *s->stream_ptr++;
    } else {
        CHECK_STREAM_PTR(s->mv_ptr, s->mv_end, 1);
        B = *s->mv_ptr++;
    }
158 159 160 161 162 163 164 165 166 167

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

    debug_interplay ("    motion byte = %d, (x, y) = (%d, %d)\n", B, x, y);
168
    return copy_from(s, &s->current_frame, x, y);
169 170
}

171
static int ipvideo_decode_block_opcode_0x4(IpvideoContext *s)
172 173 174 175 176
{
    int x, y;
    unsigned char B, BL, BH;

    /* copy a block from the previous frame; need 1 more byte */
177 178 179 180 181 182 183
    if (!s->is_16bpp) {
        CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 1);
        B = *s->stream_ptr++;
    } else {
        CHECK_STREAM_PTR(s->mv_ptr, s->mv_end, 1);
        B = *s->mv_ptr++;
    }
184 185 186 187 188 189 190

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

    debug_interplay ("    motion byte = %d, (x, y) = (%d, %d)\n", B, x, y);
191
    return copy_from(s, &s->last_frame, x, y);
192 193
}

194
static int ipvideo_decode_block_opcode_0x5(IpvideoContext *s)
195 196 197 198 199
{
    signed char x, y;

    /* copy a block from the previous frame using an expanded range;
     * need 2 more bytes */
200
    CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 2);
201

202 203
    x = *s->stream_ptr++;
    y = *s->stream_ptr++;
204 205

    debug_interplay ("    motion bytes = %d, %d\n", x, y);
206
    return copy_from(s, &s->last_frame, x, y);
207 208
}

209
static int ipvideo_decode_block_opcode_0x6(IpvideoContext *s)
210 211
{
    /* mystery opcode? skip multiple blocks? */
212
    av_log(s->avctx, AV_LOG_ERROR, "  Interplay video: Help! Mystery opcode 0x6 seen\n");
213 214 215 216 217

    /* report success */
    return 0;
}

218
static int ipvideo_decode_block_opcode_0x7(IpvideoContext *s)
219 220
{
    int x, y;
221
    unsigned char P[2];
222 223 224
    unsigned int flags;

    /* 2-color encoding */
225
    CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 2);
226

227 228
    P[0] = *s->stream_ptr++;
    P[1] = *s->stream_ptr++;
229

230
    if (P[0] <= P[1]) {
231 232

        /* need 8 more bytes from the stream */
233
        CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 8);
234 235

        for (y = 0; y < 8; y++) {
236
            flags = *s->stream_ptr++ | 0x100;
237
            for (; flags != 1; flags >>= 1)
238
                *s->pixel_ptr++ = P[flags & 1];
239
            s->pixel_ptr += s->line_inc;
240 241 242 243 244
        }

    } else {

        /* need 2 more bytes from the stream */
245
        CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 2);
246

247
        flags = bytestream_get_le16(&s->stream_ptr);
248
        for (y = 0; y < 8; y += 2) {
249
            for (x = 0; x < 8; x += 2, flags >>= 1) {
Reimar Döffinger's avatar
Reimar Döffinger committed
250 251 252 253
                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];
254
            }
255
            s->pixel_ptr += s->stride * 2;
256 257 258 259 260 261 262
        }
    }

    /* report success */
    return 0;
}

263
static int ipvideo_decode_block_opcode_0x8(IpvideoContext *s)
264 265
{
    int x, y;
266
    unsigned char P[2];
267 268 269 270
    unsigned int flags = 0;

    /* 2-color encoding for each 4x4 quadrant, or 2-color encoding on
     * either top and bottom or left and right halves */
271
    CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 2);
272

273 274
    P[0] = *s->stream_ptr++;
    P[1] = *s->stream_ptr++;
275 276 277

    if (P[0] <= P[1]) {

278
        CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 14);
279
        s->stream_ptr -= 2;
280

281 282 283 284 285
        for (y = 0; y < 16; y++) {
            // new values for each 4x4 block
            if (!(y & 3)) {
                P[0] = *s->stream_ptr++; P[1] = *s->stream_ptr++;
                flags = bytestream_get_le16(&s->stream_ptr);
286 287
            }

288
            for (x = 0; x < 4; x++, flags >>= 1)
289 290 291 292
                *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;
293 294 295 296 297
        }

    } else {

        /* need 10 more bytes */
298
        CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 10);
299

300
        if (s->stream_ptr[4] <= s->stream_ptr[5]) {
301

302
            flags = bytestream_get_le32(&s->stream_ptr);
303

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

306
            for (y = 0; y < 16; y++) {
307
                for (x = 0; x < 4; x++, flags >>= 1)
308 309 310 311 312 313 314
                    *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;
                    P[0] = *s->stream_ptr++; P[1] = *s->stream_ptr++;
                    flags = bytestream_get_le32(&s->stream_ptr);
315 316 317 318 319 320 321 322
                }
            }

        } else {

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

            for (y = 0; y < 8; y++) {
323 324 325
                if (y == 4) {
                    P[0] = *s->stream_ptr++;
                    P[1] = *s->stream_ptr++;
326
                }
327
                flags = *s->stream_ptr++ | 0x100;
328

329
                for (; flags != 1; flags >>= 1)
330
                    *s->pixel_ptr++ = P[flags & 1];
331
                s->pixel_ptr += s->line_inc;
332 333 334 335 336 337 338 339
            }
        }
    }

    /* report success */
    return 0;
}

340
static int ipvideo_decode_block_opcode_0x9(IpvideoContext *s)
341 342 343 344 345
{
    int x, y;
    unsigned char P[4];

    /* 4-color encoding */
346
    CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 4);
347

348 349
    memcpy(P, s->stream_ptr, 4);
    s->stream_ptr += 4;
350

351 352
    if (P[0] <= P[1]) {
        if (P[2] <= P[3]) {
353

Reimar Döffinger's avatar
Reimar Döffinger committed
354
            /* 1 of 4 colors for each pixel, need 16 more bytes */
355
            CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 16);
356

Reimar Döffinger's avatar
Reimar Döffinger committed
357 358 359
            for (y = 0; y < 8; y++) {
                /* get the next set of 8 2-bit flags */
                int flags = bytestream_get_le16(&s->stream_ptr);
360
                for (x = 0; x < 8; x++, flags >>= 2)
Reimar Döffinger's avatar
Reimar Döffinger committed
361 362
                    *s->pixel_ptr++ = P[flags & 0x03];
                s->pixel_ptr += s->line_inc;
363 364
            }

365
        } else {
Reimar Döffinger's avatar
Reimar Döffinger committed
366
            uint32_t flags;
367

Reimar Döffinger's avatar
Reimar Döffinger committed
368
            /* 1 of 4 colors for each 2x2 block, need 4 more bytes */
369
            CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 4);
370

Reimar Döffinger's avatar
Reimar Döffinger committed
371
            flags = bytestream_get_le32(&s->stream_ptr);
372

Reimar Döffinger's avatar
Reimar Döffinger committed
373 374 375 376 377 378 379 380
            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;
381 382
            }

383 384
        }
    } else {
385
        uint64_t flags;
386

387
        /* 1 of 4 colors for each 2x1 or 1x2 block, need 8 more bytes */
388
        CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 8);
389

390
        flags = bytestream_get_le64(&s->stream_ptr);
391
        if (P[2] <= P[3]) {
Reimar Döffinger's avatar
Reimar Döffinger committed
392 393 394 395 396 397
            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;
398
            }
399
        } else {
Reimar Döffinger's avatar
Reimar Döffinger committed
400 401 402 403 404 405
            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;
406
            }
407
        }
408 409 410 411 412 413
    }

    /* report success */
    return 0;
}

414
static int ipvideo_decode_block_opcode_0xA(IpvideoContext *s)
415 416
{
    int x, y;
417
    unsigned char P[4];
418 419 420 421
    int flags = 0;

    /* 4-color encoding for each 4x4 quadrant, or 4-color encoding on
     * either top and bottom or left and right halves */
422
    CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 24);
423

424
    if (s->stream_ptr[0] <= s->stream_ptr[1]) {
425

426
        /* 4-color encoding for each quadrant; need 32 bytes */
427
        CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 32);
428

429 430 431 432 433 434 435
        for (y = 0; y < 16; y++) {
            // new values for each 4x4 block
            if (!(y & 3)) {
                memcpy(P, s->stream_ptr, 4);
                s->stream_ptr += 4;
                flags = bytestream_get_le32(&s->stream_ptr);
            }
436

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

440 441 442
            s->pixel_ptr += s->stride - 4;
            // switch to right half
            if (y == 7) s->pixel_ptr -= 8 * s->stride - 4;
443 444 445
        }

    } else {
446 447 448
        // vertical split?
        int vert = s->stream_ptr[12] <= s->stream_ptr[13];
        uint64_t flags = 0;
449 450

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

453 454 455 456 457 458
        for (y = 0; y < 16; y++) {
            // load values for each half
            if (!(y & 7)) {
                memcpy(P, s->stream_ptr, 4);
                s->stream_ptr += 4;
                flags = bytestream_get_le64(&s->stream_ptr);
459 460
            }

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

464 465 466 467 468
            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;
469 470 471 472 473 474 475
        }
    }

    /* report success */
    return 0;
}

476
static int ipvideo_decode_block_opcode_0xB(IpvideoContext *s)
477
{
478
    int y;
479 480

    /* 64-color encoding (each pixel in block is a different color) */
481
    CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 64);
482 483

    for (y = 0; y < 8; y++) {
484 485 486
        memcpy(s->pixel_ptr, s->stream_ptr, 8);
        s->stream_ptr += 8;
        s->pixel_ptr  += s->stride;
487 488 489 490 491 492
    }

    /* report success */
    return 0;
}

493
static int ipvideo_decode_block_opcode_0xC(IpvideoContext *s)
494 495 496 497
{
    int x, y;

    /* 16-color block encoding: each 2x2 block is a different color */
498
    CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 16);
499 500 501

    for (y = 0; y < 8; y += 2) {
        for (x = 0; x < 8; x += 2) {
502 503 504
            s->pixel_ptr[x                ] =
            s->pixel_ptr[x + 1            ] =
            s->pixel_ptr[x +     s->stride] =
505
            s->pixel_ptr[x + 1 + s->stride] = *s->stream_ptr++;
506
        }
507
        s->pixel_ptr += s->stride * 2;
508 509 510 511 512 513
    }

    /* report success */
    return 0;
}

514
static int ipvideo_decode_block_opcode_0xD(IpvideoContext *s)
515
{
516
    int y;
517
    unsigned char P[2];
518 519

    /* 4-color block encoding: each 4x4 block is a different color */
520
    CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 4);
521

522 523 524 525 526 527 528
    for (y = 0; y < 8; y++) {
        if (!(y & 3)) {
            P[0] = *s->stream_ptr++;
            P[1] = *s->stream_ptr++;
        }
        memset(s->pixel_ptr,     P[0], 4);
        memset(s->pixel_ptr + 4, P[1], 4);
529
        s->pixel_ptr += s->stride;
530 531 532 533 534 535
    }

    /* report success */
    return 0;
}

536
static int ipvideo_decode_block_opcode_0xE(IpvideoContext *s)
537
{
538
    int y;
539 540 541
    unsigned char pix;

    /* 1-color encoding: the whole block is 1 solid color */
542
    CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 1);
543
    pix = *s->stream_ptr++;
544 545

    for (y = 0; y < 8; y++) {
546 547
        memset(s->pixel_ptr, pix, 8);
        s->pixel_ptr += s->stride;
548 549 550 551 552 553
    }

    /* report success */
    return 0;
}

554
static int ipvideo_decode_block_opcode_0xF(IpvideoContext *s)
555 556
{
    int x, y;
557
    unsigned char sample[2];
558 559

    /* dithered encoding */
560
    CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 2);
561 562
    sample[0] = *s->stream_ptr++;
    sample[1] = *s->stream_ptr++;
563 564 565

    for (y = 0; y < 8; y++) {
        for (x = 0; x < 8; x += 2) {
566 567
            *s->pixel_ptr++ = sample[  y & 1 ];
            *s->pixel_ptr++ = sample[!(y & 1)];
568
        }
569
        s->pixel_ptr += s->line_inc;
570 571 572 573 574 575
    }

    /* report success */
    return 0;
}

576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932
static int ipvideo_decode_block_opcode_0x6_16(IpvideoContext *s)
{
    signed char x, y;

    /* copy a block from the second last frame using an expanded range */
    CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 2);

    x = *s->stream_ptr++;
    y = *s->stream_ptr++;

    debug_interplay ("    motion bytes = %d, %d\n", x, y);
    return copy_from(s, &s->second_last_frame, x, y);
}

static int ipvideo_decode_block_opcode_0x7_16(IpvideoContext *s)
{
    int x, y;
    uint16_t P[2];
    unsigned int flags;
    uint16_t *pixel_ptr = (uint16_t*)s->pixel_ptr;

    /* 2-color encoding */
    CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 4);

    P[0] = bytestream_get_le16(&s->stream_ptr);
    P[1] = bytestream_get_le16(&s->stream_ptr);

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

        CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 8);

        for (y = 0; y < 8; y++) {
            flags = *s->stream_ptr++ | 0x100;
            for (; flags != 1; flags >>= 1)
                *pixel_ptr++ = P[flags & 1];
            pixel_ptr += s->line_inc;
        }

    } else {

        CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 2);

        flags = bytestream_get_le16(&s->stream_ptr);
        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;
}

static int ipvideo_decode_block_opcode_0x8_16(IpvideoContext *s)
{
    int x, y;
    uint16_t P[2];
    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 */
    CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 4);

    P[0] = bytestream_get_le16(&s->stream_ptr);
    P[1] = bytestream_get_le16(&s->stream_ptr);

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

        CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 24);
        s->stream_ptr -= 4;

        for (y = 0; y < 16; y++) {
            // new values for each 4x4 block
            if (!(y & 3)) {
                P[0] = bytestream_get_le16(&s->stream_ptr);
                P[1] = bytestream_get_le16(&s->stream_ptr);
                flags = bytestream_get_le16(&s->stream_ptr);
            }

            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 {

        CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 12);

        if (!(AV_RL16(s->stream_ptr + 4) & 0x8000)) {

            flags = bytestream_get_le32(&s->stream_ptr);

            /* 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;
                    P[0] = bytestream_get_le16(&s->stream_ptr);
                    P[1] = bytestream_get_le16(&s->stream_ptr);
                    flags = bytestream_get_le32(&s->stream_ptr);
                }
            }

        } else {

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

            for (y = 0; y < 8; y++) {
                if (y == 4) {
                    P[0] = bytestream_get_le16(&s->stream_ptr);
                    P[1] = bytestream_get_le16(&s->stream_ptr);
                }
                flags = *s->stream_ptr++ | 0x100;

                for (; flags != 1; flags >>= 1)
                    *pixel_ptr++ = P[flags & 1];
                pixel_ptr += s->line_inc;
            }
        }
    }

    /* report success */
    return 0;
}

static int ipvideo_decode_block_opcode_0x9_16(IpvideoContext *s)
{
    int x, y;
    uint16_t P[4];
    uint16_t *pixel_ptr = (uint16_t*)s->pixel_ptr;

    /* 4-color encoding */
    CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 8);

    for (x = 0; x < 4; x++)
        P[x] = bytestream_get_le16(&s->stream_ptr);

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

            /* 1 of 4 colors for each pixel */
            CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 16);

            for (y = 0; y < 8; y++) {
                /* get the next set of 8 2-bit flags */
                int flags = bytestream_get_le16(&s->stream_ptr);
                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 */
            CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 4);

            flags = bytestream_get_le32(&s->stream_ptr);

            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 */
        CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 8);

        flags = bytestream_get_le64(&s->stream_ptr);
        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;
}

static int ipvideo_decode_block_opcode_0xA_16(IpvideoContext *s)
{
    int x, y;
    uint16_t P[4];
    int flags = 0;
    uint16_t *pixel_ptr = (uint16_t*)s->pixel_ptr;

    /* 4-color encoding for each 4x4 quadrant, or 4-color encoding on
     * either top and bottom or left and right halves */
    CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 24);

    if (!(AV_RL16(s->stream_ptr) & 0x8000)) {

        /* 4-color encoding for each quadrant */
        CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 48);

        for (y = 0; y < 16; y++) {
            // new values for each 4x4 block
            if (!(y & 3)) {
                for (x = 0; x < 4; x++)
                    P[x] = bytestream_get_le16(&s->stream_ptr);
                flags = bytestream_get_le32(&s->stream_ptr);
            }

            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?
        int vert = !(AV_RL16(s->stream_ptr + 16) & 0x8000);
        uint64_t flags = 0;

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

        for (y = 0; y < 16; y++) {
            // load values for each half
            if (!(y & 7)) {
                for (x = 0; x < 4; x++)
                    P[x] = bytestream_get_le16(&s->stream_ptr);
                flags = bytestream_get_le64(&s->stream_ptr);
            }

            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;
        }
    }

    /* report success */
    return 0;
}

static int ipvideo_decode_block_opcode_0xB_16(IpvideoContext *s)
{
    int x, y;
    uint16_t *pixel_ptr = (uint16_t*)s->pixel_ptr;

    /* 64-color encoding (each pixel in block is a different color) */
    CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 128);

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

    /* report success */
    return 0;
}

static int ipvideo_decode_block_opcode_0xC_16(IpvideoContext *s)
{
    int x, y;
    uint16_t *pixel_ptr = (uint16_t*)s->pixel_ptr;

    /* 16-color block encoding: each 2x2 block is a different color */
    CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 32);

    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] =
            pixel_ptr[x + 1 + s->stride] = bytestream_get_le16(&s->stream_ptr);
        }
        pixel_ptr += s->stride * 2;
    }

    /* report success */
    return 0;
}

static int ipvideo_decode_block_opcode_0xD_16(IpvideoContext *s)
{
    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 */
    CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 8);

    for (y = 0; y < 8; y++) {
        if (!(y & 3)) {
            P[0] = bytestream_get_le16(&s->stream_ptr);
            P[1] = bytestream_get_le16(&s->stream_ptr);
        }
        for (x = 0; x < 8; x++)
            pixel_ptr[x] = P[x >> 2];
        pixel_ptr += s->stride;
    }

    /* report success */
    return 0;
}

static int ipvideo_decode_block_opcode_0xE_16(IpvideoContext *s)
{
    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 */
    CHECK_STREAM_PTR(s->stream_ptr, s->stream_end, 2);
    pix = bytestream_get_le16(&s->stream_ptr);

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

    /* report success */
    return 0;
}

933 934 935 936 937 938 939 940 941 942
static int (* const ipvideo_decode_block[])(IpvideoContext *s) = {
    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,
};
943

944 945 946 947 948 949 950 951 952 953 954
static int (* const ipvideo_decode_block16[])(IpvideoContext *s) = {
    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,
};

955 956 957 958 959 960
static void ipvideo_decode_opcodes(IpvideoContext *s)
{
    int x, y;
    unsigned char opcode;
    int ret;
    static int frame = 0;
961
    GetBitContext gb;
962 963 964 965

    debug_interplay("------------------ frame %d\n", frame);
    frame++;

966
    if (!s->is_16bpp) {
967 968
        /* this is PAL8, so make the palette available */
        memcpy(s->current_frame.data[1], s->avctx->palctrl->palette, PALETTE_COUNT * 4);
969

970 971 972
        s->stride = s->current_frame.linesize[0];
        s->stream_ptr = s->buf + 14;  /* data starts 14 bytes in */
        s->stream_end = s->buf + s->size;
973 974
    } else {
        s->stride = s->current_frame.linesize[0] >> 1;
975 976 977 978
        s->stream_ptr = s->buf + 16;
        s->stream_end =
        s->mv_ptr = s->buf + 14 + AV_RL16(s->buf+14);
        s->mv_end = s->buf + s->size;
979
    }
980
    s->line_inc = s->stride - 8;
981
    s->upper_motion_limit_offset = (s->avctx->height - 8) * s->current_frame.linesize[0]
982
                                  + (s->avctx->width - 8) * (1 + s->is_16bpp);
983

984
    init_get_bits(&gb, s->decoding_map, s->decoding_map_size * 8);
985 986
    for (y = 0; y < s->avctx->height; y += 8) {
        for (x = 0; x < s->avctx->width; x += 8) {
987
            opcode = get_bits(&gb, 4);
988 989

            debug_interplay("  block @ (%3d, %3d): encoding 0x%X, data ptr @ %p\n",
990
                            x, y, opcode, s->stream_ptr);
991

992
            if (!s->is_16bpp) {
993 994 995
                s->pixel_ptr = s->current_frame.data[0] + x
                              + y*s->current_frame.linesize[0];
                ret = ipvideo_decode_block[opcode](s);
996
            } else {
997 998
                s->pixel_ptr = s->current_frame.data[0] + x*2
                              + y*s->current_frame.linesize[0];
999
                ret = ipvideo_decode_block16[opcode](s);
1000
            }
1001
            if (ret != 0) {
1002
                av_log(s->avctx, AV_LOG_ERROR, " Interplay video: decode problem on frame %d, @ block (%d, %d)\n",
1003
                       frame, x, y);
1004
                return;
1005 1006 1007
            }
        }
    }
1008
    if (s->stream_end - s->stream_ptr > 1) {
1009
        av_log(s->avctx, AV_LOG_ERROR, " Interplay video: decode finished with %td bytes left over\n",
Kostya Shishkov's avatar
Kostya Shishkov committed
1010
               s->stream_end - s->stream_ptr);
1011
    }
1012 1013
}

1014
static av_cold int ipvideo_decode_init(AVCodecContext *avctx)
1015 1016 1017 1018
{
    IpvideoContext *s = avctx->priv_data;

    s->avctx = avctx;
1019

1020 1021 1022
    s->is_16bpp = avctx->bits_per_coded_sample == 16;
    avctx->pix_fmt = s->is_16bpp ? PIX_FMT_RGB555 : PIX_FMT_PAL8;
    if (!s->is_16bpp && s->avctx->palctrl == NULL) {
1023
        av_log(avctx, AV_LOG_ERROR, " Interplay video: palette expected.\n");
1024 1025 1026
        return -1;
    }

1027 1028 1029 1030
    dsputil_init(&s->dsp, avctx);

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

1032 1033 1034
    s->current_frame.data[0] = s->last_frame.data[0] =
    s->second_last_frame.data[0] = NULL;

1035 1036 1037 1038 1039
    return 0;
}

static int ipvideo_decode_frame(AVCodecContext *avctx,
                                void *data, int *data_size,
1040
                                AVPacket *avpkt)
1041
{
1042 1043
    const uint8_t *buf = avpkt->data;
    int buf_size = avpkt->size;
1044
    IpvideoContext *s = avctx->priv_data;
1045
    AVPaletteControl *palette_control = avctx->palctrl;
1046

1047 1048 1049 1050 1051
    /* 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;

1052 1053 1054
    s->decoding_map = buf;
    s->buf = buf + s->decoding_map_size;
    s->size = buf_size - s->decoding_map_size;
1055

1056
    s->current_frame.reference = 3;
1057
    if (avctx->get_buffer(avctx, &s->current_frame)) {
1058
        av_log(avctx, AV_LOG_ERROR, "  Interplay Video: get_buffer() failed\n");
1059 1060
        return -1;
    }
1061

1062
    ipvideo_decode_opcodes(s);
1063

1064
    if (!s->is_16bpp && palette_control->palette_changed) {
1065 1066 1067 1068
        palette_control->palette_changed = 0;
        s->current_frame.palette_has_changed = 1;
    }

1069 1070
    *data_size = sizeof(AVFrame);
    *(AVFrame*)data = s->current_frame;
1071

1072
    /* shuffle frames */
1073 1074 1075
    if (s->second_last_frame.data[0])
        avctx->release_buffer(avctx, &s->second_last_frame);
    s->second_last_frame = s->last_frame;
1076
    s->last_frame = s->current_frame;
1077
    s->current_frame.data[0] = NULL;  /* catch any access attempts */
1078

1079
    /* report that the buffer was completely consumed */
1080 1081 1082
    return buf_size;
}

1083
static av_cold int ipvideo_decode_end(AVCodecContext *avctx)
1084 1085 1086 1087
{
    IpvideoContext *s = avctx->priv_data;

    /* release the last frame */
1088 1089 1090 1091
    if (s->last_frame.data[0])
        avctx->release_buffer(avctx, &s->last_frame);
    if (s->second_last_frame.data[0])
        avctx->release_buffer(avctx, &s->second_last_frame);
1092 1093 1094 1095 1096 1097

    return 0;
}

AVCodec interplay_video_decoder = {
    "interplayvideo",
1098
    AVMEDIA_TYPE_VIDEO,
1099 1100 1101 1102 1103 1104 1105
    CODEC_ID_INTERPLAY_VIDEO,
    sizeof(IpvideoContext),
    ipvideo_decode_init,
    NULL,
    ipvideo_decode_end,
    ipvideo_decode_frame,
    CODEC_CAP_DR1,
1106
    .long_name = NULL_IF_CONFIG_SMALL("Interplay MVE video"),
1107
};