git.sesse.net Git - ffmpeg/blob - libavcodec/interplayvideo.c

   1 /*
   2  * Interplay MVE Video Decoder
   3  * Copyright (C) 2003 the ffmpeg project
   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 /**
  24  * @file interplayvideo.c
  25  * Interplay MVE Video Decoder by Mike Melanson (melanson@pcisys.net)
  26  * For more information about the Interplay MVE format, visit:
  27  *   http://www.pcisys.net/~melanson/codecs/interplay-mve.txt
  28  * This code is written in such a way that the identifiers match up
  29  * with the encoding descriptions in the document.
  30  *
  31  * This decoder presently only supports a PAL8 output colorspace.
  32  *
  33  * An Interplay video frame consists of 2 parts: The decoding map and
  34  * the video data. A demuxer must load these 2 parts together in a single
  35  * buffer before sending it through the stream to this decoder.
  36  */
  37
  38 #include <stdio.h>
  39 #include <stdlib.h>
  40 #include <string.h>
  41 #include <unistd.h>
  42
  43 #include "common.h"
  44 #include "avcodec.h"
  45 #include "dsputil.h"
  46
  47 #define PALETTE_COUNT 256
  48
  49 /* debugging support */
  50 #define DEBUG_INTERPLAY 0
  51 #if DEBUG_INTERPLAY
  52 #define debug_interplay(x,...) av_log(NULL, AV_LOG_DEBUG, x, __VA_ARGS__)
  53 #else
  54 static inline void debug_interplay(const char *format, ...) { }
  55 #endif
  56
  57 typedef struct IpvideoContext {
  58
  59     AVCodecContext *avctx;
  60     DSPContext dsp;
  61     AVFrame second_last_frame;
  62     AVFrame last_frame;
  63     AVFrame current_frame;
  64     unsigned char *decoding_map;
  65     int decoding_map_size;
  66
  67     unsigned char *buf;
  68     int size;
  69
  70     unsigned char *stream_ptr;
  71     unsigned char *stream_end;
  72     unsigned char *pixel_ptr;
  73     int line_inc;
  74     int stride;
  75     int upper_motion_limit_offset;
  76
  77 } IpvideoContext;
  78
  79 #define CHECK_STREAM_PTR(n) \
  80   if ((s->stream_ptr + n) > s->stream_end) { \
  81     av_log(s->avctx, AV_LOG_ERROR, "Interplay video warning: stream_ptr out of bounds (%p >= %p)\n", \
  82       s->stream_ptr + n, s->stream_end); \
  83     return -1; \
  84   }
  85
  86 #define COPY_FROM_CURRENT() \
  87     motion_offset = current_offset; \
  88     motion_offset += y * s->stride; \
  89     motion_offset += x; \
  90     if (motion_offset < 0) { \
  91         av_log(s->avctx, AV_LOG_ERROR, " Interplay video: motion offset < 0 (%d)\n", motion_offset); \
  92         return -1; \
  93     } else if (motion_offset > s->upper_motion_limit_offset) { \
  94         av_log(s->avctx, AV_LOG_ERROR, " Interplay video: motion offset above limit (%d >= %d)\n", \
  95             motion_offset, s->upper_motion_limit_offset); \
  96         return -1; \
  97     } \
  98     s->dsp.put_pixels_tab[0][0](s->pixel_ptr, \
  99         s->current_frame.data[0] + motion_offset, s->stride, 8);
 100
 101 #define COPY_FROM_PREVIOUS() \
 102     motion_offset = current_offset; \
 103     motion_offset += y * s->stride; \
 104     motion_offset += x; \
 105     if (motion_offset < 0) { \
 106         av_log(s->avctx, AV_LOG_ERROR, " Interplay video: motion offset < 0 (%d)\n", motion_offset); \
 107         return -1; \
 108     } else if (motion_offset > s->upper_motion_limit_offset) { \
 109         av_log(s->avctx, AV_LOG_ERROR, " Interplay video: motion offset above limit (%d >= %d)\n", \
 110             motion_offset, s->upper_motion_limit_offset); \
 111         return -1; \
 112     } \
 113     s->dsp.put_pixels_tab[0][0](s->pixel_ptr, \
 114         s->last_frame.data[0] + motion_offset, s->stride, 8);
 115
 116 #define COPY_FROM_SECOND_LAST() \
 117     motion_offset = current_offset; \
 118     motion_offset += y * s->stride; \
 119     motion_offset += x; \
 120     if (motion_offset < 0) { \
 121         av_log(s->avctx, AV_LOG_ERROR, " Interplay video: motion offset < 0 (%d)\n", motion_offset); \
 122         return -1; \
 123     } else if (motion_offset > s->upper_motion_limit_offset) { \
 124         av_log(s->avctx, AV_LOG_ERROR, " Interplay video: motion offset above limit (%d >= %d)\n", \
 125             motion_offset, s->upper_motion_limit_offset); \
 126         return -1; \
 127     } \
 128     s->dsp.put_pixels_tab[0][0](s->pixel_ptr, \
 129         s->second_last_frame.data[0] + motion_offset, s->stride, 8);
 130
 131 static int ipvideo_decode_block_opcode_0x0(IpvideoContext *s)
 132 {
 133     int x, y;
 134     int motion_offset;
 135     int current_offset = s->pixel_ptr - s->current_frame.data[0];
 136
 137     /* copy a block from the previous frame */
 138     x = y = 0;
 139     COPY_FROM_PREVIOUS();
 140
 141     /* report success */
 142     return 0;
 143 }
 144
 145 static int ipvideo_decode_block_opcode_0x1(IpvideoContext *s)
 146 {
 147     int x, y;
 148     int motion_offset;
 149     int current_offset = s->pixel_ptr - s->current_frame.data[0];
 150
 151     /* copy block from 2 frames ago */
 152     x = y = 0;
 153     COPY_FROM_SECOND_LAST();
 154
 155     /* report success */
 156     return 0;
 157 }
 158
 159 static int ipvideo_decode_block_opcode_0x2(IpvideoContext *s)
 160 {
 161     unsigned char B;
 162     int x, y;
 163     int motion_offset;
 164     int current_offset = s->pixel_ptr - s->current_frame.data[0];
 165
 166     /* copy block from 2 frames ago using a motion vector; need 1 more byte */
 167     CHECK_STREAM_PTR(1);
 168     B = *s->stream_ptr++;
 169
 170     if (B < 56) {
 171         x = 8 + (B % 7);
 172         y = B / 7;
 173     } else {
 174         x = -14 + ((B - 56) % 29);
 175         y =   8 + ((B - 56) / 29);
 176     }
 177
 178     debug_interplay ("    motion byte = %d, (x, y) = (%d, %d)\n", B, x, y);
 179     COPY_FROM_SECOND_LAST();
 180
 181     /* report success */
 182     return 0;
 183 }
 184
 185 static int ipvideo_decode_block_opcode_0x3(IpvideoContext *s)
 186 {
 187     unsigned char B;
 188     int x, y;
 189     int motion_offset;
 190     int current_offset = s->pixel_ptr - s->current_frame.data[0];
 191
 192     /* copy 8x8 block from current frame from an up/left block */
 193
 194     /* need 1 more byte for motion */
 195     CHECK_STREAM_PTR(1);
 196     B = *s->stream_ptr++;
 197
 198     if (B < 56) {
 199         x = -(8 + (B % 7));
 200         y = -(B / 7);
 201     } else {
 202         x = -(-14 + ((B - 56) % 29));
 203         y = -(  8 + ((B - 56) / 29));
 204     }
 205
 206     debug_interplay ("    motion byte = %d, (x, y) = (%d, %d)\n", B, x, y);
 207     COPY_FROM_CURRENT();
 208
 209     /* report success */
 210     return 0;
 211 }
 212
 213 static int ipvideo_decode_block_opcode_0x4(IpvideoContext *s)
 214 {
 215     int x, y;
 216     unsigned char B, BL, BH;
 217     int motion_offset;
 218     int current_offset = s->pixel_ptr - s->current_frame.data[0];
 219
 220     /* copy a block from the previous frame; need 1 more byte */
 221     CHECK_STREAM_PTR(1);
 222
 223     B = *s->stream_ptr++;
 224     BL = B & 0x0F;
 225     BH = (B >> 4) & 0x0F;
 226     x = -8 + BL;
 227     y = -8 + BH;
 228
 229     debug_interplay ("    motion byte = %d, (x, y) = (%d, %d)\n", B, x, y);
 230     COPY_FROM_PREVIOUS();
 231
 232     /* report success */
 233     return 0;
 234 }
 235
 236 static int ipvideo_decode_block_opcode_0x5(IpvideoContext *s)
 237 {
 238     signed char x, y;
 239     int motion_offset;
 240     int current_offset = s->pixel_ptr - s->current_frame.data[0];
 241
 242     /* copy a block from the previous frame using an expanded range;
 243      * need 2 more bytes */
 244     CHECK_STREAM_PTR(2);
 245
 246     x = *s->stream_ptr++;
 247     y = *s->stream_ptr++;
 248
 249     debug_interplay ("    motion bytes = %d, %d\n", x, y);
 250     COPY_FROM_PREVIOUS();
 251
 252     /* report success */
 253     return 0;
 254 }
 255
 256 static int ipvideo_decode_block_opcode_0x6(IpvideoContext *s)
 257 {
 258     /* mystery opcode? skip multiple blocks? */
 259     av_log(s->avctx, AV_LOG_ERROR, "  Interplay video: Help! Mystery opcode 0x6 seen\n");
 260
 261     /* report success */
 262     return 0;
 263 }
 264
 265 static int ipvideo_decode_block_opcode_0x7(IpvideoContext *s)
 266 {
 267     int x, y;
 268     unsigned char P0, P1;
 269     unsigned char B[8];
 270     unsigned int flags;
 271     int bitmask;
 272
 273     /* 2-color encoding */
 274     CHECK_STREAM_PTR(2);
 275
 276     P0 = *s->stream_ptr++;
 277     P1 = *s->stream_ptr++;
 278
 279     if (P0 <= P1) {
 280
 281         /* need 8 more bytes from the stream */
 282         CHECK_STREAM_PTR(8);
 283         for (y = 0; y < 8; y++)
 284             B[y] = *s->stream_ptr++;
 285
 286         for (y = 0; y < 8; y++) {
 287             flags = B[y];
 288             for (x = 0x01; x <= 0x80; x <<= 1) {
 289                 if (flags & x)
 290                     *s->pixel_ptr++ = P1;
 291                 else
 292                     *s->pixel_ptr++ = P0;
 293             }
 294             s->pixel_ptr += s->line_inc;
 295         }
 296
 297     } else {
 298
 299         /* need 2 more bytes from the stream */
 300         CHECK_STREAM_PTR(2);
 301         B[0] = *s->stream_ptr++;
 302         B[1] = *s->stream_ptr++;
 303
 304         flags = (B[1] << 8) | B[0];
 305         bitmask = 0x0001;
 306         for (y = 0; y < 8; y += 2) {
 307             for (x = 0; x < 8; x += 2, bitmask <<= 1) {
 308                 if (flags & bitmask) {
 309                     *(s->pixel_ptr + x) = P1;
 310                     *(s->pixel_ptr + x + 1) = P1;
 311                     *(s->pixel_ptr + s->stride + x) = P1;
 312                     *(s->pixel_ptr + s->stride + x + 1) = P1;
 313                 } else {
 314                     *(s->pixel_ptr + x) = P0;
 315                     *(s->pixel_ptr + x + 1) = P0;
 316                     *(s->pixel_ptr + s->stride + x) = P0;
 317                     *(s->pixel_ptr + s->stride + x + 1) = P0;
 318                 }
 319             }
 320             s->pixel_ptr += s->stride * 2;
 321         }
 322     }
 323
 324     /* report success */
 325     return 0;
 326 }
 327
 328 static int ipvideo_decode_block_opcode_0x8(IpvideoContext *s)
 329 {
 330     int x, y;
 331     unsigned char P[8];
 332     unsigned char B[8];
 333     unsigned int flags = 0;
 334     unsigned int bitmask = 0;
 335     unsigned char P0 = 0, P1 = 0;
 336     int lower_half = 0;
 337
 338     /* 2-color encoding for each 4x4 quadrant, or 2-color encoding on
 339      * either top and bottom or left and right halves */
 340     CHECK_STREAM_PTR(2);
 341
 342     P[0] = *s->stream_ptr++;
 343     P[1] = *s->stream_ptr++;
 344
 345     if (P[0] <= P[1]) {
 346
 347         /* need 12 more bytes */
 348         CHECK_STREAM_PTR(12);
 349         B[0] = *s->stream_ptr++;  B[1] = *s->stream_ptr++;
 350         P[2] = *s->stream_ptr++;  P[3] = *s->stream_ptr++;
 351         B[2] = *s->stream_ptr++;  B[3] = *s->stream_ptr++;
 352         P[4] = *s->stream_ptr++;  P[5] = *s->stream_ptr++;
 353         B[4] = *s->stream_ptr++;  B[5] = *s->stream_ptr++;
 354         P[6] = *s->stream_ptr++;  P[7] = *s->stream_ptr++;
 355         B[6] = *s->stream_ptr++;  B[7] = *s->stream_ptr++;
 356
 357         for (y = 0; y < 8; y++) {
 358
 359             /* time to reload flags? */
 360             if (y == 0) {
 361                 flags =
 362                     ((B[0] & 0xF0) <<  4) | ((B[4] & 0xF0) <<  8) |
 363                     ((B[0] & 0x0F)      ) | ((B[4] & 0x0F) <<  4) |
 364                     ((B[1] & 0xF0) << 20) | ((B[5] & 0xF0) << 24) |
 365                     ((B[1] & 0x0F) << 16) | ((B[5] & 0x0F) << 20);
 366                 bitmask = 0x00000001;
 367                 lower_half = 0;  /* still on top half */
 368             } else if (y == 4) {
 369                 flags =
 370                     ((B[2] & 0xF0) <<  4) | ((B[6] & 0xF0) <<  8) |
 371                     ((B[2] & 0x0F)      ) | ((B[6] & 0x0F) <<  4) |
 372                     ((B[3] & 0xF0) << 20) | ((B[7] & 0xF0) << 24) |
 373                     ((B[3] & 0x0F) << 16) | ((B[7] & 0x0F) << 20);
 374                 bitmask = 0x00000001;
 375                 lower_half = 2;
 376             }
 377
 378             for (x = 0; x < 8; x++, bitmask <<= 1) {
 379                 /* get the pixel values ready for this quadrant */
 380                 if (x == 0) {
 381                     P0 = P[lower_half + 0];
 382                     P1 = P[lower_half + 1];
 383                 } else if (x == 4) {
 384                     P0 = P[lower_half + 4];
 385                     P1 = P[lower_half + 5];
 386                 }
 387
 388                 if (flags & bitmask)
 389                     *s->pixel_ptr++ = P1;
 390                 else
 391                     *s->pixel_ptr++ = P0;
 392             }
 393             s->pixel_ptr += s->line_inc;
 394         }
 395
 396     } else {
 397
 398         /* need 10 more bytes */
 399         CHECK_STREAM_PTR(10);
 400         B[0] = *s->stream_ptr++;  B[1] = *s->stream_ptr++;
 401         B[2] = *s->stream_ptr++;  B[3] = *s->stream_ptr++;
 402         P[2] = *s->stream_ptr++;  P[3] = *s->stream_ptr++;
 403         B[4] = *s->stream_ptr++;  B[5] = *s->stream_ptr++;
 404         B[6] = *s->stream_ptr++;  B[7] = *s->stream_ptr++;
 405
 406         if (P[2] <= P[3]) {
 407
 408             /* vertical split; left & right halves are 2-color encoded */
 409
 410             for (y = 0; y < 8; y++) {
 411
 412                 /* time to reload flags? */
 413                 if (y == 0) {
 414                     flags =
 415                         ((B[0] & 0xF0) <<  4) | ((B[4] & 0xF0) <<  8) |
 416                         ((B[0] & 0x0F)      ) | ((B[4] & 0x0F) <<  4) |
 417                         ((B[1] & 0xF0) << 20) | ((B[5] & 0xF0) << 24) |
 418                         ((B[1] & 0x0F) << 16) | ((B[5] & 0x0F) << 20);
 419                     bitmask = 0x00000001;
 420                 } else if (y == 4) {
 421                     flags =
 422                         ((B[2] & 0xF0) <<  4) | ((B[6] & 0xF0) <<  8) |
 423                         ((B[2] & 0x0F)      ) | ((B[6] & 0x0F) <<  4) |
 424                         ((B[3] & 0xF0) << 20) | ((B[7] & 0xF0) << 24) |
 425                         ((B[3] & 0x0F) << 16) | ((B[7] & 0x0F) << 20);
 426                     bitmask = 0x00000001;
 427                 }
 428
 429                 for (x = 0; x < 8; x++, bitmask <<= 1) {
 430                     /* get the pixel values ready for this half */
 431                     if (x == 0) {
 432                         P0 = P[0];
 433                         P1 = P[1];
 434                     } else if (x == 4) {
 435                         P0 = P[2];
 436                         P1 = P[3];
 437                     }
 438
 439                     if (flags & bitmask)
 440                         *s->pixel_ptr++ = P1;
 441                     else
 442                         *s->pixel_ptr++ = P0;
 443                 }
 444                 s->pixel_ptr += s->line_inc;
 445             }
 446
 447         } else {
 448
 449             /* horizontal split; top & bottom halves are 2-color encoded */
 450
 451             for (y = 0; y < 8; y++) {
 452
 453                 flags = B[y];
 454                 if (y == 0) {
 455                     P0 = P[0];
 456                     P1 = P[1];
 457                 } else if (y == 4) {
 458                     P0 = P[2];
 459                     P1 = P[3];
 460                 }
 461
 462                 for (bitmask = 0x01; bitmask <= 0x80; bitmask <<= 1) {
 463
 464                     if (flags & bitmask)
 465                         *s->pixel_ptr++ = P1;
 466                     else
 467                         *s->pixel_ptr++ = P0;
 468                 }
 469                 s->pixel_ptr += s->line_inc;
 470             }
 471         }
 472     }
 473
 474     /* report success */
 475     return 0;
 476 }
 477
 478 static int ipvideo_decode_block_opcode_0x9(IpvideoContext *s)
 479 {
 480     int x, y;
 481     unsigned char P[4];
 482     unsigned char B[4];
 483     unsigned int flags = 0;
 484     int shifter = 0;
 485     unsigned char pix;
 486
 487     /* 4-color encoding */
 488     CHECK_STREAM_PTR(4);
 489
 490     for (y = 0; y < 4; y++)
 491         P[y] = *s->stream_ptr++;
 492
 493     if ((P[0] <= P[1]) && (P[2] <= P[3])) {
 494
 495         /* 1 of 4 colors for each pixel, need 16 more bytes */
 496         CHECK_STREAM_PTR(16);
 497
 498         for (y = 0; y < 8; y++) {
 499             /* get the next set of 8 2-bit flags */
 500             flags = (s->stream_ptr[1] << 8) | s->stream_ptr[0];
 501             s->stream_ptr += 2;
 502             for (x = 0, shifter = 0; x < 8; x++, shifter += 2) {
 503                 *s->pixel_ptr++ = P[(flags >> shifter) & 0x03];
 504             }
 505             s->pixel_ptr += s->line_inc;
 506         }
 507
 508     } else if ((P[0] <= P[1]) && (P[2] > P[3])) {
 509
 510         /* 1 of 4 colors for each 2x2 block, need 4 more bytes */
 511         CHECK_STREAM_PTR(4);
 512
 513         B[0] = *s->stream_ptr++;
 514         B[1] = *s->stream_ptr++;
 515         B[2] = *s->stream_ptr++;
 516         B[3] = *s->stream_ptr++;
 517         flags = (B[3] << 24) | (B[2] << 16) | (B[1] << 8) | B[0];
 518         shifter = 0;
 519
 520         for (y = 0; y < 8; y += 2) {
 521             for (x = 0; x < 8; x += 2, shifter += 2) {
 522                 pix = P[(flags >> shifter) & 0x03];
 523                 *(s->pixel_ptr + x) = pix;
 524                 *(s->pixel_ptr + x + 1) = pix;
 525                 *(s->pixel_ptr + s->stride + x) = pix;
 526                 *(s->pixel_ptr + s->stride + x + 1) = pix;
 527             }
 528             s->pixel_ptr += s->stride * 2;
 529         }
 530
 531     } else if ((P[0] > P[1]) && (P[2] <= P[3])) {
 532
 533         /* 1 of 4 colors for each 2x1 block, need 8 more bytes */
 534         CHECK_STREAM_PTR(8);
 535
 536         for (y = 0; y < 8; y++) {
 537             /* time to reload flags? */
 538             if ((y == 0) || (y == 4)) {
 539                 B[0] = *s->stream_ptr++;
 540                 B[1] = *s->stream_ptr++;
 541                 B[2] = *s->stream_ptr++;
 542                 B[3] = *s->stream_ptr++;
 543                 flags = (B[3] << 24) | (B[2] << 16) | (B[1] << 8) | B[0];
 544                 shifter = 0;
 545             }
 546             for (x = 0; x < 8; x += 2, shifter += 2) {
 547                 pix = P[(flags >> shifter) & 0x03];
 548                 *(s->pixel_ptr + x) = pix;
 549                 *(s->pixel_ptr + x + 1) = pix;
 550             }
 551             s->pixel_ptr += s->stride;
 552         }
 553
 554     } else {
 555
 556         /* 1 of 4 colors for each 1x2 block, need 8 more bytes */
 557         CHECK_STREAM_PTR(8);
 558
 559         for (y = 0; y < 8; y += 2) {
 560             /* time to reload flags? */
 561             if ((y == 0) || (y == 4)) {
 562                 B[0] = *s->stream_ptr++;
 563                 B[1] = *s->stream_ptr++;
 564                 B[2] = *s->stream_ptr++;
 565                 B[3] = *s->stream_ptr++;
 566                 flags = (B[3] << 24) | (B[2] << 16) | (B[1] << 8) | B[0];
 567                 shifter = 0;
 568             }
 569             for (x = 0; x < 8; x++, shifter += 2) {
 570                 pix = P[(flags >> shifter) & 0x03];
 571                 *(s->pixel_ptr + x) = pix;
 572                 *(s->pixel_ptr + s->stride + x) = pix;
 573             }
 574             s->pixel_ptr += s->stride * 2;
 575         }
 576     }
 577
 578     /* report success */
 579     return 0;
 580 }
 581
 582 static int ipvideo_decode_block_opcode_0xA(IpvideoContext *s)
 583 {
 584     int x, y;
 585     unsigned char P[16];
 586     unsigned char B[16];
 587     int flags = 0;
 588     int shifter = 0;
 589     int index;
 590     int split;
 591     int lower_half;
 592
 593     /* 4-color encoding for each 4x4 quadrant, or 4-color encoding on
 594      * either top and bottom or left and right halves */
 595     CHECK_STREAM_PTR(4);
 596
 597     for (y = 0; y < 4; y++)
 598         P[y] = *s->stream_ptr++;
 599
 600     if (P[0] <= P[1]) {
 601
 602         /* 4-color encoding for each quadrant; need 28 more bytes */
 603         CHECK_STREAM_PTR(28);
 604
 605         for (y = 0; y < 4; y++)
 606             B[y] = *s->stream_ptr++;
 607         for (y = 4; y < 16; y += 4) {
 608             for (x = y; x < y + 4; x++)
 609                 P[x] = *s->stream_ptr++;
 610             for (x = y; x < y + 4; x++)
 611                 B[x] = *s->stream_ptr++;
 612         }
 613
 614         for (y = 0; y < 8; y++) {
 615
 616             lower_half = (y >= 4) ? 4 : 0;
 617             flags = (B[y + 8] << 8) | B[y];
 618
 619             for (x = 0, shifter = 0; x < 8; x++, shifter += 2) {
 620                 split = (x >= 4) ? 8 : 0;
 621                 index = split + lower_half + ((flags >> shifter) & 0x03);
 622                 *s->pixel_ptr++ = P[index];
 623             }
 624
 625             s->pixel_ptr += s->line_inc;
 626         }
 627
 628     } else {
 629
 630         /* 4-color encoding for either left and right or top and bottom
 631          * halves; need 20 more bytes */
 632         CHECK_STREAM_PTR(20);
 633
 634         for (y = 0; y < 8; y++)
 635             B[y] = *s->stream_ptr++;
 636         for (y = 4; y < 8; y++)
 637             P[y] = *s->stream_ptr++;
 638         for (y = 8; y < 16; y++)
 639             B[y] = *s->stream_ptr++;
 640
 641         if (P[4] <= P[5]) {
 642
 643             /* block is divided into left and right halves */
 644             for (y = 0; y < 8; y++) {
 645
 646                 flags = (B[y + 8] << 8) | B[y];
 647                 split = 0;
 648
 649                 for (x = 0, shifter = 0; x < 8; x++, shifter += 2) {
 650                     if (x == 4)
 651                         split = 4;
 652                     *s->pixel_ptr++ = P[split + ((flags >> shifter) & 0x03)];
 653                 }
 654
 655                 s->pixel_ptr += s->line_inc;
 656             }
 657
 658         } else {
 659
 660             /* block is divided into top and bottom halves */
 661             split = 0;
 662             for (y = 0; y < 8; y++) {
 663
 664                 flags = (B[y * 2 + 1] << 8) | B[y * 2];
 665                 if (y == 4)
 666                     split = 4;
 667
 668                 for (x = 0, shifter = 0; x < 8; x++, shifter += 2)
 669                     *s->pixel_ptr++ = P[split + ((flags >> shifter) & 0x03)];
 670
 671                 s->pixel_ptr += s->line_inc;
 672             }
 673         }
 674     }
 675
 676     /* report success */
 677     return 0;
 678 }
 679
 680 static int ipvideo_decode_block_opcode_0xB(IpvideoContext *s)
 681 {
 682     int x, y;
 683
 684     /* 64-color encoding (each pixel in block is a different color) */
 685     CHECK_STREAM_PTR(64);
 686
 687     for (y = 0; y < 8; y++) {
 688         for (x = 0; x < 8; x++) {
 689             *s->pixel_ptr++ = *s->stream_ptr++;
 690         }
 691         s->pixel_ptr += s->line_inc;
 692     }
 693
 694     /* report success */
 695     return 0;
 696 }
 697
 698 static int ipvideo_decode_block_opcode_0xC(IpvideoContext *s)
 699 {
 700     int x, y;
 701     unsigned char pix;
 702
 703     /* 16-color block encoding: each 2x2 block is a different color */
 704     CHECK_STREAM_PTR(16);
 705
 706     for (y = 0; y < 8; y += 2) {
 707         for (x = 0; x < 8; x += 2) {
 708             pix = *s->stream_ptr++;
 709             *(s->pixel_ptr + x) = pix;
 710             *(s->pixel_ptr + x + 1) = pix;
 711             *(s->pixel_ptr + s->stride + x) = pix;
 712             *(s->pixel_ptr + s->stride + x + 1) = pix;
 713         }
 714         s->pixel_ptr += s->stride * 2;
 715     }
 716
 717     /* report success */
 718     return 0;
 719 }
 720
 721 static int ipvideo_decode_block_opcode_0xD(IpvideoContext *s)
 722 {
 723     int x, y;
 724     unsigned char P[4];
 725     unsigned char index = 0;
 726
 727     /* 4-color block encoding: each 4x4 block is a different color */
 728     CHECK_STREAM_PTR(4);
 729
 730     for (y = 0; y < 4; y++)
 731         P[y] = *s->stream_ptr++;
 732
 733     for (y = 0; y < 8; y++) {
 734         if (y < 4)
 735             index = 0;
 736         else
 737             index = 2;
 738
 739         for (x = 0; x < 8; x++) {
 740             if (x == 4)
 741                 index++;
 742             *s->pixel_ptr++ = P[index];
 743         }
 744         s->pixel_ptr += s->line_inc;
 745     }
 746
 747     /* report success */
 748     return 0;
 749 }
 750
 751 static int ipvideo_decode_block_opcode_0xE(IpvideoContext *s)
 752 {
 753     int x, y;
 754     unsigned char pix;
 755
 756     /* 1-color encoding: the whole block is 1 solid color */
 757     CHECK_STREAM_PTR(1);
 758     pix = *s->stream_ptr++;
 759
 760     for (y = 0; y < 8; y++) {
 761         for (x = 0; x < 8; x++) {
 762             *s->pixel_ptr++ = pix;
 763         }
 764         s->pixel_ptr += s->line_inc;
 765     }
 766
 767     /* report success */
 768     return 0;
 769 }
 770
 771 static int ipvideo_decode_block_opcode_0xF(IpvideoContext *s)
 772 {
 773     int x, y;
 774     unsigned char sample0, sample1;
 775
 776     /* dithered encoding */
 777     CHECK_STREAM_PTR(2);
 778     sample0 = *s->stream_ptr++;
 779     sample1 = *s->stream_ptr++;
 780
 781     for (y = 0; y < 8; y++) {
 782         for (x = 0; x < 8; x += 2) {
 783             if (y & 1) {
 784                 *s->pixel_ptr++ = sample1;
 785                 *s->pixel_ptr++ = sample0;
 786             } else {
 787                 *s->pixel_ptr++ = sample0;
 788                 *s->pixel_ptr++ = sample1;
 789             }
 790         }
 791         s->pixel_ptr += s->line_inc;
 792     }
 793
 794     /* report success */
 795     return 0;
 796 }
 797
 798 static int (*ipvideo_decode_block[16])(IpvideoContext *s);
 799
 800 static void ipvideo_decode_opcodes(IpvideoContext *s)
 801 {
 802     int x, y;
 803     int index = 0;
 804     unsigned char opcode;
 805     int ret;
 806     int code_counts[16];
 807     static int frame = 0;
 808
 809     debug_interplay("------------------ frame %d\n", frame);
 810     frame++;
 811
 812     for (x = 0; x < 16; x++)
 813         code_counts[x] = 0;
 814
 815     /* this is PAL8, so make the palette available */
 816     memcpy(s->current_frame.data[1], s->avctx->palctrl->palette, PALETTE_COUNT * 4);
 817
 818     s->stride = s->current_frame.linesize[0];
 819     s->stream_ptr = s->buf + 14;  /* data starts 14 bytes in */
 820     s->stream_end = s->buf + s->size;
 821     s->line_inc = s->stride - 8;
 822     s->upper_motion_limit_offset = (s->avctx->height - 8) * s->stride
 823         + s->avctx->width - 8;
 824     s->dsp = s->dsp;
 825
 826     for (y = 0; y < (s->stride * s->avctx->height); y += s->stride * 8) {
 827         for (x = y; x < y + s->avctx->width; x += 8) {
 828             /* bottom nibble first, then top nibble (which makes it
 829              * hard to use a GetBitcontext) */
 830             if (index & 1)
 831                 opcode = s->decoding_map[index >> 1] >> 4;
 832             else
 833                 opcode = s->decoding_map[index >> 1] & 0xF;
 834             index++;
 835
 836             debug_interplay("  block @ (%3d, %3d): encoding 0x%X, data ptr @ %p\n",
 837                 x - y, y / s->stride, opcode, s->stream_ptr);
 838             code_counts[opcode]++;
 839
 840             s->pixel_ptr = s->current_frame.data[0] + x;
 841             ret = ipvideo_decode_block[opcode](s);
 842             if (ret != 0) {
 843                 av_log(s->avctx, AV_LOG_ERROR, " Interplay video: decode problem on frame %d, @ block (%d, %d)\n",
 844                     frame, x - y, y / s->stride);
 845                 return;
 846             }
 847         }
 848     }
 849     if ((s->stream_ptr != s->stream_end) &&
 850         (s->stream_ptr + 1 != s->stream_end)) {
 851         av_log(s->avctx, AV_LOG_ERROR, " Interplay video: decode finished with %td bytes left over\n",
 852             s->stream_end - s->stream_ptr);
 853     }
 854 }
 855
 856 static int ipvideo_decode_init(AVCodecContext *avctx)
 857 {
 858     IpvideoContext *s = avctx->priv_data;
 859
 860     s->avctx = avctx;
 861
 862     if (s->avctx->palctrl == NULL) {
 863         av_log(avctx, AV_LOG_ERROR, " Interplay video: palette expected.\n");
 864         return -1;
 865     }
 866
 867     avctx->pix_fmt = PIX_FMT_PAL8;
 868     avctx->has_b_frames = 0;
 869     dsputil_init(&s->dsp, avctx);
 870
 871     /* decoding map contains 4 bits of information per 8x8 block */
 872     s->decoding_map_size = avctx->width * avctx->height / (8 * 8 * 2);
 873
 874     /* assign block decode functions */
 875     ipvideo_decode_block[0x0] = ipvideo_decode_block_opcode_0x0;
 876     ipvideo_decode_block[0x1] = ipvideo_decode_block_opcode_0x1;
 877     ipvideo_decode_block[0x2] = ipvideo_decode_block_opcode_0x2;
 878     ipvideo_decode_block[0x3] = ipvideo_decode_block_opcode_0x3;
 879     ipvideo_decode_block[0x4] = ipvideo_decode_block_opcode_0x4;
 880     ipvideo_decode_block[0x5] = ipvideo_decode_block_opcode_0x5;
 881     ipvideo_decode_block[0x6] = ipvideo_decode_block_opcode_0x6;
 882     ipvideo_decode_block[0x7] = ipvideo_decode_block_opcode_0x7;
 883     ipvideo_decode_block[0x8] = ipvideo_decode_block_opcode_0x8;
 884     ipvideo_decode_block[0x9] = ipvideo_decode_block_opcode_0x9;
 885     ipvideo_decode_block[0xA] = ipvideo_decode_block_opcode_0xA;
 886     ipvideo_decode_block[0xB] = ipvideo_decode_block_opcode_0xB;
 887     ipvideo_decode_block[0xC] = ipvideo_decode_block_opcode_0xC;
 888     ipvideo_decode_block[0xD] = ipvideo_decode_block_opcode_0xD;
 889     ipvideo_decode_block[0xE] = ipvideo_decode_block_opcode_0xE;
 890     ipvideo_decode_block[0xF] = ipvideo_decode_block_opcode_0xF;
 891
 892     s->current_frame.data[0] = s->last_frame.data[0] =
 893     s->second_last_frame.data[0] = NULL;
 894
 895     return 0;
 896 }
 897
 898 static int ipvideo_decode_frame(AVCodecContext *avctx,
 899                                 void *data, int *data_size,
 900                                 uint8_t *buf, int buf_size)
 901 {
 902     IpvideoContext *s = avctx->priv_data;
 903     AVPaletteControl *palette_control = avctx->palctrl;
 904
 905     /* compressed buffer needs to be large enough to at least hold an entire
 906      * decoding map */
 907     if (buf_size < s->decoding_map_size)
 908         return buf_size;
 909
 910     s->decoding_map = buf;
 911     s->buf = buf + s->decoding_map_size;
 912     s->size = buf_size - s->decoding_map_size;
 913
 914     s->current_frame.reference = 3;
 915     if (avctx->get_buffer(avctx, &s->current_frame)) {
 916         av_log(avctx, AV_LOG_ERROR, "  Interplay Video: get_buffer() failed\n");
 917         return -1;
 918     }
 919
 920     ipvideo_decode_opcodes(s);
 921
 922     if (palette_control->palette_changed) {
 923         palette_control->palette_changed = 0;
 924         s->current_frame.palette_has_changed = 1;
 925     }
 926
 927     *data_size = sizeof(AVFrame);
 928     *(AVFrame*)data = s->current_frame;
 929
 930     /* shuffle frames */
 931     if (s->second_last_frame.data[0])
 932         avctx->release_buffer(avctx, &s->second_last_frame);
 933     s->second_last_frame = s->last_frame;
 934     s->last_frame = s->current_frame;
 935     s->current_frame.data[0] = NULL;  /* catch any access attempts */
 936
 937     /* report that the buffer was completely consumed */
 938     return buf_size;
 939 }
 940
 941 static int ipvideo_decode_end(AVCodecContext *avctx)
 942 {
 943     IpvideoContext *s = avctx->priv_data;
 944
 945     /* release the last frame */
 946     if (s->last_frame.data[0])
 947         avctx->release_buffer(avctx, &s->last_frame);
 948     if (s->second_last_frame.data[0])
 949         avctx->release_buffer(avctx, &s->second_last_frame);
 950
 951     return 0;
 952 }
 953
 954 AVCodec interplay_video_decoder = {
 955     "interplayvideo",
 956     CODEC_TYPE_VIDEO,
 957     CODEC_ID_INTERPLAY_VIDEO,
 958     sizeof(IpvideoContext),
 959     ipvideo_decode_init,
 960     NULL,
 961     ipvideo_decode_end,
 962     ipvideo_decode_frame,
 963     CODEC_CAP_DR1,
 964 };