FFmpeg
 All Data Structures Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
mpeg12.c
Go to the documentation of this file.
1 /*
2  * MPEG-1/2 decoder
3  * Copyright (c) 2000, 2001 Fabrice Bellard
4  * Copyright (c) 2002-2004 Michael Niedermayer <michaelni@gmx.at>
5  *
6  * This file is part of FFmpeg.
7  *
8  * FFmpeg is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public
10  * License as published by the Free Software Foundation; either
11  * version 2.1 of the License, or (at your option) any later version.
12  *
13  * FFmpeg is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16  * Lesser General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with FFmpeg; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21  */
22 
23 /**
24  * @file
25  * MPEG-1/2 decoder
26  */
27 
28 #define UNCHECKED_BITSTREAM_READER 1
29 
30 //#define DEBUG
31 #include "internal.h"
32 #include "avcodec.h"
33 #include "dsputil.h"
34 #include "mpegvideo.h"
35 #include "libavutil/avassert.h"
36 #include "libavutil/timecode.h"
37 
38 #include "mpeg12.h"
39 #include "mpeg12data.h"
40 #include "mpeg12decdata.h"
41 #include "bytestream.h"
42 #include "vdpau_internal.h"
43 #include "xvmc_internal.h"
44 #include "thread.h"
45 
46 
47 #define MV_VLC_BITS 9
48 #define MBINCR_VLC_BITS 9
49 #define MB_PAT_VLC_BITS 9
50 #define MB_PTYPE_VLC_BITS 6
51 #define MB_BTYPE_VLC_BITS 6
52 
53 static VLC mv_vlc;
54 
55 /* as H.263, but only 17 codes */
56 static int mpeg_decode_motion(MpegEncContext *s, int fcode, int pred)
57 {
58  int code, sign, val, shift;
59 
60  code = get_vlc2(&s->gb, mv_vlc.table, MV_VLC_BITS, 2);
61  if (code == 0) {
62  return pred;
63  }
64  if (code < 0) {
65  return 0xffff;
66  }
67 
68  sign = get_bits1(&s->gb);
69  shift = fcode - 1;
70  val = code;
71  if (shift) {
72  val = (val - 1) << shift;
73  val |= get_bits(&s->gb, shift);
74  val++;
75  }
76  if (sign)
77  val = -val;
78  val += pred;
79 
80  /* modulo decoding */
81  return sign_extend(val, 5 + shift);
82 }
83 
84 static inline int mpeg1_decode_block_intra(MpegEncContext *s, int16_t *block, int n)
85 {
86  int level, dc, diff, i, j, run;
87  int component;
88  RLTable *rl = &ff_rl_mpeg1;
89  uint8_t * const scantable = s->intra_scantable.permutated;
90  const uint16_t *quant_matrix = s->intra_matrix;
91  const int qscale = s->qscale;
92 
93  /* DC coefficient */
94  component = (n <= 3 ? 0 : n - 4 + 1);
95  diff = decode_dc(&s->gb, component);
96  if (diff >= 0xffff)
97  return -1;
98  dc = s->last_dc[component];
99  dc += diff;
100  s->last_dc[component] = dc;
101  block[0] = dc * quant_matrix[0];
102  av_dlog(s->avctx, "dc=%d diff=%d\n", dc, diff);
103  i = 0;
104  {
105  OPEN_READER(re, &s->gb);
106  /* now quantify & encode AC coefficients */
107  for (;;) {
108  UPDATE_CACHE(re, &s->gb);
109  GET_RL_VLC(level, run, re, &s->gb, rl->rl_vlc[0], TEX_VLC_BITS, 2, 0);
110 
111  if (level == 127) {
112  break;
113  } else if (level != 0) {
114  i += run;
115  j = scantable[i];
116  level = (level * qscale * quant_matrix[j]) >> 4;
117  level = (level - 1) | 1;
118  level = (level ^ SHOW_SBITS(re, &s->gb, 1)) - SHOW_SBITS(re, &s->gb, 1);
119  LAST_SKIP_BITS(re, &s->gb, 1);
120  } else {
121  /* escape */
122  run = SHOW_UBITS(re, &s->gb, 6) + 1; LAST_SKIP_BITS(re, &s->gb, 6);
123  UPDATE_CACHE(re, &s->gb);
124  level = SHOW_SBITS(re, &s->gb, 8); SKIP_BITS(re, &s->gb, 8);
125  if (level == -128) {
126  level = SHOW_UBITS(re, &s->gb, 8) - 256; LAST_SKIP_BITS(re, &s->gb, 8);
127  } else if (level == 0) {
128  level = SHOW_UBITS(re, &s->gb, 8) ; LAST_SKIP_BITS(re, &s->gb, 8);
129  }
130  i += run;
131  j = scantable[i];
132  if (level < 0) {
133  level = -level;
134  level = (level * qscale * quant_matrix[j]) >> 4;
135  level = (level - 1) | 1;
136  level = -level;
137  } else {
138  level = (level * qscale * quant_matrix[j]) >> 4;
139  level = (level - 1) | 1;
140  }
141  }
142  if (i > 63) {
143  av_log(s->avctx, AV_LOG_ERROR, "ac-tex damaged at %d %d\n", s->mb_x, s->mb_y);
144  return -1;
145  }
146 
147  block[j] = level;
148  }
149  CLOSE_READER(re, &s->gb);
150  }
151  s->block_last_index[n] = i;
152  return 0;
153 }
154 
156 {
157  return mpeg1_decode_block_intra(s, block, n);
158 }
159 
160 static inline int mpeg1_decode_block_inter(MpegEncContext *s, int16_t *block, int n)
161 {
162  int level, i, j, run;
163  RLTable *rl = &ff_rl_mpeg1;
164  uint8_t * const scantable = s->intra_scantable.permutated;
165  const uint16_t *quant_matrix = s->inter_matrix;
166  const int qscale = s->qscale;
167 
168  {
169  OPEN_READER(re, &s->gb);
170  i = -1;
171  // special case for first coefficient, no need to add second VLC table
172  UPDATE_CACHE(re, &s->gb);
173  if (((int32_t)GET_CACHE(re, &s->gb)) < 0) {
174  level = (3 * qscale * quant_matrix[0]) >> 5;
175  level = (level - 1) | 1;
176  if (GET_CACHE(re, &s->gb) & 0x40000000)
177  level = -level;
178  block[0] = level;
179  i++;
180  SKIP_BITS(re, &s->gb, 2);
181  if (((int32_t)GET_CACHE(re, &s->gb)) <= (int32_t)0xBFFFFFFF)
182  goto end;
183  }
184  /* now quantify & encode AC coefficients */
185  for (;;) {
186  GET_RL_VLC(level, run, re, &s->gb, rl->rl_vlc[0], TEX_VLC_BITS, 2, 0);
187 
188  if (level != 0) {
189  i += run;
190  j = scantable[i];
191  level = ((level * 2 + 1) * qscale * quant_matrix[j]) >> 5;
192  level = (level - 1) | 1;
193  level = (level ^ SHOW_SBITS(re, &s->gb, 1)) - SHOW_SBITS(re, &s->gb, 1);
194  SKIP_BITS(re, &s->gb, 1);
195  } else {
196  /* escape */
197  run = SHOW_UBITS(re, &s->gb, 6) + 1; LAST_SKIP_BITS(re, &s->gb, 6);
198  UPDATE_CACHE(re, &s->gb);
199  level = SHOW_SBITS(re, &s->gb, 8); SKIP_BITS(re, &s->gb, 8);
200  if (level == -128) {
201  level = SHOW_UBITS(re, &s->gb, 8) - 256; SKIP_BITS(re, &s->gb, 8);
202  } else if (level == 0) {
203  level = SHOW_UBITS(re, &s->gb, 8) ; SKIP_BITS(re, &s->gb, 8);
204  }
205  i += run;
206  j = scantable[i];
207  if (level < 0) {
208  level = -level;
209  level = ((level * 2 + 1) * qscale * quant_matrix[j]) >> 5;
210  level = (level - 1) | 1;
211  level = -level;
212  } else {
213  level = ((level * 2 + 1) * qscale * quant_matrix[j]) >> 5;
214  level = (level - 1) | 1;
215  }
216  }
217  if (i > 63) {
218  av_log(s->avctx, AV_LOG_ERROR, "ac-tex damaged at %d %d\n", s->mb_x, s->mb_y);
219  return -1;
220  }
221 
222  block[j] = level;
223  if (((int32_t)GET_CACHE(re, &s->gb)) <= (int32_t)0xBFFFFFFF)
224  break;
225  UPDATE_CACHE(re, &s->gb);
226  }
227 end:
228  LAST_SKIP_BITS(re, &s->gb, 2);
229  CLOSE_READER(re, &s->gb);
230  }
231  s->block_last_index[n] = i;
232  return 0;
233 }
234 
235 /**
236  * Note: this function can read out of range and crash for corrupt streams.
237  * Changing this would eat up any speed benefits it has.
238  * Do not use "fast" flag if you need the code to be robust.
239  */
240 static inline int mpeg1_fast_decode_block_inter(MpegEncContext *s, int16_t *block, int n)
241 {
242  int level, i, j, run;
243  RLTable *rl = &ff_rl_mpeg1;
244  uint8_t * const scantable = s->intra_scantable.permutated;
245  const int qscale = s->qscale;
246 
247  {
248  OPEN_READER(re, &s->gb);
249  i = -1;
250  // special case for first coefficient, no need to add second VLC table
251  UPDATE_CACHE(re, &s->gb);
252  if (((int32_t)GET_CACHE(re, &s->gb)) < 0) {
253  level = (3 * qscale) >> 1;
254  level = (level - 1) | 1;
255  if (GET_CACHE(re, &s->gb) & 0x40000000)
256  level = -level;
257  block[0] = level;
258  i++;
259  SKIP_BITS(re, &s->gb, 2);
260  if (((int32_t)GET_CACHE(re, &s->gb)) <= (int32_t)0xBFFFFFFF)
261  goto end;
262  }
263 
264  /* now quantify & encode AC coefficients */
265  for (;;) {
266  GET_RL_VLC(level, run, re, &s->gb, rl->rl_vlc[0], TEX_VLC_BITS, 2, 0);
267 
268  if (level != 0) {
269  i += run;
270  j = scantable[i];
271  level = ((level * 2 + 1) * qscale) >> 1;
272  level = (level - 1) | 1;
273  level = (level ^ SHOW_SBITS(re, &s->gb, 1)) - SHOW_SBITS(re, &s->gb, 1);
274  SKIP_BITS(re, &s->gb, 1);
275  } else {
276  /* escape */
277  run = SHOW_UBITS(re, &s->gb, 6)+1; LAST_SKIP_BITS(re, &s->gb, 6);
278  UPDATE_CACHE(re, &s->gb);
279  level = SHOW_SBITS(re, &s->gb, 8); SKIP_BITS(re, &s->gb, 8);
280  if (level == -128) {
281  level = SHOW_UBITS(re, &s->gb, 8) - 256; SKIP_BITS(re, &s->gb, 8);
282  } else if (level == 0) {
283  level = SHOW_UBITS(re, &s->gb, 8) ; SKIP_BITS(re, &s->gb, 8);
284  }
285  i += run;
286  j = scantable[i];
287  if (level < 0) {
288  level = -level;
289  level = ((level * 2 + 1) * qscale) >> 1;
290  level = (level - 1) | 1;
291  level = -level;
292  } else {
293  level = ((level * 2 + 1) * qscale) >> 1;
294  level = (level - 1) | 1;
295  }
296  }
297 
298  block[j] = level;
299  if (((int32_t)GET_CACHE(re, &s->gb)) <= (int32_t)0xBFFFFFFF)
300  break;
301  UPDATE_CACHE(re, &s->gb);
302  }
303 end:
304  LAST_SKIP_BITS(re, &s->gb, 2);
305  CLOSE_READER(re, &s->gb);
306  }
307  s->block_last_index[n] = i;
308  return 0;
309 }
310 
311 
312 static inline int mpeg2_decode_block_non_intra(MpegEncContext *s, int16_t *block, int n)
313 {
314  int level, i, j, run;
315  RLTable *rl = &ff_rl_mpeg1;
316  uint8_t * const scantable = s->intra_scantable.permutated;
317  const uint16_t *quant_matrix;
318  const int qscale = s->qscale;
319  int mismatch;
320 
321  mismatch = 1;
322 
323  {
324  OPEN_READER(re, &s->gb);
325  i = -1;
326  if (n < 4)
327  quant_matrix = s->inter_matrix;
328  else
329  quant_matrix = s->chroma_inter_matrix;
330 
331  // special case for first coefficient, no need to add second VLC table
332  UPDATE_CACHE(re, &s->gb);
333  if (((int32_t)GET_CACHE(re, &s->gb)) < 0) {
334  level= (3 * qscale * quant_matrix[0]) >> 5;
335  if (GET_CACHE(re, &s->gb) & 0x40000000)
336  level = -level;
337  block[0] = level;
338  mismatch ^= level;
339  i++;
340  SKIP_BITS(re, &s->gb, 2);
341  if (((int32_t)GET_CACHE(re, &s->gb)) <= (int32_t)0xBFFFFFFF)
342  goto end;
343  }
344 
345  /* now quantify & encode AC coefficients */
346  for (;;) {
347  GET_RL_VLC(level, run, re, &s->gb, rl->rl_vlc[0], TEX_VLC_BITS, 2, 0);
348 
349  if (level != 0) {
350  i += run;
351  j = scantable[i];
352  level = ((level * 2 + 1) * qscale * quant_matrix[j]) >> 5;
353  level = (level ^ SHOW_SBITS(re, &s->gb, 1)) - SHOW_SBITS(re, &s->gb, 1);
354  SKIP_BITS(re, &s->gb, 1);
355  } else {
356  /* escape */
357  run = SHOW_UBITS(re, &s->gb, 6) + 1; LAST_SKIP_BITS(re, &s->gb, 6);
358  UPDATE_CACHE(re, &s->gb);
359  level = SHOW_SBITS(re, &s->gb, 12); SKIP_BITS(re, &s->gb, 12);
360 
361  i += run;
362  j = scantable[i];
363  if (level < 0) {
364  level = ((-level * 2 + 1) * qscale * quant_matrix[j]) >> 5;
365  level = -level;
366  } else {
367  level = ((level * 2 + 1) * qscale * quant_matrix[j]) >> 5;
368  }
369  }
370  if (i > 63) {
371  av_log(s->avctx, AV_LOG_ERROR, "ac-tex damaged at %d %d\n", s->mb_x, s->mb_y);
372  return -1;
373  }
374 
375  mismatch ^= level;
376  block[j] = level;
377  if (((int32_t)GET_CACHE(re, &s->gb)) <= (int32_t)0xBFFFFFFF)
378  break;
379  UPDATE_CACHE(re, &s->gb);
380  }
381 end:
382  LAST_SKIP_BITS(re, &s->gb, 2);
383  CLOSE_READER(re, &s->gb);
384  }
385  block[63] ^= (mismatch & 1);
386 
387  s->block_last_index[n] = i;
388  return 0;
389 }
390 
391 /**
392  * Note: this function can read out of range and crash for corrupt streams.
393  * Changing this would eat up any speed benefits it has.
394  * Do not use "fast" flag if you need the code to be robust.
395  */
397  int16_t *block, int n)
398 {
399  int level, i, j, run;
400  RLTable *rl = &ff_rl_mpeg1;
401  uint8_t * const scantable = s->intra_scantable.permutated;
402  const int qscale = s->qscale;
403  OPEN_READER(re, &s->gb);
404  i = -1;
405 
406  // special case for first coefficient, no need to add second VLC table
407  UPDATE_CACHE(re, &s->gb);
408  if (((int32_t)GET_CACHE(re, &s->gb)) < 0) {
409  level = (3 * qscale) >> 1;
410  if (GET_CACHE(re, &s->gb) & 0x40000000)
411  level = -level;
412  block[0] = level;
413  i++;
414  SKIP_BITS(re, &s->gb, 2);
415  if (((int32_t)GET_CACHE(re, &s->gb)) <= (int32_t)0xBFFFFFFF)
416  goto end;
417  }
418 
419  /* now quantify & encode AC coefficients */
420  for (;;) {
421  GET_RL_VLC(level, run, re, &s->gb, rl->rl_vlc[0], TEX_VLC_BITS, 2, 0);
422 
423  if (level != 0) {
424  i += run;
425  j = scantable[i];
426  level = ((level * 2 + 1) * qscale) >> 1;
427  level = (level ^ SHOW_SBITS(re, &s->gb, 1)) - SHOW_SBITS(re, &s->gb, 1);
428  SKIP_BITS(re, &s->gb, 1);
429  } else {
430  /* escape */
431  run = SHOW_UBITS(re, &s->gb, 6) + 1; LAST_SKIP_BITS(re, &s->gb, 6);
432  UPDATE_CACHE(re, &s->gb);
433  level = SHOW_SBITS(re, &s->gb, 12); SKIP_BITS(re, &s->gb, 12);
434 
435  i += run;
436  j = scantable[i];
437  if (level < 0) {
438  level = ((-level * 2 + 1) * qscale) >> 1;
439  level = -level;
440  } else {
441  level = ((level * 2 + 1) * qscale) >> 1;
442  }
443  }
444 
445  block[j] = level;
446  if (((int32_t)GET_CACHE(re, &s->gb)) <= (int32_t)0xBFFFFFFF)
447  break;
448  UPDATE_CACHE(re, &s->gb);
449  }
450 end:
451  LAST_SKIP_BITS(re, &s->gb, 2);
452  CLOSE_READER(re, &s->gb);
453  s->block_last_index[n] = i;
454  return 0;
455 }
456 
457 
458 static inline int mpeg2_decode_block_intra(MpegEncContext *s, int16_t *block, int n)
459 {
460  int level, dc, diff, i, j, run;
461  int component;
462  RLTable *rl;
463  uint8_t * const scantable = s->intra_scantable.permutated;
464  const uint16_t *quant_matrix;
465  const int qscale = s->qscale;
466  int mismatch;
467 
468  /* DC coefficient */
469  if (n < 4) {
470  quant_matrix = s->intra_matrix;
471  component = 0;
472  } else {
473  quant_matrix = s->chroma_intra_matrix;
474  component = (n & 1) + 1;
475  }
476  diff = decode_dc(&s->gb, component);
477  if (diff >= 0xffff)
478  return -1;
479  dc = s->last_dc[component];
480  dc += diff;
481  s->last_dc[component] = dc;
482  block[0] = dc << (3 - s->intra_dc_precision);
483  av_dlog(s->avctx, "dc=%d\n", block[0]);
484  mismatch = block[0] ^ 1;
485  i = 0;
486  if (s->intra_vlc_format)
487  rl = &ff_rl_mpeg2;
488  else
489  rl = &ff_rl_mpeg1;
490 
491  {
492  OPEN_READER(re, &s->gb);
493  /* now quantify & encode AC coefficients */
494  for (;;) {
495  UPDATE_CACHE(re, &s->gb);
496  GET_RL_VLC(level, run, re, &s->gb, rl->rl_vlc[0], TEX_VLC_BITS, 2, 0);
497 
498  if (level == 127) {
499  break;
500  } else if (level != 0) {
501  i += run;
502  j = scantable[i];
503  level = (level * qscale * quant_matrix[j]) >> 4;
504  level = (level ^ SHOW_SBITS(re, &s->gb, 1)) - SHOW_SBITS(re, &s->gb, 1);
505  LAST_SKIP_BITS(re, &s->gb, 1);
506  } else {
507  /* escape */
508  run = SHOW_UBITS(re, &s->gb, 6) + 1; LAST_SKIP_BITS(re, &s->gb, 6);
509  UPDATE_CACHE(re, &s->gb);
510  level = SHOW_SBITS(re, &s->gb, 12); SKIP_BITS(re, &s->gb, 12);
511  i += run;
512  j = scantable[i];
513  if (level < 0) {
514  level = (-level * qscale * quant_matrix[j]) >> 4;
515  level = -level;
516  } else {
517  level = (level * qscale * quant_matrix[j]) >> 4;
518  }
519  }
520  if (i > 63) {
521  av_log(s->avctx, AV_LOG_ERROR, "ac-tex damaged at %d %d\n", s->mb_x, s->mb_y);
522  return -1;
523  }
524 
525  mismatch ^= level;
526  block[j] = level;
527  }
528  CLOSE_READER(re, &s->gb);
529  }
530  block[63] ^= mismatch & 1;
531 
532  s->block_last_index[n] = i;
533  return 0;
534 }
535 
536 /**
537  * Note: this function can read out of range and crash for corrupt streams.
538  * Changing this would eat up any speed benefits it has.
539  * Do not use "fast" flag if you need the code to be robust.
540  */
541 static inline int mpeg2_fast_decode_block_intra(MpegEncContext *s, int16_t *block, int n)
542 {
543  int level, dc, diff, j, run;
544  int component;
545  RLTable *rl;
546  uint8_t * scantable = s->intra_scantable.permutated;
547  const uint16_t *quant_matrix;
548  const int qscale = s->qscale;
549 
550  /* DC coefficient */
551  if (n < 4) {
552  quant_matrix = s->intra_matrix;
553  component = 0;
554  } else {
555  quant_matrix = s->chroma_intra_matrix;
556  component = (n & 1) + 1;
557  }
558  diff = decode_dc(&s->gb, component);
559  if (diff >= 0xffff)
560  return -1;
561  dc = s->last_dc[component];
562  dc += diff;
563  s->last_dc[component] = dc;
564  block[0] = dc << (3 - s->intra_dc_precision);
565  if (s->intra_vlc_format)
566  rl = &ff_rl_mpeg2;
567  else
568  rl = &ff_rl_mpeg1;
569 
570  {
571  OPEN_READER(re, &s->gb);
572  /* now quantify & encode AC coefficients */
573  for (;;) {
574  UPDATE_CACHE(re, &s->gb);
575  GET_RL_VLC(level, run, re, &s->gb, rl->rl_vlc[0], TEX_VLC_BITS, 2, 0);
576 
577  if (level == 127) {
578  break;
579  } else if (level != 0) {
580  scantable += run;
581  j = *scantable;
582  level = (level * qscale * quant_matrix[j]) >> 4;
583  level = (level ^ SHOW_SBITS(re, &s->gb, 1)) - SHOW_SBITS(re, &s->gb, 1);
584  LAST_SKIP_BITS(re, &s->gb, 1);
585  } else {
586  /* escape */
587  run = SHOW_UBITS(re, &s->gb, 6) + 1; LAST_SKIP_BITS(re, &s->gb, 6);
588  UPDATE_CACHE(re, &s->gb);
589  level = SHOW_SBITS(re, &s->gb, 12); SKIP_BITS(re, &s->gb, 12);
590  scantable += run;
591  j = *scantable;
592  if (level < 0) {
593  level = (-level * qscale * quant_matrix[j]) >> 4;
594  level = -level;
595  } else {
596  level = (level * qscale * quant_matrix[j]) >> 4;
597  }
598  }
599 
600  block[j] = level;
601  }
602  CLOSE_READER(re, &s->gb);
603  }
604 
605  s->block_last_index[n] = scantable - s->intra_scantable.permutated;
606  return 0;
607 }
608 
610 
611 #define INIT_2D_VLC_RL(rl, static_size)\
612 {\
613  static RL_VLC_ELEM rl_vlc_table[static_size];\
614  INIT_VLC_STATIC(&rl.vlc, TEX_VLC_BITS, rl.n + 2,\
615  &rl.table_vlc[0][1], 4, 2,\
616  &rl.table_vlc[0][0], 4, 2, static_size);\
617 \
618  rl.rl_vlc[0] = rl_vlc_table;\
619  init_2d_vlc_rl(&rl);\
620 }
621 
622 static void init_2d_vlc_rl(RLTable *rl)
623 {
624  int i;
625 
626  for (i = 0; i < rl->vlc.table_size; i++) {
627  int code = rl->vlc.table[i][0];
628  int len = rl->vlc.table[i][1];
629  int level, run;
630 
631  if (len == 0) { // illegal code
632  run = 65;
633  level = MAX_LEVEL;
634  } else if (len<0) { //more bits needed
635  run = 0;
636  level = code;
637  } else {
638  if (code == rl->n) { //esc
639  run = 65;
640  level = 0;
641  } else if (code == rl->n+1) { //eob
642  run = 0;
643  level = 127;
644  } else {
645  run = rl->table_run [code] + 1;
646  level = rl->table_level[code];
647  }
648  }
649  rl->rl_vlc[0][i].len = len;
650  rl->rl_vlc[0][i].level = level;
651  rl->rl_vlc[0][i].run = run;
652  }
653 }
654 
656 {
657 
658  s->y_dc_scale_table =
660 
661 }
662 
664 {
665  s->last_dc[0] = 1 << (7 + s->intra_dc_precision);
666  s->last_dc[1] = s->last_dc[0];
667  s->last_dc[2] = s->last_dc[0];
668  memset(s->last_mv, 0, sizeof(s->last_mv));
669 }
670 
671 
672 /******************************************/
673 /* decoding */
674 
677 
682 
684 {
685  static int done = 0;
686 
687  if (!done) {
688  done = 1;
689 
690  INIT_VLC_STATIC(&ff_dc_lum_vlc, DC_VLC_BITS, 12,
692  ff_mpeg12_vlc_dc_lum_code, 2, 2, 512);
693  INIT_VLC_STATIC(&ff_dc_chroma_vlc, DC_VLC_BITS, 12,
695  ff_mpeg12_vlc_dc_chroma_code, 2, 2, 514);
696  INIT_VLC_STATIC(&mv_vlc, MV_VLC_BITS, 17,
697  &ff_mpeg12_mbMotionVectorTable[0][1], 2, 1,
698  &ff_mpeg12_mbMotionVectorTable[0][0], 2, 1, 518);
699  INIT_VLC_STATIC(&mbincr_vlc, MBINCR_VLC_BITS, 36,
700  &ff_mpeg12_mbAddrIncrTable[0][1], 2, 1,
701  &ff_mpeg12_mbAddrIncrTable[0][0], 2, 1, 538);
702  INIT_VLC_STATIC(&mb_pat_vlc, MB_PAT_VLC_BITS, 64,
703  &ff_mpeg12_mbPatTable[0][1], 2, 1,
704  &ff_mpeg12_mbPatTable[0][0], 2, 1, 512);
705 
706  INIT_VLC_STATIC(&mb_ptype_vlc, MB_PTYPE_VLC_BITS, 7,
707  &table_mb_ptype[0][1], 2, 1,
708  &table_mb_ptype[0][0], 2, 1, 64);
709  INIT_VLC_STATIC(&mb_btype_vlc, MB_BTYPE_VLC_BITS, 11,
710  &table_mb_btype[0][1], 2, 1,
711  &table_mb_btype[0][0], 2, 1, 64);
714 
717  }
718 }
719 
720 static inline int get_dmv(MpegEncContext *s)
721 {
722  if (get_bits1(&s->gb))
723  return 1 - (get_bits1(&s->gb) << 1);
724  else
725  return 0;
726 }
727 
728 static inline int get_qscale(MpegEncContext *s)
729 {
730  int qscale = get_bits(&s->gb, 5);
731  if (s->q_scale_type) {
732  return non_linear_qscale[qscale];
733  } else {
734  return qscale << 1;
735  }
736 }
737 
739 {
740  int16_t (*tmp)[64];
741 
742  tmp = s->pblocks[4];
743  s->pblocks[4] = s->pblocks[5];
744  s->pblocks[5] = tmp;
745 }
746 
747 /* motion type (for MPEG-2) */
748 #define MT_FIELD 1
749 #define MT_FRAME 2
750 #define MT_16X8 2
751 #define MT_DMV 3
752 
753 static int mpeg_decode_mb(MpegEncContext *s, int16_t block[12][64])
754 {
755  int i, j, k, cbp, val, mb_type, motion_type;
756  const int mb_block_count = 4 + (1 << s->chroma_format);
757 
758  av_dlog(s->avctx, "decode_mb: x=%d y=%d\n", s->mb_x, s->mb_y);
759 
760  av_assert2(s->mb_skipped == 0);
761 
762  if (s->mb_skip_run-- != 0) {
763  if (s->pict_type == AV_PICTURE_TYPE_P) {
764  s->mb_skipped = 1;
766  } else {
767  int mb_type;
768 
769  if (s->mb_x)
770  mb_type = s->current_picture.f.mb_type[s->mb_x + s->mb_y * s->mb_stride - 1];
771  else
772  mb_type = s->current_picture.f.mb_type[s->mb_width + (s->mb_y - 1) * s->mb_stride - 1]; // FIXME not sure if this is allowed in MPEG at all
773  if (IS_INTRA(mb_type)) {
774  av_log(s->avctx, AV_LOG_ERROR, "skip with previntra\n");
775  return -1;
776  }
777  s->current_picture.f.mb_type[s->mb_x + s->mb_y*s->mb_stride] =
778  mb_type | MB_TYPE_SKIP;
779 // av_assert2(s->current_picture.f.mb_type[s->mb_x + s->mb_y * s->mb_stride - 1] & (MB_TYPE_16x16 | MB_TYPE_16x8));
780 
781  if ((s->mv[0][0][0] | s->mv[0][0][1] | s->mv[1][0][0] | s->mv[1][0][1]) == 0)
782  s->mb_skipped = 1;
783  }
784 
785  return 0;
786  }
787 
788  switch (s->pict_type) {
789  default:
790  case AV_PICTURE_TYPE_I:
791  if (get_bits1(&s->gb) == 0) {
792  if (get_bits1(&s->gb) == 0) {
793  av_log(s->avctx, AV_LOG_ERROR, "invalid mb type in I Frame at %d %d\n", s->mb_x, s->mb_y);
794  return -1;
795  }
796  mb_type = MB_TYPE_QUANT | MB_TYPE_INTRA;
797  } else {
798  mb_type = MB_TYPE_INTRA;
799  }
800  break;
801  case AV_PICTURE_TYPE_P:
802  mb_type = get_vlc2(&s->gb, mb_ptype_vlc.table, MB_PTYPE_VLC_BITS, 1);
803  if (mb_type < 0) {
804  av_log(s->avctx, AV_LOG_ERROR, "invalid mb type in P Frame at %d %d\n", s->mb_x, s->mb_y);
805  return -1;
806  }
807  mb_type = ptype2mb_type[mb_type];
808  break;
809  case AV_PICTURE_TYPE_B:
810  mb_type = get_vlc2(&s->gb, mb_btype_vlc.table, MB_BTYPE_VLC_BITS, 1);
811  if (mb_type < 0) {
812  av_log(s->avctx, AV_LOG_ERROR, "invalid mb type in B Frame at %d %d\n", s->mb_x, s->mb_y);
813  return -1;
814  }
815  mb_type = btype2mb_type[mb_type];
816  break;
817  }
818  av_dlog(s->avctx, "mb_type=%x\n", mb_type);
819 // motion_type = 0; /* avoid warning */
820  if (IS_INTRA(mb_type)) {
821  s->dsp.clear_blocks(s->block[0]);
822 
823  if (!s->chroma_y_shift) {
824  s->dsp.clear_blocks(s->block[6]);
825  }
826 
827  /* compute DCT type */
828  if (s->picture_structure == PICT_FRAME && // FIXME add an interlaced_dct coded var?
829  !s->frame_pred_frame_dct) {
830  s->interlaced_dct = get_bits1(&s->gb);
831  }
832 
833  if (IS_QUANT(mb_type))
834  s->qscale = get_qscale(s);
835 
837  /* just parse them */
838  if (s->picture_structure != PICT_FRAME)
839  skip_bits1(&s->gb); /* field select */
840 
841  s->mv[0][0][0]= s->last_mv[0][0][0]= s->last_mv[0][1][0] =
842  mpeg_decode_motion(s, s->mpeg_f_code[0][0], s->last_mv[0][0][0]);
843  s->mv[0][0][1]= s->last_mv[0][0][1]= s->last_mv[0][1][1] =
844  mpeg_decode_motion(s, s->mpeg_f_code[0][1], s->last_mv[0][0][1]);
845 
846  skip_bits1(&s->gb); /* marker */
847  } else
848  memset(s->last_mv, 0, sizeof(s->last_mv)); /* reset mv prediction */
849  s->mb_intra = 1;
850  // if 1, we memcpy blocks in xvmcvideo
851  if (CONFIG_MPEG_XVMC_DECODER && s->avctx->xvmc_acceleration > 1) {
852  ff_xvmc_pack_pblocks(s, -1); // inter are always full blocks
853  if (s->swap_uv) {
854  exchange_uv(s);
855  }
856  }
857 
858  if (s->codec_id == AV_CODEC_ID_MPEG2VIDEO) {
859  if (s->flags2 & CODEC_FLAG2_FAST) {
860  for (i = 0; i < 6; i++) {
862  }
863  } else {
864  for (i = 0; i < mb_block_count; i++) {
865  if (mpeg2_decode_block_intra(s, *s->pblocks[i], i) < 0)
866  return -1;
867  }
868  }
869  } else {
870  for (i = 0; i < 6; i++) {
871  if (mpeg1_decode_block_intra(s, *s->pblocks[i], i) < 0)
872  return -1;
873  }
874  }
875  } else {
876  if (mb_type & MB_TYPE_ZERO_MV) {
877  av_assert2(mb_type & MB_TYPE_CBP);
878 
879  s->mv_dir = MV_DIR_FORWARD;
880  if (s->picture_structure == PICT_FRAME) {
881  if (s->picture_structure == PICT_FRAME
882  && !s->frame_pred_frame_dct)
883  s->interlaced_dct = get_bits1(&s->gb);
884  s->mv_type = MV_TYPE_16X16;
885  } else {
886  s->mv_type = MV_TYPE_FIELD;
887  mb_type |= MB_TYPE_INTERLACED;
888  s->field_select[0][0] = s->picture_structure - 1;
889  }
890 
891  if (IS_QUANT(mb_type))
892  s->qscale = get_qscale(s);
893 
894  s->last_mv[0][0][0] = 0;
895  s->last_mv[0][0][1] = 0;
896  s->last_mv[0][1][0] = 0;
897  s->last_mv[0][1][1] = 0;
898  s->mv[0][0][0] = 0;
899  s->mv[0][0][1] = 0;
900  } else {
901  av_assert2(mb_type & MB_TYPE_L0L1);
902  // FIXME decide if MBs in field pictures are MB_TYPE_INTERLACED
903  /* get additional motion vector type */
905  motion_type = MT_FRAME;
906  else {
907  motion_type = get_bits(&s->gb, 2);
908  if (s->picture_structure == PICT_FRAME && HAS_CBP(mb_type))
909  s->interlaced_dct = get_bits1(&s->gb);
910  }
911 
912  if (IS_QUANT(mb_type))
913  s->qscale = get_qscale(s);
914 
915  /* motion vectors */
916  s->mv_dir = (mb_type >> 13) & 3;
917  av_dlog(s->avctx, "motion_type=%d\n", motion_type);
918  switch (motion_type) {
919  case MT_FRAME: /* or MT_16X8 */
920  if (s->picture_structure == PICT_FRAME) {
921  mb_type |= MB_TYPE_16x16;
922  s->mv_type = MV_TYPE_16X16;
923  for (i = 0; i < 2; i++) {
924  if (USES_LIST(mb_type, i)) {
925  /* MT_FRAME */
926  s->mv[i][0][0]= s->last_mv[i][0][0]= s->last_mv[i][1][0] =
927  mpeg_decode_motion(s, s->mpeg_f_code[i][0], s->last_mv[i][0][0]);
928  s->mv[i][0][1]= s->last_mv[i][0][1]= s->last_mv[i][1][1] =
929  mpeg_decode_motion(s, s->mpeg_f_code[i][1], s->last_mv[i][0][1]);
930  /* full_pel: only for MPEG-1 */
931  if (s->full_pel[i]) {
932  s->mv[i][0][0] <<= 1;
933  s->mv[i][0][1] <<= 1;
934  }
935  }
936  }
937  } else {
938  mb_type |= MB_TYPE_16x8 | MB_TYPE_INTERLACED;
939  s->mv_type = MV_TYPE_16X8;
940  for (i = 0; i < 2; i++) {
941  if (USES_LIST(mb_type, i)) {
942  /* MT_16X8 */
943  for (j = 0; j < 2; j++) {
944  s->field_select[i][j] = get_bits1(&s->gb);
945  for (k = 0; k < 2; k++) {
946  val = mpeg_decode_motion(s, s->mpeg_f_code[i][k],
947  s->last_mv[i][j][k]);
948  s->last_mv[i][j][k] = val;
949  s->mv[i][j][k] = val;
950  }
951  }
952  }
953  }
954  }
955  break;
956  case MT_FIELD:
957  s->mv_type = MV_TYPE_FIELD;
958  if (s->picture_structure == PICT_FRAME) {
959  mb_type |= MB_TYPE_16x8 | MB_TYPE_INTERLACED;
960  for (i = 0; i < 2; i++) {
961  if (USES_LIST(mb_type, i)) {
962  for (j = 0; j < 2; j++) {
963  s->field_select[i][j] = get_bits1(&s->gb);
964  val = mpeg_decode_motion(s, s->mpeg_f_code[i][0],
965  s->last_mv[i][j][0]);
966  s->last_mv[i][j][0] = val;
967  s->mv[i][j][0] = val;
968  av_dlog(s->avctx, "fmx=%d\n", val);
969  val = mpeg_decode_motion(s, s->mpeg_f_code[i][1],
970  s->last_mv[i][j][1] >> 1);
971  s->last_mv[i][j][1] = val << 1;
972  s->mv[i][j][1] = val;
973  av_dlog(s->avctx, "fmy=%d\n", val);
974  }
975  }
976  }
977  } else {
979  mb_type |= MB_TYPE_16x16 | MB_TYPE_INTERLACED;
980  for (i = 0; i < 2; i++) {
981  if (USES_LIST(mb_type, i)) {
982  s->field_select[i][0] = get_bits1(&s->gb);
983  for (k = 0; k < 2; k++) {
984  val = mpeg_decode_motion(s, s->mpeg_f_code[i][k],
985  s->last_mv[i][0][k]);
986  s->last_mv[i][0][k] = val;
987  s->last_mv[i][1][k] = val;
988  s->mv[i][0][k] = val;
989  }
990  }
991  }
992  }
993  break;
994  case MT_DMV:
995  if(s->progressive_sequence){
996  av_log(s->avctx, AV_LOG_ERROR, "MT_DMV in progressive_sequence\n");
997  return -1;
998  }
999  s->mv_type = MV_TYPE_DMV;
1000  for (i = 0; i < 2; i++) {
1001  if (USES_LIST(mb_type, i)) {
1002  int dmx, dmy, mx, my, m;
1003  const int my_shift = s->picture_structure == PICT_FRAME;
1004 
1005  mx = mpeg_decode_motion(s, s->mpeg_f_code[i][0],
1006  s->last_mv[i][0][0]);
1007  s->last_mv[i][0][0] = mx;
1008  s->last_mv[i][1][0] = mx;
1009  dmx = get_dmv(s);
1010  my = mpeg_decode_motion(s, s->mpeg_f_code[i][1],
1011  s->last_mv[i][0][1] >> my_shift);
1012  dmy = get_dmv(s);
1013 
1014 
1015  s->last_mv[i][0][1] = my << my_shift;
1016  s->last_mv[i][1][1] = my << my_shift;
1017 
1018  s->mv[i][0][0] = mx;
1019  s->mv[i][0][1] = my;
1020  s->mv[i][1][0] = mx; // not used
1021  s->mv[i][1][1] = my; // not used
1022 
1023  if (s->picture_structure == PICT_FRAME) {
1024  mb_type |= MB_TYPE_16x16 | MB_TYPE_INTERLACED;
1025 
1026  // m = 1 + 2 * s->top_field_first;
1027  m = s->top_field_first ? 1 : 3;
1028 
1029  /* top -> top pred */
1030  s->mv[i][2][0] = ((mx * m + (mx > 0)) >> 1) + dmx;
1031  s->mv[i][2][1] = ((my * m + (my > 0)) >> 1) + dmy - 1;
1032  m = 4 - m;
1033  s->mv[i][3][0] = ((mx * m + (mx > 0)) >> 1) + dmx;
1034  s->mv[i][3][1] = ((my * m + (my > 0)) >> 1) + dmy + 1;
1035  } else {
1036  mb_type |= MB_TYPE_16x16;
1037 
1038  s->mv[i][2][0] = ((mx + (mx > 0)) >> 1) + dmx;
1039  s->mv[i][2][1] = ((my + (my > 0)) >> 1) + dmy;
1041  s->mv[i][2][1]--;
1042  else
1043  s->mv[i][2][1]++;
1044  }
1045  }
1046  }
1047  break;
1048  default:
1049  av_log(s->avctx, AV_LOG_ERROR, "00 motion_type at %d %d\n", s->mb_x, s->mb_y);
1050  return -1;
1051  }
1052  }
1053 
1054  s->mb_intra = 0;
1055  if (HAS_CBP(mb_type)) {
1056  s->dsp.clear_blocks(s->block[0]);
1057 
1058  cbp = get_vlc2(&s->gb, mb_pat_vlc.table, MB_PAT_VLC_BITS, 1);
1059  if (mb_block_count > 6) {
1060  cbp <<= mb_block_count - 6;
1061  cbp |= get_bits(&s->gb, mb_block_count - 6);
1062  s->dsp.clear_blocks(s->block[6]);
1063  }
1064  if (cbp <= 0) {
1065  av_log(s->avctx, AV_LOG_ERROR, "invalid cbp %d at %d %d\n", cbp, s->mb_x, s->mb_y);
1066  return -1;
1067  }
1068 
1069  //if 1, we memcpy blocks in xvmcvideo
1070  if (CONFIG_MPEG_XVMC_DECODER && s->avctx->xvmc_acceleration > 1) {
1071  ff_xvmc_pack_pblocks(s, cbp);
1072  if (s->swap_uv) {
1073  exchange_uv(s);
1074  }
1075  }
1076 
1077  if (s->codec_id == AV_CODEC_ID_MPEG2VIDEO) {
1078  if (s->flags2 & CODEC_FLAG2_FAST) {
1079  for (i = 0; i < 6; i++) {
1080  if (cbp & 32) {
1082  } else {
1083  s->block_last_index[i] = -1;
1084  }
1085  cbp += cbp;
1086  }
1087  } else {
1088  cbp <<= 12-mb_block_count;
1089 
1090  for (i = 0; i < mb_block_count; i++) {
1091  if (cbp & (1 << 11)) {
1092  if (mpeg2_decode_block_non_intra(s, *s->pblocks[i], i) < 0)
1093  return -1;
1094  } else {
1095  s->block_last_index[i] = -1;
1096  }
1097  cbp += cbp;
1098  }
1099  }
1100  } else {
1101  if (s->flags2 & CODEC_FLAG2_FAST) {
1102  for (i = 0; i < 6; i++) {
1103  if (cbp & 32) {
1104  mpeg1_fast_decode_block_inter(s, *s->pblocks[i], i);
1105  } else {
1106  s->block_last_index[i] = -1;
1107  }
1108  cbp += cbp;
1109  }
1110  } else {
1111  for (i = 0; i < 6; i++) {
1112  if (cbp & 32) {
1113  if (mpeg1_decode_block_inter(s, *s->pblocks[i], i) < 0)
1114  return -1;
1115  } else {
1116  s->block_last_index[i] = -1;
1117  }
1118  cbp += cbp;
1119  }
1120  }
1121  }
1122  } else {
1123  for (i = 0; i < 12; i++)
1124  s->block_last_index[i] = -1;
1125  }
1126  }
1127 
1128  s->current_picture.f.mb_type[s->mb_x + s->mb_y * s->mb_stride] = mb_type;
1129 
1130  return 0;
1131 }
1132 
1134 {
1135  Mpeg1Context *s = avctx->priv_data;
1137  int i;
1138 
1139  /* we need some permutation to store matrices,
1140  * until MPV_common_init() sets the real permutation. */
1141  for (i = 0; i < 64; i++)
1142  s2->dsp.idct_permutation[i]=i;
1143 
1145 
1146  s->mpeg_enc_ctx.avctx = avctx;
1147  s->mpeg_enc_ctx.flags = avctx->flags;
1148  s->mpeg_enc_ctx.flags2 = avctx->flags2;
1151 
1152  s->mpeg_enc_ctx_allocated = 0;
1154  s->repeat_field = 0;
1155  s->mpeg_enc_ctx.codec_id = avctx->codec->id;
1156  avctx->color_range = AVCOL_RANGE_MPEG;
1157  if (avctx->codec->id == AV_CODEC_ID_MPEG1VIDEO)
1159  else
1161  return 0;
1162 }
1163 
1165 {
1166  Mpeg1Context *ctx = avctx->priv_data, *ctx_from = avctx_from->priv_data;
1167  MpegEncContext *s = &ctx->mpeg_enc_ctx, *s1 = &ctx_from->mpeg_enc_ctx;
1168  int err;
1169 
1170  if (avctx == avctx_from || !ctx_from->mpeg_enc_ctx_allocated || !s1->context_initialized)
1171  return 0;
1172 
1173  err = ff_mpeg_update_thread_context(avctx, avctx_from);
1174  if (err) return err;
1175 
1176  if (!ctx->mpeg_enc_ctx_allocated)
1177  memcpy(s + 1, s1 + 1, sizeof(Mpeg1Context) - sizeof(MpegEncContext));
1178 
1179  if (!(s->pict_type == AV_PICTURE_TYPE_B || s->low_delay))
1180  s->picture_number++;
1181 
1182  return 0;
1183 }
1184 
1185 static void quant_matrix_rebuild(uint16_t *matrix, const uint8_t *old_perm,
1186  const uint8_t *new_perm)
1187 {
1188  uint16_t temp_matrix[64];
1189  int i;
1190 
1191  memcpy(temp_matrix, matrix, 64 * sizeof(uint16_t));
1192 
1193  for (i = 0; i < 64; i++) {
1194  matrix[new_perm[i]] = temp_matrix[old_perm[i]];
1195  }
1196 }
1197 
1199 #if CONFIG_MPEG_XVMC_DECODER
1202 #endif
1203 #if CONFIG_MPEG1_VDPAU_HWACCEL
1205 #endif
1208 };
1209 
1211 #if CONFIG_MPEG_XVMC_DECODER
1214 #endif
1215 #if CONFIG_MPEG2_VDPAU_HWACCEL
1217 #endif
1218 #if CONFIG_MPEG2_DXVA2_HWACCEL
1220 #endif
1221 #if CONFIG_MPEG2_VAAPI_HWACCEL
1223 #endif
1226 };
1227 
1228 static inline int uses_vdpau(AVCodecContext *avctx) {
1229  return avctx->pix_fmt == AV_PIX_FMT_VDPAU_MPEG1 || avctx->pix_fmt == AV_PIX_FMT_VDPAU_MPEG2;
1230 }
1231 
1233 {
1234  Mpeg1Context *s1 = avctx->priv_data;
1235  MpegEncContext *s = &s1->mpeg_enc_ctx;
1236 
1237  if(s->chroma_format < 2) {
1238  enum AVPixelFormat res;
1239  res = avctx->get_format(avctx,
1240  avctx->codec_id == AV_CODEC_ID_MPEG1VIDEO ?
1244  avctx->xvmc_acceleration = 0;
1245  } else if (!avctx->xvmc_acceleration) {
1246  avctx->xvmc_acceleration = 2;
1247  }
1248  return res;
1249  } else if(s->chroma_format == 2)
1250  return AV_PIX_FMT_YUV422P;
1251  else
1252  return AV_PIX_FMT_YUV444P;
1253 }
1254 
1255 /* Call this function when we know all parameters.
1256  * It may be called in different places for MPEG-1 and MPEG-2. */
1258 {
1259  Mpeg1Context *s1 = avctx->priv_data;
1260  MpegEncContext *s = &s1->mpeg_enc_ctx;
1261  uint8_t old_permutation[64];
1262 
1263  if ((s1->mpeg_enc_ctx_allocated == 0) ||
1264  avctx->coded_width != s->width ||
1265  avctx->coded_height != s->height ||
1266  s1->save_width != s->width ||
1267  s1->save_height != s->height ||
1268  s1->save_aspect_info != s->aspect_ratio_info ||
1270  0)
1271  {
1272 
1273  if (s1->mpeg_enc_ctx_allocated) {
1274  ParseContext pc = s->parse_context;
1275  s->parse_context.buffer = 0;
1276  ff_MPV_common_end(s);
1277  s->parse_context = pc;
1278  }
1279 
1280  if ((s->width == 0) || (s->height == 0))
1281  return -2;
1282 
1283  avcodec_set_dimensions(avctx, s->width, s->height);
1284  if (avctx->codec_id == AV_CODEC_ID_MPEG2VIDEO && s->bit_rate) {
1285  avctx->rc_max_rate = s->bit_rate;
1286  } else if (avctx->codec_id == AV_CODEC_ID_MPEG1VIDEO && s->bit_rate &&
1287  (s->bit_rate != 0x3FFFF*400 || s->vbv_delay != 0xFFFF)) {
1288  avctx->bit_rate = s->bit_rate;
1289  }
1291  s1->save_width = s->width;
1292  s1->save_height = s->height;
1294 
1295  /* low_delay may be forced, in this case we will have B-frames
1296  * that behave like P-frames. */
1297  avctx->has_b_frames = !s->low_delay;
1298 
1299  if (avctx->codec_id == AV_CODEC_ID_MPEG1VIDEO) {
1300  //MPEG-1 fps
1303  //MPEG-1 aspect
1305  avctx->ticks_per_frame=1;
1306  } else {//MPEG-2
1307  //MPEG-2 fps
1309  &s->avctx->time_base.num,
1312  1 << 30);
1313  avctx->ticks_per_frame = 2;
1314  //MPEG-2 aspect
1315  if (s->aspect_ratio_info > 1) {
1316  AVRational dar =
1318  (AVRational) {s1->pan_scan.width, s1->pan_scan.height}),
1319  (AVRational) {s->width, s->height});
1320 
1321  // we ignore the spec here and guess a bit as reality does not match the spec, see for example
1322  // res_change_ffmpeg_aspect.ts and sequence-display-aspect.mpg
1323  // issue1613, 621, 562
1324  if ((s1->pan_scan.width == 0) || (s1->pan_scan.height == 0) ||
1325  (av_cmp_q(dar, (AVRational) {4, 3}) && av_cmp_q(dar, (AVRational) {16, 9}))) {
1328  (AVRational) {s->width, s->height});
1329  } else {
1332  (AVRational) {s1->pan_scan.width, s1->pan_scan.height});
1333 //issue1613 4/3 16/9 -> 16/9
1334 //res_change_ffmpeg_aspect.ts 4/3 225/44 ->4/3
1335 //widescreen-issue562.mpg 4/3 16/9 -> 16/9
1336 // s->avctx->sample_aspect_ratio = av_mul_q(s->avctx->sample_aspect_ratio, (AVRational) {s->width, s->height});
1337  av_dlog(avctx, "A %d/%d\n",
1339  av_dlog(avctx, "B %d/%d\n", s->avctx->sample_aspect_ratio.num,
1341  }
1342  } else {
1343  s->avctx->sample_aspect_ratio =
1344  ff_mpeg2_aspect[s->aspect_ratio_info];
1345  }
1346  } // MPEG-2
1347 
1348  avctx->pix_fmt = mpeg_get_pixelformat(avctx);
1349  avctx->hwaccel = ff_find_hwaccel(avctx->codec->id, avctx->pix_fmt);
1350  // until then pix_fmt may be changed right after codec init
1351  if (avctx->pix_fmt == AV_PIX_FMT_XVMC_MPEG2_IDCT ||
1352  avctx->hwaccel || uses_vdpau(avctx))
1353  if (avctx->idct_algo == FF_IDCT_AUTO)
1354  avctx->idct_algo = FF_IDCT_SIMPLE;
1355 
1356  /* Quantization matrices may need reordering
1357  * if DCT permutation is changed. */
1358  memcpy(old_permutation, s->dsp.idct_permutation, 64 * sizeof(uint8_t));
1359 
1360  if (ff_MPV_common_init(s) < 0)
1361  return -2;
1362 
1363  quant_matrix_rebuild(s->intra_matrix, old_permutation, s->dsp.idct_permutation);
1364  quant_matrix_rebuild(s->inter_matrix, old_permutation, s->dsp.idct_permutation);
1365  quant_matrix_rebuild(s->chroma_intra_matrix, old_permutation, s->dsp.idct_permutation);
1366  quant_matrix_rebuild(s->chroma_inter_matrix, old_permutation, s->dsp.idct_permutation);
1367 
1368  s1->mpeg_enc_ctx_allocated = 1;
1369  }
1370  return 0;
1371 }
1372 
1374  const uint8_t *buf, int buf_size)
1375 {
1376  Mpeg1Context *s1 = avctx->priv_data;
1377  MpegEncContext *s = &s1->mpeg_enc_ctx;
1378  int ref, f_code, vbv_delay;
1379 
1380  init_get_bits(&s->gb, buf, buf_size*8);
1381 
1382  ref = get_bits(&s->gb, 10); /* temporal ref */
1383  s->pict_type = get_bits(&s->gb, 3);
1384  if (s->pict_type == 0 || s->pict_type > 3)
1385  return -1;
1386 
1387  vbv_delay = get_bits(&s->gb, 16);
1388  s->vbv_delay = vbv_delay;
1390  s->full_pel[0] = get_bits1(&s->gb);
1391  f_code = get_bits(&s->gb, 3);
1392  if (f_code == 0 && (avctx->err_recognition & (AV_EF_BITSTREAM|AV_EF_COMPLIANT)))
1393  return -1;
1394  f_code += !f_code;
1395  s->mpeg_f_code[0][0] = f_code;
1396  s->mpeg_f_code[0][1] = f_code;
1397  }
1398  if (s->pict_type == AV_PICTURE_TYPE_B) {
1399  s->full_pel[1] = get_bits1(&s->gb);
1400  f_code = get_bits(&s->gb, 3);
1401  if (f_code == 0 && (avctx->err_recognition & (AV_EF_BITSTREAM|AV_EF_COMPLIANT)))
1402  return -1;
1403  f_code += !f_code;
1404  s->mpeg_f_code[1][0] = f_code;
1405  s->mpeg_f_code[1][1] = f_code;
1406  }
1409 
1410  if (avctx->debug & FF_DEBUG_PICT_INFO)
1411  av_log(avctx, AV_LOG_DEBUG, "vbv_delay %d, ref %d type:%d\n", vbv_delay, ref, s->pict_type);
1412 
1413  s->y_dc_scale = 8;
1414  s->c_dc_scale = 8;
1415  return 0;
1416 }
1417 
1419 {
1420  MpegEncContext *s= &s1->mpeg_enc_ctx;
1421  int horiz_size_ext, vert_size_ext;
1422  int bit_rate_ext;
1423 
1424  skip_bits(&s->gb, 1); /* profile and level esc*/
1425  s->avctx->profile = get_bits(&s->gb, 3);
1426  s->avctx->level = get_bits(&s->gb, 4);
1427  s->progressive_sequence = get_bits1(&s->gb); /* progressive_sequence */
1428  s->chroma_format = get_bits(&s->gb, 2); /* chroma_format 1=420, 2=422, 3=444 */
1429  horiz_size_ext = get_bits(&s->gb, 2);
1430  vert_size_ext = get_bits(&s->gb, 2);
1431  s->width |= (horiz_size_ext << 12);
1432  s->height |= (vert_size_ext << 12);
1433  bit_rate_ext = get_bits(&s->gb, 12); /* XXX: handle it */
1434  s->bit_rate += (bit_rate_ext << 18) * 400;
1435  skip_bits1(&s->gb); /* marker */
1436  s->avctx->rc_buffer_size += get_bits(&s->gb, 8) * 1024 * 16 << 10;
1437 
1438  s->low_delay = get_bits1(&s->gb);
1439  if (s->flags & CODEC_FLAG_LOW_DELAY)
1440  s->low_delay = 1;
1441 
1442  s1->frame_rate_ext.num = get_bits(&s->gb, 2) + 1;
1443  s1->frame_rate_ext.den = get_bits(&s->gb, 5) + 1;
1444 
1445  av_dlog(s->avctx, "sequence extension\n");
1447 
1448  if (s->avctx->debug & FF_DEBUG_PICT_INFO)
1449  av_log(s->avctx, AV_LOG_DEBUG, "profile: %d, level: %d vbv buffer: %d, bitrate:%d\n",
1450  s->avctx->profile, s->avctx->level, s->avctx->rc_buffer_size, s->bit_rate);
1451 
1452 }
1453 
1455 {
1456  MpegEncContext *s = &s1->mpeg_enc_ctx;
1457  int color_description, w, h;
1458 
1459  skip_bits(&s->gb, 3); /* video format */
1460  color_description = get_bits1(&s->gb);
1461  if (color_description) {
1462  s->avctx->color_primaries = get_bits(&s->gb, 8);
1463  s->avctx->color_trc = get_bits(&s->gb, 8);
1464  s->avctx->colorspace = get_bits(&s->gb, 8);
1465  }
1466  w = get_bits(&s->gb, 14);
1467  skip_bits(&s->gb, 1); //marker
1468  h = get_bits(&s->gb, 14);
1469  // remaining 3 bits are zero padding
1470 
1471  s1->pan_scan.width = 16 * w;
1472  s1->pan_scan.height = 16 * h;
1473 
1474  if (s->avctx->debug & FF_DEBUG_PICT_INFO)
1475  av_log(s->avctx, AV_LOG_DEBUG, "sde w:%d, h:%d\n", w, h);
1476 }
1477 
1479 {
1480  MpegEncContext *s = &s1->mpeg_enc_ctx;
1481  int i, nofco;
1482 
1483  nofco = 1;
1484  if (s->progressive_sequence) {
1485  if (s->repeat_first_field) {
1486  nofco++;
1487  if (s->top_field_first)
1488  nofco++;
1489  }
1490  } else {
1491  if (s->picture_structure == PICT_FRAME) {
1492  nofco++;
1493  if (s->repeat_first_field)
1494  nofco++;
1495  }
1496  }
1497  for (i = 0; i < nofco; i++) {
1498  s1->pan_scan.position[i][0] = get_sbits(&s->gb, 16);
1499  skip_bits(&s->gb, 1); // marker
1500  s1->pan_scan.position[i][1] = get_sbits(&s->gb, 16);
1501  skip_bits(&s->gb, 1); // marker
1502  }
1503 
1504  if (s->avctx->debug & FF_DEBUG_PICT_INFO)
1505  av_log(s->avctx, AV_LOG_DEBUG, "pde (%d,%d) (%d,%d) (%d,%d)\n",
1506  s1->pan_scan.position[0][0], s1->pan_scan.position[0][1],
1507  s1->pan_scan.position[1][0], s1->pan_scan.position[1][1],
1508  s1->pan_scan.position[2][0], s1->pan_scan.position[2][1]);
1509 }
1510 
1511 static int load_matrix(MpegEncContext *s, uint16_t matrix0[64], uint16_t matrix1[64], int intra)
1512 {
1513  int i;
1514 
1515  for (i = 0; i < 64; i++) {
1516  int j = s->dsp.idct_permutation[ff_zigzag_direct[i]];
1517  int v = get_bits(&s->gb, 8);
1518  if (v == 0) {
1519  av_log(s->avctx, AV_LOG_ERROR, "matrix damaged\n");
1520  return -1;
1521  }
1522  if (intra && i == 0 && v != 8) {
1523  av_log(s->avctx, AV_LOG_DEBUG, "intra matrix specifies invalid DC quantizer %d, ignoring\n", v);
1524  v = 8; // needed by pink.mpg / issue1046
1525  }
1526  matrix0[j] = v;
1527  if (matrix1)
1528  matrix1[j] = v;
1529  }
1530  return 0;
1531 }
1532 
1534 {
1535  av_dlog(s->avctx, "matrix extension\n");
1536 
1537  if (get_bits1(&s->gb)) load_matrix(s, s->chroma_intra_matrix, s->intra_matrix, 1);
1538  if (get_bits1(&s->gb)) load_matrix(s, s->chroma_inter_matrix, s->inter_matrix, 0);
1539  if (get_bits1(&s->gb)) load_matrix(s, s->chroma_intra_matrix, NULL , 1);
1540  if (get_bits1(&s->gb)) load_matrix(s, s->chroma_inter_matrix, NULL , 0);
1541 }
1542 
1544 {
1545  MpegEncContext *s = &s1->mpeg_enc_ctx;
1546 
1547  s->full_pel[0] = s->full_pel[1] = 0;
1548  s->mpeg_f_code[0][0] = get_bits(&s->gb, 4);
1549  s->mpeg_f_code[0][1] = get_bits(&s->gb, 4);
1550  s->mpeg_f_code[1][0] = get_bits(&s->gb, 4);
1551  s->mpeg_f_code[1][1] = get_bits(&s->gb, 4);
1552  if (!s->pict_type && s1->mpeg_enc_ctx_allocated) {
1553  av_log(s->avctx, AV_LOG_ERROR, "Missing picture start code, guessing missing values\n");
1554  if (s->mpeg_f_code[1][0] == 15 && s->mpeg_f_code[1][1] == 15) {
1555  if (s->mpeg_f_code[0][0] == 15 && s->mpeg_f_code[0][1] == 15)
1557  else
1559  } else
1563  }
1564  s->mpeg_f_code[0][0] += !s->mpeg_f_code[0][0];
1565  s->mpeg_f_code[0][1] += !s->mpeg_f_code[0][1];
1566  s->mpeg_f_code[1][0] += !s->mpeg_f_code[1][0];
1567  s->mpeg_f_code[1][1] += !s->mpeg_f_code[1][1];
1568 
1569  s->intra_dc_precision = get_bits(&s->gb, 2);
1570  s->picture_structure = get_bits(&s->gb, 2);
1571  s->top_field_first = get_bits1(&s->gb);
1572  s->frame_pred_frame_dct = get_bits1(&s->gb);
1574  s->q_scale_type = get_bits1(&s->gb);
1575  s->intra_vlc_format = get_bits1(&s->gb);
1576  s->alternate_scan = get_bits1(&s->gb);
1577  s->repeat_first_field = get_bits1(&s->gb);
1578  s->chroma_420_type = get_bits1(&s->gb);
1579  s->progressive_frame = get_bits1(&s->gb);
1580 
1581 
1582  if (s->alternate_scan) {
1585  } else {
1588  }
1589 
1590  /* composite display not parsed */
1591  av_dlog(s->avctx, "intra_dc_precision=%d\n", s->intra_dc_precision);
1592  av_dlog(s->avctx, "picture_structure=%d\n", s->picture_structure);
1593  av_dlog(s->avctx, "top field first=%d\n", s->top_field_first);
1594  av_dlog(s->avctx, "repeat first field=%d\n", s->repeat_first_field);
1595  av_dlog(s->avctx, "conceal=%d\n", s->concealment_motion_vectors);
1596  av_dlog(s->avctx, "intra_vlc_format=%d\n", s->intra_vlc_format);
1597  av_dlog(s->avctx, "alternate_scan=%d\n", s->alternate_scan);
1598  av_dlog(s->avctx, "frame_pred_frame_dct=%d\n", s->frame_pred_frame_dct);
1599  av_dlog(s->avctx, "progressive_frame=%d\n", s->progressive_frame);
1600 }
1601 
1602 static int mpeg_field_start(MpegEncContext *s, const uint8_t *buf, int buf_size)
1603 {
1604  AVCodecContext *avctx = s->avctx;
1605  Mpeg1Context *s1 = (Mpeg1Context*)s;
1606 
1607  /* start frame decoding */
1608  if (s->first_field || s->picture_structure == PICT_FRAME) {
1609  if (ff_MPV_frame_start(s, avctx) < 0)
1610  return -1;
1611 
1613 
1614  /* first check if we must repeat the frame */
1616  if (s->repeat_first_field) {
1617  if (s->progressive_sequence) {
1618  if (s->top_field_first)
1620  else
1622  } else if (s->progressive_frame) {
1624  }
1625  }
1626 
1628 
1629  if (HAVE_THREADS && (avctx->active_thread_type & FF_THREAD_FRAME))
1630  ff_thread_finish_setup(avctx);
1631  } else { // second field
1632  int i;
1633 
1634  if (!s->current_picture_ptr) {
1635  av_log(s->avctx, AV_LOG_ERROR, "first field missing\n");
1636  return -1;
1637  }
1638 
1639  if (s->avctx->hwaccel &&
1641  if (s->avctx->hwaccel->end_frame(s->avctx) < 0)
1642  av_log(avctx, AV_LOG_ERROR, "hardware accelerator failed to decode first field\n");
1643  }
1644 
1645  for (i = 0; i < 4; i++) {
1649  }
1650  }
1651  }
1652 
1653  if (avctx->hwaccel) {
1654  if (avctx->hwaccel->start_frame(avctx, buf, buf_size) < 0)
1655  return -1;
1656  }
1657 
1658 // MPV_frame_start will call this function too,
1659 // but we need to call it on every field
1660  if (CONFIG_MPEG_XVMC_DECODER && s->avctx->xvmc_acceleration)
1661  if (ff_xvmc_field_start(s, avctx) < 0)
1662  return -1;
1663 
1664  return 0;
1665 }
1666 
1667 #define DECODE_SLICE_ERROR -1
1668 #define DECODE_SLICE_OK 0
1669 
1670 /**
1671  * Decode a slice.
1672  * MpegEncContext.mb_y must be set to the MB row from the startcode.
1673  * @return DECODE_SLICE_ERROR if the slice is damaged,
1674  * DECODE_SLICE_OK if this slice is OK
1675  */
1676 static int mpeg_decode_slice(MpegEncContext *s, int mb_y,
1677  const uint8_t **buf, int buf_size)
1678 {
1679  AVCodecContext *avctx = s->avctx;
1680  const int lowres = s->avctx->lowres;
1681  const int field_pic = s->picture_structure != PICT_FRAME;
1682 
1683  s->resync_mb_x =
1684  s->resync_mb_y = -1;
1685 
1686  av_assert0(mb_y < s->mb_height);
1687 
1688  init_get_bits(&s->gb, *buf, buf_size * 8);
1689  if(s->codec_id != AV_CODEC_ID_MPEG1VIDEO && s->mb_height > 2800/16)
1690  skip_bits(&s->gb, 3);
1691 
1693  s->interlaced_dct = 0;
1694 
1695  s->qscale = get_qscale(s);
1696 
1697  if (s->qscale == 0) {
1698  av_log(s->avctx, AV_LOG_ERROR, "qscale == 0\n");
1699  return -1;
1700  }
1701 
1702  /* extra slice info */
1703  while (get_bits1(&s->gb) != 0) {
1704  skip_bits(&s->gb, 8);
1705  }
1706 
1707  s->mb_x = 0;
1708 
1709  if (mb_y == 0 && s->codec_tag == AV_RL32("SLIF")) {
1710  skip_bits1(&s->gb);
1711  } else {
1712  while (get_bits_left(&s->gb) > 0) {
1713  int code = get_vlc2(&s->gb, mbincr_vlc.table, MBINCR_VLC_BITS, 2);
1714  if (code < 0) {
1715  av_log(s->avctx, AV_LOG_ERROR, "first mb_incr damaged\n");
1716  return -1;
1717  }
1718  if (code >= 33) {
1719  if (code == 33) {
1720  s->mb_x += 33;
1721  }
1722  /* otherwise, stuffing, nothing to do */
1723  } else {
1724  s->mb_x += code;
1725  break;
1726  }
1727  }
1728  }
1729 
1730  if (s->mb_x >= (unsigned)s->mb_width) {
1731  av_log(s->avctx, AV_LOG_ERROR, "initial skip overflow\n");
1732  return -1;
1733  }
1734 
1735  if (avctx->hwaccel) {
1736  const uint8_t *buf_end, *buf_start = *buf - 4; /* include start_code */
1737  int start_code = -1;
1738  buf_end = avpriv_mpv_find_start_code(buf_start + 2, *buf + buf_size, &start_code);
1739  if (buf_end < *buf + buf_size)
1740  buf_end -= 4;
1741  s->mb_y = mb_y;
1742  if (avctx->hwaccel->decode_slice(avctx, buf_start, buf_end - buf_start) < 0)
1743  return DECODE_SLICE_ERROR;
1744  *buf = buf_end;
1745  return DECODE_SLICE_OK;
1746  }
1747 
1748  s->resync_mb_x = s->mb_x;
1749  s->resync_mb_y = s->mb_y = mb_y;
1750  s->mb_skip_run = 0;
1752 
1753  if (s->mb_y == 0 && s->mb_x == 0 && (s->first_field || s->picture_structure == PICT_FRAME)) {
1754  if (s->avctx->debug & FF_DEBUG_PICT_INFO) {
1755  av_log(s->avctx, AV_LOG_DEBUG, "qp:%d fc:%2d%2d%2d%2d %s %s %s %s %s dc:%d pstruct:%d fdct:%d cmv:%d qtype:%d ivlc:%d rff:%d %s\n",
1756  s->qscale, s->mpeg_f_code[0][0], s->mpeg_f_code[0][1], s->mpeg_f_code[1][0], s->mpeg_f_code[1][1],
1757  s->pict_type == AV_PICTURE_TYPE_I ? "I" : (s->pict_type == AV_PICTURE_TYPE_P ? "P" : (s->pict_type == AV_PICTURE_TYPE_B ? "B" : "S")),
1758  s->progressive_sequence ? "ps" :"", s->progressive_frame ? "pf" : "", s->alternate_scan ? "alt" :"", s->top_field_first ? "top" :"",
1761  }
1762  }
1763 
1764  for (;;) {
1765  // If 1, we memcpy blocks in xvmcvideo.
1766  if (CONFIG_MPEG_XVMC_DECODER && s->avctx->xvmc_acceleration > 1)
1767  ff_xvmc_init_block(s); // set s->block
1768 
1769  if (mpeg_decode_mb(s, s->block) < 0)
1770  return -1;
1771 
1772  if (s->current_picture.f.motion_val[0] && !s->encoding) { // note motion_val is normally NULL unless we want to extract the MVs
1773  const int wrap = s->b8_stride;
1774  int xy = s->mb_x * 2 + s->mb_y * 2 * wrap;
1775  int b8_xy = 4 * (s->mb_x + s->mb_y * s->mb_stride);
1776  int motion_x, motion_y, dir, i;
1777 
1778  for (i = 0; i < 2; i++) {
1779  for (dir = 0; dir < 2; dir++) {
1780  if (s->mb_intra || (dir == 1 && s->pict_type != AV_PICTURE_TYPE_B)) {
1781  motion_x = motion_y = 0;
1782  } else if (s->mv_type == MV_TYPE_16X16 || (s->mv_type == MV_TYPE_FIELD && field_pic)) {
1783  motion_x = s->mv[dir][0][0];
1784  motion_y = s->mv[dir][0][1];
1785  } else /*if ((s->mv_type == MV_TYPE_FIELD) || (s->mv_type == MV_TYPE_16X8))*/ {
1786  motion_x = s->mv[dir][i][0];
1787  motion_y = s->mv[dir][i][1];
1788  }
1789 
1790  s->current_picture.f.motion_val[dir][xy ][0] = motion_x;
1791  s->current_picture.f.motion_val[dir][xy ][1] = motion_y;
1792  s->current_picture.f.motion_val[dir][xy + 1][0] = motion_x;
1793  s->current_picture.f.motion_val[dir][xy + 1][1] = motion_y;
1794  s->current_picture.f.ref_index [dir][b8_xy ] =
1795  s->current_picture.f.ref_index [dir][b8_xy + 1] = s->field_select[dir][i];
1796  av_assert2(s->field_select[dir][i] == 0 || s->field_select[dir][i] == 1);
1797  }
1798  xy += wrap;
1799  b8_xy +=2;
1800  }
1801  }
1802 
1803  s->dest[0] += 16 >> lowres;
1804  s->dest[1] +=(16 >> lowres) >> s->chroma_x_shift;
1805  s->dest[2] +=(16 >> lowres) >> s->chroma_x_shift;
1806 
1807  ff_MPV_decode_mb(s, s->block);
1808 
1809  if (++s->mb_x >= s->mb_width) {
1810  const int mb_size = 16 >> s->avctx->lowres;
1811 
1812  ff_mpeg_draw_horiz_band(s, mb_size*(s->mb_y >> field_pic), mb_size);
1814 
1815  s->mb_x = 0;
1816  s->mb_y += 1 << field_pic;
1817 
1818  if (s->mb_y >= s->mb_height) {
1819  int left = get_bits_left(&s->gb);
1820  int is_d10 = s->chroma_format == 2 && s->pict_type == AV_PICTURE_TYPE_I && avctx->profile == 0 && avctx->level == 5
1821  && s->intra_dc_precision == 2 && s->q_scale_type == 1 && s->alternate_scan == 0
1822  && s->progressive_frame == 0 /* vbv_delay == 0xBBB || 0xE10*/;
1823 
1824  if (left >= 32 && !is_d10) {
1825  GetBitContext gb = s->gb;
1826  align_get_bits(&gb);
1827  if (show_bits(&gb, 24) == 0x060E2B) {
1828  av_log(avctx, AV_LOG_DEBUG, "Invalid MXF data found in video stream\n");
1829  is_d10 = 1;
1830  }
1831  }
1832 
1833  if (left < 0 || (left && show_bits(&s->gb, FFMIN(left, 23)) && !is_d10)
1834  || ((avctx->err_recognition & (AV_EF_BITSTREAM | AV_EF_AGGRESSIVE)) && left > 8)) {
1835  av_log(avctx, AV_LOG_ERROR, "end mismatch left=%d %0X\n", left, show_bits(&s->gb, FFMIN(left, 23)));
1836  return -1;
1837  } else
1838  goto eos;
1839  }
1840 
1842  }
1843 
1844  /* skip mb handling */
1845  if (s->mb_skip_run == -1) {
1846  /* read increment again */
1847  s->mb_skip_run = 0;
1848  for (;;) {
1849  int code = get_vlc2(&s->gb, mbincr_vlc.table, MBINCR_VLC_BITS, 2);
1850  if (code < 0) {
1851  av_log(s->avctx, AV_LOG_ERROR, "mb incr damaged\n");
1852  return -1;
1853  }
1854  if (code >= 33) {
1855  if (code == 33) {
1856  s->mb_skip_run += 33;
1857  } else if (code == 35) {
1858  if (s->mb_skip_run != 0 || show_bits(&s->gb, 15) != 0) {
1859  av_log(s->avctx, AV_LOG_ERROR, "slice mismatch\n");
1860  return -1;
1861  }
1862  goto eos; /* end of slice */
1863  }
1864  /* otherwise, stuffing, nothing to do */
1865  } else {
1866  s->mb_skip_run += code;
1867  break;
1868  }
1869  }
1870  if (s->mb_skip_run) {
1871  int i;
1872  if (s->pict_type == AV_PICTURE_TYPE_I) {
1873  av_log(s->avctx, AV_LOG_ERROR, "skipped MB in I frame at %d %d\n", s->mb_x, s->mb_y);
1874  return -1;
1875  }
1876 
1877  /* skip mb */
1878  s->mb_intra = 0;
1879  for (i = 0; i < 12; i++)
1880  s->block_last_index[i] = -1;
1882  s->mv_type = MV_TYPE_16X16;
1883  else
1884  s->mv_type = MV_TYPE_FIELD;
1885  if (s->pict_type == AV_PICTURE_TYPE_P) {
1886  /* if P type, zero motion vector is implied */
1887  s->mv_dir = MV_DIR_FORWARD;
1888  s->mv[0][0][0] = s->mv[0][0][1] = 0;
1889  s->last_mv[0][0][0] = s->last_mv[0][0][1] = 0;
1890  s->last_mv[0][1][0] = s->last_mv[0][1][1] = 0;
1891  s->field_select[0][0] = (s->picture_structure - 1) & 1;
1892  } else {
1893  /* if B type, reuse previous vectors and directions */
1894  s->mv[0][0][0] = s->last_mv[0][0][0];
1895  s->mv[0][0][1] = s->last_mv[0][0][1];
1896  s->mv[1][0][0] = s->last_mv[1][0][0];
1897  s->mv[1][0][1] = s->last_mv[1][0][1];
1898  }
1899  }
1900  }
1901  }
1902 eos: // end of slice
1903  *buf += (get_bits_count(&s->gb)-1)/8;
1904  av_dlog(s, "y %d %d %d %d\n", s->resync_mb_x, s->resync_mb_y, s->mb_x, s->mb_y);
1905  return 0;
1906 }
1907 
1909 {
1910  MpegEncContext *s = *(void**)arg;
1911  const uint8_t *buf = s->gb.buffer;
1912  int mb_y = s->start_mb_y;
1913  const int field_pic = s->picture_structure != PICT_FRAME;
1914 
1915  s->er.error_count = (3 * (s->end_mb_y - s->start_mb_y) * s->mb_width) >> field_pic;
1916 
1917  for (;;) {
1918  uint32_t start_code;
1919  int ret;
1920 
1921  ret = mpeg_decode_slice(s, mb_y, &buf, s->gb.buffer_end - buf);
1922  emms_c();
1923  av_dlog(c, "ret:%d resync:%d/%d mb:%d/%d ts:%d/%d ec:%d\n",
1924  ret, s->resync_mb_x, s->resync_mb_y, s->mb_x, s->mb_y,
1925  s->start_mb_y, s->end_mb_y, s->er.error_count);
1926  if (ret < 0) {
1927  if (c->err_recognition & AV_EF_EXPLODE)
1928  return ret;
1929  if (s->resync_mb_x >= 0 && s->resync_mb_y >= 0)
1931  } else {
1933  }
1934 
1935  if (s->mb_y == s->end_mb_y)
1936  return 0;
1937 
1938  start_code = -1;
1939  buf = avpriv_mpv_find_start_code(buf, s->gb.buffer_end, &start_code);
1940  mb_y= start_code - SLICE_MIN_START_CODE;
1941  if(s->codec_id != AV_CODEC_ID_MPEG1VIDEO && s->mb_height > 2800/16)
1942  mb_y += (*buf&0xE0)<<2;
1943  mb_y <<= field_pic;
1945  mb_y++;
1946  if (mb_y < 0 || mb_y >= s->end_mb_y)
1947  return -1;
1948  }
1949 }
1950 
1951 /**
1952  * Handle slice ends.
1953  * @return 1 if it seems to be the last slice
1954  */
1955 static int slice_end(AVCodecContext *avctx, AVFrame *pict)
1956 {
1957  Mpeg1Context *s1 = avctx->priv_data;
1958  MpegEncContext *s = &s1->mpeg_enc_ctx;
1959 
1961  return 0;
1962 
1963  if (s->avctx->hwaccel) {
1964  if (s->avctx->hwaccel->end_frame(s->avctx) < 0)
1965  av_log(avctx, AV_LOG_ERROR, "hardware accelerator failed to decode picture\n");
1966  }
1967 
1968  if (CONFIG_MPEG_XVMC_DECODER && s->avctx->xvmc_acceleration)
1969  ff_xvmc_field_end(s);
1970 
1971  /* end of slice reached */
1972  if (/*s->mb_y << field_pic == s->mb_height &&*/ !s->first_field && !s->first_slice) {
1973  /* end of image */
1974 
1976 
1977  ff_er_frame_end(&s->er);
1978 
1979  ff_MPV_frame_end(s);
1980 
1981  if (s->pict_type == AV_PICTURE_TYPE_B || s->low_delay) {
1982  *pict = s->current_picture_ptr->f;
1983  ff_print_debug_info(s, pict);
1984  } else {
1985  if (avctx->active_thread_type & FF_THREAD_FRAME)
1986  s->picture_number++;
1987  /* latency of 1 frame for I- and P-frames */
1988  /* XXX: use another variable than picture_number */
1989  if (s->last_picture_ptr != NULL) {
1990  *pict = s->last_picture_ptr->f;
1991  ff_print_debug_info(s, pict);
1992  }
1993  }
1994 
1995  return 1;
1996  } else {
1997  return 0;
1998  }
1999 }
2000 
2002  const uint8_t *buf, int buf_size)
2003 {
2004  Mpeg1Context *s1 = avctx->priv_data;
2005  MpegEncContext *s = &s1->mpeg_enc_ctx;
2006  int width, height;
2007  int i, v, j;
2008 
2009  init_get_bits(&s->gb, buf, buf_size*8);
2010 
2011  width = get_bits(&s->gb, 12);
2012  height = get_bits(&s->gb, 12);
2013  s->aspect_ratio_info = get_bits(&s->gb, 4);
2014  if (s->aspect_ratio_info == 0) {
2015  av_log(avctx, AV_LOG_ERROR, "aspect ratio has forbidden 0 value\n");
2017  return -1;
2018  }
2019  s->frame_rate_index = get_bits(&s->gb, 4);
2020  if (s->frame_rate_index == 0 || s->frame_rate_index > 13)
2021  return -1;
2022  s->bit_rate = get_bits(&s->gb, 18) * 400;
2023  if (get_bits1(&s->gb) == 0) /* marker */
2024  return -1;
2025  s->width = width;
2026  s->height = height;
2027 
2028  s->avctx->rc_buffer_size = get_bits(&s->gb, 10) * 1024 * 16;
2029  skip_bits(&s->gb, 1);
2030 
2031  /* get matrix */
2032  if (get_bits1(&s->gb)) {
2034  } else {
2035  for (i = 0; i < 64; i++) {
2036  j = s->dsp.idct_permutation[i];
2038  s->intra_matrix[j] = v;
2039  s->chroma_intra_matrix[j] = v;
2040  }
2041  }
2042  if (get_bits1(&s->gb)) {
2044  } else {
2045  for (i = 0; i < 64; i++) {
2046  int j = s->dsp.idct_permutation[i];
2048  s->inter_matrix[j] = v;
2049  s->chroma_inter_matrix[j] = v;
2050  }
2051  }
2052 
2053  if (show_bits(&s->gb, 23) != 0) {
2054  av_log(s->avctx, AV_LOG_ERROR, "sequence header damaged\n");
2055  return -1;
2056  }
2057 
2058  /* we set MPEG-2 parameters so that it emulates MPEG-1 */
2059  s->progressive_sequence = 1;
2060  s->progressive_frame = 1;
2062  s->first_field = 0;
2063  s->frame_pred_frame_dct = 1;
2064  s->chroma_format = 1;
2066  s->out_format = FMT_MPEG1;
2067  s->swap_uv = 0; // AFAIK VCR2 does not have SEQ_HEADER
2068  if (s->flags & CODEC_FLAG_LOW_DELAY)
2069  s->low_delay = 1;
2070 
2071  if (s->avctx->debug & FF_DEBUG_PICT_INFO)
2072  av_log(s->avctx, AV_LOG_DEBUG, "vbv buffer: %d, bitrate:%d\n",
2073  s->avctx->rc_buffer_size, s->bit_rate);
2074 
2075  return 0;
2076 }
2077 
2079 {
2080  Mpeg1Context *s1 = avctx->priv_data;
2081  MpegEncContext *s = &s1->mpeg_enc_ctx;
2082  int i, v;
2083 
2084  /* start new MPEG-1 context decoding */
2085  s->out_format = FMT_MPEG1;
2086  if (s1->mpeg_enc_ctx_allocated) {
2087  ff_MPV_common_end(s);
2088  }
2089  s->width = avctx->coded_width;
2090  s->height = avctx->coded_height;
2091  avctx->has_b_frames = 0; // true?
2092  s->low_delay = 1;
2093 
2094  avctx->pix_fmt = mpeg_get_pixelformat(avctx);
2095  avctx->hwaccel = ff_find_hwaccel(avctx->codec->id, avctx->pix_fmt);
2096 
2097  if (avctx->pix_fmt == AV_PIX_FMT_XVMC_MPEG2_IDCT || avctx->hwaccel || uses_vdpau(avctx))
2098  if (avctx->idct_algo == FF_IDCT_AUTO)
2099  avctx->idct_algo = FF_IDCT_SIMPLE;
2100 
2101  if (ff_MPV_common_init(s) < 0)
2102  return -1;
2103  s1->mpeg_enc_ctx_allocated = 1;
2104 
2105  for (i = 0; i < 64; i++) {
2106  int j = s->dsp.idct_permutation[i];
2108  s->intra_matrix[j] = v;
2109  s->chroma_intra_matrix[j] = v;
2110 
2112  s->inter_matrix[j] = v;
2113  s->chroma_inter_matrix[j] = v;
2114  }
2115 
2116  s->progressive_sequence = 1;
2117  s->progressive_frame = 1;
2119  s->first_field = 0;
2120  s->frame_pred_frame_dct = 1;
2121  s->chroma_format = 1;
2122  if (s->codec_tag == AV_RL32("BW10")) {
2124  } else {
2125  exchange_uv(s); // common init reset pblocks, so we swap them here
2126  s->swap_uv = 1; // in case of xvmc we need to swap uv for each MB
2128  }
2129  s1->save_width = s->width;
2130  s1->save_height = s->height;
2132  return 0;
2133 }
2134 
2135 
2137  const uint8_t *p, int buf_size)
2138 {
2139  Mpeg1Context *s = avctx->priv_data;
2140  const uint8_t *buf_end = p + buf_size;
2141 
2142  if(buf_size > 29){
2143  int i;
2144  for(i=0; i<20; i++)
2145  if(!memcmp(p+i, "\0TMPGEXS\0", 9)){
2146  s->tmpgexs= 1;
2147  }
2148 
2149 /* for(i=0; !(!p[i-2] && !p[i-1] && p[i]==1) && i<buf_size; i++){
2150  av_log(avctx, AV_LOG_ERROR, "%c", p[i]);
2151  }
2152  av_log(avctx, AV_LOG_ERROR, "\n");*/
2153  }
2154 
2155  /* we parse the DTG active format information */
2156  if (buf_end - p >= 5 &&
2157  p[0] == 'D' && p[1] == 'T' && p[2] == 'G' && p[3] == '1') {
2158  int flags = p[4];
2159  p += 5;
2160  if (flags & 0x80) {
2161  /* skip event id */
2162  p += 2;
2163  }
2164  if (flags & 0x40) {
2165  if (buf_end - p < 1)
2166  return;
2167  avctx->dtg_active_format = p[0] & 0x0f;
2168  }
2169  }
2170 }
2171 
2172 static void mpeg_decode_gop(AVCodecContext *avctx,
2173  const uint8_t *buf, int buf_size)
2174 {
2175  Mpeg1Context *s1 = avctx->priv_data;
2176  MpegEncContext *s = &s1->mpeg_enc_ctx;
2177  int broken_link;
2178  int64_t tc;
2179 
2180  init_get_bits(&s->gb, buf, buf_size*8);
2181 
2182  tc = avctx->timecode_frame_start = get_bits(&s->gb, 25);
2183 
2184  s->closed_gop = get_bits1(&s->gb);
2185  /*broken_link indicate that after editing the
2186  reference frames of the first B-Frames after GOP I-Frame
2187  are missing (open gop)*/
2188  broken_link = get_bits1(&s->gb);
2189 
2190  if (s->avctx->debug & FF_DEBUG_PICT_INFO) {
2191  char tcbuf[AV_TIMECODE_STR_SIZE];
2194  "GOP (%s) closed_gop=%d broken_link=%d\n",
2195  tcbuf, s->closed_gop, broken_link);
2196  }
2197 }
2198 /**
2199  * Find the end of the current frame in the bitstream.
2200  * @return the position of the first byte of the next frame, or -1
2201  */
2203 {
2204  int i;
2205  uint32_t state = pc->state;
2206 
2207  /* EOF considered as end of frame */
2208  if (buf_size == 0)
2209  return 0;
2210 
2211 /*
2212  0 frame start -> 1/4
2213  1 first_SEQEXT -> 0/2
2214  2 first field start -> 3/0
2215  3 second_SEQEXT -> 2/0
2216  4 searching end
2217 */
2218 
2219  for (i = 0; i < buf_size; i++) {
2220  av_assert1(pc->frame_start_found >= 0 && pc->frame_start_found <= 4);
2221  if (pc->frame_start_found & 1) {
2222  if (state == EXT_START_CODE && (buf[i] & 0xF0) != 0x80)
2223  pc->frame_start_found--;
2224  else if (state == EXT_START_CODE + 2) {
2225  if ((buf[i] & 3) == 3)
2226  pc->frame_start_found = 0;
2227  else
2228  pc->frame_start_found = (pc->frame_start_found + 1) & 3;
2229  }
2230  state++;
2231  } else {
2232  i = avpriv_mpv_find_start_code(buf + i, buf + buf_size, &state) - buf - 1;
2233  if (pc->frame_start_found == 0 && state >= SLICE_MIN_START_CODE && state <= SLICE_MAX_START_CODE) {
2234  i++;
2235  pc->frame_start_found = 4;
2236  }
2237  if (state == SEQ_END_CODE) {
2238  pc->frame_start_found = 0;
2239  pc->state=-1;
2240  return i+1;
2241  }
2242  if (pc->frame_start_found == 2 && state == SEQ_START_CODE)
2243  pc->frame_start_found = 0;
2244  if (pc->frame_start_found < 4 && state == EXT_START_CODE)
2245  pc->frame_start_found++;
2246  if (pc->frame_start_found == 4 && (state & 0xFFFFFF00) == 0x100) {
2247  if (state < SLICE_MIN_START_CODE || state > SLICE_MAX_START_CODE) {
2248  pc->frame_start_found = 0;
2249  pc->state = -1;
2250  return i - 3;
2251  }
2252  }
2253  if (pc->frame_start_found == 0 && s && state == PICTURE_START_CODE) {
2254  ff_fetch_timestamp(s, i - 3, 1);
2255  }
2256  }
2257  }
2258  pc->state = state;
2259  return END_NOT_FOUND;
2260 }
2261 
2262 static int decode_chunks(AVCodecContext *avctx,
2263  AVFrame *picture, int *got_output,
2264  const uint8_t *buf, int buf_size)
2265 {
2266  Mpeg1Context *s = avctx->priv_data;
2268  const uint8_t *buf_ptr = buf;
2269  const uint8_t *buf_end = buf + buf_size;
2270  int ret, input_size;
2271  int last_code = 0;
2272  int picture_start_code_seen = 0;
2273 
2274  for (;;) {
2275  /* find next start code */
2276  uint32_t start_code = -1;
2277  buf_ptr = avpriv_mpv_find_start_code(buf_ptr, buf_end, &start_code);
2278  if (start_code > 0x1ff) {
2279  if (s2->pict_type != AV_PICTURE_TYPE_B || avctx->skip_frame <= AVDISCARD_DEFAULT) {
2280  if (HAVE_THREADS && (avctx->active_thread_type & FF_THREAD_SLICE)) {
2281  int i;
2282  av_assert0(avctx->thread_count > 1);
2283 
2284  avctx->execute(avctx, slice_decode_thread, &s2->thread_context[0], NULL, s->slice_count, sizeof(void*));
2285  for (i = 0; i < s->slice_count; i++)
2286  s2->er.error_count += s2->thread_context[i]->er.error_count;
2287  }
2288 
2289  if (CONFIG_VDPAU && uses_vdpau(avctx))
2290  ff_vdpau_mpeg_picture_complete(s2, buf, buf_size, s->slice_count);
2291 
2292 
2293  if (slice_end(avctx, picture)) {
2294  if (s2->last_picture_ptr || s2->low_delay) //FIXME merge with the stuff in mpeg_decode_slice
2295  *got_output = 1;
2296  }
2297  }
2298  s2->pict_type = 0;
2299  return FFMAX(0, buf_ptr - buf - s2->parse_context.last_index);
2300  }
2301 
2302  input_size = buf_end - buf_ptr;
2303 
2304  if (avctx->debug & FF_DEBUG_STARTCODE) {
2305  av_log(avctx, AV_LOG_DEBUG, "%3X at %td left %d\n", start_code, buf_ptr-buf, input_size);
2306  }
2307 
2308  /* prepare data for next start code */
2309  switch (start_code) {
2310  case SEQ_START_CODE:
2311  if (last_code == 0) {
2312  mpeg1_decode_sequence(avctx, buf_ptr, input_size);
2313  if(buf != avctx->extradata)
2314  s->sync=1;
2315  } else {
2316  av_log(avctx, AV_LOG_ERROR, "ignoring SEQ_START_CODE after %X\n", last_code);
2317  if (avctx->err_recognition & AV_EF_EXPLODE)
2318  return AVERROR_INVALIDDATA;
2319  }
2320  break;
2321 
2322  case PICTURE_START_CODE:
2323  if (picture_start_code_seen && s2->picture_structure == PICT_FRAME) {
2324  /* If it's a frame picture, there can't be more than one picture header.
2325  Yet, it does happen and we need to handle it. */
2326  av_log(avctx, AV_LOG_WARNING, "ignoring extra picture following a frame-picture\n");
2327  break;
2328  }
2329  picture_start_code_seen = 1;
2330 
2331  if (s2->width <= 0 || s2->height <= 0) {
2332  av_log(avctx, AV_LOG_ERROR, "%dx%d is invalid\n", s2->width, s2->height);
2333  return AVERROR_INVALIDDATA;
2334  }
2335 
2336  if(s->tmpgexs){
2337  s2->intra_dc_precision= 3;
2338  s2->intra_matrix[0]= 1;
2339  }
2340  if (HAVE_THREADS && (avctx->active_thread_type & FF_THREAD_SLICE) && s->slice_count) {
2341  int i;
2342 
2343  avctx->execute(avctx, slice_decode_thread,
2344  s2->thread_context, NULL,
2345  s->slice_count, sizeof(void*));
2346  for (i = 0; i < s->slice_count; i++)
2347  s2->er.error_count += s2->thread_context[i]->er.error_count;
2348  s->slice_count = 0;
2349  }
2350  if (last_code == 0 || last_code == SLICE_MIN_START_CODE) {
2351  ret = mpeg_decode_postinit(avctx);
2352  if (ret < 0) {
2353  av_log(avctx, AV_LOG_ERROR, "mpeg_decode_postinit() failure\n");
2354  return ret;
2355  }
2356 
2357  /* we have a complete image: we try to decompress it */
2358  if (mpeg1_decode_picture(avctx, buf_ptr, input_size) < 0)
2359  s2->pict_type = 0;
2360  s2->first_slice = 1;
2361  last_code = PICTURE_START_CODE;
2362  } else {
2363  av_log(avctx, AV_LOG_ERROR, "ignoring pic after %X\n", last_code);
2364  if (avctx->err_recognition & AV_EF_EXPLODE)
2365  return AVERROR_INVALIDDATA;
2366  }
2367  break;
2368  case EXT_START_CODE:
2369  init_get_bits(&s2->gb, buf_ptr, input_size*8);
2370 
2371  switch (get_bits(&s2->gb, 4)) {
2372  case 0x1:
2373  if (last_code == 0) {
2375  } else {
2376  av_log(avctx, AV_LOG_ERROR, "ignoring seq ext after %X\n", last_code);
2377  if (avctx->err_recognition & AV_EF_EXPLODE)
2378  return AVERROR_INVALIDDATA;
2379  }
2380  break;
2381  case 0x2:
2383  break;
2384  case 0x3:
2386  break;
2387  case 0x7:
2389  break;
2390  case 0x8:
2391  if (last_code == PICTURE_START_CODE) {
2393  } else {
2394  av_log(avctx, AV_LOG_ERROR, "ignoring pic cod ext after %X\n", last_code);
2395  if (avctx->err_recognition & AV_EF_EXPLODE)
2396  return AVERROR_INVALIDDATA;
2397  }
2398  break;
2399  }
2400  break;
2401  case USER_START_CODE:
2402  mpeg_decode_user_data(avctx, buf_ptr, input_size);
2403  break;
2404  case GOP_START_CODE:
2405  if (last_code == 0) {
2406  s2->first_field=0;
2407  mpeg_decode_gop(avctx, buf_ptr, input_size);
2408  s->sync=1;
2409  } else {
2410  av_log(avctx, AV_LOG_ERROR, "ignoring GOP_START_CODE after %X\n", last_code);
2411  if (avctx->err_recognition & AV_EF_EXPLODE)
2412  return AVERROR_INVALIDDATA;
2413  }
2414  break;
2415  default:
2416  if (start_code >= SLICE_MIN_START_CODE &&
2417  start_code <= SLICE_MAX_START_CODE && last_code == PICTURE_START_CODE) {
2418 
2419  if (s2->progressive_sequence && !s2->progressive_frame) {
2420  s2->progressive_frame = 1;
2421  av_log(s2->avctx, AV_LOG_ERROR, "interlaced frame in progressive sequence, ignoring\n");
2422  }
2423 
2424  if (s2->picture_structure == 0 || (s2->progressive_frame && s2->picture_structure != PICT_FRAME)) {
2425  av_log(s2->avctx, AV_LOG_ERROR, "picture_structure %d invalid, ignoring\n", s2->picture_structure);
2427  }
2428 
2429  if (s2->progressive_sequence && !s2->frame_pred_frame_dct) {
2430  av_log(s2->avctx, AV_LOG_WARNING, "invalid frame_pred_frame_dct\n");
2431  }
2432 
2433  if (s2->picture_structure == PICT_FRAME) {
2434  s2->first_field = 0;
2435  s2->v_edge_pos = 16 * s2->mb_height;
2436  } else {
2437  s2->first_field ^= 1;
2438  s2->v_edge_pos = 8 * s2->mb_height;
2439  memset(s2->mbskip_table, 0, s2->mb_stride * s2->mb_height);
2440  }
2441  }
2442  if (start_code >= SLICE_MIN_START_CODE &&
2443  start_code <= SLICE_MAX_START_CODE && last_code != 0) {
2444  const int field_pic = s2->picture_structure != PICT_FRAME;
2445  int mb_y = start_code - SLICE_MIN_START_CODE;
2446  last_code = SLICE_MIN_START_CODE;
2447  if(s2->codec_id != AV_CODEC_ID_MPEG1VIDEO && s2->mb_height > 2800/16)
2448  mb_y += (*buf_ptr&0xE0)<<2;
2449 
2450  mb_y <<= field_pic;
2452  mb_y++;
2453 
2454  if (mb_y >= s2->mb_height) {
2455  av_log(s2->avctx, AV_LOG_ERROR, "slice below image (%d >= %d)\n", mb_y, s2->mb_height);
2456  return -1;
2457  }
2458 
2459  if (s2->last_picture_ptr == NULL) {
2460  /* Skip B-frames if we do not have reference frames and gop is not closed */
2461  if (s2->pict_type == AV_PICTURE_TYPE_B) {
2462  if (!s2->closed_gop)
2463  break;
2464  }
2465  }
2467  s->sync=1;
2468  if (s2->next_picture_ptr == NULL) {
2469  /* Skip P-frames if we do not have a reference frame or we have an invalid header. */
2470  if (s2->pict_type == AV_PICTURE_TYPE_P && !s->sync) break;
2471  }
2472  if ((avctx->skip_frame >= AVDISCARD_NONREF && s2->pict_type == AV_PICTURE_TYPE_B) ||
2473  (avctx->skip_frame >= AVDISCARD_NONKEY && s2->pict_type != AV_PICTURE_TYPE_I) ||
2474  avctx->skip_frame >= AVDISCARD_ALL)
2475  break;
2476 
2477  if (!s->mpeg_enc_ctx_allocated)
2478  break;
2479 
2480  if (s2->codec_id == AV_CODEC_ID_MPEG2VIDEO) {
2481  if (mb_y < avctx->skip_top || mb_y >= s2->mb_height - avctx->skip_bottom)
2482  break;
2483  }
2484 
2485  if (!s2->pict_type) {
2486  av_log(avctx, AV_LOG_ERROR, "Missing picture start code\n");
2487  if (avctx->err_recognition & AV_EF_EXPLODE)
2488  return AVERROR_INVALIDDATA;
2489  break;
2490  }
2491 
2492  if (s2->first_slice) {
2493  s2->first_slice = 0;
2494  if (mpeg_field_start(s2, buf, buf_size) < 0)
2495  return -1;
2496  }
2497  if (!s2->current_picture_ptr) {
2498  av_log(avctx, AV_LOG_ERROR, "current_picture not initialized\n");
2499  return AVERROR_INVALIDDATA;
2500  }
2501 
2502  if (uses_vdpau(avctx)) {
2503  s->slice_count++;
2504  break;
2505  }
2506 
2507  if (HAVE_THREADS && (avctx->active_thread_type & FF_THREAD_SLICE)) {
2508  int threshold = (s2->mb_height * s->slice_count +
2509  s2->slice_context_count / 2) /
2510  s2->slice_context_count;
2511  av_assert0(avctx->thread_count > 1);
2512  if (threshold <= mb_y) {
2513  MpegEncContext *thread_context = s2->thread_context[s->slice_count];
2514 
2515  thread_context->start_mb_y = mb_y;
2516  thread_context->end_mb_y = s2->mb_height;
2517  if (s->slice_count) {
2518  s2->thread_context[s->slice_count-1]->end_mb_y = mb_y;
2519  ret = ff_update_duplicate_context(thread_context,
2520  s2);
2521  if (ret < 0)
2522  return ret;
2523  }
2524  init_get_bits(&thread_context->gb, buf_ptr, input_size*8);
2525  s->slice_count++;
2526  }
2527  buf_ptr += 2; // FIXME add minimum number of bytes per slice
2528  } else {
2529  ret = mpeg_decode_slice(s2, mb_y, &buf_ptr, input_size);
2530  emms_c();
2531 
2532  if (ret < 0) {
2533  if (avctx->err_recognition & AV_EF_EXPLODE)
2534  return ret;
2535  if (s2->resync_mb_x >= 0 && s2->resync_mb_y >= 0)
2537  } else {
2538  ff_er_add_slice(&s2->er, s2->resync_mb_x, s2->resync_mb_y, s2->mb_x-1, s2->mb_y, ER_AC_END | ER_DC_END | ER_MV_END);
2539  }
2540  }
2541  }
2542  break;
2543  }
2544  }
2545 }
2546 
2548  void *data, int *got_output,
2549  AVPacket *avpkt)
2550 {
2551  const uint8_t *buf = avpkt->data;
2552  int ret;
2553  int buf_size = avpkt->size;
2554  Mpeg1Context *s = avctx->priv_data;
2555  AVFrame *picture = data;
2557  av_dlog(avctx, "fill_buffer\n");
2558 
2559  if (buf_size == 0 || (buf_size == 4 && AV_RB32(buf) == SEQ_END_CODE)) {
2560  /* special case for last picture */
2561  if (s2->low_delay == 0 && s2->next_picture_ptr) {
2562  *picture = s2->next_picture_ptr->f;
2563  s2->next_picture_ptr = NULL;
2564 
2565  *got_output = 1;
2566  }
2567  return buf_size;
2568  }
2569 
2570  if (s2->flags & CODEC_FLAG_TRUNCATED) {
2571  int next = ff_mpeg1_find_frame_end(&s2->parse_context, buf, buf_size, NULL);
2572 
2573  if (ff_combine_frame(&s2->parse_context, next, (const uint8_t **)&buf, &buf_size) < 0)
2574  return buf_size;
2575  }
2576 
2577  s2->codec_tag = avpriv_toupper4(avctx->codec_tag);
2578  if (s->mpeg_enc_ctx_allocated == 0 && ( s2->codec_tag == AV_RL32("VCR2")
2579  || s2->codec_tag == AV_RL32("BW10")
2580  ))
2581  vcr2_init_sequence(avctx);
2582 
2583  s->slice_count = 0;
2584 
2585  if (avctx->extradata && !s->extradata_decoded) {
2586  ret = decode_chunks(avctx, picture, got_output, avctx->extradata, avctx->extradata_size);
2587  if(*got_output) {
2588  av_log(avctx, AV_LOG_ERROR, "picture in extradata\n");
2589  *got_output = 0;
2590  }
2591  s->extradata_decoded = 1;
2592  if (ret < 0 && (avctx->err_recognition & AV_EF_EXPLODE)) {
2593  s2->current_picture_ptr = NULL;
2594  return ret;
2595  }
2596  }
2597 
2598  ret = decode_chunks(avctx, picture, got_output, buf, buf_size);
2599  if (ret<0 || *got_output)
2600  s2->current_picture_ptr = NULL;
2601 
2602  return ret;
2603 }
2604 
2605 
2606 static void flush(AVCodecContext *avctx)
2607 {
2608  Mpeg1Context *s = avctx->priv_data;
2609 
2610  s->sync=0;
2611 
2612  ff_mpeg_flush(avctx);
2613 }
2614 
2616 {
2617  Mpeg1Context *s = avctx->priv_data;
2618 
2619  if (s->mpeg_enc_ctx_allocated)
2621  return 0;
2622 }
2623 
2625  { FF_PROFILE_MPEG2_422, "4:2:2" },
2626  { FF_PROFILE_MPEG2_HIGH, "High" },
2627  { FF_PROFILE_MPEG2_SS, "Spatially Scalable" },
2628  { FF_PROFILE_MPEG2_SNR_SCALABLE, "SNR Scalable" },
2629  { FF_PROFILE_MPEG2_MAIN, "Main" },
2630  { FF_PROFILE_MPEG2_SIMPLE, "Simple" },
2631  { FF_PROFILE_RESERVED, "Reserved" },
2632  { FF_PROFILE_RESERVED, "Reserved" },
2633  { FF_PROFILE_UNKNOWN },
2634 };
2635 
2636 
2638  .name = "mpeg1video",
2639  .type = AVMEDIA_TYPE_VIDEO,
2640  .id = AV_CODEC_ID_MPEG1VIDEO,
2641  .priv_data_size = sizeof(Mpeg1Context),
2645  .capabilities = CODEC_CAP_DRAW_HORIZ_BAND | CODEC_CAP_DR1 |
2648  .flush = flush,
2649  .max_lowres = 3,
2650  .long_name = NULL_IF_CONFIG_SMALL("MPEG-1 video"),
2652 };
2653 
2655  .name = "mpeg2video",
2656  .type = AVMEDIA_TYPE_VIDEO,
2657  .id = AV_CODEC_ID_MPEG2VIDEO,
2658  .priv_data_size = sizeof(Mpeg1Context),
2662  .capabilities = CODEC_CAP_DRAW_HORIZ_BAND | CODEC_CAP_DR1 |
2665  .flush = flush,
2666  .max_lowres = 3,
2667  .long_name = NULL_IF_CONFIG_SMALL("MPEG-2 video"),
2668  .profiles = NULL_IF_CONFIG_SMALL(mpeg2_video_profiles),
2669 };
2670 
2671 //legacy decoder
2673  .name = "mpegvideo",
2674  .type = AVMEDIA_TYPE_VIDEO,
2675  .id = AV_CODEC_ID_MPEG2VIDEO,
2676  .priv_data_size = sizeof(Mpeg1Context),
2681  .flush = flush,
2682  .max_lowres = 3,
2683  .long_name = NULL_IF_CONFIG_SMALL("MPEG-1 video"),
2684 };
2685 
2686 #if CONFIG_MPEG_XVMC_DECODER
2687 static av_cold int mpeg_mc_decode_init(AVCodecContext *avctx)
2688 {
2689  if (avctx->active_thread_type & FF_THREAD_SLICE)
2690  return -1;
2691  if (!(avctx->slice_flags & SLICE_FLAG_CODED_ORDER))
2692  return -1;
2693  if (!(avctx->slice_flags & SLICE_FLAG_ALLOW_FIELD)) {
2694  av_dlog(avctx, "mpeg12.c: XvMC decoder will work better if SLICE_FLAG_ALLOW_FIELD is set\n");
2695  }
2696  mpeg_decode_init(avctx);
2697 
2699  avctx->xvmc_acceleration = 2; // 2 - the blocks are packed!
2700 
2701  return 0;
2702 }
2703 
2704 AVCodec ff_mpeg_xvmc_decoder = {
2705  .name = "mpegvideo_xvmc",
2706  .type = AVMEDIA_TYPE_VIDEO,
2708  .priv_data_size = sizeof(Mpeg1Context),
2709  .init = mpeg_mc_decode_init,
2712  .capabilities = CODEC_CAP_DRAW_HORIZ_BAND | CODEC_CAP_DR1 |
2714  .flush = flush,
2715  .long_name = NULL_IF_CONFIG_SMALL("MPEG-1/2 video XvMC (X-Video Motion Compensation)"),
2716 };
2717 
2718 #endif
2719 
2720 #if CONFIG_MPEG_VDPAU_DECODER
2721 AVCodec ff_mpeg_vdpau_decoder = {
2722  .name = "mpegvideo_vdpau",
2723  .type = AVMEDIA_TYPE_VIDEO,
2724  .id = AV_CODEC_ID_MPEG2VIDEO,
2725  .priv_data_size = sizeof(Mpeg1Context),
2729  .capabilities = CODEC_CAP_DR1 | CODEC_CAP_TRUNCATED |
2731  .flush = flush,
2732  .long_name = NULL_IF_CONFIG_SMALL("MPEG-1/2 video (VDPAU acceleration)"),
2733 };
2734 #endif
2735 
2736 #if CONFIG_MPEG1_VDPAU_DECODER
2737 AVCodec ff_mpeg1_vdpau_decoder = {
2738  .name = "mpeg1video_vdpau",
2739  .type = AVMEDIA_TYPE_VIDEO,
2740  .id = AV_CODEC_ID_MPEG1VIDEO,
2741  .priv_data_size = sizeof(Mpeg1Context),
2745  .capabilities = CODEC_CAP_DR1 | CODEC_CAP_TRUNCATED |
2747  .flush = flush,
2748  .long_name = NULL_IF_CONFIG_SMALL("MPEG-1 video (VDPAU acceleration)"),
2749 };
2750 #endif