95059c365d1a579ef4b0389befbe3d2423879c89
[ffmpeg.git] / 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 };