?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
Remove the 3-front-channel layout from the list of channel layout
[FFMpeg-mirror/lagarith.git] / libavcodec / indeo3.c
blob5851e538e552e1bba5867de02008a08875b1eaca
1 /*
2 * Intel Indeo 3 (IV31, IV32, etc.) video decoder for ffmpeg
3 * written, produced, and directed by Alan Smithee
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 #include <stdio.h>
23 #include <stdlib.h>
24 #include <string.h>
25 #include <unistd.h>
26
27 #include "avcodec.h"
28 #include "dsputil.h"
29 #include "bytestream.h"
30
31 #include "indeo3data.h"
32
33 typedef struct
34 {
35 uint8_t *Ybuf;
36 uint8_t *Ubuf;
37 uint8_t *Vbuf;
38 unsigned short y_w, y_h;
39 unsigned short uv_w, uv_h;
40 } YUVBufs;
41
42 typedef struct Indeo3DecodeContext {
43 AVCodecContext *avctx;
44 int width, height;
45 AVFrame frame;
46
47 uint8_t *buf;
48 YUVBufs iv_frame[2];
49 YUVBufs *cur_frame;
50 YUVBufs *ref_frame;
51
52 uint8_t *ModPred;
53 uint8_t *corrector_type;
54 } Indeo3DecodeContext;
55
56 static const uint8_t corrector_type_0[24] = {
57 195, 159, 133, 115, 101, 93, 87, 77,
58 195, 159, 133, 115, 101, 93, 87, 77,
59 128, 79, 79, 79, 79, 79, 79, 79
60 };
61
62 static const uint8_t corrector_type_2[8] = { 9, 7, 6, 8, 5, 4, 3, 2 };
63
64 static av_cold int build_modpred(Indeo3DecodeContext *s)
65 {
66 int i, j;
67
68 if (!(s->ModPred = av_malloc(8 * 128)))
69 return AVERROR(ENOMEM);
70
71 for (i=0; i < 128; ++i) {
72 s->ModPred[i+0*128] = i > 126 ? 254 : 2*(i + 1 - ((i + 1) % 2));
73 s->ModPred[i+1*128] = i == 7 ? 20 :
74 i == 119 ||
75 i == 120 ? 236 : 2*(i + 2 - ((i + 1) % 3));
76 s->ModPred[i+2*128] = i > 125 ? 248 : 2*(i + 2 - ((i + 2) % 4));
77 s->ModPred[i+3*128] = 2*(i + 1 - ((i - 3) % 5));
78 s->ModPred[i+4*128] = i == 8 ? 20 : 2*(i + 1 - ((i - 3) % 6));
79 s->ModPred[i+5*128] = 2*(i + 4 - ((i + 3) % 7));
80 s->ModPred[i+6*128] = i > 123 ? 240 : 2*(i + 4 - ((i + 4) % 8));
81 s->ModPred[i+7*128] = 2*(i + 5 - ((i + 4) % 9));
82 }
83
84 if (!(s->corrector_type = av_malloc(24 * 256)))
85 return AVERROR(ENOMEM);
86
87 for (i=0; i < 24; ++i) {
88 for (j=0; j < 256; ++j) {
89 s->corrector_type[i*256+j] = j < corrector_type_0[i] ? 1 :
90 j < 248 || (i == 16 && j == 248) ? 0 :
91 corrector_type_2[j - 248];
92 }
93 }
94
95 return 0;
96 }
97
98 static av_cold int iv_alloc_frames(Indeo3DecodeContext *s)
99 {
100 int luma_width = (s->width + 3) & ~3,
101 luma_height = (s->height + 3) & ~3,
102 chroma_width = ((luma_width >> 2) + 3) & ~3,
103 chroma_height = ((luma_height >> 2) + 3) & ~3,
104 luma_pixels = luma_width * luma_height,
105 chroma_pixels = chroma_width * chroma_height,
106 i;
107 unsigned int bufsize = luma_pixels * 2 + luma_width * 3 +
108 (chroma_pixels + chroma_width) * 4;
109
110 if(!(s->buf = av_malloc(bufsize)))
111 return AVERROR(ENOMEM);
112 s->iv_frame[0].y_w = s->iv_frame[1].y_w = luma_width;
113 s->iv_frame[0].y_h = s->iv_frame[1].y_h = luma_height;
114 s->iv_frame[0].uv_w = s->iv_frame[1].uv_w = chroma_width;
115 s->iv_frame[0].uv_h = s->iv_frame[1].uv_h = chroma_height;
116
117 s->iv_frame[0].Ybuf = s->buf + luma_width;
118 i = luma_pixels + luma_width * 2;
119 s->iv_frame[1].Ybuf = s->buf + i;
120 i += (luma_pixels + luma_width);
121 s->iv_frame[0].Ubuf = s->buf + i;
122 i += (chroma_pixels + chroma_width);
123 s->iv_frame[1].Ubuf = s->buf + i;
124 i += (chroma_pixels + chroma_width);
125 s->iv_frame[0].Vbuf = s->buf + i;
126 i += (chroma_pixels + chroma_width);
127 s->iv_frame[1].Vbuf = s->buf + i;
128
129 for(i = 1; i <= luma_width; i++)
130 s->iv_frame[0].Ybuf[-i] = s->iv_frame[1].Ybuf[-i] =
131 s->iv_frame[0].Ubuf[-i] = 0x80;
132
133 for(i = 1; i <= chroma_width; i++) {
134 s->iv_frame[1].Ubuf[-i] = 0x80;
135 s->iv_frame[0].Vbuf[-i] = 0x80;
136 s->iv_frame[1].Vbuf[-i] = 0x80;
137 s->iv_frame[1].Vbuf[chroma_pixels+i-1] = 0x80;
138 }
139
140 return 0;
141 }
142
143 static av_cold void iv_free_func(Indeo3DecodeContext *s)
144 {
145 av_free(s->buf);
146 av_free(s->ModPred);
147 av_free(s->corrector_type);
148 }
149
150 struct ustr {
151 long xpos;
152 long ypos;
153 long width;
154 long height;
155 long split_flag;
156 long split_direction;
157 long usl7;
158 };
159
160
161 #define LV1_CHECK(buf1,rle_v3,lv1,lp2) \
162 if((lv1 & 0x80) != 0) { \
163 if(rle_v3 != 0) \
164 rle_v3 = 0; \
165 else { \
166 rle_v3 = 1; \
167 buf1 -= 2; \
168 } \
169 } \
170 lp2 = 4;
171
172
173 #define RLE_V3_CHECK(buf1,rle_v1,rle_v2,rle_v3) \
174 if(rle_v3 == 0) { \
175 rle_v2 = *buf1; \
176 rle_v1 = 1; \
177 if(rle_v2 > 32) { \
178 rle_v2 -= 32; \
179 rle_v1 = 0; \
180 } \
181 rle_v3 = 1; \
182 } \
183 buf1--;
184
185
186 #define LP2_CHECK(buf1,rle_v3,lp2) \
187 if(lp2 == 0 && rle_v3 != 0) \
188 rle_v3 = 0; \
189 else { \
190 buf1--; \
191 rle_v3 = 1; \
192 }
193
194
195 #define RLE_V2_CHECK(buf1,rle_v2, rle_v3,lp2) \
196 rle_v2--; \
197 if(rle_v2 == 0) { \
198 rle_v3 = 0; \
199 buf1 += 2; \
200 } \
201 lp2 = 4;
202
203 static void iv_Decode_Chunk(Indeo3DecodeContext *s,
204 uint8_t *cur, uint8_t *ref, int width, int height,
205 const uint8_t *buf1, long cb_offset, const uint8_t *hdr,
206 const uint8_t *buf2, int min_width_160)
207 {
208 uint8_t bit_buf;
209 unsigned long bit_pos, lv, lv1, lv2;
210 long *width_tbl, width_tbl_arr[10];
211 const signed char *ref_vectors;
212 uint8_t *cur_frm_pos, *ref_frm_pos, *cp, *cp2;
213 uint32_t *cur_lp, *ref_lp;
214 const uint32_t *correction_lp[2], *correctionloworder_lp[2], *correctionhighorder_lp[2];
215 uint8_t *correction_type_sp[2];
216 struct ustr strip_tbl[20], *strip;
217 int i, j, k, lp1, lp2, flag1, cmd, blks_width, blks_height, region_160_width,
218 rle_v1, rle_v2, rle_v3;
219 unsigned short res;
220
221 bit_buf = 0;
222 ref_vectors = NULL;
223
224 width_tbl = width_tbl_arr + 1;
225 i = (width < 0 ? width + 3 : width)/4;
226 for(j = -1; j < 8; j++)
227 width_tbl[j] = i * j;
228
229 strip = strip_tbl;
230
231 for(region_160_width = 0; region_160_width < (width - min_width_160); region_160_width += min_width_160);
232
233 strip->ypos = strip->xpos = 0;
234 for(strip->width = min_width_160; width > strip->width; strip->width *= 2);
235 strip->height = height;
236 strip->split_direction = 0;
237 strip->split_flag = 0;
238 strip->usl7 = 0;
239
240 bit_pos = 0;
241
242 rle_v1 = rle_v2 = rle_v3 = 0;
243
244 while(strip >= strip_tbl) {
245 if(bit_pos <= 0) {
246 bit_pos = 8;
247 bit_buf = *buf1++;
248 }
249
250 bit_pos -= 2;
251 cmd = (bit_buf >> bit_pos) & 0x03;
252
253 if(cmd == 0) {
254 strip++;
255 if(strip >= strip_tbl + FF_ARRAY_ELEMS(strip_tbl)) {
256 av_log(s->avctx, AV_LOG_WARNING, "out of range strip\n");
257 break;
258 }
259 memcpy(strip, strip-1, sizeof(*strip));
260 strip->split_flag = 1;
261 strip->split_direction = 0;
262 strip->height = (strip->height > 8 ? ((strip->height+8)>>4)<<3 : 4);
263 continue;
264 } else if(cmd == 1) {
265 strip++;
266 if(strip >= strip_tbl + FF_ARRAY_ELEMS(strip_tbl)) {
267 av_log(s->avctx, AV_LOG_WARNING, "out of range strip\n");
268 break;
269 }
270 memcpy(strip, strip-1, sizeof(*strip));
271 strip->split_flag = 1;
272 strip->split_direction = 1;
273 strip->width = (strip->width > 8 ? ((strip->width+8)>>4)<<3 : 4);
274 continue;
275 } else if(cmd == 2) {
276 if(strip->usl7 == 0) {
277 strip->usl7 = 1;
278 ref_vectors = NULL;
279 continue;
280 }
281 } else if(cmd == 3) {
282 if(strip->usl7 == 0) {
283 strip->usl7 = 1;
284 ref_vectors = (const signed char*)buf2 + (*buf1 * 2);
285 buf1++;
286 continue;
287 }
288 }
289
290 cur_frm_pos = cur + width * strip->ypos + strip->xpos;
291
292 if((blks_width = strip->width) < 0)
293 blks_width += 3;
294 blks_width >>= 2;
295 blks_height = strip->height;
296
297 if(ref_vectors != NULL) {
298 ref_frm_pos = ref + (ref_vectors[0] + strip->ypos) * width +
299 ref_vectors[1] + strip->xpos;
300 } else
301 ref_frm_pos = cur_frm_pos - width_tbl[4];
302
303 if(cmd == 2) {
304 if(bit_pos <= 0) {
305 bit_pos = 8;
306 bit_buf = *buf1++;
307 }
308
309 bit_pos -= 2;
310 cmd = (bit_buf >> bit_pos) & 0x03;
311
312 if(cmd == 0 || ref_vectors != NULL) {
313 for(lp1 = 0; lp1 < blks_width; lp1++) {
314 for(i = 0, j = 0; i < blks_height; i++, j += width_tbl[1])
315 ((uint32_t *)cur_frm_pos)[j] = ((uint32_t *)ref_frm_pos)[j];
316 cur_frm_pos += 4;
317 ref_frm_pos += 4;
318 }
319 } else if(cmd != 1)
320 return;
321 } else {
322 k = *buf1 >> 4;
323 j = *buf1 & 0x0f;
324 buf1++;
325 lv = j + cb_offset;
326
327 if((lv - 8) <= 7 && (k == 0 || k == 3 || k == 10)) {
328 cp2 = s->ModPred + ((lv - 8) << 7);
329 cp = ref_frm_pos;
330 for(i = 0; i < blks_width << 2; i++) {
331 int v = *cp >> 1;
332 *(cp++) = cp2[v];
333 }
334 }
335
336 if(k == 1 || k == 4) {
337 lv = (hdr[j] & 0xf) + cb_offset;
338 correction_type_sp[0] = s->corrector_type + (lv << 8);
339 correction_lp[0] = correction + (lv << 8);
340 lv = (hdr[j] >> 4) + cb_offset;
341 correction_lp[1] = correction + (lv << 8);
342 correction_type_sp[1] = s->corrector_type + (lv << 8);
343 } else {
344 correctionloworder_lp[0] = correctionloworder_lp[1] = correctionloworder + (lv << 8);
345 correctionhighorder_lp[0] = correctionhighorder_lp[1] = correctionhighorder + (lv << 8);
346 correction_type_sp[0] = correction_type_sp[1] = s->corrector_type + (lv << 8);
347 correction_lp[0] = correction_lp[1] = correction + (lv << 8);
348 }
349
350 switch(k) {
351 case 1:
352 case 0: /********** CASE 0 **********/
353 for( ; blks_height > 0; blks_height -= 4) {
354 for(lp1 = 0; lp1 < blks_width; lp1++) {
355 for(lp2 = 0; lp2 < 4; ) {
356 k = *buf1++;
357 cur_lp = ((uint32_t *)cur_frm_pos) + width_tbl[lp2];
358 ref_lp = ((uint32_t *)ref_frm_pos) + width_tbl[lp2];
359
360 switch(correction_type_sp[0][k]) {
361 case 0:
362 *cur_lp = le2me_32(((le2me_32(*ref_lp) >> 1) + correction_lp[lp2 & 0x01][k]) << 1);
363 lp2++;
364 break;
365 case 1:
366 res = ((le2me_16(((unsigned short *)(ref_lp))[0]) >> 1) + correction_lp[lp2 & 0x01][*buf1]) << 1;
367 ((unsigned short *)cur_lp)[0] = le2me_16(res);
368 res = ((le2me_16(((unsigned short *)(ref_lp))[1]) >> 1) + correction_lp[lp2 & 0x01][k]) << 1;
369 ((unsigned short *)cur_lp)[1] = le2me_16(res);
370 buf1++;
371 lp2++;
372 break;
373 case 2:
374 if(lp2 == 0) {
375 for(i = 0, j = 0; i < 2; i++, j += width_tbl[1])
376 cur_lp[j] = ref_lp[j];
377 lp2 += 2;
378 }
379 break;
380 case 3:
381 if(lp2 < 2) {
382 for(i = 0, j = 0; i < (3 - lp2); i++, j += width_tbl[1])
383 cur_lp[j] = ref_lp[j];
384 lp2 = 3;
385 }
386 break;
387 case 8:
388 if(lp2 == 0) {
389 RLE_V3_CHECK(buf1,rle_v1,rle_v2,rle_v3)
390
391 if(rle_v1 == 1 || ref_vectors != NULL) {
392 for(i = 0, j = 0; i < 4; i++, j += width_tbl[1])
393 cur_lp[j] = ref_lp[j];
394 }
395
396 RLE_V2_CHECK(buf1,rle_v2, rle_v3,lp2)
397 break;
398 } else {
399 rle_v1 = 1;
400 rle_v2 = *buf1 - 1;
401 }
402 case 5:
403 LP2_CHECK(buf1,rle_v3,lp2)
404 case 4:
405 for(i = 0, j = 0; i < (4 - lp2); i++, j += width_tbl[1])
406 cur_lp[j] = ref_lp[j];
407 lp2 = 4;
408 break;
409
410 case 7:
411 if(rle_v3 != 0)
412 rle_v3 = 0;
413 else {
414 buf1--;
415 rle_v3 = 1;
416 }
417 case 6:
418 if(ref_vectors != NULL) {
419 for(i = 0, j = 0; i < 4; i++, j += width_tbl[1])
420 cur_lp[j] = ref_lp[j];
421 }
422 lp2 = 4;
423 break;
424
425 case 9:
426 lv1 = *buf1++;
427 lv = (lv1 & 0x7F) << 1;
428 lv += (lv << 8);
429 lv += (lv << 16);
430 for(i = 0, j = 0; i < 4; i++, j += width_tbl[1])
431 cur_lp[j] = lv;
432
433 LV1_CHECK(buf1,rle_v3,lv1,lp2)
434 break;
435 default:
436 return;
437 }
438 }
439
440 cur_frm_pos += 4;
441 ref_frm_pos += 4;
442 }
443
444 cur_frm_pos += ((width - blks_width) * 4);
445 ref_frm_pos += ((width - blks_width) * 4);
446 }
447 break;
448
449 case 4:
450 case 3: /********** CASE 3 **********/
451 if(ref_vectors != NULL)
452 return;
453 flag1 = 1;
454
455 for( ; blks_height > 0; blks_height -= 8) {
456 for(lp1 = 0; lp1 < blks_width; lp1++) {
457 for(lp2 = 0; lp2 < 4; ) {
458 k = *buf1++;
459
460 cur_lp = ((uint32_t *)cur_frm_pos) + width_tbl[lp2 * 2];
461 ref_lp = ((uint32_t *)cur_frm_pos) + width_tbl[(lp2 * 2) - 1];
462
463 switch(correction_type_sp[lp2 & 0x01][k]) {
464 case 0:
465 cur_lp[width_tbl[1]] = le2me_32(((le2me_32(*ref_lp) >> 1) + correction_lp[lp2 & 0x01][k]) << 1);
466 if(lp2 > 0 || flag1 == 0 || strip->ypos != 0)
467 cur_lp[0] = ((cur_lp[-width_tbl[1]] >> 1) + (cur_lp[width_tbl[1]] >> 1)) & 0xFEFEFEFE;
468 else
469 cur_lp[0] = le2me_32(((le2me_32(*ref_lp) >> 1) + correction_lp[lp2 & 0x01][k]) << 1);
470 lp2++;
471 break;
472
473 case 1:
474 res = ((le2me_16(((unsigned short *)ref_lp)[0]) >> 1) + correction_lp[lp2 & 0x01][*buf1]) << 1;
475 ((unsigned short *)cur_lp)[width_tbl[2]] = le2me_16(res);
476 res = ((le2me_16(((unsigned short *)ref_lp)[1]) >> 1) + correction_lp[lp2 & 0x01][k]) << 1;
477 ((unsigned short *)cur_lp)[width_tbl[2]+1] = le2me_16(res);
478
479 if(lp2 > 0 || flag1 == 0 || strip->ypos != 0)
480 cur_lp[0] = ((cur_lp[-width_tbl[1]] >> 1) + (cur_lp[width_tbl[1]] >> 1)) & 0xFEFEFEFE;
481 else
482 cur_lp[0] = cur_lp[width_tbl[1]];
483 buf1++;
484 lp2++;
485 break;
486
487 case 2:
488 if(lp2 == 0) {
489 for(i = 0, j = 0; i < 4; i++, j += width_tbl[1])
490 cur_lp[j] = *ref_lp;
491 lp2 += 2;
492 }
493 break;
494
495 case 3:
496 if(lp2 < 2) {
497 for(i = 0, j = 0; i < 6 - (lp2 * 2); i++, j += width_tbl[1])
498 cur_lp[j] = *ref_lp;
499 lp2 = 3;
500 }
501 break;
502
503 case 6:
504 lp2 = 4;
505 break;
506
507 case 7:
508 if(rle_v3 != 0)
509 rle_v3 = 0;
510 else {
511 buf1--;
512 rle_v3 = 1;
513 }
514 lp2 = 4;
515 break;
516
517 case 8:
518 if(lp2 == 0) {
519 RLE_V3_CHECK(buf1,rle_v1,rle_v2,rle_v3)
520
521 if(rle_v1 == 1) {
522 for(i = 0, j = 0; i < 8; i++, j += width_tbl[1])
523 cur_lp[j] = ref_lp[j];
524 }
525
526 RLE_V2_CHECK(buf1,rle_v2, rle_v3,lp2)
527 break;
528 } else {
529 rle_v2 = (*buf1) - 1;
530 rle_v1 = 1;
531 }
532 case 5:
533 LP2_CHECK(buf1,rle_v3,lp2)
534 case 4:
535 for(i = 0, j = 0; i < 8 - (lp2 * 2); i++, j += width_tbl[1])
536 cur_lp[j] = *ref_lp;
537 lp2 = 4;
538 break;
539
540 case 9:
541 av_log(s->avctx, AV_LOG_ERROR, "UNTESTED.\n");
542 lv1 = *buf1++;
543 lv = (lv1 & 0x7F) << 1;
544 lv += (lv << 8);
545 lv += (lv << 16);
546
547 for(i = 0, j = 0; i < 4; i++, j += width_tbl[1])
548 cur_lp[j] = lv;
549
550 LV1_CHECK(buf1,rle_v3,lv1,lp2)
551 break;
552
553 default:
554 return;
555 }
556 }
557
558 cur_frm_pos += 4;
559 }
560
561 cur_frm_pos += (((width * 2) - blks_width) * 4);
562 flag1 = 0;
563 }
564 break;
565
566 case 10: /********** CASE 10 **********/
567 if(ref_vectors == NULL) {
568 flag1 = 1;
569
570 for( ; blks_height > 0; blks_height -= 8) {
571 for(lp1 = 0; lp1 < blks_width; lp1 += 2) {
572 for(lp2 = 0; lp2 < 4; ) {
573 k = *buf1++;
574 cur_lp = ((uint32_t *)cur_frm_pos) + width_tbl[lp2 * 2];
575 ref_lp = ((uint32_t *)cur_frm_pos) + width_tbl[(lp2 * 2) - 1];
576 lv1 = ref_lp[0];
577 lv2 = ref_lp[1];
578 if(lp2 == 0 && flag1 != 0) {
579 #ifdef WORDS_BIGENDIAN
580 lv1 = lv1 & 0xFF00FF00;
581 lv1 = (lv1 >> 8) | lv1;
582 lv2 = lv2 & 0xFF00FF00;
583 lv2 = (lv2 >> 8) | lv2;
584 #else
585 lv1 = lv1 & 0x00FF00FF;
586 lv1 = (lv1 << 8) | lv1;
587 lv2 = lv2 & 0x00FF00FF;
588 lv2 = (lv2 << 8) | lv2;
589 #endif
590 }
591
592 switch(correction_type_sp[lp2 & 0x01][k]) {
593 case 0:
594 cur_lp[width_tbl[1]] = le2me_32(((le2me_32(lv1) >> 1) + correctionloworder_lp[lp2 & 0x01][k]) << 1);
595 cur_lp[width_tbl[1]+1] = le2me_32(((le2me_32(lv2) >> 1) + correctionhighorder_lp[lp2 & 0x01][k]) << 1);
596 if(lp2 > 0 || strip->ypos != 0 || flag1 == 0) {
597 cur_lp[0] = ((cur_lp[-width_tbl[1]] >> 1) + (cur_lp[width_tbl[1]] >> 1)) & 0xFEFEFEFE;
598 cur_lp[1] = ((cur_lp[-width_tbl[1]+1] >> 1) + (cur_lp[width_tbl[1]+1] >> 1)) & 0xFEFEFEFE;
599 } else {
600 cur_lp[0] = cur_lp[width_tbl[1]];
601 cur_lp[1] = cur_lp[width_tbl[1]+1];
602 }
603 lp2++;
604 break;
605
606 case 1:
607 cur_lp[width_tbl[1]] = le2me_32(((le2me_32(lv1) >> 1) + correctionloworder_lp[lp2 & 0x01][*buf1]) << 1);
608 cur_lp[width_tbl[1]+1] = le2me_32(((le2me_32(lv2) >> 1) + correctionloworder_lp[lp2 & 0x01][k]) << 1);
609 if(lp2 > 0 || strip->ypos != 0 || flag1 == 0) {
610 cur_lp[0] = ((cur_lp[-width_tbl[1]] >> 1) + (cur_lp[width_tbl[1]] >> 1)) & 0xFEFEFEFE;
611 cur_lp[1] = ((cur_lp[-width_tbl[1]+1] >> 1) + (cur_lp[width_tbl[1]+1] >> 1)) & 0xFEFEFEFE;
612 } else {
613 cur_lp[0] = cur_lp[width_tbl[1]];
614 cur_lp[1] = cur_lp[width_tbl[1]+1];
615 }
616 buf1++;
617 lp2++;
618 break;
619
620 case 2:
621 if(lp2 == 0) {
622 if(flag1 != 0) {
623 for(i = 0, j = width_tbl[1]; i < 3; i++, j += width_tbl[1]) {
624 cur_lp[j] = lv1;
625 cur_lp[j+1] = lv2;
626 }
627 cur_lp[0] = ((cur_lp[-width_tbl[1]] >> 1) + (cur_lp[width_tbl[1]] >> 1)) & 0xFEFEFEFE;
628 cur_lp[1] = ((cur_lp[-width_tbl[1]+1] >> 1) + (cur_lp[width_tbl[1]+1] >> 1)) & 0xFEFEFEFE;
629 } else {
630 for(i = 0, j = 0; i < 4; i++, j += width_tbl[1]) {
631 cur_lp[j] = lv1;
632 cur_lp[j+1] = lv2;
633 }
634 }
635 lp2 += 2;
636 }
637 break;
638
639 case 3:
640 if(lp2 < 2) {
641 if(lp2 == 0 && flag1 != 0) {
642 for(i = 0, j = width_tbl[1]; i < 5; i++, j += width_tbl[1]) {
643 cur_lp[j] = lv1;
644 cur_lp[j+1] = lv2;
645 }
646 cur_lp[0] = ((cur_lp[-width_tbl[1]] >> 1) + (cur_lp[width_tbl[1]] >> 1)) & 0xFEFEFEFE;
647 cur_lp[1] = ((cur_lp[-width_tbl[1]+1] >> 1) + (cur_lp[width_tbl[1]+1] >> 1)) & 0xFEFEFEFE;
648 } else {
649 for(i = 0, j = 0; i < 6 - (lp2 * 2); i++, j += width_tbl[1]) {
650 cur_lp[j] = lv1;
651 cur_lp[j+1] = lv2;
652 }
653 }
654 lp2 = 3;
655 }
656 break;
657
658 case 8:
659 if(lp2 == 0) {
660 RLE_V3_CHECK(buf1,rle_v1,rle_v2,rle_v3)
661 if(rle_v1 == 1) {
662 if(flag1 != 0) {
663 for(i = 0, j = width_tbl[1]; i < 7; i++, j += width_tbl[1]) {
664 cur_lp[j] = lv1;
665 cur_lp[j+1] = lv2;
666 }
667 cur_lp[0] = ((cur_lp[-width_tbl[1]] >> 1) + (cur_lp[width_tbl[1]] >> 1)) & 0xFEFEFEFE;
668 cur_lp[1] = ((cur_lp[-width_tbl[1]+1] >> 1) + (cur_lp[width_tbl[1]+1] >> 1)) & 0xFEFEFEFE;
669 } else {
670 for(i = 0, j = 0; i < 8; i++, j += width_tbl[1]) {
671 cur_lp[j] = lv1;
672 cur_lp[j+1] = lv2;
673 }
674 }
675 }
676 RLE_V2_CHECK(buf1,rle_v2, rle_v3,lp2)
677 break;
678 } else {
679 rle_v1 = 1;
680 rle_v2 = (*buf1) - 1;
681 }
682 case 5:
683 LP2_CHECK(buf1,rle_v3,lp2)
684 case 4:
685 if(lp2 == 0 && flag1 != 0) {
686 for(i = 0, j = width_tbl[1]; i < 7; i++, j += width_tbl[1]) {
687 cur_lp[j] = lv1;
688 cur_lp[j+1] = lv2;
689 }
690 cur_lp[0] = ((cur_lp[-width_tbl[1]] >> 1) + (cur_lp[width_tbl[1]] >> 1)) & 0xFEFEFEFE;
691 cur_lp[1] = ((cur_lp[-width_tbl[1]+1] >> 1) + (cur_lp[width_tbl[1]+1] >> 1)) & 0xFEFEFEFE;
692 } else {
693 for(i = 0, j = 0; i < 8 - (lp2 * 2); i++, j += width_tbl[1]) {
694 cur_lp[j] = lv1;
695 cur_lp[j+1] = lv2;
696 }
697 }
698 lp2 = 4;
699 break;
700
701 case 6:
702 lp2 = 4;
703 break;
704
705 case 7:
706 if(lp2 == 0) {
707 if(rle_v3 != 0)
708 rle_v3 = 0;
709 else {
710 buf1--;
711 rle_v3 = 1;
712 }
713 lp2 = 4;
714 }
715 break;
716
717 case 9:
718 av_log(s->avctx, AV_LOG_ERROR, "UNTESTED.\n");
719 lv1 = *buf1;
720 lv = (lv1 & 0x7F) << 1;
721 lv += (lv << 8);
722 lv += (lv << 16);
723 for(i = 0, j = 0; i < 8; i++, j += width_tbl[1])
724 cur_lp[j] = lv;
725 LV1_CHECK(buf1,rle_v3,lv1,lp2)
726 break;
727
728 default:
729 return;
730 }
731 }
732
733 cur_frm_pos += 8;
734 }
735
736 cur_frm_pos += (((width * 2) - blks_width) * 4);
737 flag1 = 0;
738 }
739 } else {
740 for( ; blks_height > 0; blks_height -= 8) {
741 for(lp1 = 0; lp1 < blks_width; lp1 += 2) {
742 for(lp2 = 0; lp2 < 4; ) {
743 k = *buf1++;
744 cur_lp = ((uint32_t *)cur_frm_pos) + width_tbl[lp2 * 2];
745 ref_lp = ((uint32_t *)ref_frm_pos) + width_tbl[lp2 * 2];
746
747 switch(correction_type_sp[lp2 & 0x01][k]) {
748 case 0:
749 lv1 = correctionloworder_lp[lp2 & 0x01][k];
750 lv2 = correctionhighorder_lp[lp2 & 0x01][k];
751 cur_lp[0] = le2me_32(((le2me_32(ref_lp[0]) >> 1) + lv1) << 1);
752 cur_lp[1] = le2me_32(((le2me_32(ref_lp[1]) >> 1) + lv2) << 1);
753 cur_lp[width_tbl[1]] = le2me_32(((le2me_32(ref_lp[width_tbl[1]]) >> 1) + lv1) << 1);
754 cur_lp[width_tbl[1]+1] = le2me_32(((le2me_32(ref_lp[width_tbl[1]+1]) >> 1) + lv2) << 1);
755 lp2++;
756 break;
757
758 case 1:
759 lv1 = correctionloworder_lp[lp2 & 0x01][*buf1++];
760 lv2 = correctionloworder_lp[lp2 & 0x01][k];
761 cur_lp[0] = le2me_32(((le2me_32(ref_lp[0]) >> 1) + lv1) << 1);
762 cur_lp[1] = le2me_32(((le2me_32(ref_lp[1]) >> 1) + lv2) << 1);
763 cur_lp[width_tbl[1]] = le2me_32(((le2me_32(ref_lp[width_tbl[1]]) >> 1) + lv1) << 1);
764 cur_lp[width_tbl[1]+1] = le2me_32(((le2me_32(ref_lp[width_tbl[1]+1]) >> 1) + lv2) << 1);
765 lp2++;
766 break;
767
768 case 2:
769 if(lp2 == 0) {
770 for(i = 0, j = 0; i < 4; i++, j += width_tbl[1]) {
771 cur_lp[j] = ref_lp[j];
772 cur_lp[j+1] = ref_lp[j+1];
773 }
774 lp2 += 2;
775 }
776 break;
777
778 case 3:
779 if(lp2 < 2) {
780 for(i = 0, j = 0; i < 6 - (lp2 * 2); i++, j += width_tbl[1]) {
781 cur_lp[j] = ref_lp[j];
782 cur_lp[j+1] = ref_lp[j+1];
783 }
784 lp2 = 3;
785 }
786 break;
787
788 case 8:
789 if(lp2 == 0) {
790 RLE_V3_CHECK(buf1,rle_v1,rle_v2,rle_v3)
791 for(i = 0, j = 0; i < 8; i++, j += width_tbl[1]) {
792 ((uint32_t *)cur_frm_pos)[j] = ((uint32_t *)ref_frm_pos)[j];
793 ((uint32_t *)cur_frm_pos)[j+1] = ((uint32_t *)ref_frm_pos)[j+1];
794 }
795 RLE_V2_CHECK(buf1,rle_v2, rle_v3,lp2)
796 break;
797 } else {
798 rle_v1 = 1;
799 rle_v2 = (*buf1) - 1;
800 }
801 case 5:
802 case 7:
803 LP2_CHECK(buf1,rle_v3,lp2)
804 case 6:
805 case 4:
806 for(i = 0, j = 0; i < 8 - (lp2 * 2); i++, j += width_tbl[1]) {
807 cur_lp[j] = ref_lp[j];
808 cur_lp[j+1] = ref_lp[j+1];
809 }
810 lp2 = 4;
811 break;
812
813 case 9:
814 av_log(s->avctx, AV_LOG_ERROR, "UNTESTED.\n");
815 lv1 = *buf1;
816 lv = (lv1 & 0x7F) << 1;
817 lv += (lv << 8);
818 lv += (lv << 16);
819 for(i = 0, j = 0; i < 8; i++, j += width_tbl[1])
820 ((uint32_t *)cur_frm_pos)[j] = ((uint32_t *)cur_frm_pos)[j+1] = lv;
821 LV1_CHECK(buf1,rle_v3,lv1,lp2)
822 break;
823
824 default:
825 return;
826 }
827 }
828
829 cur_frm_pos += 8;
830 ref_frm_pos += 8;
831 }
832
833 cur_frm_pos += (((width * 2) - blks_width) * 4);
834 ref_frm_pos += (((width * 2) - blks_width) * 4);
835 }
836 }
837 break;
838
839 case 11: /********** CASE 11 **********/
840 if(ref_vectors == NULL)
841 return;
842
843 for( ; blks_height > 0; blks_height -= 8) {
844 for(lp1 = 0; lp1 < blks_width; lp1++) {
845 for(lp2 = 0; lp2 < 4; ) {
846 k = *buf1++;
847 cur_lp = ((uint32_t *)cur_frm_pos) + width_tbl[lp2 * 2];
848 ref_lp = ((uint32_t *)ref_frm_pos) + width_tbl[lp2 * 2];
849
850 switch(correction_type_sp[lp2 & 0x01][k]) {
851 case 0:
852 cur_lp[0] = le2me_32(((le2me_32(*ref_lp) >> 1) + correction_lp[lp2 & 0x01][k]) << 1);
853 cur_lp[width_tbl[1]] = le2me_32(((le2me_32(ref_lp[width_tbl[1]]) >> 1) + correction_lp[lp2 & 0x01][k]) << 1);
854 lp2++;
855 break;
856
857 case 1:
858 lv1 = (unsigned short)(correction_lp[lp2 & 0x01][*buf1++]);
859 lv2 = (unsigned short)(correction_lp[lp2 & 0x01][k]);
860 res = (unsigned short)(((le2me_16(((unsigned short *)ref_lp)[0]) >> 1) + lv1) << 1);
861 ((unsigned short *)cur_lp)[0] = le2me_16(res);
862 res = (unsigned short)(((le2me_16(((unsigned short *)ref_lp)[1]) >> 1) + lv2) << 1);
863 ((unsigned short *)cur_lp)[1] = le2me_16(res);
864 res = (unsigned short)(((le2me_16(((unsigned short *)ref_lp)[width_tbl[2]]) >> 1) + lv1) << 1);
865 ((unsigned short *)cur_lp)[width_tbl[2]] = le2me_16(res);
866 res = (unsigned short)(((le2me_16(((unsigned short *)ref_lp)[width_tbl[2]+1]) >> 1) + lv2) << 1);
867 ((unsigned short *)cur_lp)[width_tbl[2]+1] = le2me_16(res);
868 lp2++;
869 break;
870
871 case 2:
872 if(lp2 == 0) {
873 for(i = 0, j = 0; i < 4; i++, j += width_tbl[1])
874 cur_lp[j] = ref_lp[j];
875 lp2 += 2;
876 }
877 break;
878
879 case 3:
880 if(lp2 < 2) {
881 for(i = 0, j = 0; i < 6 - (lp2 * 2); i++, j += width_tbl[1])
882 cur_lp[j] = ref_lp[j];
883 lp2 = 3;
884 }
885 break;
886
887 case 8:
888 if(lp2 == 0) {
889 RLE_V3_CHECK(buf1,rle_v1,rle_v2,rle_v3)
890
891 for(i = 0, j = 0; i < 8; i++, j += width_tbl[1])
892 cur_lp[j] = ref_lp[j];
893
894 RLE_V2_CHECK(buf1,rle_v2, rle_v3,lp2)
895 break;
896 } else {
897 rle_v1 = 1;
898 rle_v2 = (*buf1) - 1;
899 }
900 case 5:
901 case 7:
902 LP2_CHECK(buf1,rle_v3,lp2)
903 case 4:
904 case 6:
905 for(i = 0, j = 0; i < 8 - (lp2 * 2); i++, j += width_tbl[1])
906 cur_lp[j] = ref_lp[j];
907 lp2 = 4;
908 break;
909
910 case 9:
911 av_log(s->avctx, AV_LOG_ERROR, "UNTESTED.\n");
912 lv1 = *buf1++;
913 lv = (lv1 & 0x7F) << 1;
914 lv += (lv << 8);
915 lv += (lv << 16);
916 for(i = 0, j = 0; i < 4; i++, j += width_tbl[1])
917 cur_lp[j] = lv;
918 LV1_CHECK(buf1,rle_v3,lv1,lp2)
919 break;
920
921 default:
922 return;
923 }
924 }
925
926 cur_frm_pos += 4;
927 ref_frm_pos += 4;
928 }
929
930 cur_frm_pos += (((width * 2) - blks_width) * 4);
931 ref_frm_pos += (((width * 2) - blks_width) * 4);
932 }
933 break;
934
935 default:
936 return;
937 }
938 }
939
940 for( ; strip >= strip_tbl; strip--) {
941 if(strip->split_flag != 0) {
942 strip->split_flag = 0;
943 strip->usl7 = (strip-1)->usl7;
944
945 if(strip->split_direction) {
946 strip->xpos += strip->width;
947 strip->width = (strip-1)->width - strip->width;
948 if(region_160_width <= strip->xpos && width < strip->width + strip->xpos)
949 strip->width = width - strip->xpos;
950 } else {
951 strip->ypos += strip->height;
952 strip->height = (strip-1)->height - strip->height;
953 }
954 break;
955 }
956 }
957 }
958 }
959
960 static av_cold int indeo3_decode_init(AVCodecContext *avctx)
961 {
962 Indeo3DecodeContext *s = avctx->priv_data;
963 int ret = 0;
964
965 s->avctx = avctx;
966 s->width = avctx->width;
967 s->height = avctx->height;
968 avctx->pix_fmt = PIX_FMT_YUV410P;
969
970 if (!(ret = build_modpred(s)))
971 ret = iv_alloc_frames(s);
972 if (ret)
973 iv_free_func(s);
974
975 return ret;
976 }
977
978 static int iv_decode_frame(Indeo3DecodeContext *s,
979 const uint8_t *buf, int buf_size)
980 {
981 unsigned int image_width, image_height,
982 chroma_width, chroma_height;
983 unsigned long flags, cb_offset, data_size,
984 y_offset, v_offset, u_offset, mc_vector_count;
985 const uint8_t *hdr_pos, *buf_pos;
986
987 buf_pos = buf;
988 buf_pos += 18; /* skip OS header (16 bytes) and version number */
989
990 flags = bytestream_get_le16(&buf_pos);
991 data_size = bytestream_get_le32(&buf_pos);
992 cb_offset = *buf_pos++;
993 buf_pos += 3; /* skip reserved byte and checksum */
994 image_height = bytestream_get_le16(&buf_pos);
995 image_width = bytestream_get_le16(&buf_pos);
996
997 if(avcodec_check_dimensions(NULL, image_width, image_height))
998 return -1;
999
1000 chroma_height = ((image_height >> 2) + 3) & 0x7ffc;
1001 chroma_width = ((image_width >> 2) + 3) & 0x7ffc;
1002 y_offset = bytestream_get_le32(&buf_pos);
1003 v_offset = bytestream_get_le32(&buf_pos);
1004 u_offset = bytestream_get_le32(&buf_pos);
1005 buf_pos += 4; /* reserved */
1006 hdr_pos = buf_pos;
1007 if(data_size == 0x80) return 4;
1008
1009 if(FFMAX3(y_offset, v_offset, u_offset) >= buf_size-16) {
1010 av_log(s->avctx, AV_LOG_ERROR, "y/u/v offset outside buffer\n");
1011 return -1;
1012 }
1013
1014 if(flags & 0x200) {
1015 s->cur_frame = s->iv_frame + 1;
1016 s->ref_frame = s->iv_frame;
1017 } else {
1018 s->cur_frame = s->iv_frame;
1019 s->ref_frame = s->iv_frame + 1;
1020 }
1021
1022 buf_pos = buf + 16 + y_offset;
1023 mc_vector_count = bytestream_get_le32(&buf_pos);
1024 if(2LL*mc_vector_count >= buf_size-16-y_offset) {
1025 av_log(s->avctx, AV_LOG_ERROR, "mc_vector_count too large\n");
1026 return -1;
1027 }
1028
1029 iv_Decode_Chunk(s, s->cur_frame->Ybuf, s->ref_frame->Ybuf, image_width,
1030 image_height, buf_pos + mc_vector_count * 2, cb_offset, hdr_pos, buf_pos,
1031 FFMIN(image_width, 160));
1032
1033 if (!(s->avctx->flags & CODEC_FLAG_GRAY))
1034 {
1035
1036 buf_pos = buf + 16 + v_offset;
1037 mc_vector_count = bytestream_get_le32(&buf_pos);
1038 if(2LL*mc_vector_count >= buf_size-16-v_offset) {
1039 av_log(s->avctx, AV_LOG_ERROR, "mc_vector_count too large\n");
1040 return -1;
1041 }
1042
1043 iv_Decode_Chunk(s, s->cur_frame->Vbuf, s->ref_frame->Vbuf, chroma_width,
1044 chroma_height, buf_pos + mc_vector_count * 2, cb_offset, hdr_pos, buf_pos,
1045 FFMIN(chroma_width, 40));
1046
1047 buf_pos = buf + 16 + u_offset;
1048 mc_vector_count = bytestream_get_le32(&buf_pos);
1049 if(2LL*mc_vector_count >= buf_size-16-u_offset) {
1050 av_log(s->avctx, AV_LOG_ERROR, "mc_vector_count too large\n");
1051 return -1;
1052 }
1053
1054 iv_Decode_Chunk(s, s->cur_frame->Ubuf, s->ref_frame->Ubuf, chroma_width,
1055 chroma_height, buf_pos + mc_vector_count * 2, cb_offset, hdr_pos, buf_pos,
1056 FFMIN(chroma_width, 40));
1057
1058 }
1059
1060 return 8;
1061 }
1062
1063 static int indeo3_decode_frame(AVCodecContext *avctx,
1064 void *data, int *data_size,
1065 AVPacket *avpkt)
1066 {
1067 const uint8_t *buf = avpkt->data;
1068 int buf_size = avpkt->size;
1069 Indeo3DecodeContext *s=avctx->priv_data;
1070 uint8_t *src, *dest;
1071 int y;
1072
1073 if (iv_decode_frame(s, buf, buf_size) < 0)
1074 return -1;
1075
1076 if(s->frame.data[0])
1077 avctx->release_buffer(avctx, &s->frame);
1078
1079 s->frame.reference = 0;
1080 if(avctx->get_buffer(avctx, &s->frame) < 0) {
1081 av_log(s->avctx, AV_LOG_ERROR, "get_buffer() failed\n");
1082 return -1;
1083 }
1084
1085 src = s->cur_frame->Ybuf;
1086 dest = s->frame.data[0];
1087 for (y = 0; y < s->height; y++) {
1088 memcpy(dest, src, s->cur_frame->y_w);
1089 src += s->cur_frame->y_w;
1090 dest += s->frame.linesize[0];
1091 }
1092
1093 if (!(s->avctx->flags & CODEC_FLAG_GRAY))
1094 {
1095 src = s->cur_frame->Ubuf;
1096 dest = s->frame.data[1];
1097 for (y = 0; y < s->height / 4; y++) {
1098 memcpy(dest, src, s->cur_frame->uv_w);
1099 src += s->cur_frame->uv_w;
1100 dest += s->frame.linesize[1];
1101 }
1102
1103 src = s->cur_frame->Vbuf;
1104 dest = s->frame.data[2];
1105 for (y = 0; y < s->height / 4; y++) {
1106 memcpy(dest, src, s->cur_frame->uv_w);
1107 src += s->cur_frame->uv_w;
1108 dest += s->frame.linesize[2];
1109 }
1110 }
1111
1112 *data_size=sizeof(AVFrame);
1113 *(AVFrame*)data= s->frame;
1114
1115 return buf_size;
1116 }
1117
1118 static av_cold int indeo3_decode_end(AVCodecContext *avctx)
1119 {
1120 Indeo3DecodeContext *s = avctx->priv_data;
1121
1122 iv_free_func(s);
1123
1124 return 0;
1125 }
1126
1127 AVCodec indeo3_decoder = {
1128 "indeo3",
1129 CODEC_TYPE_VIDEO,
1130 CODEC_ID_INDEO3,
1131 sizeof(Indeo3DecodeContext),
1132 indeo3_decode_init,
1133 NULL,
1134 indeo3_decode_end,
1135 indeo3_decode_frame,
1136 CODEC_CAP_DR1,
1137 NULL,
1138 .long_name = NULL_IF_CONFIG_SMALL("Intel Indeo 3"),
1139 };
ffmpeg Lagarith lossless decoder