Add support for CUDA FFI.

Author: hgaiser

build.rs

@@ -645,7 +645,7 @@ fn main() {
     let statik = env::var("CARGO_FEATURE_STATIC").is_ok();
     let ffmpeg_major_version: u32 = env!("CARGO_PKG_VERSION_MAJOR").parse().unwrap();
 
-    let include_paths: Vec<PathBuf> = if env::var("CARGO_FEATURE_BUILD").is_ok() {
+    let mut include_paths: Vec<PathBuf> = if env::var("CARGO_FEATURE_BUILD").is_ok() {
         println!(
             "cargo:rustc-link-search=native={}",
             search().join("lib").to_string_lossy()
@@ -742,6 +742,14 @@ fn main() {
             .include_paths
     };
 
+    include_paths.extend(
+        pkg_config::Config::new()
+            .statik(statik)
+            .probe("cuda")
+            .unwrap()
+            .include_paths
+    );
+
     if statik && cfg!(target_os = "macos") {
         let frameworks = vec![
             "AppKit",
@@ -1236,6 +1244,7 @@ fn main() {
         .header(search_include(&include_paths, "libavutil/hash.h"))
         .header(search_include(&include_paths, "libavutil/hmac.h"))
         .header(search_include(&include_paths, "libavutil/hwcontext.h"))
+        .header(search_include(&include_paths, "libavutil/hwcontext_cuda.h"))
         .header(search_include(&include_paths, "libavutil/imgutils.h"))
         .header(search_include(&include_paths, "libavutil/lfg.h"))
         .header(search_include(&include_paths, "libavutil/log.h"))