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	goto error;	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	if (res < 0) {	1976	if (res < 0) {
1977	p_nvenc->nvEncUnlockBitstream(ctx->nvencoder, tmpoutsurf->output_surface);	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	.priv_class = &h264_nvenc_class,	287	.priv_class = &h264_nvenc_class,
288	.defaults = defaults,	288	.defaults = defaults,
289	.capabilities = AV_CODEC_CAP_DELAY \| AV_CODEC_CAP_HARDWARE \|	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	.caps_internal = FF_CODEC_CAP_INIT_CLEANUP,	291	.caps_internal = FF_CODEC_CAP_INIT_CLEANUP,
292	.pix_fmts = ff_nvenc_pix_fmts,	292	.pix_fmts = ff_nvenc_pix_fmts,
293	.wrapper_name = "nvenc",	293	.wrapper_name = "nvenc",

libavcodec/nvenc_hevc.c+1−1

@@ -237,7 +237,7 @@ AVCodec ff_hevc_nvenc_encoder = { · 236 unmodified lines
237	.defaults = defaults,	237	.defaults = defaults,
238	.pix_fmts = ff_nvenc_pix_fmts,	238	.pix_fmts = ff_nvenc_pix_fmts,
239	.capabilities = AV_CODEC_CAP_DELAY \| AV_CODEC_CAP_HARDWARE \|	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	.caps_internal = FF_CODEC_CAP_INIT_CLEANUP,	241	.caps_internal = FF_CODEC_CAP_INIT_CLEANUP,
242	.wrapper_name = "nvenc",	242	.wrapper_name = "nvenc",
243	.hw_configs = ff_nvenc_hw_configs,	243	.hw_configs = ff_nvenc_hw_configs,