Commit | Line | Data |
---|---|---|
2ba45a60 DM |
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 | }; |