| Line | Branch | Exec | Source |
|---|---|---|---|
| 1 | /* | ||
| 2 | * AV1 helper functions for muxers | ||
| 3 | * Copyright (c) 2018 James Almer <jamrial@gmail.com> | ||
| 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 | #include "libavutil/avassert.h" | ||
| 23 | #include "libavutil/mem.h" | ||
| 24 | #include "libavutil/pixfmt.h" | ||
| 25 | #include "libavcodec/av1.h" | ||
| 26 | #include "libavcodec/av1_parse.h" | ||
| 27 | #include "libavcodec/defs.h" | ||
| 28 | #include "libavcodec/put_bits.h" | ||
| 29 | #include "av1.h" | ||
| 30 | #include "avio.h" | ||
| 31 | #include "avio_internal.h" | ||
| 32 | |||
| 33 | 60 | static int av1_filter_obus(AVIOContext *pb, const uint8_t *buf, | |
| 34 | int size, int *offset) | ||
| 35 | { | ||
| 36 | 60 | const uint8_t *start = buf, *end = buf + size; | |
| 37 | int off; | ||
| 38 | enum { | ||
| 39 | START_NOT_FOUND, | ||
| 40 | START_FOUND, | ||
| 41 | END_FOUND, | ||
| 42 | OFFSET_IMPOSSIBLE, | ||
| 43 | 60 | } state = START_NOT_FOUND; | |
| 44 | |||
| 45 | 60 | off = size = 0; | |
| 46 |
2/2✓ Branch 0 taken 146 times.
✓ Branch 1 taken 60 times.
|
206 | while (buf < end) { |
| 47 | int64_t obu_size; | ||
| 48 | int start_pos, type, temporal_id, spatial_id; | ||
| 49 | 146 | int len = parse_obu_header(buf, end - buf, &obu_size, &start_pos, | |
| 50 | &type, &temporal_id, &spatial_id); | ||
| 51 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 146 times.
|
146 | if (len < 0) |
| 52 | ✗ | return len; | |
| 53 | |||
| 54 |
2/2✓ Branch 0 taken 60 times.
✓ Branch 1 taken 86 times.
|
146 | switch (type) { |
| 55 | 60 | case AV1_OBU_TEMPORAL_DELIMITER: | |
| 56 | case AV1_OBU_REDUNDANT_FRAME_HEADER: | ||
| 57 | case AV1_OBU_TILE_LIST: | ||
| 58 | case AV1_OBU_PADDING: | ||
| 59 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 60 times.
|
60 | if (state == START_FOUND) |
| 60 | ✗ | state = END_FOUND; | |
| 61 | 60 | break; | |
| 62 | 86 | default: | |
| 63 |
2/2✓ Branch 0 taken 60 times.
✓ Branch 1 taken 26 times.
|
86 | if (state == START_NOT_FOUND) { |
| 64 | 60 | off = buf - start; | |
| 65 | 60 | state = START_FOUND; | |
| 66 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 26 times.
|
26 | } else if (state == END_FOUND) { |
| 67 | ✗ | state = OFFSET_IMPOSSIBLE; | |
| 68 | } | ||
| 69 |
2/2✓ Branch 0 taken 46 times.
✓ Branch 1 taken 40 times.
|
86 | if (pb) |
| 70 | 46 | avio_write(pb, buf, len); | |
| 71 | 86 | size += len; | |
| 72 | 86 | break; | |
| 73 | } | ||
| 74 | 146 | buf += len; | |
| 75 | } | ||
| 76 | |||
| 77 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 60 times.
|
60 | if (offset) |
| 78 | ✗ | *offset = state != OFFSET_IMPOSSIBLE ? off : -1; | |
| 79 | |||
| 80 | 60 | return size; | |
| 81 | } | ||
| 82 | |||
| 83 | 60 | int ff_av1_filter_obus(AVIOContext *pb, const uint8_t *buf, int size) | |
| 84 | { | ||
| 85 | 60 | return av1_filter_obus(pb, buf, size, NULL); | |
| 86 | } | ||
| 87 | |||
| 88 | ✗ | int ff_av1_filter_obus_buf(const uint8_t *in, uint8_t **out, | |
| 89 | int *size, int *offset) | ||
| 90 | { | ||
| 91 | FFIOContext pb; | ||
| 92 | uint8_t *buf; | ||
| 93 | int len, off, ret; | ||
| 94 | |||
| 95 | ✗ | len = ret = av1_filter_obus(NULL, in, *size, &off); | |
| 96 | ✗ | if (ret < 0) { | |
| 97 | ✗ | return ret; | |
| 98 | } | ||
| 99 | ✗ | if (off >= 0) { | |
| 100 | ✗ | *out = (uint8_t *)in; | |
| 101 | ✗ | *size = len; | |
| 102 | ✗ | *offset = off; | |
| 103 | |||
| 104 | ✗ | return 0; | |
| 105 | } | ||
| 106 | |||
| 107 | ✗ | buf = av_malloc(len + AV_INPUT_BUFFER_PADDING_SIZE); | |
| 108 | ✗ | if (!buf) | |
| 109 | ✗ | return AVERROR(ENOMEM); | |
| 110 | |||
| 111 | ✗ | ffio_init_write_context(&pb, buf, len); | |
| 112 | |||
| 113 | ✗ | ret = av1_filter_obus(&pb.pub, in, *size, NULL); | |
| 114 | av_assert1(ret == len); | ||
| 115 | |||
| 116 | ✗ | memset(buf + len, 0, AV_INPUT_BUFFER_PADDING_SIZE); | |
| 117 | |||
| 118 | ✗ | *out = buf; | |
| 119 | ✗ | *size = len; | |
| 120 | ✗ | *offset = 0; | |
| 121 | |||
| 122 | ✗ | return 0; | |
| 123 | } | ||
| 124 | |||
| 125 | ✗ | static inline void uvlc(GetBitContext *gb) | |
| 126 | { | ||
| 127 | ✗ | int leading_zeros = 0; | |
| 128 | |||
| 129 | ✗ | while (get_bits_left(gb)) { | |
| 130 | ✗ | if (get_bits1(gb)) | |
| 131 | ✗ | break; | |
| 132 | ✗ | leading_zeros++; | |
| 133 | } | ||
| 134 | |||
| 135 | ✗ | if (leading_zeros >= 32) | |
| 136 | ✗ | return; | |
| 137 | |||
| 138 | ✗ | skip_bits_long(gb, leading_zeros); | |
| 139 | } | ||
| 140 | |||
| 141 | 4 | static int parse_color_config(AV1SequenceParameters *seq_params, GetBitContext *gb) | |
| 142 | { | ||
| 143 | 4 | int twelve_bit = 0; | |
| 144 | 4 | int high_bitdepth = get_bits1(gb); | |
| 145 |
1/4✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
|
4 | if (seq_params->profile == AV_PROFILE_AV1_PROFESSIONAL && high_bitdepth) |
| 146 | ✗ | twelve_bit = get_bits1(gb); | |
| 147 | |||
| 148 | 4 | seq_params->bitdepth = 8 + (high_bitdepth * 2) + (twelve_bit * 2); | |
| 149 | |||
| 150 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
|
4 | if (seq_params->profile == AV_PROFILE_AV1_HIGH) |
| 151 | ✗ | seq_params->monochrome = 0; | |
| 152 | else | ||
| 153 | 4 | seq_params->monochrome = get_bits1(gb); | |
| 154 | |||
| 155 | 4 | seq_params->color_description_present_flag = get_bits1(gb); | |
| 156 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
|
4 | if (seq_params->color_description_present_flag) { |
| 157 | ✗ | seq_params->color_primaries = get_bits(gb, 8); | |
| 158 | ✗ | seq_params->transfer_characteristics = get_bits(gb, 8); | |
| 159 | ✗ | seq_params->matrix_coefficients = get_bits(gb, 8); | |
| 160 | } else { | ||
| 161 | 4 | seq_params->color_primaries = AVCOL_PRI_UNSPECIFIED; | |
| 162 | 4 | seq_params->transfer_characteristics = AVCOL_TRC_UNSPECIFIED; | |
| 163 | 4 | seq_params->matrix_coefficients = AVCOL_SPC_UNSPECIFIED; | |
| 164 | } | ||
| 165 | |||
| 166 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
|
4 | if (seq_params->monochrome) { |
| 167 | ✗ | seq_params->color_range = get_bits1(gb); | |
| 168 | ✗ | seq_params->chroma_subsampling_x = 1; | |
| 169 | ✗ | seq_params->chroma_subsampling_y = 1; | |
| 170 | ✗ | seq_params->chroma_sample_position = 0; | |
| 171 | ✗ | return 0; | |
| 172 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
|
4 | } else if (seq_params->color_primaries == AVCOL_PRI_BT709 && |
| 173 | ✗ | seq_params->transfer_characteristics == AVCOL_TRC_IEC61966_2_1 && | |
| 174 | ✗ | seq_params->matrix_coefficients == AVCOL_SPC_RGB) { | |
| 175 | ✗ | seq_params->chroma_subsampling_x = 0; | |
| 176 | ✗ | seq_params->chroma_subsampling_y = 0; | |
| 177 | } else { | ||
| 178 | 4 | seq_params->color_range = get_bits1(gb); | |
| 179 | |||
| 180 |
1/2✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
|
4 | if (seq_params->profile == AV_PROFILE_AV1_MAIN) { |
| 181 | 4 | seq_params->chroma_subsampling_x = 1; | |
| 182 | 4 | seq_params->chroma_subsampling_y = 1; | |
| 183 | ✗ | } else if (seq_params->profile == AV_PROFILE_AV1_HIGH) { | |
| 184 | ✗ | seq_params->chroma_subsampling_x = 0; | |
| 185 | ✗ | seq_params->chroma_subsampling_y = 0; | |
| 186 | } else { | ||
| 187 | ✗ | if (twelve_bit) { | |
| 188 | ✗ | seq_params->chroma_subsampling_x = get_bits1(gb); | |
| 189 | ✗ | if (seq_params->chroma_subsampling_x) | |
| 190 | ✗ | seq_params->chroma_subsampling_y = get_bits1(gb); | |
| 191 | else | ||
| 192 | ✗ | seq_params->chroma_subsampling_y = 0; | |
| 193 | } else { | ||
| 194 | ✗ | seq_params->chroma_subsampling_x = 1; | |
| 195 | ✗ | seq_params->chroma_subsampling_y = 0; | |
| 196 | } | ||
| 197 | } | ||
| 198 |
2/4✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 4 times.
✗ Branch 3 not taken.
|
4 | if (seq_params->chroma_subsampling_x && seq_params->chroma_subsampling_y) |
| 199 | 4 | seq_params->chroma_sample_position = get_bits(gb, 2); | |
| 200 | } | ||
| 201 | |||
| 202 | 4 | skip_bits1(gb); // separate_uv_delta_q | |
| 203 | |||
| 204 | 4 | return 0; | |
| 205 | } | ||
| 206 | |||
| 207 | 4 | static int parse_sequence_header(AV1SequenceParameters *seq_params, const uint8_t *buf, int size) | |
| 208 | { | ||
| 209 | GetBitContext gb; | ||
| 210 | int reduced_still_picture_header; | ||
| 211 | int frame_width_bits_minus_1, frame_height_bits_minus_1; | ||
| 212 | int size_bits, ret; | ||
| 213 | |||
| 214 | 4 | size_bits = get_obu_bit_length(buf, size, AV1_OBU_SEQUENCE_HEADER); | |
| 215 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
|
4 | if (size_bits < 0) |
| 216 | ✗ | return size_bits; | |
| 217 | |||
| 218 | 4 | ret = init_get_bits(&gb, buf, size_bits); | |
| 219 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
|
4 | if (ret < 0) |
| 220 | ✗ | return ret; | |
| 221 | |||
| 222 | 4 | memset(seq_params, 0, sizeof(*seq_params)); | |
| 223 | |||
| 224 | 4 | seq_params->profile = get_bits(&gb, 3); | |
| 225 | |||
| 226 | 4 | skip_bits1(&gb); // still_picture | |
| 227 | 4 | reduced_still_picture_header = get_bits1(&gb); | |
| 228 | |||
| 229 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
|
4 | if (reduced_still_picture_header) { |
| 230 | ✗ | seq_params->level = get_bits(&gb, 5); | |
| 231 | ✗ | seq_params->tier = 0; | |
| 232 | } else { | ||
| 233 | int initial_display_delay_present_flag, operating_points_cnt_minus_1; | ||
| 234 | int decoder_model_info_present_flag, buffer_delay_length_minus_1; | ||
| 235 | |||
| 236 |
2/2✓ Branch 1 taken 2 times.
✓ Branch 2 taken 2 times.
|
4 | if (get_bits1(&gb)) { // timing_info_present_flag |
| 237 | 2 | skip_bits_long(&gb, 32); // num_units_in_display_tick | |
| 238 | 2 | skip_bits_long(&gb, 32); // time_scale | |
| 239 | |||
| 240 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 2 times.
|
2 | if (get_bits1(&gb)) // equal_picture_interval |
| 241 | ✗ | uvlc(&gb); // num_ticks_per_picture_minus_1 | |
| 242 | |||
| 243 | 2 | decoder_model_info_present_flag = get_bits1(&gb); | |
| 244 |
1/2✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
|
2 | if (decoder_model_info_present_flag) { |
| 245 | 2 | buffer_delay_length_minus_1 = get_bits(&gb, 5); | |
| 246 | 2 | skip_bits_long(&gb, 32); // num_units_in_decoding_tick | |
| 247 | 2 | skip_bits(&gb, 10); // buffer_removal_time_length_minus_1 (5) | |
| 248 | // frame_presentation_time_length_minus_1 (5) | ||
| 249 | } | ||
| 250 | } else | ||
| 251 | 2 | decoder_model_info_present_flag = 0; | |
| 252 | |||
| 253 | 4 | initial_display_delay_present_flag = get_bits1(&gb); | |
| 254 | |||
| 255 | 4 | operating_points_cnt_minus_1 = get_bits(&gb, 5); | |
| 256 |
2/2✓ Branch 0 taken 4 times.
✓ Branch 1 taken 4 times.
|
8 | for (int i = 0; i <= operating_points_cnt_minus_1; i++) { |
| 257 | int seq_level_idx, seq_tier; | ||
| 258 | |||
| 259 | 4 | skip_bits(&gb, 12); // operating_point_idc | |
| 260 | 4 | seq_level_idx = get_bits(&gb, 5); | |
| 261 | |||
| 262 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
|
4 | if (seq_level_idx > 7) |
| 263 | ✗ | seq_tier = get_bits1(&gb); | |
| 264 | else | ||
| 265 | 4 | seq_tier = 0; | |
| 266 | |||
| 267 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 2 times.
|
4 | if (decoder_model_info_present_flag) { |
| 268 |
1/2✓ Branch 1 taken 2 times.
✗ Branch 2 not taken.
|
2 | if (get_bits1(&gb)) { // decoder_model_present_for_this_op |
| 269 | 2 | skip_bits_long(&gb, buffer_delay_length_minus_1 + 1); // decoder_buffer_delay | |
| 270 | 2 | skip_bits_long(&gb, buffer_delay_length_minus_1 + 1); // encoder_buffer_delay | |
| 271 | 2 | skip_bits1(&gb); // low_delay_mode_flag | |
| 272 | } | ||
| 273 | } | ||
| 274 | |||
| 275 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 2 times.
|
4 | if (initial_display_delay_present_flag) { |
| 276 |
1/2✓ Branch 1 taken 2 times.
✗ Branch 2 not taken.
|
2 | if (get_bits1(&gb)) // initial_display_delay_present_for_this_op |
| 277 | 2 | skip_bits(&gb, 4); // initial_display_delay_minus_1 | |
| 278 | } | ||
| 279 | |||
| 280 |
1/2✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
|
4 | if (i == 0) { |
| 281 | 4 | seq_params->level = seq_level_idx; | |
| 282 | 4 | seq_params->tier = seq_tier; | |
| 283 | } | ||
| 284 | } | ||
| 285 | } | ||
| 286 | |||
| 287 | 4 | frame_width_bits_minus_1 = get_bits(&gb, 4); | |
| 288 | 4 | frame_height_bits_minus_1 = get_bits(&gb, 4); | |
| 289 | |||
| 290 | 4 | skip_bits(&gb, frame_width_bits_minus_1 + 1); // max_frame_width_minus_1 | |
| 291 | 4 | skip_bits(&gb, frame_height_bits_minus_1 + 1); // max_frame_height_minus_1 | |
| 292 | |||
| 293 |
1/2✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
|
4 | if (!reduced_still_picture_header) { |
| 294 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 4 times.
|
4 | if (get_bits1(&gb)) // frame_id_numbers_present_flag |
| 295 | ✗ | skip_bits(&gb, 7); // delta_frame_id_length_minus_2 (4), additional_frame_id_length_minus_1 (3) | |
| 296 | } | ||
| 297 | |||
| 298 | 4 | skip_bits(&gb, 3); // use_128x128_superblock (1), enable_filter_intra (1), enable_intra_edge_filter (1) | |
| 299 | |||
| 300 |
1/2✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
|
4 | if (!reduced_still_picture_header) { |
| 301 | int enable_order_hint, seq_force_screen_content_tools; | ||
| 302 | |||
| 303 | 4 | skip_bits(&gb, 4); // enable_interintra_compound (1), enable_masked_compound (1) | |
| 304 | // enable_warped_motion (1), enable_dual_filter (1) | ||
| 305 | |||
| 306 | 4 | enable_order_hint = get_bits1(&gb); | |
| 307 |
1/2✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
|
4 | if (enable_order_hint) |
| 308 | 4 | skip_bits(&gb, 2); // enable_jnt_comp (1), enable_ref_frame_mvs (1) | |
| 309 | |||
| 310 |
1/2✓ Branch 1 taken 4 times.
✗ Branch 2 not taken.
|
4 | if (get_bits1(&gb)) // seq_choose_screen_content_tools |
| 311 | 4 | seq_force_screen_content_tools = 2; | |
| 312 | else | ||
| 313 | ✗ | seq_force_screen_content_tools = get_bits1(&gb); | |
| 314 | |||
| 315 |
1/2✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
|
4 | if (seq_force_screen_content_tools) { |
| 316 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 4 times.
|
4 | if (!get_bits1(&gb)) // seq_choose_integer_mv |
| 317 | ✗ | skip_bits1(&gb); // seq_force_integer_mv | |
| 318 | } | ||
| 319 | |||
| 320 |
1/2✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
|
4 | if (enable_order_hint) |
| 321 | 4 | skip_bits(&gb, 3); // order_hint_bits_minus_1 | |
| 322 | } | ||
| 323 | |||
| 324 | 4 | skip_bits(&gb, 3); // enable_superres (1), enable_cdef (1), enable_restoration (1) | |
| 325 | |||
| 326 | 4 | parse_color_config(seq_params, &gb); | |
| 327 | |||
| 328 | 4 | skip_bits1(&gb); // film_grain_params_present | |
| 329 | |||
| 330 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 4 times.
|
4 | if (get_bits_left(&gb)) |
| 331 | ✗ | return AVERROR_INVALIDDATA; | |
| 332 | |||
| 333 | 4 | return 0; | |
| 334 | } | ||
| 335 | |||
| 336 | ✗ | int ff_av1_parse_seq_header(AV1SequenceParameters *seq, const uint8_t *buf, int size) | |
| 337 | { | ||
| 338 | int is_av1c; | ||
| 339 | |||
| 340 | ✗ | if (size <= 0) | |
| 341 | ✗ | return AVERROR_INVALIDDATA; | |
| 342 | |||
| 343 | ✗ | is_av1c = !!(buf[0] & 0x80); | |
| 344 | ✗ | if (is_av1c) { | |
| 345 | GetBitContext gb; | ||
| 346 | ✗ | int ret, version = buf[0] & 0x7F; | |
| 347 | |||
| 348 | ✗ | if (version != 1 || size < 4) | |
| 349 | ✗ | return AVERROR_INVALIDDATA; | |
| 350 | |||
| 351 | ✗ | ret = init_get_bits8(&gb, buf, 4); | |
| 352 | ✗ | if (ret < 0) | |
| 353 | ✗ | return ret; | |
| 354 | |||
| 355 | ✗ | memset(seq, 0, sizeof(*seq)); | |
| 356 | |||
| 357 | ✗ | skip_bits(&gb, 8); | |
| 358 | ✗ | seq->profile = get_bits(&gb, 3); | |
| 359 | ✗ | seq->level = get_bits(&gb, 5); | |
| 360 | ✗ | seq->tier = get_bits(&gb, 1); | |
| 361 | ✗ | seq->bitdepth = get_bits(&gb, 1) * 2 + 8; | |
| 362 | ✗ | seq->bitdepth += get_bits(&gb, 1) * 2; | |
| 363 | ✗ | seq->monochrome = get_bits(&gb, 1); | |
| 364 | ✗ | seq->chroma_subsampling_x = get_bits(&gb, 1); | |
| 365 | ✗ | seq->chroma_subsampling_y = get_bits(&gb, 1); | |
| 366 | ✗ | seq->chroma_sample_position = get_bits(&gb, 2); | |
| 367 | ✗ | seq->color_primaries = AVCOL_PRI_UNSPECIFIED; | |
| 368 | ✗ | seq->transfer_characteristics = AVCOL_TRC_UNSPECIFIED; | |
| 369 | ✗ | seq->matrix_coefficients = AVCOL_SPC_UNSPECIFIED; | |
| 370 | |||
| 371 | ✗ | size -= 4; | |
| 372 | ✗ | buf += 4; | |
| 373 | } | ||
| 374 | |||
| 375 | ✗ | while (size > 0) { | |
| 376 | int64_t obu_size; | ||
| 377 | int start_pos, type, temporal_id, spatial_id; | ||
| 378 | ✗ | int len = parse_obu_header(buf, size, &obu_size, &start_pos, | |
| 379 | &type, &temporal_id, &spatial_id); | ||
| 380 | ✗ | if (len < 0) | |
| 381 | ✗ | return len; | |
| 382 | |||
| 383 | ✗ | switch (type) { | |
| 384 | ✗ | case AV1_OBU_SEQUENCE_HEADER: | |
| 385 | ✗ | if (!obu_size) | |
| 386 | ✗ | return AVERROR_INVALIDDATA; | |
| 387 | |||
| 388 | ✗ | return parse_sequence_header(seq, buf + start_pos, obu_size); | |
| 389 | ✗ | default: | |
| 390 | ✗ | break; | |
| 391 | } | ||
| 392 | ✗ | size -= len; | |
| 393 | ✗ | buf += len; | |
| 394 | } | ||
| 395 | |||
| 396 | ✗ | return is_av1c ? 0 : AVERROR_INVALIDDATA; | |
| 397 | } | ||
| 398 | |||
| 399 | 5 | int ff_isom_write_av1c(AVIOContext *pb, const uint8_t *buf, int size, | |
| 400 | int write_seq_header) | ||
| 401 | { | ||
| 402 | AVIOContext *meta_pb; | ||
| 403 | AV1SequenceParameters seq_params; | ||
| 404 | PutBitContext pbc; | ||
| 405 | uint8_t header[4], *meta; | ||
| 406 | const uint8_t *seq; | ||
| 407 | 5 | int ret, nb_seq = 0, seq_size, meta_size; | |
| 408 | |||
| 409 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 5 times.
|
5 | if (size <= 0) |
| 410 | ✗ | return AVERROR_INVALIDDATA; | |
| 411 | |||
| 412 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 4 times.
|
5 | if (buf[0] & 0x80) { |
| 413 | // first bit is nonzero, the passed data does not consist purely of | ||
| 414 | // OBUs. Expect that the data is already in AV1CodecConfigurationRecord | ||
| 415 | // format. | ||
| 416 | 1 | int config_record_version = buf[0] & 0x7f; | |
| 417 |
2/4✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 1 times.
|
1 | if (config_record_version != 1 || size < 4) { |
| 418 | ✗ | return AVERROR_INVALIDDATA; | |
| 419 | } | ||
| 420 | |||
| 421 | 1 | avio_write(pb, buf, size); | |
| 422 | |||
| 423 | 1 | return 0; | |
| 424 | } | ||
| 425 | |||
| 426 | 4 | ret = avio_open_dyn_buf(&meta_pb); | |
| 427 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
|
4 | if (ret < 0) |
| 428 | ✗ | return ret; | |
| 429 | |||
| 430 |
2/2✓ Branch 0 taken 4 times.
✓ Branch 1 taken 4 times.
|
8 | while (size > 0) { |
| 431 | int64_t obu_size; | ||
| 432 | int start_pos, type, temporal_id, spatial_id; | ||
| 433 | 4 | int len = parse_obu_header(buf, size, &obu_size, &start_pos, | |
| 434 | &type, &temporal_id, &spatial_id); | ||
| 435 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
|
4 | if (len < 0) { |
| 436 | ✗ | ret = len; | |
| 437 | ✗ | goto fail; | |
| 438 | } | ||
| 439 | |||
| 440 |
1/3✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
|
4 | switch (type) { |
| 441 | 4 | case AV1_OBU_SEQUENCE_HEADER: | |
| 442 | 4 | nb_seq++; | |
| 443 |
2/4✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 4 times.
|
4 | if (!obu_size || nb_seq > 1) { |
| 444 | ✗ | ret = AVERROR_INVALIDDATA; | |
| 445 | ✗ | goto fail; | |
| 446 | } | ||
| 447 | 4 | ret = parse_sequence_header(&seq_params, buf + start_pos, obu_size); | |
| 448 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
|
4 | if (ret < 0) |
| 449 | ✗ | goto fail; | |
| 450 | |||
| 451 | 4 | seq = buf; | |
| 452 | 4 | seq_size = len; | |
| 453 | 4 | break; | |
| 454 | ✗ | case AV1_OBU_METADATA: | |
| 455 | ✗ | if (!obu_size) { | |
| 456 | ✗ | ret = AVERROR_INVALIDDATA; | |
| 457 | ✗ | goto fail; | |
| 458 | } | ||
| 459 | ✗ | avio_write(meta_pb, buf, len); | |
| 460 | ✗ | break; | |
| 461 | ✗ | default: | |
| 462 | ✗ | break; | |
| 463 | } | ||
| 464 | 4 | size -= len; | |
| 465 | 4 | buf += len; | |
| 466 | } | ||
| 467 | |||
| 468 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
|
4 | if (!nb_seq) { |
| 469 | ✗ | ret = AVERROR_INVALIDDATA; | |
| 470 | ✗ | goto fail; | |
| 471 | } | ||
| 472 | |||
| 473 | 4 | init_put_bits(&pbc, header, sizeof(header)); | |
| 474 | |||
| 475 | 4 | put_bits(&pbc, 1, 1); // marker | |
| 476 | 4 | put_bits(&pbc, 7, 1); // version | |
| 477 | 4 | put_bits(&pbc, 3, seq_params.profile); | |
| 478 | 4 | put_bits(&pbc, 5, seq_params.level); | |
| 479 | 4 | put_bits(&pbc, 1, seq_params.tier); | |
| 480 | 4 | put_bits(&pbc, 1, seq_params.bitdepth > 8); | |
| 481 | 4 | put_bits(&pbc, 1, seq_params.bitdepth == 12); | |
| 482 | 4 | put_bits(&pbc, 1, seq_params.monochrome); | |
| 483 | 4 | put_bits(&pbc, 1, seq_params.chroma_subsampling_x); | |
| 484 | 4 | put_bits(&pbc, 1, seq_params.chroma_subsampling_y); | |
| 485 | 4 | put_bits(&pbc, 2, seq_params.chroma_sample_position); | |
| 486 | 4 | put_bits(&pbc, 8, 0); // padding | |
| 487 | 4 | flush_put_bits(&pbc); | |
| 488 | |||
| 489 | 4 | avio_write(pb, header, sizeof(header)); | |
| 490 |
1/2✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
|
4 | if (write_seq_header) { |
| 491 | 4 | avio_write(pb, seq, seq_size); | |
| 492 | } | ||
| 493 | |||
| 494 | 4 | meta_size = avio_get_dyn_buf(meta_pb, &meta); | |
| 495 |
1/2✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
|
4 | if (meta_size) |
| 496 | ✗ | avio_write(pb, meta, meta_size); | |
| 497 | |||
| 498 | 4 | fail: | |
| 499 | 4 | ffio_free_dyn_buf(&meta_pb); | |
| 500 | |||
| 501 | 4 | return ret; | |
| 502 | } | ||
| 503 |