avcodec/nvenc: use AVCodecContext.get_encode_buffer()

James Almer committed 5 years ago

Commit 238e08b1 parent b86af933 files · +3 −3Browse files

Unified · Split

3 files

libavcodec/nvenc.c +1 −1
libavcodec/nvenc_h264.c +1 −1
libavcodec/nvenc_hevc.c +1 −1

libavcodec/nvenc.c+1−1

@@ -1971,7 +1971,7 @@ static int process_output_surface(AVCodecContext avctx, AVPacket pkt, NvencSur · 1970 unmodified lines
1971	1971	goto error;
1972	1972	}
1973	1973
1974		− res = av_new_packet(pkt, lock_params.bitstreamSizeInBytes);
	1974	+ res = ff_get_encode_buffer(avctx, pkt, lock_params.bitstreamSizeInBytes, 0);
1975	1975
1976	1976	if (res < 0) {
1977	1977	p_nvenc->nvEncUnlockBitstream(ctx->nvencoder, tmpoutsurf->output_surface);

libavcodec/nvenc_h264.c+1−1

@@ -287,7 +287,7 @@ AVCodec ff_h264_nvenc_encoder = { · 286 unmodified lines
287	287	.priv_class = &h264_nvenc_class,
288	288	.defaults = defaults,
289	289	.capabilities = AV_CODEC_CAP_DELAY \| AV_CODEC_CAP_HARDWARE \|
290		− AV_CODEC_CAP_ENCODER_FLUSH,
	290	+ AV_CODEC_CAP_ENCODER_FLUSH \| AV_CODEC_CAP_DR1,
291	291	.caps_internal = FF_CODEC_CAP_INIT_CLEANUP,
292	292	.pix_fmts = ff_nvenc_pix_fmts,
293	293	.wrapper_name = "nvenc",

libavcodec/nvenc_hevc.c+1−1

@@ -237,7 +237,7 @@ AVCodec ff_hevc_nvenc_encoder = { · 236 unmodified lines
237	237	.defaults = defaults,
238	238	.pix_fmts = ff_nvenc_pix_fmts,
239	239	.capabilities = AV_CODEC_CAP_DELAY \| AV_CODEC_CAP_HARDWARE \|
240		− AV_CODEC_CAP_ENCODER_FLUSH,
	240	+ AV_CODEC_CAP_ENCODER_FLUSH \| AV_CODEC_CAP_DR1,
241	241	.caps_internal = FF_CODEC_CAP_INIT_CLEANUP,
242	242	.wrapper_name = "nvenc",
243	243	.hw_configs = ff_nvenc_hw_configs,