FFmpeg  4.4
dvdec.c
Go to the documentation of this file.
1 /*
2  * DV decoder
3  * Copyright (c) 2002 Fabrice Bellard
4  * Copyright (c) 2004 Roman Shaposhnik
5  *
6  * 50 Mbps (DVCPRO50) support
7  * Copyright (c) 2006 Daniel Maas <dmaas@maasdigital.com>
8  *
9  * 100 Mbps (DVCPRO HD) support
10  * Initial code by Daniel Maas <dmaas@maasdigital.com> (funded by BBC R&D)
11  * Final code by Roman Shaposhnik
12  *
13  * Many thanks to Dan Dennedy <dan@dennedy.org> for providing wealth
14  * of DV technical info.
15  *
16  * This file is part of FFmpeg.
17  *
18  * FFmpeg is free software; you can redistribute it and/or
19  * modify it under the terms of the GNU Lesser General Public
20  * License as published by the Free Software Foundation; either
21  * version 2.1 of the License, or (at your option) any later version.
22  *
23  * FFmpeg is distributed in the hope that it will be useful,
24  * but WITHOUT ANY WARRANTY; without even the implied warranty of
25  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
26  * Lesser General Public License for more details.
27  *
28  * You should have received a copy of the GNU Lesser General Public
29  * License along with FFmpeg; if not, write to the Free Software
30  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
31  */
32 
33 /**
34  * @file
35  * DV decoder
36  */
37 
38 #include "libavutil/avassert.h"
39 #include "libavutil/imgutils.h"
40 #include "libavutil/internal.h"
41 #include "libavutil/mem_internal.h"
42 #include "libavutil/pixdesc.h"
43 
44 #include "avcodec.h"
45 #include "dv.h"
46 #include "dv_profile_internal.h"
47 #include "dvdata.h"
48 #include "get_bits.h"
49 #include "internal.h"
50 #include "put_bits.h"
51 #include "simple_idct.h"
52 #include "thread.h"
53 
54 typedef struct BlockInfo {
55  const uint32_t *factor_table;
57  uint8_t pos; /* position in block */
58  void (*idct_put)(uint8_t *dest, ptrdiff_t stride, int16_t *block);
62 } BlockInfo;
63 
64 static const int dv_iweight_bits = 14;
65 
66 static const uint16_t dv_iweight_88[64] = {
67  32768, 16705, 16705, 17734, 17032, 17734, 18205, 18081,
68  18081, 18205, 18725, 18562, 19195, 18562, 18725, 19266,
69  19091, 19705, 19705, 19091, 19266, 21407, 19643, 20267,
70  20228, 20267, 19643, 21407, 22725, 21826, 20853, 20806,
71  20806, 20853, 21826, 22725, 23170, 23170, 21407, 21400,
72  21407, 23170, 23170, 24598, 23786, 22018, 22018, 23786,
73  24598, 25251, 24465, 22654, 24465, 25251, 25972, 25172,
74  25172, 25972, 26722, 27969, 26722, 29692, 29692, 31521,
75 };
76 static const uint16_t dv_iweight_248[64] = {
77  32768, 16384, 16705, 16705, 17734, 17734, 17734, 17734,
78  18081, 18081, 18725, 18725, 21407, 21407, 19091, 19091,
79  19195, 19195, 18205, 18205, 18725, 18725, 19705, 19705,
80  20267, 20267, 21826, 21826, 23170, 23170, 20806, 20806,
81  20267, 20267, 19266, 19266, 21407, 21407, 20853, 20853,
82  21400, 21400, 23786, 23786, 24465, 24465, 22018, 22018,
83  23170, 23170, 22725, 22725, 24598, 24598, 24465, 24465,
84  25172, 25172, 27969, 27969, 25972, 25972, 29692, 29692
85 };
86 
87 /**
88  * The "inverse" DV100 weights are actually just the spec weights (zig-zagged).
89  */
90 static const uint16_t dv_iweight_1080_y[64] = {
91  128, 16, 16, 17, 17, 17, 18, 18,
92  18, 18, 18, 18, 19, 18, 18, 19,
93  19, 19, 19, 19, 19, 42, 38, 40,
94  40, 40, 38, 42, 44, 43, 41, 41,
95  41, 41, 43, 44, 45, 45, 42, 42,
96  42, 45, 45, 48, 46, 43, 43, 46,
97  48, 49, 48, 44, 48, 49, 101, 98,
98  98, 101, 104, 109, 104, 116, 116, 123,
99 };
100 static const uint16_t dv_iweight_1080_c[64] = {
101  128, 16, 16, 17, 17, 17, 25, 25,
102  25, 25, 26, 25, 26, 25, 26, 26,
103  26, 27, 27, 26, 26, 42, 38, 40,
104  40, 40, 38, 42, 44, 43, 41, 41,
105  41, 41, 43, 44, 91, 91, 84, 84,
106  84, 91, 91, 96, 93, 86, 86, 93,
107  96, 197, 191, 177, 191, 197, 203, 197,
108  197, 203, 209, 219, 209, 232, 232, 246,
109 };
110 static const uint16_t dv_iweight_720_y[64] = {
111  128, 16, 16, 17, 17, 17, 18, 18,
112  18, 18, 18, 18, 19, 18, 18, 19,
113  19, 19, 19, 19, 19, 42, 38, 40,
114  40, 40, 38, 42, 44, 43, 41, 41,
115  41, 41, 43, 44, 68, 68, 63, 63,
116  63, 68, 68, 96, 92, 86, 86, 92,
117  96, 98, 96, 88, 96, 98, 202, 196,
118  196, 202, 208, 218, 208, 232, 232, 246,
119 };
120 static const uint16_t dv_iweight_720_c[64] = {
121  128, 24, 24, 26, 26, 26, 36, 36,
122  36, 36, 36, 36, 38, 36, 36, 38,
123  38, 38, 38, 38, 38, 84, 76, 80,
124  80, 80, 76, 84, 88, 86, 82, 82,
125  82, 82, 86, 88, 182, 182, 168, 168,
126  168, 182, 182, 192, 186, 192, 172, 186,
127  192, 394, 382, 354, 382, 394, 406, 394,
128  394, 406, 418, 438, 418, 464, 464, 492,
129 };
130 
132 {
133  int j, i, c, s;
134  uint32_t *factor1 = &ctx->idct_factor[0],
135  *factor2 = &ctx->idct_factor[DV_PROFILE_IS_HD(d) ? 4096 : 2816];
136 
137  if (DV_PROFILE_IS_HD(d)) {
138  /* quantization quanta by QNO for DV100 */
139  static const uint8_t dv100_qstep[16] = {
140  1, /* QNO = 0 and 1 both have no quantization */
141  1,
142  2, 3, 4, 5, 6, 7, 8, 16, 18, 20, 22, 24, 28, 52
143  };
144  const uint16_t *iweight1, *iweight2;
145 
146  if (d->height == 720) {
147  iweight1 = &dv_iweight_720_y[0];
148  iweight2 = &dv_iweight_720_c[0];
149  } else {
150  iweight1 = &dv_iweight_1080_y[0];
151  iweight2 = &dv_iweight_1080_c[0];
152  }
153  for (c = 0; c < 4; c++) {
154  for (s = 0; s < 16; s++) {
155  for (i = 0; i < 64; i++) {
156  *factor1++ = (dv100_qstep[s] << (c + 9)) * iweight1[i];
157  *factor2++ = (dv100_qstep[s] << (c + 9)) * iweight2[i];
158  }
159  }
160  }
161  } else {
162  static const uint8_t dv_quant_areas[4] = { 6, 21, 43, 64 };
163  const uint16_t *iweight1 = &dv_iweight_88[0];
164  for (j = 0; j < 2; j++, iweight1 = &dv_iweight_248[0]) {
165  for (s = 0; s < 22; s++) {
166  for (i = c = 0; c < 4; c++) {
167  for (; i < dv_quant_areas[c]; i++) {
168  *factor1 = iweight1[i] << (ff_dv_quant_shifts[s][c] + 1);
169  *factor2++ = (*factor1++) << 1;
170  }
171  }
172  }
173  }
174  }
175 }
176 
178 {
179  DVVideoContext *s = avctx->priv_data;
180  int i;
181 
182  ff_idctdsp_init(&s->idsp, avctx);
183 
184  for (i = 0; i < 64; i++)
185  s->dv_zigzag[0][i] = s->idsp.idct_permutation[ff_zigzag_direct[i]];
186 
187  if (avctx->lowres){
188  for (i = 0; i < 64; i++){
189  int j = ff_dv_zigzag248_direct[i];
190  s->dv_zigzag[1][i] = s->idsp.idct_permutation[(j & 7) + (j & 8) * 4 + (j & 48) / 2];
191  }
192  }else
193  memcpy(s->dv_zigzag[1], ff_dv_zigzag248_direct, sizeof(s->dv_zigzag[1]));
194 
195  s->idct_put[0] = s->idsp.idct_put;
196  s->idct_put[1] = ff_simple_idct248_put;
197 
198  return ff_dvvideo_init(avctx);
199 }
200 
201 /* decode AC coefficients */
202 static void dv_decode_ac(GetBitContext *gb, BlockInfo *mb, int16_t *block)
203 {
204  int last_index = gb->size_in_bits;
205  const uint8_t *scan_table = mb->scan_table;
206  const uint32_t *factor_table = mb->factor_table;
207  int pos = mb->pos;
208  int partial_bit_count = mb->partial_bit_count;
209  int level, run, vlc_len, index;
210 
211  OPEN_READER_NOSIZE(re, gb);
212  UPDATE_CACHE(re, gb);
213 
214  /* if we must parse a partial VLC, we do it here */
215  if (partial_bit_count > 0) {
216  re_cache = re_cache >> partial_bit_count |
217  mb->partial_bit_buffer;
218  re_index -= partial_bit_count;
219  mb->partial_bit_count = 0;
220  }
221 
222  /* get the AC coefficients until last_index is reached */
223  for (;;) {
224  ff_dlog(NULL, "%2d: bits=%04"PRIx32" index=%u\n",
225  pos, SHOW_UBITS(re, gb, 16), re_index);
226  /* our own optimized GET_RL_VLC */
227  index = NEG_USR32(re_cache, TEX_VLC_BITS);
228  vlc_len = ff_dv_rl_vlc[index].len;
229  if (vlc_len < 0) {
230  index = NEG_USR32((unsigned) re_cache << TEX_VLC_BITS, -vlc_len) +
232  vlc_len = TEX_VLC_BITS - vlc_len;
233  }
236 
237  /* gotta check if we're still within gb boundaries */
238  if (re_index + vlc_len > last_index) {
239  /* should be < 16 bits otherwise a codeword could have been parsed */
240  mb->partial_bit_count = last_index - re_index;
241  mb->partial_bit_buffer = re_cache & ~(-1u >> mb->partial_bit_count);
242  re_index = last_index;
243  break;
244  }
245  re_index += vlc_len;
246 
247  ff_dlog(NULL, "run=%d level=%d\n", run, level);
248  pos += run;
249  if (pos >= 64)
250  break;
251 
252  level = (level * factor_table[pos] + (1 << (dv_iweight_bits - 1))) >>
254  block[scan_table[pos]] = level;
255 
256  UPDATE_CACHE(re, gb);
257  }
258  CLOSE_READER(re, gb);
259  mb->pos = pos;
260 }
261 
262 static inline void bit_copy(PutBitContext *pb, GetBitContext *gb)
263 {
264  int bits_left = get_bits_left(gb);
265  while (bits_left >= MIN_CACHE_BITS) {
267  bits_left -= MIN_CACHE_BITS;
268  }
269  if (bits_left > 0)
270  put_bits(pb, bits_left, get_bits(gb, bits_left));
271 }
272 
274 {
275  int i, j;
276 
277  for (i = 0; i < 4; i++) {
278  for (j = 0; j < 8; j++)
279  p[j] = av_clip_uint8(block[j]);
280  block += 8;
281  p += stride;
282  }
283 }
284 
286  int stride, int16_t *blocks)
287 {
288  s->idsp.idct(blocks + 0*64);
289  s->idsp.idct(blocks + 1*64);
290 
291  put_block_8x4(blocks+0*64, data, stride<<1);
292  put_block_8x4(blocks+0*64 + 4*8, data + 8, stride<<1);
293  put_block_8x4(blocks+1*64, data + stride, stride<<1);
294  put_block_8x4(blocks+1*64 + 4*8, data + 8 + stride, stride<<1);
295 }
296 
298  int stride, int16_t *blocks)
299 {
300  s->idsp.idct(blocks + 0*64);
301  s->idsp.idct(blocks + 1*64);
302  s->idsp.idct(blocks + 2*64);
303  s->idsp.idct(blocks + 3*64);
304 
305  put_block_8x4(blocks+0*64, data, stride<<1);
306  put_block_8x4(blocks+0*64 + 4*8, data + 16, stride<<1);
307  put_block_8x4(blocks+1*64, data + 8, stride<<1);
308  put_block_8x4(blocks+1*64 + 4*8, data + 24, stride<<1);
309  put_block_8x4(blocks+2*64, data + stride, stride<<1);
310  put_block_8x4(blocks+2*64 + 4*8, data + 16 + stride, stride<<1);
311  put_block_8x4(blocks+3*64, data + 8 + stride, stride<<1);
312  put_block_8x4(blocks+3*64 + 4*8, data + 24 + stride, stride<<1);
313 }
314 
315 /* mb_x and mb_y are in units of 8 pixels */
316 static int dv_decode_video_segment(AVCodecContext *avctx, void *arg)
317 {
318  DVVideoContext *s = avctx->priv_data;
319  DVwork_chunk *work_chunk = arg;
320  int quant, dc, dct_mode, class1, j;
321  int mb_index, mb_x, mb_y, last_index;
322  int y_stride, linesize;
323  int16_t *block, *block1;
324  int c_offset;
325  uint8_t *y_ptr;
326  const uint8_t *buf_ptr;
327  PutBitContext pb, vs_pb;
328  GetBitContext gb;
329  BlockInfo mb_data[5 * DV_MAX_BPM], *mb, *mb1;
330  LOCAL_ALIGNED_16(int16_t, sblock, [5 * DV_MAX_BPM], [64]);
331  LOCAL_ALIGNED_16(uint8_t, mb_bit_buffer, [80 + AV_INPUT_BUFFER_PADDING_SIZE]); /* allow some slack */
332  LOCAL_ALIGNED_16(uint8_t, vs_bit_buffer, [80 * 5 + AV_INPUT_BUFFER_PADDING_SIZE]); /* allow some slack */
333  const int log2_blocksize = 3-s->avctx->lowres;
334  int is_field_mode[5];
335  int vs_bit_buffer_damaged = 0;
336  int mb_bit_buffer_damaged[5] = {0};
337  int retried = 0;
338  int sta;
339 
340  av_assert1((((int) mb_bit_buffer) & 7) == 0);
341  av_assert1((((int) vs_bit_buffer) & 7) == 0);
342 
343 retry:
344 
345  memset(sblock, 0, 5 * DV_MAX_BPM * sizeof(*sblock));
346 
347  /* pass 1: read DC and AC coefficients in blocks */
348  buf_ptr = &s->buf[work_chunk->buf_offset * 80];
349  block1 = &sblock[0][0];
350  mb1 = mb_data;
351  init_put_bits(&vs_pb, vs_bit_buffer, 5 * 80);
352  for (mb_index = 0; mb_index < 5; mb_index++, mb1 += s->sys->bpm, block1 += s->sys->bpm * 64) {
353  /* skip header */
354  quant = buf_ptr[3] & 0x0f;
355  if (avctx->error_concealment) {
356  if ((buf_ptr[3] >> 4) == 0x0E)
357  vs_bit_buffer_damaged = 1;
358  if (!mb_index) {
359  sta = buf_ptr[3] >> 4;
360  } else if (sta != (buf_ptr[3] >> 4))
361  vs_bit_buffer_damaged = 1;
362  }
363  buf_ptr += 4;
364  init_put_bits(&pb, mb_bit_buffer, 80);
365  mb = mb1;
366  block = block1;
367  is_field_mode[mb_index] = 0;
368  for (j = 0; j < s->sys->bpm; j++) {
369  last_index = s->sys->block_sizes[j];
370  init_get_bits(&gb, buf_ptr, last_index);
371 
372  /* get the DC */
373  dc = get_sbits(&gb, 9);
374  dct_mode = get_bits1(&gb);
375  class1 = get_bits(&gb, 2);
376  if (DV_PROFILE_IS_HD(s->sys)) {
377  mb->idct_put = s->idct_put[0];
378  mb->scan_table = s->dv_zigzag[0];
379  mb->factor_table = &s->idct_factor[(j >= 4) * 4 * 16 * 64 +
380  class1 * 16 * 64 +
381  quant * 64];
382  is_field_mode[mb_index] |= !j && dct_mode;
383  } else {
384  mb->idct_put = s->idct_put[dct_mode && log2_blocksize == 3];
385  mb->scan_table = s->dv_zigzag[dct_mode];
386  mb->factor_table =
387  &s->idct_factor[(class1 == 3) * 2 * 22 * 64 +
388  dct_mode * 22 * 64 +
389  (quant + ff_dv_quant_offset[class1]) * 64];
390  }
391  dc = dc * 4;
392  /* convert to unsigned because 128 is not added in the
393  * standard IDCT */
394  dc += 1024;
395  block[0] = dc;
396  buf_ptr += last_index >> 3;
397  mb->pos = 0;
398  mb->partial_bit_count = 0;
399 
400  ff_dlog(avctx, "MB block: %d, %d ", mb_index, j);
401  dv_decode_ac(&gb, mb, block);
402 
403  /* write the remaining bits in a new buffer only if the
404  * block is finished */
405  if (mb->pos >= 64)
406  bit_copy(&pb, &gb);
407  if (mb->pos >= 64 && mb->pos < 127)
408  vs_bit_buffer_damaged = mb_bit_buffer_damaged[mb_index] = 1;
409 
410  block += 64;
411  mb++;
412  }
413 
414  if (mb_bit_buffer_damaged[mb_index] > 0)
415  continue;
416 
417  /* pass 2: we can do it just after */
418  ff_dlog(avctx, "***pass 2 size=%d MB#=%d\n", put_bits_count(&pb), mb_index);
419  block = block1;
420  mb = mb1;
421  init_get_bits(&gb, mb_bit_buffer, put_bits_count(&pb));
422  put_bits32(&pb, 0); // padding must be zeroed
423  flush_put_bits(&pb);
424  for (j = 0; j < s->sys->bpm; j++, block += 64, mb++) {
425  if (mb->pos < 64 && get_bits_left(&gb) > 0) {
426  dv_decode_ac(&gb, mb, block);
427  /* if still not finished, no need to parse other blocks */
428  if (mb->pos < 64)
429  break;
430  if (mb->pos < 127)
431  vs_bit_buffer_damaged = mb_bit_buffer_damaged[mb_index] = 1;
432  }
433  }
434  /* all blocks are finished, so the extra bytes can be used at
435  * the video segment level */
436  if (j >= s->sys->bpm)
437  bit_copy(&vs_pb, &gb);
438  }
439 
440  /* we need a pass over the whole video segment */
441  ff_dlog(avctx, "***pass 3 size=%d\n", put_bits_count(&vs_pb));
442  block = &sblock[0][0];
443  mb = mb_data;
444  init_get_bits(&gb, vs_bit_buffer, put_bits_count(&vs_pb));
445  put_bits32(&vs_pb, 0); // padding must be zeroed
446  flush_put_bits(&vs_pb);
447  for (mb_index = 0; mb_index < 5; mb_index++) {
448  for (j = 0; j < s->sys->bpm; j++) {
449  if (mb->pos < 64 && get_bits_left(&gb) > 0 && !vs_bit_buffer_damaged) {
450  ff_dlog(avctx, "start %d:%d\n", mb_index, j);
451  dv_decode_ac(&gb, mb, block);
452  }
453 
454  if (mb->pos >= 64 && mb->pos < 127) {
455  av_log(avctx, AV_LOG_ERROR,
456  "AC EOB marker is absent pos=%d\n", mb->pos);
457  vs_bit_buffer_damaged = 1;
458  }
459  block += 64;
460  mb++;
461  }
462  }
463  if (vs_bit_buffer_damaged && !retried) {
464  av_log(avctx, AV_LOG_ERROR, "Concealing bitstream errors\n");
465  retried = 1;
466  goto retry;
467  }
468 
469  /* compute idct and place blocks */
470  block = &sblock[0][0];
471  mb = mb_data;
472  for (mb_index = 0; mb_index < 5; mb_index++) {
473  dv_calculate_mb_xy(s, work_chunk, mb_index, &mb_x, &mb_y);
474 
475  /* idct_put'ting luminance */
476  if ((s->sys->pix_fmt == AV_PIX_FMT_YUV420P) ||
477  (s->sys->pix_fmt == AV_PIX_FMT_YUV411P && mb_x >= (704 / 8)) ||
478  (s->sys->height >= 720 && mb_y != 134)) {
479  y_stride = (s->frame->linesize[0] <<
480  ((!is_field_mode[mb_index]) * log2_blocksize));
481  } else {
482  y_stride = (2 << log2_blocksize);
483  }
484  y_ptr = s->frame->data[0] +
485  ((mb_y * s->frame->linesize[0] + mb_x) << log2_blocksize);
486  if (mb_y == 134 && is_field_mode[mb_index]) {
487  dv100_idct_put_last_row_field_luma(s, y_ptr, s->frame->linesize[0], block);
488  } else {
489  linesize = s->frame->linesize[0] << is_field_mode[mb_index];
490  mb[0].idct_put(y_ptr, linesize, block + 0 * 64);
491  if (s->sys->video_stype == 4) { /* SD 422 */
492  mb[2].idct_put(y_ptr + (1 << log2_blocksize), linesize, block + 2 * 64);
493  } else {
494  mb[1].idct_put(y_ptr + (1 << log2_blocksize), linesize, block + 1 * 64);
495  mb[2].idct_put(y_ptr + y_stride, linesize, block + 2 * 64);
496  mb[3].idct_put(y_ptr + (1 << log2_blocksize) + y_stride, linesize, block + 3 * 64);
497  }
498  }
499  mb += 4;
500  block += 4 * 64;
501 
502  /* idct_put'ting chrominance */
503  c_offset = (((mb_y >> (s->sys->pix_fmt == AV_PIX_FMT_YUV420P)) * s->frame->linesize[1] +
504  (mb_x >> ((s->sys->pix_fmt == AV_PIX_FMT_YUV411P) ? 2 : 1))) << log2_blocksize);
505  for (j = 2; j; j--) {
506  uint8_t *c_ptr = s->frame->data[j] + c_offset;
507  if (s->sys->pix_fmt == AV_PIX_FMT_YUV411P && mb_x >= (704 / 8)) {
508  uint64_t aligned_pixels[64 / 8];
509  uint8_t *pixels = (uint8_t *) aligned_pixels;
510  uint8_t *c_ptr1, *ptr1;
511  int x, y;
512  mb->idct_put(pixels, 8, block);
513  for (y = 0; y < (1 << log2_blocksize); y++, c_ptr += s->frame->linesize[j], pixels += 8) {
514  ptr1 = pixels + ((1 << (log2_blocksize))>>1);
515  c_ptr1 = c_ptr + (s->frame->linesize[j] << log2_blocksize);
516  for (x = 0; x < (1 << FFMAX(log2_blocksize - 1, 0)); x++) {
517  c_ptr[x] = pixels[x];
518  c_ptr1[x] = ptr1[x];
519  }
520  }
521  block += 64;
522  mb++;
523  } else {
524  y_stride = (mb_y == 134) ? (1 << log2_blocksize) :
525  s->frame->linesize[j] << ((!is_field_mode[mb_index]) * log2_blocksize);
526  if (mb_y == 134 && is_field_mode[mb_index]) {
527  dv100_idct_put_last_row_field_chroma(s, c_ptr, s->frame->linesize[j], block);
528  mb += 2;
529  block += 2*64;
530  } else {
531  linesize = s->frame->linesize[j] << is_field_mode[mb_index];
532  (mb++)->idct_put(c_ptr, linesize, block);
533  block += 64;
534  if (s->sys->bpm == 8) {
535  (mb++)->idct_put(c_ptr + y_stride, linesize, block);
536  block += 64;
537  }
538  }
539  }
540  }
541  }
542  return 0;
543 }
544 
545 /* NOTE: exactly one frame must be given (120000 bytes for NTSC,
546  * 144000 bytes for PAL - or twice those for 50Mbps) */
547 static int dvvideo_decode_frame(AVCodecContext *avctx, void *data,
548  int *got_frame, AVPacket *avpkt)
549 {
550  uint8_t *buf = avpkt->data;
551  int buf_size = avpkt->size;
552  DVVideoContext *s = avctx->priv_data;
553  ThreadFrame frame = { .f = data };
554  const uint8_t *vsc_pack;
555  int apt, is16_9, ret;
556  const AVDVProfile *sys;
557 
558  sys = ff_dv_frame_profile(avctx, s->sys, buf, buf_size);
559  if (!sys || buf_size < sys->frame_size) {
560  av_log(avctx, AV_LOG_ERROR, "could not find dv frame profile\n");
561  return -1; /* NOTE: we only accept several full frames */
562  }
563 
564  if (sys != s->sys) {
565  ret = ff_dv_init_dynamic_tables(s, sys);
566  if (ret < 0) {
567  av_log(avctx, AV_LOG_ERROR, "Error initializing the work tables.\n");
568  return ret;
569  }
570  dv_init_weight_tables(s, sys);
571  s->sys = sys;
572  }
573 
574  s->frame = frame.f;
575  frame.f->key_frame = 1;
577  avctx->pix_fmt = s->sys->pix_fmt;
578  avctx->framerate = av_inv_q(s->sys->time_base);
579 
580  ret = ff_set_dimensions(avctx, s->sys->width, s->sys->height);
581  if (ret < 0)
582  return ret;
583 
584  /* Determine the codec's sample_aspect ratio from the packet */
585  vsc_pack = buf + 80 * 5 + 48 + 5;
586  if (*vsc_pack == dv_video_control) {
587  apt = buf[4] & 0x07;
588  is16_9 = (vsc_pack[2] & 0x07) == 0x02 ||
589  (!apt && (vsc_pack[2] & 0x07) == 0x07);
590  ff_set_sar(avctx, s->sys->sar[is16_9]);
591  }
592 
593  if ((ret = ff_thread_get_buffer(avctx, &frame, 0)) < 0)
594  return ret;
595 
596  /* Determine the codec's field order from the packet */
597  if ( *vsc_pack == dv_video_control ) {
598  if (avctx->height == 720) {
599  frame.f->interlaced_frame = 0;
600  frame.f->top_field_first = 0;
601  } else if (avctx->height == 1080) {
602  frame.f->interlaced_frame = 1;
603  frame.f->top_field_first = (vsc_pack[3] & 0x40) == 0x40;
604  } else {
605  frame.f->interlaced_frame = (vsc_pack[3] & 0x10) == 0x10;
606  frame.f->top_field_first = !(vsc_pack[3] & 0x40);
607  }
608  }
609 
610  s->buf = buf;
611  avctx->execute(avctx, dv_decode_video_segment, s->work_chunks, NULL,
612  dv_work_pool_size(s->sys), sizeof(DVwork_chunk));
613 
614  emms_c();
615 
616  /* return image */
617  *got_frame = 1;
618 
619  return s->sys->frame_size;
620 }
621 
623  .name = "dvvideo",
624  .long_name = NULL_IF_CONFIG_SMALL("DV (Digital Video)"),
625  .type = AVMEDIA_TYPE_VIDEO,
626  .id = AV_CODEC_ID_DVVIDEO,
627  .priv_data_size = sizeof(DVVideoContext),
631  .max_lowres = 3,
632 };
#define av_always_inline
Definition: attributes.h:45
#define av_cold
Definition: attributes.h:88
uint8_t pi<< 24) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_U8, uint8_t,(*(const uint8_t *) pi - 0x80) *(1.0f/(1<< 7))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_U8, uint8_t,(*(const uint8_t *) pi - 0x80) *(1.0/(1<< 7))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S16, int16_t,(*(const int16_t *) pi >> 8)+0x80) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S16, int16_t, *(const int16_t *) pi *(1.0f/(1<< 15))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S16, int16_t, *(const int16_t *) pi *(1.0/(1<< 15))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S32, int32_t,(*(const int32_t *) pi >> 24)+0x80) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S32, int32_t, *(const int32_t *) pi *(1.0f/(1U<< 31))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S32, int32_t, *(const int32_t *) pi *(1.0/(1U<< 31))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_FLT, float, av_clip_uint8(lrintf(*(const float *) pi *(1<< 7))+0x80)) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_FLT, float, av_clip_int16(lrintf(*(const float *) pi *(1<< 15)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_FLT, float, av_clipl_int32(llrintf(*(const float *) pi *(1U<< 31)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_DBL, double, av_clip_uint8(lrint(*(const double *) pi *(1<< 7))+0x80)) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_DBL, double, av_clip_int16(lrint(*(const double *) pi *(1<< 15)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_DBL, double, av_clipl_int32(llrint(*(const double *) pi *(1U<< 31)))) #define SET_CONV_FUNC_GROUP(ofmt, ifmt) static void set_generic_function(AudioConvert *ac) { } void ff_audio_convert_free(AudioConvert **ac) { if(! *ac) return;ff_dither_free(&(*ac) ->dc);av_freep(ac);} AudioConvert *ff_audio_convert_alloc(AVAudioResampleContext *avr, enum AVSampleFormat out_fmt, enum AVSampleFormat in_fmt, int channels, int sample_rate, int apply_map) { AudioConvert *ac;int in_planar, out_planar;ac=av_mallocz(sizeof(*ac));if(!ac) return NULL;ac->avr=avr;ac->out_fmt=out_fmt;ac->in_fmt=in_fmt;ac->channels=channels;ac->apply_map=apply_map;if(avr->dither_method !=AV_RESAMPLE_DITHER_NONE &&av_get_packed_sample_fmt(out_fmt)==AV_SAMPLE_FMT_S16 &&av_get_bytes_per_sample(in_fmt) > 2) { ac->dc=ff_dither_alloc(avr, out_fmt, in_fmt, channels, sample_rate, apply_map);if(!ac->dc) { av_free(ac);return NULL;} return ac;} in_planar=ff_sample_fmt_is_planar(in_fmt, channels);out_planar=ff_sample_fmt_is_planar(out_fmt, channels);if(in_planar==out_planar) { ac->func_type=CONV_FUNC_TYPE_FLAT;ac->planes=in_planar ? ac->channels :1;} else if(in_planar) ac->func_type=CONV_FUNC_TYPE_INTERLEAVE;else ac->func_type=CONV_FUNC_TYPE_DEINTERLEAVE;set_generic_function(ac);if(ARCH_AARCH64) ff_audio_convert_init_aarch64(ac);if(ARCH_ARM) ff_audio_convert_init_arm(ac);if(ARCH_X86) ff_audio_convert_init_x86(ac);return ac;} int ff_audio_convert(AudioConvert *ac, AudioData *out, AudioData *in) { int use_generic=1;int len=in->nb_samples;int p;if(ac->dc) { av_log(ac->avr, AV_LOG_TRACE, "%d samples - audio_convert: %s to %s (dithered)\n", len, av_get_sample_fmt_name(ac->in_fmt), av_get_sample_fmt_name(ac->out_fmt));return ff_convert_dither(ac-> dc
uint8_t
simple assert() macros that are a bit more flexible than ISO C assert().
#define av_assert1(cond)
assert() equivalent, that does not lie in speed critical code.
Definition: avassert.h:53
Libavcodec external API header.
static av_cold int init(AVCodecContext *avctx)
Definition: avrndec.c:31
#define s(width, name)
Definition: cbs_vp9.c:257
#define FFMAX(a, b)
Definition: common.h:103
#define av_clip_uint8
Definition: common.h:128
#define av_restrict
Definition: config.h:11
#define NULL
Definition: coverity.c:32
static void decode(AVCodecContext *dec_ctx, AVPacket *pkt, AVFrame *frame, FILE *outfile)
Definition: decode_audio.c:71
static AVFrame * frame
const AVDVProfile * ff_dv_frame_profile(AVCodecContext *codec, const AVDVProfile *sys, const uint8_t *frame, unsigned buf_size)
Get a DV profile for the provided compressed frame.
Definition: dv_profile.c:260
const uint8_t ff_dv_quant_offset[4]
Definition: dvdata.c:70
const uint8_t ff_dv_zigzag248_direct[64]
Definition: dvdata.c:33
const uint8_t ff_dv_quant_shifts[22][4]
Definition: dvdata.c:45
static av_always_inline void put_block_8x4(int16_t *block, uint8_t *av_restrict p, int stride)
Definition: dvdec.c:273
static const uint16_t dv_iweight_1080_y[64]
The "inverse" DV100 weights are actually just the spec weights (zig-zagged).
Definition: dvdec.c:90
AVCodec ff_dvvideo_decoder
Definition: dvdec.c:622
static const uint16_t dv_iweight_720_c[64]
Definition: dvdec.c:120
static av_cold int dvvideo_decode_init(AVCodecContext *avctx)
Definition: dvdec.c:177
static const uint16_t dv_iweight_248[64]
Definition: dvdec.c:76
static void bit_copy(PutBitContext *pb, GetBitContext *gb)
Definition: dvdec.c:262
static void dv100_idct_put_last_row_field_chroma(DVVideoContext *s, uint8_t *data, int stride, int16_t *blocks)
Definition: dvdec.c:285
static const uint16_t dv_iweight_720_y[64]
Definition: dvdec.c:110
static const uint16_t dv_iweight_88[64]
Definition: dvdec.c:66
static void dv100_idct_put_last_row_field_luma(DVVideoContext *s, uint8_t *data, int stride, int16_t *blocks)
Definition: dvdec.c:297
static void dv_init_weight_tables(DVVideoContext *ctx, const AVDVProfile *d)
Definition: dvdec.c:131
static const int dv_iweight_bits
Definition: dvdec.c:64
static const uint16_t dv_iweight_1080_c[64]
Definition: dvdec.c:100
static int dv_decode_video_segment(AVCodecContext *avctx, void *arg)
Definition: dvdec.c:316
static void dv_decode_ac(GetBitContext *gb, BlockInfo *mb, int16_t *block)
Definition: dvdec.c:202
static int dvvideo_decode_frame(AVCodecContext *avctx, void *data, int *got_frame, AVPacket *avpkt)
Definition: dvdec.c:547
float re
Definition: fft.c:82
bitstream reader API header.
static int get_sbits(GetBitContext *s, int n)
Definition: get_bits.h:359
#define CLOSE_READER(name, gb)
Definition: get_bits.h:149
static int get_bits_left(GetBitContext *gb)
Definition: get_bits.h:849
#define SHOW_UBITS(name, gb, num)
Definition: get_bits.h:211
static unsigned int get_bits1(GetBitContext *s)
Definition: get_bits.h:498
#define OPEN_READER_NOSIZE(name, gb)
Definition: get_bits.h:133
#define UPDATE_CACHE(name, gb)
Definition: get_bits.h:178
#define MIN_CACHE_BITS
Definition: get_bits.h:128
static unsigned int get_bits(GetBitContext *s, int n)
Read 1-25 bits.
Definition: get_bits.h:379
static int init_get_bits(GetBitContext *s, const uint8_t *buffer, int bit_size)
Initialize GetBitContext.
Definition: get_bits.h:659
#define AV_CODEC_CAP_DR1
Codec uses get_buffer() or get_encode_buffer() for allocating buffers and supports custom allocators.
Definition: codec.h:52
#define AV_CODEC_CAP_SLICE_THREADS
Codec supports slice-based (or partition-based) multithreading.
Definition: codec.h:112
#define AV_CODEC_CAP_FRAME_THREADS
Codec supports frame-level multithreading.
Definition: codec.h:108
@ AV_CODEC_ID_DVVIDEO
Definition: codec_id.h:73
#define AV_INPUT_BUFFER_PADDING_SIZE
Required number of additionally allocated bytes at the end of the input bitstream for decoding.
Definition: avcodec.h:215
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:194
static av_always_inline AVRational av_inv_q(AVRational q)
Invert a rational.
Definition: rational.h:159
@ AVMEDIA_TYPE_VIDEO
Definition: avutil.h:201
@ AV_PICTURE_TYPE_I
Intra.
Definition: avutil.h:274
int index
Definition: gxfenc.c:89
av_cold void ff_idctdsp_init(IDCTDSPContext *c, AVCodecContext *avctx)
Definition: idctdsp.c:238
misc image utilities
int i
Definition: input.c:407
static void put_bits(Jpeg2000EncoderContext *s, int val, int n)
put n times val bit
Definition: j2kenc.c:218
static void idct_put(FourXContext *f, int x, int y)
Definition: 4xm.c:557
av_cold int ff_dvvideo_init(AVCodecContext *avctx)
Definition: dv.c:198
RL_VLC_ELEM ff_dv_rl_vlc[1664]
Definition: dv.c:52
int ff_dv_init_dynamic_tables(DVVideoContext *ctx, const AVDVProfile *d)
Definition: dv.c:175
#define TEX_VLC_BITS
Definition: dv.h:99
static void dv_calculate_mb_xy(DVVideoContext *s, DVwork_chunk *work_chunk, int m, int *mb_x, int *mb_y)
Definition: dv.h:117
@ dv_video_control
Definition: dv.h:78
#define DV_MAX_BPM
maximum number of blocks per macroblock in any DV format
Definition: dv.h:97
static int dv_work_pool_size(const AVDVProfile *d)
Definition: dv.h:107
#define DV_PROFILE_IS_HD(p)
Definition: dv.h:84
int ff_set_sar(AVCodecContext *avctx, AVRational sar)
Check that the provided sample aspect ratio is valid and set it on the codec context.
Definition: utils.c:99
int ff_set_dimensions(AVCodecContext *s, int width, int height)
Check that the provided frame dimensions are valid and set them on the codec context.
Definition: utils.c:84
const char * arg
Definition: jacosubdec.c:66
common internal API header
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification.
Definition: internal.h:117
#define emms_c()
Definition: internal.h:54
int stride
Definition: mace.c:144
const uint8_t ff_zigzag_direct[64]
Definition: mathtables.c:98
#define NEG_USR32(a, s)
Definition: mathops.h:166
#define LOCAL_ALIGNED_16(t, v,...)
Definition: mem_internal.h:130
const char data[16]
Definition: mxf.c:142
int frame_size
Definition: mxfenc.c:2206
@ AV_PIX_FMT_YUV420P
planar YUV 4:2:0, 12bpp, (1 Cr & Cb sample per 2x2 Y samples)
Definition: pixfmt.h:66
@ AV_PIX_FMT_YUV411P
planar YUV 4:1:1, 12bpp, (1 Cr & Cb sample per 4x1 Y samples)
Definition: pixfmt.h:73
FF_ENABLE_DEPRECATION_WARNINGS int ff_thread_get_buffer(AVCodecContext *avctx, ThreadFrame *f, int flags)
Wrapper around get_buffer() for frame-multithreaded codecs.
bitstream writer API
static void init_put_bits(PutBitContext *s, uint8_t *buffer, int buffer_size)
Initialize the PutBitContext s.
Definition: put_bits.h:57
static void av_unused put_bits32(PutBitContext *s, uint32_t value)
Write exactly 32 bits into a bitstream.
Definition: put_bits.h:263
static int put_bits_count(PutBitContext *s)
Definition: put_bits.h:76
static void flush_put_bits(PutBitContext *s)
Pad the end of the output stream with zeros.
Definition: put_bits.h:110
typedef void(RENAME(mix_any_func_type))
void ff_simple_idct248_put(uint8_t *dest, ptrdiff_t line_size, int16_t *block)
Definition: simple_idct.c:106
unsigned int pos
Definition: spdifenc.c:412
main external API structure.
Definition: avcodec.h:536
enum AVPixelFormat pix_fmt
Pixel format, see AV_PIX_FMT_xxx.
Definition: avcodec.h:746
int error_concealment
error concealment flags
Definition: avcodec.h:1613
AVRational framerate
Definition: avcodec.h:2071
int(* execute)(struct AVCodecContext *c, int(*func)(struct AVCodecContext *c2, void *arg), void *arg2, int *ret, int count, int size)
The codec may call this to execute several independent things.
Definition: avcodec.h:1824
int lowres
low resolution decoding, 1-> 1/2 size, 2->1/4 size
Definition: avcodec.h:1754
void * priv_data
Definition: avcodec.h:563
AVCodec.
Definition: codec.h:197
const char * name
Name of the codec implementation.
Definition: codec.h:204
int key_frame
1 -> keyframe, 0-> not
Definition: frame.h:396
int top_field_first
If the content is interlaced, is top field displayed first.
Definition: frame.h:470
int interlaced_frame
The content of the picture is interlaced.
Definition: frame.h:465
enum AVPictureType pict_type
Picture type of the frame.
Definition: frame.h:401
This structure stores compressed data.
Definition: packet.h:346
int size
Definition: packet.h:370
uint8_t * data
Definition: packet.h:369
uint8_t pos
Definition: dvdec.c:57
void(* idct_put)(uint8_t *dest, ptrdiff_t stride, int16_t *block)
Definition: dvdec.c:58
uint32_t partial_bit_buffer
Definition: dvdec.c:60
const uint32_t * factor_table
Definition: dvdec.c:55
const uint8_t * scan_table
Definition: dvdec.c:56
int shift_offset
Definition: dvdec.c:61
uint8_t partial_bit_count
Definition: dvdec.c:59
uint16_t buf_offset
Definition: dv.h:37
int size_in_bits
Definition: get_bits.h:68
int16_t level
Definition: vlc.h:33
uint8_t run
Definition: vlc.h:35
int8_t len
Definition: vlc.h:34
uint8_t run
Definition: svq3.c:205
uint8_t level
Definition: svq3.c:206
#define ff_dlog(a,...)
#define av_log(a,...)
static int16_t block[64]
Definition: dct.c:116
static int16_t block1[64]
Definition: dct.c:117
AVFormatContext * ctx
Definition: movenc.c:48
#define mb
if(ret< 0)
Definition: vf_mcdeint.c:282
const uint8_t * quant
static double c[64]