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
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
|
#! Last Change: Mon Apr 21 07:00 PM 2008 J
"""Code to support special facilities to scons which are only useful for
numpy.core, hence not put into numpy.distutils.scons"""
import sys
import os
from os.path import join as pjoin, dirname as pdirname, basename as pbasename
from copy import deepcopy
from code_generators.generate_array_api import \
do_generate_api as nowrap_do_generate_array_api
from code_generators.generate_ufunc_api import \
do_generate_api as nowrap_do_generate_ufunc_api
from numscons.numdist import process_c_str as process_str
from numscons.core.utils import rsplit, isstring
try:
from numscons import distutils_dirs_emitter
except ImportError:
raise ImportError("You need numscons >= 0.5.2")
import SCons.Node
import SCons
from SCons.Builder import Builder
from SCons.Action import Action
def split_ext(string):
sp = rsplit(string, '.', 1)
if len(sp) == 1:
return (sp[0], '')
else:
return sp
#------------------------------------
# Ufunc and multiarray API generators
#------------------------------------
def do_generate_array_api(target, source, env):
nowrap_do_generate_array_api([str(i) for i in target],
[str(i) for i in source])
return 0
def do_generate_ufunc_api(target, source, env):
nowrap_do_generate_ufunc_api([str(i) for i in target],
[str(i) for i in source])
return 0
def generate_api_emitter(target, source, env):
"""Returns the list of targets generated by the code generator for array
api and ufunc api."""
base, ext = split_ext(str(target[0]))
dir = pdirname(base)
ba = pbasename(base)
h = pjoin(dir, '__' + ba + '.h')
c = pjoin(dir, '__' + ba + '.c')
txt = base + '.txt'
#print h, c, txt
t = [h, c, txt]
return (t, source)
#-------------------------
# From template generators
#-------------------------
# XXX: this is general and can be used outside numpy.core.
def do_generate_from_template(targetfile, sourcefile, env):
t = open(targetfile, 'w')
s = open(sourcefile, 'r')
allstr = s.read()
s.close()
writestr = process_str(allstr)
t.write(writestr)
t.close()
return 0
def generate_from_template(target, source, env):
for t, s in zip(target, source):
do_generate_from_template(str(t), str(s), env)
def generate_from_template_emitter(target, source, env):
base, ext = split_ext(pbasename(str(source[0])))
t = pjoin(pdirname(str(target[0])), base)
return ([t], source)
#----------------
# umath generator
#----------------
def do_generate_umath(targetfile, sourcefile, env):
t = open(targetfile, 'w')
from code_generators import generate_umath
code = generate_umath.make_code(generate_umath.defdict, generate_umath.__file__)
t.write(code)
t.close()
def generate_umath(target, source, env):
for t, s in zip(target, source):
do_generate_umath(str(t), str(s), env)
def generate_umath_emitter(target, source, env):
t = str(target[0]) + '.c'
return ([t], source)
#-----------------------------------------
# Other functions related to configuration
#-----------------------------------------
def CheckBrokenMathlib(context, mathlib):
src = """
/* check whether libm is broken */
#include <math.h>
int main(int argc, char *argv[])
{
return exp(-720.) > 1.0; /* typically an IEEE denormal */
}
"""
try:
oldLIBS = deepcopy(context.env['LIBS'])
except:
oldLIBS = []
try:
context.Message("Checking if math lib %s is usable for numpy ... " % mathlib)
context.env.AppendUnique(LIBS = mathlib)
st = context.TryRun(src, '.c')
finally:
context.env['LIBS'] = oldLIBS
if st[0]:
context.Result(' Yes !')
else:
context.Result(' No !')
return st[0]
def check_mlib(config, mlib):
"""Return 1 if mlib is available and usable by numpy, 0 otherwise.
mlib can be a string (one library), or a list of libraries."""
# Check the libraries in mlib are linkable
if len(mlib) > 0:
# XXX: put an autoadd argument to 0 here and add an autoadd argument to
# CheckBroekenMathlib (otherwise we may add bogus libraries, the ones
# which do not path the CheckBrokenMathlib test).
st = config.CheckLib(mlib)
if not st:
return 0
# Check the mlib is usable by numpy
return config.CheckBrokenMathlib(mlib)
def check_mlibs(config, mlibs):
for mlib in mlibs:
if check_mlib(config, mlib):
return mlib
# No mlib was found.
raise SCons.Errors.UserError("No usable mathlib was found: chose another "\
"one using the MATHLIB env variable, eg "\
"'MATHLIB=m python setup.py build'")
def is_npy_no_signal():
"""Return True if the NPY_NO_SIGNAL symbol must be defined in configuration
header."""
return sys.platform == 'win32'
def define_no_smp():
"""Returns True if we should define NPY_NOSMP, False otherwise."""
#--------------------------------
# Checking SMP and thread options
#--------------------------------
# Python 2.3 causes a segfault when
# trying to re-acquire the thread-state
# which is done in error-handling
# ufunc code. NPY_ALLOW_C_API and friends
# cause the segfault. So, we disable threading
# for now.
if sys.version[:5] < '2.4.2':
nosmp = 1
else:
# Perhaps a fancier check is in order here.
# so that threads are only enabled if there
# are actually multiple CPUS? -- but
# threaded code can be nice even on a single
# CPU so that long-calculating code doesn't
# block.
try:
nosmp = os.environ['NPY_NOSMP']
nosmp = 1
except KeyError:
nosmp = 0
return nosmp == 1
array_api_gen_bld = Builder(action = Action(do_generate_array_api, '$ARRAPIGENCOMSTR'),
emitter = [generate_api_emitter,
distutils_dirs_emitter])
ufunc_api_gen_bld = Builder(action = Action(do_generate_ufunc_api, '$UFUNCAPIGENCOMSTR'),
emitter = [generate_api_emitter,
distutils_dirs_emitter])
template_bld = Builder(action = Action(generate_from_template, '$TEMPLATECOMSTR'),
emitter = [generate_from_template_emitter,
distutils_dirs_emitter])
umath_bld = Builder(action = Action(generate_umath, '$UMATHCOMSTR'),
emitter = [generate_umath_emitter, distutils_dirs_emitter])
|
