2 * AV1 helper functions for muxers
3 * Copyright (c) 2018 James Almer <jamrial@gmail.com>
5 * This file is part of FFmpeg.
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.
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.
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
22 #include "libavutil/mem.h"
23 #include "libavcodec/av1.h"
24 #include "libavcodec/av1_parse.h"
25 #include "libavcodec/profiles.h"
26 #include "libavcodec/put_bits.h"
30 int ff_av1_filter_obus(AVIOContext *pb, const uint8_t *buf, int size)
32 const uint8_t *end = buf + size;
34 int start_pos, type, temporal_id, spatial_id;
38 int len = parse_obu_header(buf, end - buf, &obu_size, &start_pos,
39 &type, &temporal_id, &spatial_id);
44 case AV1_OBU_TEMPORAL_DELIMITER:
45 case AV1_OBU_REDUNDANT_FRAME_HEADER:
46 case AV1_OBU_TILE_LIST:
50 avio_write(pb, buf, len);
60 int ff_av1_filter_obus_buf(const uint8_t *buf, uint8_t **out, int *size)
65 ret = avio_open_dyn_buf(&pb);
69 ret = ff_av1_filter_obus(pb, buf, *size);
74 *size = avio_close_dyn_buf(pb, out);
79 typedef struct AV1SequenceParameters {
81 uint8_t seq_level_idx_0;
83 uint8_t high_bitdepth;
86 uint8_t chroma_subsampling_x;
87 uint8_t chroma_subsampling_y;
88 uint8_t chroma_sample_position;
89 } AV1SequenceParameters;
91 static inline void uvlc(GetBitContext *gb)
93 int leading_zeros = 0;
95 while (get_bits_left(gb)) {
101 if (leading_zeros >= 32)
104 skip_bits_long(gb, leading_zeros);
107 static int parse_color_config(AV1SequenceParameters *seq_params, GetBitContext *gb)
109 int color_primaries, transfer_characteristics, matrix_coefficients;
111 seq_params->high_bitdepth = get_bits1(gb);
112 if (seq_params->seq_profile == FF_PROFILE_AV1_PROFESSIONAL && seq_params->high_bitdepth)
113 seq_params->twelve_bit = get_bits1(gb);
115 if (seq_params->seq_profile == FF_PROFILE_AV1_HIGH)
116 seq_params->monochrome = 0;
118 seq_params->monochrome = get_bits1(gb);
120 if (get_bits1(gb)) { // color_description_present_flag
121 color_primaries = get_bits(gb, 8);
122 transfer_characteristics = get_bits(gb, 8);
123 matrix_coefficients = get_bits(gb, 8);
125 color_primaries = AVCOL_PRI_UNSPECIFIED;
126 transfer_characteristics = AVCOL_TRC_UNSPECIFIED;
127 matrix_coefficients = AVCOL_SPC_UNSPECIFIED;
130 if (seq_params->monochrome) {
131 skip_bits1(gb); // color_range
132 seq_params->chroma_subsampling_x = 1;
133 seq_params->chroma_subsampling_y = 1;
134 seq_params->chroma_sample_position = 0;
136 } else if (color_primaries == AVCOL_PRI_BT709 &&
137 transfer_characteristics == AVCOL_TRC_IEC61966_2_1 &&
138 matrix_coefficients == AVCOL_SPC_RGB) {
139 seq_params->chroma_subsampling_x = 0;
140 seq_params->chroma_subsampling_y = 0;
142 skip_bits1(gb); // color_range
144 if (seq_params->seq_profile == FF_PROFILE_AV1_MAIN) {
145 seq_params->chroma_subsampling_x = 1;
146 seq_params->chroma_subsampling_y = 1;
147 } else if (seq_params->seq_profile == FF_PROFILE_AV1_HIGH) {
148 seq_params->chroma_subsampling_x = 0;
149 seq_params->chroma_subsampling_y = 0;
151 if (seq_params->twelve_bit) {
152 seq_params->chroma_subsampling_x = get_bits1(gb);
153 if (seq_params->chroma_subsampling_x)
154 seq_params->chroma_subsampling_y = get_bits1(gb);
156 seq_params->chroma_subsampling_y = 0;
158 seq_params->chroma_subsampling_x = 1;
159 seq_params->chroma_subsampling_y = 0;
162 if (seq_params->chroma_subsampling_x && seq_params->chroma_subsampling_y)
163 seq_params->chroma_sample_position = get_bits(gb, 2);
166 skip_bits1(gb); // separate_uv_delta_q
171 static int parse_sequence_header(AV1SequenceParameters *seq_params, const uint8_t *buf, int size)
174 int reduced_still_picture_header;
175 int frame_width_bits_minus_1, frame_height_bits_minus_1;
178 size_bits = get_obu_bit_length(buf, size, AV1_OBU_SEQUENCE_HEADER);
182 ret = init_get_bits(&gb, buf, size_bits);
186 memset(seq_params, 0, sizeof(*seq_params));
188 seq_params->seq_profile = get_bits(&gb, 3);
190 skip_bits1(&gb); // still_picture
191 reduced_still_picture_header = get_bits1(&gb);
193 if (reduced_still_picture_header) {
194 seq_params->seq_level_idx_0 = get_bits(&gb, 5);
195 seq_params->seq_tier_0 = 0;
197 int initial_display_delay_present_flag, operating_points_cnt_minus_1;
198 int decoder_model_info_present_flag, buffer_delay_length_minus_1;
200 if (get_bits1(&gb)) { // timing_info_present_flag
201 skip_bits_long(&gb, 32); // num_units_in_display_tick
202 skip_bits_long(&gb, 32); // time_scale
204 if (get_bits1(&gb)) // equal_picture_interval
205 uvlc(&gb); // num_ticks_per_picture_minus_1
207 decoder_model_info_present_flag = get_bits1(&gb);
208 if (decoder_model_info_present_flag) {
209 buffer_delay_length_minus_1 = get_bits(&gb, 5);
210 skip_bits_long(&gb, 32); // num_units_in_decoding_tick
211 skip_bits(&gb, 10); // buffer_removal_time_length_minus_1 (5)
212 // frame_presentation_time_length_minus_1 (5)
215 decoder_model_info_present_flag = 0;
217 initial_display_delay_present_flag = get_bits1(&gb);
219 operating_points_cnt_minus_1 = get_bits(&gb, 5);
220 for (int i = 0; i <= operating_points_cnt_minus_1; i++) {
221 int seq_level_idx, seq_tier;
223 skip_bits(&gb, 12); // operating_point_idc
224 seq_level_idx = get_bits(&gb, 5);
226 if (seq_level_idx > 7)
227 seq_tier = get_bits1(&gb);
231 if (decoder_model_info_present_flag) {
232 if (get_bits1(&gb)) { // decoder_model_present_for_this_op
233 skip_bits_long(&gb, buffer_delay_length_minus_1 + 1); // decoder_buffer_delay
234 skip_bits_long(&gb, buffer_delay_length_minus_1 + 1); // encoder_buffer_delay
235 skip_bits1(&gb); // low_delay_mode_flag
239 if (initial_display_delay_present_flag) {
240 if (get_bits1(&gb)) // initial_display_delay_present_for_this_op
241 skip_bits(&gb, 4); // initial_display_delay_minus_1
245 seq_params->seq_level_idx_0 = seq_level_idx;
246 seq_params->seq_tier_0 = seq_tier;
251 frame_width_bits_minus_1 = get_bits(&gb, 4);
252 frame_height_bits_minus_1 = get_bits(&gb, 4);
254 skip_bits(&gb, frame_width_bits_minus_1 + 1); // max_frame_width_minus_1
255 skip_bits(&gb, frame_height_bits_minus_1 + 1); // max_frame_height_minus_1
257 if (!reduced_still_picture_header) {
258 if (get_bits1(&gb)) // frame_id_numbers_present_flag
259 skip_bits(&gb, 7); // delta_frame_id_length_minus_2 (4), additional_frame_id_length_minus_1 (3)
262 skip_bits(&gb, 3); // use_128x128_superblock (1), enable_filter_intra (1), enable_intra_edge_filter (1)
264 if (!reduced_still_picture_header) {
265 int enable_order_hint, seq_force_screen_content_tools;
267 skip_bits(&gb, 4); // enable_intraintra_compound (1), enable_masked_compound (1)
268 // enable_warped_motion (1), enable_dual_filter (1)
270 enable_order_hint = get_bits1(&gb);
271 if (enable_order_hint)
272 skip_bits(&gb, 2); // enable_jnt_comp (1), enable_ref_frame_mvs (1)
274 if (get_bits1(&gb)) // seq_choose_screen_content_tools
275 seq_force_screen_content_tools = 2;
277 seq_force_screen_content_tools = get_bits1(&gb);
279 if (seq_force_screen_content_tools) {
280 if (!get_bits1(&gb)) // seq_choose_integer_mv
281 skip_bits1(&gb); // seq_force_integer_mv
284 if (enable_order_hint)
285 skip_bits(&gb, 3); // order_hint_bits_minus_1
288 skip_bits(&gb, 3); // enable_superres (1), enable_cdef (1), enable_restoration (1)
290 parse_color_config(seq_params, &gb);
292 skip_bits1(&gb); // film_grain_params_present
294 if (get_bits_left(&gb))
295 return AVERROR_INVALIDDATA;
300 int ff_isom_write_av1c(AVIOContext *pb, const uint8_t *buf, int size)
302 AVIOContext *seq_pb = NULL, *meta_pb = NULL;
303 AV1SequenceParameters seq_params;
306 uint8_t *seq = NULL, *meta = NULL;
308 int start_pos, type, temporal_id, spatial_id;
309 int ret, nb_seq = 0, seq_size, meta_size;
312 return AVERROR_INVALIDDATA;
314 ret = avio_open_dyn_buf(&seq_pb);
317 ret = avio_open_dyn_buf(&meta_pb);
322 int len = parse_obu_header(buf, size, &obu_size, &start_pos,
323 &type, &temporal_id, &spatial_id);
330 case AV1_OBU_SEQUENCE_HEADER:
332 if (!obu_size || nb_seq > 1) {
333 ret = AVERROR_INVALIDDATA;
336 ret = parse_sequence_header(&seq_params, buf + start_pos, obu_size);
340 avio_write(seq_pb, buf, len);
342 case AV1_OBU_METADATA:
344 ret = AVERROR_INVALIDDATA;
347 avio_write(meta_pb, buf, len);
356 seq_size = avio_close_dyn_buf(seq_pb, &seq);
358 ret = AVERROR_INVALIDDATA;
362 init_put_bits(&pbc, header, sizeof(header));
364 put_bits(&pbc, 1, 1); // marker
365 put_bits(&pbc, 7, 1); // version
366 put_bits(&pbc, 3, seq_params.seq_profile);
367 put_bits(&pbc, 5, seq_params.seq_level_idx_0);
368 put_bits(&pbc, 1, seq_params.seq_tier_0);
369 put_bits(&pbc, 1, seq_params.high_bitdepth);
370 put_bits(&pbc, 1, seq_params.twelve_bit);
371 put_bits(&pbc, 1, seq_params.monochrome);
372 put_bits(&pbc, 1, seq_params.chroma_subsampling_x);
373 put_bits(&pbc, 1, seq_params.chroma_subsampling_y);
374 put_bits(&pbc, 2, seq_params.chroma_sample_position);
375 flush_put_bits(&pbc);
377 avio_write(pb, header, sizeof(header));
378 avio_write(pb, seq, seq_size);
380 meta_size = avio_close_dyn_buf(meta_pb, &meta);
382 avio_write(pb, meta, meta_size);
386 avio_close_dyn_buf(seq_pb, &seq);
388 avio_close_dyn_buf(meta_pb, &meta);