78 lines
2.1 KiB
Python
78 lines
2.1 KiB
Python
import os
|
|
import shutil
|
|
import sys
|
|
from distutils.core import Distribution, Extension
|
|
from distutils.command import build_ext
|
|
|
|
from Cython.Build import cythonize
|
|
|
|
|
|
def build() -> None:
|
|
compile_args = []
|
|
|
|
if sys.platform == "linux":
|
|
compile_args += [
|
|
"-march=native",
|
|
"-O3",
|
|
"-msse",
|
|
"-msse2",
|
|
"-mfma",
|
|
"-mfpmath=sse",
|
|
"-Wno-pointer-sign",
|
|
"-Wno-unused-variable",
|
|
]
|
|
|
|
extensions = [
|
|
Extension(
|
|
"pyModeS.c_common",
|
|
["pyModeS/c_common.pyx"],
|
|
extra_compile_args=compile_args,
|
|
),
|
|
Extension(
|
|
"pyModeS.decoder.flarm.decode",
|
|
[
|
|
"pyModeS/decoder/flarm/decode.pyx",
|
|
"pyModeS/decoder/flarm/core.c",
|
|
],
|
|
extra_compile_args=compile_args,
|
|
include_dirs=["pyModeS/decoder/flarm"],
|
|
),
|
|
# Extension(
|
|
# "pyModeS.extra.demod2400.core",
|
|
# [
|
|
# "pyModeS/extra/demod2400/core.pyx",
|
|
# "pyModeS/extra/demod2400/demod2400.c",
|
|
# ],
|
|
# extra_compile_args=compile_args,
|
|
# include_dirs=["pyModeS/extra/demod2400"],
|
|
# libraries=["m"],
|
|
# ),
|
|
]
|
|
|
|
ext_modules = cythonize(
|
|
extensions,
|
|
compiler_directives={"binding": True, "language_level": 3},
|
|
)
|
|
|
|
distribution = Distribution(
|
|
{"name": "extended", "ext_modules": ext_modules}
|
|
)
|
|
distribution.package_dir = "extended" # type: ignore
|
|
|
|
cmd = build_ext.build_ext(distribution)
|
|
cmd.verbose = True # type: ignore
|
|
cmd.ensure_finalized() # type: ignore
|
|
cmd.run()
|
|
|
|
# Copy built extensions back to the project
|
|
for output in cmd.get_outputs():
|
|
relative_extension = os.path.relpath(output, cmd.build_lib)
|
|
shutil.copyfile(output, relative_extension)
|
|
mode = os.stat(relative_extension).st_mode
|
|
mode |= (mode & 0o444) >> 2
|
|
os.chmod(relative_extension, mode)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
build()
|