Collaboration diagram for wheel_target_hook.WheelBuildHook:

Public Member Functions
	initialize (self, version, build_data)

	read_env (self)

	prepare_binaries (self)

	get_libs_dir (self)

	get_runtime_build_dir (self)

	copy_libraries (self, src_dir, target_dir, subdir=None)

	recreate_dir (self, dir_path)

	create_build_tag (self)

Data Fields
	DEFAULT_PRODUCT_DIR

	product_dir

	platform

	glibc_version

	whl_binaries_target_dir

Protected Member Functions
	_read_env (self, env_var_name)

	_validate_platform (self, value)

	_validate_product_dir (self, value)

	_validate_glibc_version (self, value)

Protected Attributes
	_validate_platform

	_validate_glibc_version

	_validate_product_dir

Detailed Description

This file contains the code used by the runtime/pyproject.toml config file. It's executed when the user attempts to build
a python wheel for the ONERT API. The code in this file is executed before the target is built.

Definition at line 9 of file wheel_target_hook.py.

Member Function Documentation

◆ _read_env()

wheel_target_hook.WheelBuildHook._read_env	(	self,
		env_var_name
	)

protected

Definition at line 136 of file wheel_target_hook.py.

    def _read_env(self, env_var_name):
        validators = {
            "PLATFORM": self._validate_platform,
            "GLIBC_VERSION": self._validate_glibc_version,
            "PRODUCT_DIR": self._validate_product_dir,
        }
 
        value = os.environ.get(env_var_name, None)
        validate = validators.get(env_var_name)
        if validate is not None:
            return validate(value)
        else:
            return value
 

Referenced by wheel_target_hook.WheelBuildHook.read_env().

◆ _validate_glibc_version()

wheel_target_hook.WheelBuildHook._validate_glibc_version	(	self,
		value
	)

protected

Definition at line 173 of file wheel_target_hook.py.

    def _validate_glibc_version(self, value):
        if value is None:
            return value
 
        # accept the X.Y format
        m = re.search(r"^[0-9]+\.[0-9]+$", value)
        if m is not None:
            return value.replace(".", "_")
 
        # or look for X_Y if the previous test failed
        m = re.search(r"^[0-9]+_[0-9]+$", value)
        if m is None:
            raise ValueError(
                "Incorrect version of glibc specified. Please use the following format: X_Y or X.Y"
            )
        else:
            return value

◆ _validate_platform()

wheel_target_hook.WheelBuildHook._validate_platform	(	self,
		value
	)

protected

Definition at line 150 of file wheel_target_hook.py.

    def _validate_platform(self, value):
        supported_platforms = ["x86_64", "armv7l", "aarch64"]
        if value is None:
            print(
                f" |> No platform specified. Creating a wheel for '{supported_platforms[0]}'"
            )
            return supported_platforms[0]
        elif value not in supported_platforms:
            raise ValueError(f"""Unsupported platform detected: {value}.
                Please use one of the following values: {','.join(supported_platforms)}"""
                             )
        else:
            return value
 

◆ _validate_product_dir()

wheel_target_hook.WheelBuildHook._validate_product_dir	(	self,
		value
	)

protected

Definition at line 164 of file wheel_target_hook.py.

    def _validate_product_dir(self, value):
        if value is None:
            value = self.DEFAULT_PRODUCT_DIR
 
        if not os.path.exists(value):
            raise FileNotFoundError(f"The path with the build does not exist: '{value}'")
        else:
            return value
 

References wheel_target_hook.WheelBuildHook.DEFAULT_PRODUCT_DIR.

◆ copy_libraries()

wheel_target_hook.WheelBuildHook.copy_libraries	(	self,
		src_dir,
		target_dir,
		subdir = `None`
	)

Copy all .so files found in src_dir to the target_dir
If subdir is provided copy from src_dir/subdir to target_dir/subdir

Definition at line 93 of file wheel_target_hook.py.

    def copy_libraries(self, src_dir, target_dir, subdir=None):
        """
        Copy all .so files found in src_dir to the target_dir
        If subdir is provided copy from src_dir/subdir to target_dir/subdir
        """
        if subdir is not None:
            src_dir = os.path.join(src_dir, subdir)
            target_dir = os.path.join(target_dir, subdir)
 
        os.makedirs(target_dir, exist_ok=True)
 
        for file in filter(lambda file: file.endswith(".so"), os.listdir(src_dir)):
            src_path = os.path.join(src_dir, file)
            tgt_path = os.path.join(target_dir, file)
            shutil.copy(src_path, tgt_path)
            print(f" |> Copied {src_path} to {tgt_path}")
 

Referenced by wheel_target_hook.WheelBuildHook.prepare_binaries().

◆ create_build_tag()

wheel_target_hook.WheelBuildHook.create_build_tag ( self )

Create the most appropriate build tag that will be used to name the wheel

Definition at line 117 of file wheel_target_hook.py.

    def create_build_tag(self):
        """Create the most appropriate build tag that will be used to name the wheel"""
 
        # first get the tag using the usual way build backends do it
        tag = next(sys_tags())
 
        # now create the part of the build tag that will be overridden
        # use 'manylinux' + glibc version (if provided) + the platform string
        tag_platform = "manylinux"
        if self.glibc_version is not None:
            tag_platform = f"{tag_platform}_{self.glibc_version}_{self.platform}"
        else:
            tag_platform = f"{tag_platform}_{self.platform}"
 
        # compose the final tag and override the tag.platform part with our own string
        build_tag = f"{tag.interpreter}-{tag.abi}-{tag_platform}"
        print(f" |> Created build_tag: {build_tag}")
        return build_tag
 

References wheel_target_hook.WheelBuildHook.glibc_version.

◆ get_libs_dir()

wheel_target_hook.WheelBuildHook.get_libs_dir ( self )

Retrieve the path of a directory where the required shared libraries are

Definition at line 65 of file wheel_target_hook.py.

    def get_libs_dir(self):
        """Retrieve the path of a directory where the required shared libraries are"""
        runtime_build_dir = self.get_runtime_build_dir()
        if not os.path.exists(runtime_build_dir):
            raise FileExistsError(
                f"The expected runtime build dir does not exist: {runtime_build_dir}")
 
        print(f" |> runtime_build_dir={runtime_build_dir}")
 
        possible_lib_dirs = ["lib64", "lib32", "lib"]
        for lib_dir in possible_lib_dirs:
            libs_dir_path = os.path.join(runtime_build_dir, lib_dir)
            if os.path.exists(libs_dir_path):
                return libs_dir_path
 
        raise FileNotFoundError(f"No lib directory found in {runtime_build_dir}")
 

References wheel_target_hook.WheelBuildHook.get_runtime_build_dir().

Referenced by wheel_target_hook.WheelBuildHook.prepare_binaries().

◆ get_runtime_build_dir()

wheel_target_hook.WheelBuildHook.get_runtime_build_dir ( self )

Retrieve the path of a directory where the runtime's binaries are (the build tree's root)

Definition at line 82 of file wheel_target_hook.py.

    def get_runtime_build_dir(self):
        """Retrieve the path of a directory where the runtime's binaries are (the build tree's root)"""
 
        if self.product_dir != self.DEFAULT_PRODUCT_DIR:
            # In case the product directory was passed as an environment variable use this path
            # as a custom root directory of the build tree
            return self.product_dir
        else:
            # TODO - add the debug build support (via env variables)
            return os.path.join(self.product_dir, f"{self.platform}-linux.release/out")
 

References wheel_target_hook.WheelBuildHook.DEFAULT_PRODUCT_DIR, and wheel_target_hook.WheelBuildHook.product_dir.

Referenced by wheel_target_hook.WheelBuildHook.get_libs_dir().

◆ initialize()

wheel_target_hook.WheelBuildHook.initialize	(	self,
		version,
		build_data
	)

Definition at line 14 of file wheel_target_hook.py.

    def initialize(self, version, build_data):
        super().initialize(version, build_data)
 
        THIS_FILE_DIR = os.path.dirname(os.path.abspath(__file__))
        self.DEFAULT_PRODUCT_DIR = os.path.normpath(
            os.path.join(THIS_FILE_DIR, "../../../Product"))
 
        self.product_dir = None
        self.platform = None
        self.glibc_version = None
 
        # read the environment variables that can be used to override some build settings
        self.read_env()
 
        # a temporary build dir used by the wheel build system
        tmp_build_dir = "_build_"
        self.recreate_dir(tmp_build_dir)
 
        # this is the path where the native libraries are expected to land in the wheel
        # the files copied to this location will eventually be added to the wheel by the build system
        # the whole structure of subdirectories will be reflected in the final wheel
        self.whl_binaries_target_dir = os.path.join(tmp_build_dir, "native")
 
        # gather the required binaries in the temporary build directory
        self.prepare_binaries()
 
        # include all contents of the build directory in the 'onert' subdirectory in the wheel
        # at this point the temporary build directory should be populated with the required files
        build_data["force_include"][tmp_build_dir] = "onert"
 
        build_data["pure_python"] = False
        build_data["infer_tag"] = False
        build_data["tag"] = self.create_build_tag()
 

References wheel_target_hook.WheelBuildHook.initialize().

Referenced by wheel_target_hook.WheelBuildHook.initialize().

◆ prepare_binaries()

wheel_target_hook.WheelBuildHook.prepare_binaries ( self )

Definition at line 55 of file wheel_target_hook.py.

    def prepare_binaries(self):
        # the main directory in the runtime's build tree containing the .so files
        # those files need to be copied to whl_binaries_target_dir before they are added to the wheel
        src_libs_base_dir = self.get_libs_dir()
 
        self.copy_libraries(src_libs_base_dir, self.whl_binaries_target_dir)
        self.copy_libraries(src_libs_base_dir, self.whl_binaries_target_dir, "nnfw")
        self.copy_libraries(src_libs_base_dir, self.whl_binaries_target_dir,
                            "nnfw/backend")
 

References wheel_target_hook.WheelBuildHook.copy_libraries(), wheel_target_hook.WheelBuildHook.get_libs_dir(), and wheel_target_hook.WheelBuildHook.whl_binaries_target_dir.

◆ read_env()

wheel_target_hook.WheelBuildHook.read_env ( self )

Read the relevant environment variables or use the defaults

Definition at line 48 of file wheel_target_hook.py.

    def read_env(self):
        """Read the relevant environment variables or use the defaults"""
 
        self.product_dir = self._read_env("PRODUCT_DIR")
        self.platform = self._read_env("PLATFORM")
        self.glibc_version = self._read_env("GLIBC_VERSION")
 

References wheel_target_hook.WheelBuildHook._read_env(), wheel_target_hook.WheelBuildHook.glibc_version, wheel_target_hook.WheelBuildHook.platform, and wheel_target_hook.WheelBuildHook.product_dir.

◆ recreate_dir()

wheel_target_hook.WheelBuildHook.recreate_dir	(	self,
		dir_path
	)

Delete a directory (if it exists) and create it again but empty

Definition at line 110 of file wheel_target_hook.py.

    def recreate_dir(self, dir_path):
        """Delete a directory (if it exists) and create it again but empty"""
        if os.path.exists(dir_path):
            print(f" |> Deleting existing directory '{dir_path}'...")
            shutil.rmtree(dir_path)
        os.makedirs(dir_path)
 

Field Documentation

◆ _validate_glibc_version

wheel_target_hook.WheelBuildHook._validate_glibc_version

protected

Definition at line 139 of file wheel_target_hook.py.

◆ _validate_platform

wheel_target_hook.WheelBuildHook._validate_platform

protected

Definition at line 138 of file wheel_target_hook.py.

◆ _validate_product_dir

wheel_target_hook.WheelBuildHook._validate_product_dir

protected

Definition at line 140 of file wheel_target_hook.py.

◆ DEFAULT_PRODUCT_DIR

wheel_target_hook.WheelBuildHook.DEFAULT_PRODUCT_DIR

Definition at line 18 of file wheel_target_hook.py.

Referenced by wheel_target_hook.WheelBuildHook._validate_product_dir(), and wheel_target_hook.WheelBuildHook.get_runtime_build_dir().

◆ glibc_version

wheel_target_hook.WheelBuildHook.glibc_version

Definition at line 23 of file wheel_target_hook.py.

Referenced by wheel_target_hook.WheelBuildHook.create_build_tag(), and wheel_target_hook.WheelBuildHook.read_env().

◆ platform

wheel_target_hook.WheelBuildHook.platform

Definition at line 22 of file wheel_target_hook.py.

Referenced by wheel_target_hook.WheelBuildHook.read_env().

◆ product_dir

wheel_target_hook.WheelBuildHook.product_dir

Definition at line 21 of file wheel_target_hook.py.

Referenced by wheel_target_hook.WheelBuildHook.get_runtime_build_dir(), and wheel_target_hook.WheelBuildHook.read_env().

◆ whl_binaries_target_dir

wheel_target_hook.WheelBuildHook.whl_binaries_target_dir

Definition at line 35 of file wheel_target_hook.py.

Referenced by wheel_target_hook.WheelBuildHook.prepare_binaries().

The documentation for this class was generated from the following file:

runtime/infra/python/wheel_target_hook.py

Public Member Functions

Data Fields

Protected Member Functions

Protected Attributes

Detailed Description

Member Function Documentation

◆ _read_env()

◆ _validate_glibc_version()

◆ _validate_platform()

◆ _validate_product_dir()

◆ copy_libraries()

◆ create_build_tag()

◆ get_libs_dir()

◆ get_runtime_build_dir()

◆ initialize()

◆ prepare_binaries()

◆ read_env()

◆ recreate_dir()

Field Documentation

◆ _validate_glibc_version

◆ _validate_platform

◆ _validate_product_dir

◆ DEFAULT_PRODUCT_DIR

◆ glibc_version

◆ platform

◆ product_dir

◆ whl_binaries_target_dir