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.

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