summaryrefslogtreecommitdiffstats
path: root/compat/cuda/dynlink_loader.h
diff options
context:
space:
mode:
Diffstat (limited to 'compat/cuda/dynlink_loader.h')
-rw-r--r--compat/cuda/dynlink_loader.h273
1 files changed, 19 insertions, 254 deletions
diff --git a/compat/cuda/dynlink_loader.h b/compat/cuda/dynlink_loader.h
index fa43782..9f93465 100644
--- a/compat/cuda/dynlink_loader.h
+++ b/compat/cuda/dynlink_loader.h
@@ -1,268 +1,33 @@
/*
- * This copyright notice applies to this header file only:
+ * This file is part of FFmpeg.
*
- * Copyright (c) 2016
+ * FFmpeg is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
*
- * Permission is hereby granted, free of charge, to any person
- * obtaining a copy of this software and associated documentation
- * files (the "Software"), to deal in the Software without
- * restriction, including without limitation the rights to use,
- * copy, modify, merge, publish, distribute, sublicense, and/or sell
- * copies of the software, and to permit persons to whom the
- * software is furnished to do so, subject to the following
- * conditions:
+ * FFmpeg is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
*
- * The above copyright notice and this permission notice shall be
- * included in all copies or substantial portions of the Software.
- *
- * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
- * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
- * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
- * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
- * HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
- * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
- * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
- * OTHER DEALINGS IN THE SOFTWARE.
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with FFmpeg; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
*/
#ifndef AV_COMPAT_CUDA_DYNLINK_LOADER_H
#define AV_COMPAT_CUDA_DYNLINK_LOADER_H
-#include "compat/cuda/dynlink_cuda.h"
-#include "compat/cuda/dynlink_nvcuvid.h"
-#include "compat/nvenc/nvEncodeAPI.h"
-#include "compat/w32dlfcn.h"
-
#include "libavutil/log.h"
-#include "libavutil/error.h"
-
-#if defined(_WIN32)
-# define LIB_HANDLE HMODULE
-#else
-# define LIB_HANDLE void*
-#endif
-
-#if defined(_WIN32) || defined(__CYGWIN__)
-# define CUDA_LIBNAME "nvcuda.dll"
-# define NVCUVID_LIBNAME "nvcuvid.dll"
-# if ARCH_X86_64
-# define NVENC_LIBNAME "nvEncodeAPI64.dll"
-# else
-# define NVENC_LIBNAME "nvEncodeAPI.dll"
-# endif
-#else
-# define CUDA_LIBNAME "libcuda.so.1"
-# define NVCUVID_LIBNAME "libnvcuvid.so.1"
-# define NVENC_LIBNAME "libnvidia-encode.so.1"
-#endif
-
-#define LOAD_LIBRARY(l, path) \
- do { \
- if (!((l) = dlopen(path, RTLD_LAZY))) { \
- av_log(logctx, AV_LOG_ERROR, "Cannot load %s\n", path); \
- ret = AVERROR_UNKNOWN; \
- goto error; \
- } \
- av_log(logctx, AV_LOG_TRACE, "Loaded lib: %s\n", path); \
- } while (0)
-
-#define LOAD_SYMBOL(fun, tp, symbol) \
- do { \
- if (!((f->fun) = (tp*)dlsym(f->lib, symbol))) { \
- av_log(logctx, AV_LOG_ERROR, "Cannot load %s\n", symbol); \
- ret = AVERROR_UNKNOWN; \
- goto error; \
- } \
- av_log(logctx, AV_LOG_TRACE, "Loaded sym: %s\n", symbol); \
- } while (0)
-
-#define LOAD_SYMBOL_OPT(fun, tp, symbol) \
- do { \
- if (!((f->fun) = (tp*)dlsym(f->lib, symbol))) { \
- av_log(logctx, AV_LOG_DEBUG, "Cannot load optional %s\n", symbol); \
- } else { \
- av_log(logctx, AV_LOG_TRACE, "Loaded sym: %s\n", symbol); \
- } \
- } while (0)
-
-#define GENERIC_LOAD_FUNC_PREAMBLE(T, n, N) \
- T *f; \
- int ret; \
- \
- n##_free_functions(functions); \
- \
- f = *functions = av_mallocz(sizeof(*f)); \
- if (!f) \
- return AVERROR(ENOMEM); \
- \
- LOAD_LIBRARY(f->lib, N);
-
-#define GENERIC_LOAD_FUNC_FINALE(n) \
- return 0; \
-error: \
- n##_free_functions(functions); \
- return ret;
-
-#define GENERIC_FREE_FUNC() \
- if (!functions) \
- return; \
- if (*functions && (*functions)->lib) \
- dlclose((*functions)->lib); \
- av_freep(functions);
-
-#ifdef AV_COMPAT_DYNLINK_CUDA_H
-typedef struct CudaFunctions {
- tcuInit *cuInit;
- tcuDeviceGetCount *cuDeviceGetCount;
- tcuDeviceGet *cuDeviceGet;
- tcuDeviceGetName *cuDeviceGetName;
- tcuDeviceComputeCapability *cuDeviceComputeCapability;
- tcuCtxCreate_v2 *cuCtxCreate;
- tcuCtxPushCurrent_v2 *cuCtxPushCurrent;
- tcuCtxPopCurrent_v2 *cuCtxPopCurrent;
- tcuCtxDestroy_v2 *cuCtxDestroy;
- tcuMemAlloc_v2 *cuMemAlloc;
- tcuMemFree_v2 *cuMemFree;
- tcuMemcpy2D_v2 *cuMemcpy2D;
- tcuGetErrorName *cuGetErrorName;
- tcuGetErrorString *cuGetErrorString;
-
- LIB_HANDLE lib;
-} CudaFunctions;
-#else
-typedef struct CudaFunctions CudaFunctions;
-#endif
-
-typedef struct CuvidFunctions {
- tcuvidGetDecoderCaps *cuvidGetDecoderCaps;
- tcuvidCreateDecoder *cuvidCreateDecoder;
- tcuvidDestroyDecoder *cuvidDestroyDecoder;
- tcuvidDecodePicture *cuvidDecodePicture;
- tcuvidMapVideoFrame *cuvidMapVideoFrame;
- tcuvidUnmapVideoFrame *cuvidUnmapVideoFrame;
- tcuvidCtxLockCreate *cuvidCtxLockCreate;
- tcuvidCtxLockDestroy *cuvidCtxLockDestroy;
- tcuvidCtxLock *cuvidCtxLock;
- tcuvidCtxUnlock *cuvidCtxUnlock;
-
- tcuvidCreateVideoSource *cuvidCreateVideoSource;
- tcuvidCreateVideoSourceW *cuvidCreateVideoSourceW;
- tcuvidDestroyVideoSource *cuvidDestroyVideoSource;
- tcuvidSetVideoSourceState *cuvidSetVideoSourceState;
- tcuvidGetVideoSourceState *cuvidGetVideoSourceState;
- tcuvidGetSourceVideoFormat *cuvidGetSourceVideoFormat;
- tcuvidGetSourceAudioFormat *cuvidGetSourceAudioFormat;
- tcuvidCreateVideoParser *cuvidCreateVideoParser;
- tcuvidParseVideoData *cuvidParseVideoData;
- tcuvidDestroyVideoParser *cuvidDestroyVideoParser;
-
- LIB_HANDLE lib;
-} CuvidFunctions;
-
-typedef NVENCSTATUS NVENCAPI tNvEncodeAPICreateInstance(NV_ENCODE_API_FUNCTION_LIST *functionList);
-typedef NVENCSTATUS NVENCAPI tNvEncodeAPIGetMaxSupportedVersion(uint32_t* version);
-
-typedef struct NvencFunctions {
- tNvEncodeAPICreateInstance *NvEncodeAPICreateInstance;
- tNvEncodeAPIGetMaxSupportedVersion *NvEncodeAPIGetMaxSupportedVersion;
-
- LIB_HANDLE lib;
-} NvencFunctions;
-
-#ifdef AV_COMPAT_DYNLINK_CUDA_H
-static inline void cuda_free_functions(CudaFunctions **functions)
-{
- GENERIC_FREE_FUNC();
-}
-#endif
-
-static inline void cuvid_free_functions(CuvidFunctions **functions)
-{
- GENERIC_FREE_FUNC();
-}
-
-static inline void nvenc_free_functions(NvencFunctions **functions)
-{
- GENERIC_FREE_FUNC();
-}
-
-#ifdef AV_COMPAT_DYNLINK_CUDA_H
-static inline int cuda_load_functions(CudaFunctions **functions, void *logctx)
-{
- GENERIC_LOAD_FUNC_PREAMBLE(CudaFunctions, cuda, CUDA_LIBNAME);
-
- LOAD_SYMBOL(cuInit, tcuInit, "cuInit");
- LOAD_SYMBOL(cuDeviceGetCount, tcuDeviceGetCount, "cuDeviceGetCount");
- LOAD_SYMBOL(cuDeviceGet, tcuDeviceGet, "cuDeviceGet");
- LOAD_SYMBOL(cuDeviceGetName, tcuDeviceGetName, "cuDeviceGetName");
- LOAD_SYMBOL(cuDeviceComputeCapability, tcuDeviceComputeCapability, "cuDeviceComputeCapability");
- LOAD_SYMBOL(cuCtxCreate, tcuCtxCreate_v2, "cuCtxCreate_v2");
- LOAD_SYMBOL(cuCtxPushCurrent, tcuCtxPushCurrent_v2, "cuCtxPushCurrent_v2");
- LOAD_SYMBOL(cuCtxPopCurrent, tcuCtxPopCurrent_v2, "cuCtxPopCurrent_v2");
- LOAD_SYMBOL(cuCtxDestroy, tcuCtxDestroy_v2, "cuCtxDestroy_v2");
- LOAD_SYMBOL(cuMemAlloc, tcuMemAlloc_v2, "cuMemAlloc_v2");
- LOAD_SYMBOL(cuMemFree, tcuMemFree_v2, "cuMemFree_v2");
- LOAD_SYMBOL(cuMemcpy2D, tcuMemcpy2D_v2, "cuMemcpy2D_v2");
- LOAD_SYMBOL(cuGetErrorName, tcuGetErrorName, "cuGetErrorName");
- LOAD_SYMBOL(cuGetErrorString, tcuGetErrorString, "cuGetErrorString");
-
- GENERIC_LOAD_FUNC_FINALE(cuda);
-}
-#endif
-
-static inline int cuvid_load_functions(CuvidFunctions **functions, void *logctx)
-{
- GENERIC_LOAD_FUNC_PREAMBLE(CuvidFunctions, cuvid, NVCUVID_LIBNAME);
-
- LOAD_SYMBOL_OPT(cuvidGetDecoderCaps, tcuvidGetDecoderCaps, "cuvidGetDecoderCaps");
- LOAD_SYMBOL(cuvidCreateDecoder, tcuvidCreateDecoder, "cuvidCreateDecoder");
- LOAD_SYMBOL(cuvidDestroyDecoder, tcuvidDestroyDecoder, "cuvidDestroyDecoder");
- LOAD_SYMBOL(cuvidDecodePicture, tcuvidDecodePicture, "cuvidDecodePicture");
-#ifdef __CUVID_DEVPTR64
- LOAD_SYMBOL(cuvidMapVideoFrame, tcuvidMapVideoFrame, "cuvidMapVideoFrame64");
- LOAD_SYMBOL(cuvidUnmapVideoFrame, tcuvidUnmapVideoFrame, "cuvidUnmapVideoFrame64");
-#else
- LOAD_SYMBOL(cuvidMapVideoFrame, tcuvidMapVideoFrame, "cuvidMapVideoFrame");
- LOAD_SYMBOL(cuvidUnmapVideoFrame, tcuvidUnmapVideoFrame, "cuvidUnmapVideoFrame");
-#endif
- LOAD_SYMBOL(cuvidCtxLockCreate, tcuvidCtxLockCreate, "cuvidCtxLockCreate");
- LOAD_SYMBOL(cuvidCtxLockDestroy, tcuvidCtxLockDestroy, "cuvidCtxLockDestroy");
- LOAD_SYMBOL(cuvidCtxLock, tcuvidCtxLock, "cuvidCtxLock");
- LOAD_SYMBOL(cuvidCtxUnlock, tcuvidCtxUnlock, "cuvidCtxUnlock");
-
- LOAD_SYMBOL(cuvidCreateVideoSource, tcuvidCreateVideoSource, "cuvidCreateVideoSource");
- LOAD_SYMBOL(cuvidCreateVideoSourceW, tcuvidCreateVideoSourceW, "cuvidCreateVideoSourceW");
- LOAD_SYMBOL(cuvidDestroyVideoSource, tcuvidDestroyVideoSource, "cuvidDestroyVideoSource");
- LOAD_SYMBOL(cuvidSetVideoSourceState, tcuvidSetVideoSourceState, "cuvidSetVideoSourceState");
- LOAD_SYMBOL(cuvidGetVideoSourceState, tcuvidGetVideoSourceState, "cuvidGetVideoSourceState");
- LOAD_SYMBOL(cuvidGetSourceVideoFormat, tcuvidGetSourceVideoFormat, "cuvidGetSourceVideoFormat");
- LOAD_SYMBOL(cuvidGetSourceAudioFormat, tcuvidGetSourceAudioFormat, "cuvidGetSourceAudioFormat");
- LOAD_SYMBOL(cuvidCreateVideoParser, tcuvidCreateVideoParser, "cuvidCreateVideoParser");
- LOAD_SYMBOL(cuvidParseVideoData, tcuvidParseVideoData, "cuvidParseVideoData");
- LOAD_SYMBOL(cuvidDestroyVideoParser, tcuvidDestroyVideoParser, "cuvidDestroyVideoParser");
-
- GENERIC_LOAD_FUNC_FINALE(cuvid);
-}
-
-static inline int nvenc_load_functions(NvencFunctions **functions, void *logctx)
-{
- GENERIC_LOAD_FUNC_PREAMBLE(NvencFunctions, nvenc, NVENC_LIBNAME);
-
- LOAD_SYMBOL(NvEncodeAPICreateInstance, tNvEncodeAPICreateInstance, "NvEncodeAPICreateInstance");
- LOAD_SYMBOL(NvEncodeAPIGetMaxSupportedVersion, tNvEncodeAPIGetMaxSupportedVersion, "NvEncodeAPIGetMaxSupportedVersion");
+#include "compat/w32dlfcn.h"
- GENERIC_LOAD_FUNC_FINALE(nvenc);
-}
+#define FFNV_LOAD_FUNC(path) dlopen((path), RTLD_LAZY)
+#define FFNV_SYM_FUNC(lib, sym) dlsym((lib), (sym))
+#define FFNV_FREE_FUNC(lib) dlclose(lib)
+#define FFNV_LOG_FUNC(logctx, msg, ...) av_log(logctx, AV_LOG_ERROR, msg, __VA_ARGS__)
+#define FFNV_DEBUG_LOG_FUNC(logctx, msg, ...) av_log(logctx, AV_LOG_DEBUG, msg, __VA_ARGS__)
-#undef GENERIC_LOAD_FUNC_PREAMBLE
-#undef LOAD_LIBRARY
-#undef LOAD_SYMBOL
-#undef GENERIC_LOAD_FUNC_FINALE
-#undef GENERIC_FREE_FUNC
-#undef CUDA_LIBNAME
-#undef NVCUVID_LIBNAME
-#undef NVENC_LIBNAME
-#undef LIB_HANDLE
+#include <ffnvcodec/dynlink_loader.h>
#endif
-
OpenPOWER on IntegriCloud