cbs_h264: Add support for pan-scan rectangle SEI messages
[ffmpeg.git] / libavcodec / cbs_h264_syntax_template.c
1 /*
2  * This file is part of FFmpeg.
3  *
4  * FFmpeg is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * FFmpeg is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with FFmpeg; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17  */
18
19 static int FUNC(rbsp_trailing_bits)(CodedBitstreamContext *ctx, RWContext *rw)
20 {
21     int err;
22
23     fixed(1, rbsp_stop_one_bit, 1);
24     while (byte_alignment(rw) != 0)
25         fixed(1, rbsp_alignment_zero_bit, 0);
26
27     return 0;
28 }
29
30 static int FUNC(nal_unit_header)(CodedBitstreamContext *ctx, RWContext *rw,
31                                  H264RawNALUnitHeader *current,
32                                  uint32_t valid_type_mask)
33 {
34     int err;
35
36     u(1, forbidden_zero_bit, 0, 0);
37     u(2, nal_ref_idc,        0, 3);
38     u(5, nal_unit_type,      0, 31);
39
40     if (!(1 << current->nal_unit_type & valid_type_mask)) {
41         av_log(ctx->log_ctx, AV_LOG_ERROR, "Invalid NAL unit type %d.\n",
42                current->nal_unit_type);
43         return AVERROR_INVALIDDATA;
44     }
45
46     if (current->nal_unit_type == 14 ||
47         current->nal_unit_type == 20 ||
48         current->nal_unit_type == 21) {
49         if (current->nal_unit_type != 21)
50             flag(svc_extension_flag);
51         else
52             flag(avc_3d_extension_flag);
53
54         if (current->svc_extension_flag) {
55             av_log(ctx->log_ctx, AV_LOG_ERROR, "SVC not supported.\n");
56             return AVERROR_PATCHWELCOME;
57
58         } else if (current->avc_3d_extension_flag) {
59             av_log(ctx->log_ctx, AV_LOG_ERROR, "3DAVC not supported.\n");
60             return AVERROR_PATCHWELCOME;
61
62         } else {
63             av_log(ctx->log_ctx, AV_LOG_ERROR, "MVC not supported.\n");
64             return AVERROR_PATCHWELCOME;
65         }
66     }
67
68     return 0;
69 }
70
71 static int FUNC(scaling_list)(CodedBitstreamContext *ctx, RWContext *rw,
72                               H264RawScalingList *current,
73                               int size_of_scaling_list)
74 {
75     int err, i, scale;
76
77     scale = 8;
78     for (i = 0; i < size_of_scaling_list; i++) {
79         ses(delta_scale[i], -128, +127, 1, i);
80         scale = (scale + current->delta_scale[i] + 256) % 256;
81         if (scale == 0)
82             break;
83     }
84
85     return 0;
86 }
87
88 static int FUNC(hrd_parameters)(CodedBitstreamContext *ctx, RWContext *rw,
89                                 H264RawHRD *current)
90 {
91     int err, i;
92
93     ue(cpb_cnt_minus1, 0, 31);
94     u(4, bit_rate_scale, 0, 15);
95     u(4, cpb_size_scale, 0, 15);
96
97     for (i = 0; i <= current->cpb_cnt_minus1; i++) {
98         ues(bit_rate_value_minus1[i], 0, UINT32_MAX - 1, 1, i);
99         ues(cpb_size_value_minus1[i], 0, UINT32_MAX - 1, 1, i);
100         flags(cbr_flag[i], 1, i);
101     }
102
103     u(5, initial_cpb_removal_delay_length_minus1, 0, 31);
104     u(5, cpb_removal_delay_length_minus1,         0, 31);
105     u(5, dpb_output_delay_length_minus1,          0, 31);
106     u(5, time_offset_length,                      0, 31);
107
108     return 0;
109 }
110
111 static int FUNC(vui_parameters)(CodedBitstreamContext *ctx, RWContext *rw,
112                                 H264RawVUI *current, H264RawSPS *sps)
113 {
114     int err;
115
116     flag(aspect_ratio_info_present_flag);
117     if (current->aspect_ratio_info_present_flag) {
118         u(8, aspect_ratio_idc, 0, 255);
119         if (current->aspect_ratio_idc == 255) {
120             u(16, sar_width,  0, 65535);
121             u(16, sar_height, 0, 65535);
122         }
123     } else {
124         infer(aspect_ratio_idc, 0);
125     }
126
127     flag(overscan_info_present_flag);
128     if (current->overscan_info_present_flag)
129         flag(overscan_appropriate_flag);
130
131     flag(video_signal_type_present_flag);
132     if (current->video_signal_type_present_flag) {
133         u(3, video_format, 0, 7);
134         flag(video_full_range_flag);
135         flag(colour_description_present_flag);
136         if (current->colour_description_present_flag) {
137             u(8, colour_primaries,         0, 255);
138             u(8, transfer_characteristics, 0, 255);
139             u(8, matrix_coefficients,      0, 255);
140         }
141     } else {
142         infer(video_format,             5);
143         infer(video_full_range_flag,    0);
144         infer(colour_primaries,         2);
145         infer(transfer_characteristics, 2);
146         infer(matrix_coefficients,      2);
147     }
148
149     flag(chroma_loc_info_present_flag);
150     if (current->chroma_loc_info_present_flag) {
151         ue(chroma_sample_loc_type_top_field,    0, 5);
152         ue(chroma_sample_loc_type_bottom_field, 0, 5);
153     } else {
154         infer(chroma_sample_loc_type_top_field,    0);
155         infer(chroma_sample_loc_type_bottom_field, 0);
156     }
157
158     flag(timing_info_present_flag);
159     if (current->timing_info_present_flag) {
160         u(32, num_units_in_tick, 1, UINT32_MAX);
161         u(32, time_scale,        1, UINT32_MAX);
162         flag(fixed_frame_rate_flag);
163     } else {
164         infer(fixed_frame_rate_flag, 0);
165     }
166
167     flag(nal_hrd_parameters_present_flag);
168     if (current->nal_hrd_parameters_present_flag)
169         CHECK(FUNC(hrd_parameters)(ctx, rw, &current->nal_hrd_parameters));
170
171     flag(vcl_hrd_parameters_present_flag);
172     if (current->vcl_hrd_parameters_present_flag)
173         CHECK(FUNC(hrd_parameters)(ctx, rw, &current->vcl_hrd_parameters));
174
175     if (current->nal_hrd_parameters_present_flag ||
176         current->vcl_hrd_parameters_present_flag)
177         flag(low_delay_hrd_flag);
178     else
179         infer(low_delay_hrd_flag, 1 - current->fixed_frame_rate_flag);
180
181     flag(pic_struct_present_flag);
182
183     flag(bitstream_restriction_flag);
184     if (current->bitstream_restriction_flag) {
185         flag(motion_vectors_over_pic_boundaries_flag);
186         ue(max_bytes_per_pic_denom, 0, 16);
187         ue(max_bits_per_mb_denom,   0, 16);
188         ue(log2_max_mv_length_horizontal, 0, 16);
189         ue(log2_max_mv_length_vertical,   0, 16);
190         ue(max_num_reorder_frames,  0, H264_MAX_DPB_FRAMES);
191         ue(max_dec_frame_buffering, 0, H264_MAX_DPB_FRAMES);
192     } else {
193         infer(motion_vectors_over_pic_boundaries_flag, 1);
194         infer(max_bytes_per_pic_denom, 2);
195         infer(max_bits_per_mb_denom,   1);
196         infer(log2_max_mv_length_horizontal, 16);
197         infer(log2_max_mv_length_vertical,   16);
198
199         if ((sps->profile_idc ==  44 || sps->profile_idc ==  86 ||
200              sps->profile_idc == 110 || sps->profile_idc == 110 ||
201              sps->profile_idc == 122 || sps->profile_idc == 244) &&
202             sps->constraint_set3_flag) {
203             infer(max_num_reorder_frames,  0);
204             infer(max_dec_frame_buffering, 0);
205         } else {
206             infer(max_num_reorder_frames,  H264_MAX_DPB_FRAMES);
207             infer(max_dec_frame_buffering, H264_MAX_DPB_FRAMES);
208         }
209     }
210
211     return 0;
212 }
213
214 static int FUNC(sps)(CodedBitstreamContext *ctx, RWContext *rw,
215                      H264RawSPS *current)
216 {
217     int err, i;
218
219     HEADER("Sequence Parameter Set");
220
221     CHECK(FUNC(nal_unit_header)(ctx, rw, &current->nal_unit_header,
222                                 1 << H264_NAL_SPS));
223
224     u(8, profile_idc, 0, 255);
225
226     flag(constraint_set0_flag);
227     flag(constraint_set1_flag);
228     flag(constraint_set2_flag);
229     flag(constraint_set3_flag);
230     flag(constraint_set4_flag);
231     flag(constraint_set5_flag);
232
233     u(2, reserved_zero_2bits,  0, 0);
234
235     u(8, level_idc, 0, 255);
236
237     ue(seq_parameter_set_id, 0, 31);
238
239     if (current->profile_idc == 100 || current->profile_idc == 110 ||
240         current->profile_idc == 122 || current->profile_idc == 244 ||
241         current->profile_idc ==  44 || current->profile_idc ==  83 ||
242         current->profile_idc ==  86 || current->profile_idc == 118 ||
243         current->profile_idc == 128 || current->profile_idc == 138) {
244         ue(chroma_format_idc, 0, 3);
245
246         if (current->chroma_format_idc == 3)
247             flag(separate_colour_plane_flag);
248         else
249             infer(separate_colour_plane_flag, 0);
250
251         ue(bit_depth_luma_minus8,   0, 6);
252         ue(bit_depth_chroma_minus8, 0, 6);
253
254         flag(qpprime_y_zero_transform_bypass_flag);
255
256         flag(seq_scaling_matrix_present_flag);
257         if (current->seq_scaling_matrix_present_flag) {
258             for (i = 0; i < ((current->chroma_format_idc != 3) ? 8 : 12); i++) {
259                 flags(seq_scaling_list_present_flag[i], 1, i);
260                 if (current->seq_scaling_list_present_flag[i]) {
261                     if (i < 6)
262                         CHECK(FUNC(scaling_list)(ctx, rw,
263                                                  &current->scaling_list_4x4[i],
264                                                  16));
265                     else
266                         CHECK(FUNC(scaling_list)(ctx, rw,
267                                                  &current->scaling_list_8x8[i - 6],
268                                                  64));
269                 }
270             }
271         }
272     } else {
273         infer(chroma_format_idc, current->profile_idc == 183 ? 0 : 1);
274
275         infer(separate_colour_plane_flag, 0);
276         infer(bit_depth_luma_minus8,      0);
277         infer(bit_depth_chroma_minus8,    0);
278     }
279
280     ue(log2_max_frame_num_minus4, 0, 12);
281     ue(pic_order_cnt_type, 0, 2);
282
283     if (current->pic_order_cnt_type == 0) {
284         ue(log2_max_pic_order_cnt_lsb_minus4, 0, 12);
285     } else if (current->pic_order_cnt_type == 1) {
286         flag(delta_pic_order_always_zero_flag);
287         se(offset_for_non_ref_pic,         INT32_MIN + 1, INT32_MAX);
288         se(offset_for_top_to_bottom_field, INT32_MIN + 1, INT32_MAX);
289         ue(num_ref_frames_in_pic_order_cnt_cycle, 0, 255);
290
291         for (i = 0; i < current->num_ref_frames_in_pic_order_cnt_cycle; i++)
292             ses(offset_for_ref_frame[i], INT32_MIN + 1, INT32_MAX, 1, i);
293     }
294
295     ue(max_num_ref_frames, 0, H264_MAX_DPB_FRAMES);
296     flag(gaps_in_frame_num_allowed_flag);
297
298     ue(pic_width_in_mbs_minus1,        0, H264_MAX_MB_WIDTH);
299     ue(pic_height_in_map_units_minus1, 0, H264_MAX_MB_HEIGHT);
300
301     flag(frame_mbs_only_flag);
302     if (!current->frame_mbs_only_flag)
303         flag(mb_adaptive_frame_field_flag);
304
305     flag(direct_8x8_inference_flag);
306
307     flag(frame_cropping_flag);
308     if (current->frame_cropping_flag) {
309         ue(frame_crop_left_offset,   0, H264_MAX_WIDTH);
310         ue(frame_crop_right_offset,  0, H264_MAX_WIDTH);
311         ue(frame_crop_top_offset,    0, H264_MAX_HEIGHT);
312         ue(frame_crop_bottom_offset, 0, H264_MAX_HEIGHT);
313     }
314
315     flag(vui_parameters_present_flag);
316     if (current->vui_parameters_present_flag)
317         CHECK(FUNC(vui_parameters)(ctx, rw, &current->vui, current));
318
319     CHECK(FUNC(rbsp_trailing_bits)(ctx, rw));
320
321     return 0;
322 }
323
324 static int FUNC(sps_extension)(CodedBitstreamContext *ctx, RWContext *rw,
325                                H264RawSPSExtension *current)
326 {
327     int err;
328
329     HEADER("Sequence Parameter Set Extension");
330
331     CHECK(FUNC(nal_unit_header)(ctx, rw, &current->nal_unit_header,
332                                 1 << H264_NAL_SPS_EXT));
333
334     ue(seq_parameter_set_id, 0, 31);
335
336     ue(aux_format_idc, 0, 3);
337
338     if (current->aux_format_idc != 0) {
339         int bits;
340
341         ue(bit_depth_aux_minus8, 0, 4);
342         flag(alpha_incr_flag);
343
344         bits = current->bit_depth_aux_minus8 + 9;
345         u(bits, alpha_opaque_value,      0, MAX_UINT_BITS(bits));
346         u(bits, alpha_transparent_value, 0, MAX_UINT_BITS(bits));
347     }
348
349     flag(additional_extension_flag);
350
351     CHECK(FUNC(rbsp_trailing_bits)(ctx, rw));
352
353     return 0;
354 }
355
356 static int FUNC(pps)(CodedBitstreamContext *ctx, RWContext *rw,
357                      H264RawPPS *current)
358 {
359     CodedBitstreamH264Context *h264 = ctx->priv_data;
360     const H264RawSPS *sps;
361     int err, i;
362
363     HEADER("Picture Parameter Set");
364
365     CHECK(FUNC(nal_unit_header)(ctx, rw, &current->nal_unit_header,
366                                 1 << H264_NAL_PPS));
367
368     ue(pic_parameter_set_id, 0, 255);
369     ue(seq_parameter_set_id, 0, 31);
370
371     sps = h264->sps[current->seq_parameter_set_id];
372     if (!sps) {
373         av_log(ctx->log_ctx, AV_LOG_ERROR, "SPS id %d not available.\n",
374                current->seq_parameter_set_id);
375         return AVERROR_INVALIDDATA;
376     }
377
378     flag(entropy_coding_mode_flag);
379     flag(bottom_field_pic_order_in_frame_present_flag);
380
381     ue(num_slice_groups_minus1, 0, 7);
382     if (current->num_slice_groups_minus1 > 0) {
383         unsigned int pic_size;
384         int iGroup;
385
386         pic_size = (sps->pic_width_in_mbs_minus1 + 1) *
387                    (sps->pic_height_in_map_units_minus1 + 1);
388
389         ue(slice_group_map_type, 0, 6);
390
391         if (current->slice_group_map_type == 0) {
392             for (iGroup = 0; iGroup <= current->num_slice_groups_minus1; iGroup++)
393                 ues(run_length_minus1[iGroup], 0, pic_size - 1, 1, iGroup);
394
395         } else if (current->slice_group_map_type == 2) {
396             for (iGroup = 0; iGroup < current->num_slice_groups_minus1; iGroup++) {
397                 ues(top_left[iGroup],       0, pic_size - 1, 1, iGroup);
398                 ues(bottom_right[iGroup],
399                     current->top_left[iGroup], pic_size - 1, 1, iGroup);
400             }
401         } else if (current->slice_group_map_type == 3 ||
402                    current->slice_group_map_type == 4 ||
403                    current->slice_group_map_type == 5) {
404             flag(slice_group_change_direction_flag);
405             ue(slice_group_change_rate_minus1, 0, pic_size - 1);
406         } else if (current->slice_group_map_type == 6) {
407             ue(pic_size_in_map_units_minus1, pic_size - 1, pic_size - 1);
408
409             allocate(current->slice_group_id,
410                      current->pic_size_in_map_units_minus1 + 1);
411             for (i = 0; i <= current->pic_size_in_map_units_minus1; i++)
412                 us(av_log2(2 * current->num_slice_groups_minus1 + 1),
413                    slice_group_id[i], 0, current->num_slice_groups_minus1, 1, i);
414         }
415     }
416
417     ue(num_ref_idx_l0_default_active_minus1, 0, 31);
418     ue(num_ref_idx_l1_default_active_minus1, 0, 31);
419
420     flag(weighted_pred_flag);
421     u(2, weighted_bipred_idc, 0, 2);
422
423     se(pic_init_qp_minus26, -26 - 6 * sps->bit_depth_luma_minus8, +25);
424     se(pic_init_qs_minus26, -26, +25);
425     se(chroma_qp_index_offset, -12, +12);
426
427     flag(deblocking_filter_control_present_flag);
428     flag(constrained_intra_pred_flag);
429     flag(redundant_pic_cnt_present_flag);
430
431     if (more_rbsp_data(current->more_rbsp_data))
432     {
433         flag(transform_8x8_mode_flag);
434
435         flag(pic_scaling_matrix_present_flag);
436         if (current->pic_scaling_matrix_present_flag) {
437             for (i = 0; i < 6 + (((sps->chroma_format_idc != 3) ? 2 : 6) *
438                                  current->transform_8x8_mode_flag); i++) {
439                 flags(pic_scaling_list_present_flag[i], 1, i);
440                 if (current->pic_scaling_list_present_flag[i]) {
441                     if (i < 6)
442                         CHECK(FUNC(scaling_list)(ctx, rw,
443                                                  &current->scaling_list_4x4[i],
444                                                  16));
445                     else
446                         CHECK(FUNC(scaling_list)(ctx, rw,
447                                                  &current->scaling_list_8x8[i - 6],
448                                                  64));
449                 }
450             }
451         }
452
453         se(second_chroma_qp_index_offset, -12, +12);
454     } else {
455         infer(transform_8x8_mode_flag, 0);
456         infer(pic_scaling_matrix_present_flag, 0);
457         infer(second_chroma_qp_index_offset, current->chroma_qp_index_offset);
458     }
459
460     CHECK(FUNC(rbsp_trailing_bits)(ctx, rw));
461
462     return 0;
463 }
464
465 static int FUNC(sei_buffering_period)(CodedBitstreamContext *ctx, RWContext *rw,
466                                       H264RawSEIBufferingPeriod *current)
467 {
468     CodedBitstreamH264Context *h264 = ctx->priv_data;
469     const H264RawSPS *sps;
470     int err, i, length;
471
472     ue(seq_parameter_set_id, 0, 31);
473
474     sps = h264->sps[current->seq_parameter_set_id];
475     if (!sps) {
476         av_log(ctx->log_ctx, AV_LOG_ERROR, "SPS id %d not available.\n",
477                current->seq_parameter_set_id);
478         return AVERROR_INVALIDDATA;
479     }
480     h264->active_sps = sps;
481
482     if (sps->vui.nal_hrd_parameters_present_flag) {
483         for (i = 0; i <= sps->vui.nal_hrd_parameters.cpb_cnt_minus1; i++) {
484             length = sps->vui.nal_hrd_parameters.initial_cpb_removal_delay_length_minus1 + 1;
485             xu(length, initial_cpb_removal_delay[SchedSelIdx],
486                current->nal.initial_cpb_removal_delay[i],
487                1, MAX_UINT_BITS(length), 1, i);
488             xu(length, initial_cpb_removal_delay_offset[SchedSelIdx],
489                current->nal.initial_cpb_removal_delay_offset[i],
490                0, MAX_UINT_BITS(length), 1, i);
491         }
492     }
493
494     if (sps->vui.vcl_hrd_parameters_present_flag) {
495         for (i = 0; i <= sps->vui.vcl_hrd_parameters.cpb_cnt_minus1; i++) {
496             length = sps->vui.vcl_hrd_parameters.initial_cpb_removal_delay_length_minus1 + 1;
497             xu(length, initial_cpb_removal_delay[SchedSelIdx],
498                current->vcl.initial_cpb_removal_delay[i],
499                1, MAX_UINT_BITS(length), 1, i);
500             xu(length, initial_cpb_removal_delay_offset[SchedSelIdx],
501                current->vcl.initial_cpb_removal_delay_offset[i],
502                0, MAX_UINT_BITS(length), 1, i);
503         }
504     }
505
506     return 0;
507 }
508
509 static int FUNC(sei_pic_timestamp)(CodedBitstreamContext *ctx, RWContext *rw,
510                                    H264RawSEIPicTimestamp *current)
511 {
512     CodedBitstreamH264Context *h264 = ctx->priv_data;
513     const H264RawSPS *sps;
514     uint8_t time_offset_length;
515     int err;
516
517     u(2, ct_type, 0, 2);
518     flag(nuit_field_based_flag);
519     u(5, counting_type, 0, 6);
520     flag(full_timestamp_flag);
521     flag(discontinuity_flag);
522     flag(cnt_dropped_flag);
523     u(8, n_frames, 0, 255);
524     if (current->full_timestamp_flag) {
525             u(6, seconds_value, 0, 59);
526             u(6, minutes_value, 0, 59);
527             u(5, hours_value,   0, 23);
528     } else {
529         flag(seconds_flag);
530         if (current->seconds_flag) {
531             u(6, seconds_value, 0, 59);
532             flag(minutes_flag);
533             if (current->minutes_flag) {
534                 u(6, minutes_value, 0, 59);
535                 flag(hours_flag);
536                 if (current->hours_flag)
537                     u(5, hours_value, 0, 23);
538             }
539         }
540     }
541
542     sps = h264->active_sps;
543     if (sps->vui.nal_hrd_parameters_present_flag)
544         time_offset_length = sps->vui.nal_hrd_parameters.time_offset_length;
545     else if (sps->vui.vcl_hrd_parameters_present_flag)
546         time_offset_length = sps->vui.vcl_hrd_parameters.time_offset_length;
547     else
548         time_offset_length = 24;
549
550     if (time_offset_length > 0)
551         u(time_offset_length, time_offset,
552           0, MAX_UINT_BITS(time_offset_length));
553     else
554         infer(time_offset, 0);
555
556     return 0;
557 }
558
559 static int FUNC(sei_pic_timing)(CodedBitstreamContext *ctx, RWContext *rw,
560                                 H264RawSEIPicTiming *current)
561 {
562     CodedBitstreamH264Context *h264 = ctx->priv_data;
563     const H264RawSPS *sps;
564     int err;
565
566     sps = h264->active_sps;
567     if (!sps) {
568         // If there is exactly one possible SPS but it is not yet active
569         // then just assume that it should be the active one.
570         int i, k = -1;
571         for (i = 0; i < H264_MAX_SPS_COUNT; i++) {
572             if (h264->sps[i]) {
573                 if (k >= 0) {
574                     k = -1;
575                     break;
576                 }
577                 k = i;
578             }
579         }
580         if (k >= 0)
581             sps = h264->sps[k];
582     }
583     if (!sps) {
584         av_log(ctx->log_ctx, AV_LOG_ERROR,
585                "No active SPS for pic_timing.\n");
586         return AVERROR_INVALIDDATA;
587     }
588
589     if (sps->vui.nal_hrd_parameters_present_flag ||
590         sps->vui.vcl_hrd_parameters_present_flag) {
591         const H264RawHRD *hrd;
592
593         if (sps->vui.nal_hrd_parameters_present_flag)
594             hrd = &sps->vui.nal_hrd_parameters;
595         else if (sps->vui.vcl_hrd_parameters_present_flag)
596             hrd = &sps->vui.vcl_hrd_parameters;
597         else {
598             av_log(ctx->log_ctx, AV_LOG_ERROR,
599                    "No HRD parameters for pic_timing.\n");
600             return AVERROR_INVALIDDATA;
601         }
602
603         u(hrd->cpb_removal_delay_length_minus1 + 1, cpb_removal_delay,
604           0, MAX_UINT_BITS(hrd->cpb_removal_delay_length_minus1 + 1));
605         u(hrd->dpb_output_delay_length_minus1 + 1, dpb_output_delay,
606           0, MAX_UINT_BITS(hrd->dpb_output_delay_length_minus1 + 1));
607     }
608
609     if (sps->vui.pic_struct_present_flag) {
610         static const int num_clock_ts[9] = {
611             1, 1, 1, 2, 2, 3, 3, 2, 3
612         };
613         int i;
614
615         u(4, pic_struct, 0, 8);
616         if (current->pic_struct > 8)
617             return AVERROR_INVALIDDATA;
618
619         for (i = 0; i < num_clock_ts[current->pic_struct]; i++) {
620             flags(clock_timestamp_flag[i], 1, i);
621             if (current->clock_timestamp_flag[i])
622                 CHECK(FUNC(sei_pic_timestamp)(ctx, rw, &current->timestamp[i]));
623         }
624     }
625
626     return 0;
627 }
628
629 static int FUNC(sei_pan_scan_rect)(CodedBitstreamContext *ctx, RWContext *rw,
630                                    H264RawSEIPanScanRect *current)
631 {
632     int err, i;
633
634     ue(pan_scan_rect_id, 0, UINT32_MAX - 1);
635     flag(pan_scan_rect_cancel_flag);
636
637     if (!current->pan_scan_rect_cancel_flag) {
638         ue(pan_scan_cnt_minus1, 0, 2);
639
640         for (i = 0; i <= current->pan_scan_cnt_minus1; i++) {
641             ses(pan_scan_rect_left_offset[i],   INT32_MIN + 1, INT32_MAX, 1, i);
642             ses(pan_scan_rect_right_offset[i],  INT32_MIN + 1, INT32_MAX, 1, i);
643             ses(pan_scan_rect_top_offset[i],    INT32_MIN + 1, INT32_MAX, 1, i);
644             ses(pan_scan_rect_bottom_offset[i], INT32_MIN + 1, INT32_MAX, 1, i);
645         }
646
647         ue(pan_scan_rect_repetition_period, 0, 16384);
648     }
649
650     return 0;
651 }
652
653 static int FUNC(sei_user_data_registered)(CodedBitstreamContext *ctx, RWContext *rw,
654                                           H264RawSEIUserDataRegistered *current,
655                                           uint32_t *payload_size)
656 {
657     int err, i, j;
658
659     u(8, itu_t_t35_country_code, 0x00, 0xff);
660     if (current->itu_t_t35_country_code != 0xff)
661         i = 1;
662     else {
663         u(8, itu_t_t35_country_code_extension_byte, 0x00, 0xff);
664         i = 2;
665     }
666
667 #ifdef READ
668     if (*payload_size < i) {
669         av_log(ctx->log_ctx, AV_LOG_ERROR,
670                "Invalid SEI user data registered payload.\n");
671         return AVERROR_INVALIDDATA;
672     }
673     current->data_length = *payload_size - i;
674 #else
675     *payload_size = i + current->data_length;
676 #endif
677
678     allocate(current->data, current->data_length);
679     for (j = 0; j < current->data_length; j++)
680         xu(8, itu_t_t35_payload_byte[i], current->data[j], 0x00, 0xff, 1, i + j);
681
682     return 0;
683 }
684
685 static int FUNC(sei_user_data_unregistered)(CodedBitstreamContext *ctx, RWContext *rw,
686                                             H264RawSEIUserDataUnregistered *current,
687                                             uint32_t *payload_size)
688 {
689     int err, i;
690
691 #ifdef READ
692     if (*payload_size < 16) {
693         av_log(ctx->log_ctx, AV_LOG_ERROR,
694                "Invalid SEI user data unregistered payload.\n");
695         return AVERROR_INVALIDDATA;
696     }
697     current->data_length = *payload_size - 16;
698 #else
699     *payload_size = 16 + current->data_length;
700 #endif
701
702     for (i = 0; i < 16; i++)
703         us(8, uuid_iso_iec_11578[i], 0x00, 0xff, 1, i);
704
705     allocate(current->data, current->data_length);
706
707     for (i = 0; i < current->data_length; i++)
708         xu(8, user_data_payload_byte[i], current->data[i], 0x00, 0xff, 1, i);
709
710     return 0;
711 }
712
713 static int FUNC(sei_recovery_point)(CodedBitstreamContext *ctx, RWContext *rw,
714                                     H264RawSEIRecoveryPoint *current)
715 {
716     int err;
717
718     ue(recovery_frame_cnt, 0, 65535);
719     flag(exact_match_flag);
720     flag(broken_link_flag);
721     u(2, changing_slice_group_idc, 0, 2);
722
723     return 0;
724 }
725
726 static int FUNC(sei_display_orientation)(CodedBitstreamContext *ctx, RWContext *rw,
727                                          H264RawSEIDisplayOrientation *current)
728 {
729     int err;
730
731     flag(display_orientation_cancel_flag);
732     if (!current->display_orientation_cancel_flag) {
733         flag(hor_flip);
734         flag(ver_flip);
735         u(16, anticlockwise_rotation, 0, 65535);
736         ue(display_orientation_repetition_period, 0, 16384);
737         flag(display_orientation_extension_flag);
738     }
739
740     return 0;
741 }
742
743 static int FUNC(sei_payload)(CodedBitstreamContext *ctx, RWContext *rw,
744                              H264RawSEIPayload *current)
745 {
746     int err, i;
747     int start_position, end_position;
748
749 #ifdef READ
750     start_position = get_bits_count(rw);
751 #else
752     start_position = put_bits_count(rw);
753 #endif
754
755     switch (current->payload_type) {
756     case H264_SEI_TYPE_BUFFERING_PERIOD:
757         CHECK(FUNC(sei_buffering_period)
758               (ctx, rw, &current->payload.buffering_period));
759         break;
760     case H264_SEI_TYPE_PIC_TIMING:
761         CHECK(FUNC(sei_pic_timing)
762               (ctx, rw, &current->payload.pic_timing));
763         break;
764     case H264_SEI_TYPE_PAN_SCAN_RECT:
765         CHECK(FUNC(sei_pan_scan_rect)
766               (ctx, rw, &current->payload.pan_scan_rect));
767         break;
768     case H264_SEI_TYPE_FILLER_PAYLOAD:
769         {
770             for (i = 0; i  < current->payload_size; i++)
771                 fixed(8, ff_byte, 0xff);
772         }
773         break;
774     case H264_SEI_TYPE_USER_DATA_REGISTERED:
775         CHECK(FUNC(sei_user_data_registered)
776               (ctx, rw, &current->payload.user_data_registered, &current->payload_size));
777         break;
778     case H264_SEI_TYPE_USER_DATA_UNREGISTERED:
779         CHECK(FUNC(sei_user_data_unregistered)
780               (ctx, rw, &current->payload.user_data_unregistered, &current->payload_size));
781         break;
782     case H264_SEI_TYPE_RECOVERY_POINT:
783         CHECK(FUNC(sei_recovery_point)
784               (ctx, rw, &current->payload.recovery_point));
785         break;
786     case H264_SEI_TYPE_DISPLAY_ORIENTATION:
787         CHECK(FUNC(sei_display_orientation)
788               (ctx, rw, &current->payload.display_orientation));
789         break;
790     default:
791         {
792 #ifdef READ
793             current->payload.other.data_length = current->payload_size;
794 #endif
795             allocate(current->payload.other.data, current->payload.other.data_length);
796             for (i = 0; i < current->payload.other.data_length; i++)
797                 xu(8, payload_byte[i], current->payload.other.data[i], 0, 255, 1, i);
798         }
799     }
800
801     if (byte_alignment(rw)) {
802         fixed(1, bit_equal_to_one, 1);
803         while (byte_alignment(rw))
804             fixed(1, bit_equal_to_zero, 0);
805     }
806
807 #ifdef READ
808     end_position = get_bits_count(rw);
809     if (end_position < start_position + 8 * current->payload_size) {
810         av_log(ctx->log_ctx, AV_LOG_ERROR, "Incorrect SEI payload length: "
811                "header %"PRIu32" bits, actually %d bits.\n",
812                8 * current->payload_size,
813                end_position - start_position);
814         return AVERROR_INVALIDDATA;
815     }
816 #else
817     end_position = put_bits_count(rw);
818     current->payload_size = (end_position - start_position) / 8;
819 #endif
820
821     return 0;
822 }
823
824 static int FUNC(sei)(CodedBitstreamContext *ctx, RWContext *rw,
825                      H264RawSEI *current)
826 {
827     int err, k;
828
829     HEADER("Supplemental Enhancement Information");
830
831     CHECK(FUNC(nal_unit_header)(ctx, rw, &current->nal_unit_header,
832                                 1 << H264_NAL_SEI));
833
834 #ifdef READ
835     for (k = 0; k < H264_MAX_SEI_PAYLOADS; k++) {
836         uint32_t payload_type = 0;
837         uint32_t payload_size = 0;
838         uint32_t tmp;
839
840         while (show_bits(rw, 8) == 0xff) {
841             fixed(8, ff_byte, 0xff);
842             payload_type += 255;
843         }
844         xu(8, last_payload_type_byte, tmp, 0, 254, 0);
845         payload_type += tmp;
846
847         while (show_bits(rw, 8) == 0xff) {
848             fixed(8, ff_byte, 0xff);
849             payload_size += 255;
850         }
851         xu(8, last_payload_size_byte, tmp, 0, 254, 0);
852         payload_size += tmp;
853
854         current->payload[k].payload_type = payload_type;
855         current->payload[k].payload_size = payload_size;
856
857         CHECK(FUNC(sei_payload)(ctx, rw, &current->payload[k]));
858
859         if (!cbs_h2645_read_more_rbsp_data(rw))
860             break;
861     }
862     if (k >= H264_MAX_SEI_PAYLOADS) {
863         av_log(ctx->log_ctx, AV_LOG_ERROR, "Too many payloads in "
864                "SEI message: found %d.\n", k);
865         return AVERROR_INVALIDDATA;
866     }
867     current->payload_count = k + 1;
868 #else
869     for (k = 0; k < current->payload_count; k++) {
870         PutBitContext start_state;
871         uint32_t tmp;
872         int need_size, i;
873
874         // Somewhat clumsy: we write the payload twice when
875         // we don't know the size in advance.  This will mess
876         // with trace output, but is otherwise harmless.
877         start_state = *rw;
878         need_size = !current->payload[k].payload_size;
879         for (i = 0; i < 1 + need_size; i++) {
880             *rw = start_state;
881
882             tmp = current->payload[k].payload_type;
883             while (tmp >= 255) {
884                 fixed(8, ff_byte, 0xff);
885                 tmp -= 255;
886             }
887             xu(8, last_payload_type_byte, tmp, 0, 254, 0);
888
889             tmp = current->payload[k].payload_size;
890             while (tmp >= 255) {
891                 fixed(8, ff_byte, 0xff);
892                 tmp -= 255;
893             }
894             xu(8, last_payload_size_byte, tmp, 0, 254, 0);
895
896             CHECK(FUNC(sei_payload)(ctx, rw, &current->payload[k]));
897         }
898     }
899 #endif
900
901     CHECK(FUNC(rbsp_trailing_bits)(ctx, rw));
902
903     return 0;
904 }
905
906 static int FUNC(aud)(CodedBitstreamContext *ctx, RWContext *rw,
907                      H264RawAUD *current)
908 {
909     int err;
910
911     HEADER("Access Unit Delimiter");
912
913     CHECK(FUNC(nal_unit_header)(ctx, rw, &current->nal_unit_header,
914                                 1 << H264_NAL_AUD));
915
916     u(3, primary_pic_type, 0, 7);
917
918     CHECK(FUNC(rbsp_trailing_bits)(ctx, rw));
919
920     return 0;
921 }
922
923 static int FUNC(ref_pic_list_modification)(CodedBitstreamContext *ctx, RWContext *rw,
924                                            H264RawSliceHeader *current)
925 {
926     CodedBitstreamH264Context *h264 = ctx->priv_data;
927     const H264RawSPS *sps = h264->active_sps;
928     int err, i, mopn;
929
930     if (current->slice_type % 5 != 2 &&
931         current->slice_type % 5 != 4) {
932         flag(ref_pic_list_modification_flag_l0);
933         if (current->ref_pic_list_modification_flag_l0) {
934             for (i = 0; i < H264_MAX_RPLM_COUNT; i++) {
935                 xue(modification_of_pic_nums_idc,
936                     current->rplm_l0[i].modification_of_pic_nums_idc, 0, 3, 0);
937
938                 mopn = current->rplm_l0[i].modification_of_pic_nums_idc;
939                 if (mopn == 3)
940                     break;
941
942                 if (mopn == 0 || mopn == 1)
943                     xue(abs_diff_pic_num_minus1,
944                         current->rplm_l0[i].abs_diff_pic_num_minus1,
945                         0, (1 + current->field_pic_flag) *
946                         (1 << (sps->log2_max_frame_num_minus4 + 4)), 0);
947                 else if (mopn == 2)
948                     xue(long_term_pic_num,
949                         current->rplm_l0[i].long_term_pic_num,
950                         0, sps->max_num_ref_frames - 1, 0);
951             }
952         }
953     }
954
955     if (current->slice_type % 5 == 1) {
956         flag(ref_pic_list_modification_flag_l1);
957         if (current->ref_pic_list_modification_flag_l1) {
958             for (i = 0; i < H264_MAX_RPLM_COUNT; i++) {
959                 xue(modification_of_pic_nums_idc,
960                     current->rplm_l1[i].modification_of_pic_nums_idc, 0, 3, 0);
961
962                 mopn = current->rplm_l1[i].modification_of_pic_nums_idc;
963                 if (mopn == 3)
964                     break;
965
966                 if (mopn == 0 || mopn == 1)
967                     xue(abs_diff_pic_num_minus1,
968                         current->rplm_l1[i].abs_diff_pic_num_minus1,
969                         0, (1 + current->field_pic_flag) *
970                         (1 << (sps->log2_max_frame_num_minus4 + 4)), 0);
971                 else if (mopn == 2)
972                     xue(long_term_pic_num,
973                         current->rplm_l1[i].long_term_pic_num,
974                         0, sps->max_num_ref_frames - 1, 0);
975             }
976         }
977     }
978
979     return 0;
980 }
981
982 static int FUNC(pred_weight_table)(CodedBitstreamContext *ctx, RWContext *rw,
983                                    H264RawSliceHeader *current)
984 {
985     CodedBitstreamH264Context *h264 = ctx->priv_data;
986     const H264RawSPS *sps = h264->active_sps;
987     int chroma;
988     int err, i, j;
989
990     ue(luma_log2_weight_denom, 0, 7);
991
992     chroma = !sps->separate_colour_plane_flag && sps->chroma_format_idc != 0;
993     if (chroma)
994         ue(chroma_log2_weight_denom, 0, 7);
995
996     for (i = 0; i <= current->num_ref_idx_l0_active_minus1; i++) {
997         flags(luma_weight_l0_flag[i], 1, i);
998         if (current->luma_weight_l0_flag[i]) {
999             ses(luma_weight_l0[i], -128, +127, 1, i);
1000             ses(luma_offset_l0[i], -128, +127, 1, i);
1001         }
1002         if (chroma) {
1003             flags(chroma_weight_l0_flag[i], 1, i);
1004             if (current->chroma_weight_l0_flag[i]) {
1005                 for (j = 0; j < 2; j++) {
1006                     ses(chroma_weight_l0[i][j], -128, +127, 2, i, j);
1007                     ses(chroma_offset_l0[i][j], -128, +127, 2, i, j);
1008                 }
1009             }
1010         }
1011     }
1012
1013     if (current->slice_type % 5 == 1) {
1014         for (i = 0; i <= current->num_ref_idx_l1_active_minus1; i++) {
1015             flags(luma_weight_l1_flag[i], 1, i);
1016             if (current->luma_weight_l1_flag[i]) {
1017                 ses(luma_weight_l1[i], -128, +127, 1, i);
1018                 ses(luma_offset_l1[i], -128, +127, 1, i);
1019             }
1020             if (chroma) {
1021                 flags(chroma_weight_l1_flag[i], 1, i);
1022                 if (current->chroma_weight_l1_flag[i]) {
1023                     for (j = 0; j < 2; j++) {
1024                         ses(chroma_weight_l1[i][j], -128, +127, 2, i, j);
1025                         ses(chroma_offset_l1[i][j], -128, +127, 2, i, j);
1026                     }
1027                 }
1028             }
1029         }
1030     }
1031
1032     return 0;
1033 }
1034
1035 static int FUNC(dec_ref_pic_marking)(CodedBitstreamContext *ctx, RWContext *rw,
1036                                      H264RawSliceHeader *current, int idr_pic_flag)
1037 {
1038     CodedBitstreamH264Context *h264 = ctx->priv_data;
1039     const H264RawSPS *sps = h264->active_sps;
1040     int err, i;
1041     uint32_t mmco;
1042
1043     if (idr_pic_flag) {
1044         flag(no_output_of_prior_pics_flag);
1045         flag(long_term_reference_flag);
1046     } else {
1047         flag(adaptive_ref_pic_marking_mode_flag);
1048         if (current->adaptive_ref_pic_marking_mode_flag) {
1049             for (i = 0; i < H264_MAX_MMCO_COUNT; i++) {
1050                 xue(memory_management_control_operation,
1051                     current->mmco[i].memory_management_control_operation,
1052                     0, 6, 0);
1053
1054                 mmco = current->mmco[i].memory_management_control_operation;
1055                 if (mmco == 0)
1056                     break;
1057
1058                 if (mmco == 1 || mmco == 3)
1059                     xue(difference_of_pic_nums_minus1,
1060                         current->mmco[i].difference_of_pic_nums_minus1,
1061                         0, INT32_MAX, 0);
1062                 if (mmco == 2)
1063                     xue(long_term_pic_num,
1064                         current->mmco[i].long_term_pic_num,
1065                         0, sps->max_num_ref_frames - 1, 0);
1066                 if (mmco == 3 || mmco == 6)
1067                     xue(long_term_frame_idx,
1068                         current->mmco[i].long_term_frame_idx,
1069                         0, sps->max_num_ref_frames - 1, 0);
1070                 if (mmco == 4)
1071                     xue(max_long_term_frame_idx_plus1,
1072                         current->mmco[i].max_long_term_frame_idx_plus1,
1073                         0, sps->max_num_ref_frames, 0);
1074             }
1075             if (i == H264_MAX_MMCO_COUNT) {
1076                 av_log(ctx->log_ctx, AV_LOG_ERROR, "Too many "
1077                        "memory management control operations.\n");
1078                 return AVERROR_INVALIDDATA;
1079             }
1080         }
1081     }
1082
1083     return 0;
1084 }
1085
1086 static int FUNC(slice_header)(CodedBitstreamContext *ctx, RWContext *rw,
1087                               H264RawSliceHeader *current)
1088 {
1089     CodedBitstreamH264Context *h264 = ctx->priv_data;
1090     const H264RawSPS *sps;
1091     const H264RawPPS *pps;
1092     int err;
1093     int idr_pic_flag;
1094     int slice_type_i, slice_type_p, slice_type_b;
1095     int slice_type_si, slice_type_sp;
1096
1097     HEADER("Slice Header");
1098
1099     CHECK(FUNC(nal_unit_header)(ctx, rw, &current->nal_unit_header,
1100                                 1 << H264_NAL_SLICE     |
1101                                 1 << H264_NAL_IDR_SLICE |
1102                                 1 << H264_NAL_AUXILIARY_SLICE));
1103
1104     if (current->nal_unit_header.nal_unit_type == H264_NAL_AUXILIARY_SLICE) {
1105         if (!h264->last_slice_nal_unit_type) {
1106             av_log(ctx->log_ctx, AV_LOG_ERROR, "Auxiliary slice "
1107                    "is not decodable without the main picture "
1108                    "in the same access unit.\n");
1109             return AVERROR_INVALIDDATA;
1110         }
1111     } else {
1112         h264->last_slice_nal_unit_type =
1113             current->nal_unit_header.nal_unit_type;
1114     }
1115     idr_pic_flag = h264->last_slice_nal_unit_type == H264_NAL_IDR_SLICE;
1116
1117     ue(first_mb_in_slice, 0, H264_MAX_MB_PIC_SIZE - 1);
1118     ue(slice_type, 0, 9);
1119
1120     slice_type_i  = current->slice_type % 5 == 2;
1121     slice_type_p  = current->slice_type % 5 == 0;
1122     slice_type_b  = current->slice_type % 5 == 1;
1123     slice_type_si = current->slice_type % 5 == 4;
1124     slice_type_sp = current->slice_type % 5 == 3;
1125
1126     if (idr_pic_flag && !(slice_type_i || slice_type_si)) {
1127         av_log(ctx->log_ctx, AV_LOG_ERROR, "Invalid slice type %d "
1128                "for IDR picture.\n", current->slice_type);
1129         return AVERROR_INVALIDDATA;
1130     }
1131
1132     ue(pic_parameter_set_id, 0, 255);
1133
1134     pps = h264->pps[current->pic_parameter_set_id];
1135     if (!pps) {
1136         av_log(ctx->log_ctx, AV_LOG_ERROR, "PPS id %d not available.\n",
1137                current->pic_parameter_set_id);
1138         return AVERROR_INVALIDDATA;
1139     }
1140     h264->active_pps = pps;
1141
1142     sps = h264->sps[pps->seq_parameter_set_id];
1143     if (!sps) {
1144         av_log(ctx->log_ctx, AV_LOG_ERROR, "SPS id %d not available.\n",
1145                pps->seq_parameter_set_id);
1146         return AVERROR_INVALIDDATA;
1147     }
1148     h264->active_sps = sps;
1149
1150     if (sps->separate_colour_plane_flag)
1151         u(2, colour_plane_id, 0, 2);
1152
1153     u(sps->log2_max_frame_num_minus4 + 4, frame_num,
1154       0, MAX_UINT_BITS(sps->log2_max_frame_num_minus4 + 4));
1155
1156     if (!sps->frame_mbs_only_flag) {
1157         flag(field_pic_flag);
1158         if (current->field_pic_flag)
1159             flag(bottom_field_flag);
1160         else
1161             infer(bottom_field_flag, 0);
1162     } else {
1163         infer(field_pic_flag,    0);
1164         infer(bottom_field_flag, 0);
1165     }
1166
1167     if (idr_pic_flag)
1168         ue(idr_pic_id, 0, 65535);
1169
1170     if (sps->pic_order_cnt_type == 0) {
1171         u(sps->log2_max_pic_order_cnt_lsb_minus4 + 4, pic_order_cnt_lsb,
1172           0, MAX_UINT_BITS(sps->log2_max_pic_order_cnt_lsb_minus4 + 4));
1173         if (pps->bottom_field_pic_order_in_frame_present_flag &&
1174             !current->field_pic_flag)
1175             se(delta_pic_order_cnt_bottom, INT32_MIN + 1, INT32_MAX);
1176
1177     } else if (sps->pic_order_cnt_type == 1) {
1178         if (!sps->delta_pic_order_always_zero_flag) {
1179             se(delta_pic_order_cnt[0], INT32_MIN + 1, INT32_MAX);
1180             if (pps->bottom_field_pic_order_in_frame_present_flag &&
1181                 !current->field_pic_flag)
1182                 se(delta_pic_order_cnt[1], INT32_MIN + 1, INT32_MAX);
1183             else
1184                 infer(delta_pic_order_cnt[1], 0);
1185         } else {
1186             infer(delta_pic_order_cnt[0], 0);
1187             infer(delta_pic_order_cnt[1], 0);
1188         }
1189     }
1190
1191     if (pps->redundant_pic_cnt_present_flag)
1192         ue(redundant_pic_cnt, 0, 127);
1193
1194     if (slice_type_b)
1195         flag(direct_spatial_mv_pred_flag);
1196
1197     if (slice_type_p || slice_type_sp || slice_type_b) {
1198         flag(num_ref_idx_active_override_flag);
1199         if (current->num_ref_idx_active_override_flag) {
1200             ue(num_ref_idx_l0_active_minus1, 0, 31);
1201             if (slice_type_b)
1202                 ue(num_ref_idx_l1_active_minus1, 0, 31);
1203         } else {
1204             infer(num_ref_idx_l0_active_minus1,
1205                   pps->num_ref_idx_l0_default_active_minus1);
1206             infer(num_ref_idx_l1_active_minus1,
1207                   pps->num_ref_idx_l1_default_active_minus1);
1208         }
1209     }
1210
1211     if (current->nal_unit_header.nal_unit_type == 20 ||
1212         current->nal_unit_header.nal_unit_type == 21) {
1213         av_log(ctx->log_ctx, AV_LOG_ERROR, "MVC / 3DAVC not supported.\n");
1214         return AVERROR_PATCHWELCOME;
1215     } else {
1216         CHECK(FUNC(ref_pic_list_modification)(ctx, rw, current));
1217     }
1218
1219     if ((pps->weighted_pred_flag && (slice_type_p || slice_type_sp)) ||
1220         (pps->weighted_bipred_idc == 1 && slice_type_b)) {
1221         CHECK(FUNC(pred_weight_table)(ctx, rw, current));
1222     }
1223
1224     if (current->nal_unit_header.nal_ref_idc != 0) {
1225         CHECK(FUNC(dec_ref_pic_marking)(ctx, rw, current, idr_pic_flag));
1226     }
1227
1228     if (pps->entropy_coding_mode_flag &&
1229         !slice_type_i && !slice_type_si) {
1230         ue(cabac_init_idc, 0, 2);
1231     }
1232
1233     se(slice_qp_delta, - 51 - 6 * sps->bit_depth_luma_minus8,
1234                        + 51 + 6 * sps->bit_depth_luma_minus8);
1235     if (slice_type_sp || slice_type_si) {
1236         if (slice_type_sp)
1237             flag(sp_for_switch_flag);
1238         se(slice_qs_delta, -51, +51);
1239     }
1240
1241     if (pps->deblocking_filter_control_present_flag) {
1242         ue(disable_deblocking_filter_idc, 0, 2);
1243         if (current->disable_deblocking_filter_idc != 1) {
1244             se(slice_alpha_c0_offset_div2, -6, +6);
1245             se(slice_beta_offset_div2,     -6, +6);
1246         } else {
1247             infer(slice_alpha_c0_offset_div2, 0);
1248             infer(slice_beta_offset_div2,     0);
1249         }
1250     } else {
1251         infer(disable_deblocking_filter_idc, 0);
1252         infer(slice_alpha_c0_offset_div2,    0);
1253         infer(slice_beta_offset_div2,        0);
1254     }
1255
1256     if (pps->num_slice_groups_minus1 > 0 &&
1257         pps->slice_group_map_type >= 3 &&
1258         pps->slice_group_map_type <= 5) {
1259         unsigned int pic_size, max, bits;
1260
1261         pic_size = (sps->pic_width_in_mbs_minus1 + 1) *
1262                    (sps->pic_height_in_map_units_minus1 + 1);
1263         max = (pic_size + pps->slice_group_change_rate_minus1) /
1264               (pps->slice_group_change_rate_minus1 + 1);
1265         bits = av_log2(2 * max - 1);
1266
1267         u(bits, slice_group_change_cycle, 0, max);
1268     }
1269
1270     if (pps->entropy_coding_mode_flag) {
1271         while (byte_alignment(rw))
1272             fixed(1, cabac_alignment_one_bit, 1);
1273     }
1274
1275     return 0;
1276 }
1277
1278 static int FUNC(filler)(CodedBitstreamContext *ctx, RWContext *rw,
1279                         H264RawFiller *current)
1280 {
1281     int err;
1282
1283     HEADER("Filler Data");
1284
1285     CHECK(FUNC(nal_unit_header)(ctx, rw, &current->nal_unit_header,
1286                                 1 << H264_NAL_FILLER_DATA));
1287
1288 #ifdef READ
1289     while (show_bits(rw, 8) == 0xff) {
1290         fixed(8, ff_byte, 0xff);
1291         ++current->filler_size;
1292     }
1293 #else
1294     {
1295         uint32_t i;
1296         for (i = 0; i < current->filler_size; i++)
1297             fixed(8, ff_byte, 0xff);
1298     }
1299 #endif
1300
1301     CHECK(FUNC(rbsp_trailing_bits)(ctx, rw));
1302
1303     return 0;
1304 }