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.

1119 lines
37KB

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