FFmpeg
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
h264_slice.c
Go to the documentation of this file.
1 /*
2  * H.26L/H.264/AVC/JVT/14496-10/... decoder
3  * Copyright (c) 2003 Michael Niedermayer <michaelni@gmx.at>
4  *
5  * This file is part of FFmpeg.
6  *
7  * FFmpeg is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * FFmpeg is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with FFmpeg; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21 
22 /**
23  * @file
24  * H.264 / AVC / MPEG-4 part10 codec.
25  * @author Michael Niedermayer <michaelni@gmx.at>
26  */
27 
28 #include "libavutil/avassert.h"
29 #include "libavutil/imgutils.h"
30 #include "libavutil/timer.h"
31 #include "internal.h"
32 #include "cabac.h"
33 #include "cabac_functions.h"
34 #include "error_resilience.h"
35 #include "avcodec.h"
36 #include "h264.h"
37 #include "h264data.h"
38 #include "h264chroma.h"
39 #include "h264_mvpred.h"
40 #include "golomb.h"
41 #include "mathops.h"
42 #include "mpegutils.h"
43 #include "rectangle.h"
44 #include "thread.h"
45 
46 static const uint8_t field_scan[16+1] = {
47  0 + 0 * 4, 0 + 1 * 4, 1 + 0 * 4, 0 + 2 * 4,
48  0 + 3 * 4, 1 + 1 * 4, 1 + 2 * 4, 1 + 3 * 4,
49  2 + 0 * 4, 2 + 1 * 4, 2 + 2 * 4, 2 + 3 * 4,
50  3 + 0 * 4, 3 + 1 * 4, 3 + 2 * 4, 3 + 3 * 4,
51 };
52 
53 static const uint8_t field_scan8x8[64+1] = {
54  0 + 0 * 8, 0 + 1 * 8, 0 + 2 * 8, 1 + 0 * 8,
55  1 + 1 * 8, 0 + 3 * 8, 0 + 4 * 8, 1 + 2 * 8,
56  2 + 0 * 8, 1 + 3 * 8, 0 + 5 * 8, 0 + 6 * 8,
57  0 + 7 * 8, 1 + 4 * 8, 2 + 1 * 8, 3 + 0 * 8,
58  2 + 2 * 8, 1 + 5 * 8, 1 + 6 * 8, 1 + 7 * 8,
59  2 + 3 * 8, 3 + 1 * 8, 4 + 0 * 8, 3 + 2 * 8,
60  2 + 4 * 8, 2 + 5 * 8, 2 + 6 * 8, 2 + 7 * 8,
61  3 + 3 * 8, 4 + 1 * 8, 5 + 0 * 8, 4 + 2 * 8,
62  3 + 4 * 8, 3 + 5 * 8, 3 + 6 * 8, 3 + 7 * 8,
63  4 + 3 * 8, 5 + 1 * 8, 6 + 0 * 8, 5 + 2 * 8,
64  4 + 4 * 8, 4 + 5 * 8, 4 + 6 * 8, 4 + 7 * 8,
65  5 + 3 * 8, 6 + 1 * 8, 6 + 2 * 8, 5 + 4 * 8,
66  5 + 5 * 8, 5 + 6 * 8, 5 + 7 * 8, 6 + 3 * 8,
67  7 + 0 * 8, 7 + 1 * 8, 6 + 4 * 8, 6 + 5 * 8,
68  6 + 6 * 8, 6 + 7 * 8, 7 + 2 * 8, 7 + 3 * 8,
69  7 + 4 * 8, 7 + 5 * 8, 7 + 6 * 8, 7 + 7 * 8,
70 };
71 
72 static const uint8_t field_scan8x8_cavlc[64+1] = {
73  0 + 0 * 8, 1 + 1 * 8, 2 + 0 * 8, 0 + 7 * 8,
74  2 + 2 * 8, 2 + 3 * 8, 2 + 4 * 8, 3 + 3 * 8,
75  3 + 4 * 8, 4 + 3 * 8, 4 + 4 * 8, 5 + 3 * 8,
76  5 + 5 * 8, 7 + 0 * 8, 6 + 6 * 8, 7 + 4 * 8,
77  0 + 1 * 8, 0 + 3 * 8, 1 + 3 * 8, 1 + 4 * 8,
78  1 + 5 * 8, 3 + 1 * 8, 2 + 5 * 8, 4 + 1 * 8,
79  3 + 5 * 8, 5 + 1 * 8, 4 + 5 * 8, 6 + 1 * 8,
80  5 + 6 * 8, 7 + 1 * 8, 6 + 7 * 8, 7 + 5 * 8,
81  0 + 2 * 8, 0 + 4 * 8, 0 + 5 * 8, 2 + 1 * 8,
82  1 + 6 * 8, 4 + 0 * 8, 2 + 6 * 8, 5 + 0 * 8,
83  3 + 6 * 8, 6 + 0 * 8, 4 + 6 * 8, 6 + 2 * 8,
84  5 + 7 * 8, 6 + 4 * 8, 7 + 2 * 8, 7 + 6 * 8,
85  1 + 0 * 8, 1 + 2 * 8, 0 + 6 * 8, 3 + 0 * 8,
86  1 + 7 * 8, 3 + 2 * 8, 2 + 7 * 8, 4 + 2 * 8,
87  3 + 7 * 8, 5 + 2 * 8, 4 + 7 * 8, 5 + 4 * 8,
88  6 + 3 * 8, 6 + 5 * 8, 7 + 3 * 8, 7 + 7 * 8,
89 };
90 
91 // zigzag_scan8x8_cavlc[i] = zigzag_scan8x8[(i/4) + 16*(i%4)]
92 static const uint8_t zigzag_scan8x8_cavlc[64+1] = {
93  0 + 0 * 8, 1 + 1 * 8, 1 + 2 * 8, 2 + 2 * 8,
94  4 + 1 * 8, 0 + 5 * 8, 3 + 3 * 8, 7 + 0 * 8,
95  3 + 4 * 8, 1 + 7 * 8, 5 + 3 * 8, 6 + 3 * 8,
96  2 + 7 * 8, 6 + 4 * 8, 5 + 6 * 8, 7 + 5 * 8,
97  1 + 0 * 8, 2 + 0 * 8, 0 + 3 * 8, 3 + 1 * 8,
98  3 + 2 * 8, 0 + 6 * 8, 4 + 2 * 8, 6 + 1 * 8,
99  2 + 5 * 8, 2 + 6 * 8, 6 + 2 * 8, 5 + 4 * 8,
100  3 + 7 * 8, 7 + 3 * 8, 4 + 7 * 8, 7 + 6 * 8,
101  0 + 1 * 8, 3 + 0 * 8, 0 + 4 * 8, 4 + 0 * 8,
102  2 + 3 * 8, 1 + 5 * 8, 5 + 1 * 8, 5 + 2 * 8,
103  1 + 6 * 8, 3 + 5 * 8, 7 + 1 * 8, 4 + 5 * 8,
104  4 + 6 * 8, 7 + 4 * 8, 5 + 7 * 8, 6 + 7 * 8,
105  0 + 2 * 8, 2 + 1 * 8, 1 + 3 * 8, 5 + 0 * 8,
106  1 + 4 * 8, 2 + 4 * 8, 6 + 0 * 8, 4 + 3 * 8,
107  0 + 7 * 8, 4 + 4 * 8, 7 + 2 * 8, 3 + 6 * 8,
108  5 + 5 * 8, 6 + 5 * 8, 6 + 6 * 8, 7 + 7 * 8,
109 };
110 
111 static void release_unused_pictures(H264Context *h, int remove_current)
112 {
113  int i;
114 
115  /* release non reference frames */
116  for (i = 0; i < H264_MAX_PICTURE_COUNT; i++) {
117  if (h->DPB[i].f->buf[0] && !h->DPB[i].reference &&
118  (remove_current || &h->DPB[i] != h->cur_pic_ptr)) {
119  ff_h264_unref_picture(h, &h->DPB[i]);
120  }
121  }
122 }
123 
124 static int alloc_scratch_buffers(H264SliceContext *sl, int linesize)
125 {
126  const H264Context *h = sl->h264;
127  int alloc_size = FFALIGN(FFABS(linesize) + 32, 32);
128 
129  av_fast_malloc(&sl->bipred_scratchpad, &sl->bipred_scratchpad_allocated, 16 * 6 * alloc_size);
130  // edge emu needs blocksize + filter length - 1
131  // (= 21x21 for H.264)
132  av_fast_malloc(&sl->edge_emu_buffer, &sl->edge_emu_buffer_allocated, alloc_size * 2 * 21);
133 
135  h->mb_width * 16 * 3 * sizeof(uint8_t) * 2);
137  h->mb_width * 16 * 3 * sizeof(uint8_t) * 2);
138 
139  if (!sl->bipred_scratchpad || !sl->edge_emu_buffer ||
140  !sl->top_borders[0] || !sl->top_borders[1]) {
143  av_freep(&sl->top_borders[0]);
144  av_freep(&sl->top_borders[1]);
145 
148  sl->top_borders_allocated[0] = 0;
149  sl->top_borders_allocated[1] = 0;
150  return AVERROR(ENOMEM);
151  }
152 
153  return 0;
154 }
155 
157 {
158  const int big_mb_num = h->mb_stride * (h->mb_height + 1) + 1;
159  const int mb_array_size = h->mb_stride * h->mb_height;
160  const int b4_stride = h->mb_width * 4 + 1;
161  const int b4_array_size = b4_stride * h->mb_height * 4;
162 
163  h->qscale_table_pool = av_buffer_pool_init(big_mb_num + h->mb_stride,
165  h->mb_type_pool = av_buffer_pool_init((big_mb_num + h->mb_stride) *
166  sizeof(uint32_t), av_buffer_allocz);
167  h->motion_val_pool = av_buffer_pool_init(2 * (b4_array_size + 4) *
168  sizeof(int16_t), av_buffer_allocz);
169  h->ref_index_pool = av_buffer_pool_init(4 * mb_array_size, av_buffer_allocz);
170 
171  if (!h->qscale_table_pool || !h->mb_type_pool || !h->motion_val_pool ||
172  !h->ref_index_pool) {
177  return AVERROR(ENOMEM);
178  }
179 
180  return 0;
181 }
182 
184 {
185  int i, ret = 0;
186 
187  av_assert0(!pic->f->data[0]);
188 
189  pic->tf.f = pic->f;
190  ret = ff_thread_get_buffer(h->avctx, &pic->tf, pic->reference ?
192  if (ret < 0)
193  goto fail;
194 
195  pic->crop = h->ps.sps->crop;
196  pic->crop_top = h->ps.sps->crop_top;
197  pic->crop_left= h->ps.sps->crop_left;
198 
199  if (h->avctx->hwaccel) {
200  const AVHWAccel *hwaccel = h->avctx->hwaccel;
202  if (hwaccel->frame_priv_data_size) {
204  if (!pic->hwaccel_priv_buf)
205  return AVERROR(ENOMEM);
207  }
208  }
209  if (CONFIG_GRAY && !h->avctx->hwaccel && h->flags & AV_CODEC_FLAG_GRAY && pic->f->data[2]) {
210  int h_chroma_shift, v_chroma_shift;
212  &h_chroma_shift, &v_chroma_shift);
213 
214  for(i=0; i<AV_CEIL_RSHIFT(pic->f->height, v_chroma_shift); i++) {
215  memset(pic->f->data[1] + pic->f->linesize[1]*i,
216  0x80, AV_CEIL_RSHIFT(pic->f->width, h_chroma_shift));
217  memset(pic->f->data[2] + pic->f->linesize[2]*i,
218  0x80, AV_CEIL_RSHIFT(pic->f->width, h_chroma_shift));
219  }
220  }
221 
222  if (!h->qscale_table_pool) {
223  ret = init_table_pools(h);
224  if (ret < 0)
225  goto fail;
226  }
227 
230  if (!pic->qscale_table_buf || !pic->mb_type_buf)
231  goto fail;
232 
233  pic->mb_type = (uint32_t*)pic->mb_type_buf->data + 2 * h->mb_stride + 1;
234  pic->qscale_table = pic->qscale_table_buf->data + 2 * h->mb_stride + 1;
235 
236  for (i = 0; i < 2; i++) {
239  if (!pic->motion_val_buf[i] || !pic->ref_index_buf[i])
240  goto fail;
241 
242  pic->motion_val[i] = (int16_t (*)[2])pic->motion_val_buf[i]->data + 4;
243  pic->ref_index[i] = pic->ref_index_buf[i]->data;
244  }
245 
246  return 0;
247 fail:
248  ff_h264_unref_picture(h, pic);
249  return (ret < 0) ? ret : AVERROR(ENOMEM);
250 }
251 
252 static inline int pic_is_unused(H264Context *h, H264Picture *pic)
253 {
254  if (!pic->f->buf[0])
255  return 1;
256  return 0;
257 }
258 
260 {
261  int i;
262 
263  for (i = 0; i < H264_MAX_PICTURE_COUNT; i++) {
264  if (pic_is_unused(h, &h->DPB[i]))
265  break;
266  }
267  if (i == H264_MAX_PICTURE_COUNT)
268  return AVERROR_INVALIDDATA;
269 
270  return i;
271 }
272 
273 
274 #define IN_RANGE(a, b, size) (((void*)(a) >= (void*)(b)) && ((void*)(a) < (void*)((b) + (size))))
275 
276 #define REBASE_PICTURE(pic, new_ctx, old_ctx) \
277  (((pic) && (pic) >= (old_ctx)->DPB && \
278  (pic) < (old_ctx)->DPB + H264_MAX_PICTURE_COUNT) ? \
279  &(new_ctx)->DPB[(pic) - (old_ctx)->DPB] : NULL)
280 
282  H264Context *new_base,
283  H264Context *old_base)
284 {
285  int i;
286 
287  for (i = 0; i < count; i++) {
288  av_assert1(!from[i] ||
289  IN_RANGE(from[i], old_base, 1) ||
290  IN_RANGE(from[i], old_base->DPB, H264_MAX_PICTURE_COUNT));
291  to[i] = REBASE_PICTURE(from[i], new_base, old_base);
292  }
293 }
294 
296 
298  const AVCodecContext *src)
299 {
300  H264Context *h = dst->priv_data, *h1 = src->priv_data;
301  int inited = h->context_initialized, err = 0;
302  int need_reinit = 0;
303  int i, ret;
304 
305  if (dst == src)
306  return 0;
307 
308  // We can't fail if SPS isn't set at it breaks current skip_frame code
309  //if (!h1->ps.sps)
310  // return AVERROR_INVALIDDATA;
311 
312  if (inited &&
313  (h->width != h1->width ||
314  h->height != h1->height ||
315  h->mb_width != h1->mb_width ||
316  h->mb_height != h1->mb_height ||
317  !h->ps.sps ||
318  h->ps.sps->bit_depth_luma != h1->ps.sps->bit_depth_luma ||
319  h->ps.sps->chroma_format_idc != h1->ps.sps->chroma_format_idc ||
320  h->ps.sps->colorspace != h1->ps.sps->colorspace)) {
321  need_reinit = 1;
322  }
323 
324  /* copy block_offset since frame_start may not be called */
325  memcpy(h->block_offset, h1->block_offset, sizeof(h->block_offset));
326 
327  // SPS/PPS
328  for (i = 0; i < FF_ARRAY_ELEMS(h->ps.sps_list); i++) {
329  av_buffer_unref(&h->ps.sps_list[i]);
330  if (h1->ps.sps_list[i]) {
331  h->ps.sps_list[i] = av_buffer_ref(h1->ps.sps_list[i]);
332  if (!h->ps.sps_list[i])
333  return AVERROR(ENOMEM);
334  }
335  }
336  for (i = 0; i < FF_ARRAY_ELEMS(h->ps.pps_list); i++) {
337  av_buffer_unref(&h->ps.pps_list[i]);
338  if (h1->ps.pps_list[i]) {
339  h->ps.pps_list[i] = av_buffer_ref(h1->ps.pps_list[i]);
340  if (!h->ps.pps_list[i])
341  return AVERROR(ENOMEM);
342  }
343  }
344 
347  h->ps.pps = NULL;
348  h->ps.sps = NULL;
349  if (h1->ps.pps_ref) {
350  h->ps.pps_ref = av_buffer_ref(h1->ps.pps_ref);
351  if (!h->ps.pps_ref)
352  return AVERROR(ENOMEM);
353  h->ps.pps = (const PPS*)h->ps.pps_ref->data;
354  }
355  if (h1->ps.sps_ref) {
356  h->ps.sps_ref = av_buffer_ref(h1->ps.sps_ref);
357  if (!h->ps.sps_ref)
358  return AVERROR(ENOMEM);
359  h->ps.sps = (SPS*)h->ps.sps_ref->data;
360  }
361 
362  if (need_reinit || !inited) {
363  h->width = h1->width;
364  h->height = h1->height;
365  h->mb_height = h1->mb_height;
366  h->mb_width = h1->mb_width;
367  h->mb_num = h1->mb_num;
368  h->mb_stride = h1->mb_stride;
369  h->b_stride = h1->b_stride;
370 
371  if (h->context_initialized || h1->context_initialized) {
372  if ((err = h264_slice_header_init(h)) < 0) {
373  av_log(h->avctx, AV_LOG_ERROR, "h264_slice_header_init() failed");
374  return err;
375  }
376  }
377  /* copy block_offset since frame_start may not be called */
378  memcpy(h->block_offset, h1->block_offset, sizeof(h->block_offset));
379  }
380 
381  h->avctx->coded_height = h1->avctx->coded_height;
382  h->avctx->coded_width = h1->avctx->coded_width;
383  h->avctx->width = h1->avctx->width;
384  h->avctx->height = h1->avctx->height;
385  h->coded_picture_number = h1->coded_picture_number;
386  h->first_field = h1->first_field;
387  h->picture_structure = h1->picture_structure;
388  h->droppable = h1->droppable;
389  h->backup_width = h1->backup_width;
390  h->backup_height = h1->backup_height;
391  h->backup_pix_fmt = h1->backup_pix_fmt;
392 
393  for (i = 0; i < H264_MAX_PICTURE_COUNT; i++) {
394  ff_h264_unref_picture(h, &h->DPB[i]);
395  if (h1->DPB[i].f->buf[0] &&
396  (ret = ff_h264_ref_picture(h, &h->DPB[i], &h1->DPB[i])) < 0)
397  return ret;
398  }
399 
400  h->cur_pic_ptr = REBASE_PICTURE(h1->cur_pic_ptr, h, h1);
402  if (h1->cur_pic.f->buf[0]) {
403  ret = ff_h264_ref_picture(h, &h->cur_pic, &h1->cur_pic);
404  if (ret < 0)
405  return ret;
406  }
407 
408  h->enable_er = h1->enable_er;
409  h->workaround_bugs = h1->workaround_bugs;
410  h->droppable = h1->droppable;
411 
412  // extradata/NAL handling
413  h->is_avc = h1->is_avc;
414  h->nal_length_size = h1->nal_length_size;
415  h->sei.unregistered.x264_build = h1->sei.unregistered.x264_build;
416 
417  memcpy(&h->poc, &h1->poc, sizeof(h->poc));
418 
419  h->curr_pic_num = h1->curr_pic_num;
420  h->max_pic_num = h1->max_pic_num;
421 
422  memcpy(h->default_ref, h1->default_ref, sizeof(h->default_ref));
423  memcpy(h->short_ref, h1->short_ref, sizeof(h->short_ref));
424  memcpy(h->long_ref, h1->long_ref, sizeof(h->long_ref));
425  memcpy(h->delayed_pic, h1->delayed_pic, sizeof(h->delayed_pic));
426  memcpy(h->last_pocs, h1->last_pocs, sizeof(h->last_pocs));
427 
428  h->next_output_pic = h1->next_output_pic;
429  h->next_outputed_poc = h1->next_outputed_poc;
430 
431  memcpy(h->mmco, h1->mmco, sizeof(h->mmco));
432  h->mmco_index = h1->mmco_index;
433  h->mmco_reset = h1->mmco_reset;
434  h->long_ref_count = h1->long_ref_count;
435  h->short_ref_count = h1->short_ref_count;
436 
437  copy_picture_range(h->short_ref, h1->short_ref, 32, h, h1);
438  copy_picture_range(h->long_ref, h1->long_ref, 32, h, h1);
439  copy_picture_range(h->delayed_pic, h1->delayed_pic,
440  MAX_DELAYED_PIC_COUNT + 2, h, h1);
441 
442  h->frame_recovered = h1->frame_recovered;
443 
444  if (!h->cur_pic_ptr)
445  return 0;
446 
447  if (!h->droppable) {
449  h->poc.prev_poc_msb = h->poc.poc_msb;
450  h->poc.prev_poc_lsb = h->poc.poc_lsb;
451  }
454 
455  h->recovery_frame = h1->recovery_frame;
456 
457  return err;
458 }
459 
461 {
462  H264Picture *pic;
463  int i, ret;
464  const int pixel_shift = h->pixel_shift;
465  int c[4] = {
466  1<<(h->ps.sps->bit_depth_luma-1),
467  1<<(h->ps.sps->bit_depth_chroma-1),
468  1<<(h->ps.sps->bit_depth_chroma-1),
469  -1
470  };
471 
472  if (!ff_thread_can_start_frame(h->avctx)) {
473  av_log(h->avctx, AV_LOG_ERROR, "Attempt to start a frame outside SETUP state\n");
474  return -1;
475  }
476 
478  h->cur_pic_ptr = NULL;
479 
480  i = find_unused_picture(h);
481  if (i < 0) {
482  av_log(h->avctx, AV_LOG_ERROR, "no frame buffer available\n");
483  return i;
484  }
485  pic = &h->DPB[i];
486 
487  pic->reference = h->droppable ? 0 : h->picture_structure;
490  pic->frame_num = h->poc.frame_num;
491  /*
492  * Zero key_frame here; IDR markings per slice in frame or fields are ORed
493  * in later.
494  * See decode_nal_units().
495  */
496  pic->f->key_frame = 0;
497  pic->mmco_reset = 0;
498  pic->recovered = 0;
499  pic->invalid_gap = 0;
501 
502  pic->f->pict_type = h->slice_ctx[0].slice_type;
503 
504  if ((ret = alloc_picture(h, pic)) < 0)
505  return ret;
506  if(!h->frame_recovered && !h->avctx->hwaccel
509 #endif
510  )
511  ff_color_frame(pic->f, c);
512 
513  h->cur_pic_ptr = pic;
515  if (CONFIG_ERROR_RESILIENCE) {
517  }
518 
519  if ((ret = ff_h264_ref_picture(h, &h->cur_pic, h->cur_pic_ptr)) < 0)
520  return ret;
521 
522  for (i = 0; i < h->nb_slice_ctx; i++) {
523  h->slice_ctx[i].linesize = h->cur_pic_ptr->f->linesize[0];
524  h->slice_ctx[i].uvlinesize = h->cur_pic_ptr->f->linesize[1];
525  }
526 
527  if (CONFIG_ERROR_RESILIENCE && h->enable_er) {
531  }
532 
533  for (i = 0; i < 16; i++) {
534  h->block_offset[i] = (4 * ((scan8[i] - scan8[0]) & 7) << pixel_shift) + 4 * pic->f->linesize[0] * ((scan8[i] - scan8[0]) >> 3);
535  h->block_offset[48 + i] = (4 * ((scan8[i] - scan8[0]) & 7) << pixel_shift) + 8 * pic->f->linesize[0] * ((scan8[i] - scan8[0]) >> 3);
536  }
537  for (i = 0; i < 16; i++) {
538  h->block_offset[16 + i] =
539  h->block_offset[32 + i] = (4 * ((scan8[i] - scan8[0]) & 7) << pixel_shift) + 4 * pic->f->linesize[1] * ((scan8[i] - scan8[0]) >> 3);
540  h->block_offset[48 + 16 + i] =
541  h->block_offset[48 + 32 + i] = (4 * ((scan8[i] - scan8[0]) & 7) << pixel_shift) + 8 * pic->f->linesize[1] * ((scan8[i] - scan8[0]) >> 3);
542  }
543 
544  /* We mark the current picture as non-reference after allocating it, so
545  * that if we break out due to an error it can be released automatically
546  * in the next ff_mpv_frame_start().
547  */
548  h->cur_pic_ptr->reference = 0;
549 
550  h->cur_pic_ptr->field_poc[0] = h->cur_pic_ptr->field_poc[1] = INT_MAX;
551 
552  h->next_output_pic = NULL;
553 
554  h->postpone_filter = 0;
555 
556  assert(h->cur_pic_ptr->long_ref == 0);
557 
558  return 0;
559 }
560 
562  uint8_t *src_y,
563  uint8_t *src_cb, uint8_t *src_cr,
564  int linesize, int uvlinesize,
565  int simple)
566 {
567  uint8_t *top_border;
568  int top_idx = 1;
569  const int pixel_shift = h->pixel_shift;
570  int chroma444 = CHROMA444(h);
571  int chroma422 = CHROMA422(h);
572 
573  src_y -= linesize;
574  src_cb -= uvlinesize;
575  src_cr -= uvlinesize;
576 
577  if (!simple && FRAME_MBAFF(h)) {
578  if (sl->mb_y & 1) {
579  if (!MB_MBAFF(sl)) {
580  top_border = sl->top_borders[0][sl->mb_x];
581  AV_COPY128(top_border, src_y + 15 * linesize);
582  if (pixel_shift)
583  AV_COPY128(top_border + 16, src_y + 15 * linesize + 16);
584  if (simple || !CONFIG_GRAY || !(h->flags & AV_CODEC_FLAG_GRAY)) {
585  if (chroma444) {
586  if (pixel_shift) {
587  AV_COPY128(top_border + 32, src_cb + 15 * uvlinesize);
588  AV_COPY128(top_border + 48, src_cb + 15 * uvlinesize + 16);
589  AV_COPY128(top_border + 64, src_cr + 15 * uvlinesize);
590  AV_COPY128(top_border + 80, src_cr + 15 * uvlinesize + 16);
591  } else {
592  AV_COPY128(top_border + 16, src_cb + 15 * uvlinesize);
593  AV_COPY128(top_border + 32, src_cr + 15 * uvlinesize);
594  }
595  } else if (chroma422) {
596  if (pixel_shift) {
597  AV_COPY128(top_border + 32, src_cb + 15 * uvlinesize);
598  AV_COPY128(top_border + 48, src_cr + 15 * uvlinesize);
599  } else {
600  AV_COPY64(top_border + 16, src_cb + 15 * uvlinesize);
601  AV_COPY64(top_border + 24, src_cr + 15 * uvlinesize);
602  }
603  } else {
604  if (pixel_shift) {
605  AV_COPY128(top_border + 32, src_cb + 7 * uvlinesize);
606  AV_COPY128(top_border + 48, src_cr + 7 * uvlinesize);
607  } else {
608  AV_COPY64(top_border + 16, src_cb + 7 * uvlinesize);
609  AV_COPY64(top_border + 24, src_cr + 7 * uvlinesize);
610  }
611  }
612  }
613  }
614  } else if (MB_MBAFF(sl)) {
615  top_idx = 0;
616  } else
617  return;
618  }
619 
620  top_border = sl->top_borders[top_idx][sl->mb_x];
621  /* There are two lines saved, the line above the top macroblock
622  * of a pair, and the line above the bottom macroblock. */
623  AV_COPY128(top_border, src_y + 16 * linesize);
624  if (pixel_shift)
625  AV_COPY128(top_border + 16, src_y + 16 * linesize + 16);
626 
627  if (simple || !CONFIG_GRAY || !(h->flags & AV_CODEC_FLAG_GRAY)) {
628  if (chroma444) {
629  if (pixel_shift) {
630  AV_COPY128(top_border + 32, src_cb + 16 * linesize);
631  AV_COPY128(top_border + 48, src_cb + 16 * linesize + 16);
632  AV_COPY128(top_border + 64, src_cr + 16 * linesize);
633  AV_COPY128(top_border + 80, src_cr + 16 * linesize + 16);
634  } else {
635  AV_COPY128(top_border + 16, src_cb + 16 * linesize);
636  AV_COPY128(top_border + 32, src_cr + 16 * linesize);
637  }
638  } else if (chroma422) {
639  if (pixel_shift) {
640  AV_COPY128(top_border + 32, src_cb + 16 * uvlinesize);
641  AV_COPY128(top_border + 48, src_cr + 16 * uvlinesize);
642  } else {
643  AV_COPY64(top_border + 16, src_cb + 16 * uvlinesize);
644  AV_COPY64(top_border + 24, src_cr + 16 * uvlinesize);
645  }
646  } else {
647  if (pixel_shift) {
648  AV_COPY128(top_border + 32, src_cb + 8 * uvlinesize);
649  AV_COPY128(top_border + 48, src_cr + 8 * uvlinesize);
650  } else {
651  AV_COPY64(top_border + 16, src_cb + 8 * uvlinesize);
652  AV_COPY64(top_border + 24, src_cr + 8 * uvlinesize);
653  }
654  }
655  }
656 }
657 
658 /**
659  * Initialize implicit_weight table.
660  * @param field 0/1 initialize the weight for interlaced MBAFF
661  * -1 initializes the rest
662  */
663 static void implicit_weight_table(const H264Context *h, H264SliceContext *sl, int field)
664 {
665  int ref0, ref1, i, cur_poc, ref_start, ref_count0, ref_count1;
666 
667  for (i = 0; i < 2; i++) {
668  sl->pwt.luma_weight_flag[i] = 0;
669  sl->pwt.chroma_weight_flag[i] = 0;
670  }
671 
672  if (field < 0) {
673  if (h->picture_structure == PICT_FRAME) {
674  cur_poc = h->cur_pic_ptr->poc;
675  } else {
676  cur_poc = h->cur_pic_ptr->field_poc[h->picture_structure - 1];
677  }
678  if (sl->ref_count[0] == 1 && sl->ref_count[1] == 1 && !FRAME_MBAFF(h) &&
679  sl->ref_list[0][0].poc + (int64_t)sl->ref_list[1][0].poc == 2 * cur_poc) {
680  sl->pwt.use_weight = 0;
681  sl->pwt.use_weight_chroma = 0;
682  return;
683  }
684  ref_start = 0;
685  ref_count0 = sl->ref_count[0];
686  ref_count1 = sl->ref_count[1];
687  } else {
688  cur_poc = h->cur_pic_ptr->field_poc[field];
689  ref_start = 16;
690  ref_count0 = 16 + 2 * sl->ref_count[0];
691  ref_count1 = 16 + 2 * sl->ref_count[1];
692  }
693 
694  sl->pwt.use_weight = 2;
695  sl->pwt.use_weight_chroma = 2;
696  sl->pwt.luma_log2_weight_denom = 5;
698 
699  for (ref0 = ref_start; ref0 < ref_count0; ref0++) {
700  int64_t poc0 = sl->ref_list[0][ref0].poc;
701  for (ref1 = ref_start; ref1 < ref_count1; ref1++) {
702  int w = 32;
703  if (!sl->ref_list[0][ref0].parent->long_ref && !sl->ref_list[1][ref1].parent->long_ref) {
704  int poc1 = sl->ref_list[1][ref1].poc;
705  int td = av_clip_int8(poc1 - poc0);
706  if (td) {
707  int tb = av_clip_int8(cur_poc - poc0);
708  int tx = (16384 + (FFABS(td) >> 1)) / td;
709  int dist_scale_factor = (tb * tx + 32) >> 8;
710  if (dist_scale_factor >= -64 && dist_scale_factor <= 128)
711  w = 64 - dist_scale_factor;
712  }
713  }
714  if (field < 0) {
715  sl->pwt.implicit_weight[ref0][ref1][0] =
716  sl->pwt.implicit_weight[ref0][ref1][1] = w;
717  } else {
718  sl->pwt.implicit_weight[ref0][ref1][field] = w;
719  }
720  }
721  }
722 }
723 
724 /**
725  * initialize scan tables
726  */
728 {
729  int i;
730  for (i = 0; i < 16; i++) {
731 #define TRANSPOSE(x) ((x) >> 2) | (((x) << 2) & 0xF)
733  h->field_scan[i] = TRANSPOSE(field_scan[i]);
734 #undef TRANSPOSE
735  }
736  for (i = 0; i < 64; i++) {
737 #define TRANSPOSE(x) ((x) >> 3) | (((x) & 7) << 3)
742 #undef TRANSPOSE
743  }
744  if (h->ps.sps->transform_bypass) { // FIXME same ugly
745  memcpy(h->zigzag_scan_q0 , ff_zigzag_scan , sizeof(h->zigzag_scan_q0 ));
746  memcpy(h->zigzag_scan8x8_q0 , ff_zigzag_direct , sizeof(h->zigzag_scan8x8_q0 ));
748  memcpy(h->field_scan_q0 , field_scan , sizeof(h->field_scan_q0 ));
749  memcpy(h->field_scan8x8_q0 , field_scan8x8 , sizeof(h->field_scan8x8_q0 ));
751  } else {
752  memcpy(h->zigzag_scan_q0 , h->zigzag_scan , sizeof(h->zigzag_scan_q0 ));
753  memcpy(h->zigzag_scan8x8_q0 , h->zigzag_scan8x8 , sizeof(h->zigzag_scan8x8_q0 ));
755  memcpy(h->field_scan_q0 , h->field_scan , sizeof(h->field_scan_q0 ));
756  memcpy(h->field_scan8x8_q0 , h->field_scan8x8 , sizeof(h->field_scan8x8_q0 ));
758  }
759 }
760 
761 static enum AVPixelFormat get_pixel_format(H264Context *h, int force_callback)
762 {
763 #define HWACCEL_MAX (CONFIG_H264_DXVA2_HWACCEL + \
764  CONFIG_H264_D3D11VA_HWACCEL + \
765  CONFIG_H264_VAAPI_HWACCEL + \
766  (CONFIG_H264_VDA_HWACCEL * 2) + \
767  CONFIG_H264_VIDEOTOOLBOX_HWACCEL + \
768  CONFIG_H264_VDPAU_HWACCEL)
770  const enum AVPixelFormat *choices = pix_fmts;
771  int i;
772 
773  switch (h->ps.sps->bit_depth_luma) {
774  case 9:
775  if (CHROMA444(h)) {
776  if (h->avctx->colorspace == AVCOL_SPC_RGB) {
777  *fmt++ = AV_PIX_FMT_GBRP9;
778  } else
779  *fmt++ = AV_PIX_FMT_YUV444P9;
780  } else if (CHROMA422(h))
781  *fmt++ = AV_PIX_FMT_YUV422P9;
782  else
783  *fmt++ = AV_PIX_FMT_YUV420P9;
784  break;
785  case 10:
786  if (CHROMA444(h)) {
787  if (h->avctx->colorspace == AVCOL_SPC_RGB) {
788  *fmt++ = AV_PIX_FMT_GBRP10;
789  } else
790  *fmt++ = AV_PIX_FMT_YUV444P10;
791  } else if (CHROMA422(h))
792  *fmt++ = AV_PIX_FMT_YUV422P10;
793  else
794  *fmt++ = AV_PIX_FMT_YUV420P10;
795  break;
796  case 12:
797  if (CHROMA444(h)) {
798  if (h->avctx->colorspace == AVCOL_SPC_RGB) {
799  *fmt++ = AV_PIX_FMT_GBRP12;
800  } else
801  *fmt++ = AV_PIX_FMT_YUV444P12;
802  } else if (CHROMA422(h))
803  *fmt++ = AV_PIX_FMT_YUV422P12;
804  else
805  *fmt++ = AV_PIX_FMT_YUV420P12;
806  break;
807  case 14:
808  if (CHROMA444(h)) {
809  if (h->avctx->colorspace == AVCOL_SPC_RGB) {
810  *fmt++ = AV_PIX_FMT_GBRP14;
811  } else
812  *fmt++ = AV_PIX_FMT_YUV444P14;
813  } else if (CHROMA422(h))
814  *fmt++ = AV_PIX_FMT_YUV422P14;
815  else
816  *fmt++ = AV_PIX_FMT_YUV420P14;
817  break;
818  case 8:
819 #if CONFIG_H264_VDPAU_HWACCEL
820  *fmt++ = AV_PIX_FMT_VDPAU;
821 #endif
822  if (CHROMA444(h)) {
823  if (h->avctx->colorspace == AVCOL_SPC_RGB)
824  *fmt++ = AV_PIX_FMT_GBRP;
825  else if (h->avctx->color_range == AVCOL_RANGE_JPEG)
826  *fmt++ = AV_PIX_FMT_YUVJ444P;
827  else
828  *fmt++ = AV_PIX_FMT_YUV444P;
829  } else if (CHROMA422(h)) {
831  *fmt++ = AV_PIX_FMT_YUVJ422P;
832  else
833  *fmt++ = AV_PIX_FMT_YUV422P;
834  } else {
835 #if CONFIG_H264_DXVA2_HWACCEL
836  *fmt++ = AV_PIX_FMT_DXVA2_VLD;
837 #endif
838 #if CONFIG_H264_D3D11VA_HWACCEL
839  *fmt++ = AV_PIX_FMT_D3D11VA_VLD;
840 #endif
841 #if CONFIG_H264_VAAPI_HWACCEL
842  *fmt++ = AV_PIX_FMT_VAAPI;
843 #endif
844 #if CONFIG_H264_VDA_HWACCEL
845  *fmt++ = AV_PIX_FMT_VDA_VLD;
846  *fmt++ = AV_PIX_FMT_VDA;
847 #endif
848 #if CONFIG_H264_VIDEOTOOLBOX_HWACCEL
849  *fmt++ = AV_PIX_FMT_VIDEOTOOLBOX;
850 #endif
851  if (h->avctx->codec->pix_fmts)
852  choices = h->avctx->codec->pix_fmts;
853  else if (h->avctx->color_range == AVCOL_RANGE_JPEG)
854  *fmt++ = AV_PIX_FMT_YUVJ420P;
855  else
856  *fmt++ = AV_PIX_FMT_YUV420P;
857  }
858  break;
859  default:
861  "Unsupported bit depth %d\n", h->ps.sps->bit_depth_luma);
862  return AVERROR_INVALIDDATA;
863  }
864 
865  *fmt = AV_PIX_FMT_NONE;
866 
867  for (i=0; choices[i] != AV_PIX_FMT_NONE; i++)
868  if (choices[i] == h->avctx->pix_fmt && !force_callback)
869  return choices[i];
870  return ff_thread_get_format(h->avctx, choices);
871 }
872 
873 /* export coded and cropped frame dimensions to AVCodecContext */
875 {
876  SPS *sps = h->ps.sps;
877  int width = h->width - (sps->crop_right + sps->crop_left);
878  int height = h->height - (sps->crop_top + sps->crop_bottom);
879  av_assert0(sps->crop_right + sps->crop_left < (unsigned)h->width);
880  av_assert0(sps->crop_top + sps->crop_bottom < (unsigned)h->height);
881 
882  /* handle container cropping */
883  if (FFALIGN(h->avctx->width, 16) == FFALIGN(width, 16) &&
884  FFALIGN(h->avctx->height, 16) == FFALIGN(height, 16) &&
885  h->avctx->width <= width &&
886  h->avctx->height <= height
887  ) {
888  width = h->avctx->width;
889  height = h->avctx->height;
890  }
891 
892  if (width <= 0 || height <= 0) {
893  av_log(h->avctx, AV_LOG_ERROR, "Invalid cropped dimensions: %dx%d.\n",
894  width, height);
896  return AVERROR_INVALIDDATA;
897 
898  av_log(h->avctx, AV_LOG_WARNING, "Ignoring cropping information.\n");
899  sps->crop_bottom =
900  sps->crop_top =
901  sps->crop_right =
902  sps->crop_left =
903  sps->crop = 0;
904 
905  width = h->width;
906  height = h->height;
907  }
908 
909  h->avctx->coded_width = h->width;
910  h->avctx->coded_height = h->height;
911  h->avctx->width = width;
912  h->avctx->height = height;
913 
914  return 0;
915 }
916 
918 {
919  const SPS *sps = h->ps.sps;
920  int i, ret;
921 
922  ff_set_sar(h->avctx, sps->sar);
924  &h->chroma_x_shift, &h->chroma_y_shift);
925 
926  if (sps->timing_info_present_flag) {
927  int64_t den = sps->time_scale;
928  if (h->sei.unregistered.x264_build < 44U)
929  den *= 2;
931  sps->num_units_in_tick * h->avctx->ticks_per_frame, den, 1 << 30);
932  }
933 
935 
936  h->first_field = 0;
937  h->prev_interlaced_frame = 1;
938 
939  init_scan_tables(h);
940  ret = ff_h264_alloc_tables(h);
941  if (ret < 0) {
942  av_log(h->avctx, AV_LOG_ERROR, "Could not allocate memory\n");
943  goto fail;
944  }
945 
946 #if FF_API_CAP_VDPAU
947  if (h->avctx->codec &&
949  (sps->bit_depth_luma != 8 || sps->chroma_format_idc > 1)) {
951  "VDPAU decoding does not support video colorspace.\n");
952  ret = AVERROR_INVALIDDATA;
953  goto fail;
954  }
955 #endif
956 
957  if (sps->bit_depth_luma < 8 || sps->bit_depth_luma > 14 ||
958  sps->bit_depth_luma == 11 || sps->bit_depth_luma == 13
959  ) {
960  av_log(h->avctx, AV_LOG_ERROR, "Unsupported bit depth %d\n",
961  sps->bit_depth_luma);
962  ret = AVERROR_INVALIDDATA;
963  goto fail;
964  }
965 
966  h->cur_bit_depth_luma =
969  h->pixel_shift = sps->bit_depth_luma > 8;
971  h->bit_depth_luma = sps->bit_depth_luma;
972 
974  sps->chroma_format_idc);
978  sps->chroma_format_idc);
980 
981  if (!HAVE_THREADS || !(h->avctx->active_thread_type & FF_THREAD_SLICE)) {
982  ret = ff_h264_slice_context_init(h, &h->slice_ctx[0]);
983  if (ret < 0) {
984  av_log(h->avctx, AV_LOG_ERROR, "context_init() failed.\n");
985  goto fail;
986  }
987  } else {
988  for (i = 0; i < h->nb_slice_ctx; i++) {
989  H264SliceContext *sl = &h->slice_ctx[i];
990 
991  sl->h264 = h;
992  sl->intra4x4_pred_mode = h->intra4x4_pred_mode + i * 8 * 2 * h->mb_stride;
993  sl->mvd_table[0] = h->mvd_table[0] + i * 8 * 2 * h->mb_stride;
994  sl->mvd_table[1] = h->mvd_table[1] + i * 8 * 2 * h->mb_stride;
995 
996  if ((ret = ff_h264_slice_context_init(h, sl)) < 0) {
997  av_log(h->avctx, AV_LOG_ERROR, "context_init() failed.\n");
998  goto fail;
999  }
1000  }
1001  }
1002 
1003  h->context_initialized = 1;
1004 
1005  return 0;
1006 fail:
1008  h->context_initialized = 0;
1009  return ret;
1010 }
1011 
1013 {
1014  switch (a) {
1018  default:
1019  return a;
1020  }
1021 }
1022 
1023 /**
1024  * Decode a slice header.
1025  * This will (re)initialize the decoder and call h264_frame_start() as needed.
1026  *
1027  * @param h h264context
1028  *
1029  * @return 0 if okay, <0 if an error occurred
1030  */
1032 {
1033  const SPS *sps;
1034  const PPS *pps;
1035  unsigned int first_mb_in_slice;
1036  unsigned int pps_id;
1037  int ret;
1038  unsigned int slice_type, tmp, i, j;
1039  int last_pic_structure, last_pic_droppable;
1040  int must_reinit;
1041  int needs_reinit = 0;
1042  int field_pic_flag, bottom_field_flag;
1043  int first_slice = sl == h->slice_ctx && !h->current_slice;
1044  int frame_num, droppable, picture_structure;
1045  int mb_aff_frame, last_mb_aff_frame;
1046 
1047  if (first_slice)
1049 
1050  first_mb_in_slice = get_ue_golomb_long(&sl->gb);
1051 
1052  if (first_mb_in_slice == 0) { // FIXME better field boundary detection
1053  if (h->current_slice) {
1054  if (h->setup_finished) {
1055  av_log(h->avctx, AV_LOG_ERROR, "Too many fields\n");
1056  return AVERROR_INVALIDDATA;
1057  }
1058  if (h->max_contexts > 1) {
1059  if (!h->single_decode_warning) {
1060  av_log(h->avctx, AV_LOG_WARNING, "Cannot decode multiple access units as slice threads\n");
1061  h->single_decode_warning = 1;
1062  }
1063  h->max_contexts = 1;
1064  return SLICE_SINGLETHREAD;
1065  }
1066 
1067  if (h->cur_pic_ptr && FIELD_PICTURE(h) && h->first_field) {
1068  ret = ff_h264_field_end(h, h->slice_ctx, 1);
1069  h->current_slice = 0;
1070  if (ret < 0)
1071  return ret;
1072  } else if (h->cur_pic_ptr && !FIELD_PICTURE(h) && !h->first_field && h->nal_unit_type == NAL_IDR_SLICE) {
1073  av_log(h, AV_LOG_WARNING, "Broken frame packetizing\n");
1074  ret = ff_h264_field_end(h, h->slice_ctx, 1);
1075  h->current_slice = 0;
1076  ff_thread_report_progress(&h->cur_pic_ptr->tf, INT_MAX, 0);
1077  ff_thread_report_progress(&h->cur_pic_ptr->tf, INT_MAX, 1);
1078  h->cur_pic_ptr = NULL;
1079  if (ret < 0)
1080  return ret;
1081  } else
1082  return AVERROR_INVALIDDATA;
1083  }
1084 
1085  if (!h->first_field) {
1086  if (h->cur_pic_ptr && !h->droppable) {
1089  }
1090  h->cur_pic_ptr = NULL;
1091  }
1092  }
1093 
1094  if (!h->current_slice)
1095  av_assert0(sl == h->slice_ctx);
1096 
1097  slice_type = get_ue_golomb_31(&sl->gb);
1098  if (slice_type > 9) {
1100  "slice type %d too large at %d\n",
1101  slice_type, first_mb_in_slice);
1102  return AVERROR_INVALIDDATA;
1103  }
1104  if (slice_type > 4) {
1105  slice_type -= 5;
1106  sl->slice_type_fixed = 1;
1107  } else
1108  sl->slice_type_fixed = 0;
1109 
1110  slice_type = ff_h264_golomb_to_pict_type[slice_type];
1111  sl->slice_type = slice_type;
1112  sl->slice_type_nos = slice_type & 3;
1113 
1114  if (h->nal_unit_type == NAL_IDR_SLICE &&
1116  av_log(h->avctx, AV_LOG_ERROR, "A non-intra slice in an IDR NAL unit.\n");
1117  return AVERROR_INVALIDDATA;
1118  }
1119 
1120  if (h->current_slice == 0 && !h->first_field) {
1121  if (
1122  (h->avctx->skip_frame >= AVDISCARD_NONREF && !h->nal_ref_idc) ||
1126  h->avctx->skip_frame >= AVDISCARD_ALL) {
1127  return SLICE_SKIPED;
1128  }
1129  }
1130 
1131  pps_id = get_ue_golomb(&sl->gb);
1132  if (pps_id >= MAX_PPS_COUNT) {
1133  av_log(h->avctx, AV_LOG_ERROR, "pps_id %u out of range\n", pps_id);
1134  return AVERROR_INVALIDDATA;
1135  }
1136  if (!h->ps.pps_list[pps_id]) {
1138  "non-existing PPS %u referenced\n",
1139  pps_id);
1140  return AVERROR_INVALIDDATA;
1141  }
1142  if (h->au_pps_id >= 0 && pps_id != h->au_pps_id) {
1144  "PPS change from %d to %d forbidden\n",
1145  h->au_pps_id, pps_id);
1146  return AVERROR_INVALIDDATA;
1147  }
1148 
1149  pps = (const PPS*)h->ps.pps_list[pps_id]->data;
1150 
1151  if (!h->ps.sps_list[pps->sps_id]) {
1153  "non-existing SPS %u referenced\n",
1154  pps->sps_id);
1155  return AVERROR_INVALIDDATA;
1156  }
1157 
1158  if (first_slice) {
1159  av_buffer_unref(&h->ps.pps_ref);
1160  h->ps.pps = NULL;
1161  h->ps.pps_ref = av_buffer_ref(h->ps.pps_list[pps_id]);
1162  if (!h->ps.pps_ref)
1163  return AVERROR(ENOMEM);
1164  h->ps.pps = (const PPS*)h->ps.pps_ref->data;
1165  } else {
1166  if (h->ps.pps->sps_id != pps->sps_id ||
1167  h->ps.pps->transform_8x8_mode != pps->transform_8x8_mode /*||
1168  (h->setup_finished && h->ps.pps != pps)*/) {
1169  av_log(h->avctx, AV_LOG_ERROR, "PPS changed between slices\n");
1170  return AVERROR_INVALIDDATA;
1171  }
1172  }
1173 
1174  if (h->ps.sps != (const SPS*)h->ps.sps_list[h->ps.pps->sps_id]->data ||
1175  pps->sps_id != h->current_sps_id) {
1176 
1177  if (!first_slice) {
1179  "SPS changed in the middle of the frame\n");
1180  return AVERROR_INVALIDDATA;
1181  }
1182 
1183  av_buffer_unref(&h->ps.sps_ref);
1184  h->ps.sps = NULL;
1185  h->ps.sps_ref = av_buffer_ref(h->ps.sps_list[h->ps.pps->sps_id]);
1186  if (!h->ps.sps_ref)
1187  return AVERROR(ENOMEM);
1188  h->ps.sps = (const SPS*)h->ps.sps_ref->data;
1189 
1190  if (h->mb_width != h->ps.sps->mb_width ||
1191  h->mb_height != h->ps.sps->mb_height * (2 - h->ps.sps->frame_mbs_only_flag) ||
1194  )
1195  needs_reinit = 1;
1196 
1197  if (h->bit_depth_luma != h->ps.sps->bit_depth_luma ||
1199  needs_reinit = 1;
1200  }
1201 
1202  pps = h->ps.pps;
1203  sps = h->ps.sps;
1204 
1205  must_reinit = (h->context_initialized &&
1206  ( 16*sps->mb_width != h->avctx->coded_width
1207  || 16*sps->mb_height * (2 - sps->frame_mbs_only_flag) != h->avctx->coded_height
1208  || h->cur_bit_depth_luma != sps->bit_depth_luma
1210  || h->mb_width != sps->mb_width
1211  || h->mb_height != sps->mb_height * (2 - sps->frame_mbs_only_flag)
1212  ));
1213  if (h->avctx->pix_fmt == AV_PIX_FMT_NONE
1215  must_reinit = 1;
1216 
1217  if (first_slice && av_cmp_q(sps->sar, h->avctx->sample_aspect_ratio))
1218  must_reinit = 1;
1219 
1220  if (!h->setup_finished) {
1221  h->avctx->profile = ff_h264_get_profile(sps);
1222  h->avctx->level = sps->level_idc;
1223  h->avctx->refs = sps->ref_frame_count;
1224 
1225  h->mb_width = sps->mb_width;
1226  h->mb_height = sps->mb_height * (2 - sps->frame_mbs_only_flag);
1227  h->mb_num = h->mb_width * h->mb_height;
1228  h->mb_stride = h->mb_width + 1;
1229 
1230  h->b_stride = h->mb_width * 4;
1231 
1232  h->chroma_y_shift = sps->chroma_format_idc <= 1; // 400 uses yuv420p
1233 
1234  h->width = 16 * h->mb_width;
1235  h->height = 16 * h->mb_height;
1236 
1237  ret = init_dimensions(h);
1238  if (ret < 0)
1239  return ret;
1240 
1241  if (sps->video_signal_type_present_flag) {
1242  h->avctx->color_range = sps->full_range > 0 ? AVCOL_RANGE_JPEG
1243  : AVCOL_RANGE_MPEG;
1245  if (h->avctx->colorspace != sps->colorspace)
1246  needs_reinit = 1;
1248  h->avctx->color_trc = sps->color_trc;
1249  h->avctx->colorspace = sps->colorspace;
1250  }
1251  }
1252  }
1253 
1254  if (h->context_initialized &&
1255  (must_reinit || needs_reinit)) {
1256  h->context_initialized = 0;
1257  if (sl != h->slice_ctx) {
1259  "changing width %d -> %d / height %d -> %d on "
1260  "slice %d\n",
1261  h->width, h->avctx->coded_width,
1262  h->height, h->avctx->coded_height,
1263  h->current_slice + 1);
1264  return AVERROR_INVALIDDATA;
1265  }
1266 
1267  av_assert1(first_slice);
1268 
1270 
1271  if ((ret = get_pixel_format(h, 1)) < 0)
1272  return ret;
1273  h->avctx->pix_fmt = ret;
1274 
1275  av_log(h->avctx, AV_LOG_INFO, "Reinit context to %dx%d, "
1276  "pix_fmt: %s\n", h->width, h->height, av_get_pix_fmt_name(h->avctx->pix_fmt));
1277 
1278  if ((ret = h264_slice_header_init(h)) < 0) {
1280  "h264_slice_header_init() failed\n");
1281  return ret;
1282  }
1283  }
1284  if (!h->context_initialized) {
1285  if (sl != h->slice_ctx) {
1287  "Cannot (re-)initialize context during parallel decoding.\n");
1288  return AVERROR_PATCHWELCOME;
1289  }
1290 
1291  if ((ret = get_pixel_format(h, 1)) < 0)
1292  return ret;
1293  h->avctx->pix_fmt = ret;
1294 
1295  if ((ret = h264_slice_header_init(h)) < 0) {
1297  "h264_slice_header_init() failed\n");
1298  return ret;
1299  }
1300  }
1301 
1302  frame_num = get_bits(&sl->gb, sps->log2_max_frame_num);
1303  if (!first_slice) {
1304  if (h->poc.frame_num != frame_num) {
1305  av_log(h->avctx, AV_LOG_ERROR, "Frame num change from %d to %d\n",
1306  h->poc.frame_num, frame_num);
1307  return AVERROR_INVALIDDATA;
1308  }
1309  }
1310 
1311  if (!h->setup_finished)
1312  h->poc.frame_num = frame_num;
1313 
1314  sl->mb_mbaff = 0;
1315  mb_aff_frame = 0;
1316  last_mb_aff_frame = h->mb_aff_frame;
1317  last_pic_structure = h->picture_structure;
1318  last_pic_droppable = h->droppable;
1319 
1320  droppable = h->nal_ref_idc == 0;
1321  if (sps->frame_mbs_only_flag) {
1322  picture_structure = PICT_FRAME;
1323  } else {
1324  if (!h->ps.sps->direct_8x8_inference_flag && slice_type == AV_PICTURE_TYPE_B) {
1325  av_log(h->avctx, AV_LOG_ERROR, "This stream was generated by a broken encoder, invalid 8x8 inference\n");
1326  return -1;
1327  }
1328  field_pic_flag = get_bits1(&sl->gb);
1329 
1330  if (field_pic_flag) {
1331  bottom_field_flag = get_bits1(&sl->gb);
1332  picture_structure = PICT_TOP_FIELD + bottom_field_flag;
1333  } else {
1334  picture_structure = PICT_FRAME;
1335  mb_aff_frame = sps->mb_aff;
1336  }
1337  }
1338 
1339  if (h->current_slice) {
1340  if (last_pic_structure != picture_structure ||
1341  last_pic_droppable != droppable ||
1342  last_mb_aff_frame != mb_aff_frame) {
1344  "Changing field mode (%d -> %d) between slices is not allowed\n",
1345  last_pic_structure, h->picture_structure);
1346  return AVERROR_INVALIDDATA;
1347  } else if (!h->cur_pic_ptr) {
1349  "unset cur_pic_ptr on slice %d\n",
1350  h->current_slice + 1);
1351  return AVERROR_INVALIDDATA;
1352  }
1353  }
1354 
1355  if (!h->setup_finished) {
1356  h->droppable = droppable;
1357  h->picture_structure = picture_structure;
1358  h->mb_aff_frame = mb_aff_frame;
1359  }
1360  sl->mb_field_decoding_flag = picture_structure != PICT_FRAME;
1361 
1362  if (h->current_slice == 0) {
1363  /* Shorten frame num gaps so we don't have to allocate reference
1364  * frames just to throw them away */
1365  if (h->poc.frame_num != h->poc.prev_frame_num) {
1366  int unwrap_prev_frame_num = h->poc.prev_frame_num;
1367  int max_frame_num = 1 << sps->log2_max_frame_num;
1368 
1369  if (unwrap_prev_frame_num > h->poc.frame_num)
1370  unwrap_prev_frame_num -= max_frame_num;
1371 
1372  if ((h->poc.frame_num - unwrap_prev_frame_num) > sps->ref_frame_count) {
1373  unwrap_prev_frame_num = (h->poc.frame_num - sps->ref_frame_count) - 1;
1374  if (unwrap_prev_frame_num < 0)
1375  unwrap_prev_frame_num += max_frame_num;
1376 
1377  h->poc.prev_frame_num = unwrap_prev_frame_num;
1378  }
1379  }
1380 
1381  /* See if we have a decoded first field looking for a pair...
1382  * Here, we're using that to see if we should mark previously
1383  * decode frames as "finished".
1384  * We have to do that before the "dummy" in-between frame allocation,
1385  * since that can modify h->cur_pic_ptr. */
1386  if (h->first_field) {
1387  av_assert0(h->cur_pic_ptr);
1388  av_assert0(h->cur_pic_ptr->f->buf[0]);
1389  assert(h->cur_pic_ptr->reference != DELAYED_PIC_REF);
1390 
1391  /* Mark old field/frame as completed */
1392  if (h->cur_pic_ptr->tf.owner == h->avctx) {
1394  last_pic_structure == PICT_BOTTOM_FIELD);
1395  }
1396 
1397  /* figure out if we have a complementary field pair */
1398  if (!FIELD_PICTURE(h) || h->picture_structure == last_pic_structure) {
1399  /* Previous field is unmatched. Don't display it, but let it
1400  * remain for reference if marked as such. */
1401  if (last_pic_structure != PICT_FRAME) {
1403  last_pic_structure == PICT_TOP_FIELD);
1404  }
1405  } else {
1406  if (h->cur_pic_ptr->frame_num != h->poc.frame_num) {
1407  /* This and previous field were reference, but had
1408  * different frame_nums. Consider this field first in
1409  * pair. Throw away previous field except for reference
1410  * purposes. */
1411  if (last_pic_structure != PICT_FRAME) {
1413  last_pic_structure == PICT_TOP_FIELD);
1414  }
1415  } else {
1416  /* Second field in complementary pair */
1417  if (!((last_pic_structure == PICT_TOP_FIELD &&
1419  (last_pic_structure == PICT_BOTTOM_FIELD &&
1422  "Invalid field mode combination %d/%d\n",
1423  last_pic_structure, h->picture_structure);
1424  h->picture_structure = last_pic_structure;
1425  h->droppable = last_pic_droppable;
1426  return AVERROR_INVALIDDATA;
1427  } else if (last_pic_droppable != h->droppable) {
1429  "Found reference and non-reference fields in the same frame, which");
1430  h->picture_structure = last_pic_structure;
1431  h->droppable = last_pic_droppable;
1432  return AVERROR_PATCHWELCOME;
1433  }
1434  }
1435  }
1436  }
1437 
1438  while (h->poc.frame_num != h->poc.prev_frame_num && !h->first_field &&
1439  h->poc.frame_num != (h->poc.prev_frame_num + 1) % (1 << sps->log2_max_frame_num)) {
1440  H264Picture *prev = h->short_ref_count ? h->short_ref[0] : NULL;
1441  av_log(h->avctx, AV_LOG_DEBUG, "Frame num gap %d %d\n",
1442  h->poc.frame_num, h->poc.prev_frame_num);
1444  for(i=0; i<FF_ARRAY_ELEMS(h->last_pocs); i++)
1445  h->last_pocs[i] = INT_MIN;
1446  ret = h264_frame_start(h);
1447  if (ret < 0) {
1448  h->first_field = 0;
1449  return ret;
1450  }
1451 
1452  h->poc.prev_frame_num++;
1453  h->poc.prev_frame_num %= 1 << sps->log2_max_frame_num;
1456  ff_thread_report_progress(&h->cur_pic_ptr->tf, INT_MAX, 0);
1457  ff_thread_report_progress(&h->cur_pic_ptr->tf, INT_MAX, 1);
1459  if (ret < 0 && (h->avctx->err_recognition & AV_EF_EXPLODE))
1460  return ret;
1462  if (ret < 0 && (h->avctx->err_recognition & AV_EF_EXPLODE))
1463  return ret;
1464  /* Error concealment: If a ref is missing, copy the previous ref
1465  * in its place.
1466  * FIXME: Avoiding a memcpy would be nice, but ref handling makes
1467  * many assumptions about there being no actual duplicates.
1468  * FIXME: This does not copy padding for out-of-frame motion
1469  * vectors. Given we are concealing a lost frame, this probably
1470  * is not noticeable by comparison, but it should be fixed. */
1471  if (h->short_ref_count) {
1472  if (prev &&
1473  h->short_ref[0]->f->width == prev->f->width &&
1474  h->short_ref[0]->f->height == prev->f->height &&
1475  h->short_ref[0]->f->format == prev->f->format) {
1476  av_image_copy(h->short_ref[0]->f->data,
1477  h->short_ref[0]->f->linesize,
1478  (const uint8_t **)prev->f->data,
1479  prev->f->linesize,
1480  prev->f->format,
1481  prev->f->width,
1482  prev->f->height);
1483  h->short_ref[0]->poc = prev->poc + 2;
1484  }
1485  h->short_ref[0]->frame_num = h->poc.prev_frame_num;
1486  }
1487  }
1488 
1489  /* See if we have a decoded first field looking for a pair...
1490  * We're using that to see whether to continue decoding in that
1491  * frame, or to allocate a new one. */
1492  if (h->first_field) {
1493  av_assert0(h->cur_pic_ptr);
1494  av_assert0(h->cur_pic_ptr->f->buf[0]);
1495  assert(h->cur_pic_ptr->reference != DELAYED_PIC_REF);
1496 
1497  /* figure out if we have a complementary field pair */
1498  if (!FIELD_PICTURE(h) || h->picture_structure == last_pic_structure) {
1499  /* Previous field is unmatched. Don't display it, but let it
1500  * remain for reference if marked as such. */
1501  h->missing_fields ++;
1502  h->cur_pic_ptr = NULL;
1503  h->first_field = FIELD_PICTURE(h);
1504  } else {
1505  h->missing_fields = 0;
1506  if (h->cur_pic_ptr->frame_num != h->poc.frame_num) {
1509  /* This and the previous field had different frame_nums.
1510  * Consider this field first in pair. Throw away previous
1511  * one except for reference purposes. */
1512  h->first_field = 1;
1513  h->cur_pic_ptr = NULL;
1514  } else {
1515  /* Second field in complementary pair */
1516  h->first_field = 0;
1517  }
1518  }
1519  } else {
1520  /* Frame or first field in a potentially complementary pair */
1521  h->first_field = FIELD_PICTURE(h);
1522  }
1523 
1524  if (!FIELD_PICTURE(h) || h->first_field) {
1525  if (h264_frame_start(h) < 0) {
1526  h->first_field = 0;
1527  return AVERROR_INVALIDDATA;
1528  }
1529  } else {
1531  }
1532  /* Some macroblocks can be accessed before they're available in case
1533  * of lost slices, MBAFF or threading. */
1534  if (FIELD_PICTURE(h)) {
1535  for(i = (h->picture_structure == PICT_BOTTOM_FIELD); i<h->mb_height; i++)
1536  memset(h->slice_table + i*h->mb_stride, -1, (h->mb_stride - (i+1==h->mb_height)) * sizeof(*h->slice_table));
1537  } else {
1538  memset(h->slice_table, -1,
1539  (h->mb_height * h->mb_stride - 1) * sizeof(*h->slice_table));
1540  }
1541  }
1542 
1543  av_assert1(h->mb_num == h->mb_width * h->mb_height);
1544  if (first_mb_in_slice << FIELD_OR_MBAFF_PICTURE(h) >= h->mb_num ||
1545  first_mb_in_slice >= h->mb_num) {
1546  av_log(h->avctx, AV_LOG_ERROR, "first_mb_in_slice overflow\n");
1547  return AVERROR_INVALIDDATA;
1548  }
1549  sl->resync_mb_x = sl->mb_x = first_mb_in_slice % h->mb_width;
1550  sl->resync_mb_y = sl->mb_y = (first_mb_in_slice / h->mb_width) <<
1553  sl->resync_mb_y = sl->mb_y = sl->mb_y + 1;
1554  av_assert1(sl->mb_y < h->mb_height);
1555 
1556  if (h->picture_structure == PICT_FRAME) {
1557  h->curr_pic_num = h->poc.frame_num;
1558  h->max_pic_num = 1 << sps->log2_max_frame_num;
1559  } else {
1560  h->curr_pic_num = 2 * h->poc.frame_num + 1;
1561  h->max_pic_num = 1 << (sps->log2_max_frame_num + 1);
1562  }
1563 
1564  if (h->nal_unit_type == NAL_IDR_SLICE)
1565  get_ue_golomb_long(&sl->gb); /* idr_pic_id */
1566 
1567  if (sps->poc_type == 0) {
1568  int poc_lsb = get_bits(&sl->gb, sps->log2_max_poc_lsb);
1569 
1570  if (!h->setup_finished)
1571  h->poc.poc_lsb = poc_lsb;
1572 
1573  if (pps->pic_order_present == 1 && h->picture_structure == PICT_FRAME) {
1574  int delta_poc_bottom = get_se_golomb(&sl->gb);
1575  if (!h->setup_finished)
1576  h->poc.delta_poc_bottom = delta_poc_bottom;
1577  }
1578  }
1579 
1580  if (sps->poc_type == 1 && !sps->delta_pic_order_always_zero_flag) {
1581  int delta_poc = get_se_golomb(&sl->gb);
1582 
1583  if (!h->setup_finished)
1584  h->poc.delta_poc[0] = delta_poc;
1585 
1586  if (pps->pic_order_present == 1 && h->picture_structure == PICT_FRAME) {
1587  delta_poc = get_se_golomb(&sl->gb);
1588 
1589  if (!h->setup_finished)
1590  h->poc.delta_poc[1] = delta_poc;
1591  }
1592  }
1593 
1594  if (!h->setup_finished)
1596  sps, &h->poc, h->picture_structure, h->nal_ref_idc);
1597 
1598  if (pps->redundant_pic_cnt_present)
1599  sl->redundant_pic_count = get_ue_golomb(&sl->gb);
1600 
1601  if (sl->slice_type_nos == AV_PICTURE_TYPE_B)
1602  sl->direct_spatial_mv_pred = get_bits1(&sl->gb);
1603 
1605  &sl->gb, pps, sl->slice_type_nos,
1606  h->picture_structure, h->avctx);
1607  if (ret < 0)
1608  return ret;
1609 
1610  if (sl->slice_type_nos != AV_PICTURE_TYPE_I) {
1612  if (ret < 0) {
1613  sl->ref_count[1] = sl->ref_count[0] = 0;
1614  return ret;
1615  }
1616  }
1617 
1618  if ((pps->weighted_pred && sl->slice_type_nos == AV_PICTURE_TYPE_P) ||
1619  (pps->weighted_bipred_idc == 1 &&
1621  ff_h264_pred_weight_table(&sl->gb, sps, sl->ref_count,
1622  sl->slice_type_nos, &sl->pwt, h->avctx);
1623  else if (pps->weighted_bipred_idc == 2 &&
1625  implicit_weight_table(h, sl, -1);
1626  } else {
1627  sl->pwt.use_weight = 0;
1628  for (i = 0; i < 2; i++) {
1629  sl->pwt.luma_weight_flag[i] = 0;
1630  sl->pwt.chroma_weight_flag[i] = 0;
1631  }
1632  }
1633 
1634  // If frame-mt is enabled, only update mmco tables for the first slice
1635  // in a field. Subsequent slices can temporarily clobber h->mmco_index
1636  // or h->mmco, which will cause ref list mix-ups and decoding errors
1637  // further down the line. This may break decoding if the first slice is
1638  // corrupt, thus we only do this if frame-mt is enabled.
1639  if (h->nal_ref_idc) {
1640  ret = ff_h264_decode_ref_pic_marking(h, &sl->gb,
1642  h->current_slice == 0);
1643  if (ret < 0 && (h->avctx->err_recognition & AV_EF_EXPLODE))
1644  return AVERROR_INVALIDDATA;
1645  }
1646 
1647  if (FRAME_MBAFF(h)) {
1649 
1650  if (pps->weighted_bipred_idc == 2 && sl->slice_type_nos == AV_PICTURE_TYPE_B) {
1651  implicit_weight_table(h, sl, 0);
1652  implicit_weight_table(h, sl, 1);
1653  }
1654  }
1655 
1659 
1660  if (sl->slice_type_nos != AV_PICTURE_TYPE_I && pps->cabac) {
1661  tmp = get_ue_golomb_31(&sl->gb);
1662  if (tmp > 2) {
1663  av_log(h->avctx, AV_LOG_ERROR, "cabac_init_idc %u overflow\n", tmp);
1664  return AVERROR_INVALIDDATA;
1665  }
1666  sl->cabac_init_idc = tmp;
1667  }
1668 
1669  sl->last_qscale_diff = 0;
1670  tmp = pps->init_qp + get_se_golomb(&sl->gb);
1671  if (tmp > 51 + 6 * (sps->bit_depth_luma - 8)) {
1672  av_log(h->avctx, AV_LOG_ERROR, "QP %u out of range\n", tmp);
1673  return AVERROR_INVALIDDATA;
1674  }
1675  sl->qscale = tmp;
1676  sl->chroma_qp[0] = get_chroma_qp(h, 0, sl->qscale);
1677  sl->chroma_qp[1] = get_chroma_qp(h, 1, sl->qscale);
1678  // FIXME qscale / qp ... stuff
1679  if (sl->slice_type == AV_PICTURE_TYPE_SP)
1680  get_bits1(&sl->gb); /* sp_for_switch_flag */
1681  if (sl->slice_type == AV_PICTURE_TYPE_SP ||
1683  get_se_golomb(&sl->gb); /* slice_qs_delta */
1684 
1685  sl->deblocking_filter = 1;
1686  sl->slice_alpha_c0_offset = 0;
1687  sl->slice_beta_offset = 0;
1689  tmp = get_ue_golomb_31(&sl->gb);
1690  if (tmp > 2) {
1692  "deblocking_filter_idc %u out of range\n", tmp);
1693  return AVERROR_INVALIDDATA;
1694  }
1695  sl->deblocking_filter = tmp;
1696  if (sl->deblocking_filter < 2)
1697  sl->deblocking_filter ^= 1; // 1<->0
1698 
1699  if (sl->deblocking_filter) {
1700  sl->slice_alpha_c0_offset = get_se_golomb(&sl->gb) * 2;
1701  sl->slice_beta_offset = get_se_golomb(&sl->gb) * 2;
1702  if (sl->slice_alpha_c0_offset > 12 ||
1703  sl->slice_alpha_c0_offset < -12 ||
1704  sl->slice_beta_offset > 12 ||
1705  sl->slice_beta_offset < -12) {
1707  "deblocking filter parameters %d %d out of range\n",
1709  return AVERROR_INVALIDDATA;
1710  }
1711  }
1712  }
1713 
1714  if (h->avctx->skip_loop_filter >= AVDISCARD_ALL ||
1716  h->nal_unit_type != NAL_IDR_SLICE) ||
1722  h->nal_ref_idc == 0))
1723  sl->deblocking_filter = 0;
1724 
1725  if (sl->deblocking_filter == 1 && h->max_contexts > 1) {
1726  if (h->avctx->flags2 & AV_CODEC_FLAG2_FAST) {
1727  /* Cheat slightly for speed:
1728  * Do not bother to deblock across slices. */
1729  sl->deblocking_filter = 2;
1730  } else {
1731  h->postpone_filter = 1;
1732  }
1733  }
1734  sl->qp_thresh = 15 -
1736  FFMAX3(0,
1737  pps->chroma_qp_index_offset[0],
1738  pps->chroma_qp_index_offset[1]) +
1739  6 * (sps->bit_depth_luma - 8);
1740 
1741  sl->slice_num = ++h->current_slice;
1742 
1743  if (sl->slice_num)
1744  h->slice_row[(sl->slice_num-1)&(MAX_SLICES-1)]= sl->resync_mb_y;
1745  if ( h->slice_row[sl->slice_num&(MAX_SLICES-1)] + 3 >= sl->resync_mb_y
1746  && h->slice_row[sl->slice_num&(MAX_SLICES-1)] <= sl->resync_mb_y
1747  && sl->slice_num >= MAX_SLICES) {
1748  //in case of ASO this check needs to be updated depending on how we decide to assign slice numbers in this case
1749  av_log(h->avctx, AV_LOG_WARNING, "Possibly too many slices (%d >= %d), increase MAX_SLICES and recompile if there are artifacts\n", sl->slice_num, MAX_SLICES);
1750  }
1751 
1752  for (j = 0; j < 2; j++) {
1753  int id_list[16];
1754  int *ref2frm = h->ref2frm[sl->slice_num & (MAX_SLICES - 1)][j];
1755  for (i = 0; i < 16; i++) {
1756  id_list[i] = 60;
1757  if (j < sl->list_count && i < sl->ref_count[j] &&
1758  sl->ref_list[j][i].parent->f->buf[0]) {
1759  int k;
1760  AVBuffer *buf = sl->ref_list[j][i].parent->f->buf[0]->buffer;
1761  for (k = 0; k < h->short_ref_count; k++)
1762  if (h->short_ref[k]->f->buf[0]->buffer == buf) {
1763  id_list[i] = k;
1764  break;
1765  }
1766  for (k = 0; k < h->long_ref_count; k++)
1767  if (h->long_ref[k] && h->long_ref[k]->f->buf[0]->buffer == buf) {
1768  id_list[i] = h->short_ref_count + k;
1769  break;
1770  }
1771  }
1772  }
1773 
1774  ref2frm[0] =
1775  ref2frm[1] = -1;
1776  for (i = 0; i < 16; i++)
1777  ref2frm[i + 2] = 4 * id_list[i] + (sl->ref_list[j][i].reference & 3);
1778  ref2frm[18 + 0] =
1779  ref2frm[18 + 1] = -1;
1780  for (i = 16; i < 48; i++)
1781  ref2frm[i + 4] = 4 * id_list[(i - 16) >> 1] +
1782  (sl->ref_list[j][i].reference & 3);
1783  }
1784 
1785  h->au_pps_id = pps_id;
1786  h->current_sps_id = h->ps.pps->sps_id;
1787 
1788  if (h->avctx->debug & FF_DEBUG_PICT_INFO) {
1790  "slice:%d %s mb:%d %c%s%s pps:%u frame:%d poc:%d/%d ref:%d/%d qp:%d loop:%d:%d:%d weight:%d%s %s\n",
1791  sl->slice_num,
1792  (h->picture_structure == PICT_FRAME ? "F" : h->picture_structure == PICT_TOP_FIELD ? "T" : "B"),
1793  first_mb_in_slice,
1795  sl->slice_type_fixed ? " fix" : "",
1796  h->nal_unit_type == NAL_IDR_SLICE ? " IDR" : "",
1797  pps_id, h->poc.frame_num,
1798  h->cur_pic_ptr->field_poc[0],
1799  h->cur_pic_ptr->field_poc[1],
1800  sl->ref_count[0], sl->ref_count[1],
1801  sl->qscale,
1802  sl->deblocking_filter,
1804  sl->pwt.use_weight,
1805  sl->pwt.use_weight == 1 && sl->pwt.use_weight_chroma ? "c" : "",
1806  sl->slice_type == AV_PICTURE_TYPE_B ? (sl->direct_spatial_mv_pred ? "SPAT" : "TEMP") : "");
1807  }
1808 
1809  return 0;
1810 }
1811 
1813 {
1814  switch (sl->slice_type) {
1815  case AV_PICTURE_TYPE_P:
1816  return 0;
1817  case AV_PICTURE_TYPE_B:
1818  return 1;
1819  case AV_PICTURE_TYPE_I:
1820  return 2;
1821  case AV_PICTURE_TYPE_SP:
1822  return 3;
1823  case AV_PICTURE_TYPE_SI:
1824  return 4;
1825  default:
1826  return AVERROR_INVALIDDATA;
1827  }
1828 }
1829 
1831  H264SliceContext *sl,
1832  int mb_type, int top_xy,
1833  int left_xy[LEFT_MBS],
1834  int top_type,
1835  int left_type[LEFT_MBS],
1836  int mb_xy, int list)
1837 {
1838  int b_stride = h->b_stride;
1839  int16_t(*mv_dst)[2] = &sl->mv_cache[list][scan8[0]];
1840  int8_t *ref_cache = &sl->ref_cache[list][scan8[0]];
1841  if (IS_INTER(mb_type) || IS_DIRECT(mb_type)) {
1842  if (USES_LIST(top_type, list)) {
1843  const int b_xy = h->mb2b_xy[top_xy] + 3 * b_stride;
1844  const int b8_xy = 4 * top_xy + 2;
1845  const int *ref2frm = h->ref2frm[h->slice_table[top_xy] & (MAX_SLICES - 1)][list] + (MB_MBAFF(sl) ? 20 : 2);
1846  AV_COPY128(mv_dst - 1 * 8, h->cur_pic.motion_val[list][b_xy + 0]);
1847  ref_cache[0 - 1 * 8] =
1848  ref_cache[1 - 1 * 8] = ref2frm[h->cur_pic.ref_index[list][b8_xy + 0]];
1849  ref_cache[2 - 1 * 8] =
1850  ref_cache[3 - 1 * 8] = ref2frm[h->cur_pic.ref_index[list][b8_xy + 1]];
1851  } else {
1852  AV_ZERO128(mv_dst - 1 * 8);
1853  AV_WN32A(&ref_cache[0 - 1 * 8], ((LIST_NOT_USED) & 0xFF) * 0x01010101u);
1854  }
1855 
1856  if (!IS_INTERLACED(mb_type ^ left_type[LTOP])) {
1857  if (USES_LIST(left_type[LTOP], list)) {
1858  const int b_xy = h->mb2b_xy[left_xy[LTOP]] + 3;
1859  const int b8_xy = 4 * left_xy[LTOP] + 1;
1860  const int *ref2frm = h->ref2frm[h->slice_table[left_xy[LTOP]] & (MAX_SLICES - 1)][list] + (MB_MBAFF(sl) ? 20 : 2);
1861  AV_COPY32(mv_dst - 1 + 0, h->cur_pic.motion_val[list][b_xy + b_stride * 0]);
1862  AV_COPY32(mv_dst - 1 + 8, h->cur_pic.motion_val[list][b_xy + b_stride * 1]);
1863  AV_COPY32(mv_dst - 1 + 16, h->cur_pic.motion_val[list][b_xy + b_stride * 2]);
1864  AV_COPY32(mv_dst - 1 + 24, h->cur_pic.motion_val[list][b_xy + b_stride * 3]);
1865  ref_cache[-1 + 0] =
1866  ref_cache[-1 + 8] = ref2frm[h->cur_pic.ref_index[list][b8_xy + 2 * 0]];
1867  ref_cache[-1 + 16] =
1868  ref_cache[-1 + 24] = ref2frm[h->cur_pic.ref_index[list][b8_xy + 2 * 1]];
1869  } else {
1870  AV_ZERO32(mv_dst - 1 + 0);
1871  AV_ZERO32(mv_dst - 1 + 8);
1872  AV_ZERO32(mv_dst - 1 + 16);
1873  AV_ZERO32(mv_dst - 1 + 24);
1874  ref_cache[-1 + 0] =
1875  ref_cache[-1 + 8] =
1876  ref_cache[-1 + 16] =
1877  ref_cache[-1 + 24] = LIST_NOT_USED;
1878  }
1879  }
1880  }
1881 
1882  if (!USES_LIST(mb_type, list)) {
1883  fill_rectangle(mv_dst, 4, 4, 8, pack16to32(0, 0), 4);
1884  AV_WN32A(&ref_cache[0 * 8], ((LIST_NOT_USED) & 0xFF) * 0x01010101u);
1885  AV_WN32A(&ref_cache[1 * 8], ((LIST_NOT_USED) & 0xFF) * 0x01010101u);
1886  AV_WN32A(&ref_cache[2 * 8], ((LIST_NOT_USED) & 0xFF) * 0x01010101u);
1887  AV_WN32A(&ref_cache[3 * 8], ((LIST_NOT_USED) & 0xFF) * 0x01010101u);
1888  return;
1889  }
1890 
1891  {
1892  int8_t *ref = &h->cur_pic.ref_index[list][4 * mb_xy];
1893  const int *ref2frm = h->ref2frm[sl->slice_num & (MAX_SLICES - 1)][list] + (MB_MBAFF(sl) ? 20 : 2);
1894  uint32_t ref01 = (pack16to32(ref2frm[ref[0]], ref2frm[ref[1]]) & 0x00FF00FF) * 0x0101;
1895  uint32_t ref23 = (pack16to32(ref2frm[ref[2]], ref2frm[ref[3]]) & 0x00FF00FF) * 0x0101;
1896  AV_WN32A(&ref_cache[0 * 8], ref01);
1897  AV_WN32A(&ref_cache[1 * 8], ref01);
1898  AV_WN32A(&ref_cache[2 * 8], ref23);
1899  AV_WN32A(&ref_cache[3 * 8], ref23);
1900  }
1901 
1902  {
1903  int16_t(*mv_src)[2] = &h->cur_pic.motion_val[list][4 * sl->mb_x + 4 * sl->mb_y * b_stride];
1904  AV_COPY128(mv_dst + 8 * 0, mv_src + 0 * b_stride);
1905  AV_COPY128(mv_dst + 8 * 1, mv_src + 1 * b_stride);
1906  AV_COPY128(mv_dst + 8 * 2, mv_src + 2 * b_stride);
1907  AV_COPY128(mv_dst + 8 * 3, mv_src + 3 * b_stride);
1908  }
1909 }
1910 
1911 /**
1912  * @return non zero if the loop filter can be skipped
1913  */
1914 static int fill_filter_caches(const H264Context *h, H264SliceContext *sl, int mb_type)
1915 {
1916  const int mb_xy = sl->mb_xy;
1917  int top_xy, left_xy[LEFT_MBS];
1918  int top_type, left_type[LEFT_MBS];
1919  uint8_t *nnz;
1920  uint8_t *nnz_cache;
1921 
1922  top_xy = mb_xy - (h->mb_stride << MB_FIELD(sl));
1923 
1924  left_xy[LBOT] = left_xy[LTOP] = mb_xy - 1;
1925  if (FRAME_MBAFF(h)) {
1926  const int left_mb_field_flag = IS_INTERLACED(h->cur_pic.mb_type[mb_xy - 1]);
1927  const int curr_mb_field_flag = IS_INTERLACED(mb_type);
1928  if (sl->mb_y & 1) {
1929  if (left_mb_field_flag != curr_mb_field_flag)
1930  left_xy[LTOP] -= h->mb_stride;
1931  } else {
1932  if (curr_mb_field_flag)
1933  top_xy += h->mb_stride &
1934  (((h->cur_pic.mb_type[top_xy] >> 7) & 1) - 1);
1935  if (left_mb_field_flag != curr_mb_field_flag)
1936  left_xy[LBOT] += h->mb_stride;
1937  }
1938  }
1939 
1940  sl->top_mb_xy = top_xy;
1941  sl->left_mb_xy[LTOP] = left_xy[LTOP];
1942  sl->left_mb_xy[LBOT] = left_xy[LBOT];
1943  {
1944  /* For sufficiently low qp, filtering wouldn't do anything.
1945  * This is a conservative estimate: could also check beta_offset
1946  * and more accurate chroma_qp. */
1947  int qp_thresh = sl->qp_thresh; // FIXME strictly we should store qp_thresh for each mb of a slice
1948  int qp = h->cur_pic.qscale_table[mb_xy];
1949  if (qp <= qp_thresh &&
1950  (left_xy[LTOP] < 0 ||
1951  ((qp + h->cur_pic.qscale_table[left_xy[LTOP]] + 1) >> 1) <= qp_thresh) &&
1952  (top_xy < 0 ||
1953  ((qp + h->cur_pic.qscale_table[top_xy] + 1) >> 1) <= qp_thresh)) {
1954  if (!FRAME_MBAFF(h))
1955  return 1;
1956  if ((left_xy[LTOP] < 0 ||
1957  ((qp + h->cur_pic.qscale_table[left_xy[LBOT]] + 1) >> 1) <= qp_thresh) &&
1958  (top_xy < h->mb_stride ||
1959  ((qp + h->cur_pic.qscale_table[top_xy - h->mb_stride] + 1) >> 1) <= qp_thresh))
1960  return 1;
1961  }
1962  }
1963 
1964  top_type = h->cur_pic.mb_type[top_xy];
1965  left_type[LTOP] = h->cur_pic.mb_type[left_xy[LTOP]];
1966  left_type[LBOT] = h->cur_pic.mb_type[left_xy[LBOT]];
1967  if (sl->deblocking_filter == 2) {
1968  if (h->slice_table[top_xy] != sl->slice_num)
1969  top_type = 0;
1970  if (h->slice_table[left_xy[LBOT]] != sl->slice_num)
1971  left_type[LTOP] = left_type[LBOT] = 0;
1972  } else {
1973  if (h->slice_table[top_xy] == 0xFFFF)
1974  top_type = 0;
1975  if (h->slice_table[left_xy[LBOT]] == 0xFFFF)
1976  left_type[LTOP] = left_type[LBOT] = 0;
1977  }
1978  sl->top_type = top_type;
1979  sl->left_type[LTOP] = left_type[LTOP];
1980  sl->left_type[LBOT] = left_type[LBOT];
1981 
1982  if (IS_INTRA(mb_type))
1983  return 0;
1984 
1985  fill_filter_caches_inter(h, sl, mb_type, top_xy, left_xy,
1986  top_type, left_type, mb_xy, 0);
1987  if (sl->list_count == 2)
1988  fill_filter_caches_inter(h, sl, mb_type, top_xy, left_xy,
1989  top_type, left_type, mb_xy, 1);
1990 
1991  nnz = h->non_zero_count[mb_xy];
1992  nnz_cache = sl->non_zero_count_cache;
1993  AV_COPY32(&nnz_cache[4 + 8 * 1], &nnz[0]);
1994  AV_COPY32(&nnz_cache[4 + 8 * 2], &nnz[4]);
1995  AV_COPY32(&nnz_cache[4 + 8 * 3], &nnz[8]);
1996  AV_COPY32(&nnz_cache[4 + 8 * 4], &nnz[12]);
1997  sl->cbp = h->cbp_table[mb_xy];
1998 
1999  if (top_type) {
2000  nnz = h->non_zero_count[top_xy];
2001  AV_COPY32(&nnz_cache[4 + 8 * 0], &nnz[3 * 4]);
2002  }
2003 
2004  if (left_type[LTOP]) {
2005  nnz = h->non_zero_count[left_xy[LTOP]];
2006  nnz_cache[3 + 8 * 1] = nnz[3 + 0 * 4];
2007  nnz_cache[3 + 8 * 2] = nnz[3 + 1 * 4];
2008  nnz_cache[3 + 8 * 3] = nnz[3 + 2 * 4];
2009  nnz_cache[3 + 8 * 4] = nnz[3 + 3 * 4];
2010  }
2011 
2012  /* CAVLC 8x8dct requires NNZ values for residual decoding that differ
2013  * from what the loop filter needs */
2014  if (!CABAC(h) && h->ps.pps->transform_8x8_mode) {
2015  if (IS_8x8DCT(top_type)) {
2016  nnz_cache[4 + 8 * 0] =
2017  nnz_cache[5 + 8 * 0] = (h->cbp_table[top_xy] & 0x4000) >> 12;
2018  nnz_cache[6 + 8 * 0] =
2019  nnz_cache[7 + 8 * 0] = (h->cbp_table[top_xy] & 0x8000) >> 12;
2020  }
2021  if (IS_8x8DCT(left_type[LTOP])) {
2022  nnz_cache[3 + 8 * 1] =
2023  nnz_cache[3 + 8 * 2] = (h->cbp_table[left_xy[LTOP]] & 0x2000) >> 12; // FIXME check MBAFF
2024  }
2025  if (IS_8x8DCT(left_type[LBOT])) {
2026  nnz_cache[3 + 8 * 3] =
2027  nnz_cache[3 + 8 * 4] = (h->cbp_table[left_xy[LBOT]] & 0x8000) >> 12; // FIXME check MBAFF
2028  }
2029 
2030  if (IS_8x8DCT(mb_type)) {
2031  nnz_cache[scan8[0]] =
2032  nnz_cache[scan8[1]] =
2033  nnz_cache[scan8[2]] =
2034  nnz_cache[scan8[3]] = (sl->cbp & 0x1000) >> 12;
2035 
2036  nnz_cache[scan8[0 + 4]] =
2037  nnz_cache[scan8[1 + 4]] =
2038  nnz_cache[scan8[2 + 4]] =
2039  nnz_cache[scan8[3 + 4]] = (sl->cbp & 0x2000) >> 12;
2040 
2041  nnz_cache[scan8[0 + 8]] =
2042  nnz_cache[scan8[1 + 8]] =
2043  nnz_cache[scan8[2 + 8]] =
2044  nnz_cache[scan8[3 + 8]] = (sl->cbp & 0x4000) >> 12;
2045 
2046  nnz_cache[scan8[0 + 12]] =
2047  nnz_cache[scan8[1 + 12]] =
2048  nnz_cache[scan8[2 + 12]] =
2049  nnz_cache[scan8[3 + 12]] = (sl->cbp & 0x8000) >> 12;
2050  }
2051  }
2052 
2053  return 0;
2054 }
2055 
2056 static void loop_filter(const H264Context *h, H264SliceContext *sl, int start_x, int end_x)
2057 {
2058  uint8_t *dest_y, *dest_cb, *dest_cr;
2059  int linesize, uvlinesize, mb_x, mb_y;
2060  const int end_mb_y = sl->mb_y + FRAME_MBAFF(h);
2061  const int old_slice_type = sl->slice_type;
2062  const int pixel_shift = h->pixel_shift;
2063  const int block_h = 16 >> h->chroma_y_shift;
2064 
2065  if (h->postpone_filter)
2066  return;
2067 
2068  if (sl->deblocking_filter) {
2069  for (mb_x = start_x; mb_x < end_x; mb_x++)
2070  for (mb_y = end_mb_y - FRAME_MBAFF(h); mb_y <= end_mb_y; mb_y++) {
2071  int mb_xy, mb_type;
2072  mb_xy = sl->mb_xy = mb_x + mb_y * h->mb_stride;
2073  mb_type = h->cur_pic.mb_type[mb_xy];
2074 
2075  if (FRAME_MBAFF(h))
2076  sl->mb_mbaff =
2077  sl->mb_field_decoding_flag = !!IS_INTERLACED(mb_type);
2078 
2079  sl->mb_x = mb_x;
2080  sl->mb_y = mb_y;
2081  dest_y = h->cur_pic.f->data[0] +
2082  ((mb_x << pixel_shift) + mb_y * sl->linesize) * 16;
2083  dest_cb = h->cur_pic.f->data[1] +
2084  (mb_x << pixel_shift) * (8 << CHROMA444(h)) +
2085  mb_y * sl->uvlinesize * block_h;
2086  dest_cr = h->cur_pic.f->data[2] +
2087  (mb_x << pixel_shift) * (8 << CHROMA444(h)) +
2088  mb_y * sl->uvlinesize * block_h;
2089  // FIXME simplify above
2090 
2091  if (MB_FIELD(sl)) {
2092  linesize = sl->mb_linesize = sl->linesize * 2;
2093  uvlinesize = sl->mb_uvlinesize = sl->uvlinesize * 2;
2094  if (mb_y & 1) { // FIXME move out of this function?
2095  dest_y -= sl->linesize * 15;
2096  dest_cb -= sl->uvlinesize * (block_h - 1);
2097  dest_cr -= sl->uvlinesize * (block_h - 1);
2098  }
2099  } else {
2100  linesize = sl->mb_linesize = sl->linesize;
2101  uvlinesize = sl->mb_uvlinesize = sl->uvlinesize;
2102  }
2103  backup_mb_border(h, sl, dest_y, dest_cb, dest_cr, linesize,
2104  uvlinesize, 0);
2105  if (fill_filter_caches(h, sl, mb_type))
2106  continue;
2107  sl->chroma_qp[0] = get_chroma_qp(h, 0, h->cur_pic.qscale_table[mb_xy]);
2108  sl->chroma_qp[1] = get_chroma_qp(h, 1, h->cur_pic.qscale_table[mb_xy]);
2109 
2110  if (FRAME_MBAFF(h)) {
2111  ff_h264_filter_mb(h, sl, mb_x, mb_y, dest_y, dest_cb, dest_cr,
2112  linesize, uvlinesize);
2113  } else {
2114  ff_h264_filter_mb_fast(h, sl, mb_x, mb_y, dest_y, dest_cb,
2115  dest_cr, linesize, uvlinesize);
2116  }
2117  }
2118  }
2119  sl->slice_type = old_slice_type;
2120  sl->mb_x = end_x;
2121  sl->mb_y = end_mb_y - FRAME_MBAFF(h);
2122  sl->chroma_qp[0] = get_chroma_qp(h, 0, sl->qscale);
2123  sl->chroma_qp[1] = get_chroma_qp(h, 1, sl->qscale);
2124 }
2125 
2127 {
2128  const int mb_xy = sl->mb_x + sl->mb_y * h->mb_stride;
2129  int mb_type = (h->slice_table[mb_xy - 1] == sl->slice_num) ?
2130  h->cur_pic.mb_type[mb_xy - 1] :
2131  (h->slice_table[mb_xy - h->mb_stride] == sl->slice_num) ?
2132  h->cur_pic.mb_type[mb_xy - h->mb_stride] : 0;
2133  sl->mb_mbaff = sl->mb_field_decoding_flag = IS_INTERLACED(mb_type) ? 1 : 0;
2134 }
2135 
2136 /**
2137  * Draw edges and report progress for the last MB row.
2138  */
2140 {
2141  int top = 16 * (sl->mb_y >> FIELD_PICTURE(h));
2142  int pic_height = 16 * h->mb_height >> FIELD_PICTURE(h);
2143  int height = 16 << FRAME_MBAFF(h);
2144  int deblock_border = (16 + 4) << FRAME_MBAFF(h);
2145 
2146  if (sl->deblocking_filter) {
2147  if ((top + height) >= pic_height)
2148  height += deblock_border;
2149  top -= deblock_border;
2150  }
2151 
2152  if (top >= pic_height || (top + height) < 0)
2153  return;
2154 
2155  height = FFMIN(height, pic_height - top);
2156  if (top < 0) {
2157  height = top + height;
2158  top = 0;
2159  }
2160 
2161  ff_h264_draw_horiz_band(h, sl, top, height);
2162 
2163  if (h->droppable || sl->h264->slice_ctx[0].er.error_occurred)
2164  return;
2165 
2166  ff_thread_report_progress(&h->cur_pic_ptr->tf, top + height - 1,
2168 }
2169 
2171  int startx, int starty,
2172  int endx, int endy, int status)
2173 {
2174  if (!sl->h264->enable_er)
2175  return;
2176 
2177  if (CONFIG_ERROR_RESILIENCE) {
2178  ERContext *er = &sl->h264->slice_ctx[0].er;
2179 
2180  ff_er_add_slice(er, startx, starty, endx, endy, status);
2181  }
2182 }
2183 
2184 static int decode_slice(struct AVCodecContext *avctx, void *arg)
2185 {
2186  H264SliceContext *sl = arg;
2187  const H264Context *h = sl->h264;
2188  int lf_x_start = sl->mb_x;
2189  int orig_deblock = sl->deblocking_filter;
2190  int ret;
2191 
2192  sl->linesize = h->cur_pic_ptr->f->linesize[0];
2193  sl->uvlinesize = h->cur_pic_ptr->f->linesize[1];
2194 
2195  ret = alloc_scratch_buffers(sl, sl->linesize);
2196  if (ret < 0)
2197  return ret;
2198 
2199  sl->mb_skip_run = -1;
2200 
2201  av_assert0(h->block_offset[15] == (4 * ((scan8[15] - scan8[0]) & 7) << h->pixel_shift) + 4 * sl->linesize * ((scan8[15] - scan8[0]) >> 3));
2202 
2203  if (h->postpone_filter)
2204  sl->deblocking_filter = 0;
2205 
2206  sl->is_complex = FRAME_MBAFF(h) || h->picture_structure != PICT_FRAME ||
2207  avctx->codec_id != AV_CODEC_ID_H264 ||
2208  (CONFIG_GRAY && (h->flags & AV_CODEC_FLAG_GRAY));
2209 
2211  const int start_i = av_clip(sl->resync_mb_x + sl->resync_mb_y * h->mb_width, 0, h->mb_num - 1);
2212  if (start_i) {
2213  int prev_status = h->slice_ctx[0].er.error_status_table[h->slice_ctx[0].er.mb_index2xy[start_i - 1]];
2214  prev_status &= ~ VP_START;
2215  if (prev_status != (ER_MV_END | ER_DC_END | ER_AC_END))
2216  h->slice_ctx[0].er.error_occurred = 1;
2217  }
2218  }
2219 
2220  if (h->ps.pps->cabac) {
2221  /* realign */
2222  align_get_bits(&sl->gb);
2223 
2224  /* init cabac */
2225  ret = ff_init_cabac_decoder(&sl->cabac,
2226  sl->gb.buffer + get_bits_count(&sl->gb) / 8,
2227  (get_bits_left(&sl->gb) + 7) / 8);
2228  if (ret < 0)
2229  return ret;
2230 
2232 
2233  for (;;) {
2234  // START_TIMER
2235  int ret, eos;
2236  if (sl->mb_x + sl->mb_y * h->mb_width >= sl->next_slice_idx) {
2237  av_log(h->avctx, AV_LOG_ERROR, "Slice overlaps with next at %d\n",
2238  sl->next_slice_idx);
2239  er_add_slice(sl, sl->resync_mb_x, sl->resync_mb_y, sl->mb_x,
2240  sl->mb_y, ER_MB_ERROR);
2241  return AVERROR_INVALIDDATA;
2242  }
2243 
2244  ret = ff_h264_decode_mb_cabac(h, sl);
2245  // STOP_TIMER("decode_mb_cabac")
2246 
2247  if (ret >= 0)
2248  ff_h264_hl_decode_mb(h, sl);
2249 
2250  // FIXME optimal? or let mb_decode decode 16x32 ?
2251  if (ret >= 0 && FRAME_MBAFF(h)) {
2252  sl->mb_y++;
2253 
2254  ret = ff_h264_decode_mb_cabac(h, sl);
2255 
2256  if (ret >= 0)
2257  ff_h264_hl_decode_mb(h, sl);
2258  sl->mb_y--;
2259  }
2260  eos = get_cabac_terminate(&sl->cabac);
2261 
2262  if ((h->workaround_bugs & FF_BUG_TRUNCATED) &&
2263  sl->cabac.bytestream > sl->cabac.bytestream_end + 2) {
2264  er_add_slice(sl, sl->resync_mb_x, sl->resync_mb_y, sl->mb_x - 1,
2265  sl->mb_y, ER_MB_END);
2266  if (sl->mb_x >= lf_x_start)
2267  loop_filter(h, sl, lf_x_start, sl->mb_x + 1);
2268  goto finish;
2269  }
2270  if (sl->cabac.bytestream > sl->cabac.bytestream_end + 2 )
2271  av_log(h->avctx, AV_LOG_DEBUG, "bytestream overread %"PTRDIFF_SPECIFIER"\n", sl->cabac.bytestream_end - sl->cabac.bytestream);
2272  if (ret < 0 || sl->cabac.bytestream > sl->cabac.bytestream_end + 4) {
2274  "error while decoding MB %d %d, bytestream %"PTRDIFF_SPECIFIER"\n",
2275  sl->mb_x, sl->mb_y,
2276  sl->cabac.bytestream_end - sl->cabac.bytestream);
2277  er_add_slice(sl, sl->resync_mb_x, sl->resync_mb_y, sl->mb_x,
2278  sl->mb_y, ER_MB_ERROR);
2279  return AVERROR_INVALIDDATA;
2280  }
2281 
2282  if (++sl->mb_x >= h->mb_width) {
2283  loop_filter(h, sl, lf_x_start, sl->mb_x);
2284  sl->mb_x = lf_x_start = 0;
2285  decode_finish_row(h, sl);
2286  ++sl->mb_y;
2287  if (FIELD_OR_MBAFF_PICTURE(h)) {
2288  ++sl->mb_y;
2289  if (FRAME_MBAFF(h) && sl->mb_y < h->mb_height)
2291  }
2292  }
2293 
2294  if (eos || sl->mb_y >= h->mb_height) {
2295  ff_tlog(h->avctx, "slice end %d %d\n",
2296  get_bits_count(&sl->gb), sl->gb.size_in_bits);
2297  er_add_slice(sl, sl->resync_mb_x, sl->resync_mb_y, sl->mb_x - 1,
2298  sl->mb_y, ER_MB_END);
2299  if (sl->mb_x > lf_x_start)
2300  loop_filter(h, sl, lf_x_start, sl->mb_x);
2301  goto finish;
2302  }
2303  }
2304  } else {
2305  for (;;) {
2306  int ret;
2307 
2308  if (sl->mb_x + sl->mb_y * h->mb_width >= sl->next_slice_idx) {
2309  av_log(h->avctx, AV_LOG_ERROR, "Slice overlaps with next at %d\n",
2310  sl->next_slice_idx);
2311  er_add_slice(sl, sl->resync_mb_x, sl->resync_mb_y, sl->mb_x,
2312  sl->mb_y, ER_MB_ERROR);
2313  return AVERROR_INVALIDDATA;
2314  }
2315 
2316  ret = ff_h264_decode_mb_cavlc(h, sl);
2317 
2318  if (ret >= 0)
2319  ff_h264_hl_decode_mb(h, sl);
2320 
2321  // FIXME optimal? or let mb_decode decode 16x32 ?
2322  if (ret >= 0 && FRAME_MBAFF(h)) {
2323  sl->mb_y++;
2324  ret = ff_h264_decode_mb_cavlc(h, sl);
2325 
2326  if (ret >= 0)
2327  ff_h264_hl_decode_mb(h, sl);
2328  sl->mb_y--;
2329  }
2330 
2331  if (ret < 0) {
2333  "error while decoding MB %d %d\n", sl->mb_x, sl->mb_y);
2334  er_add_slice(sl, sl->resync_mb_x, sl->resync_mb_y, sl->mb_x,
2335  sl->mb_y, ER_MB_ERROR);
2336  return ret;
2337  }
2338 
2339  if (++sl->mb_x >= h->mb_width) {
2340  loop_filter(h, sl, lf_x_start, sl->mb_x);
2341  sl->mb_x = lf_x_start = 0;
2342  decode_finish_row(h, sl);
2343  ++sl->mb_y;
2344  if (FIELD_OR_MBAFF_PICTURE(h)) {
2345  ++sl->mb_y;
2346  if (FRAME_MBAFF(h) && sl->mb_y < h->mb_height)
2348  }
2349  if (sl->mb_y >= h->mb_height) {
2350  ff_tlog(h->avctx, "slice end %d %d\n",
2351  get_bits_count(&sl->gb), sl->gb.size_in_bits);
2352 
2353  if ( get_bits_left(&sl->gb) == 0
2354  || get_bits_left(&sl->gb) > 0 && !(h->avctx->err_recognition & AV_EF_AGGRESSIVE)) {
2355  er_add_slice(sl, sl->resync_mb_x, sl->resync_mb_y,
2356  sl->mb_x - 1, sl->mb_y, ER_MB_END);
2357 
2358  goto finish;
2359  } else {
2360  er_add_slice(sl, sl->resync_mb_x, sl->resync_mb_y,
2361  sl->mb_x, sl->mb_y, ER_MB_END);
2362 
2363  return AVERROR_INVALIDDATA;
2364  }
2365  }
2366  }
2367 
2368  if (get_bits_left(&sl->gb) <= 0 && sl->mb_skip_run <= 0) {
2369  ff_tlog(h->avctx, "slice end %d %d\n",
2370  get_bits_count(&sl->gb), sl->gb.size_in_bits);
2371 
2372  if (get_bits_left(&sl->gb) == 0) {
2373  er_add_slice(sl, sl->resync_mb_x, sl->resync_mb_y,
2374  sl->mb_x - 1, sl->mb_y, ER_MB_END);
2375  if (sl->mb_x > lf_x_start)
2376  loop_filter(h, sl, lf_x_start, sl->mb_x);
2377 
2378  goto finish;
2379  } else {
2380  er_add_slice(sl, sl->resync_mb_x, sl->resync_mb_y, sl->mb_x,
2381  sl->mb_y, ER_MB_ERROR);
2382 
2383  return AVERROR_INVALIDDATA;
2384  }
2385  }
2386  }
2387  }
2388 
2389 finish:
2390  sl->deblocking_filter = orig_deblock;
2391  return 0;
2392 }
2393 
2394 /**
2395  * Call decode_slice() for each context.
2396  *
2397  * @param h h264 master context
2398  * @param context_count number of contexts to execute
2399  */
2400 int ff_h264_execute_decode_slices(H264Context *h, unsigned context_count)
2401 {
2402  AVCodecContext *const avctx = h->avctx;
2403  H264SliceContext *sl;
2404  int i, j;
2405 
2406  av_assert0(context_count && h->slice_ctx[context_count - 1].mb_y < h->mb_height);
2407 
2408  h->slice_ctx[0].next_slice_idx = INT_MAX;
2409 
2410  if (h->avctx->hwaccel
2411 #if FF_API_CAP_VDPAU
2413 #endif
2414  )
2415  return 0;
2416  if (context_count == 1) {
2417  int ret;
2418 
2419  h->slice_ctx[0].next_slice_idx = h->mb_width * h->mb_height;
2420  h->postpone_filter = 0;
2421 
2422  ret = decode_slice(avctx, &h->slice_ctx[0]);
2423  h->mb_y = h->slice_ctx[0].mb_y;
2424  return ret;
2425  } else {
2426  av_assert0(context_count > 0);
2427  for (i = 0; i < context_count; i++) {
2428  int next_slice_idx = h->mb_width * h->mb_height;
2429  int slice_idx;
2430 
2431  sl = &h->slice_ctx[i];
2432  if (CONFIG_ERROR_RESILIENCE) {
2433  sl->er.error_count = 0;
2434  }
2435 
2436  /* make sure none of those slices overlap */
2437  slice_idx = sl->mb_y * h->mb_width + sl->mb_x;
2438  for (j = 0; j < context_count; j++) {
2439  H264SliceContext *sl2 = &h->slice_ctx[j];
2440  int slice_idx2 = sl2->mb_y * h->mb_width + sl2->mb_x;
2441 
2442  if (i == j || slice_idx2 < slice_idx)
2443  continue;
2444  next_slice_idx = FFMIN(next_slice_idx, slice_idx2);
2445  }
2446  sl->next_slice_idx = next_slice_idx;
2447  }
2448 
2449  avctx->execute(avctx, decode_slice, h->slice_ctx,
2450  NULL, context_count, sizeof(h->slice_ctx[0]));
2451 
2452  /* pull back stuff from slices to master context */
2453  sl = &h->slice_ctx[context_count - 1];
2454  h->mb_y = sl->mb_y;
2455  if (CONFIG_ERROR_RESILIENCE) {
2456  for (i = 1; i < context_count; i++)
2458  }
2459 
2460  if (h->postpone_filter) {
2461  h->postpone_filter = 0;
2462 
2463  for (i = 0; i < context_count; i++) {
2464  int y_end, x_end;
2465 
2466  sl = &h->slice_ctx[i];
2467  y_end = FFMIN(sl->mb_y + 1, h->mb_height);
2468  x_end = (sl->mb_y >= h->mb_height) ? h->mb_width : sl->mb_x;
2469 
2470  for (j = sl->resync_mb_y; j < y_end; j += 1 + FIELD_OR_MBAFF_PICTURE(h)) {
2471  sl->mb_y = j;
2472  loop_filter(h, sl, j > sl->resync_mb_y ? 0 : sl->resync_mb_x,
2473  j == y_end - 1 ? x_end : h->mb_width);
2474  }
2475  }
2476  }
2477  }
2478 
2479  return 0;
2480 }
int chroma_format_idc
Definition: h264.h:140
int video_signal_type_present_flag
Definition: h264.h:165
struct H264Context * h264
Definition: h264.h:317
#define AV_EF_AGGRESSIVE
consider things that a sane encoder should not do as an error
Definition: avcodec.h:2952
#define ff_tlog(ctx,...)
Definition: internal.h:65
void ff_h264_unref_picture(H264Context *h, H264Picture *pic)
Definition: h264_picture.c:47
av_cold void ff_videodsp_init(VideoDSPContext *ctx, int bpc)
Definition: videodsp.c:38
#define NULL
Definition: coverity.c:32
int ff_thread_can_start_frame(AVCodecContext *avctx)
const struct AVCodec * codec
Definition: avcodec.h:1658
AVRational framerate
Definition: avcodec.h:3338
discard all frames except keyframes
Definition: avcodec.h:783
void ff_h264_flush_change(H264Context *h)
Definition: h264.c:756
int workaround_bugs
Definition: h264.h:494
int long_ref
1->long term reference 0->short term reference
Definition: h264.h:289
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:59
int single_decode_warning
1 if the single thread fallback warning has already been displayed, 0 otherwise.
Definition: h264.h:629
int sei_recovery_frame_cnt
Definition: h264.h:298
enum AVPixelFormat backup_pix_fmt
Definition: h264.h:487
H264POCContext poc
Definition: h264.h:578
int mb_num
Definition: h264.h:557
void av_buffer_unref(AVBufferRef **buf)
Free a given reference and automatically free the buffer if there are no more references to it...
Definition: buffer.c:124
int ff_h264_decode_ref_pic_list_reordering(H264Context *h, H264SliceContext *sl)
Definition: h264_refs.c:273
int mb_aff_frame
Definition: h264.h:527
int recovery_frame_cnt
recovery_frame_cnt
Definition: h264_sei.h:109
int16_t mv_cache[2][5 *8][2]
Motion vector cache.
Definition: h264.h:429
#define AV_PIX_FMT_YUV444P14
Definition: pixfmt.h:351
static int get_se_golomb(GetBitContext *gb)
read signed exp golomb code.
Definition: golomb.h:183
#define CHROMA444(h)
Definition: h264.h:103
#define LEFT_MBS
Definition: h264.h:79
int edge_emu_buffer_allocated
Definition: h264.h:417
int coded_width
Bitstream width / height, may be different from width/height e.g.
Definition: avcodec.h:1851
static void decode_finish_row(const H264Context *h, H264SliceContext *sl)
Draw edges and report progress for the last MB row.
Definition: h264_slice.c:2139
const char * fmt
Definition: avisynth_c.h:632
#define H264_MAX_PICTURE_COUNT
Definition: h264.h:51
int first_field
Definition: h264.h:529
int ff_h264_decode_mb_cavlc(const H264Context *h, H264SliceContext *sl)
Decode a macroblock.
Definition: h264_cavlc.c:702
uint8_t field_scan8x8_q0[64]
Definition: h264.h:551
planar YUV 4:4:4, 24bpp, (1 Cr & Cb sample per 1x1 Y samples)
Definition: pixfmt.h:67
misc image utilities
static unsigned int get_bits(GetBitContext *s, int n)
Read 1-25 bits.
Definition: get_bits.h:247
#define AV_LOG_WARNING
Something somehow does not look correct.
Definition: log.h:182
#define ER_MB_END
AVFrame * f
Definition: thread.h:36
int weighted_bipred_idc
Definition: h264.h:208
int left_mb_xy[LEFT_MBS]
Definition: h264.h:350
int chroma_qp_index_offset[2]
Definition: h264.h:211
AVBufferRef * sps_list[MAX_SPS_COUNT]
Definition: h264.h:230
const uint8_t * bytestream_end
Definition: cabac.h:49
AVBufferRef * buf[AV_NUM_DATA_POINTERS]
AVBuffer references backing the data for this frame.
Definition: frame.h:363
hardware decoding through Videotoolbox
Definition: pixfmt.h:295
H264ChromaContext h264chroma
Definition: h264.h:461
uint16_t * cbp_table
Definition: h264.h:534
int luma_weight_flag[2]
7.4.3.2 luma_weight_lX_flag
Definition: h264_parse.h:34
MMCO mmco[MAX_MMCO_COUNT]
memory management control operations buffer.
Definition: h264.h:601
static void implicit_weight_table(const H264Context *h, H264SliceContext *sl, int field)
Initialize implicit_weight table.
Definition: h264_slice.c:663
#define MAX_PPS_COUNT
Definition: h264.h:54
Sequence parameter set.
Definition: h264.h:136
enum AVColorRange color_range
MPEG vs JPEG YUV range.
Definition: avcodec.h:2385
int mb_y
Definition: h264.h:554
int coded_picture_number
Definition: h264.h:490
planar GBR 4:4:4 24bpp
Definition: pixfmt.h:180
int num
numerator
Definition: rational.h:44
AVBufferRef * mb_type_buf
Definition: h264.h:273
int bipred_scratchpad_allocated
Definition: h264.h:416
#define DELAYED_PIC_REF
Value of Picture.reference when Picture is not a reference picture, but is held for delayed output...
Definition: diracdec.c:64
void ff_h264_direct_ref_list_init(const H264Context *const h, H264SliceContext *sl)
Definition: h264_direct.c:110
#define AV_PIX_FMT_GBRP10
Definition: pixfmt.h:357
#define VP_START
< current MB is the first after a resync marker
AVBufferPool * mb_type_pool
Definition: h264.h:686
int ff_h264_init_poc(int pic_field_poc[2], int *pic_poc, const SPS *sps, H264POCContext *pc, int picture_structure, int nal_ref_idc)
Definition: h264_parse.c:244
int chroma_x_shift
Definition: h264.h:479
void ff_h264_filter_mb_fast(const H264Context *h, H264SliceContext *sl, int mb_x, int mb_y, uint8_t *img_y, uint8_t *img_cb, uint8_t *img_cr, unsigned int linesize, unsigned int uvlinesize)
const uint8_t * buffer
Definition: get_bits.h:56
Picture parameter set.
Definition: h264.h:200
int crop
Definition: h264.h:300
void ff_h264_draw_horiz_band(const H264Context *h, H264SliceContext *sl, int y, int height)
Definition: h264.c:102
AVRational sample_aspect_ratio
sample aspect ratio (0 if unknown) That is the width of a pixel divided by the height of the pixel...
Definition: avcodec.h:2060
int16_t(*[2] motion_val)[2]
Definition: h264.h:271
int flags
Definition: h264.h:493
enum AVPixelFormat pix_fmt
Pixel format, see AV_PIX_FMT_xxx.
Definition: avcodec.h:1877
int frame_mbs_only_flag
Definition: h264.h:153
int mb_height
Definition: h264.h:555
H264Picture * delayed_pic[MAX_DELAYED_PIC_COUNT+2]
Definition: h264.h:593
static int av_cmp_q(AVRational a, AVRational b)
Compare two rationals.
Definition: rational.h:66
int is_avc
Used to parse AVC variant of H.264.
Definition: h264.h:568
int mmco_index
Definition: h264.h:602
H264Ref default_ref[2]
Definition: h264.h:590
av_cold void ff_h264_pred_init(H264PredContext *h, int codec_id, const int bit_depth, int chroma_format_idc)
Set the intra prediction function pointers.
Definition: h264pred.c:411
AVBufferPool * ref_index_pool
Definition: h264.h:688
void ff_h264_free_tables(H264Context *h)
Definition: h264.c:137
uint8_t zigzag_scan8x8_cavlc[64]
Definition: h264.h:543
#define AV_PIX_FMT_YUV420P12
Definition: pixfmt.h:345
av_cold void ff_h264chroma_init(H264ChromaContext *c, int bit_depth)
Definition: h264chroma.c:41
ERPicture last_pic
int next_slice_idx
Definition: h264.h:374
static const uint8_t zigzag_scan8x8_cavlc[64+1]
Definition: h264_slice.c:92
H264Context.
Definition: h264.h:456
discard all non intra frames
Definition: avcodec.h:782
discard all
Definition: avcodec.h:784
AVFrame * f
Definition: h264.h:264
int bits_per_raw_sample
Bits per sample/pixel of internal libavcodec pixel/sample format.
Definition: avcodec.h:3049
uint32_t num_units_in_tick
Definition: h264.h:172
void ff_h264_filter_mb(const H264Context *h, H264SliceContext *sl, int mb_x, int mb_y, uint8_t *img_y, uint8_t *img_cb, uint8_t *img_cr, unsigned int linesize, unsigned int uvlinesize)
static const uint8_t field_scan[16+1]
Definition: h264_slice.c:46
H264Picture * long_ref[32]
Definition: h264.h:592
int profile
profile
Definition: avcodec.h:3153
int picture_structure
Definition: h264.h:528
order of coefficients is actually GBR, also IEC 61966-2-1 (sRGB)
Definition: pixfmt.h:435
#define AV_WN32A(p, v)
Definition: intreadwrite.h:538
#define AV_COPY32(d, s)
Definition: intreadwrite.h:586
unsigned current_sps_id
id of the current SPS
Definition: h264.h:520
unsigned int ref_count[2]
num_ref_idx_l0/1_active_minus1 + 1
Definition: h264.h:404
#define IN_RANGE(a, b, size)
Definition: h264_slice.c:274
#define REBASE_PICTURE(pic, new_ctx, old_ctx)
Definition: h264_slice.c:276
int ff_h264_update_thread_context(AVCodecContext *dst, const AVCodecContext *src)
Definition: h264_slice.c:297
Switching Intra.
Definition: avutil.h:270
int setup_finished
Definition: h264.h:675
enum AVDiscard skip_frame
Skip decoding for selected frames.
Definition: avcodec.h:3264
H264SEIContext sei
Definition: h264.h:683
struct AVHWAccel * hwaccel
Hardware accelerator in use.
Definition: avcodec.h:2968
unsigned int crop_top
frame_cropping_rect_top_offset
Definition: h264.h:161
#define USES_LIST(a, list)
Definition: mpegutils.h:101
void ff_color_frame(AVFrame *frame, const int color[4])
Definition: utils.c:700
#define av_assert0(cond)
assert() equivalent, that is always enabled.
Definition: avassert.h:37
int is_complex
Definition: h264.h:376
#define IS_8x8DCT(a)
Definition: h264.h:110
const uint8_t * bytestream
Definition: cabac.h:48
int ref2frm[MAX_SLICES][2][64]
reference to frame number lists, used in the loop filter, the first 2 are for -2,-1 ...
Definition: h264.h:689
int deblocking_filter_parameters_present
deblocking_filter_parameters_present_flag
Definition: h264.h:212
#define MB_FIELD(sl)
Definition: h264.h:76
void void avpriv_request_sample(void *avc, const char *msg,...) av_printf_format(2
Log a generic warning message about a missing feature.
const PPS * pps
Definition: h264.h:236
static enum AVPixelFormat non_j_pixfmt(enum AVPixelFormat a)
Definition: h264_slice.c:1012
uint8_t
int full_range
Definition: h264.h:166
unsigned int crop_left
frame_cropping_rect_left_offset
Definition: h264.h:159
int gaps_in_frame_num_allowed_flag
Definition: h264.h:150
int slice_alpha_c0_offset
Definition: h264.h:333
int poc
Definition: h264.h:310
int field_picture
whether or not picture was encoded in separate fields
Definition: h264.h:293
int bit_depth_chroma
bit_depth_chroma_minus8 + 8
Definition: h264.h:190
#define FF_DEBUG_PICT_INFO
Definition: avcodec.h:2889
enum AVColorPrimaries color_primaries
Definition: h264.h:168
int poc
frame POC
Definition: h264.h:283
void ff_h264_direct_dist_scale_factor(const H264Context *const h, H264SliceContext *sl)
Definition: h264_direct.c:51
int frame_num_offset
for POC type 2
Definition: h264_parse.h:50
int slice_type
Definition: h264.h:322
int chroma_weight_flag[2]
7.4.3.2 chroma_weight_lX_flag
Definition: h264_parse.h:35
Multithreading support functions.
#define ER_MB_ERROR
int cabac
entropy_coding_mode_flag
Definition: h264.h:202
const char * from
Definition: jacosubdec.c:65
unsigned int crop_right
frame_cropping_rect_right_offset
Definition: h264.h:160
void av_fast_mallocz(void *ptr, unsigned int *size, size_t min_size)
Allocate a buffer, reusing the given one if large enough.
Definition: mem.c:504
#define AV_PIX_FMT_GBRP9
Definition: pixfmt.h:356
uint8_t(*[2] top_borders)[(16 *3)*2]
Definition: h264.h:415
int invalid_gap
Definition: h264.h:297
#define FF_API_CAP_VDPAU
Definition: version.h:73
ERPicture cur_pic
int frame_recovered
Initial frame has been completely recovered.
Definition: h264.h:665
int height
Definition: h264.h:478
#define PICT_BOTTOM_FIELD
Definition: mpegutils.h:38
#define height
int transform_bypass
qpprime_y_zero_transform_bypass_flag
Definition: h264.h:141
static void finish(void)
Definition: movenc.c:338
static int get_bits_count(const GetBitContext *s)
Definition: get_bits.h:199
char av_get_picture_type_char(enum AVPictureType pict_type)
Return a single letter to describe the given picture type pict_type.
Definition: utils.c:91
#define AV_CODEC_CAP_HWACCEL_VDPAU
Codec can export data for HW decoding (VDPAU).
Definition: avcodec.h:992
planar YUV 4:2:2, 16bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV422P and setting col...
Definition: pixfmt.h:75
#define ER_MV_END
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:225
int redundant_pic_cnt_present
redundant_pic_cnt_present_flag
Definition: h264.h:214
int chroma_y_shift
Definition: h264.h:479
#define MAX_DELAYED_PIC_COUNT
Definition: h264.h:58
static void fill_rectangle(SDL_Surface *screen, int x, int y, int w, int h, int color, int update)
Definition: ffplay.c:801
AVBufferRef * qscale_table_buf
Definition: h264.h:267
#define AV_PIX_FMT_YUV422P12
Definition: pixfmt.h:346
H264Picture * parent
Definition: h264.h:313
int ff_h264_decode_mb_cabac(const H264Context *h, H264SliceContext *sl)
Decode a CABAC coded macroblock.
Definition: h264_cabac.c:1915
#define AV_CODEC_FLAG_GRAY
Only decode/encode grayscale.
Definition: avcodec.h:858
high precision timer, useful to profile code
int recovered
picture at IDR or recovery point + recovery count
Definition: h264.h:296
#define AV_COPY64(d, s)
Definition: intreadwrite.h:590
#define FFALIGN(x, a)
Definition: macros.h:48
int chroma_qp[2]
Definition: h264.h:327
#define av_log(a,...)
AVCodecContext * owner
Definition: thread.h:37
int resync_mb_x
Definition: h264.h:371
int last_pocs[MAX_DELAYED_PIC_COUNT]
Definition: h264.h:594
const char * to
Definition: webvttdec.c:34
int width
Definition: h264.h:478
static int h264_frame_start(H264Context *h)
Definition: h264_slice.c:460
H.264 / AVC / MPEG-4 part10 codec.
void ff_h264_fill_mbaff_ref_list(H264SliceContext *sl)
Definition: h264_refs.c:404
int slice_num
Definition: h264.h:321
#define U(x)
Definition: vp56_arith.h:37
#define HWACCEL_MAX
void ff_h264_hl_decode_mb(const H264Context *h, H264SliceContext *sl)
Definition: h264_mb.c:798
static int get_bits_left(GetBitContext *gb)
Definition: get_bits.h:568
H264Picture DPB[H264_MAX_PICTURE_COUNT]
Definition: h264.h:465
int mb_aff
mb_adaptive_frame_field_flag
Definition: h264.h:154
enum AVColorTransferCharacteristic color_trc
Definition: h264.h:169
H264PredContext hpc
Definition: h264.h:502
AVBufferRef * sps_ref
Definition: h264.h:234
int chroma_log2_weight_denom
Definition: h264_parse.h:33
int width
width and height of the video frame
Definition: frame.h:236
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:176
#define td
Definition: regdef.h:70
static int get_ue_golomb(GetBitContext *gb)
Read an unsigned Exp-Golomb code in the range 0 to 8190.
Definition: golomb.h:53
static int alloc_scratch_buffers(H264SliceContext *sl, int linesize)
Definition: h264_slice.c:124
int poc_type
pic_order_cnt_type
Definition: h264.h:143
void ff_er_add_slice(ERContext *s, int startx, int starty, int endx, int endy, int status)
Add a slice.
int context_initialized
Definition: h264.h:492
#define PTRDIFF_SPECIFIER
Definition: internal.h:251
ERContext er
Definition: h264.h:319
int nal_unit_type
Definition: h264.h:563
int ff_h264_get_profile(const SPS *sps)
Compute profile from profile_idc and constraint_set?_flags.
Definition: h264_parse.c:479
discard all bidirectional frames
Definition: avcodec.h:781
#define AVERROR(e)
Definition: error.h:43
void * hwaccel_picture_private
hardware accelerator private data
Definition: h264.h:277
int backup_height
Definition: h264.h:486
int av_pix_fmt_get_chroma_sub_sample(enum AVPixelFormat pix_fmt, int *h_shift, int *v_shift)
Utility function to access log2_chroma_w log2_chroma_h from the pixel format AVPixFmtDescriptor.
Definition: pixdesc.c:2250
int active_thread_type
Which multithreading methods are in use by the codec.
Definition: avcodec.h:3098
static const uint8_t field_scan8x8[64+1]
Definition: h264_slice.c:53
int backup_width
Backup frame properties: needed, because they can be different between returned frame and last decode...
Definition: h264.h:485
int capabilities
Codec capabilities.
Definition: avcodec.h:3561
const uint8_t ff_zigzag_scan[16+1]
Definition: mathtables.c:109
#define AV_LOG_DEBUG
Stuff which is only useful for libav* developers.
Definition: log.h:197
static int init_dimensions(H264Context *h)
Definition: h264_slice.c:874
const char * arg
Definition: jacosubdec.c:66
int deblocking_filter
disable_deblocking_filter_idc with 1 <-> 0
Definition: h264.h:332
uint8_t(*[2] mvd_table)[2]
Definition: h264.h:538
int prev_interlaced_frame
Complement sei_pic_struct SEI_PIC_STRUCT_TOP_BOTTOM and SEI_PIC_STRUCT_BOTTOM_TOP indicate interlaced...
Definition: h264.h:639
static int fill_filter_caches(const H264Context *h, H264SliceContext *sl, int mb_type)
Definition: h264_slice.c:1914
ThreadFrame tf
Definition: h264.h:265
simple assert() macros that are a bit more flexible than ISO C assert().
int weighted_pred
weighted_pred_flag
Definition: h264.h:207
#define PICT_TOP_FIELD
Definition: mpegutils.h:37
H264QpelContext h264qpel
Definition: h264.h:462
#define AV_PIX_FMT_YUV444P10
Definition: pixfmt.h:344
int direct_spatial_mv_pred
Definition: h264.h:388
#define CABAC(h)
Definition: h264.h:98
H264SEIUnregistered unregistered
Definition: h264_sei.h:150
HW acceleration through VDA, data[3] contains a CVPixelBufferRef.
Definition: pixfmt.h:223
int frame_num
frame_num (raw frame_num from slice header)
Definition: h264.h:284
const uint8_t ff_h264_golomb_to_pict_type[5]
Definition: h264data.c:37
static av_always_inline uint32_t pack16to32(unsigned a, unsigned b)
Definition: h264.h:817
#define MAX_SLICES
Definition: dxva2_hevc.c:32
GLsizei count
Definition: opengl_enc.c:109
int ff_h264_get_slice_type(const H264SliceContext *sl)
Reconstruct bitstream slice_type.
Definition: h264_slice.c:1812
#define fail()
Definition: checkasm.h:81
int delta_pic_order_always_zero_flag
Definition: h264.h:145
void av_image_copy(uint8_t *dst_data[4], int dst_linesizes[4], const uint8_t *src_data[4], const int src_linesizes[4], enum AVPixelFormat pix_fmt, int width, int height)
Copy image in src_data to dst_data.
Definition: imgutils.c:302
int * mb_index2xy
#define FIELD_OR_MBAFF_PICTURE(h)
Definition: h264.h:95
int slice_type_nos
S free slice type (SI/SP are remapped to I/P)
Definition: h264.h:323
int mb_skip_run
Definition: h264.h:375
uint8_t zigzag_scan8x8[64]
Definition: h264.h:542
AVBufferRef * hwaccel_priv_buf
Definition: h264.h:276
planar YUV 4:2:2, 16bpp, (1 Cr & Cb sample per 2x1 Y samples)
Definition: pixfmt.h:66
static const uint8_t scan8[16 *3+3]
Definition: h264.h:801
int crop_left
Definition: h264.h:301
int crop
frame_cropping_flag
Definition: h264.h:156
uint8_t * error_status_table
int av_reduce(int *dst_num, int *dst_den, int64_t num, int64_t den, int64_t max)
Reduce a fraction.
Definition: rational.c:35
int ff_h264_parse_ref_count(int *plist_count, int ref_count[2], GetBitContext *gb, const PPS *pps, int slice_type_nos, int picture_structure, void *logctx)
Definition: h264_parse.c:191
int nal_length_size
Number of bytes used for nal length (1, 2 or 4)
Definition: h264.h:569
int ff_h264_pred_weight_table(GetBitContext *gb, const SPS *sps, const int *ref_count, int slice_type_nos, H264PredWeightTable *pwt, void *logctx)
Definition: h264_parse.c:25
useful rectangle filling function
#define AV_PIX_FMT_YUV422P9
Definition: pixfmt.h:339
int refs
number of reference frames
Definition: avcodec.h:2329
int prev_poc_msb
poc_msb of the last reference pic for POC type 0
Definition: h264_parse.h:48
AVBufferRef * motion_val_buf[2]
Definition: h264.h:270
int ref_frame_count
num_ref_frames
Definition: h264.h:149
enum AVPixelFormat * pix_fmts
array of supported pixel formats, or NULL if unknown, array is terminated by -1
Definition: avcodec.h:3563
enum AVPictureType pict_type
Picture type of the frame.
Definition: frame.h:258
int err_recognition
Error recognition; may misdetect some more or less valid parts as errors.
Definition: avcodec.h:2936
#define av_assert1(cond)
assert() equivalent, that does not lie in speed critical code.
Definition: avassert.h:53
#define FF_THREAD_FRAME
Decode more than one frame at once.
Definition: avcodec.h:3090
#define FFMIN(a, b)
Definition: common.h:96
uint16_t * slice_table
slice_table_base + 2*mb_stride + 1
Definition: h264.h:524
static void copy_picture_range(H264Picture **to, H264Picture **from, int count, H264Context *new_base, H264Context *old_base)
Definition: h264_slice.c:281
av_cold void ff_h264dsp_init(H264DSPContext *c, const int bit_depth, const int chroma_format_idc)
Definition: h264dsp.c:67
uint8_t field_scan8x8_cavlc[64]
Definition: h264.h:546
#define IS_DIRECT(a)
Definition: mpegutils.h:86
CABACContext cabac
Cabac.
Definition: h264.h:448
int colour_description_present_flag
Definition: h264.h:167
int reference
Definition: h264.h:295
static void er_add_slice(H264SliceContext *sl, int startx, int starty, int endx, int endy, int status)
Definition: h264_slice.c:2170
planar YUV 4:2:0, 12bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV420P and setting col...
Definition: pixfmt.h:74
AVRational sar
Definition: h264.h:164
#define width
#define FIELD_PICTURE(h)
Definition: h264.h:78
int width
picture width / height.
Definition: avcodec.h:1836
int redundant_pic_count
Definition: h264.h:381
int nb_slice_ctx
Definition: h264.h:471
H264PredWeightTable pwt
Definition: h264.h:336
int long_ref_count
number of actual long term references
Definition: h264.h:605
#define ER_DC_END
uint32_t * mb_type
Definition: h264.h:274
int ff_h264_ref_picture(H264Context *h, H264Picture *dst, H264Picture *src)
Definition: h264_picture.c:68
int top_mb_xy
Definition: h264.h:348
void ff_thread_report_progress(ThreadFrame *f, int n, int field)
Notify later decoding threads when part of their reference picture is ready.
int size_in_bits
Definition: get_bits.h:58
int ff_init_cabac_decoder(CABACContext *c, const uint8_t *buf, int buf_size)
Definition: cabac.c:177
enum AVColorPrimaries color_primaries
Chromaticity coordinates of the source primaries.
Definition: avcodec.h:2364
#define FFABS(a)
Absolute value, Note, INT_MIN / INT64_MIN result in undefined behavior as they are not representable ...
Definition: common.h:72
#define FF_THREAD_SLICE
Decode more than one part of a single frame at once.
Definition: avcodec.h:3091
Context Adaptive Binary Arithmetic Coder inline functions.
int level
level
Definition: avcodec.h:3242
int init_qp
pic_init_qp_minus26 + 26
Definition: h264.h:209
int mmco_reset
Definition: h264.h:603
H264SliceContext * slice_ctx
Definition: h264.h:470
int direct_8x8_inference_flag
Definition: h264.h:155
#define AV_EF_EXPLODE
abort decoding on minor error detection
Definition: avcodec.h:2947
int reference
Definition: h264.h:309
int ff_h264_decode_slice_header(H264Context *h, H264SliceContext *sl)
Decode a slice header.
Definition: h264_slice.c:1031
int max_pic_num
max_frame_num or 2 * max_frame_num for field pics.
Definition: h264.h:588
int ticks_per_frame
For some codecs, the time base is closer to the field rate than the frame rate.
Definition: avcodec.h:1795
int top_borders_allocated[2]
Definition: h264.h:418
int ff_generate_sliding_window_mmcos(H264Context *h, int first_slice)
Definition: h264_refs.c:587
static int pic_is_unused(H264Context *h, H264Picture *pic)
Definition: h264_slice.c:252
int curr_pic_num
frame_num for frames or 2 * frame_num + 1 for field pics.
Definition: h264.h:583
#define src
Definition: vp9dsp.c:530
static void init_scan_tables(H264Context *h)
initialize scan tables
Definition: h264_slice.c:727
static int av_unused get_cabac_terminate(CABACContext *c)
#define MB_MBAFF(h)
Definition: h264.h:75
#define AV_PIX_FMT_YUV444P9
Definition: pixfmt.h:340
#define AV_PIX_FMT_GBRP14
Definition: pixfmt.h:359
void ff_h264_init_cabac_states(const H264Context *h, H264SliceContext *sl)
Definition: h264_cabac.c:1264
#define FF_ARRAY_ELEMS(a)
the normal 2^n-1 "JPEG" YUV ranges
Definition: pixfmt.h:457
static int init_table_pools(H264Context *h)
Definition: h264_slice.c:156
uint8_t * edge_emu_buffer
Definition: h264.h:414
int top_type
Definition: h264.h:353
static unsigned get_ue_golomb_long(GetBitContext *gb)
Read an unsigned Exp-Golomb code in the range 0 to UINT32_MAX-1.
Definition: golomb.h:85
#define AVERROR_PATCHWELCOME
Not yet implemented in FFmpeg, patches welcome.
Definition: error.h:62
int pic_order_present
pic_order_present_flag
Definition: h264.h:203
uint8_t zigzag_scan_q0[16]
Definition: h264.h:547
int format
format of the frame, -1 if unknown or unset Values correspond to enum AVPixelFormat for video frames...
Definition: frame.h:248
int bit_depth_luma
luma bit depth from sps to detect changes
Definition: h264.h:571
int chroma_format_idc
chroma format from sps to detect changes
Definition: h264.h:572
VideoDSPContext vdsp
Definition: h264.h:459
int timing_info_present_flag
Definition: h264.h:171
int ff_h264_execute_ref_pic_marking(H264Context *h, MMCO *mmco, int mmco_count)
Execute the reference picture marking (memory management control operations).
Definition: h264_refs.c:619
int coded_picture_number
picture number in bitstream order
Definition: frame.h:285
int ff_h264_slice_context_init(H264Context *h, H264SliceContext *sl)
Init context Allocate buffers which are not shared amongst multiple threads.
Definition: h264.c:238
int mb_stride
Definition: h264.h:556
#define AV_LOG_INFO
Standard information.
Definition: log.h:187
int postpone_filter
Definition: h264.h:499
#define IS_INTERLACED(a)
Definition: mpegutils.h:85
AVCodecContext * avctx
Definition: h264.h:458
uint8_t zigzag_scan8x8_q0[64]
Definition: h264.h:548
#define AV_PIX_FMT_YUV420P14
Definition: pixfmt.h:349
Libavcodec external API header.
int last_qscale_diff
Definition: h264.h:329
AVBufferRef * pps_list[MAX_PPS_COUNT]
Definition: h264.h:231
enum AVCodecID codec_id
Definition: avcodec.h:1666
static int get_ue_golomb_31(GetBitContext *gb)
read unsigned exp golomb code, constraint to a max of 31.
Definition: golomb.h:100
int delta_poc_bottom
Definition: h264_parse.h:45
ERPicture next_pic
int linesize[AV_NUM_DATA_POINTERS]
For video, size in bytes of each picture line.
Definition: frame.h:215
H264Picture * short_ref[32]
Definition: h264.h:591
int next_outputed_poc
Definition: h264.h:596
#define LTOP
Definition: h264.h:80
#define AV_CODEC_FLAG2_FAST
Allow non spec compliant speedup tricks.
Definition: avcodec.h:898
int field_poc[2]
top/bottom POC
Definition: h264.h:282
int debug
debug
Definition: avcodec.h:2888
int ff_thread_get_buffer(AVCodecContext *avctx, ThreadFrame *f, int flags)
Wrapper around get_buffer() for frame-multithreaded codecs.
int max_contexts
Max number of threads / contexts.
Definition: h264.h:623
int recovery_frame
recovery_frame is the frame_num at which the next frame should be fully constructed.
Definition: h264.h:652
main external API structure.
Definition: avcodec.h:1649
int qp_thresh
QP threshold to skip loopfilter.
Definition: h264.h:328
uint8_t * data
The data buffer.
Definition: buffer.h:89
int ff_h264_alloc_tables(H264Context *h)
Allocate tables.
Definition: h264.c:180
void * buf
Definition: avisynth_c.h:553
int resync_mb_y
Definition: h264.h:372
int implicit_weight[48][48][2]
Definition: h264_parse.h:39
int8_t * qscale_table
Definition: h264.h:268
AVBufferRef * av_buffer_allocz(int size)
Same as av_buffer_alloc(), except the returned buffer will be initialized to zero.
Definition: buffer.c:82
AVBuffer * buffer
Definition: buffer.h:82
static const uint8_t field_scan8x8_cavlc[64+1]
Definition: h264_slice.c:72
static unsigned int get_bits1(GetBitContext *s)
Definition: get_bits.h:299
#define AV_PIX_FMT_YUV420P10
Definition: pixfmt.h:341
int coded_height
Definition: avcodec.h:1851
Switching Predicted.
Definition: avutil.h:271
int prev_frame_num
frame_num of the last pic for POC type 1/2
Definition: h264_parse.h:52
uint8_t non_zero_count_cache[15 *8]
non zero coeff count cache.
Definition: h264.h:424
#define CHROMA422(h)
Definition: h264.h:102
void av_buffer_pool_uninit(AVBufferPool **ppool)
Mark the pool as being available for freeing.
Definition: buffer.c:274
enum AVColorSpace colorspace
YUV colorspace type.
Definition: avcodec.h:2378
enum AVColorTransferCharacteristic color_trc
Color Transfer Characteristic.
Definition: avcodec.h:2371
HW acceleration through VDPAU, Picture.data[3] contains a VdpVideoSurface.
Definition: pixfmt.h:208
int8_t * ref_index[2]
Definition: h264.h:280
A reference counted buffer type.
int pixel_shift
0 for 8-bit H.264, 1 for high-bit-depth H.264
Definition: h264.h:475
int mmco_reset
MMCO_RESET set this 1.
Definition: h264.h:285
int ff_h264_decode_ref_pic_marking(H264Context *h, GetBitContext *gb, int first_slice)
Definition: h264_refs.c:828
H264Picture * cur_pic_ptr
Definition: h264.h:466
const uint8_t ff_zigzag_direct[64]
Definition: mathtables.c:98
ptrdiff_t mb_uvlinesize
Definition: h264.h:367
static int h264_slice_header_init(H264Context *h)
Definition: h264_slice.c:917
int mb_mbaff
mb_aff_frame && mb_field_decoding_flag
Definition: h264.h:379
enum AVPixelFormat ff_thread_get_format(AVCodecContext *avctx, const enum AVPixelFormat *fmt)
Wrapper around get_format() for frame-multithreaded codecs.
int enable_er
Definition: h264.h:681
#define AV_PIX_FMT_YUV420P9
Definition: pixfmt.h:338
#define IS_INTER(a)
Definition: mpegutils.h:81
unsigned int sps_id
Definition: h264.h:201
#define TRANSPOSE(x)
int log2_max_poc_lsb
log2_max_pic_order_cnt_lsb_minus4
Definition: h264.h:144
H264SEIRecoveryPoint recovery_point
Definition: h264_sei.h:151
ptrdiff_t mb_linesize
may be equal to s->linesize or s->linesize * 2, for mbaff
Definition: h264.h:366
HW decoding through DXVA2, Picture.data[3] contains a LPDIRECT3DSURFACE9 pointer. ...
Definition: pixfmt.h:148
int16_t slice_row[MAX_SLICES]
to detect when MAX_SLICES is too low
Definition: h264.h:679
static enum AVPixelFormat pix_fmts[]
Definition: libkvazaar.c:262
static int alloc_picture(H264Context *h, H264Picture *pic)
Definition: h264_slice.c:183
ptrdiff_t linesize
Definition: h264.h:365
int block_offset[2 *(16 *3)]
block_offset[ 0..23] for frame macroblocks block_offset[24..47] for field macroblocks ...
Definition: h264.h:513
uint32_t time_scale
Definition: h264.h:173
#define AV_PIX_FMT_YUV422P14
Definition: pixfmt.h:350
#define AV_PIX_FMT_GBRP12
Definition: pixfmt.h:358
int transform_8x8_mode
transform_8x8_mode_flag
Definition: h264.h:215
ptrdiff_t uvlinesize
Definition: h264.h:365
#define AV_PIX_FMT_YUV422P10
Definition: pixfmt.h:342
unsigned int list_count
Definition: h264.h:405
uint8_t zigzag_scan[16]
Definition: h264.h:541
#define AV_PIX_FMT_YUV444P12
Definition: pixfmt.h:348
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
Definition: frame.h:198
int prev_poc_lsb
poc_lsb of the last reference pic for POC type 0
Definition: h264_parse.h:49
static void release_unused_pictures(H264Context *h, int remove_current)
Definition: h264_slice.c:111
#define LBOT
Definition: h264.h:81
#define AV_ZERO128(d)
Definition: intreadwrite.h:622
the normal 219*2^(n-8) "MPEG" YUV ranges
Definition: pixfmt.h:456
int left_type[LEFT_MBS]
Definition: h264.h:355
hardware decoding through VDA
Definition: pixfmt.h:179
discard all non reference
Definition: avcodec.h:780
AVBufferPool * qscale_table_pool
Definition: h264.h:685
H264Picture * next_output_pic
Definition: h264.h:595
volatile int error_count
int mb_height
pic_height_in_map_units_minus1 + 1
Definition: h264.h:152
AVBufferPool * motion_val_pool
Definition: h264.h:687
void av_fast_malloc(void *ptr, unsigned int *size, size_t min_size)
Allocate a buffer, reusing the given one if large enough.
Definition: mem.c:499
planar YUV 4:2:0, 12bpp, (1 Cr & Cb sample per 2x2 Y samples)
Definition: pixfmt.h:62
#define SLICE_SINGLETHREAD
Definition: h264.h:993
common internal api header.
if(ret< 0)
Definition: vf_mcdeint.c:282
AVBufferPool * av_buffer_pool_init(int size, AVBufferRef *(*alloc)(int size))
Allocate and initialize a buffer pool.
Definition: buffer.c:237
static int ref[MAX_W *MAX_W]
Definition: jpeg2000dwt.c:107
#define AV_COPY128(d, s)
Definition: intreadwrite.h:594
static enum AVPixelFormat get_pixel_format(H264Context *h, int force_callback)
Definition: h264_slice.c:761
AVBufferRef * pps_ref
Definition: h264.h:233
int log2_max_frame_num
log2_max_frame_num_minus4 + 4
Definition: h264.h:142
int missing_fields
Definition: h264.h:669
static double c[64]
H264ParamSets ps
Definition: h264.h:574
H.264 / AVC / MPEG-4 part10 motion vector prediction.
AVBufferRef * av_buffer_ref(AVBufferRef *buf)
Create a new reference to an AVBuffer.
Definition: buffer.c:92
Bi-dir predicted.
Definition: avutil.h:268
planar YUV 4:4:4, 24bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV444P and setting col...
Definition: pixfmt.h:76
int cur_chroma_format_idc
Definition: h264.h:677
int8_t * intra4x4_pred_mode
Definition: h264.h:345
enum AVDiscard skip_loop_filter
Skip loop filtering for selected frames.
Definition: avcodec.h:3250
int den
denominator
Definition: rational.h:45
static void predict_field_decoding_flag(const H264Context *h, H264SliceContext *sl)
Definition: h264_slice.c:2126
int bit_depth_luma
bit_depth_luma_minus8 + 8
Definition: h264.h:189
static av_always_inline int get_chroma_qp(const H264Context *h, int t, int qscale)
Get the chroma qp.
Definition: h264.h:838
#define IS_INTRA(x, y)
int delta_poc[2]
Definition: h264_parse.h:46
void * priv_data
Definition: avcodec.h:1691
int ff_h264_field_end(H264Context *h, H264SliceContext *sl, int in_setup)
Definition: h264_picture.c:154
#define PICT_FRAME
Definition: mpegutils.h:39
int ff_h264_execute_decode_slices(H264Context *h, unsigned context_count)
Call decode_slice() for each context.
Definition: h264_slice.c:2400
static av_always_inline void backup_mb_border(const H264Context *h, H264SliceContext *sl, uint8_t *src_y, uint8_t *src_cb, uint8_t *src_cr, int linesize, int uvlinesize, int simple)
Definition: h264_slice.c:561
void ff_h264_set_erpic(ERPicture *dst, H264Picture *src)
Definition: h264_picture.c:131
uint8_t zigzag_scan8x8_cavlc_q0[64]
Definition: h264.h:549
int8_t ref_cache[2][5 *8]
Definition: h264.h:430
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:3119
#define SLICE_SKIPED
Definition: h264.h:994
#define FRAME_MBAFF(h)
Definition: h264.h:77
int frame_priv_data_size
Size of per-frame hardware accelerator private data.
Definition: avcodec.h:3759
#define FF_BUG_TRUNCATED
Definition: avcodec.h:2852
static uint8_t tmp[8]
Definition: des.c:38
H264Picture cur_pic
Definition: h264.h:467
int key_frame
1 -> keyframe, 0-> not
Definition: frame.h:253
#define AV_ZERO32(d)
Definition: intreadwrite.h:614
int mb_width
Definition: h264.h:555
static int find_unused_picture(H264Context *h)
Definition: h264_slice.c:259
int current_slice
current slice number, used to initialize slice_num of each thread/context
Definition: h264.h:615
static const uint8_t * align_get_bits(GetBitContext *s)
Definition: get_bits.h:445
int mb_width
pic_width_in_mbs_minus1 + 1
Definition: h264.h:151
int flags2
AV_CODEC_FLAG2_*.
Definition: avcodec.h:1751
uint32_t * mb2b_xy
Definition: h264.h:515
H264Ref ref_list[2][48]
0..15: frame refs, 16..47: mbaff field refs.
Definition: h264.h:406
uint8_t field_scan8x8_cavlc_q0[64]
Definition: h264.h:552
int cur_bit_depth_luma
Definition: h264.h:678
AVBufferRef * ref_index_buf[2]
Definition: h264.h:279
int au_pps_id
pps_id of current access unit
Definition: h264.h:522
SPS * sps
Definition: h264.h:238
H264DSPContext h264dsp
Definition: h264.h:460
void ff_er_frame_start(ERContext *s)
int height
Definition: frame.h:236
int crop_top
Definition: h264.h:302
uint8_t(*[2] mvd_table)[2]
Definition: h264.h:443
uint8_t field_scan8x8[64]
Definition: h264.h:545
int slice_type_fixed
Definition: h264.h:324
static av_always_inline void fill_filter_caches_inter(const H264Context *h, H264SliceContext *sl, int mb_type, int top_xy, int left_xy[LEFT_MBS], int top_type, int left_type[LEFT_MBS], int mb_xy, int list)
Definition: h264_slice.c:1830
#define av_freep(p)
int prev_frame_num_offset
for POC type 2
Definition: h264_parse.h:51
#define av_always_inline
Definition: attributes.h:39
HW decoding through Direct3D11, Picture.data[3] contains a ID3D11VideoDecoderOutputView pointer...
Definition: pixfmt.h:242
int slice_beta_offset
Definition: h264.h:334
int8_t * intra4x4_pred_mode
Definition: h264.h:501
AVBufferRef * av_buffer_pool_get(AVBufferPool *pool)
Allocate a new AVBuffer, reusing an old buffer from the pool when available.
Definition: buffer.c:380
#define ER_AC_END
static int decode_slice(struct AVCodecContext *avctx, void *arg)
Definition: h264_slice.c:2184
const char * av_get_pix_fmt_name(enum AVPixelFormat pix_fmt)
Return the short name for a pixel format, NULL in case pix_fmt is unknown.
Definition: pixdesc.c:2138
uint8_t field_scan_q0[16]
Definition: h264.h:550
#define LIST_NOT_USED
Definition: h264.h:506
int mb_field_decoding_flag
Definition: h264.h:378
uint8_t(* non_zero_count)[48]
Definition: h264.h:504
static void loop_filter(const H264Context *h, H264SliceContext *sl, int start_x, int end_x)
Definition: h264_slice.c:2056
unsigned int crop_bottom
frame_cropping_rect_bottom_offset
Definition: h264.h:162
exp golomb vlc stuff
uint8_t * bipred_scratchpad
Definition: h264.h:413
AVPixelFormat
Pixel format.
Definition: pixfmt.h:60
av_cold void ff_h264qpel_init(H264QpelContext *c, int bit_depth)
Definition: h264qpel.c:49
#define AV_GET_BUFFER_FLAG_REF
The decoder will keep a reference to the frame and may reuse it later.
Definition: avcodec.h:1341
int droppable
Definition: h264.h:489
int level_idc
Definition: h264.h:139
int nal_ref_idc
Definition: h264.h:562
GetBitContext gb
Definition: h264.h:318
uint8_t field_scan[16]
Definition: h264.h:544
int cabac_init_idc
Definition: h264.h:450
for(j=16;j >0;--j)
#define FFMAX3(a, b, c)
Definition: common.h:95
int b_stride
Definition: h264.h:517
Predicted.
Definition: avutil.h:267
#define tb
Definition: regdef.h:68
Context Adaptive Binary Arithmetic Coder.
#define AV_CEIL_RSHIFT(a, b)
Definition: common.h:58
int short_ref_count
number of actual short term references
Definition: h264.h:606
enum AVColorSpace colorspace
Definition: h264.h:170