From: "mypopy@gmail.com" <mypopy@gmail.com> To: FFmpeg development discussions and patches <ffmpeg-devel@ffmpeg.org> Cc: Thomas Siedel <thomas.ff@spin-digital.com> Subject: Re: [FFmpeg-devel] [PATCH v3 05/10] avformat: add demuxer and probe support for H266/VVC Date: Mon, 7 Nov 2022 10:28:04 +0800 Message-ID: <CACYjbn17ohTJjoY8_DY8XpHmVESrgGKC4+usUfo15=-6KDkLfQ@mail.gmail.com> (raw) In-Reply-To: <20221103122158.27169-6-thomas.ff@spin-digital.com> On Thu, Nov 3, 2022 at 8:23 PM Thomas Siedel <thomas.ff@spin-digital.com> wrote: > > Add demuxer to probe raw vvc and parse vvcc byte stream format. > > Signed-off-by: Thomas Siedel <thomas.ff@spin-digital.com> > --- > libavformat/Makefile | 1 + > libavformat/allformats.c | 1 + > libavformat/demux.c | 7 +- > libavformat/vvc.c | 919 +++++++++++++++++++++++++++++++++++++++ > libavformat/vvc.h | 99 +++++ > libavformat/vvcdec.c | 61 +++ > 6 files changed, 1086 insertions(+), 2 deletions(-) > create mode 100644 libavformat/vvc.c > create mode 100644 libavformat/vvc.h > create mode 100644 libavformat/vvcdec.c > > diff --git a/libavformat/Makefile b/libavformat/Makefile > index d7f198bf39..00ab4ded89 100644 > --- a/libavformat/Makefile > +++ b/libavformat/Makefile > @@ -595,6 +595,7 @@ OBJS-$(CONFIG_VOC_MUXER) += vocenc.o voc.o > OBJS-$(CONFIG_VPK_DEMUXER) += vpk.o > OBJS-$(CONFIG_VPLAYER_DEMUXER) += vplayerdec.o subtitles.o > OBJS-$(CONFIG_VQF_DEMUXER) += vqf.o > +OBJS-$(CONFIG_VVC_DEMUXER) += vvcdec.o rawdec.o > OBJS-$(CONFIG_W64_DEMUXER) += wavdec.o w64.o pcm.o > OBJS-$(CONFIG_W64_MUXER) += wavenc.o w64.o > OBJS-$(CONFIG_WAV_DEMUXER) += wavdec.o pcm.o > diff --git a/libavformat/allformats.c b/libavformat/allformats.c > index 47c419a009..a4e3822681 100644 > --- a/libavformat/allformats.c > +++ b/libavformat/allformats.c > @@ -474,6 +474,7 @@ extern const AVOutputFormat ff_voc_muxer; > extern const AVInputFormat ff_vpk_demuxer; > extern const AVInputFormat ff_vplayer_demuxer; > extern const AVInputFormat ff_vqf_demuxer; > +extern const AVInputFormat ff_vvc_demuxer; > extern const AVInputFormat ff_w64_demuxer; > extern const AVOutputFormat ff_w64_muxer; > extern const AVInputFormat ff_wav_demuxer; > diff --git a/libavformat/demux.c b/libavformat/demux.c > index 2dfd82a63c..8dbde23fcd 100644 > --- a/libavformat/demux.c > +++ b/libavformat/demux.c > @@ -120,6 +120,7 @@ static int set_codec_from_probe_data(AVFormatContext *s, AVStream *st, > { "mp3", AV_CODEC_ID_MP3, AVMEDIA_TYPE_AUDIO }, > { "mpegvideo", AV_CODEC_ID_MPEG2VIDEO, AVMEDIA_TYPE_VIDEO }, > { "truehd", AV_CODEC_ID_TRUEHD, AVMEDIA_TYPE_AUDIO }, > + { "vvc", AV_CODEC_ID_VVC, AVMEDIA_TYPE_VIDEO }, > { 0 } > }; > int score; > @@ -743,7 +744,8 @@ static int64_t select_from_pts_buffer(AVStream *st, int64_t *pts_buffer, int64_t > { > FFStream *const sti = ffstream(st); > int onein_oneout = st->codecpar->codec_id != AV_CODEC_ID_H264 && > - st->codecpar->codec_id != AV_CODEC_ID_HEVC; > + st->codecpar->codec_id != AV_CODEC_ID_HEVC && > + st->codecpar->codec_id != AV_CODEC_ID_VVC; > > if (!onein_oneout) { > int delay = sti->avctx->has_b_frames; > @@ -933,7 +935,8 @@ static void compute_pkt_fields(AVFormatContext *s, AVStream *st, > int64_t offset; > AVRational duration; > int onein_oneout = st->codecpar->codec_id != AV_CODEC_ID_H264 && > - st->codecpar->codec_id != AV_CODEC_ID_HEVC; > + st->codecpar->codec_id != AV_CODEC_ID_HEVC && > + st->codecpar->codec_id != AV_CODEC_ID_VVC; > > if (s->flags & AVFMT_FLAG_NOFILLIN) > return; > diff --git a/libavformat/vvc.c b/libavformat/vvc.c > new file mode 100644 > index 0000000000..b27a522009 > --- /dev/null > +++ b/libavformat/vvc.c > @@ -0,0 +1,919 @@ > +/* > + * VVC helper functions for muxers > + * > + * Copyright (C) 2022, Thomas Siedel > + * > + * This file is part of FFmpeg. > + * > + * FFmpeg is free software; you can redistribute it and/or > + * modify it under the terms of the GNU Lesser General Public > + * License as published by the Free Software Foundation; either > + * version 2.1 of the License, or (at your option) any later version. > + * > + * FFmpeg is distributed in the hope that it will be useful, > + * but WITHOUT ANY WARRANTY; without even the implied warranty of > + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU > + * Lesser General Public License for more details. > + * > + * You should have received a copy of the GNU Lesser General Public > + * License along with FFmpeg; if not, write to the Free Software > + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA > + */ > + > +#include "libavcodec/get_bits.h" > +#include "libavcodec/golomb.h" > +#include "libavcodec/vvc.h" > +#include "libavutil/intreadwrite.h" > +#include "avc.h" > +#include "avio.h" > +#include "avio_internal.h" > +#include "vvc.h" > + > +typedef struct VVCCNALUnitArray { > + uint8_t array_completeness; > + uint8_t NAL_unit_type; > + uint16_t num_nalus; > + uint16_t *nal_unit_length; > + uint8_t **nal_unit; > +} VVCCNALUnitArray; > + > +typedef struct VVCPTLRecord { > + uint8_t num_bytes_constraint_info; > + uint8_t general_profile_idc; > + uint8_t general_tier_flag; > + uint8_t general_level_idc; > + uint8_t ptl_frame_only_constraint_flag; > + uint8_t ptl_multilayer_enabled_flag; > + uint8_t general_constraint_info[9]; > + uint8_t *ptl_sublayer_level_present_flag; > + uint8_t *sublayer_level_idc; > + uint8_t ptl_num_sub_profiles; > + uint32_t *general_sub_profile_idc; > +} VVCPTLRecord; > + > +typedef struct VVCDecoderConfigurationRecord { > + uint8_t lengthSizeMinusOne; > + uint8_t ptl_present_flag; > + uint16_t ols_idx; > + uint8_t num_sublayers; > + uint8_t constant_frame_rate; > + uint8_t chroma_format_idc; > + uint8_t bit_depth_minus8; > + VVCPTLRecord ptl; > + uint16_t max_picture_width; > + uint16_t max_picture_height; > + uint16_t avg_frame_rate; > + uint8_t num_of_arrays; > + VVCCNALUnitArray *array; > +} VVCDecoderConfigurationRecord; > + > +typedef struct VVCCProfileTierLevel { > + uint8_t profile_idc; > + uint8_t tier_flag; > + uint8_t general_level_idc; > + uint8_t ptl_frame_only_constraint_flag; > + uint8_t ptl_multilayer_enabled_flag; > +// general_constraint_info > + uint8_t gci_present_flag; > + uint8_t gci_general_constraints[9]; > + uint8_t gci_num_reserved_bits; > +// end general_constraint_info > + uint8_t *ptl_sublayer_level_present_flag; > + uint8_t *sublayer_level_idc; > + uint8_t ptl_num_sub_profiles; > + uint32_t *general_sub_profile_idc; > +} VVCCProfileTierLevel; > + > + > +static void vvcc_update_ptl(VVCDecoderConfigurationRecord *vvcc, > + VVCCProfileTierLevel *ptl) > +{ > + /* > + * The level indication general_level_idc must indicate a level of > + * capability equal to or greater than the highest level indicated for the > + * highest tier in all the parameter sets. > + */ > + if (vvcc->ptl.general_tier_flag < ptl->tier_flag) > + vvcc->ptl.general_level_idc = ptl->general_level_idc; > + else > + vvcc->ptl.general_level_idc = FFMAX(vvcc->ptl.general_level_idc, ptl->general_level_idc); > + > + /* > + * The tier indication general_tier_flag must indicate a tier equal to or > + * greater than the highest tier indicated in all the parameter sets. > + */ > + vvcc->ptl.general_tier_flag = FFMAX(vvcc->ptl.general_tier_flag, ptl->tier_flag); > + > + /* > + * The profile indication general_profile_idc must indicate a profile to > + * which the stream associated with this configuration record conforms. > + * > + * If the sequence parameter sets are marked with different profiles, then > + * the stream may need examination to determine which profile, if any, the > + * entire stream conforms to. If the entire stream is not examined, or the > + * examination reveals that there is no profile to which the entire stream > + * conforms, then the entire stream must be split into two or more > + * sub-streams with separate configuration records in which these rules can > + * be met. > + * > + * Note: set the profile to the highest value for the sake of simplicity. > + */ > + vvcc->ptl.general_profile_idc = FFMAX(vvcc->ptl.general_profile_idc, ptl->profile_idc); > + > + /* > + * Each bit in flags may only be set if all > + * the parameter sets set that bit. > + */ > + vvcc->ptl.ptl_frame_only_constraint_flag &= ptl->ptl_frame_only_constraint_flag; > + vvcc->ptl.ptl_multilayer_enabled_flag &= ptl->ptl_multilayer_enabled_flag; > + > + /* > + * Constraints Info > + */ > + if(ptl->gci_present_flag) { > + vvcc->ptl.num_bytes_constraint_info = 9; > + memcpy(&vvcc->ptl.general_constraint_info[0], &ptl->gci_general_constraints[0], sizeof(uint8_t)*9); > + > + } else { > + vvcc->ptl.num_bytes_constraint_info = 1; > + memset(&vvcc->ptl.general_constraint_info[0], 0, sizeof(uint8_t)*9); > + } > + > + /* > + * Each bit in flags may only be set if one of > + * the parameter sets set that bit. > + */ > + vvcc->ptl.ptl_sublayer_level_present_flag = (uint8_t *) malloc(sizeof(uint8_t)*vvcc->num_sublayers-1); > + vvcc->ptl.sublayer_level_idc = (uint8_t *) malloc(sizeof(uint8_t)*vvcc->num_sublayers-1); > + > + memset(vvcc->ptl.ptl_sublayer_level_present_flag, 0, sizeof(uint8_t)*vvcc->num_sublayers-1); > + memset(vvcc->ptl.sublayer_level_idc, 0, sizeof(uint8_t)*vvcc->num_sublayers-1); > + > + for(int i = vvcc->num_sublayers - 2; i >= 0; i--) > + { > + vvcc->ptl.ptl_sublayer_level_present_flag[i] |= ptl->ptl_sublayer_level_present_flag[i]; > + if(vvcc->ptl.ptl_sublayer_level_present_flag[i]){ > + vvcc->ptl.sublayer_level_idc[i] = FFMAX(vvcc->ptl.sublayer_level_idc[i], ptl->sublayer_level_idc[i]); > + } > + else{ > + if(i == vvcc->num_sublayers - 1){ > + vvcc->ptl.sublayer_level_idc[i] = vvcc->ptl.general_level_idc; > + }else{ > + vvcc->ptl.sublayer_level_idc[i] = vvcc->ptl.sublayer_level_idc[i+1]; > + } > + } > + } > + > + vvcc->ptl.ptl_num_sub_profiles = FFMAX(vvcc->ptl.ptl_num_sub_profiles, ptl->ptl_num_sub_profiles); > + if( vvcc->ptl.ptl_num_sub_profiles ){ > + vvcc->ptl.general_sub_profile_idc = (uint32_t *) malloc(sizeof(uint32_t)*vvcc->ptl.ptl_num_sub_profiles); > + for(int i = 0; i < vvcc->ptl.ptl_num_sub_profiles; i++) { > + vvcc->ptl.general_sub_profile_idc[i] = ptl->general_sub_profile_idc[i]; > + } > + } > + else > + { > + vvcc->ptl.general_sub_profile_idc = (uint32_t *) malloc(sizeof(uint32_t)); > + } > +} > + > +static void vvcc_parse_ptl(GetBitContext *gb, > + VVCDecoderConfigurationRecord *vvcc, > + unsigned int profileTierPresentFlag, > + unsigned int max_sub_layers_minus1) > +{ > + VVCCProfileTierLevel general_ptl; > + int j; > + > + if(profileTierPresentFlag) { > + general_ptl.profile_idc = get_bits(gb, 7); > + general_ptl.tier_flag = get_bits1(gb); > + } > + general_ptl.general_level_idc = get_bits(gb, 8); > + > + general_ptl.ptl_frame_only_constraint_flag = get_bits1(gb); > + general_ptl.ptl_multilayer_enabled_flag = get_bits1(gb); > + if(profileTierPresentFlag) { // parse constraint info > + general_ptl.gci_present_flag = get_bits1(gb); > + if(general_ptl.gci_present_flag) { > + for (j = 0; j < 8; j++) > + general_ptl.gci_general_constraints[j] = get_bits(gb, 8); > + general_ptl.gci_general_constraints[8] = 0; > + general_ptl.gci_general_constraints[8] = get_bits(gb, 7); > + > + general_ptl.gci_num_reserved_bits = get_bits(gb, 8); > + skip_bits(gb, general_ptl.gci_num_reserved_bits); > + } > + while(gb->index % 8 != 0) > + skip_bits1(gb); > + } > + > + general_ptl.ptl_sublayer_level_present_flag = (uint8_t *) malloc(sizeof(uint8_t)*max_sub_layers_minus1); > + for(int i = max_sub_layers_minus1-1; i >= 0; i--) { > + general_ptl.ptl_sublayer_level_present_flag[i] = get_bits1(gb); > + } > + while(gb->index%8 != 0) > + skip_bits1(gb); > + > + general_ptl.sublayer_level_idc = (uint8_t *) malloc(sizeof(uint8_t)*max_sub_layers_minus1); > + for(int i = max_sub_layers_minus1-1; i >= 0; i--) { > + if( general_ptl.ptl_sublayer_level_present_flag[i] ) > + general_ptl.sublayer_level_idc[i] = get_bits(gb, 8); > + } > + > + if(profileTierPresentFlag) { > + general_ptl.ptl_num_sub_profiles = get_bits(gb, 8); > + if( general_ptl.ptl_num_sub_profiles) { > + general_ptl.general_sub_profile_idc = (uint32_t *) malloc(sizeof(uint32_t)*general_ptl.ptl_num_sub_profiles); > + for(int i = 0; i < general_ptl.ptl_num_sub_profiles; i++) { > + general_ptl.general_sub_profile_idc[i] = get_bits_long(gb, 32); > + } > + } > + else > + { > + general_ptl.general_sub_profile_idc = (uint32_t *) malloc(sizeof(uint32_t)); > + } > + } > + > + vvcc_update_ptl(vvcc, &general_ptl); > + > + free(general_ptl.ptl_sublayer_level_present_flag); > + free(general_ptl.sublayer_level_idc); > + free(general_ptl.general_sub_profile_idc); > +} > + > +static int vvcc_parse_vps(GetBitContext *gb, > + VVCDecoderConfigurationRecord *vvcc) > +{ > + unsigned int vps_max_layers_minus1; > + unsigned int vps_max_sub_layers_minus1; > + unsigned int vps_default_ptl_dpb_hrd_max_tid_flag; > + unsigned int vps_all_independant_layer_flag; > + unsigned int vps_each_layer_is_an_ols_flag; > + unsigned int vps_ols_mode_idc; > + in FFmpeg, the indent size is 4 > + unsigned int *vps_pt_present_flag; > + unsigned int *vps_ptl_max_tid; > + unsigned int vps_num_ptls_minus1 = 0; > + > + > + /* > + * vps_video_parameter_set_id u(4) > + */ > + skip_bits(gb, 4); > + > + vps_max_layers_minus1 = get_bits(gb, 6); > + vps_max_sub_layers_minus1 = get_bits(gb, 3); > + > + /* > + * numTemporalLayers greater than 1 indicates that the stream to which this > + * configuration record applies is temporally scalable and the contained > + * number of temporal layers (also referred to as temporal sub-layer or > + * sub-layer in ISO/IEC 23008-2) is equal to numTemporalLayers. Value 1 > + * indicates that the stream is not temporally scalable. Value 0 indicates > + * that it is unknown whether the stream is temporally scalable. > + */ > + vvcc->num_sublayers = FFMAX(vvcc->num_sublayers, > + vps_max_sub_layers_minus1 + 1); > + > + > + if(vps_max_layers_minus1 > 0 && vps_max_sub_layers_minus1 > 0) > + vps_default_ptl_dpb_hrd_max_tid_flag = get_bits1(gb); //vps_default_ptl_dpb_hrd_max_tid_flags > + if(vps_max_layers_minus1 > 0) > + vps_all_independant_layer_flag = get_bits1(gb); > + > + for(int i =0; i <= vps_max_layers_minus1; i++) { > + skip_bits(gb, 6); //vps_default_ptl_dpb_hrd_max_tid_flagsvps_layer_id[i] > + if(i > 0 && !vps_all_independant_layer_flag) { > + if(get_bits1(gb)) { // vps_independant_layer_flag > + unsigned int vps_max_tid_ref_present_flag = get_bits1(gb); > + for (int j =0; j<i; j++) { > + if(vps_max_tid_ref_present_flag && get_bits1(gb)) // vps_direct_ref_layer_flag[i][j] > + skip_bits(gb, 3); //vps_max_tid_il_ref_pics_plus1 > + } > + } > + } > + } > + > + if(vps_max_layers_minus1 > 0) { > + if(vps_all_independant_layer_flag) > + vps_each_layer_is_an_ols_flag = get_bits1(gb); > + if(vps_each_layer_is_an_ols_flag) { > + if(!vps_all_independant_layer_flag) > + vps_ols_mode_idc = get_bits(gb, 2); > + if(vps_ols_mode_idc == 2) { > + unsigned int vps_num_output_layer_sets_minus2 = get_bits(gb, 8); > + for( int i = 1; i <= vps_num_output_layer_sets_minus2 +1; i++) { > + for( int j = 0; j <= vps_max_layers_minus1; j++) { > + skip_bits1(gb); > + } > + } > + } > + } > + vps_num_ptls_minus1 = get_bits(gb, 8); > + } > + > + vps_pt_present_flag = (unsigned int *) malloc(sizeof(unsigned int) * (vps_num_ptls_minus1+1)); > + vps_ptl_max_tid = (unsigned int *) malloc(sizeof(unsigned int) * (vps_num_ptls_minus1+1)); > + for(int i = 0; i <= vps_num_ptls_minus1; i++) { > + if(i>0) > + vps_pt_present_flag[i] = get_bits1(gb); > + if(!vps_default_ptl_dpb_hrd_max_tid_flag) > + vps_ptl_max_tid[i] = get_bits(gb, 3); > + } > + > + while(gb->index%8 != 0) > + skip_bits1(gb); > + > + for(int i = 0; i <= vps_num_ptls_minus1; i++) { > + vvcc_parse_ptl(gb, vvcc, vps_pt_present_flag[i], vps_ptl_max_tid[i]); > + } > + > + free(vps_pt_present_flag); > + free(vps_ptl_max_tid); > + > + /* nothing useful for vvcc past this point */ > + return 0; > +} > + > +static int vvcc_parse_sps(GetBitContext *gb, > + VVCDecoderConfigurationRecord *vvcc) > +{ > + unsigned int sps_max_sub_layers_minus1, log2_ctu_size_minus5; > + //unsigned int num_short_term_ref_pic_sets, num_delta_pocs[VVC_MAX_REF_PIC_LISTS]; > + //unsigned int sps_chroma_format_idc; > + unsigned int sps_subpic_same_size_flag, sps_pic_height_max_in_luma_sample, sps_pic_width_max_in_luma_sample; > + unsigned int sps_independant_subpics_flag; > + unsigned int flag; > + > + skip_bits(gb, 8); // sps_seq_parameter_set_id && sps_video_parameter_set_id > + sps_max_sub_layers_minus1 = get_bits (gb, 3); > + > + /* > + * numTemporalLayers greater than 1 indicates that the stream to which this > + * configuration record applies is temporally scalable and the contained > + * number of temporal layers (also referred to as temporal sub-layer or > + * sub-layer in ISO/IEC 23008-2) is equal to numTemporalLayers. Value 1 > + * indicates that the stream is not temporally scalable. Value 0 indicates > + * that it is unknown whether the stream is temporally scalable. > + */ > + vvcc->num_sublayers = FFMAX(vvcc->num_sublayers, > + sps_max_sub_layers_minus1 + 1); > + > + vvcc->chroma_format_idc = get_bits(gb, 2); > + log2_ctu_size_minus5 = get_bits(gb, 2); > + > + if(get_bits1(gb)) //sps_ptl_dpb_hrd_params_present_flag > + vvcc_parse_ptl(gb, vvcc, 1, sps_max_sub_layers_minus1); > + > + flag = get_bits(gb, 1); //skip_bits1(gb); //sps_gdr_enabled_flag > + flag = get_bits(gb, 1); //sps_ref_pic_resampling_enabled_flag > + if(flag){ //sps_ref_pic_resampling_enabled_flag > + flag = get_bits(gb, 1); //skip_bits1(gb); //sps_res_change_in_clvs_allowed_flag > + } > + > + sps_pic_width_max_in_luma_sample = get_ue_golomb_long(gb); > + vvcc->max_picture_width = FFMAX(vvcc->max_picture_width , sps_pic_width_max_in_luma_sample); > + sps_pic_height_max_in_luma_sample= get_ue_golomb_long(gb); > + vvcc->max_picture_height = FFMAX(vvcc->max_picture_height , sps_pic_height_max_in_luma_sample); > + > + if(get_bits1(gb)) { > + get_ue_golomb_long(gb); // sps_conf_win_left_offset > + get_ue_golomb_long(gb); // sps_conf_win_right_offset > + get_ue_golomb_long(gb); // sps_conf_win_top_offset > + get_ue_golomb_long(gb); // sps_conf_win_bottom_offset > + } > + > + if(get_bits1(gb)) { // sps_subpic_info_present_flag > + unsigned int sps_num_subpics_minus1 = get_ue_golomb_long(gb); > + if(sps_num_subpics_minus1 > 0) { // sps_num_subpics_minus1 > + sps_independant_subpics_flag = get_bits1(gb); > + sps_subpic_same_size_flag = get_bits1(gb); > + } > + for(int i = 0; sps_num_subpics_minus1 > 0 && i <= sps_num_subpics_minus1; i++) { > + if(!sps_subpic_same_size_flag || i == 0) { > + int len = FFMIN(log2_ctu_size_minus5 +5, 16); > + if(i > 0 && sps_pic_width_max_in_luma_sample > 128) > + skip_bits(gb, len); > + if(i > 0 && sps_pic_height_max_in_luma_sample > 128) > + skip_bits(gb, len); > + if(i < sps_num_subpics_minus1 && sps_pic_width_max_in_luma_sample > 128) > + skip_bits(gb, len); > + if(i < sps_num_subpics_minus1 && sps_pic_height_max_in_luma_sample > 128) > + skip_bits(gb, len); > + } > + if(!sps_independant_subpics_flag) { > + skip_bits(gb, 2); // sps_subpic_treated_as_pic_flag && sps_loop_filter_across_subpic_enabled_flag > + } > + } > + get_ue_golomb_long(gb); // sps_subpic_id_len_minus1 > + if(get_bits1(gb)) { // sps_subpic_id_mapping_explicitly_signalled_flag > + if(get_bits1(gb)) // sps_subpic_id_mapping_present_flag > + for(int i = 0; i <= sps_num_subpics_minus1; i++) { > + skip_bits1(gb); // sps_subpic_id[i] > + } > + } > + } > + vvcc->bit_depth_minus8 = get_ue_golomb_long(gb); > + > + /* nothing useful for vvcc past this point */ > + return 0; > +} > + > +static int vvcc_parse_pps(GetBitContext *gb, > + VVCDecoderConfigurationRecord *vvcc) > +{ > + > + // Nothing of importance to parse in PPS > + /* nothing useful for vvcc past this point */ > + return 0; > +} > + > +static void nal_unit_parse_header(GetBitContext *gb, uint8_t *nal_type) > +{ > + /* > + * forbidden_zero_bit u(1) > + * nuh_reserved_zero_bit u(1) > + * nuh_layer_id u(6) > + */ > + skip_bits(gb, 8); > + *nal_type = get_bits(gb, 5); > + > + /* > + * nuh_temporal_id_plus1 u(3) > + */ > + skip_bits(gb, 3); > +} > + > +static int vvcc_array_add_nal_unit(uint8_t *nal_buf, uint32_t nal_size, > + uint8_t nal_type, int ps_array_completeness, > + VVCDecoderConfigurationRecord *vvcc) > +{ > + int ret; > + uint8_t index; > + uint16_t num_nalus; > + VVCCNALUnitArray *array; > + > + for (index = 0; index < vvcc->num_of_arrays; index++) > + if (vvcc->array[index].NAL_unit_type == nal_type) > + break; > + > + if (index >= vvcc->num_of_arrays) { > + uint8_t i; > + > + ret = av_reallocp_array(&vvcc->array, index + 1, sizeof(VVCCNALUnitArray)); > + if (ret < 0) > + return ret; > + > + for (i = vvcc->num_of_arrays; i <= index; i++) > + memset(&vvcc->array[i], 0, sizeof(VVCCNALUnitArray)); > + vvcc->num_of_arrays = index + 1; > + } > + > + array = &vvcc->array[index]; > + num_nalus = array->num_nalus; > + > + ret = av_reallocp_array(&array->nal_unit, num_nalus + 1, sizeof(uint8_t*)); > + if (ret < 0) > + return ret; > + > + ret = av_reallocp_array(&array->nal_unit_length, num_nalus + 1, sizeof(uint16_t)); > + if (ret < 0) > + return ret; > + > + array->nal_unit [num_nalus] = nal_buf; > + array->nal_unit_length[num_nalus] = nal_size; > + array->NAL_unit_type = nal_type; > + array->num_nalus++; > + > + /* > + * When the sample entry name is ‘vvc1’, the default and mandatory value of > + * array_completeness is 1 for arrays of all types of parameter sets, and 0 > + * for all other arrays. When the sample entry name is ‘hev1’, the default > + * value of array_completeness is 0 for all arrays. > + */ > + if (nal_type == VVC_VPS_NUT || nal_type == VVC_SPS_NUT || nal_type == VVC_PPS_NUT) > + array->array_completeness = ps_array_completeness; > + > + return 0; > +} > + > +static int vvcc_add_nal_unit(uint8_t *nal_buf, uint32_t nal_size, > + int ps_array_completeness, > + VVCDecoderConfigurationRecord *vvcc) > +{ > + int ret = 0; > + GetBitContext gbc; > + uint8_t nal_type; > + uint8_t *rbsp_buf; > + uint32_t rbsp_size; > + > + rbsp_buf = ff_nal_unit_extract_rbsp(nal_buf, nal_size, &rbsp_size, 2); > + if (!rbsp_buf) { > + ret = AVERROR(ENOMEM); > + goto end; > + } > + > + ret = init_get_bits8(&gbc, rbsp_buf, rbsp_size); > + if (ret < 0) > + goto end; > + > + nal_unit_parse_header(&gbc, &nal_type); > + > + /* > + * Note: only 'declarative' SEI messages are allowed in > + * vvcc. Perhaps the SEI playload type should be checked > + * and non-declarative SEI messages discarded? > + */ > + switch (nal_type) { > + case VVC_OPI_NUT: > + case VVC_VPS_NUT: > + case VVC_SPS_NUT: > + case VVC_PPS_NUT: > + case VVC_PREFIX_SEI_NUT: > + case VVC_SUFFIX_SEI_NUT: > + ret = vvcc_array_add_nal_unit(nal_buf, nal_size, nal_type, > + ps_array_completeness, vvcc); > + if (ret < 0) > + goto end; > + else if (nal_type == VVC_VPS_NUT) > + ret = vvcc_parse_vps(&gbc, vvcc); > + else if (nal_type == VVC_SPS_NUT) > + ret = vvcc_parse_sps(&gbc, vvcc); > + else if (nal_type == VVC_PPS_NUT) > + ret = vvcc_parse_pps(&gbc, vvcc); > + else if (nal_type == VVC_OPI_NUT) { > + // not yet supported > + } > + if (ret < 0) > + goto end; > + break; > + default: > + ret = AVERROR_INVALIDDATA; > + goto end; > + } > + > +end: > + av_free(rbsp_buf); > + return ret; > +} > + > +static void vvcc_init(VVCDecoderConfigurationRecord *vvcc) > +{ > + memset(vvcc, 0, sizeof(VVCDecoderConfigurationRecord)); > + vvcc->lengthSizeMinusOne = 3; // 4 bytes > + > + vvcc->ptl.num_bytes_constraint_info = 1; > + > + vvcc->ptl_present_flag = 1; > +} > + > +static void vvcc_close(VVCDecoderConfigurationRecord *vvcc) > +{ > + uint8_t i; > + > + for (i = 0; i < vvcc->num_of_arrays; i++) { > + vvcc->array[i].num_nalus = 0; > + av_freep(&vvcc->array[i].nal_unit); > + av_freep(&vvcc->array[i].nal_unit_length); > + } > + > + free(vvcc->ptl.ptl_sublayer_level_present_flag); > + free(vvcc->ptl.sublayer_level_idc); > + free(vvcc->ptl.general_sub_profile_idc); > + > + vvcc->num_of_arrays = 0; > + av_freep(&vvcc->array); > +} > + > +static int vvcc_write(AVIOContext *pb, VVCDecoderConfigurationRecord *vvcc) > +{ > + uint8_t i; > + uint16_t j, vps_count = 0, sps_count = 0, pps_count = 0; > + unsigned char *buf = NULL; > + /* > + * It's unclear how to properly compute these fields, so > + * let's always set them to values meaning 'unspecified'. > + */ > + vvcc->avg_frame_rate = 0; > + vvcc->constant_frame_rate = 1; > + > + av_log(NULL, AV_LOG_TRACE, "lengthSizeMinusOne: %"PRIu8"\n", > + vvcc->lengthSizeMinusOne); > + av_log(NULL, AV_LOG_TRACE, "ptl_present_flag: %"PRIu8"\n", > + vvcc->ptl_present_flag); > + av_log(NULL, AV_LOG_TRACE, "ols_idx: %"PRIu16"\n", > + vvcc->ols_idx); > + av_log(NULL, AV_LOG_TRACE, "num_sublayers: %"PRIu8"\n", > + vvcc->num_sublayers); > + av_log(NULL, AV_LOG_TRACE, "constant_frame_rate: %"PRIu8"\n", > + vvcc->constant_frame_rate); > + av_log(NULL, AV_LOG_TRACE, "chroma_format_idc: %"PRIu8"\n", > + vvcc->chroma_format_idc); > + > + av_log(NULL, AV_LOG_TRACE, "bit_depth_minus8: %"PRIu8"\n", > + vvcc->bit_depth_minus8); > + av_log(NULL, AV_LOG_TRACE, "num_bytes_constraint_info: %"PRIu8"\n", > + vvcc->ptl.num_bytes_constraint_info); > + av_log(NULL, AV_LOG_TRACE, "general_profile_idc: %"PRIu8"\n", > + vvcc->ptl.general_profile_idc); > + av_log(NULL, AV_LOG_TRACE, "general_tier_flag: %"PRIu8"\n", > + vvcc->ptl.general_tier_flag); > + av_log(NULL, AV_LOG_TRACE, "general_level_idc: %"PRIu8"\n", > + vvcc->ptl.general_level_idc); > + av_log(NULL, AV_LOG_TRACE, "ptl_frame_only_constraint_flag: %"PRIu8"\n", > + vvcc->ptl.ptl_frame_only_constraint_flag); > + av_log(NULL, AV_LOG_TRACE, "ptl_multilayer_enabled_flag: %"PRIu8"\n", > + vvcc->ptl.ptl_multilayer_enabled_flag); > + for (i = 0; i < vvcc->ptl.num_bytes_constraint_info; i++) { > + av_log(NULL, AV_LOG_TRACE, "general_constraint_info[%d]: %"PRIu8"\n", > + i, vvcc->ptl.general_constraint_info[i]); > + } > + > + for (i = 0; i < vvcc->num_sublayers-1; i++) { > + av_log(NULL, AV_LOG_TRACE, "ptl_sublayer_level_present_flag[%"PRIu8"]: %"PRIu8"\n", > + i, vvcc->ptl.ptl_sublayer_level_present_flag[i]); > + av_log(NULL, AV_LOG_TRACE, "sublayer_level_idc[%"PRIu8"]: %"PRIu8"\n", > + i, vvcc->ptl.sublayer_level_idc[i]); > + } > + > + av_log(NULL, AV_LOG_TRACE, "num_sub_profiles: %"PRIu8"\n", > + vvcc->ptl.ptl_num_sub_profiles); > + > + for (i = 0; i < vvcc->ptl.ptl_num_sub_profiles; i++) { > + av_log(NULL, AV_LOG_TRACE, "general_sub_profile_idc[%"PRIu8"]: %"PRIx32"\n", > + i, vvcc->ptl.general_sub_profile_idc[i]); > + } > + > + av_log(NULL, AV_LOG_TRACE, "max_picture_width: %"PRIu16"\n", > + vvcc->max_picture_width); > + av_log(NULL, AV_LOG_TRACE, "max_picture_height: %"PRIu16"\n", > + vvcc->max_picture_height); > + av_log(NULL, AV_LOG_TRACE, "avg_frame_rate: %"PRIu16"\n", > + vvcc->avg_frame_rate); > + > + av_log(NULL, AV_LOG_TRACE, "num_of_arrays: %"PRIu8"\n", > + vvcc->num_of_arrays); > + for (i = 0; i < vvcc->num_of_arrays; i++) { > + av_log(NULL, AV_LOG_TRACE, "array_completeness[%"PRIu8"]: %"PRIu8"\n", > + i, vvcc->array[i].array_completeness); > + av_log(NULL, AV_LOG_TRACE, "NAL_unit_type[%"PRIu8"]: %"PRIu8"\n", > + i, vvcc->array[i].NAL_unit_type); > + av_log(NULL, AV_LOG_TRACE, "num_nalus[%"PRIu8"]: %"PRIu16"\n", > + i, vvcc->array[i].num_nalus); > + for (j = 0; j < vvcc->array[i].num_nalus; j++) > + av_log(NULL, AV_LOG_TRACE, > + "nal_unit_length[%"PRIu8"][%"PRIu16"]: %"PRIu16"\n", > + i, j, vvcc->array[i].nal_unit_length[j]); > + } > + > + /* > + * We need at least one of each: VPS and SPS. > + */ > + for (i = 0; i < vvcc->num_of_arrays; i++) > + switch (vvcc->array[i].NAL_unit_type) { > + case VVC_VPS_NUT: > + vps_count += vvcc->array[i].num_nalus; > + break; > + case VVC_SPS_NUT: > + sps_count += vvcc->array[i].num_nalus; > + break; > + case VVC_PPS_NUT: > + pps_count += vvcc->array[i].num_nalus; > + break; > + default: > + break; > + } > + > + if (!sps_count || sps_count > VVC_MAX_SPS_COUNT) > + return AVERROR_INVALIDDATA; > + > + /* bit(5) reserved = ‘11111’b; > + unsigned int (2) LengthSizeMinusOne > + unsigned int (1) ptl_present_flag */ > + avio_w8(pb, vvcc->lengthSizeMinusOne << 1 | vvcc->ptl_present_flag | 0xf8); > + > + if(vvcc->ptl_present_flag) { > + /* > + * unsigned int(9) ols_idx; > + * unsigned int(3) num_sublayers; > + * unsigned int(2) constant_frame_rate; > + * unsigned int(2) chroma_format_idc; */ > + avio_wb16(pb, vvcc->ols_idx << 7 | vvcc->num_sublayers << 4 | vvcc->constant_frame_rate << 2 | vvcc->chroma_format_idc); > + > + /* unsigned int(3) bit_depth_minus8; > + bit(5) reserved = ‘11111’b;*/ > + avio_w8(pb, vvcc->bit_depth_minus8 << 5 | 0x1f); > + > + //VVCPTLRecord > + > + /* bit(2) reserved = ‘00’b; > + unsigned int (6) num_bytes_constraint_info */ > + avio_w8(pb, vvcc->ptl.num_bytes_constraint_info & 0x3f); > + > + /* unsigned int (7) general_profile_idc > + unsigned int (1) general_tier_flag */ > + avio_w8(pb, vvcc->ptl.general_profile_idc << 1 | vvcc->ptl.general_tier_flag); > + > + /* unsigned int (8) general_level_idc */ > + avio_w8(pb, vvcc->ptl.general_level_idc); > + > + /* > + * unsigned int (1) ptl_frame_only_constraint_flag > + * unsigned int (1) ptl_multilayer_enabled_flag > + * unsigned int (8*num_bytes_constraint_info -2) general_constraint_info */ > + buf = (unsigned char *) malloc(sizeof(unsigned char) * vvcc->ptl.num_bytes_constraint_info); > + *buf = vvcc->ptl.ptl_frame_only_constraint_flag << vvcc->ptl.num_bytes_constraint_info*8-1 | vvcc->ptl.ptl_multilayer_enabled_flag << vvcc->ptl.num_bytes_constraint_info*8-2 | *vvcc->ptl.general_constraint_info >> 2; > + avio_write(pb, buf, vvcc->ptl.num_bytes_constraint_info); > + free(buf); > + > + if(vvcc->num_sublayers > 1) { > + uint8_t ptl_sublayer_level_present_flags = 0; > + for(int i = vvcc->num_sublayers -2; i >= 0; i--) { > + ptl_sublayer_level_present_flags = (ptl_sublayer_level_present_flags << 1 | vvcc->ptl.ptl_sublayer_level_present_flag[i]); > + } > + avio_w8(pb, ptl_sublayer_level_present_flags); > + } > + > + for(int i = vvcc->num_sublayers -2; i >= 0; i--) { > + if(vvcc->ptl.ptl_sublayer_level_present_flag[i]) > + avio_w8(pb, vvcc->ptl.sublayer_level_idc[i]); > + } > + > + /* unsigned int(8) num_sub_profiles; */ > + avio_w8(pb, vvcc->ptl.ptl_num_sub_profiles); > + > + for(int j = 0; j < vvcc->ptl.ptl_num_sub_profiles; j++) { > + /* unsigned int(32) general_sub_profile_idc[j]; */ > + avio_wb32(pb, vvcc->ptl.general_sub_profile_idc[j]); > + } > + > + //End of VvcPTLRecord > + > + /* > + * unsigned int(16) max_picture_width;*/ > + avio_wb16(pb, vvcc->max_picture_width); > + > + /* > + * unsigned int(16) max_picture_height;*/ > + avio_wb16(pb, vvcc->max_picture_height); > + > + /* > + * unsigned int(16) avg_frame_rate; */ > + avio_wb16(pb, vvcc->avg_frame_rate); > + } > + > + /* unsigned int(8) num_of_arrays; */ > + avio_w8(pb, vvcc->num_of_arrays); > + > + for (i = 0; i < vvcc->num_of_arrays; i++) { > + /* > + * bit(1) array_completeness; > + * unsigned int(2) reserved = 0; > + * unsigned int(5) NAL_unit_type; > + */ > + avio_w8(pb, vvcc->array[i].array_completeness << 7 | > + vvcc->array[i].NAL_unit_type & 0x1f); > + /* unsigned int(16) num_nalus; */ > + if(vvcc->array[i].NAL_unit_type != VVC_DCI_NUT && vvcc->array[i].NAL_unit_type != VVC_OPI_NUT) > + avio_wb16(pb, vvcc->array[i].num_nalus); > + for (j = 0; j < vvcc->array[i].num_nalus; j++) { > + /* unsigned int(16) nal_unit_length; */ > + avio_wb16(pb, vvcc->array[i].nal_unit_length[j]); > + > + /* bit(8*nal_unit_length) nal_unit; */ > + avio_write(pb, vvcc->array[i].nal_unit[j], vvcc->array[i].nal_unit_length[j]); > + } > + } > + > + return 0; > +} > + > +int ff_vvc_annexb2mp4(AVIOContext *pb, const uint8_t *buf_in, > + int size, int filter_ps, int *ps_count) > +{ > + int num_ps = 0, ret = 0; > + uint8_t *buf, *end, *start = NULL; > + > + if (!filter_ps) { > + ret = ff_avc_parse_nal_units(pb, buf_in, size); > + goto end; > + } > + > + ret = ff_avc_parse_nal_units_buf(buf_in, &start, &size); > + if (ret < 0) > + goto end; > + > + ret = 0; > + buf = start; > + end = start + size; > + > + while (end - buf > 4) { > + uint32_t len = FFMIN(AV_RB32(buf), end - buf - 4); > + uint8_t type = (buf[5] >> 3); > + > + buf += 4; > + > + switch (type) { > + case VVC_VPS_NUT: > + case VVC_SPS_NUT: > + case VVC_PPS_NUT: > + num_ps++; > + break; > + default: > + ret += 4 + len; > + avio_wb32(pb, len); > + avio_write(pb, buf, len); > + break; > + } > + > + buf += len; > + } > + > +end: > + av_free(start); > + if (ps_count) > + *ps_count = num_ps; > + return ret; > +} > + > +int ff_vvc_annexb2mp4_buf(const uint8_t *buf_in, uint8_t **buf_out, > + int *size, int filter_ps, int *ps_count) > +{ > + AVIOContext *pb; > + int ret; > + > + ret = avio_open_dyn_buf(&pb); > + if (ret < 0) > + return ret; > + > + ret = ff_vvc_annexb2mp4(pb, buf_in, *size, filter_ps, ps_count); > + if (ret < 0) { > + ffio_free_dyn_buf(&pb); > + return ret; > + } > + > + *size = avio_close_dyn_buf(pb, buf_out); > + > + return 0; > +} > + > +int ff_isom_write_vvcc(AVIOContext *pb, const uint8_t *data, > + int size, int ps_array_completeness) > +{ > + VVCDecoderConfigurationRecord vvcc; > + uint8_t *buf, *end, *start; > + int ret; > + > + if (size < 6) { > + /* We can't write a valid vvcc from the provided data */ > + return AVERROR_INVALIDDATA; > + } else if (*data == 1) { > + /* Data is already vvcc-formatted */ > + avio_write(pb, data, size); > + return 0; > + } else if (!(AV_RB24(data) == 1 || AV_RB32(data) == 1)) { > + /* Not a valid Annex B start code prefix */ > + return AVERROR_INVALIDDATA; > + } > + > + ret = ff_avc_parse_nal_units_buf(data, &start, &size); > + if (ret < 0) > + return ret; > + > + vvcc_init(&vvcc); > + > + buf = start; > + end = start + size; > + > + while (end - buf > 4) { > + uint32_t len = FFMIN(AV_RB32(buf), end - buf - 4); > + uint8_t type = (buf[5] >> 3); > + > + buf += 4; > + > + switch (type) { > + case VVC_OPI_NUT: > + case VVC_VPS_NUT: > + case VVC_SPS_NUT: > + case VVC_PPS_NUT: > + case VVC_PREFIX_SEI_NUT: > + case VVC_SUFFIX_SEI_NUT: > + ret = vvcc_add_nal_unit(buf, len, ps_array_completeness, &vvcc); > + if (ret < 0) > + goto end; > + break; > + default: > + break; > + } > + > + buf += len; > + } > + > + ret = vvcc_write(pb, &vvcc); > + > +end: > + vvcc_close(&vvcc); > + av_free(start); > + return ret; > +} > diff --git a/libavformat/vvc.h b/libavformat/vvc.h > new file mode 100644 > index 0000000000..cdf4f6d3f5 > --- /dev/null > +++ b/libavformat/vvc.h > @@ -0,0 +1,99 @@ > +/* > + * VVC helper functions for muxers > + * > + * This file is part of FFmpeg. > + * > + * FFmpeg is free software; you can redistribute it and/or > + * modify it under the terms of the GNU Lesser General Public > + * License as published by the Free Software Foundation; either > + * version 2.1 of the License, or (at your option) any later version. > + * > + * FFmpeg is distributed in the hope that it will be useful, > + * but WITHOUT ANY WARRANTY; without even the implied warranty of > + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU > + * Lesser General Public License for more details. > + * > + * You should have received a copy of the GNU Lesser General Public > + * License along with FFmpeg; if not, write to the Free Software > + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA > + */ > + > +/** > + * @file > + * internal header for VVC (de)muxer utilities > + */ > + > +#ifndef AVFORMAT_VVC_H > +#define AVFORMAT_VVC_H > + > +#include <stdint.h> > +#include "avio.h" > + > +/** > + * Writes Annex B formatted VVC NAL units to the provided AVIOContext. > + * > + * The NAL units are converted to an MP4-compatible format (start code prefixes > + * are replaced by 4-byte size fields, as per ISO/IEC 14496-15). > + * > + * If filter_ps is non-zero, any VVC parameter sets found in the input will be > + * discarded, and *ps_count will be set to the number of discarded PS NAL units. > + * > + * @param pb address of the AVIOContext where the data shall be written > + * @param buf_in address of the buffer holding the input data > + * @param size size (in bytes) of the input buffer > + * @param filter_ps whether to write parameter set NAL units to the output (0) > + * or to discard them (non-zero) > + * @param ps_count address of the variable where the number of discarded > + * parameter set NAL units shall be written, may be NULL > + * @return the amount (in bytes) of data written in case of success, a negative > + * value corresponding to an AVERROR code in case of failure > + */ > +int ff_vvc_annexb2mp4(AVIOContext *pb, const uint8_t *buf_in, > + int size, int filter_ps, int *ps_count); > + > +/** > + * Writes Annex B formatted VVC NAL units to a data buffer. > + * > + * The NAL units are converted to an MP4-compatible format (start code prefixes > + * are replaced by 4-byte size fields, as per ISO/IEC 14496-15). > + * > + * If filter_ps is non-zero, any VVC parameter sets found in the input will be > + * discarded, and *ps_count will be set to the number of discarded PS NAL units. > + * > + * On success, *size holds the size (in bytes) of the output data buffer. > + * > + * @param buf_in address of the buffer holding the input data > + * @param size address of the variable holding the size (in bytes) of the input > + * buffer (on input) and of the output buffer (on success) > + * @param buf_out on success, address of the variable holding the address of > + * the output buffer > + * @param filter_ps whether to write parameter set NAL units to the output (0) > + * or to discard them (non-zero) > + * @param ps_count address of the variable where the number of discarded > + * parameter set NAL units shall be written, may be NULL > + * @return 0 in case of success, a negative value corresponding to an AVERROR > + * code in case of failure > + * @note *buf_out will be treated as uninitialized on input and won't be freed. > + */ > +int ff_vvc_annexb2mp4_buf(const uint8_t *buf_in, uint8_t **buf_out, > + int *size, int filter_ps, int *ps_count); > + > +/** > + * Writes VVC extradata (parameter sets, declarative SEI NAL units) to the > + * provided AVIOContext. > + * > + * If the extradata is Annex B format, it gets converted to vvcC format before > + * writing. > + * > + * @param pb address of the AVIOContext where the vvcC shall be written > + * @param data address of the buffer holding the data needed to write the vvcC > + * @param size size (in bytes) of the data buffer > + * @param ps_array_completeness whether all parameter sets are in the vvcC (1) > + * or there may be additional parameter sets in the bitstream (0) > + * @return >=0 in case of success, a negative value corresponding to an AVERROR > + * code in case of failure > + */ > +int ff_isom_write_vvcc(AVIOContext *pb, const uint8_t *data, > + int size, int ps_array_completeness); > + > +#endif /* AVFORMAT_VVC_H */ > diff --git a/libavformat/vvcdec.c b/libavformat/vvcdec.c > new file mode 100644 > index 0000000000..304f900966 > --- /dev/null > +++ b/libavformat/vvcdec.c > @@ -0,0 +1,61 @@ > +/* > + * RAW VVC video demuxer > + * Copyright (c) 2020 Nuo Mi <nuomi2021@gmail.com> > + * > + * This file is part of FFmpeg. > + * > + * FFmpeg is free software; you can redistribute it and/or > + * modify it under the terms of the GNU Lesser General Public > + * License as published by the Free Software Foundation; either > + * version 2.1 of the License, or (at your option) any later version. > + * > + * FFmpeg is distributed in the hope that it will be useful, > + * but WITHOUT ANY WARRANTY; without even the implied warranty of > + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU > + * Lesser General Public License for more details. > + * > + * You should have received a copy of the GNU Lesser General Public > + * License along with FFmpeg; if not, write to the Free Software > + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA > + */ > + > +#include "libavcodec/vvc.h" > + > +#include "avformat.h" > +#include "rawdec.h" > + > +static int vvc_probe(const AVProbeData *p) > +{ > + uint32_t code = -1; > + int sps = 0, pps = 0, irap = 0; > + int i; > + > + for (i = 0; i < p->buf_size - 1; i++) { > + code = (code << 8) + p->buf[i]; > + if ((code & 0xffffff00) == 0x100) { > + uint8_t nal2 = p->buf[i + 1]; > + int type = (nal2 & 0xF8) >> 3; > + > + if (code & 0x80) // forbidden_zero_bit > + return 0; > + > + if ((nal2 & 0x7) == 0) // nuh_temporal_id_plus1 > + return 0; > + > + switch (type) { > + case VVC_SPS_NUT: sps++; break; > + case VVC_PPS_NUT: pps++; break; > + case VVC_IDR_N_LP: > + case VVC_IDR_W_RADL: > + case VVC_CRA_NUT: > + case VVC_GDR_NUT: irap++; break; > + } > + } > + } > + > + if (sps && pps && irap) > + return AVPROBE_SCORE_EXTENSION + 1; // 1 more than .mpg > + return 0; > +} > + > +FF_DEF_RAWVIDEO_DEMUXER(vvc, "raw VVC video", vvc_probe, "h266,266,vvc", AV_CODEC_ID_VVC) > -- > 2.25.1 > > _______________________________________________ > ffmpeg-devel mailing list > ffmpeg-devel@ffmpeg.org > https://ffmpeg.org/mailman/listinfo/ffmpeg-devel > > To unsubscribe, visit link above, or email > ffmpeg-devel-request@ffmpeg.org with subject "unsubscribe". -- ======================================= Jun zhao/赵军 +++++++++++++++++++++++++++++++++++++++ _______________________________________________ ffmpeg-devel mailing list ffmpeg-devel@ffmpeg.org https://ffmpeg.org/mailman/listinfo/ffmpeg-devel To unsubscribe, visit link above, or email ffmpeg-devel-request@ffmpeg.org with subject "unsubscribe".
next prev parent reply other threads:[~2022-11-07 2:28 UTC|newest] Thread overview: 15+ messages / expand[flat|nested] mbox.gz Atom feed top 2022-11-03 12:21 [FFmpeg-devel] [PATCH v3 00/10] Add " Thomas Siedel 2022-11-03 12:21 ` [FFmpeg-devel] [PATCH v3 01/10] avcodec: add enum types " Thomas Siedel 2022-11-03 12:21 ` [FFmpeg-devel] [PATCH v3 02/10] avcodec: add cbs " Thomas Siedel 2022-11-03 15:57 ` Lynne 2022-11-03 16:02 ` Andreas Rheinhardt 2022-11-03 12:21 ` [FFmpeg-devel] [PATCH v3 03/10] avcodec: add bitstream parser " Thomas Siedel 2022-11-03 12:21 ` [FFmpeg-devel] [PATCH v3 04/10] avcodec: add MP4 to annexb support " Thomas Siedel 2022-11-03 12:21 ` [FFmpeg-devel] [PATCH v3 05/10] avformat: add demuxer and probe " Thomas Siedel 2022-11-07 2:28 ` mypopy [this message] 2022-11-15 10:43 ` Thomas Siedel 2022-11-03 12:21 ` [FFmpeg-devel] [PATCH v3 06/10] avformat: add muxer " Thomas Siedel 2022-11-03 12:21 ` [FFmpeg-devel] [PATCH v3 07/10] avcodec: add external decoder libvvdec " Thomas Siedel 2022-11-03 12:21 ` [FFmpeg-devel] [PATCH v3 08/10] avcodec: add external encoder libvvenc " Thomas Siedel 2022-11-03 12:21 ` [FFmpeg-devel] [PATCH v3 09/10] avformat: add ts stream types " Thomas Siedel 2022-11-03 12:21 ` [FFmpeg-devel] [PATCH v3 10/10] avcodec: increase minor version " Thomas Siedel
Reply instructions: You may reply publicly to this message via plain-text email using any one of the following methods: * Save the following mbox file, import it into your mail client, and reply-to-all from there: mbox Avoid top-posting and favor interleaved quoting: https://en.wikipedia.org/wiki/Posting_style#Interleaved_style * Reply using the --to, --cc, and --in-reply-to switches of git-send-email(1): git send-email \ --in-reply-to='CACYjbn17ohTJjoY8_DY8XpHmVESrgGKC4+usUfo15=-6KDkLfQ@mail.gmail.com' \ --to=mypopy@gmail.com \ --cc=ffmpeg-devel@ffmpeg.org \ --cc=thomas.ff@spin-digital.com \ /path/to/YOUR_REPLY https://kernel.org/pub/software/scm/git/docs/git-send-email.html * If your mail client supports setting the In-Reply-To header via mailto: links, try the mailto: link
Git Inbox Mirror of the ffmpeg-devel mailing list - see https://ffmpeg.org/mailman/listinfo/ffmpeg-devel This inbox may be cloned and mirrored by anyone: git clone --mirror https://master.gitmailbox.com/ffmpegdev/0 ffmpegdev/git/0.git # If you have public-inbox 1.1+ installed, you may # initialize and index your mirror using the following commands: public-inbox-init -V2 ffmpegdev ffmpegdev/ https://master.gitmailbox.com/ffmpegdev \ ffmpegdev@gitmailbox.com public-inbox-index ffmpegdev Example config snippet for mirrors. AGPL code for this site: git clone https://public-inbox.org/public-inbox.git