mirror of https://git.ffmpeg.org/ffmpeg.git
avcodec/h265_metadata: filter parameter sets in packet side data
Signed-off-by: James Almer <jamrial@gmail.com>
This commit is contained in:
parent
502dacdc50
commit
1a9684c08a
|
@ -336,6 +336,57 @@ static int h265_metadata_update_sps(AVBSFContext *bsf,
|
|||
return 0;
|
||||
}
|
||||
|
||||
static int h265_metadata_update_side_data(AVBSFContext *bsf, AVPacket *pkt)
|
||||
{
|
||||
H265MetadataContext *ctx = bsf->priv_data;
|
||||
CodedBitstreamFragment *au = &ctx->access_unit;
|
||||
uint8_t *side_data;
|
||||
int side_data_size;
|
||||
int err, i;
|
||||
|
||||
side_data = av_packet_get_side_data(pkt, AV_PKT_DATA_NEW_EXTRADATA,
|
||||
&side_data_size);
|
||||
if (!side_data_size)
|
||||
return 0;
|
||||
|
||||
err = ff_cbs_read(ctx->cbc, au, side_data, side_data_size);
|
||||
if (err < 0) {
|
||||
av_log(bsf, AV_LOG_ERROR, "Failed to read extradata from packet side data.\n");
|
||||
return err;
|
||||
}
|
||||
|
||||
if (ctx->level == LEVEL_AUTO && !ctx->level_guess)
|
||||
h265_metadata_guess_level(bsf, au);
|
||||
|
||||
for (i = 0; i < au->nb_units; i++) {
|
||||
if (au->units[i].type == HEVC_NAL_VPS) {
|
||||
err = h265_metadata_update_vps(bsf, au->units[i].content);
|
||||
if (err < 0)
|
||||
return err;
|
||||
}
|
||||
if (au->units[i].type == HEVC_NAL_SPS) {
|
||||
err = h265_metadata_update_sps(bsf, au->units[i].content);
|
||||
if (err < 0)
|
||||
return err;
|
||||
}
|
||||
}
|
||||
|
||||
err = ff_cbs_write_fragment_data(ctx->cbc, au);
|
||||
if (err < 0) {
|
||||
av_log(bsf, AV_LOG_ERROR, "Failed to write extradata into packet side data.\n");
|
||||
return err;
|
||||
}
|
||||
|
||||
side_data = av_packet_new_side_data(pkt, AV_PKT_DATA_NEW_EXTRADATA, au->data_size);
|
||||
if (!side_data)
|
||||
return AVERROR(ENOMEM);
|
||||
memcpy(side_data, au->data, au->data_size);
|
||||
|
||||
ff_cbs_fragment_reset(ctx->cbc, au);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int h265_metadata_filter(AVBSFContext *bsf, AVPacket *pkt)
|
||||
{
|
||||
H265MetadataContext *ctx = bsf->priv_data;
|
||||
|
@ -346,6 +397,10 @@ static int h265_metadata_filter(AVBSFContext *bsf, AVPacket *pkt)
|
|||
if (err < 0)
|
||||
return err;
|
||||
|
||||
err = h265_metadata_update_side_data(bsf, pkt);
|
||||
if (err < 0)
|
||||
goto fail;
|
||||
|
||||
err = ff_cbs_read_packet(ctx->cbc, au, pkt);
|
||||
if (err < 0) {
|
||||
av_log(bsf, AV_LOG_ERROR, "Failed to read packet.\n");
|
||||
|
|
Loading…
Reference in New Issue