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.

1140 lines
39KB

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