From mboxrd@z Thu Jan 1 00:00:00 1970 Return-Path: Received: from ffbox0-bg.mplayerhq.hu (ffbox0-bg.ffmpeg.org [79.124.17.100]) by master.gitmailbox.com (Postfix) with ESMTP id D957A435E8 for ; Mon, 18 Jul 2022 17:41:39 +0000 (UTC) Received: from [127.0.1.1] (localhost [127.0.0.1]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTP id AC67068B8F6; Mon, 18 Jul 2022 20:41:36 +0300 (EEST) Received: from mail-oa1-f53.google.com (mail-oa1-f53.google.com [209.85.160.53]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTPS id 7E71068B6D6 for ; Mon, 18 Jul 2022 20:41:29 +0300 (EEST) Received: by mail-oa1-f53.google.com with SMTP id 586e51a60fabf-10bffc214ffso25684660fac.1 for ; Mon, 18 Jul 2022 10:41:29 -0700 (PDT) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=gmail.com; s=20210112; h=message-id:date:mime-version:user-agent:subject:content-language:to :references:from:in-reply-to:content-transfer-encoding; bh=i/1QhwzUthgRhtmWVLH6JKWe3E3fxNJHpJqfqx8Z/t0=; b=SHEdClq2yAJYTMwlpxIXDEN/0vWvjANAaZJfKwEC739CuOepmEbne80KCqd5naCeTb MIecmJqlvWLbSNTH870uQlbg+NlX1BbUQN5gMBm2GqIM/8t3yaFIM7jYjVhsYw1ntTku QT4TcSAaRumzQreum8u1PstzI+/DItPe+pnjef4ZBOzJLEomCBDw14alozMsqQ6y+c3U 2NM3aZu/mC6GDsQ+vqMpoJHL61uFteJb/t/vncEiC/Blos6R5RoEMRQMNhgfUWmmQ+ne 9Ns5c9kTT0fVoYqjpApTMzdnZEq8rz216N0cAmBdh5qNYnrfX1SgaNaPH56w3BLAMZdg kXpA== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20210112; h=x-gm-message-state:message-id:date:mime-version:user-agent:subject :content-language:to:references:from:in-reply-to :content-transfer-encoding; bh=i/1QhwzUthgRhtmWVLH6JKWe3E3fxNJHpJqfqx8Z/t0=; b=7riBozXra3ztNXqtk3jpfsvlSF79wAsO7zkp8wv97X1jfyCadh3Ym5Bz43FrzbdHE4 P98yO1XWVphRLtuXT3LSjoAYoqMctXUT2HVC1NDnotgMiMUc85CJTVYg8EDTYEJ3zwpH 64bIJRzuR5sIa4xabi/yaI6CgOZs3NVzBz13/V81B9OTvg3KgMt44UV3L1TuskGNS4CQ 5bvsuKBHKNJb6/j6cLu73ij43uoId+UDIxRnXYeHLY444MT2oNwNDRcuDRWouidGRsVJ Nd22iAxaeW1tq0/Fj8ZlJU8IXkSTAAyGpnqPMkpzrQUfH24RnyTCW2hT6SYmh4PmZpf3 GMKg== X-Gm-Message-State: AJIora95gnusXyLpI8SD9gQ2EPVQJHI9RvMIABPB/4T/qc9KgR40NRVh fhRBC5EFIlsrFCYdnNF5nhB5Y6TPdiebSQ== X-Google-Smtp-Source: AGRyM1uhWw3mrlLMP+qwjsfBcioo90Z69n7Q2krfU3kUyFrIIXI4ApQbTTf5svXTZe/3rbNZxgDJ0g== X-Received: by 2002:a05:6870:4720:b0:10c:5d1d:aecb with SMTP id b32-20020a056870472000b0010c5d1daecbmr17641208oaq.217.1658166085782; Mon, 18 Jul 2022 10:41:25 -0700 (PDT) Received: from [192.168.0.13] ([186.136.131.204]) by smtp.gmail.com with ESMTPSA id l8-20020a4ab2c8000000b0035eb4e5a6cbsm5049693ooo.33.2022.07.18.10.41.24 for (version=TLS1_3 cipher=TLS_AES_128_GCM_SHA256 bits=128/128); Mon, 18 Jul 2022 10:41:25 -0700 (PDT) Message-ID: <4de6238e-1c97-bcb5-f185-f0d58df9419c@gmail.com> Date: Mon, 18 Jul 2022 14:41:27 -0300 MIME-Version: 1.0 User-Agent: Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:91.0) Gecko/20100101 Thunderbird/91.11.0 Content-Language: en-US To: ffmpeg-devel@ffmpeg.org References: <20220717220619.GO2088045@pb2> <20220718071225.4446-1-anton@khirnov.net> From: James Almer In-Reply-To: <20220718071225.4446-1-anton@khirnov.net> Subject: Re: [FFmpeg-devel] [PATCH] lavc/libx264: support AV_CODEC_CAP_ENCODER_RECON_FRAME X-BeenThere: ffmpeg-devel@ffmpeg.org X-Mailman-Version: 2.1.29 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 Content-Transfer-Encoding: 7bit Content-Type: text/plain; charset="us-ascii"; Format="flowed" Errors-To: ffmpeg-devel-bounces@ffmpeg.org Sender: "ffmpeg-devel" Archived-At: List-Archive: List-Post: On 7/18/2022 4:12 AM, Anton Khirnov wrote: > --- > libavcodec/libx264.c | 52 +++++++++++++++++++++++++++++++++++++++++++- > 1 file changed, 51 insertions(+), 1 deletion(-) > > diff --git a/libavcodec/libx264.c b/libavcodec/libx264.c > index 98ec030865..5e360682e6 100644 > --- a/libavcodec/libx264.c > +++ b/libavcodec/libx264.c > @@ -311,6 +311,25 @@ static void free_picture(AVCodecContext *ctx) > pic->extra_sei.num_payloads = 0; > } > > +static enum AVPixelFormat csp_to_pixfmt(int csp) > +{ > + switch (csp) { > +#ifdef X264_CSP_I400 > + case X264_CSP_I400: return AV_PIX_FMT_GRAY8; > + case X264_CSP_I400 | X264_CSP_HIGH_DEPTH: return AV_PIX_FMT_GRAY10; > +#endif > + case X264_CSP_I420: return AV_PIX_FMT_YUV420P; > + case X264_CSP_I420 | X264_CSP_HIGH_DEPTH: return AV_PIX_FMT_YUV420P10; > + case X264_CSP_I422: return AV_PIX_FMT_YUV422P; > + case X264_CSP_I422 | X264_CSP_HIGH_DEPTH: return AV_PIX_FMT_YUV422P10; > + case X264_CSP_I444: return AV_PIX_FMT_YUV444P; > + case X264_CSP_I444 | X264_CSP_HIGH_DEPTH: return AV_PIX_FMT_YUV444P10; > + case X264_CSP_NV12: return AV_PIX_FMT_NV12; > + case X264_CSP_NV16: return AV_PIX_FMT_NV16; > + }; > + return AV_PIX_FMT_NONE; > +} > + > static int X264_frame(AVCodecContext *ctx, AVPacket *pkt, const AVFrame *frame, > int *got_packet) > { > @@ -496,6 +515,33 @@ static int X264_frame(AVCodecContext *ctx, AVPacket *pkt, const AVFrame *frame, > if (x264_encoder_encode(x4->enc, &nal, &nnal, frame? &x4->pic: NULL, &pic_out) < 0) > return AVERROR_EXTERNAL; > > + if (nnal && (ctx->flags & AV_CODEC_FLAG_RECON_FRAME)) { > + AVCodecInternal *avci = ctx->internal; > + > + av_frame_unref(avci->recon_frame); > + > + avci->recon_frame->format = csp_to_pixfmt(pic_out.img.i_csp); > + if (avci->recon_frame->format == AV_PIX_FMT_NONE) { > + av_log(ctx, AV_LOG_ERROR, > + "Unhandled reconstructed frame colorspace: %d\n", > + pic_out.img.i_csp); > + return AVERROR(ENOSYS); > + } > + > + avci->recon_frame->width = ctx->width; > + avci->recon_frame->height = ctx->height; > + for (int i = 0; i < pic_out.img.i_plane; i++) { > + avci->recon_frame->data[i] = pic_out.img.plane[i]; > + avci->recon_frame->linesize[i] = pic_out.img.i_stride[i]; > + } > + > + ret = av_frame_make_writable(avci->recon_frame); Why not ff_encode_alloc_frame() for this? Can csp_to_pixfmt(pic_out.img.i_csp) be different than ctx->pix_fmt? > + if (ret < 0) { > + av_frame_unref(avci->recon_frame); > + return ret; > + } > + } > + > ret = encode_nals(ctx, pkt, nal, nnal); > if (ret < 0) > return ret; > @@ -928,6 +974,9 @@ static av_cold int X264_init(AVCodecContext *avctx) > if (avctx->flags & AV_CODEC_FLAG_GLOBAL_HEADER) > x4->params.b_repeat_headers = 0; > > + if (avctx->flags & AV_CODEC_FLAG_RECON_FRAME) > + x4->params.b_full_recon = 1; > + > if(x4->x264opts){ > const char *p= x4->x264opts; > while(p){ > @@ -1223,7 +1272,8 @@ FFCodec ff_libx264_encoder = { > .p.id = AV_CODEC_ID_H264, > .p.capabilities = AV_CODEC_CAP_DR1 | AV_CODEC_CAP_DELAY | > AV_CODEC_CAP_OTHER_THREADS | > - AV_CODEC_CAP_ENCODER_REORDERED_OPAQUE, > + AV_CODEC_CAP_ENCODER_REORDERED_OPAQUE | > + AV_CODEC_CAP_ENCODER_RECON_FRAME, > .p.priv_class = &x264_class, > .p.wrapper_name = "libx264", > .priv_data_size = sizeof(X264Context), _______________________________________________ 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".