Nvidia nvenc cuvid + ffmpeg 硬编解码

ffmpeg 命令行，可指定GPU运算：ffmpeg -hwaccel_output_format cuda -vcodec h264_cuvid -i input.mp4 -vcodec h264_nvenc -gpu 0 -acodec copy -f mp4 -y output.mp4

ffmpeg SDK调用: 

av_find_encoder_by_name(“h264_nvenc”);
av_find_encoder_by_name(“hevc_nvenc”);

 

nvenc.c

av_cold int ff_nvenc_encode_init(AVCodecContext *avctx)

static av_cold int nvenc_setup_device(AVCodecContext *avctx)

static av_cold int nvenc_check_device(AVCodecContext *avctx, int idx)

static av_cold int nvenc_open_session(AVCodecContext *avctx)

{

    NV_ENC_OPEN_ENCODE_SESSION_EX_PARAMS params = { 0 };
    NvencContext *ctx = avctx->priv_data;
    NV_ENCODE_API_FUNCTION_LIST *p_nvenc = &ctx->nvenc_dload_funcs.nvenc_funcs;
    NVENCSTATUS ret;
    params.version    = NV_ENC_OPEN_ENCODE_SESSION_EX_PARAMS_VER;
    params.apiVersion = NVENCAPI_VERSION;
    if (ctx->d3d11_device) {
        para