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