coriolis/builder.py

152 lines
6.6 KiB
Python
Raw Normal View History

2022-12-05 19:48:49 -06:00
import io
import glob
2022-12-22 04:57:53 -06:00
import ninja
2022-11-24 09:53:12 -06:00
import os
import platform
import re
import subprocess
import sys
2022-12-05 19:48:49 -06:00
2022-11-24 09:53:12 -06:00
from distutils.version import LooseVersion
2022-12-02 13:30:53 -06:00
from distutils.dir_util import copy_tree, remove_tree
2022-12-08 09:07:03 -06:00
from distutils.sysconfig import get_python_inc
2022-12-13 21:24:07 -06:00
from distutils import sysconfig
2022-11-24 09:53:12 -06:00
from typing import Any, Dict
from setuptools.command.build_ext import build_ext
from setuptools.extension import Extension
2022-12-13 21:24:07 -06:00
2022-11-24 09:53:12 -06:00
class CMakeExtension(Extension):
name: str # exists, even though IDE doesn't find it
def __init__(self, name: str, sourcedir: str="") -> None:
super().__init__(name, sources=[])
self.sourcedir = os.path.abspath(sourcedir)
self.sourcedir_rel = sourcedir
class ExtensionBuilder(build_ext):
def run(self) -> None:
self.validate_cmake()
super().run()
def build_extension(self, ext: Extension) -> None:
if isinstance(ext, CMakeExtension):
self.build_cmake_extension(ext)
else:
super().build_extension(ext)
def validate_cmake(self) -> None:
cmake_extensions = [x for x in self.extensions if isinstance(x, CMakeExtension)]
if len(cmake_extensions) > 0:
try:
out = subprocess.check_output(["cmake", "--version"])
except OSError:
raise RuntimeError(
2022-11-24 09:53:12 -06:00
"CMake must be installed to build the following extensions: "
+ ", ".join(e.name for e in cmake_extensions)
)
if platform.system() == "Windows":
cmake_version = LooseVersion(re.search(r"version\s*([\d.]+)", out.decode()).group(1)) # type: ignore
if cmake_version < "3.1.0":
raise RuntimeError("CMake >= 3.1.0 is required on Windows")
def build_cmake_extension(self, ext: CMakeExtension) -> None:
extdir = os.path.abspath(os.path.dirname(self.get_ext_fullpath(ext.name)))
cmake_args = []
2022-11-24 09:53:12 -06:00
2022-12-22 04:57:53 -06:00
ninja_executable_path = os.path.join(ninja.BIN_DIR, "ninja")
cmake_args += ["-GNinja","-DCMAKE_MAKE_PROGRAM:FILEPATH=" + ninja_executable_path]
2022-11-24 09:53:12 -06:00
cfg = "Debug" if self.debug else "Release"
# cfg = 'Debug'
build_args = ["--config", cfg]
install_args = ["--config", cfg]
2022-12-22 03:33:53 -06:00
cmake_args += ["-DCMAKE_BUILD_TYPE=" + cfg]
2022-11-24 09:53:12 -06:00
if platform.system() == "Windows":
if sys.maxsize > 2 ** 32:
cmake_args += ["-A", "x64"]
env = os.environ.copy()
env["CXXFLAGS"] = '{} -DVERSION_INFO=\\"{}\\"'.format(env.get("CXXFLAGS", ""), self.distribution.get_version())
2023-07-15 07:28:43 -05:00
2022-11-24 09:53:12 -06:00
build_dir = os.path.join(self.build_temp, ext.sourcedir_rel)
2022-12-02 13:42:41 -06:00
install_dir = os.path.join(extdir, 'Coriolis')
2022-11-24 09:53:12 -06:00
os.makedirs(build_dir,exist_ok=True)
2023-07-15 07:28:43 -05:00
if "USE_CCACHE" in env:
cmake_args += ["-DCMAKE_CXX_COMPILER_LAUNCHER=ccache"]
2022-11-24 09:53:12 -06:00
cmake_args += [f"-DCMAKE_MODULE_PATH={os.path.abspath('bootstrap/cmake_modules')};"
2022-12-02 13:42:41 -06:00
f"{os.path.join(install_dir, 'share/cmake/Modules')}"]
cmake_args += [f"-DCMAKE_INSTALL_PREFIX={install_dir}"]
cmake_args += [f"-DPython_CORIOLISLIB={install_dir}"]
cmake_args += [f"-DPython_CORIOLISARCH={install_dir}"]
cmake_args += [f"-DSYS_CONF_DIR={install_dir}"]
cmake_args += [f"-DCORIOLIS_TOP={install_dir}"]
cmake_args += [f"-DCORIOLIS_USER_TOP={install_dir}"]
cmake_args += [f"-DUSE_MANYLINUX=TRUE"]
2022-12-08 09:07:03 -06:00
2022-12-09 09:51:19 -06:00
cmake_args += [f"-DPython_EXECUTABLE={sys.executable}"]
2022-12-08 09:07:03 -06:00
cmake_args += ["-DPOETRY=1"]
cmake_args += ["-DWITH_QT5=1"]
cmake_args += ["-DCMAKE_BUILD_RPATH_USE_ORIGIN=1"]
cmake_args += ["-DCMAKE_SKIP_BUILD_RPATH=FALSE"]
cmake_args += ["-DCMAKE_BUILD_WITH_INSTALL_RPATH=FALSE"]
cmake_args += ["-DCMAKE_INSTALL_RPATH='$ORIGIN/lib:$ORIGIN'"]
cmake_args += ["-DCMAKE_INSTALL_RPATH_USE_LINK_PATH=TRUE"]
2022-11-24 09:53:12 -06:00
2022-12-13 21:24:07 -06:00
print(f"Using cmake args: {cmake_args}")
2022-12-22 05:08:14 -06:00
subprocess.check_call(["cmake", "--trace-redirect=build.cmake.trace", "--trace-expand", ext.sourcedir] + cmake_args, cwd=build_dir, env=env)
2022-11-24 09:53:12 -06:00
subprocess.check_call(["cmake", "--build", "."] + build_args, cwd=build_dir)
2022-12-02 13:42:41 -06:00
subprocess.check_call(["cmake", "--install", ".", "--prefix", install_dir] + install_args, cwd=build_dir)
if os.path.exists(os.path.join(install_dir, "bin")):
copy_tree(os.path.join(install_dir, "bin"), os.path.join(install_dir,"data/bin"))
remove_tree(os.path.join(install_dir, "bin"))
2022-12-05 19:48:49 -06:00
proc = subprocess.Popen(['file'] + glob.glob(os.path.join(install_dir,"data/bin/*")), stdout=subprocess.PIPE)
for line in io.TextIOWrapper(proc.stdout, encoding="utf-8"):
if 'ELF' in line: #TODO support other OSs
f = line.split(':')[0]
print(f"fixing up {f}")
subprocess.check_call(["patchelf", "--set-rpath", '$ORIGIN/../../lib', f])
2022-11-24 09:53:12 -06:00
2023-01-09 11:29:43 -06:00
# Remove docs
if os.path.exists(os.path.join(install_dir, "share", "doc")):
remove_tree(os.path.join(install_dir, "share", "doc"))
2022-11-24 09:53:12 -06:00
def build(setup_kwargs: Dict[str, Any]) -> None:
cmake_modules = [
CMakeExtension("coloquinte", sourcedir="coloquinte"),
CMakeExtension("Hurricane", sourcedir="hurricane"),
CMakeExtension("crlcore", sourcedir="crlcore"),
CMakeExtension("flute", sourcedir="flute"),
CMakeExtension("etesian", sourcedir="etesian"),
CMakeExtension("anabatic", sourcedir="anabatic"),
CMakeExtension("katana", sourcedir="katana"),
CMakeExtension("equinox", sourcedir="equinox"),
CMakeExtension("solstice", sourcedir="solstice"),
CMakeExtension("oroshi", sourcedir="oroshi"),
CMakeExtension("bora", sourcedir="bora"),
CMakeExtension("karakaze", sourcedir="karakaze"),
#CMakeExtension("knik", sourcedir="knik"),
#CMakeExtension("unicorn", sourcedir="unicorn"),
CMakeExtension("tutorial", sourcedir="tutorial"),
CMakeExtension("cumulus", sourcedir="cumulus"),
CMakeExtension("stratus1", sourcedir="stratus1"),
CMakeExtension("documentation", sourcedir="documentation"),
CMakeExtension("unittests", sourcedir="unittests")
]
2022-11-24 09:53:12 -06:00
ext_modules = cmake_modules
setup_kwargs.update(
{
"ext_modules": ext_modules,
"cmdclass": dict(build_ext=ExtensionBuilder),
"zip_safe": False,
}
)