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.

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