Merge pull request #11 from stitchEm/linux_pkg_sys

Linux pkg sys

IO/src/rtmp/CMakeLists.txt

@@ -105,6 +105,23 @@ endif(RTMP_QUICKSYNC)
 
 
 if(RTMP_NVENC)
+
+  if(LINUX)
+    include(FetchContent)
+    FetchContent_Declare(
+      nvenc_api
+      GIT_REPOSITORY https://git.videolan.org/git/ffmpeg/nv-codec-headers.git
+      GIT_TAG n8.2.15.8
+    )
+
+
+    FetchContent_GetProperties(nvenc_api)
+    if(NOT nvenc_api_POPULATED)
+      FetchContent_Populate(nvenc_api)
+      include_directories(${nvenc_api_SOURCE_DIR}/include/ffnvcodec/)
+    endif()
+  endif()
+
   set(SOURCE_FILES
       ${SOURCE_FILES}
       src/cuvid.cpp
@@ -117,6 +134,8 @@ if(RTMP_NVENC)
       include/nvenc.hpp
       )
   find_package(CUDA REQUIRED)
+
+
 elseif(RTMP_NVENC_M)
   set(SOURCE_FILES
       ${SOURCE_FILES}

cmake/CUDA.cmake

@@ -38,7 +38,7 @@ if (LINUX)
   else(TEGRA_DEMO)
     find_package(GLEW)
     find_library(CUDA cuda PATHS "${LINUX_CUDA_PATH}/stubs")
-    find_library(CUDART cudart PATHS ${LINUX_CUDA_PATH} NO_DEFAULT_PATH)
+    find_library(CUDART cudart PATHS ${LINUX_CUDA_PATH})
     find_library(NVML nvidia-ml PATHS "${LINUX_CUDA_PATH}/stubs")
   endif(TEGRA_DEMO)
   include_directories(${CMAKE_EXTERNAL_DEPS}/include)