From patchwork Mon Aug 19 02:37:57 2019 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Andriy Gelman X-Patchwork-Id: 14579 Return-Path: X-Original-To: patchwork@ffaux-bg.ffmpeg.org Delivered-To: patchwork@ffaux-bg.ffmpeg.org Received: from ffbox0-bg.mplayerhq.hu (ffbox0-bg.ffmpeg.org [79.124.17.100]) by ffaux.localdomain (Postfix) with ESMTP id 020AC44887E for ; Mon, 19 Aug 2019 05:45:02 +0300 (EEST) Received: from [127.0.1.1] (localhost [127.0.0.1]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTP id C912F68A9BB; Mon, 19 Aug 2019 05:45:01 +0300 (EEST) X-Original-To: ffmpeg-devel@ffmpeg.org Delivered-To: ffmpeg-devel@ffmpeg.org Received: from mail-qk1-f195.google.com (mail-qk1-f195.google.com [209.85.222.195]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTPS id EA4A5689D13 for ; Mon, 19 Aug 2019 05:44:55 +0300 (EEST) Received: by mail-qk1-f195.google.com with SMTP id m2so269196qki.12 for ; Sun, 18 Aug 2019 19:44:55 -0700 (PDT) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=gmail.com; s=20161025; h=from:to:cc:subject:date:message-id:in-reply-to:references :mime-version:content-transfer-encoding; bh=kzbuu85Nn1ZXujnbJ3kMSPsdjeayDDsa6qFoUocSCMA=; b=XK6yNS1kyYme5vGxHA0PQaq6EEsFTyBoL+MMOaoio/xd95HVcDDnp92oHCJ6EHtz57 boEEoDSgmgpiMYKwKsxPgvUXDYoStMQKfZEh7tFYZZ/QlkyGPkK2cP+vyNKA1QZPNz5i GaV42X5F34Ukxwt54hp7vJFVfMCwmBoErwI+c4rL7F6o+WAEHHdG6s3kqeLTJtVDT5bE bgxXQLf95+NjK9OukK/azJu6ca4P75ya/KCzbEe9QMI3oiWull+sbH0GR90JlPFlNi0Y B6nA/w8bFepTXjufBK6bEv8naQvBQ7M3c78b5avdnsRBWIze+aN7rON8KZOYZ2wGdqQG hriA== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20161025; h=x-gm-message-state:from:to:cc:subject:date:message-id:in-reply-to :references:mime-version:content-transfer-encoding; bh=kzbuu85Nn1ZXujnbJ3kMSPsdjeayDDsa6qFoUocSCMA=; b=G/EpoQOOngnkAHl/RgSnFq0bFicrJWemCZPnO24LFSXZwjx/2aqCorFYBbCHkCfV82 QsJQjZq+4t17kFnGodRdljrMFgp3rJua/eLzIWADNWfyOk6nPufQbWI4PQSXzeUxOV2Y m/dR8jDKDBkUqbtjK8gzg2TK7sJ5esTGQTU1WvmSfcgorIuv+UMwgy4eGNF6gcOB7Kip gkKukSQDJwWeqvXmCKmDg5gNMCFNgTIL2nf3/96V6WDN/Wjx4dmNCJx3so3V0iPMN5Ld UUXk1pcjarORwGJapMy7PqNmx3F+rgcAAyMQucvkAvyuKS2F7DzW6+AAwsAFLi8EtMET GKkQ== X-Gm-Message-State: APjAAAW1CBT40zI9T1dQflGUcyUpM9Kjb8+dQvUjZSi3nWqhI3SahcSQ LeTiXFAxDtWX/K5e7WCElGRgx/p/ X-Google-Smtp-Source: APXvYqwelilL3whAnjqgFQiRtGDpBDuID9yu2wL13ilgM+ukdR9P25HwHctCxEiFwiF5cVDWMu8OzA== X-Received: by 2002:a37:395:: with SMTP id 143mr19361854qkd.317.1566182326292; Sun, 18 Aug 2019 19:38:46 -0700 (PDT) Received: from localhost.localdomain (c-24-60-253-205.hsd1.ma.comcast.net. [24.60.253.205]) by smtp.gmail.com with ESMTPSA id u187sm6645599qkh.110.2019.08.18.19.38.45 (version=TLS1_3 cipher=TLS_AES_256_GCM_SHA384 bits=256/256); Sun, 18 Aug 2019 19:38:45 -0700 (PDT) From: Andriy Gelman X-Google-Original-From: Andriy Gelman To: ffmpeg-devel@ffmpeg.org Date: Sun, 18 Aug 2019 22:37:57 -0400 Message-Id: <20190819023757.20900-1-andriy.gelman@gmail.com> X-Mailer: git-send-email 2.22.0 In-Reply-To: <2fa17704-7cee-4d29-5c02-2c4f8a805fc0@gmail.com> References: <2fa17704-7cee-4d29-5c02-2c4f8a805fc0@gmail.com> MIME-Version: 1.0 Subject: [FFmpeg-devel] [PATCH v2] hevc_mp4toannexb: Insert correct parameter sets before IRAP X-BeenThere: ffmpeg-devel@ffmpeg.org X-Mailman-Version: 2.1.20 Precedence: list List-Id: FFmpeg development discussions and patches List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , Reply-To: FFmpeg development discussions and patches Cc: Andriy Gelman Errors-To: ffmpeg-devel-bounces@ffmpeg.org Sender: "ffmpeg-devel" From: Andriy Gelman Fixes #7799 Currently, the mp4toannexb filter always inserts the same extradata at the start of the first IRAP unit. As in ticket #7799, this can lead to decoding errors if modified parameter sets are signalled in-band. This commit keeps track of the vps/sps/pps parameter sets during the conversion. The correct combination is inserted at the start of the first IRAP unit instead of the original extradata. SEI prefix nal units are also cached and inserted after the parameter sets. This commit also makes an update to the hevc-bsf-mp4toannexb fate test since the result before this patch contained duplicate vps/sps/pps parameter sets in-band. --- libavcodec/hevc_mp4toannexb_bsf.c | 475 +++++++++++++++++++++++++++--- tests/fate/hevc.mak | 2 +- 2 files changed, 437 insertions(+), 40 deletions(-) diff --git a/libavcodec/hevc_mp4toannexb_bsf.c b/libavcodec/hevc_mp4toannexb_bsf.c index 09bce5b34c..7f3d68252d 100644 --- a/libavcodec/hevc_mp4toannexb_bsf.c +++ b/libavcodec/hevc_mp4toannexb_bsf.c @@ -28,19 +28,212 @@ #include "bsf.h" #include "bytestream.h" #include "hevc.h" - -#define MIN_HEVCC_LENGTH 23 +#include "h2645_parse.h" +#include "hevc_ps.h" +#include "golomb.h" + +#define MIN_HEVCC_LENGTH 23 +#define PROFILE_WITHOUT_IDC_BITS 88 +#define IS_IRAP(s) ((s)->type >= 16 && (s)->type <= 23) +#define IS_PARAMSET(s) ((s)->type >= 32 && (s)->type <= 34) + +#define WRITE_NAL(pkt, prev_size, in_buffer) do { \ + AV_WB32((pkt)->data + (prev_size), 1); \ + prev_size += 4; \ + memcpy((pkt)->data + (prev_size), (in_buffer)->data, (in_buffer)->size); \ + prev_size += (in_buffer)->size; \ +} while (0) + +typedef struct Param { + AVBufferRef *raw_data; /*store a copy of the raw data to construct extradata*/ + int ref; /*stores the ref of the higher level parameter set*/ +} Param; + +/*modified version of HEVCParamSets to store bytestream and reference to previous level*/ +typedef struct ParamSets { + Param vps_list[HEVC_MAX_VPS_COUNT]; + Param sps_list[HEVC_MAX_SPS_COUNT]; + Param pps_list[HEVC_MAX_PPS_COUNT]; + + AVBufferRef *sei_prefix; +} ParamSets; typedef struct HEVCBSFContext { - uint8_t length_size; - int extradata_parsed; + uint8_t length_size; + int extradata_parsed; + ParamSets ps; /*make own of version of HEVCParamSets store copy of the bytestream*/ } HEVCBSFContext; + +static int parse_vps(AVBSFContext* ctx, H2645NAL *nal) +{ + int vps_id = 0; + Param *param_ptr; + + HEVCBSFContext *s = ctx->priv_data; + ParamSets *ps = &s->ps; + + GetBitContext *gb = &nal->gb; + int nal_size = nal->raw_size; + + vps_id = get_bits(gb, 4); + if (vps_id >= HEVC_MAX_VPS_COUNT) { + av_log(ctx, AV_LOG_ERROR, "VPS id out of range: %d\n", vps_id); + return AVERROR_INVALIDDATA; + } + + param_ptr = &ps->vps_list[vps_id]; + /*init raw_data if needed*/ + if (!param_ptr->raw_data) { + param_ptr->raw_data = av_buffer_allocz(nal_size); + param_ptr->ref = 0; + if (!param_ptr->raw_data) + return AVERROR(ENOMEM); + } + + if (param_ptr->raw_data && param_ptr->raw_data->size == nal_size && + !memcmp(param_ptr->raw_data->data, nal->raw_data, nal_size)) { + av_log(ctx, AV_LOG_DEBUG, "Parsed VPS id: %d. Copy already exists in parameter set\n", vps_id); + } else { + AVBufferRef *vps_buf = av_buffer_allocz(nal_size); + if (!vps_buf) + return AVERROR(ENOMEM); + + /*copy raw data into vps_buf buffer and replace existing copy in ps*/ + memcpy(vps_buf->data, nal->raw_data, nal_size); + av_buffer_unref(¶m_ptr->raw_data); + param_ptr->raw_data = vps_buf; + } + return 0; +} + +static int parse_sps(AVBSFContext *ctx, H2645NAL *nal) +{ + int sps_id, vps_ref, max_sub_layers_minus1; + int i; + Param *param_ptr; + + HEVCBSFContext *s = (HEVCBSFContext*)ctx->priv_data; + ParamSets *ps = &s->ps; + + GetBitContext *gb = &nal->gb; + int nal_size = nal->raw_size; + + uint8_t sub_layer_profile_present_flag[HEVC_MAX_SUB_LAYERS]; + uint8_t sub_layer_level_present_flag[HEVC_MAX_SUB_LAYERS]; + + vps_ref = get_bits(gb, 4); + if (vps_ref >= HEVC_MAX_VPS_COUNT) { + av_log(ctx, AV_LOG_ERROR, "VPS id out of range: %d\n", vps_ref); + return AVERROR_INVALIDDATA; + } + + max_sub_layers_minus1 = get_bits(gb, 3); + skip_bits1(gb); /*sps_temporal_id_nesting_flag*/ + skip_bits(gb, PROFILE_WITHOUT_IDC_BITS); /*profile_tier_level*/ + skip_bits(gb, 8); /*general_level_idc*/ + + for (i = 0; i < max_sub_layers_minus1; ++i) { + sub_layer_profile_present_flag[i] = get_bits1(gb); + sub_layer_level_present_flag[i] = get_bits1(gb); + } + + if (max_sub_layers_minus1 > 0) + for (i = max_sub_layers_minus1; i < 8; i++) + skip_bits(gb, 2); // reserved_zero_2bits[i] + + for (i = 0; i < max_sub_layers_minus1; ++i) { + if (sub_layer_profile_present_flag[i]) + skip_bits(gb, PROFILE_WITHOUT_IDC_BITS); /*profile_tier_level*/ + if (sub_layer_level_present_flag[i]) + skip_bits(gb, 8); /*sub_layer_level_idc*/ + } + + /*we only need the sps_id index*/ + sps_id = get_ue_golomb_long(gb); + if (sps_id >= HEVC_MAX_SPS_COUNT) { + av_log(ctx, AV_LOG_ERROR, "SPS id out of range: %d\n", sps_id); + return AVERROR_INVALIDDATA; + } + + param_ptr = &ps->sps_list[sps_id]; + if (!param_ptr->raw_data) { + param_ptr->raw_data = av_buffer_allocz(nal_size); + param_ptr->ref = 0; + if (!param_ptr->raw_data) + return AVERROR(ENOMEM); + } + + if (param_ptr->raw_data && param_ptr->raw_data->size == nal_size && + !memcmp(param_ptr->raw_data->data, nal->raw_data, nal_size)) { + av_log(ctx, AV_LOG_DEBUG, "Parsed SPS id: %d. Copy already exists in parameter set\n", sps_id); + } else { + AVBufferRef *sps_buf = av_buffer_allocz(nal_size); + if (!sps_buf) + return AVERROR(ENOMEM); + + /*copy raw data into vps_buf buffer and replace existing copy in ps*/ + memcpy(sps_buf->data, nal->raw_data, nal_size); + av_buffer_unref(¶m_ptr->raw_data); + param_ptr->raw_data = sps_buf; + param_ptr->ref = vps_ref; + } + return 0; +} + +static int parse_pps(AVBSFContext *ctx, H2645NAL *nal) +{ + int pps_id, sps_ref; + Param *param_ptr; + + HEVCBSFContext *s = ctx->priv_data; + ParamSets *ps = &s->ps; + + GetBitContext *gb = &nal->gb; + int nal_size = nal->raw_size; + + pps_id = get_ue_golomb_long(gb); + if (pps_id >= HEVC_MAX_PPS_COUNT) { + av_log(ctx, AV_LOG_ERROR, "PPS id out of range: %d\n", pps_id); + return AVERROR_INVALIDDATA; + } + + sps_ref = get_ue_golomb_long(gb); + if (sps_ref >= HEVC_MAX_SPS_COUNT) { + av_log(ctx, AV_LOG_ERROR, "SPS id out of range: %d\n", sps_ref); + return AVERROR_INVALIDDATA; + } + + param_ptr = &ps->pps_list[pps_id]; + if (!param_ptr->raw_data) { + param_ptr->raw_data = av_buffer_allocz(nal_size); + param_ptr->ref = 0; + if (!param_ptr->raw_data) + return AVERROR(ENOMEM); + } + + if (param_ptr->raw_data && param_ptr->raw_data->size == nal_size && + !memcmp(param_ptr->raw_data->data, nal->raw_data, nal_size)) { + av_log(ctx, AV_LOG_DEBUG, "Parsed PPS id: %d. Copy already exists in parameter set\n", pps_id); + } else { + AVBufferRef *pps_buf = av_buffer_allocz(nal_size); + if (!pps_buf) + return AVERROR(ENOMEM); + + /*copy raw data into vps_buf buffer and replace existing copy in ps*/ + memcpy(pps_buf->data, nal->raw_data, nal_size); + av_buffer_unref(¶m_ptr->raw_data); + param_ptr->raw_data = pps_buf; + param_ptr->ref = sps_ref; + } + return 0; +} + static int hevc_extradata_to_annexb(AVBSFContext *ctx) { - GetByteContext gb; int length_size, num_arrays, i, j; int ret = 0; + GetByteContext gb; uint8_t *new_extradata = NULL; size_t new_extradata_size = 0; @@ -94,10 +287,95 @@ fail: return ret; } +static int avcc_to_annexb(AVBSFContext *ctx, AVPacket **pkt, int nalu_length_size) +{ + int ret, i, prev_size; + GetByteContext gb; + + AVPacket* pkt_out = av_packet_alloc(); + if (!pkt_out) + return AVERROR(ENOMEM); + + ret = av_new_packet(pkt_out, 0); + if (ret < 0) + goto fail; + + bytestream2_init(&gb, (*pkt)->data, (*pkt)->size); + while (bytestream2_get_bytes_left(&gb)) { + uint32_t nalu_size = 0; + for (i = 0; i < nalu_length_size; i++) + nalu_size = (nalu_size << 8) | bytestream2_get_byte(&gb); + + if (SIZE_MAX - nalu_size < 4) { + ret = AVERROR_INVALIDDATA; + goto fail; + } + + prev_size = pkt_out->size; + ret = av_grow_packet(pkt_out, 4 + nalu_size); + if (ret < 0) + goto fail; + + AV_WB32(pkt_out->data + prev_size, 1); + ret = bytestream2_get_buffer(&gb, pkt_out->data + prev_size + 4, nalu_size); + if (ret != nalu_size) + av_log(ctx, AV_LOG_WARNING, "Corrupted avcc nal unit. Read %d/%d bytes\n", ret, nalu_size); + } + + ret = av_packet_copy_props(pkt_out, *pkt); + if (ret < 0) + goto fail; + + /*unref avcc version and replace pkt with pkt_out*/ + av_packet_free(pkt); + *pkt = pkt_out; + + return 0; + +fail: + av_packet_free(&pkt_out); + return ret; +} + +static int update_sei(AVBufferRef **sei, H2645NAL *nal) +{ + AVBufferRef *tmp; + + av_buffer_unref(sei); + tmp = av_buffer_alloc(nal->raw_size); + if (!tmp) + return AVERROR(ENOMEM); + + memcpy(tmp->data, nal->raw_data, nal->raw_size); + *sei = tmp; + + return 0; +} + +static int update_paramset(AVBSFContext *ctx, H2645NAL *nal) +{ + int ret; + switch (nal->type) { + case (HEVC_NAL_VPS): + if (ret = parse_vps(ctx, nal) < 0) + return ret; + break; + case (HEVC_NAL_SPS): + if (ret = parse_sps(ctx, nal) < 0) + return ret; + break; + case (HEVC_NAL_PPS): + if (ret = parse_pps(ctx, nal) < 0) + return ret; + } + return 0; +} + static int hevc_mp4toannexb_init(AVBSFContext *ctx) { HEVCBSFContext *s = ctx->priv_data; - int ret; + H2645Packet pkt; + int i, ret = 0; if (ctx->par_in->extradata_size < MIN_HEVCC_LENGTH || AV_RB24(ctx->par_in->extradata) == 1 || @@ -110,7 +388,114 @@ static int hevc_mp4toannexb_init(AVBSFContext *ctx) return ret; s->length_size = ret; s->extradata_parsed = 1; + + memset(&pkt, 0, sizeof(H2645Packet)); + ret = ff_h2645_packet_split(&pkt, ctx->par_out->extradata, ctx->par_out->extradata_size, + ctx, 0, 0, AV_CODEC_ID_HEVC, 1, 0); + if (ret < 0) + goto done; + + for (i = 0; i < pkt.nb_nals; ++i) { + H2645NAL *nal = &pkt.nals[i]; + + /*current segmentation algorithm includes next 0x00 from next nal unit*/ + if (nal->raw_data[nal->raw_size - 1] == 0x00) + nal->raw_size--; + + if (IS_PARAMSET(nal)) { + ret = update_paramset(ctx, nal); + if (ret < 0) + goto done; + continue; + } + + if (nal->type == HEVC_NAL_SEI_PREFIX) { + ret = update_sei(&s->ps.sei_prefix, nal); + if (ret < 0) + goto done; + } + } + } + +done: + ff_h2645_packet_uninit(&pkt); + return ret; +} + +static void ps_uninit(ParamSets* ps) +{ + int i; + for (i = 0; i < FF_ARRAY_ELEMS(ps->vps_list); i++) + av_buffer_unref(&ps->vps_list[i].raw_data); + for (i = 0; i < FF_ARRAY_ELEMS(ps->sps_list); i++) + av_buffer_unref(&ps->sps_list[i].raw_data); + for (i = 0; i < FF_ARRAY_ELEMS(ps->pps_list); i++) + av_buffer_unref(&ps->pps_list[i].raw_data); + + av_buffer_unref(&ps->sei_prefix); +} + +static void hevc_mp4toannexb_close(AVBSFContext *ctx) +{ + HEVCBSFContext *s = (HEVCBSFContext*)ctx->priv_data; + ps_uninit(&s->ps); +} + +static int write_extradata(AVBSFContext *ctx, AVPacket *pkt_out, H2645NAL *nal_irap) +{ + int ref, ret, prev_size; + int new_extradata_size = 0; + + HEVCBSFContext *s = (HEVCBSFContext*)ctx->priv_data; + ParamSets *ps = &s->ps; + GetBitContext *gb = &nal_irap->gb; + + /* currently active pps parameter set */ + const Param *vps; + const Param *sps; + const Param *pps; + + skip_bits1(gb); /*first_slice_ic_pic_flag*/ + skip_bits1(gb); /*no_output_of_prior_pics_flag*/ + + ref = get_ue_golomb_long(gb); + if (ref >= HEVC_MAX_PPS_COUNT || ps->pps_list[ref].raw_data == NULL) { + av_log(ctx, AV_LOG_ERROR, "Invalid PPS: %d\n", ref); + return AVERROR_INVALIDDATA; + } + pps = &ps->pps_list[ref]; + new_extradata_size += pps->raw_data->size + 4; + ref = pps->ref; + + if (ref >= HEVC_MAX_SPS_COUNT || ps->sps_list[ref].raw_data == NULL) { + av_log(ctx, AV_LOG_ERROR, "Invalid SPS: %d\n", ref); + return AVERROR_INVALIDDATA; + } + sps = &ps->sps_list[ref]; + new_extradata_size += sps->raw_data->size + 4; + ref = sps->ref; + + if (ref >= HEVC_MAX_VPS_COUNT || ps->vps_list[ref].raw_data == NULL) { + av_log(ctx, AV_LOG_ERROR, "Invalid VPS: %d\n", ref); + return AVERROR_INVALIDDATA; } + vps = &ps->vps_list[ref]; + new_extradata_size += vps->raw_data->size + 4; + + if (ps->sei_prefix) + new_extradata_size += ps->sei_prefix->size + 4; + + prev_size = pkt_out->size; + ret = av_grow_packet(pkt_out, new_extradata_size); + if (ret < 0) + return AVERROR(ENOMEM); + + WRITE_NAL(pkt_out, prev_size, vps->raw_data); + WRITE_NAL(pkt_out, prev_size, sps->raw_data); + WRITE_NAL(pkt_out, prev_size, pps->raw_data); + + if (ps->sei_prefix) + WRITE_NAL(pkt_out, prev_size, ps->sei_prefix); return 0; } @@ -119,62 +504,73 @@ static int hevc_mp4toannexb_filter(AVBSFContext *ctx, AVPacket *out) { HEVCBSFContext *s = ctx->priv_data; AVPacket *in; - GetByteContext gb; - - int got_irap = 0; - int i, ret = 0; + H2645Packet pkt; + int i, prev_size; + int ret = 0; + int first_irap = 0; ret = ff_bsf_get_packet(ctx, &in); if (ret < 0) return ret; + /*output the annexb nalu if extradata is not parsed*/ if (!s->extradata_parsed) { av_packet_move_ref(out, in); av_packet_free(&in); return 0; } - bytestream2_init(&gb, in->data, in->size); + /*convert packet from avcc to annexb*/ + ret = avcc_to_annexb(ctx, &in, s->length_size); + if (ret < 0) { + av_packet_free(&in); + return ret; + } - while (bytestream2_get_bytes_left(&gb)) { - uint32_t nalu_size = 0; - int nalu_type; - int is_irap, add_extradata, extra_size, prev_size; + /*segment annexb packet into nal units*/ + memset(&pkt, 0, sizeof(H2645Packet)); + ret = ff_h2645_packet_split(&pkt, in->data, in->size, ctx, 0, 0, AV_CODEC_ID_HEVC, 1, 0); + if (ret < 0) + goto done; - for (i = 0; i < s->length_size; i++) - nalu_size = (nalu_size << 8) | bytestream2_get_byte(&gb); + for (i = 0; i < pkt.nb_nals; ++i) { - nalu_type = (bytestream2_peek_byte(&gb) >> 1) & 0x3f; + H2645NAL *nal = &pkt.nals[i]; - /* prepend extradata to IRAP frames */ - is_irap = nalu_type >= 16 && nalu_type <= 23; - add_extradata = is_irap && !got_irap; - extra_size = add_extradata * ctx->par_out->extradata_size; - got_irap |= is_irap; + /*current segmentation algorithm includes next 0x00 from next nal unit*/ + if (nal->raw_data[nal->raw_size - 1] == 0x00) + nal->raw_size--; - if (SIZE_MAX - nalu_size < 4 || - SIZE_MAX - 4 - nalu_size < extra_size) { - ret = AVERROR_INVALIDDATA; - goto fail; + if (IS_PARAMSET(nal)) { + ret = update_paramset(ctx, nal); + if (ret < 0) + goto done; + continue; } - prev_size = out->size; + if (nal->type == HEVC_NAL_SEI_PREFIX) { + ret = update_sei(&s->ps.sei_prefix, nal); + if (ret < 0) + goto done; + continue; + } - ret = av_grow_packet(out, 4 + nalu_size + extra_size); - if (ret < 0) - goto fail; + if (!first_irap && IS_IRAP(nal)) { + ret = write_extradata(ctx, out, nal); + if (ret < 0) + goto done; + first_irap = 1; + } - if (add_extradata) - memcpy(out->data + prev_size, ctx->par_out->extradata, extra_size); - AV_WB32(out->data + prev_size + extra_size, 1); - bytestream2_get_buffer(&gb, out->data + prev_size + 4 + extra_size, nalu_size); + /*write to output packet*/ + prev_size = out->size; + av_grow_packet(out, nal->raw_size + 4); + WRITE_NAL(out, prev_size, nal); } - ret = av_packet_copy_props(out, in); - if (ret < 0) - goto fail; +done: + ff_h2645_packet_uninit(&pkt); -fail: if (ret < 0) av_packet_unref(out); av_packet_free(&in); @@ -190,6 +586,7 @@ const AVBitStreamFilter ff_hevc_mp4toannexb_bsf = { .name = "hevc_mp4toannexb", .priv_data_size = sizeof(HEVCBSFContext), .init = hevc_mp4toannexb_init, + .close = hevc_mp4toannexb_close, .filter = hevc_mp4toannexb_filter, .codec_ids = codec_ids, }; diff --git a/tests/fate/hevc.mak b/tests/fate/hevc.mak index 559c3898bc..46f130f4d1 100644 --- a/tests/fate/hevc.mak +++ b/tests/fate/hevc.mak @@ -238,7 +238,7 @@ FATE_HEVC-$(call ALLYES, HEVC_DEMUXER MOV_DEMUXER HEVC_MP4TOANNEXB_BSF MOV_MUXER fate-hevc-bsf-mp4toannexb: tests/data/hevc-mp4.mov fate-hevc-bsf-mp4toannexb: CMD = md5 -i $(TARGET_PATH)/tests/data/hevc-mp4.mov -c:v copy -fflags +bitexact -f hevc fate-hevc-bsf-mp4toannexb: CMP = oneline -fate-hevc-bsf-mp4toannexb: REF = 1873662a3af1848c37e4eb25722c8df9 +fate-hevc-bsf-mp4toannexb: REF = d41d8cd98f00b204e9800998ecf8427e fate-hevc-skiploopfilter: CMD = framemd5 -skip_loop_filter nokey -i $(TARGET_SAMPLES)/hevc-conformance/SAO_D_Samsung_5.bit -sws_flags bitexact FATE_HEVC += fate-hevc-skiploopfilter