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, DCTELEM *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, DCTELEM *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  */
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 
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  DCTELEM *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, DCTELEM *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  */
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  DCTELEM (*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 
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 )
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 
1612  ff_er_frame_start(s);
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_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 < 0 || (left && show_bits(&s->gb, FFMIN(left, 23)) && !is_d10)
1825  || ((avctx->err_recognition & (AV_EF_BITSTREAM | AV_EF_AGGRESSIVE)) && left > 8)) {
1826  av_log(avctx, AV_LOG_ERROR, "end mismatch left=%d %0X\n", left, show_bits(&s->gb, FFMIN(left, 23)));
1827  return -1;
1828  } else
1829  goto eos;
1830  }
1831 
1833  }
1834 
1835  /* skip mb handling */
1836  if (s->mb_skip_run == -1) {
1837  /* read increment again */
1838  s->mb_skip_run = 0;
1839  for (;;) {
1840  int code = get_vlc2(&s->gb, mbincr_vlc.table, MBINCR_VLC_BITS, 2);
1841  if (code < 0) {
1842  av_log(s->avctx, AV_LOG_ERROR, "mb incr damaged\n");
1843  return -1;
1844  }
1845  if (code >= 33) {
1846  if (code == 33) {
1847  s->mb_skip_run += 33;
1848  } else if (code == 35) {
1849  if (s->mb_skip_run != 0 || show_bits(&s->gb, 15) != 0) {
1850  av_log(s->avctx, AV_LOG_ERROR, "slice mismatch\n");
1851  return -1;
1852  }
1853  goto eos; /* end of slice */
1854  }
1855  /* otherwise, stuffing, nothing to do */
1856  } else {
1857  s->mb_skip_run += code;
1858  break;
1859  }
1860  }
1861  if (s->mb_skip_run) {
1862  int i;
1863  if (s->pict_type == AV_PICTURE_TYPE_I) {
1864  av_log(s->avctx, AV_LOG_ERROR, "skipped MB in I frame at %d %d\n", s->mb_x, s->mb_y);
1865  return -1;
1866  }
1867 
1868  /* skip mb */
1869  s->mb_intra = 0;
1870  for (i = 0; i < 12; i++)
1871  s->block_last_index[i] = -1;
1873  s->mv_type = MV_TYPE_16X16;
1874  else
1875  s->mv_type = MV_TYPE_FIELD;
1876  if (s->pict_type == AV_PICTURE_TYPE_P) {
1877  /* if P type, zero motion vector is implied */
1878  s->mv_dir = MV_DIR_FORWARD;
1879  s->mv[0][0][0] = s->mv[0][0][1] = 0;
1880  s->last_mv[0][0][0] = s->last_mv[0][0][1] = 0;
1881  s->last_mv[0][1][0] = s->last_mv[0][1][1] = 0;
1882  s->field_select[0][0] = (s->picture_structure - 1) & 1;
1883  } else {
1884  /* if B type, reuse previous vectors and directions */
1885  s->mv[0][0][0] = s->last_mv[0][0][0];
1886  s->mv[0][0][1] = s->last_mv[0][0][1];
1887  s->mv[1][0][0] = s->last_mv[1][0][0];
1888  s->mv[1][0][1] = s->last_mv[1][0][1];
1889  }
1890  }
1891  }
1892  }
1893 eos: // end of slice
1894  *buf += (get_bits_count(&s->gb)-1)/8;
1895  av_dlog(s, "y %d %d %d %d\n", s->resync_mb_x, s->resync_mb_y, s->mb_x, s->mb_y);
1896  return 0;
1897 }
1898 
1900 {
1901  MpegEncContext *s = *(void**)arg;
1902  const uint8_t *buf = s->gb.buffer;
1903  int mb_y = s->start_mb_y;
1904  const int field_pic = s->picture_structure != PICT_FRAME;
1905 
1906  s->error_count = (3 * (s->end_mb_y - s->start_mb_y) * s->mb_width) >> field_pic;
1907 
1908  for (;;) {
1909  uint32_t start_code;
1910  int ret;
1911 
1912  ret = mpeg_decode_slice(s, mb_y, &buf, s->gb.buffer_end - buf);
1913  emms_c();
1914  av_dlog(c, "ret:%d resync:%d/%d mb:%d/%d ts:%d/%d ec:%d\n",
1915  ret, s->resync_mb_x, s->resync_mb_y, s->mb_x, s->mb_y,
1916  s->start_mb_y, s->end_mb_y, s->error_count);
1917  if (ret < 0) {
1918  if (c->err_recognition & AV_EF_EXPLODE)
1919  return ret;
1920  if (s->resync_mb_x >= 0 && s->resync_mb_y >= 0)
1922  } else {
1924  }
1925 
1926  if (s->mb_y == s->end_mb_y)
1927  return 0;
1928 
1929  start_code = -1;
1930  buf = avpriv_mpv_find_start_code(buf, s->gb.buffer_end, &start_code);
1931  mb_y= start_code - SLICE_MIN_START_CODE;
1932  if(s->codec_id != AV_CODEC_ID_MPEG1VIDEO && s->mb_height > 2800/16)
1933  mb_y += (*buf&0xE0)<<2;
1934  mb_y <<= field_pic;
1936  mb_y++;
1937  if (mb_y < 0 || mb_y >= s->end_mb_y)
1938  return -1;
1939  }
1940 }
1941 
1942 /**
1943  * Handle slice ends.
1944  * @return 1 if it seems to be the last slice
1945  */
1946 static int slice_end(AVCodecContext *avctx, AVFrame *pict)
1947 {
1948  Mpeg1Context *s1 = avctx->priv_data;
1949  MpegEncContext *s = &s1->mpeg_enc_ctx;
1950 
1952  return 0;
1953 
1954  if (s->avctx->hwaccel) {
1955  if (s->avctx->hwaccel->end_frame(s->avctx) < 0)
1956  av_log(avctx, AV_LOG_ERROR, "hardware accelerator failed to decode picture\n");
1957  }
1958 
1959  if (CONFIG_MPEG_XVMC_DECODER && s->avctx->xvmc_acceleration)
1960  ff_xvmc_field_end(s);
1961 
1962  /* end of slice reached */
1963  if (/*s->mb_y << field_pic == s->mb_height &&*/ !s->first_field && !s->first_slice) {
1964  /* end of image */
1965 
1967 
1968  ff_er_frame_end(s);
1969 
1970  ff_MPV_frame_end(s);
1971 
1972  if (s->pict_type == AV_PICTURE_TYPE_B || s->low_delay) {
1973  *pict = s->current_picture_ptr->f;
1974  ff_print_debug_info(s, pict);
1975  } else {
1976  if (avctx->active_thread_type & FF_THREAD_FRAME)
1977  s->picture_number++;
1978  /* latency of 1 frame for I- and P-frames */
1979  /* XXX: use another variable than picture_number */
1980  if (s->last_picture_ptr != NULL) {
1981  *pict = s->last_picture_ptr->f;
1982  ff_print_debug_info(s, pict);
1983  }
1984  }
1985 
1986  return 1;
1987  } else {
1988  return 0;
1989  }
1990 }
1991 
1993  const uint8_t *buf, int buf_size)
1994 {
1995  Mpeg1Context *s1 = avctx->priv_data;
1996  MpegEncContext *s = &s1->mpeg_enc_ctx;
1997  int width, height;
1998  int i, v, j;
1999 
2000  init_get_bits(&s->gb, buf, buf_size*8);
2001 
2002  width = get_bits(&s->gb, 12);
2003  height = get_bits(&s->gb, 12);
2004  s->aspect_ratio_info = get_bits(&s->gb, 4);
2005  if (s->aspect_ratio_info == 0) {
2006  av_log(avctx, AV_LOG_ERROR, "aspect ratio has forbidden 0 value\n");
2008  return -1;
2009  }
2010  s->frame_rate_index = get_bits(&s->gb, 4);
2011  if (s->frame_rate_index == 0 || s->frame_rate_index > 13)
2012  return -1;
2013  s->bit_rate = get_bits(&s->gb, 18) * 400;
2014  if (get_bits1(&s->gb) == 0) /* marker */
2015  return -1;
2016  s->width = width;
2017  s->height = height;
2018 
2019  s->avctx->rc_buffer_size = get_bits(&s->gb, 10) * 1024 * 16;
2020  skip_bits(&s->gb, 1);
2021 
2022  /* get matrix */
2023  if (get_bits1(&s->gb)) {
2025  } else {
2026  for (i = 0; i < 64; i++) {
2027  j = s->dsp.idct_permutation[i];
2029  s->intra_matrix[j] = v;
2030  s->chroma_intra_matrix[j] = v;
2031  }
2032  }
2033  if (get_bits1(&s->gb)) {
2035  } else {
2036  for (i = 0; i < 64; i++) {
2037  int j = s->dsp.idct_permutation[i];
2039  s->inter_matrix[j] = v;
2040  s->chroma_inter_matrix[j] = v;
2041  }
2042  }
2043 
2044  if (show_bits(&s->gb, 23) != 0) {
2045  av_log(s->avctx, AV_LOG_ERROR, "sequence header damaged\n");
2046  return -1;
2047  }
2048 
2049  /* we set MPEG-2 parameters so that it emulates MPEG-1 */
2050  s->progressive_sequence = 1;
2051  s->progressive_frame = 1;
2053  s->first_field = 0;
2054  s->frame_pred_frame_dct = 1;
2055  s->chroma_format = 1;
2057  s->out_format = FMT_MPEG1;
2058  s->swap_uv = 0; // AFAIK VCR2 does not have SEQ_HEADER
2059  if (s->flags & CODEC_FLAG_LOW_DELAY)
2060  s->low_delay = 1;
2061 
2062  if (s->avctx->debug & FF_DEBUG_PICT_INFO)
2063  av_log(s->avctx, AV_LOG_DEBUG, "vbv buffer: %d, bitrate:%d\n",
2064  s->avctx->rc_buffer_size, s->bit_rate);
2065 
2066  return 0;
2067 }
2068 
2070 {
2071  Mpeg1Context *s1 = avctx->priv_data;
2072  MpegEncContext *s = &s1->mpeg_enc_ctx;
2073  int i, v;
2074 
2075  /* start new MPEG-1 context decoding */
2076  s->out_format = FMT_MPEG1;
2077  if (s1->mpeg_enc_ctx_allocated) {
2078  ff_MPV_common_end(s);
2079  }
2080  s->width = avctx->coded_width;
2081  s->height = avctx->coded_height;
2082  avctx->has_b_frames = 0; // true?
2083  s->low_delay = 1;
2084 
2085  avctx->pix_fmt = mpeg_get_pixelformat(avctx);
2086  avctx->hwaccel = ff_find_hwaccel(avctx->codec->id, avctx->pix_fmt);
2087 
2088  if( avctx->pix_fmt == AV_PIX_FMT_XVMC_MPEG2_IDCT || avctx->hwaccel )
2089  if (avctx->idct_algo == FF_IDCT_AUTO)
2090  avctx->idct_algo = FF_IDCT_SIMPLE;
2091 
2092  if (ff_MPV_common_init(s) < 0)
2093  return -1;
2094  s1->mpeg_enc_ctx_allocated = 1;
2095 
2096  for (i = 0; i < 64; i++) {
2097  int j = s->dsp.idct_permutation[i];
2099  s->intra_matrix[j] = v;
2100  s->chroma_intra_matrix[j] = v;
2101 
2103  s->inter_matrix[j] = v;
2104  s->chroma_inter_matrix[j] = v;
2105  }
2106 
2107  s->progressive_sequence = 1;
2108  s->progressive_frame = 1;
2110  s->first_field = 0;
2111  s->frame_pred_frame_dct = 1;
2112  s->chroma_format = 1;
2113  if (s->codec_tag == AV_RL32("BW10")) {
2115  } else {
2116  exchange_uv(s); // common init reset pblocks, so we swap them here
2117  s->swap_uv = 1; // in case of xvmc we need to swap uv for each MB
2119  }
2120  s1->save_width = s->width;
2121  s1->save_height = s->height;
2123  return 0;
2124 }
2125 
2126 
2128  const uint8_t *p, int buf_size)
2129 {
2130  Mpeg1Context *s = avctx->priv_data;
2131  const uint8_t *buf_end = p + buf_size;
2132 
2133  if(buf_size > 29){
2134  int i;
2135  for(i=0; i<20; i++)
2136  if(!memcmp(p+i, "\0TMPGEXS\0", 9)){
2137  s->tmpgexs= 1;
2138  }
2139 
2140 /* for(i=0; !(!p[i-2] && !p[i-1] && p[i]==1) && i<buf_size; i++){
2141  av_log(avctx, AV_LOG_ERROR, "%c", p[i]);
2142  }
2143  av_log(avctx, AV_LOG_ERROR, "\n");*/
2144  }
2145 
2146  /* we parse the DTG active format information */
2147  if (buf_end - p >= 5 &&
2148  p[0] == 'D' && p[1] == 'T' && p[2] == 'G' && p[3] == '1') {
2149  int flags = p[4];
2150  p += 5;
2151  if (flags & 0x80) {
2152  /* skip event id */
2153  p += 2;
2154  }
2155  if (flags & 0x40) {
2156  if (buf_end - p < 1)
2157  return;
2158  avctx->dtg_active_format = p[0] & 0x0f;
2159  }
2160  }
2161 }
2162 
2163 static void mpeg_decode_gop(AVCodecContext *avctx,
2164  const uint8_t *buf, int buf_size)
2165 {
2166  Mpeg1Context *s1 = avctx->priv_data;
2167  MpegEncContext *s = &s1->mpeg_enc_ctx;
2168  int broken_link;
2169  int64_t tc;
2170 
2171  init_get_bits(&s->gb, buf, buf_size*8);
2172 
2173  tc = avctx->timecode_frame_start = get_bits(&s->gb, 25);
2174 
2175  s->closed_gop = get_bits1(&s->gb);
2176  /*broken_link indicate that after editing the
2177  reference frames of the first B-Frames after GOP I-Frame
2178  are missing (open gop)*/
2179  broken_link = get_bits1(&s->gb);
2180 
2181  if (s->avctx->debug & FF_DEBUG_PICT_INFO) {
2182  char tcbuf[AV_TIMECODE_STR_SIZE];
2185  "GOP (%s) closed_gop=%d broken_link=%d\n",
2186  tcbuf, s->closed_gop, broken_link);
2187  }
2188 }
2189 /**
2190  * Find the end of the current frame in the bitstream.
2191  * @return the position of the first byte of the next frame, or -1
2192  */
2194 {
2195  int i;
2196  uint32_t state = pc->state;
2197 
2198  /* EOF considered as end of frame */
2199  if (buf_size == 0)
2200  return 0;
2201 
2202 /*
2203  0 frame start -> 1/4
2204  1 first_SEQEXT -> 0/2
2205  2 first field start -> 3/0
2206  3 second_SEQEXT -> 2/0
2207  4 searching end
2208 */
2209 
2210  for (i = 0; i < buf_size; i++) {
2211  av_assert1(pc->frame_start_found >= 0 && pc->frame_start_found <= 4);
2212  if (pc->frame_start_found & 1) {
2213  if (state == EXT_START_CODE && (buf[i] & 0xF0) != 0x80)
2214  pc->frame_start_found--;
2215  else if (state == EXT_START_CODE + 2) {
2216  if ((buf[i] & 3) == 3)
2217  pc->frame_start_found = 0;
2218  else
2219  pc->frame_start_found = (pc->frame_start_found + 1) & 3;
2220  }
2221  state++;
2222  } else {
2223  i = avpriv_mpv_find_start_code(buf + i, buf + buf_size, &state) - buf - 1;
2224  if (pc->frame_start_found == 0 && state >= SLICE_MIN_START_CODE && state <= SLICE_MAX_START_CODE) {
2225  i++;
2226  pc->frame_start_found = 4;
2227  }
2228  if (state == SEQ_END_CODE) {
2229  pc->frame_start_found = 0;
2230  pc->state=-1;
2231  return i+1;
2232  }
2233  if (pc->frame_start_found == 2 && state == SEQ_START_CODE)
2234  pc->frame_start_found = 0;
2235  if (pc->frame_start_found < 4 && state == EXT_START_CODE)
2236  pc->frame_start_found++;
2237  if (pc->frame_start_found == 4 && (state & 0xFFFFFF00) == 0x100) {
2238  if (state < SLICE_MIN_START_CODE || state > SLICE_MAX_START_CODE) {
2239  pc->frame_start_found = 0;
2240  pc->state = -1;
2241  return i - 3;
2242  }
2243  }
2244  if (pc->frame_start_found == 0 && s && state == PICTURE_START_CODE) {
2245  ff_fetch_timestamp(s, i - 3, 1);
2246  }
2247  }
2248  }
2249  pc->state = state;
2250  return END_NOT_FOUND;
2251 }
2252 
2253 static int decode_chunks(AVCodecContext *avctx,
2254  AVFrame *picture, int *got_output,
2255  const uint8_t *buf, int buf_size)
2256 {
2257  Mpeg1Context *s = avctx->priv_data;
2259  const uint8_t *buf_ptr = buf;
2260  const uint8_t *buf_end = buf + buf_size;
2261  int ret, input_size;
2262  int last_code = 0;
2263 
2264  for (;;) {
2265  /* find next start code */
2266  uint32_t start_code = -1;
2267  buf_ptr = avpriv_mpv_find_start_code(buf_ptr, buf_end, &start_code);
2268  if (start_code > 0x1ff) {
2269  if (s2->pict_type != AV_PICTURE_TYPE_B || avctx->skip_frame <= AVDISCARD_DEFAULT) {
2270  if (HAVE_THREADS && (avctx->active_thread_type & FF_THREAD_SLICE)) {
2271  int i;
2272  av_assert0(avctx->thread_count > 1);
2273 
2274  avctx->execute(avctx, slice_decode_thread, &s2->thread_context[0], NULL, s->slice_count, sizeof(void*));
2275  for (i = 0; i < s->slice_count; i++)
2276  s2->error_count += s2->thread_context[i]->error_count;
2277  }
2278 
2279  if (CONFIG_VDPAU && uses_vdpau(avctx))
2280  ff_vdpau_mpeg_picture_complete(s2, buf, buf_size, s->slice_count);
2281 
2282 
2283  if (slice_end(avctx, picture)) {
2284  if (s2->last_picture_ptr || s2->low_delay) //FIXME merge with the stuff in mpeg_decode_slice
2285  *got_output = 1;
2286  }
2287  }
2288  s2->pict_type = 0;
2289  return FFMAX(0, buf_ptr - buf - s2->parse_context.last_index);
2290  }
2291 
2292  input_size = buf_end - buf_ptr;
2293 
2294  if (avctx->debug & FF_DEBUG_STARTCODE) {
2295  av_log(avctx, AV_LOG_DEBUG, "%3X at %td left %d\n", start_code, buf_ptr-buf, input_size);
2296  }
2297 
2298  /* prepare data for next start code */
2299  switch (start_code) {
2300  case SEQ_START_CODE:
2301  if (last_code == 0) {
2302  mpeg1_decode_sequence(avctx, buf_ptr, input_size);
2303  if(buf != avctx->extradata)
2304  s->sync=1;
2305  } else {
2306  av_log(avctx, AV_LOG_ERROR, "ignoring SEQ_START_CODE after %X\n", last_code);
2307  if (avctx->err_recognition & AV_EF_EXPLODE)
2308  return AVERROR_INVALIDDATA;
2309  }
2310  break;
2311 
2312  case PICTURE_START_CODE:
2313  if (s2->width <= 0 || s2->height <= 0) {
2314  av_log(avctx, AV_LOG_ERROR, "%dx%d is invalid\n", s2->width, s2->height);
2315  return AVERROR_INVALIDDATA;
2316  }
2317 
2318  if(s->tmpgexs){
2319  s2->intra_dc_precision= 3;
2320  s2->intra_matrix[0]= 1;
2321  }
2322  if (HAVE_THREADS && (avctx->active_thread_type & FF_THREAD_SLICE) && s->slice_count) {
2323  int i;
2324 
2325  avctx->execute(avctx, slice_decode_thread,
2326  s2->thread_context, NULL,
2327  s->slice_count, sizeof(void*));
2328  for (i = 0; i < s->slice_count; i++)
2329  s2->error_count += s2->thread_context[i]->error_count;
2330  s->slice_count = 0;
2331  }
2332  if (last_code == 0 || last_code == SLICE_MIN_START_CODE) {
2333  ret = mpeg_decode_postinit(avctx);
2334  if (ret < 0) {
2335  av_log(avctx, AV_LOG_ERROR, "mpeg_decode_postinit() failure\n");
2336  return ret;
2337  }
2338 
2339  /* we have a complete image: we try to decompress it */
2340  if (mpeg1_decode_picture(avctx, buf_ptr, input_size) < 0)
2341  s2->pict_type = 0;
2342  s2->first_slice = 1;
2343  last_code = PICTURE_START_CODE;
2344  } else {
2345  av_log(avctx, AV_LOG_ERROR, "ignoring pic after %X\n", last_code);
2346  if (avctx->err_recognition & AV_EF_EXPLODE)
2347  return AVERROR_INVALIDDATA;
2348  }
2349  break;
2350  case EXT_START_CODE:
2351  init_get_bits(&s2->gb, buf_ptr, input_size*8);
2352 
2353  switch (get_bits(&s2->gb, 4)) {
2354  case 0x1:
2355  if (last_code == 0) {
2357  } else {
2358  av_log(avctx, AV_LOG_ERROR, "ignoring seq ext after %X\n", last_code);
2359  if (avctx->err_recognition & AV_EF_EXPLODE)
2360  return AVERROR_INVALIDDATA;
2361  }
2362  break;
2363  case 0x2:
2365  break;
2366  case 0x3:
2368  break;
2369  case 0x7:
2371  break;
2372  case 0x8:
2373  if (last_code == PICTURE_START_CODE) {
2375  } else {
2376  av_log(avctx, AV_LOG_ERROR, "ignoring pic cod ext after %X\n", last_code);
2377  if (avctx->err_recognition & AV_EF_EXPLODE)
2378  return AVERROR_INVALIDDATA;
2379  }
2380  break;
2381  }
2382  break;
2383  case USER_START_CODE:
2384  mpeg_decode_user_data(avctx, buf_ptr, input_size);
2385  break;
2386  case GOP_START_CODE:
2387  if (last_code == 0) {
2388  s2->first_field=0;
2389  mpeg_decode_gop(avctx, buf_ptr, input_size);
2390  s->sync=1;
2391  } else {
2392  av_log(avctx, AV_LOG_ERROR, "ignoring GOP_START_CODE after %X\n", last_code);
2393  if (avctx->err_recognition & AV_EF_EXPLODE)
2394  return AVERROR_INVALIDDATA;
2395  }
2396  break;
2397  default:
2398  if (start_code >= SLICE_MIN_START_CODE &&
2399  start_code <= SLICE_MAX_START_CODE && last_code == PICTURE_START_CODE) {
2400 
2401  if (s2->progressive_sequence && !s2->progressive_frame) {
2402  s2->progressive_frame = 1;
2403  av_log(s2->avctx, AV_LOG_ERROR, "interlaced frame in progressive sequence, ignoring\n");
2404  }
2405 
2406  if (s2->picture_structure == 0 || (s2->progressive_frame && s2->picture_structure != PICT_FRAME)) {
2407  av_log(s2->avctx, AV_LOG_ERROR, "picture_structure %d invalid, ignoring\n", s2->picture_structure);
2409  }
2410 
2411  if (s2->progressive_sequence && !s2->frame_pred_frame_dct) {
2412  av_log(s2->avctx, AV_LOG_WARNING, "invalid frame_pred_frame_dct\n");
2413  }
2414 
2415  if (s2->picture_structure == PICT_FRAME) {
2416  s2->first_field = 0;
2417  s2->v_edge_pos = 16 * s2->mb_height;
2418  } else {
2419  s2->first_field ^= 1;
2420  s2->v_edge_pos = 8 * s2->mb_height;
2421  memset(s2->mbskip_table, 0, s2->mb_stride * s2->mb_height);
2422  }
2423  }
2424  if (start_code >= SLICE_MIN_START_CODE &&
2425  start_code <= SLICE_MAX_START_CODE && last_code != 0) {
2426  const int field_pic = s2->picture_structure != PICT_FRAME;
2427  int mb_y = start_code - SLICE_MIN_START_CODE;
2428  last_code = SLICE_MIN_START_CODE;
2429  if(s2->codec_id != AV_CODEC_ID_MPEG1VIDEO && s2->mb_height > 2800/16)
2430  mb_y += (*buf_ptr&0xE0)<<2;
2431 
2432  mb_y <<= field_pic;
2434  mb_y++;
2435 
2436  if (mb_y >= s2->mb_height) {
2437  av_log(s2->avctx, AV_LOG_ERROR, "slice below image (%d >= %d)\n", mb_y, s2->mb_height);
2438  return -1;
2439  }
2440 
2441  if (s2->last_picture_ptr == NULL) {
2442  /* Skip B-frames if we do not have reference frames and gop is not closed */
2443  if (s2->pict_type == AV_PICTURE_TYPE_B) {
2444  if (!s2->closed_gop)
2445  break;
2446  }
2447  }
2449  s->sync=1;
2450  if (s2->next_picture_ptr == NULL) {
2451  /* Skip P-frames if we do not have a reference frame or we have an invalid header. */
2452  if (s2->pict_type == AV_PICTURE_TYPE_P && !s->sync) break;
2453  }
2454  if ((avctx->skip_frame >= AVDISCARD_NONREF && s2->pict_type == AV_PICTURE_TYPE_B) ||
2455  (avctx->skip_frame >= AVDISCARD_NONKEY && s2->pict_type != AV_PICTURE_TYPE_I) ||
2456  avctx->skip_frame >= AVDISCARD_ALL)
2457  break;
2458 
2459  if (!s->mpeg_enc_ctx_allocated)
2460  break;
2461 
2462  if (s2->codec_id == AV_CODEC_ID_MPEG2VIDEO) {
2463  if (mb_y < avctx->skip_top || mb_y >= s2->mb_height - avctx->skip_bottom)
2464  break;
2465  }
2466 
2467  if (!s2->pict_type) {
2468  av_log(avctx, AV_LOG_ERROR, "Missing picture start code\n");
2469  if (avctx->err_recognition & AV_EF_EXPLODE)
2470  return AVERROR_INVALIDDATA;
2471  break;
2472  }
2473 
2474  if (s2->first_slice) {
2475  s2->first_slice = 0;
2476  if (mpeg_field_start(s2, buf, buf_size) < 0)
2477  return -1;
2478  }
2479  if (!s2->current_picture_ptr) {
2480  av_log(avctx, AV_LOG_ERROR, "current_picture not initialized\n");
2481  return AVERROR_INVALIDDATA;
2482  }
2483 
2484  if (uses_vdpau(avctx)) {
2485  s->slice_count++;
2486  break;
2487  }
2488 
2489  if (HAVE_THREADS && (avctx->active_thread_type & FF_THREAD_SLICE)) {
2490  int threshold = (s2->mb_height * s->slice_count +
2491  s2->slice_context_count / 2) /
2492  s2->slice_context_count;
2493  av_assert0(avctx->thread_count > 1);
2494  if (threshold <= mb_y) {
2495  MpegEncContext *thread_context = s2->thread_context[s->slice_count];
2496 
2497  thread_context->start_mb_y = mb_y;
2498  thread_context->end_mb_y = s2->mb_height;
2499  if (s->slice_count) {
2500  s2->thread_context[s->slice_count-1]->end_mb_y = mb_y;
2501  ret = ff_update_duplicate_context(thread_context,
2502  s2);
2503  if (ret < 0)
2504  return ret;
2505  }
2506  init_get_bits(&thread_context->gb, buf_ptr, input_size*8);
2507  s->slice_count++;
2508  }
2509  buf_ptr += 2; // FIXME add minimum number of bytes per slice
2510  } else {
2511  ret = mpeg_decode_slice(s2, mb_y, &buf_ptr, input_size);
2512  emms_c();
2513 
2514  if (ret < 0) {
2515  if (avctx->err_recognition & AV_EF_EXPLODE)
2516  return ret;
2517  if (s2->resync_mb_x >= 0 && s2->resync_mb_y >= 0)
2519  } else {
2521  }
2522  }
2523  }
2524  break;
2525  }
2526  }
2527 }
2528 
2530  void *data, int *got_output,
2531  AVPacket *avpkt)
2532 {
2533  const uint8_t *buf = avpkt->data;
2534  int ret;
2535  int buf_size = avpkt->size;
2536  Mpeg1Context *s = avctx->priv_data;
2537  AVFrame *picture = data;
2539  av_dlog(avctx, "fill_buffer\n");
2540 
2541  if (buf_size == 0 || (buf_size == 4 && AV_RB32(buf) == SEQ_END_CODE)) {
2542  /* special case for last picture */
2543  if (s2->low_delay == 0 && s2->next_picture_ptr) {
2544  *picture = s2->next_picture_ptr->f;
2545  s2->next_picture_ptr = NULL;
2546 
2547  *got_output = 1;
2548  }
2549  return buf_size;
2550  }
2551 
2552  if (s2->flags & CODEC_FLAG_TRUNCATED) {
2553  int next = ff_mpeg1_find_frame_end(&s2->parse_context, buf, buf_size, NULL);
2554 
2555  if (ff_combine_frame(&s2->parse_context, next, (const uint8_t **)&buf, &buf_size) < 0)
2556  return buf_size;
2557  }
2558 
2559  s2->codec_tag = avpriv_toupper4(avctx->codec_tag);
2560  if (s->mpeg_enc_ctx_allocated == 0 && ( s2->codec_tag == AV_RL32("VCR2")
2561  || s2->codec_tag == AV_RL32("BW10")
2562  ))
2563  vcr2_init_sequence(avctx);
2564 
2565  s->slice_count = 0;
2566 
2567  if (avctx->extradata && !s->extradata_decoded) {
2568  ret = decode_chunks(avctx, picture, got_output, avctx->extradata, avctx->extradata_size);
2569  if(*got_output) {
2570  av_log(avctx, AV_LOG_ERROR, "picture in extradata\n");
2571  *got_output = 0;
2572  }
2573  s->extradata_decoded = 1;
2574  if (ret < 0 && (avctx->err_recognition & AV_EF_EXPLODE)) {
2575  s2->current_picture_ptr = NULL;
2576  return ret;
2577  }
2578  }
2579 
2580  ret = decode_chunks(avctx, picture, got_output, buf, buf_size);
2581  if (ret<0 || *got_output)
2582  s2->current_picture_ptr = NULL;
2583 
2584  return ret;
2585 }
2586 
2587 
2588 static void flush(AVCodecContext *avctx)
2589 {
2590  Mpeg1Context *s = avctx->priv_data;
2591 
2592  s->sync=0;
2593 
2594  ff_mpeg_flush(avctx);
2595 }
2596 
2598 {
2599  Mpeg1Context *s = avctx->priv_data;
2600 
2601  if (s->mpeg_enc_ctx_allocated)
2603  return 0;
2604 }
2605 
2607  { FF_PROFILE_MPEG2_422, "4:2:2" },
2608  { FF_PROFILE_MPEG2_HIGH, "High" },
2609  { FF_PROFILE_MPEG2_SS, "Spatially Scalable" },
2610  { FF_PROFILE_MPEG2_SNR_SCALABLE, "SNR Scalable" },
2611  { FF_PROFILE_MPEG2_MAIN, "Main" },
2612  { FF_PROFILE_MPEG2_SIMPLE, "Simple" },
2613  { FF_PROFILE_RESERVED, "Reserved" },
2614  { FF_PROFILE_RESERVED, "Reserved" },
2615  { FF_PROFILE_UNKNOWN },
2616 };
2617 
2618 
2620  .name = "mpeg1video",
2621  .type = AVMEDIA_TYPE_VIDEO,
2622  .id = AV_CODEC_ID_MPEG1VIDEO,
2623  .priv_data_size = sizeof(Mpeg1Context),
2627  .capabilities = CODEC_CAP_DRAW_HORIZ_BAND | CODEC_CAP_DR1 |
2630  .flush = flush,
2631  .max_lowres = 3,
2632  .long_name = NULL_IF_CONFIG_SMALL("MPEG-1 video"),
2634 };
2635 
2637  .name = "mpeg2video",
2638  .type = AVMEDIA_TYPE_VIDEO,
2639  .id = AV_CODEC_ID_MPEG2VIDEO,
2640  .priv_data_size = sizeof(Mpeg1Context),
2644  .capabilities = CODEC_CAP_DRAW_HORIZ_BAND | CODEC_CAP_DR1 |
2647  .flush = flush,
2648  .max_lowres = 3,
2649  .long_name = NULL_IF_CONFIG_SMALL("MPEG-2 video"),
2650  .profiles = NULL_IF_CONFIG_SMALL(mpeg2_video_profiles),
2651 };
2652 
2653 //legacy decoder
2655  .name = "mpegvideo",
2656  .type = AVMEDIA_TYPE_VIDEO,
2657  .id = AV_CODEC_ID_MPEG2VIDEO,
2658  .priv_data_size = sizeof(Mpeg1Context),
2663  .flush = flush,
2664  .max_lowres = 3,
2665  .long_name = NULL_IF_CONFIG_SMALL("MPEG-1 video"),
2666 };
2667 
2668 #if CONFIG_MPEG_XVMC_DECODER
2669 static av_cold int mpeg_mc_decode_init(AVCodecContext *avctx)
2670 {
2671  if (avctx->active_thread_type & FF_THREAD_SLICE)
2672  return -1;
2673  if (!(avctx->slice_flags & SLICE_FLAG_CODED_ORDER))
2674  return -1;
2675  if (!(avctx->slice_flags & SLICE_FLAG_ALLOW_FIELD)) {
2676  av_dlog(avctx, "mpeg12.c: XvMC decoder will work better if SLICE_FLAG_ALLOW_FIELD is set\n");
2677  }
2678  mpeg_decode_init(avctx);
2679 
2681  avctx->xvmc_acceleration = 2; // 2 - the blocks are packed!
2682 
2683  return 0;
2684 }
2685 
2686 AVCodec ff_mpeg_xvmc_decoder = {
2687  .name = "mpegvideo_xvmc",
2688  .type = AVMEDIA_TYPE_VIDEO,
2690  .priv_data_size = sizeof(Mpeg1Context),
2691  .init = mpeg_mc_decode_init,
2694  .capabilities = CODEC_CAP_DRAW_HORIZ_BAND | CODEC_CAP_DR1 |
2696  .flush = flush,
2697  .long_name = NULL_IF_CONFIG_SMALL("MPEG-1/2 video XvMC (X-Video Motion Compensation)"),
2698 };
2699 
2700 #endif
2701 
2702 #if CONFIG_MPEG_VDPAU_DECODER
2703 AVCodec ff_mpeg_vdpau_decoder = {
2704  .name = "mpegvideo_vdpau",
2705  .type = AVMEDIA_TYPE_VIDEO,
2706  .id = AV_CODEC_ID_MPEG2VIDEO,
2707  .priv_data_size = sizeof(Mpeg1Context),
2711  .capabilities = CODEC_CAP_DR1 | CODEC_CAP_TRUNCATED |
2713  .flush = flush,
2714  .long_name = NULL_IF_CONFIG_SMALL("MPEG-1/2 video (VDPAU acceleration)"),
2715 };
2716 #endif
2717 
2718 #if CONFIG_MPEG1_VDPAU_DECODER
2719 AVCodec ff_mpeg1_vdpau_decoder = {
2720  .name = "mpeg1video_vdpau",
2721  .type = AVMEDIA_TYPE_VIDEO,
2722  .id = AV_CODEC_ID_MPEG1VIDEO,
2723  .priv_data_size = sizeof(Mpeg1Context),
2727  .capabilities = CODEC_CAP_DR1 | CODEC_CAP_TRUNCATED |
2729  .flush = flush,
2730  .long_name = NULL_IF_CONFIG_SMALL("MPEG-1 video (VDPAU acceleration)"),
2731 };
2732 #endif