/* * Copyright (C) 2017 Ericsson AB. All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions * are met: * * 1. Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * 2. Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer * in the documentation and/or other materials provided with the * distribution. * * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */ #ifdef HAVE_CONFIG_H #include "config.h" #endif #include #ifdef HAVE_NVCODEC_DGPU #include "gstnvdec.h" #include "gstnvenc.h" #include "gstnvav1dec.h" #include "gstnvh264dec.h" #include "gstnvh265dec.h" #include "gstnvvp8dec.h" #include "gstnvvp9dec.h" #include "gstnvdecoder.h" #endif #include "gstcudamemorycopy.h" #include "gstcudaconvertscale.h" #include #ifdef G_OS_WIN32 #include #endif #ifdef HAVE_NVCODEC_DGPU #include "gstnvh264encoder.h" #include "gstnvh265encoder.h" #include "gstnvav1encoder.h" #include "gstnvjpegenc.h" #endif #include "gstcudaipcsink.h" #include "gstcudaipcsrc.h" #include "gstnvcodecutils.h" #include GST_DEBUG_CATEGORY (gst_nvcodec_debug); GST_DEBUG_CATEGORY (gst_nvdec_debug); GST_DEBUG_CATEGORY (gst_nvenc_debug); GST_DEBUG_CATEGORY (gst_nv_decoder_debug); #define GST_CAT_DEFAULT gst_nvcodec_debug #ifdef G_OS_WIN32 #define CUDA_LIBNAME "nvcuda.dll" #define NVCUVID_LIBNAME "nvcuvid.dll" #ifdef _WIN64 #define NVENC_LIBNAME "nvEncodeAPI64.dll" #else #define NVENC_LIBNAME "nvEncodeAPI.dll" #endif #define NVRTC_LIBNAME "nvrtc64_*_0.dll" #else /* G_OS_WIN32 */ #define CUDA_LIBNAME "libcuda.so.1" #define NVCUVID_LIBNAME "libnvcuvid.so.1" #define NVENC_LIBNAME "libnvidia-encode.so.1" #define NVRTC_LIBNAME "libnvrtc.so" #endif /* G_OS_WIN32 */ static void plugin_deinit (gpointer data) { gst_cuda_ipc_client_deinit (); } static gboolean check_runtime_compiler (void) { /* *INDENT-OFF* */ const gchar *nvrtc_test_source = "__global__ void\n" "my_kernel (void) {}"; /* *INDENT-ON* */ gchar *test_ptx; if (!gst_cuda_nvrtc_load_library ()) return FALSE; test_ptx = gst_cuda_nvrtc_compile (nvrtc_test_source); if (!test_ptx) return FALSE; g_free (test_ptx); return TRUE; } static gboolean plugin_init (GstPlugin * plugin) { CUresult cuda_ret; const char *err_name = NULL, *err_desc = NULL; gint dev_count = 0; guint i; #ifdef HAVE_NVCODEC_DGPU gboolean nvdec_available = TRUE; gboolean nvenc_available = TRUE; /* hardcoded minimum supported version */ guint api_major_ver = 8; guint api_minor_ver = 1; GList *h264_enc_cdata = NULL; GList *h265_enc_cdata = NULL; GList *av1_enc_cdata = NULL; #endif gboolean have_nvrtc = FALSE; GST_DEBUG_CATEGORY_INIT (gst_nvcodec_debug, "nvcodec", 0, "nvcodec"); GST_DEBUG_CATEGORY_INIT (gst_nvdec_debug, "nvdec", 0, "nvdec"); GST_DEBUG_CATEGORY_INIT (gst_nvenc_debug, "nvenc", 0, "nvenc"); GST_DEBUG_CATEGORY_INIT (gst_nv_decoder_debug, "nvdecoder", 0, "nvdecoder"); if (!gst_cuda_load_library ()) { gst_plugin_add_status_warning (plugin, "CUDA library \"" CUDA_LIBNAME "\" was not found."); return TRUE; } #ifdef HAVE_NVCODEC_DGPU /* get available API version from nvenc and it will be passed to * nvdec */ if (!gst_nvenc_load_library (&api_major_ver, &api_minor_ver)) { gst_plugin_add_status_warning (plugin, "NVENC library \"" NVENC_LIBNAME "\" was not found."); nvenc_available = FALSE; } if (!gst_cuvid_load_library (api_major_ver, api_minor_ver)) { GST_WARNING ("Failed to load nvdec library version %u.%u", api_major_ver, api_minor_ver); gst_plugin_add_status_warning (plugin, "NVDEC library \"" NVCUVID_LIBNAME "\" was not found."); nvdec_available = FALSE; } if (!nvdec_available && !nvenc_available) return TRUE; #endif cuda_ret = CuInit (0); if (cuda_ret != CUDA_SUCCESS) { CuGetErrorName (cuda_ret, &err_name); CuGetErrorString (cuda_ret, &err_desc); GST_ERROR ("Failed to init cuda, cuInit ret: 0x%x: %s: %s", (int) cuda_ret, err_name, err_desc); /* to abort if GST_CUDA_CRITICAL_ERRORS is configured */ gst_cuda_result (CUDA_ERROR_NO_DEVICE); gst_plugin_add_status_error (plugin, N_("Unable to initialize CUDA library.")); return TRUE; } cuda_ret = CuDeviceGetCount (&dev_count); if (cuda_ret != CUDA_SUCCESS || !dev_count) { CuGetErrorName (cuda_ret, &err_name); CuGetErrorString (cuda_ret, &err_desc); GST_ERROR ("No available device, cuDeviceGetCount ret: 0x%x: %s %s", (int) cuda_ret, err_name, err_desc); gst_plugin_add_status_warning (plugin, N_("No NVIDIA graphics cards detected!")); return TRUE; } have_nvrtc = check_runtime_compiler (); if (!have_nvrtc) { gst_plugin_add_status_info (plugin, "CUDA runtime compilation library \"" NVRTC_LIBNAME "\" was not found, " "check CUDA toolkit package installation"); } for (i = 0; i < dev_count; i++) { GstCudaContext *context = gst_cuda_context_new (i); #if defined(G_OS_WIN32) || defined(HAVE_NVCODEC_DGPU) gint64 adapter_luid = 0; #endif if (!context) { GST_WARNING ("Failed to create context for device %d", i); continue; } #ifdef G_OS_WIN32 g_object_get (context, "dxgi-adapter-luid", &adapter_luid, NULL); #endif #ifdef HAVE_NVCODEC_DGPU if (nvdec_available) { gint j; for (j = 0; j < cudaVideoCodec_NumCodecs; j++) { GstCaps *sink_template = NULL; GstCaps *src_template = NULL; cudaVideoCodec codec = (cudaVideoCodec) j; gboolean register_cuviddec = FALSE; if (gst_nv_decoder_check_device_caps (context, codec, &sink_template, &src_template)) { const gchar *codec_name = gst_cuda_video_codec_to_string (codec); GST_INFO ("CUDA video codec %s, sink template %" GST_PTR_FORMAT "src template %" GST_PTR_FORMAT, codec_name, sink_template, src_template); switch (codec) { case cudaVideoCodec_H264: /* higher than avdec_h264 */ gst_nv_h264_dec_register (plugin, i, adapter_luid, GST_RANK_PRIMARY + 1, sink_template, src_template); break; case cudaVideoCodec_HEVC: /* higher than avdec_h265 */ gst_nv_h265_dec_register (plugin, i, adapter_luid, GST_RANK_PRIMARY + 1, sink_template, src_template); break; case cudaVideoCodec_VP8: gst_nv_vp8_dec_register (plugin, i, adapter_luid, GST_RANK_PRIMARY, sink_template, src_template); break; case cudaVideoCodec_VP9: gst_nv_vp9_dec_register (plugin, i, adapter_luid, GST_RANK_PRIMARY, sink_template, src_template); break; case cudaVideoCodec_AV1: /* rust dav1ddec has "primary" rank */ gst_nv_av1_dec_register (plugin, i, adapter_luid, GST_RANK_PRIMARY + 1, sink_template, src_template); break; default: register_cuviddec = TRUE; break; } if (register_cuviddec) { gst_nvdec_plugin_init (plugin, i, codec, codec_name, sink_template, src_template); } gst_caps_unref (sink_template); gst_caps_unref (src_template); } } } if (nvenc_available) { GstNvEncoderClassData *cdata; #ifdef G_OS_WIN32 if (g_win32_check_windows_version (6, 0, 0, G_WIN32_OS_ANY)) { GstD3D11Device *d3d11_device; d3d11_device = gst_d3d11_device_new_for_adapter_luid (adapter_luid, D3D11_CREATE_DEVICE_BGRA_SUPPORT); if (!d3d11_device) { GST_WARNING ("Failed to d3d11 create device"); } else { cdata = gst_nv_h264_encoder_register_d3d11 (plugin, d3d11_device, GST_RANK_NONE); if (cdata) h264_enc_cdata = g_list_append (h264_enc_cdata, cdata); cdata = gst_nv_h265_encoder_register_d3d11 (plugin, d3d11_device, GST_RANK_NONE); if (cdata) h265_enc_cdata = g_list_append (h265_enc_cdata, cdata); cdata = gst_nv_av1_encoder_register_d3d11 (plugin, d3d11_device, GST_RANK_NONE); if (cdata) av1_enc_cdata = g_list_append (av1_enc_cdata, cdata); gst_object_unref (d3d11_device); } } #endif cdata = gst_nv_h264_encoder_register_cuda (plugin, context, GST_RANK_PRIMARY + 1); if (cdata) h264_enc_cdata = g_list_append (h264_enc_cdata, cdata); cdata = gst_nv_h265_encoder_register_cuda (plugin, context, GST_RANK_PRIMARY + 1); if (cdata) h265_enc_cdata = g_list_append (h265_enc_cdata, cdata); cdata = gst_nv_av1_encoder_register_cuda (plugin, context, GST_RANK_PRIMARY + 1); if (cdata) av1_enc_cdata = g_list_append (av1_enc_cdata, cdata); } gst_nv_jpeg_enc_register (plugin, context, GST_RANK_NONE, have_nvrtc); #endif gst_object_unref (context); } #ifdef HAVE_NVCODEC_DGPU if (h264_enc_cdata) { gst_nv_h264_encoder_register_auto_select (plugin, h264_enc_cdata, GST_RANK_NONE); } if (h265_enc_cdata) { gst_nv_h265_encoder_register_auto_select (plugin, h265_enc_cdata, GST_RANK_NONE); } if (av1_enc_cdata) { gst_nv_av1_encoder_register_auto_select (plugin, av1_enc_cdata, GST_RANK_NONE); } #endif gst_cuda_memory_copy_register (plugin, GST_RANK_NONE); if (have_nvrtc) { gst_element_register (plugin, "cudaconvert", GST_RANK_NONE, GST_TYPE_CUDA_CONVERT); gst_element_register (plugin, "cudascale", GST_RANK_NONE, GST_TYPE_CUDA_SCALE); gst_element_register (plugin, "cudaconvertscale", GST_RANK_NONE, GST_TYPE_CUDA_CONVERT_SCALE); } gst_element_register (plugin, "cudaipcsink", GST_RANK_NONE, GST_TYPE_CUDA_IPC_SINK); gst_element_register (plugin, "cudaipcsrc", GST_RANK_NONE, GST_TYPE_CUDA_IPC_SRC); gst_cuda_memory_init_once (); if (gst_cuda_nvmm_init_once ()) GST_INFO ("Enable NVMM support"); g_object_set_data_full (G_OBJECT (plugin), "plugin-nvcodec-shutdown", (gpointer) "shutdown-data", (GDestroyNotify) plugin_deinit); return TRUE; } GST_PLUGIN_DEFINE (GST_VERSION_MAJOR, GST_VERSION_MINOR, nvcodec, "GStreamer NVCODEC plugin", plugin_init, VERSION, "LGPL", GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN)