You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

1148 lines
39KB

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