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.

1135 lines
38KB

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