1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
|
"""Build script for mypyc C runtime library and C API unit tests.
The tests are written in C++ and use the Google Test framework.
"""
from __future__ import annotations
import os
import subprocess
import sys
from distutils import ccompiler, sysconfig
from typing import Any
# we'll import stuff from the source tree, let's ensure is on the sys path
sys.path.insert(0, os.path.dirname(os.path.realpath(__file__)))
import build_setup # noqa: F401
from setuptools import Extension, setup
from setuptools.command.build_ext import build_ext
C_APIS_TO_TEST = [
"init.c",
"int_ops.c",
"float_ops.c",
"list_ops.c",
"exc_ops.c",
"generic_ops.c",
"pythonsupport.c",
]
class BuildExtGtest(build_ext):
def get_library_names(self) -> list[str]:
return ["gtest"]
def run(self) -> None:
# Build Google Test, the C++ framework we use for testing C code.
# The source code for Google Test is copied to this repository.
gtest_dir = os.path.abspath(
os.path.join(os.path.dirname(__file__), "..", "external", "googletest")
)
os.makedirs(self.build_temp, exist_ok=True)
subprocess.check_call(
["make", "-f", os.path.join(gtest_dir, "make", "Makefile"), f"GTEST_DIR={gtest_dir}"],
cwd=self.build_temp,
)
self.library_dirs = [self.build_temp]
return build_ext.run(self)
if "--run-capi-tests" in sys.argv:
sys.argv.pop()
kwargs: dict[str, Any]
if sys.platform == "darwin":
kwargs = {"language": "c++"}
compile_args: list[str] = []
else:
kwargs = {}
compile_args = ["--std=c++11"]
setup(
name="test_capi",
version="0.1",
ext_modules=[
Extension(
"test_capi",
["test_capi.cc"] + C_APIS_TO_TEST,
depends=["CPy.h", "mypyc_util.h", "pythonsupport.h"],
extra_compile_args=["-Wno-unused-function", "-Wno-sign-compare"] + compile_args,
libraries=["gtest"],
include_dirs=["../external/googletest", "../external/googletest/include"],
**kwargs,
)
],
cmdclass={"build_ext": BuildExtGtest},
)
else:
compiler = ccompiler.new_compiler()
sysconfig.customize_compiler(compiler)
cflags: list[str] = []
if compiler.compiler_type == "unix": # type: ignore[attr-defined]
cflags += ["-O3", "-Wno-unused-function"]
elif compiler.compiler_type == "msvc": # type: ignore[attr-defined]
cflags += ["/O2"]
setup(
ext_modules=[
Extension(
"librt.internal",
[
"librt_internal.c",
"init.c",
"int_ops.c",
"exc_ops.c",
"pythonsupport.c",
"getargsfast.c",
],
include_dirs=["."],
extra_compile_args=cflags,
),
Extension(
"librt.strings",
[
"librt_strings.c",
"init.c",
"int_ops.c",
"exc_ops.c",
"pythonsupport.c",
"getargsfast.c",
],
include_dirs=["."],
extra_compile_args=cflags,
),
Extension(
"librt.base64",
[
"librt_base64.c",
"base64/lib.c",
"base64/codec_choose.c",
"base64/tables/tables.c",
"base64/arch/generic/codec.c",
"base64/arch/ssse3/codec.c",
"base64/arch/sse41/codec.c",
"base64/arch/sse42/codec.c",
"base64/arch/avx/codec.c",
"base64/arch/avx2/codec.c",
"base64/arch/avx512/codec.c",
"base64/arch/neon32/codec.c",
"base64/arch/neon64/codec.c",
],
include_dirs=[".", "base64"],
extra_compile_args=cflags,
),
]
)
|
