avcodec/librav1e: use AVCodecContext.get_encode_buffer()

Signed-off-by: James Almer <jamrial@gmail.com>
This commit is contained in:
James Almer 2021-03-12 20:50:22 -03:00
parent b615e8a782
commit d51a6884ab
1 changed files with 3 additions and 2 deletions

View File

@ -532,7 +532,7 @@ retry:
return AVERROR_UNKNOWN; return AVERROR_UNKNOWN;
} }
ret = av_new_packet(pkt, rpkt->len); ret = ff_get_encode_buffer(avctx, pkt, rpkt->len, 0);
if (ret < 0) { if (ret < 0) {
av_log(avctx, AV_LOG_ERROR, "Could not allocate packet.\n"); av_log(avctx, AV_LOG_ERROR, "Could not allocate packet.\n");
rav1e_packet_unref(rpkt); rav1e_packet_unref(rpkt);
@ -624,7 +624,8 @@ AVCodec ff_librav1e_encoder = {
.priv_class = &class, .priv_class = &class,
.defaults = librav1e_defaults, .defaults = librav1e_defaults,
.pix_fmts = librav1e_pix_fmts, .pix_fmts = librav1e_pix_fmts,
.capabilities = AV_CODEC_CAP_DELAY | AV_CODEC_CAP_OTHER_THREADS, .capabilities = AV_CODEC_CAP_DELAY | AV_CODEC_CAP_OTHER_THREADS |
AV_CODEC_CAP_DR1,
.caps_internal = FF_CODEC_CAP_INIT_CLEANUP | FF_CODEC_CAP_AUTO_THREADS, .caps_internal = FF_CODEC_CAP_INIT_CLEANUP | FF_CODEC_CAP_AUTO_THREADS,
.wrapper_name = "librav1e", .wrapper_name = "librav1e",
}; };