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.

918 lines
30KB

  1. /*
  2. * MSMPEG4 backend for encoder and decoder
  3. * Copyright (c) 2001 Fabrice Bellard
  4. * Copyright (c) 2002-2004 Michael Niedermayer <michaelni@gmx.at>
  5. *
  6. * msmpeg4v1 & v2 stuff by Michael Niedermayer <michaelni@gmx.at>
  7. *
  8. * This file is part of Libav.
  9. *
  10. * Libav is free software; you can redistribute it and/or
  11. * modify it under the terms of the GNU Lesser General Public
  12. * License as published by the Free Software Foundation; either
  13. * version 2.1 of the License, or (at your option) any later version.
  14. *
  15. * Libav is distributed in the hope that it will be useful,
  16. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  18. * Lesser General Public License for more details.
  19. *
  20. * You should have received a copy of the GNU Lesser General Public
  21. * License along with Libav; if not, write to the Free Software
  22. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  23. */
  24. #include "avcodec.h"
  25. #include "internal.h"
  26. #include "mpegutils.h"
  27. #include "mpegvideo.h"
  28. #include "msmpeg4.h"
  29. #include "h263.h"
  30. #include "mpeg4video.h"
  31. #include "msmpeg4data.h"
  32. #include "vc1data.h"
  33. #include "wmv2.h"
  34. #define DC_VLC_BITS 9
  35. #define V2_INTRA_CBPC_VLC_BITS 3
  36. #define V2_MB_TYPE_VLC_BITS 7
  37. #define MV_VLC_BITS 9
  38. #define V2_MV_VLC_BITS 9
  39. #define TEX_VLC_BITS 9
  40. #define DEFAULT_INTER_INDEX 3
  41. static inline int msmpeg4v1_pred_dc(MpegEncContext * s, int n,
  42. int32_t **dc_val_ptr)
  43. {
  44. int i;
  45. if (n < 4) {
  46. i= 0;
  47. } else {
  48. i= n-3;
  49. }
  50. *dc_val_ptr= &s->last_dc[i];
  51. return s->last_dc[i];
  52. }
  53. /****************************************/
  54. /* decoding stuff */
  55. VLC ff_mb_non_intra_vlc[4];
  56. static VLC v2_dc_lum_vlc;
  57. static VLC v2_dc_chroma_vlc;
  58. static VLC v2_intra_cbpc_vlc;
  59. static VLC v2_mb_type_vlc;
  60. static VLC v2_mv_vlc;
  61. VLC ff_inter_intra_vlc;
  62. /* This is identical to H.263 except that its range is multiplied by 2. */
  63. static int msmpeg4v2_decode_motion(MpegEncContext * s, int pred, int f_code)
  64. {
  65. int code, val, sign, shift;
  66. code = get_vlc2(&s->gb, v2_mv_vlc.table, V2_MV_VLC_BITS, 2);
  67. ff_dlog(s, "MV code %d at %d %d pred: %d\n", code, s->mb_x,s->mb_y, pred);
  68. if (code < 0)
  69. return 0xffff;
  70. if (code == 0)
  71. return pred;
  72. sign = get_bits1(&s->gb);
  73. shift = f_code - 1;
  74. val = code;
  75. if (shift) {
  76. val = (val - 1) << shift;
  77. val |= get_bits(&s->gb, shift);
  78. val++;
  79. }
  80. if (sign)
  81. val = -val;
  82. val += pred;
  83. if (val <= -64)
  84. val += 64;
  85. else if (val >= 64)
  86. val -= 64;
  87. return val;
  88. }
  89. static int msmpeg4v12_decode_mb(MpegEncContext *s, int16_t block[6][64])
  90. {
  91. int cbp, code, i;
  92. if (s->pict_type == AV_PICTURE_TYPE_P) {
  93. if (s->use_skip_mb_code) {
  94. if (get_bits1(&s->gb)) {
  95. /* skip mb */
  96. s->mb_intra = 0;
  97. for(i=0;i<6;i++)
  98. s->block_last_index[i] = -1;
  99. s->mv_dir = MV_DIR_FORWARD;
  100. s->mv_type = MV_TYPE_16X16;
  101. s->mv[0][0][0] = 0;
  102. s->mv[0][0][1] = 0;
  103. s->mb_skipped = 1;
  104. return 0;
  105. }
  106. }
  107. if(s->msmpeg4_version==2)
  108. code = get_vlc2(&s->gb, v2_mb_type_vlc.table, V2_MB_TYPE_VLC_BITS, 1);
  109. else
  110. code = get_vlc2(&s->gb, ff_h263_inter_MCBPC_vlc.table, INTER_MCBPC_VLC_BITS, 2);
  111. if(code<0 || code>7){
  112. av_log(s->avctx, AV_LOG_ERROR, "cbpc %d invalid at %d %d\n", code, s->mb_x, s->mb_y);
  113. return -1;
  114. }
  115. s->mb_intra = code >>2;
  116. cbp = code & 0x3;
  117. } else {
  118. s->mb_intra = 1;
  119. if(s->msmpeg4_version==2)
  120. cbp= get_vlc2(&s->gb, v2_intra_cbpc_vlc.table, V2_INTRA_CBPC_VLC_BITS, 1);
  121. else
  122. cbp= get_vlc2(&s->gb, ff_h263_intra_MCBPC_vlc.table, INTRA_MCBPC_VLC_BITS, 1);
  123. if(cbp<0 || cbp>3){
  124. av_log(s->avctx, AV_LOG_ERROR, "cbpc %d invalid at %d %d\n", cbp, s->mb_x, s->mb_y);
  125. return -1;
  126. }
  127. }
  128. if (!s->mb_intra) {
  129. int mx, my, cbpy;
  130. cbpy= get_vlc2(&s->gb, ff_h263_cbpy_vlc.table, CBPY_VLC_BITS, 1);
  131. if(cbpy<0){
  132. av_log(s->avctx, AV_LOG_ERROR, "cbpy %d invalid at %d %d\n", cbp, s->mb_x, s->mb_y);
  133. return -1;
  134. }
  135. cbp|= cbpy<<2;
  136. if(s->msmpeg4_version==1 || (cbp&3) != 3) cbp^= 0x3C;
  137. ff_h263_pred_motion(s, 0, 0, &mx, &my);
  138. mx= msmpeg4v2_decode_motion(s, mx, 1);
  139. my= msmpeg4v2_decode_motion(s, my, 1);
  140. s->mv_dir = MV_DIR_FORWARD;
  141. s->mv_type = MV_TYPE_16X16;
  142. s->mv[0][0][0] = mx;
  143. s->mv[0][0][1] = my;
  144. } else {
  145. if(s->msmpeg4_version==2){
  146. s->ac_pred = get_bits1(&s->gb);
  147. cbp|= get_vlc2(&s->gb, ff_h263_cbpy_vlc.table, CBPY_VLC_BITS, 1)<<2; //FIXME check errors
  148. } else{
  149. s->ac_pred = 0;
  150. cbp|= get_vlc2(&s->gb, ff_h263_cbpy_vlc.table, CBPY_VLC_BITS, 1)<<2; //FIXME check errors
  151. if(s->pict_type==AV_PICTURE_TYPE_P) cbp^=0x3C;
  152. }
  153. }
  154. s->bdsp.clear_blocks(s->block[0]);
  155. for (i = 0; i < 6; i++) {
  156. if (ff_msmpeg4_decode_block(s, block[i], i, (cbp >> (5 - i)) & 1, NULL) < 0)
  157. {
  158. av_log(s->avctx, AV_LOG_ERROR, "\nerror while decoding block: %d x %d (%d)\n", s->mb_x, s->mb_y, i);
  159. return -1;
  160. }
  161. }
  162. return 0;
  163. }
  164. static int msmpeg4v34_decode_mb(MpegEncContext *s, int16_t block[6][64])
  165. {
  166. int cbp, code, i;
  167. uint8_t *coded_val;
  168. uint32_t * const mb_type_ptr = &s->current_picture.mb_type[s->mb_x + s->mb_y*s->mb_stride];
  169. if (s->pict_type == AV_PICTURE_TYPE_P) {
  170. if (s->use_skip_mb_code) {
  171. if (get_bits1(&s->gb)) {
  172. /* skip mb */
  173. s->mb_intra = 0;
  174. for(i=0;i<6;i++)
  175. s->block_last_index[i] = -1;
  176. s->mv_dir = MV_DIR_FORWARD;
  177. s->mv_type = MV_TYPE_16X16;
  178. s->mv[0][0][0] = 0;
  179. s->mv[0][0][1] = 0;
  180. s->mb_skipped = 1;
  181. *mb_type_ptr = MB_TYPE_SKIP | MB_TYPE_L0 | MB_TYPE_16x16;
  182. return 0;
  183. }
  184. }
  185. code = get_vlc2(&s->gb, ff_mb_non_intra_vlc[DEFAULT_INTER_INDEX].table, MB_NON_INTRA_VLC_BITS, 3);
  186. if (code < 0)
  187. return -1;
  188. //s->mb_intra = (code & 0x40) ? 0 : 1;
  189. s->mb_intra = (~code & 0x40) >> 6;
  190. cbp = code & 0x3f;
  191. } else {
  192. s->mb_intra = 1;
  193. code = get_vlc2(&s->gb, ff_msmp4_mb_i_vlc.table, MB_INTRA_VLC_BITS, 2);
  194. if (code < 0)
  195. return -1;
  196. /* predict coded block pattern */
  197. cbp = 0;
  198. for(i=0;i<6;i++) {
  199. int val = ((code >> (5 - i)) & 1);
  200. if (i < 4) {
  201. int pred = ff_msmpeg4_coded_block_pred(s, i, &coded_val);
  202. val = val ^ pred;
  203. *coded_val = val;
  204. }
  205. cbp |= val << (5 - i);
  206. }
  207. }
  208. if (!s->mb_intra) {
  209. int mx, my;
  210. if(s->per_mb_rl_table && cbp){
  211. s->rl_table_index = decode012(&s->gb);
  212. s->rl_chroma_table_index = s->rl_table_index;
  213. }
  214. ff_h263_pred_motion(s, 0, 0, &mx, &my);
  215. if (ff_msmpeg4_decode_motion(s, &mx, &my) < 0)
  216. return -1;
  217. s->mv_dir = MV_DIR_FORWARD;
  218. s->mv_type = MV_TYPE_16X16;
  219. s->mv[0][0][0] = mx;
  220. s->mv[0][0][1] = my;
  221. *mb_type_ptr = MB_TYPE_L0 | MB_TYPE_16x16;
  222. } else {
  223. ff_dlog(s, "I at %d %d %d %06X\n", s->mb_x, s->mb_y,
  224. ((cbp & 3) ? 1 : 0) +((cbp & 0x3C)? 2 : 0),
  225. show_bits(&s->gb, 24));
  226. s->ac_pred = get_bits1(&s->gb);
  227. *mb_type_ptr = MB_TYPE_INTRA;
  228. if(s->inter_intra_pred){
  229. s->h263_aic_dir= get_vlc2(&s->gb, ff_inter_intra_vlc.table, INTER_INTRA_VLC_BITS, 1);
  230. ff_dlog(s, "%d%d %d %d/",
  231. s->ac_pred, s->h263_aic_dir, s->mb_x, s->mb_y);
  232. }
  233. if(s->per_mb_rl_table && cbp){
  234. s->rl_table_index = decode012(&s->gb);
  235. s->rl_chroma_table_index = s->rl_table_index;
  236. }
  237. }
  238. s->bdsp.clear_blocks(s->block[0]);
  239. for (i = 0; i < 6; i++) {
  240. if (ff_msmpeg4_decode_block(s, block[i], i, (cbp >> (5 - i)) & 1, NULL) < 0)
  241. {
  242. av_log(s->avctx, AV_LOG_ERROR, "\nerror while decoding block: %d x %d (%d)\n", s->mb_x, s->mb_y, i);
  243. return -1;
  244. }
  245. }
  246. return 0;
  247. }
  248. /* init all vlc decoding tables */
  249. av_cold int ff_msmpeg4_decode_init(AVCodecContext *avctx)
  250. {
  251. MpegEncContext *s = avctx->priv_data;
  252. static int done = 0;
  253. int i;
  254. MVTable *mv;
  255. if (ff_h263_decode_init(avctx) < 0)
  256. return -1;
  257. ff_msmpeg4_common_init(s);
  258. if (!done) {
  259. done = 1;
  260. for(i=0;i<NB_RL_TABLES;i++) {
  261. ff_rl_init(&ff_rl_table[i], ff_static_rl_table_store[i]);
  262. }
  263. INIT_VLC_RL(ff_rl_table[0], 642);
  264. INIT_VLC_RL(ff_rl_table[1], 1104);
  265. INIT_VLC_RL(ff_rl_table[2], 554);
  266. INIT_VLC_RL(ff_rl_table[3], 940);
  267. INIT_VLC_RL(ff_rl_table[4], 962);
  268. INIT_VLC_RL(ff_rl_table[5], 554);
  269. mv = &ff_mv_tables[0];
  270. INIT_VLC_STATIC(&mv->vlc, MV_VLC_BITS, mv->n + 1,
  271. mv->table_mv_bits, 1, 1,
  272. mv->table_mv_code, 2, 2, 3714);
  273. mv = &ff_mv_tables[1];
  274. INIT_VLC_STATIC(&mv->vlc, MV_VLC_BITS, mv->n + 1,
  275. mv->table_mv_bits, 1, 1,
  276. mv->table_mv_code, 2, 2, 2694);
  277. INIT_VLC_STATIC(&ff_msmp4_dc_luma_vlc[0], DC_VLC_BITS, 120,
  278. &ff_table0_dc_lum[0][1], 8, 4,
  279. &ff_table0_dc_lum[0][0], 8, 4, 1158);
  280. INIT_VLC_STATIC(&ff_msmp4_dc_chroma_vlc[0], DC_VLC_BITS, 120,
  281. &ff_table0_dc_chroma[0][1], 8, 4,
  282. &ff_table0_dc_chroma[0][0], 8, 4, 1118);
  283. INIT_VLC_STATIC(&ff_msmp4_dc_luma_vlc[1], DC_VLC_BITS, 120,
  284. &ff_table1_dc_lum[0][1], 8, 4,
  285. &ff_table1_dc_lum[0][0], 8, 4, 1476);
  286. INIT_VLC_STATIC(&ff_msmp4_dc_chroma_vlc[1], DC_VLC_BITS, 120,
  287. &ff_table1_dc_chroma[0][1], 8, 4,
  288. &ff_table1_dc_chroma[0][0], 8, 4, 1216);
  289. INIT_VLC_STATIC(&v2_dc_lum_vlc, DC_VLC_BITS, 512,
  290. &ff_v2_dc_lum_table[0][1], 8, 4,
  291. &ff_v2_dc_lum_table[0][0], 8, 4, 1472);
  292. INIT_VLC_STATIC(&v2_dc_chroma_vlc, DC_VLC_BITS, 512,
  293. &ff_v2_dc_chroma_table[0][1], 8, 4,
  294. &ff_v2_dc_chroma_table[0][0], 8, 4, 1506);
  295. INIT_VLC_STATIC(&v2_intra_cbpc_vlc, V2_INTRA_CBPC_VLC_BITS, 4,
  296. &ff_v2_intra_cbpc[0][1], 2, 1,
  297. &ff_v2_intra_cbpc[0][0], 2, 1, 8);
  298. INIT_VLC_STATIC(&v2_mb_type_vlc, V2_MB_TYPE_VLC_BITS, 8,
  299. &ff_v2_mb_type[0][1], 2, 1,
  300. &ff_v2_mb_type[0][0], 2, 1, 128);
  301. INIT_VLC_STATIC(&v2_mv_vlc, V2_MV_VLC_BITS, 33,
  302. &ff_mvtab[0][1], 2, 1,
  303. &ff_mvtab[0][0], 2, 1, 538);
  304. INIT_VLC_STATIC(&ff_mb_non_intra_vlc[0], MB_NON_INTRA_VLC_BITS, 128,
  305. &ff_wmv2_inter_table[0][0][1], 8, 4,
  306. &ff_wmv2_inter_table[0][0][0], 8, 4, 1636);
  307. INIT_VLC_STATIC(&ff_mb_non_intra_vlc[1], MB_NON_INTRA_VLC_BITS, 128,
  308. &ff_wmv2_inter_table[1][0][1], 8, 4,
  309. &ff_wmv2_inter_table[1][0][0], 8, 4, 2648);
  310. INIT_VLC_STATIC(&ff_mb_non_intra_vlc[2], MB_NON_INTRA_VLC_BITS, 128,
  311. &ff_wmv2_inter_table[2][0][1], 8, 4,
  312. &ff_wmv2_inter_table[2][0][0], 8, 4, 1532);
  313. INIT_VLC_STATIC(&ff_mb_non_intra_vlc[3], MB_NON_INTRA_VLC_BITS, 128,
  314. &ff_wmv2_inter_table[3][0][1], 8, 4,
  315. &ff_wmv2_inter_table[3][0][0], 8, 4, 2488);
  316. INIT_VLC_STATIC(&ff_msmp4_mb_i_vlc, MB_INTRA_VLC_BITS, 64,
  317. &ff_msmp4_mb_i_table[0][1], 4, 2,
  318. &ff_msmp4_mb_i_table[0][0], 4, 2, 536);
  319. INIT_VLC_STATIC(&ff_inter_intra_vlc, INTER_INTRA_VLC_BITS, 4,
  320. &ff_table_inter_intra[0][1], 2, 1,
  321. &ff_table_inter_intra[0][0], 2, 1, 8);
  322. }
  323. switch(s->msmpeg4_version){
  324. case 1:
  325. case 2:
  326. s->decode_mb= msmpeg4v12_decode_mb;
  327. break;
  328. case 3:
  329. case 4:
  330. s->decode_mb= msmpeg4v34_decode_mb;
  331. break;
  332. case 5:
  333. if (CONFIG_WMV2_DECODER)
  334. s->decode_mb= ff_wmv2_decode_mb;
  335. case 6:
  336. //FIXME + TODO VC1 decode mb
  337. break;
  338. }
  339. s->slice_height= s->mb_height; //to avoid 1/0 if the first frame is not a keyframe
  340. return 0;
  341. }
  342. int ff_msmpeg4_decode_picture_header(MpegEncContext * s)
  343. {
  344. int code;
  345. if(s->msmpeg4_version==1){
  346. int start_code = get_bits_long(&s->gb, 32);
  347. if(start_code!=0x00000100){
  348. av_log(s->avctx, AV_LOG_ERROR, "invalid startcode\n");
  349. return -1;
  350. }
  351. skip_bits(&s->gb, 5); // frame number */
  352. }
  353. s->pict_type = get_bits(&s->gb, 2) + 1;
  354. if (s->pict_type != AV_PICTURE_TYPE_I &&
  355. s->pict_type != AV_PICTURE_TYPE_P){
  356. av_log(s->avctx, AV_LOG_ERROR, "invalid picture type\n");
  357. return -1;
  358. }
  359. #if 0
  360. {
  361. static int had_i=0;
  362. if(s->pict_type == AV_PICTURE_TYPE_I) had_i=1;
  363. if(!had_i) return -1;
  364. }
  365. #endif
  366. s->chroma_qscale= s->qscale = get_bits(&s->gb, 5);
  367. if(s->qscale==0){
  368. av_log(s->avctx, AV_LOG_ERROR, "invalid qscale\n");
  369. return -1;
  370. }
  371. if (s->pict_type == AV_PICTURE_TYPE_I) {
  372. code = get_bits(&s->gb, 5);
  373. if(s->msmpeg4_version==1){
  374. if(code==0 || code>s->mb_height){
  375. av_log(s->avctx, AV_LOG_ERROR, "invalid slice height %d\n", code);
  376. return -1;
  377. }
  378. s->slice_height = code;
  379. }else{
  380. /* 0x17: one slice, 0x18: two slices, ... */
  381. if (code < 0x17){
  382. av_log(s->avctx, AV_LOG_ERROR, "error, slice code was %X\n", code);
  383. return -1;
  384. }
  385. s->slice_height = s->mb_height / (code - 0x16);
  386. }
  387. switch(s->msmpeg4_version){
  388. case 1:
  389. case 2:
  390. s->rl_chroma_table_index = 2;
  391. s->rl_table_index = 2;
  392. s->dc_table_index = 0; //not used
  393. break;
  394. case 3:
  395. s->rl_chroma_table_index = decode012(&s->gb);
  396. s->rl_table_index = decode012(&s->gb);
  397. s->dc_table_index = get_bits1(&s->gb);
  398. break;
  399. case 4:
  400. ff_msmpeg4_decode_ext_header(s, (2+5+5+17+7)/8);
  401. if(s->bit_rate > MBAC_BITRATE) s->per_mb_rl_table= get_bits1(&s->gb);
  402. else s->per_mb_rl_table= 0;
  403. if(!s->per_mb_rl_table){
  404. s->rl_chroma_table_index = decode012(&s->gb);
  405. s->rl_table_index = decode012(&s->gb);
  406. }
  407. s->dc_table_index = get_bits1(&s->gb);
  408. s->inter_intra_pred= 0;
  409. break;
  410. }
  411. s->no_rounding = 1;
  412. if(s->avctx->debug&FF_DEBUG_PICT_INFO)
  413. av_log(s->avctx, AV_LOG_DEBUG, "qscale:%d rlc:%d rl:%d dc:%d mbrl:%d slice:%d \n",
  414. s->qscale,
  415. s->rl_chroma_table_index,
  416. s->rl_table_index,
  417. s->dc_table_index,
  418. s->per_mb_rl_table,
  419. s->slice_height);
  420. } else {
  421. switch(s->msmpeg4_version){
  422. case 1:
  423. case 2:
  424. if(s->msmpeg4_version==1)
  425. s->use_skip_mb_code = 1;
  426. else
  427. s->use_skip_mb_code = get_bits1(&s->gb);
  428. s->rl_table_index = 2;
  429. s->rl_chroma_table_index = s->rl_table_index;
  430. s->dc_table_index = 0; //not used
  431. s->mv_table_index = 0;
  432. break;
  433. case 3:
  434. s->use_skip_mb_code = get_bits1(&s->gb);
  435. s->rl_table_index = decode012(&s->gb);
  436. s->rl_chroma_table_index = s->rl_table_index;
  437. s->dc_table_index = get_bits1(&s->gb);
  438. s->mv_table_index = get_bits1(&s->gb);
  439. break;
  440. case 4:
  441. s->use_skip_mb_code = get_bits1(&s->gb);
  442. if(s->bit_rate > MBAC_BITRATE) s->per_mb_rl_table= get_bits1(&s->gb);
  443. else s->per_mb_rl_table= 0;
  444. if(!s->per_mb_rl_table){
  445. s->rl_table_index = decode012(&s->gb);
  446. s->rl_chroma_table_index = s->rl_table_index;
  447. }
  448. s->dc_table_index = get_bits1(&s->gb);
  449. s->mv_table_index = get_bits1(&s->gb);
  450. s->inter_intra_pred= (s->width*s->height < 320*240 && s->bit_rate<=II_BITRATE);
  451. break;
  452. }
  453. if(s->avctx->debug&FF_DEBUG_PICT_INFO)
  454. av_log(s->avctx, AV_LOG_DEBUG, "skip:%d rl:%d rlc:%d dc:%d mv:%d mbrl:%d qp:%d \n",
  455. s->use_skip_mb_code,
  456. s->rl_table_index,
  457. s->rl_chroma_table_index,
  458. s->dc_table_index,
  459. s->mv_table_index,
  460. s->per_mb_rl_table,
  461. s->qscale);
  462. if(s->flipflop_rounding){
  463. s->no_rounding ^= 1;
  464. }else{
  465. s->no_rounding = 0;
  466. }
  467. }
  468. ff_dlog(s->avctx, "%d %d %d %d %d\n", s->pict_type, s->bit_rate,
  469. s->inter_intra_pred, s->width, s->height);
  470. s->esc3_level_length= 0;
  471. s->esc3_run_length= 0;
  472. return 0;
  473. }
  474. int ff_msmpeg4_decode_ext_header(MpegEncContext * s, int buf_size)
  475. {
  476. int left= buf_size*8 - get_bits_count(&s->gb);
  477. int length= s->msmpeg4_version>=3 ? 17 : 16;
  478. /* the alt_bitstream reader could read over the end so we need to check it */
  479. if(left>=length && left<length+8)
  480. {
  481. skip_bits(&s->gb, 5); /* fps */
  482. s->bit_rate= get_bits(&s->gb, 11)*1024;
  483. if(s->msmpeg4_version>=3)
  484. s->flipflop_rounding= get_bits1(&s->gb);
  485. else
  486. s->flipflop_rounding= 0;
  487. }
  488. else if(left<length+8)
  489. {
  490. s->flipflop_rounding= 0;
  491. if(s->msmpeg4_version != 2)
  492. av_log(s->avctx, AV_LOG_ERROR, "ext header missing, %d left\n", left);
  493. }
  494. else
  495. {
  496. av_log(s->avctx, AV_LOG_ERROR, "I-frame too long, ignoring ext header\n");
  497. }
  498. return 0;
  499. }
  500. static int msmpeg4_decode_dc(MpegEncContext * s, int n, int *dir_ptr)
  501. {
  502. int level, pred;
  503. if(s->msmpeg4_version<=2){
  504. if (n < 4) {
  505. level = get_vlc2(&s->gb, v2_dc_lum_vlc.table, DC_VLC_BITS, 3);
  506. } else {
  507. level = get_vlc2(&s->gb, v2_dc_chroma_vlc.table, DC_VLC_BITS, 3);
  508. }
  509. if (level < 0)
  510. return -1;
  511. level-=256;
  512. }else{ //FIXME optimize use unified tables & index
  513. if (n < 4) {
  514. level = get_vlc2(&s->gb, ff_msmp4_dc_luma_vlc[s->dc_table_index].table, DC_VLC_BITS, 3);
  515. } else {
  516. level = get_vlc2(&s->gb, ff_msmp4_dc_chroma_vlc[s->dc_table_index].table, DC_VLC_BITS, 3);
  517. }
  518. if (level < 0){
  519. av_log(s->avctx, AV_LOG_ERROR, "illegal dc vlc\n");
  520. return -1;
  521. }
  522. if (level == DC_MAX) {
  523. level = get_bits(&s->gb, 8);
  524. if (get_bits1(&s->gb))
  525. level = -level;
  526. } else if (level != 0) {
  527. if (get_bits1(&s->gb))
  528. level = -level;
  529. }
  530. }
  531. if(s->msmpeg4_version==1){
  532. int32_t *dc_val;
  533. pred = msmpeg4v1_pred_dc(s, n, &dc_val);
  534. level += pred;
  535. /* update predictor */
  536. *dc_val= level;
  537. }else{
  538. int16_t *dc_val;
  539. pred = ff_msmpeg4_pred_dc(s, n, &dc_val, dir_ptr);
  540. level += pred;
  541. /* update predictor */
  542. if (n < 4) {
  543. *dc_val = level * s->y_dc_scale;
  544. } else {
  545. *dc_val = level * s->c_dc_scale;
  546. }
  547. }
  548. return level;
  549. }
  550. int ff_msmpeg4_decode_block(MpegEncContext * s, int16_t * block,
  551. int n, int coded, const uint8_t *scan_table)
  552. {
  553. int level, i, last, run, run_diff;
  554. int av_uninit(dc_pred_dir);
  555. RLTable *rl;
  556. RL_VLC_ELEM *rl_vlc;
  557. int qmul, qadd;
  558. if (s->mb_intra) {
  559. qmul=1;
  560. qadd=0;
  561. /* DC coef */
  562. level = msmpeg4_decode_dc(s, n, &dc_pred_dir);
  563. if (level < 0){
  564. av_log(s->avctx, AV_LOG_ERROR, "dc overflow- block: %d qscale: %d//\n", n, s->qscale);
  565. if(s->inter_intra_pred) level=0;
  566. else return -1;
  567. }
  568. if (n < 4) {
  569. rl = &ff_rl_table[s->rl_table_index];
  570. if(level > 256*s->y_dc_scale){
  571. av_log(s->avctx, AV_LOG_ERROR, "dc overflow+ L qscale: %d//\n", s->qscale);
  572. if(!s->inter_intra_pred) return -1;
  573. }
  574. } else {
  575. rl = &ff_rl_table[3 + s->rl_chroma_table_index];
  576. if(level > 256*s->c_dc_scale){
  577. av_log(s->avctx, AV_LOG_ERROR, "dc overflow+ C qscale: %d//\n", s->qscale);
  578. if(!s->inter_intra_pred) return -1;
  579. }
  580. }
  581. block[0] = level;
  582. run_diff = s->msmpeg4_version >= 4;
  583. i = 0;
  584. if (!coded) {
  585. goto not_coded;
  586. }
  587. if (s->ac_pred) {
  588. if (dc_pred_dir == 0)
  589. scan_table = s->intra_v_scantable.permutated; /* left */
  590. else
  591. scan_table = s->intra_h_scantable.permutated; /* top */
  592. } else {
  593. scan_table = s->intra_scantable.permutated;
  594. }
  595. rl_vlc= rl->rl_vlc[0];
  596. } else {
  597. qmul = s->qscale << 1;
  598. qadd = (s->qscale - 1) | 1;
  599. i = -1;
  600. rl = &ff_rl_table[3 + s->rl_table_index];
  601. if(s->msmpeg4_version==2)
  602. run_diff = 0;
  603. else
  604. run_diff = 1;
  605. if (!coded) {
  606. s->block_last_index[n] = i;
  607. return 0;
  608. }
  609. if(!scan_table)
  610. scan_table = s->inter_scantable.permutated;
  611. rl_vlc= rl->rl_vlc[s->qscale];
  612. }
  613. {
  614. OPEN_READER(re, &s->gb);
  615. for(;;) {
  616. UPDATE_CACHE(re, &s->gb);
  617. GET_RL_VLC(level, run, re, &s->gb, rl_vlc, TEX_VLC_BITS, 2, 0);
  618. if (level==0) {
  619. int cache;
  620. cache= GET_CACHE(re, &s->gb);
  621. /* escape */
  622. if (s->msmpeg4_version==1 || (cache&0x80000000)==0) {
  623. if (s->msmpeg4_version==1 || (cache&0x40000000)==0) {
  624. /* third escape */
  625. if(s->msmpeg4_version!=1) LAST_SKIP_BITS(re, &s->gb, 2);
  626. UPDATE_CACHE(re, &s->gb);
  627. if(s->msmpeg4_version<=3){
  628. last= SHOW_UBITS(re, &s->gb, 1); SKIP_CACHE(re, &s->gb, 1);
  629. run= SHOW_UBITS(re, &s->gb, 6); SKIP_CACHE(re, &s->gb, 6);
  630. level= SHOW_SBITS(re, &s->gb, 8);
  631. SKIP_COUNTER(re, &s->gb, 1+6+8);
  632. }else{
  633. int sign;
  634. last= SHOW_UBITS(re, &s->gb, 1); SKIP_BITS(re, &s->gb, 1);
  635. if(!s->esc3_level_length){
  636. int ll;
  637. ff_dlog(s->avctx, "ESC-3 %X at %d %d\n",
  638. show_bits(&s->gb, 24), s->mb_x, s->mb_y);
  639. if(s->qscale<8){
  640. ll= SHOW_UBITS(re, &s->gb, 3); SKIP_BITS(re, &s->gb, 3);
  641. if(ll==0){
  642. ll= 8+SHOW_UBITS(re, &s->gb, 1); SKIP_BITS(re, &s->gb, 1);
  643. }
  644. }else{
  645. ll=2;
  646. while(ll<8 && SHOW_UBITS(re, &s->gb, 1)==0){
  647. ll++;
  648. SKIP_BITS(re, &s->gb, 1);
  649. }
  650. if(ll<8) SKIP_BITS(re, &s->gb, 1);
  651. }
  652. s->esc3_level_length= ll;
  653. s->esc3_run_length= SHOW_UBITS(re, &s->gb, 2) + 3; SKIP_BITS(re, &s->gb, 2);
  654. UPDATE_CACHE(re, &s->gb);
  655. }
  656. run= SHOW_UBITS(re, &s->gb, s->esc3_run_length);
  657. SKIP_BITS(re, &s->gb, s->esc3_run_length);
  658. sign= SHOW_UBITS(re, &s->gb, 1);
  659. SKIP_BITS(re, &s->gb, 1);
  660. level= SHOW_UBITS(re, &s->gb, s->esc3_level_length);
  661. SKIP_BITS(re, &s->gb, s->esc3_level_length);
  662. if(sign) level= -level;
  663. }
  664. //level = level * qmul + (level>0) * qadd - (level<=0) * qadd ;
  665. if (level>0) level= level * qmul + qadd;
  666. else level= level * qmul - qadd;
  667. i+= run + 1;
  668. if(last) i+=192;
  669. } else {
  670. /* second escape */
  671. SKIP_BITS(re, &s->gb, 2);
  672. GET_RL_VLC(level, run, re, &s->gb, rl_vlc, TEX_VLC_BITS, 2, 1);
  673. i+= run + rl->max_run[run>>7][level/qmul] + run_diff; //FIXME opt indexing
  674. level = (level ^ SHOW_SBITS(re, &s->gb, 1)) - SHOW_SBITS(re, &s->gb, 1);
  675. LAST_SKIP_BITS(re, &s->gb, 1);
  676. }
  677. } else {
  678. /* first escape */
  679. SKIP_BITS(re, &s->gb, 1);
  680. GET_RL_VLC(level, run, re, &s->gb, rl_vlc, TEX_VLC_BITS, 2, 1);
  681. i+= run;
  682. level = level + rl->max_level[run>>7][(run-1)&63] * qmul;//FIXME opt indexing
  683. level = (level ^ SHOW_SBITS(re, &s->gb, 1)) - SHOW_SBITS(re, &s->gb, 1);
  684. LAST_SKIP_BITS(re, &s->gb, 1);
  685. }
  686. } else {
  687. i+= run;
  688. level = (level ^ SHOW_SBITS(re, &s->gb, 1)) - SHOW_SBITS(re, &s->gb, 1);
  689. LAST_SKIP_BITS(re, &s->gb, 1);
  690. }
  691. if (i > 62){
  692. i-= 192;
  693. if(i&(~63)){
  694. const int left= get_bits_left(&s->gb);
  695. if (((i + 192 == 64 && level / qmul == -1) ||
  696. !(s->avctx->err_recognition & AV_EF_BITSTREAM)) &&
  697. left >= 0) {
  698. av_log(s->avctx, AV_LOG_ERROR, "ignoring overflow at %d %d\n", s->mb_x, s->mb_y);
  699. break;
  700. }else{
  701. av_log(s->avctx, AV_LOG_ERROR, "ac-tex damaged at %d %d\n", s->mb_x, s->mb_y);
  702. return -1;
  703. }
  704. }
  705. block[scan_table[i]] = level;
  706. break;
  707. }
  708. block[scan_table[i]] = level;
  709. }
  710. CLOSE_READER(re, &s->gb);
  711. }
  712. not_coded:
  713. if (s->mb_intra) {
  714. ff_mpeg4_pred_ac(s, block, n, dc_pred_dir);
  715. if (s->ac_pred) {
  716. i = 63; /* XXX: not optimal */
  717. }
  718. }
  719. if(s->msmpeg4_version>=4 && i>0) i=63; //FIXME/XXX optimize
  720. s->block_last_index[n] = i;
  721. return 0;
  722. }
  723. int ff_msmpeg4_decode_motion(MpegEncContext * s,
  724. int *mx_ptr, int *my_ptr)
  725. {
  726. MVTable *mv;
  727. int code, mx, my;
  728. mv = &ff_mv_tables[s->mv_table_index];
  729. code = get_vlc2(&s->gb, mv->vlc.table, MV_VLC_BITS, 2);
  730. if (code < 0){
  731. av_log(s->avctx, AV_LOG_ERROR, "illegal MV code at %d %d\n", s->mb_x, s->mb_y);
  732. return -1;
  733. }
  734. if (code == mv->n) {
  735. mx = get_bits(&s->gb, 6);
  736. my = get_bits(&s->gb, 6);
  737. } else {
  738. mx = mv->table_mvx[code];
  739. my = mv->table_mvy[code];
  740. }
  741. mx += *mx_ptr - 32;
  742. my += *my_ptr - 32;
  743. /* WARNING : they do not do exactly modulo encoding */
  744. if (mx <= -64)
  745. mx += 64;
  746. else if (mx >= 64)
  747. mx -= 64;
  748. if (my <= -64)
  749. my += 64;
  750. else if (my >= 64)
  751. my -= 64;
  752. *mx_ptr = mx;
  753. *my_ptr = my;
  754. return 0;
  755. }
  756. AVCodec ff_msmpeg4v1_decoder = {
  757. .name = "msmpeg4v1",
  758. .long_name = NULL_IF_CONFIG_SMALL("MPEG-4 part 2 Microsoft variant version 1"),
  759. .type = AVMEDIA_TYPE_VIDEO,
  760. .id = AV_CODEC_ID_MSMPEG4V1,
  761. .priv_data_size = sizeof(MpegEncContext),
  762. .init = ff_msmpeg4_decode_init,
  763. .close = ff_h263_decode_end,
  764. .decode = ff_h263_decode_frame,
  765. .capabilities = AV_CODEC_CAP_DRAW_HORIZ_BAND | AV_CODEC_CAP_DR1,
  766. .pix_fmts = (const enum AVPixelFormat[]) {
  767. AV_PIX_FMT_YUV420P,
  768. AV_PIX_FMT_NONE
  769. },
  770. };
  771. AVCodec ff_msmpeg4v2_decoder = {
  772. .name = "msmpeg4v2",
  773. .long_name = NULL_IF_CONFIG_SMALL("MPEG-4 part 2 Microsoft variant version 2"),
  774. .type = AVMEDIA_TYPE_VIDEO,
  775. .id = AV_CODEC_ID_MSMPEG4V2,
  776. .priv_data_size = sizeof(MpegEncContext),
  777. .init = ff_msmpeg4_decode_init,
  778. .close = ff_h263_decode_end,
  779. .decode = ff_h263_decode_frame,
  780. .capabilities = AV_CODEC_CAP_DRAW_HORIZ_BAND | AV_CODEC_CAP_DR1,
  781. .pix_fmts = (const enum AVPixelFormat[]) {
  782. AV_PIX_FMT_YUV420P,
  783. AV_PIX_FMT_NONE
  784. },
  785. };
  786. AVCodec ff_msmpeg4v3_decoder = {
  787. .name = "msmpeg4",
  788. .long_name = NULL_IF_CONFIG_SMALL("MPEG-4 part 2 Microsoft variant version 3"),
  789. .type = AVMEDIA_TYPE_VIDEO,
  790. .id = AV_CODEC_ID_MSMPEG4V3,
  791. .priv_data_size = sizeof(MpegEncContext),
  792. .init = ff_msmpeg4_decode_init,
  793. .close = ff_h263_decode_end,
  794. .decode = ff_h263_decode_frame,
  795. .capabilities = AV_CODEC_CAP_DRAW_HORIZ_BAND | AV_CODEC_CAP_DR1,
  796. .pix_fmts = (const enum AVPixelFormat[]) {
  797. AV_PIX_FMT_YUV420P,
  798. AV_PIX_FMT_NONE
  799. },
  800. };
  801. AVCodec ff_wmv1_decoder = {
  802. .name = "wmv1",
  803. .long_name = NULL_IF_CONFIG_SMALL("Windows Media Video 7"),
  804. .type = AVMEDIA_TYPE_VIDEO,
  805. .id = AV_CODEC_ID_WMV1,
  806. .priv_data_size = sizeof(MpegEncContext),
  807. .init = ff_msmpeg4_decode_init,
  808. .close = ff_h263_decode_end,
  809. .decode = ff_h263_decode_frame,
  810. .capabilities = AV_CODEC_CAP_DRAW_HORIZ_BAND | AV_CODEC_CAP_DR1,
  811. .pix_fmts = (const enum AVPixelFormat[]) {
  812. AV_PIX_FMT_YUV420P,
  813. AV_PIX_FMT_NONE
  814. },
  815. };