#!/usr/bin/env python # Copyright 2024-2025 Gentoo Authors # Distributed under the terms of the GNU General Public License v2 # # # Takes the manifests/*.xml file from the NVIDIA CUDA Toolkit as input. # Those files contain a nested tree of package items. # # For each package node a if block is output that checks the package's # name attribute against the environment provided array SKIP_COMPONENTS # and skips that whole package if it's listed in it. # # Each item contains four nodes that reference things to install, # `dir`, `file`, `desktopFile`, & `pcfile`, and metadata that further # detail these. This script will output calls to do* calls that # use the metadata. # - dodir would create an empty file - we handle this in dofile # - dofile copies a regex file glob # - dodesktopFile creates a .desktop file # - dopcfile creates a pkgconfig file # # The resulting bash code can be run inside src_install(). # # Usage: python parse_manifest.py import argparse import xml.etree.ElementTree import defusedxml.ElementTree from pathlib import Path ind = "\t" indent = 1 def main(): parser = argparse.ArgumentParser() parser.add_argument('filename') # positional argument args = parser.parse_args() basedir = Path(args.filename).parents[1] et = defusedxml.ElementTree.parse(args.filename) # Find all packages under the package with the id main (i.e. under "CUDA Installer") for e in et.findall("[@id='main']/package"): def p_package(el: xml.etree.ElementTree.Element, level: int = 0): skip = { "Documentation", # obsolete "Driver", # unused # "Kernel Objects", # split # "Demo Suite", # "Visual Tools", # old eclipse # "nsight", # old java # "nvvp", # "cuda-gdb-src" } name = el.get("name") # trim leading CUDA and trailing version if name.startswith("CUDA"): name2 = ' '.join(name.split(" ")[1:-1]) else: name2 = name if name2 in skip: return # avoid having to deal with whitespaces in bash name2 = name2.replace(" ", "_") path = "" print(f"{ind * (level + 0) * indent}if ! has {name2} \"${{SKIP_COMPONENTS[@]}}\"; then # \"{name}\"") # output attributes from unhandled tags for child in el: if child.tag == "package": continue if child.tag == "file": continue if child.tag == "desktopFile": continue if child.tag == "pcfile": continue for attrib in child.attrib: print(f"{child.tag} {attrib}={child.attrib[attrib]}") # CUDA Installer # only for CUDA Installer # unused for node in el.findall("./name"): print(f"{ind * (level + 1) * indent}# {node.tag}: \"{node.text}\"") # compiler # category. We use the package name instead. # unused # for node in el.findall("./type"): # print(f"{ind * (level + 1) * indent}# {node.tag}: \"{node.text}\"") # 1 # probably sorting for the tui installer # unused # for node in el.findall("./priority"): # print(f"{ind * (level + 1) * indent}# {node.tag}: \"{node.text}\"") # # proably for tui installer # unused # for node in el.findall("./single-selection"): # print(f"{ind * (level + 1) * indent}# {node.tag}: \"{node.text}\"") # 2.24.2 # version of the installed kernel object (Kernel Objects only) # unused for node in el.findall("./koversion"): print(f"{ind * (level + 1) * indent}# {node.tag}: \"{node.text}\"") # /usr/local/cuda-12.8 # overrides the install location # unused # for node in el.findall("./installPath"): # print(f"{ind * (level + 1) * indent}# {node.tag}: \"{node.text}\"") # ./builds/cuda_cccl/ # path where the package files are found # we cd into it for node in el.findall("./buildPath"): path = node.text.removeprefix('./') print(f"{ind * (level + 1) * indent}cd \"${{S}}/{path}\" || die \"cd ${{S}}/{path} failed\"") print() # bin # would install empty dirs # unused # for node in el.findall("./dir"): # pass # .* # targets/x86_64-linux/lib/.*\.so # regex glob of files to install. for node in el.findall("./file"): # unescape '.*' -> '*' & '\.' -> '.' file = (node.text .replace(".*", "*") .replace(r"\.", ".") .replace("x86_64", "${narch}") .replace("sbsa", "${narch}") ) # optional dir offset, we merge it into path dir = "" if "dir" in node.attrib: dir = f" \"{Path(node.attrib["dir"])}\"" filepath = basedir / path / file # ignore existing symlinks ( include, lib* ) and the uninstallers if not filepath.is_symlink() and not file.endswith("-uninstaller"): print(f"{ind * (level + 1) * indent}dofile \"{file}\"{dir}") # opencl-12.8.pc # create a pkgconfig file for the given description and lib name/version for node in el.findall("./pcfile"): offset = node.text.rfind('-') if offset == -1: raise RuntimeError(f"failed to split pcfile {node.text}") lib_name = node.text[:offset] if not node.text.endswith('.pc'): raise RuntimeError(f"pcfile does not end in '.pc' {node.text}") lib_version = node.text[offset+1:-3] if "description" not in node.attrib: raise RuntimeError(f"no description for {node.text}") subdir = "" if "subdir" in node.attrib: subdir = f" \"{node.attrib["subdir"]}\"" print(f"{ind * (level + 1) * indent}dopcfile " f"\"{lib_name}\" " f"\"{lib_version}\" " f"\"{node.attrib["description"]}\"{subdir}") # # create a .desktop file for node in el.findall("./desktopFile"): print(f"{ind * (level + 1) * indent}dodesktopFile \\") print(f"{ind * (level + 2) * indent}\"{node.attrib["filename"]}\" \\") print(f"{ind * (level + 2) * indent}\"{node.attrib["name"]}\" \\") print(f"{ind * (level + 2) * indent}\"{node.attrib["categories"]}\" \\") print(f"{ind * (level + 2) * indent}\"{node.attrib["keywords"]}\" \\") print(f"{ind * (level + 2) * indent}\"{node.attrib["iconPath"]}\" \\") print(f"{ind * (level + 2) * indent}\"{node.attrib["execPath"]}\" \\") print(f"{ind * (level + 2) * indent}\"{node.attrib["tryExecPath"]}\"") # iterator over all nested packages for node in el.findall("./package"): p_package(node, level + 1) print(f"{ind * (level + 0) * indent}fi") p_package(e) if __name__ == "__main__": main()