mirror of
https://github.com/RYDE-WORK/pybind11.git
synced 2026-01-22 23:15:36 +08:00
1be0a0a610
For single-file extensions, a convenient pattern offered by cython
is to place the source files directly in the python source tree
(`foo/__init__.py`, `foo/ext.pyx`), deriving the package names from
their filesystem location. Adapt this pattern for pybind11, using an
`intree_extensions` helper, which should be thought of as the moral
equivalent to `cythonize`.
Differences with cythonize: I chose not to include globbing support
(`intree_extensions(glob.glob("**/*.cpp"))` seems sufficient), nor to
provide extension-customization kwargs (directly setting the attributes
on the resulting Pybind11Extension objects seems sufficient).
We could choose to have `intree_extension` (singular instead) and make
users write `[*map(intree_extension, glob.glob("**/*.cpp"))]`; no strong
opinion here.
Co-authored-by: Aaron Gokaslan <skylion.aaron@gmail.com>
65 lines
2.0 KiB
Python
65 lines
2.0 KiB
Python
# IMPORTANT: Should stay in sync with setup_helpers.py (mostly checked by CI /
|
|
# pre-commit).
|
|
|
|
from typing import Any, Callable, Dict, Iterator, List, Optional, Type, TypeVar, Union
|
|
from types import TracebackType
|
|
|
|
from distutils.command.build_ext import build_ext as _build_ext # type: ignore
|
|
from distutils.extension import Extension as _Extension
|
|
import distutils.ccompiler
|
|
import contextlib
|
|
|
|
WIN: bool
|
|
PY2: bool
|
|
MACOS: bool
|
|
STD_TMPL: str
|
|
|
|
class Pybind11Extension(_Extension):
|
|
def _add_cflags(self, *flags: str) -> None: ...
|
|
def _add_lflags(self, *flags: str) -> None: ...
|
|
def __init__(
|
|
self, *args: Any, cxx_std: int = 0, language: str = "c++", **kwargs: Any
|
|
) -> None: ...
|
|
@property
|
|
def cxx_std(self) -> int: ...
|
|
@cxx_std.setter
|
|
def cxx_std(self, level: int) -> None: ...
|
|
|
|
@contextlib.contextmanager
|
|
def tmp_chdir() -> Iterator[str]: ...
|
|
def has_flag(compiler: distutils.ccompiler.CCompiler, flag: str) -> bool: ...
|
|
def auto_cpp_level(compiler: distutils.ccompiler.CCompiler) -> Union[int, str]: ...
|
|
|
|
class build_ext(_build_ext): # type: ignore
|
|
def build_extensions(self) -> None: ...
|
|
|
|
def intree_extensions(
|
|
paths: Iterator[str], package_dir: Optional[Dict[str, str]] = None
|
|
) -> List[Pybind11Extension]: ...
|
|
def no_recompile(obj: str, src: str) -> bool: ...
|
|
def naive_recompile(obj: str, src: str) -> bool: ...
|
|
|
|
T = TypeVar("T", bound="ParallelCompile")
|
|
|
|
class ParallelCompile:
|
|
envvar: Optional[str]
|
|
default: int
|
|
max: int
|
|
needs_recompile: Callable[[str, str], bool]
|
|
def __init__(
|
|
self,
|
|
envvar: Optional[str] = None,
|
|
default: int = 0,
|
|
max: int = 0,
|
|
needs_recompile: Callable[[str, str], bool] = no_recompile,
|
|
) -> None: ...
|
|
def function(self) -> Any: ...
|
|
def install(self: T) -> T: ...
|
|
def __enter__(self: T) -> T: ...
|
|
def __exit__(
|
|
self,
|
|
exc_type: Optional[Type[BaseException]],
|
|
exc_value: Optional[BaseException],
|
|
traceback: Optional[TracebackType],
|
|
) -> None: ...
|