-
Notifications
You must be signed in to change notification settings - Fork 5
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Switch to poetry and pyproject.toml (#61)
- Loading branch information
1 parent
bfb3362
commit 84c66fa
Showing
13 changed files
with
8,362 additions
and
3,966 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,109 @@ | ||
|
|
||
| import os | ||
| import sys | ||
| from pathlib import Path | ||
| import shutil | ||
| import subprocess | ||
| from typing import List, Any | ||
|
|
||
|
|
||
| PACKAGE_DIR = Path(__file__).parent / "openjpeg" | ||
| OPENJPEG_SRC = PACKAGE_DIR / "src" / "openjpeg" / "src" / "lib" / "openjp2" | ||
| INTERFACE_SRC = PACKAGE_DIR / "src" / "interface" | ||
|
|
||
|
|
||
| def build(setup_kwargs: Any) -> None: | ||
| from setuptools import Extension | ||
| from setuptools.dist import Distribution | ||
| import Cython.Compiler.Options | ||
| from Cython.Build import build_ext, cythonize | ||
| import numpy | ||
|
|
||
| setup_oj() | ||
|
|
||
| ext = Extension( | ||
| "_openjpeg", | ||
| [os.fspath(p) for p in get_source_files()], | ||
| language="c", | ||
| include_dirs=[ | ||
| os.fspath(OPENJPEG_SRC), | ||
| os.fspath(INTERFACE_SRC), | ||
| numpy.get_include(), | ||
| # distutils.sysconfig.get_python_inc(), | ||
| ], | ||
| extra_compile_args=[], | ||
| extra_link_args=[], | ||
| ) | ||
|
|
||
| ext_modules = cythonize( | ||
| [ext], | ||
| include_path=ext.include_dirs, | ||
| language_level=3, | ||
| ) | ||
|
|
||
| dist = Distribution({"ext_modules": ext_modules}) | ||
| cmd = build_ext(dist) | ||
| cmd.ensure_finalized() | ||
| cmd.run() | ||
|
|
||
| for output in cmd.get_outputs(): | ||
| output = Path(output) | ||
| relative_ext = output.relative_to(cmd.build_lib) | ||
| shutil.copyfile(output, relative_ext) | ||
|
|
||
| return setup_kwargs | ||
|
|
||
|
|
||
| def get_source_files() -> List[Path]: | ||
| """Return a list of paths to the source files to be compiled.""" | ||
| source_files = [ | ||
| INTERFACE_SRC / "decode.c", | ||
| INTERFACE_SRC / "color.c", | ||
| ] | ||
| for fname in OPENJPEG_SRC.glob("*"): | ||
| if fname.parts[-1].startswith("test"): | ||
| continue | ||
|
|
||
| if fname.parts[-1].startswith("bench"): | ||
| continue | ||
|
|
||
| if fname.suffix == ".c": | ||
| source_files.append(fname) | ||
|
|
||
| source_files = [p.relative_to(PACKAGE_DIR.parent) for p in source_files] | ||
| source_files.insert(0, Path("openjpeg/_openjpeg.pyx")) | ||
|
|
||
| return source_files | ||
|
|
||
|
|
||
| def setup_oj() -> None: | ||
| """Run custom cmake.""" | ||
| base_dir = os.path.join("openjpeg", "src", "openjpeg") | ||
|
|
||
| # Copy custom CMakeLists.txt file to openjpeg base dir | ||
| shutil.copy( | ||
| os.path.join("build_tools", "cmake", "CMakeLists.txt"), | ||
| base_dir | ||
| ) | ||
| build_dir = os.path.join(base_dir, "build") | ||
| if os.path.exists(build_dir): | ||
| shutil.rmtree(build_dir) | ||
|
|
||
| try: | ||
| os.remove(INTERFACE_SRC / "opj_config.h") | ||
| os.remove(INTERFACE_SRC / "opj_config_private.h") | ||
| except: | ||
| pass | ||
|
|
||
| os.mkdir(build_dir) | ||
| fpath = os.path.abspath(base_dir) | ||
| cur_dir = os.getcwd() | ||
| os.chdir(build_dir) | ||
| subprocess.call(['cmake', fpath]) | ||
| os.chdir(cur_dir) | ||
|
|
||
| # Turn off JPIP | ||
| if os.path.exists(INTERFACE_SRC / "opj_config.h"): | ||
| with open(INTERFACE_SRC / "opj_config.h", "a") as f: | ||
| f.write("\n") | ||
| f.write("#define USE_JPIP 0") |
Oops, something went wrong.