lbry-android-sdk/p4a/pythonforandroid/recipes/lapack/__init__.py
2022-12-02 15:15:34 -05:00

79 lines
3 KiB
Python

'''
known to build with cmake version 3.23.2 and NDK r21e.
See https://gitlab.kitware.com/cmake/cmake/-/issues/18739
'''
from pythonforandroid.recipe import Recipe
from pythonforandroid.logger import shprint
from pythonforandroid.util import current_directory, ensure_dir, BuildInterruptingException
from multiprocessing import cpu_count
from os.path import join
import sh
import shutil
from os import environ
from pythonforandroid.util import build_platform
arch_to_sysroot = {'armeabi': 'arm', 'armeabi-v7a': 'arm', 'arm64-v8a': 'arm64'}
def arch_to_toolchain(arch):
if 'arm' in arch.arch:
return arch.command_prefix
return arch.arch
class LapackRecipe(Recipe):
name = 'lapack'
version = 'v3.10.1'
url = 'https://github.com/Reference-LAPACK/lapack/archive/{version}.tar.gz'
libdir = 'build/install/lib'
built_libraries = {'libblas.so': libdir, 'liblapack.so': libdir, 'libcblas.so': libdir}
def get_recipe_env(self, arch):
env = super().get_recipe_env(arch)
ndk_dir = environ.get("LEGACY_NDK")
if ndk_dir is None:
raise BuildInterruptingException("Please set the environment variable 'LEGACY_NDK' to point to a NDK location with gcc/gfortran support (supported NDK version: 'r21e')")
GCC_VER = '4.9'
HOST = build_platform
sysroot_suffix = arch_to_sysroot.get(arch.arch, arch.arch)
sysroot = f"{ndk_dir}/platforms/{env['NDK_API']}/arch-{sysroot_suffix}"
FC = f"{ndk_dir}/toolchains/{arch_to_toolchain(arch)}-{GCC_VER}/prebuilt/{HOST}/bin/{arch.command_prefix}-gfortran"
env['FC'] = f'{FC} --sysroot={sysroot}'
if shutil.which(FC) is None:
raise BuildInterruptingException(f"{FC} not found. See https://github.com/mzakharo/android-gfortran")
return env
def build_arch(self, arch):
source_dir = self.get_build_dir(arch.arch)
build_target = join(source_dir, 'build')
install_target = join(build_target, 'install')
ensure_dir(build_target)
with current_directory(build_target):
env = self.get_recipe_env(arch)
ndk_dir = environ["LEGACY_NDK"]
shprint(sh.rm, '-rf', 'CMakeFiles/', 'CMakeCache.txt', _env=env)
opts = [
'-DCMAKE_SYSTEM_NAME=Android',
'-DCMAKE_POSITION_INDEPENDENT_CODE=1',
'-DCMAKE_ANDROID_ARCH_ABI={arch}'.format(arch=arch.arch),
'-DCMAKE_ANDROID_NDK=' + ndk_dir,
'-DCMAKE_ANDROID_API={api}'.format(api=self.ctx.ndk_api),
'-DCMAKE_BUILD_TYPE=Release',
'-DCMAKE_INSTALL_PREFIX={}'.format(install_target),
'-DCBLAS=ON',
'-DBUILD_SHARED_LIBS=ON',
]
if arch.arch == 'armeabi-v7a':
opts.append('-DCMAKE_ANDROID_ARM_NEON=ON')
shprint(sh.cmake, source_dir, *opts, _env=env)
shprint(sh.make, '-j' + str(cpu_count()), _env=env)
shprint(sh.make, 'install', _env=env)
recipe = LapackRecipe()