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.

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