Compile nvidia headers

Author: WyattBlue

.github/workflows/build-ffmpeg.yml

@@ -63,7 +63,7 @@ jobs:
       - name: Build FFmpeg
         env:
           CIBW_ARCHS: ${{ matrix.arch }}
-          CIBW_BEFORE_BUILD: python scripts/build-ffmpeg.py /tmp/vendor --commercial
+          CIBW_BEFORE_BUILD: python scripts/build-ffmpeg.py /tmp/vendor --commercial --enable-cuda
           CIBW_BEFORE_BUILD_WINDOWS: python scripts\build-ffmpeg.py C:\cibw\vendor --commercial
           CIBW_BUILD: cp39-*
           CIBW_REPAIR_WHEEL_COMMAND_LINUX: LD_LIBRARY_PATH=/tmp/vendor/lib:$LD_LIBRARY_PATH auditwheel repair -w {dest_dir} {wheel}

scripts/build-ffmpeg.py

@@ -37,7 +37,7 @@
         source_url="https://download.gnome.org/sources/libxml2/2.9/libxml2-2.9.13.tar.xz",
         build_arguments=["--without-python"],
         community=True,
-    )
+    ),
 ]
 
 gnutls_group = [
@@ -187,7 +187,9 @@
         build_arguments=(
             [r"-DOPENSSL_ROOT_DIR=C:\Program Files\OpenSSL"]
             if plat == "Windows"
-            else ["-DENABLE_ENCRYPTION=OFF"] if plat == "Darwin" else [""]
+            else ["-DENABLE_ENCRYPTION=OFF"]
+            if plat == "Darwin"
+            else [""]
         ),
         community=True,
     ),
@@ -201,15 +203,21 @@
     build_system="meson",
 )
 
+nvheaders = Package(
+    name="nv-codec-headers",
+    source_url="https://github.com/FFmpeg/nv-codec-headers/archive/refs/tags/n13.0.19.0.tar.gz",
+    build_system="make",
+)
+
 ffmpeg_package = Package(
     name="ffmpeg",
     source_url="https://ffmpeg.org/releases/ffmpeg-7.1.tar.xz",
     build_arguments=[],
-    build_parallel= plat != "Windows",
+    build_parallel=plat != "Windows",
 )
 
 
-def download_tars(use_gnutls):
+def download_tars(use_gnutls: bool, community: bool) -> None:
     # Try to download all tars at the start.
     # If there is an curl error, do nothing, then try again in `main()`
 
@@ -218,6 +226,8 @@ def download_tars(use_gnutls):
         local_libs += gnutls_group
 
     for package in local_libs + codec_group:
+        if not community and package.community:
+            continue
         tarball = os.path.join(
             os.path.abspath("source"),
             package.source_filename or package.source_url.split("/")[-1],
@@ -236,6 +246,9 @@ def main():
     parser.add_argument("destination")
     parser.add_argument("--community", action="store_true")
     parser.add_argument("--commercial", action="store_true")
+    parser.add_argument(
+        "--enable-cuda", action="store_true", help="Enable NVIDIA CUDA support"
+    )
 
     args = parser.parse_args()
 
@@ -244,6 +257,7 @@ def main():
 
     dest_dir = args.destination
     community = args.community
+    enable_cuda = args.enable_cuda and plat == "Linux"
     del args
 
     output_dir = os.path.abspath("output")
@@ -261,7 +275,7 @@ def main():
     builder = Builder(dest_dir=dest_dir)
     builder.create_directories()
 
-    download_tars(use_gnutls)
+    download_tars(use_gnutls, community)
 
     # install packages
     available_tools = set()
@@ -329,6 +343,12 @@ def main():
         "--enable-zlib",
         "--enable-version3",
     ]
+
+    if enable_cuda:
+        ffmpeg_package.build_arguments.extend(["--enable-nvenc", "--enable-nvdec"])
+        # NVIDIA codecs require nonfree
+        ffmpeg_package.build_arguments.append("--enable-nonfree")
+
     if not community:
         ffmpeg_package.build_arguments.extend(
             ["--enable-libopenh264", "--disable-libx264"]
@@ -349,6 +369,8 @@ def main():
 
     if use_gnutls:
         library_group += gnutls_group
+    if enable_cuda:
+        library_group += [nvheaders]
 
     package_groups = [library_group + codec_group, [ffmpeg_package]]
     packages = [p for p_list in package_groups for p in p_list]

scripts/cibuildpkg.py

@@ -165,6 +165,8 @@ def build(self, package: Package, *, for_builder: bool = False):
                 self._build_with_cmake(package, for_builder=for_builder)
             elif package.build_system == "meson":
                 self._build_with_meson(package, for_builder=for_builder)
+            elif package.build_system == "make":
+                self._build_with_make(package, for_builder=for_builder)
             else:
                 self._build_with_autoconf(package, for_builder=for_builder)
 
@@ -196,6 +198,33 @@ def create_directories(self) -> None:
             separator=os.pathsep,
         )
 
+    def _build_with_make(self, package: Package, for_builder: bool) -> None:
+        assert package.build_system == "make"
+        package_path = os.path.join(self.build_dir, package.name)
+        package_source_path = os.path.join(package_path, package.source_dir)
+
+        # Get environment and prefix
+        env = self._environment(for_builder=for_builder)
+        prefix = self._prefix(for_builder=for_builder)
+
+        # Build package
+        with chdir(package_source_path):
+            make_command = ["make"] + make_args(parallel=package.build_parallel)
+            install_command = ["make", "install"]
+
+            # Add PREFIX to both make and install commands
+            prefix_arg = f"PREFIX={self._mangle_path(prefix)}"
+            make_command.append(prefix_arg)
+            install_command.append(prefix_arg)
+
+            # Add any additional build arguments
+            make_command.extend(package.build_arguments)
+            install_command.extend(package.build_arguments)
+
+            # Run build and install
+            run(make_command, env=env)
+            run(install_command, env=env)
+
     def _build_with_autoconf(self, package: Package, for_builder: bool) -> None:
         assert package.build_system == "autoconf"
         package_path = os.path.join(self.build_dir, package.name)
@@ -405,9 +434,9 @@ def _extract(self, package: Package) -> None:
                 prefixes = set()
                 for name in tar.getnames():
                     prefixes.add(name.split("/")[0])
-                assert (
-                    len(prefixes) == 1
-                ), "cannot strip path components, multiple prefixes found"
+                assert len(prefixes) == 1, (
+                    "cannot strip path components, multiple prefixes found"
+                )
                 prefix = list(prefixes)[0]
             else:
                 prefix = ""