-
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.
- Loading branch information
1 parent
7ac26f0
commit 68ff710
Showing
12 changed files
with
8,414 additions
and
3,790 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.