From c7a0addf93949e89cfacd17cfa59396ef4203598 Mon Sep 17 00:00:00 2001 From: Andreas Rheinhardt Date: Tue, 27 Apr 2021 16:58:16 +0200 Subject: avcodec/rawenc: Allow user-supplied buffers Trivial for an encoder that knows the size of the output packet in advance. Signed-off-by: Andreas Rheinhardt --- libavcodec/rawenc.c | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'libavcodec/rawenc.c') diff --git a/libavcodec/rawenc.c b/libavcodec/rawenc.c index 73b0b92916..bd992239a5 100644 --- a/libavcodec/rawenc.c +++ b/libavcodec/rawenc.c @@ -25,6 +25,7 @@ */ #include "avcodec.h" +#include "encode.h" #include "raw.h" #include "internal.h" #include "libavutil/pixdesc.h" @@ -53,7 +54,7 @@ static int raw_encode(AVCodecContext *avctx, AVPacket *pkt, if (ret < 0) return ret; - if ((ret = ff_alloc_packet2(avctx, pkt, ret, ret)) < 0) + if ((ret = ff_get_encode_buffer(avctx, pkt, ret, 0)) < 0) return ret; if ((ret = av_image_copy_to_buffer(pkt->data, pkt->size, (const uint8_t **)frame->data, frame->linesize, @@ -85,6 +86,7 @@ const AVCodec ff_rawvideo_encoder = { .long_name = NULL_IF_CONFIG_SMALL("raw video"), .type = AVMEDIA_TYPE_VIDEO, .id = AV_CODEC_ID_RAWVIDEO, + .capabilities = AV_CODEC_CAP_DR1, .init = raw_encode_init, .encode2 = raw_encode, .caps_internal = FF_CODEC_CAP_INIT_THREADSAFE, -- cgit v1.2.3