mirror of
https://git.ffmpeg.org/ffmpeg.git
synced 2024-12-13 10:44:59 +00:00
0e4dfa4709
This reverts commit 41b73ae883
.
This patch was pushed extremely quickly, without giving developers the
time to object.
151 lines
5.1 KiB
C
151 lines
5.1 KiB
C
/*
|
|
* 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 "libavutil/common.h"
|
|
#include "libavutil/opt.h"
|
|
|
|
#include "bsf.h"
|
|
#include "bsf_internal.h"
|
|
#include "cbs.h"
|
|
#include "cbs_bsf.h"
|
|
#include "cbs_h266.h"
|
|
#include "vvc.h"
|
|
|
|
#define IS_H266_SLICE(nut) (nut <= VVC_RASL_NUT || (nut >= VVC_IDR_W_RADL && nut <= VVC_GDR_NUT))
|
|
|
|
typedef struct H266MetadataContext {
|
|
CBSBSFContext common;
|
|
|
|
H266RawAUD aud_nal;
|
|
|
|
int aud;
|
|
} H266MetadataContext;
|
|
|
|
static int h266_metadata_update_fragment(AVBSFContext *bsf, AVPacket *pkt,
|
|
CodedBitstreamFragment *pu)
|
|
{
|
|
H266MetadataContext *ctx = bsf->priv_data;
|
|
int err, i;
|
|
|
|
// If an AUD is present, it must be the first NAL unit.
|
|
if (pu->nb_units && pu->units[0].type == VVC_AUD_NUT) {
|
|
if (ctx->aud == BSF_ELEMENT_REMOVE)
|
|
ff_cbs_delete_unit(pu, 0);
|
|
} else if ( pkt && ctx->aud == BSF_ELEMENT_INSERT) {
|
|
const H266RawSlice *first_slice = NULL;
|
|
const H266RawPictureHeader *ph = NULL;
|
|
H266RawAUD *aud = &ctx->aud_nal;
|
|
int pic_type = 0, temporal_id = 8, layer_id = 0;
|
|
for (i = 0; i < pu->nb_units; i++) {
|
|
const H266RawNALUnitHeader *nal = pu->units[i].content;
|
|
if (!nal)
|
|
continue;
|
|
if (nal->nuh_temporal_id_plus1 < temporal_id + 1)
|
|
temporal_id = nal->nuh_temporal_id_plus1 - 1;
|
|
if ( nal->nal_unit_type == VVC_PH_NUT ) {
|
|
const H266RawPH *header = pu->units[i].content;
|
|
ph = &header->ph_picture_header;
|
|
} else if (IS_H266_SLICE(nal->nal_unit_type)) {
|
|
const H266RawSlice *slice = pu->units[i].content;
|
|
layer_id = nal->nuh_layer_id;
|
|
if (slice->header.sh_slice_type == VVC_SLICE_TYPE_B &&
|
|
pic_type < 2)
|
|
pic_type = 2;
|
|
if (slice->header.sh_slice_type == VVC_SLICE_TYPE_P &&
|
|
pic_type < 1)
|
|
pic_type = 1;
|
|
if (!first_slice) {
|
|
first_slice = slice;
|
|
if (first_slice->header.
|
|
sh_picture_header_in_slice_header_flag)
|
|
ph = &first_slice->header.sh_picture_header;
|
|
else if (!ph)
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
if (!ph) {
|
|
av_log(bsf, AV_LOG_ERROR, "no avaliable picture header");
|
|
return AVERROR_INVALIDDATA;
|
|
}
|
|
|
|
aud->nal_unit_header = (H266RawNALUnitHeader) {
|
|
.nal_unit_type = VVC_AUD_NUT,
|
|
.nuh_layer_id = layer_id,
|
|
.nuh_temporal_id_plus1 = temporal_id + 1,
|
|
};
|
|
aud->aud_pic_type = pic_type;
|
|
aud->aud_irap_or_gdr_flag = ph->ph_gdr_or_irap_pic_flag;
|
|
|
|
err = ff_cbs_insert_unit_content(pu, 0, VVC_AUD_NUT, aud, NULL);
|
|
if (err < 0) {
|
|
av_log(bsf, AV_LOG_ERROR, "Failed to insert AUD.\n");
|
|
return err;
|
|
}
|
|
}
|
|
|
|
/* TODO: implement more metadata parsing, like VUI, Levels etc. */
|
|
//for (i = 0; i < pu->nb_units; i++) {
|
|
// if (pu->units[i].type == VVC_SPS_NUT) {
|
|
// }
|
|
//}
|
|
return 0;
|
|
}
|
|
|
|
static const CBSBSFType h266_metadata_type = {
|
|
.codec_id = AV_CODEC_ID_VVC,
|
|
.fragment_name = "access unit",
|
|
.unit_name = "NAL unit",
|
|
.update_fragment = &h266_metadata_update_fragment,
|
|
};
|
|
|
|
static int h266_metadata_init(AVBSFContext *bsf)
|
|
{
|
|
return ff_cbs_bsf_generic_init(bsf, &h266_metadata_type);
|
|
}
|
|
|
|
#define OFFSET(x) offsetof(H266MetadataContext, x)
|
|
#define FLAGS (AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_BSF_PARAM)
|
|
static const AVOption h266_metadata_options[] = {
|
|
BSF_ELEMENT_OPTIONS_PIR("aud", "Access Unit Delimiter NAL units",
|
|
aud, FLAGS),
|
|
|
|
{ NULL }
|
|
};
|
|
|
|
static const AVClass h266_metadata_class = {
|
|
.class_name = "h266_metadata_bsf",
|
|
.item_name = av_default_item_name,
|
|
.option = h266_metadata_options,
|
|
.version = LIBAVUTIL_VERSION_INT,
|
|
};
|
|
|
|
static const enum AVCodecID h266_metadata_codec_ids[] = {
|
|
AV_CODEC_ID_VVC, AV_CODEC_ID_NONE,
|
|
};
|
|
|
|
const FFBitStreamFilter ff_vvc_metadata_bsf = {
|
|
.p.name = "vvc_metadata",
|
|
.p.codec_ids = h266_metadata_codec_ids,
|
|
.p.priv_class = &h266_metadata_class,
|
|
.priv_data_size = sizeof(H266MetadataContext),
|
|
.init = &h266_metadata_init,
|
|
.close = &ff_cbs_bsf_generic_close,
|
|
.filter = &ff_cbs_bsf_generic_filter,
|
|
};
|