summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat
-rw-r--r--numpy/build_utils/__init__.py172
-rw-r--r--numpy/core/bscript66
2 files changed, 238 insertions, 0 deletions
diff --git a/numpy/build_utils/__init__.py b/numpy/build_utils/__init__.py
index 0fa3f0ff4..d25458fc6 100644
--- a/numpy/build_utils/__init__.py
+++ b/numpy/build_utils/__init__.py
@@ -220,3 +220,175 @@ def check_type_size(conf, type_name, expected_sizes=None, **kw):
conf.post_check(**kw)
return size
+@waflib.Configure.conf
+def check_functions_at_once(self, funcs, **kw):
+ header = []
+ header = ['#ifdef __cplusplus']
+ header.append('extern "C" {')
+ header.append('#endif')
+ for f in funcs:
+ header.append("\tchar %s();" % f)
+ # Handle MSVC intrinsics: force MS compiler to make a function
+ # call. Useful to test for some functions when built with
+ # optimization on, to avoid build error because the intrinsic
+ # and our 'fake' test declaration do not match.
+ header.append("#ifdef _MSC_VER")
+ header.append("#pragma function(%s)" % f)
+ header.append("#endif")
+ header.append('#ifdef __cplusplus')
+ header.append('};')
+ header.append('#endif')
+ funcs_decl = "\n".join(header)
+
+ tmp = []
+ for f in funcs:
+ tmp.append("\t%s();" % f)
+ tmp = "\n".join(tmp)
+
+ code = r"""
+%(include)s
+%(funcs_decl)s
+
+int main (void)
+{
+ %(tmp)s
+ return 0;
+}
+""" % {"tmp": tmp, "include": to_header(kw), "funcs_decl": funcs_decl}
+ kw["code"] = code
+ if not "features" in kw:
+ kw["features"] = ["c", "cprogram"]
+
+ msg = ", ".join(funcs)
+ if len(msg) > 30:
+ _funcs = list(funcs)
+ msg = []
+ while len(", ".join(msg)) < 30 and _funcs:
+ msg.append(_funcs.pop(0))
+ msg = ", ".join(msg) + ",..."
+ if "lib" in kw:
+ kw["msg"] = "Checking for functions %s in library %r" % (msg, kw["lib"])
+ else:
+ kw["msg"] = "Checking for functions %s" % msg
+
+ validate_arguments(self, kw)
+ try_compile(self, kw)
+ ret = kw["success"]
+
+ # We set the config.h define here because we need to define several of them
+ # in one shot
+ if ret == 0:
+ for f in funcs:
+ self.define_with_comment("HAVE_%s" % sanitize_string(f), 1,
+ "/* Define to 1 if you have the `%s' function */" % f)
+
+ self.post_check(**kw)
+ if not kw.get('execute', False):
+ return ret == 0
+ return ret
+
+@waflib.Configure.conf
+def post_check(self, *k, **kw):
+ "set the variables after a test was run successfully"
+
+ is_success = 0
+ if kw['execute']:
+ if kw['success'] is not None:
+ is_success = kw['success']
+ else:
+ is_success = (kw['success'] == 0)
+
+ def define_or_stuff():
+ nm = kw['define_name']
+ cmt = kw.get('define_comment', None)
+ value = kw.get("define_value", is_success)
+ if kw['execute'] and kw.get('define_ret', None) and isinstance(is_success, str):
+ self.define_with_comment(kw['define_name'], value, cmt, quote=kw.get('quote', 1))
+ else:
+ self.define_cond(kw['define_name'], value, cmt)
+
+ if 'define_name' in kw:
+ define_or_stuff()
+
+ if is_success and 'uselib_store' in kw:
+ from waflib.Tools import ccroot
+
+ # TODO see get_uselib_vars from ccroot.py
+ _vars = set([])
+ for x in kw['features']:
+ if x in ccroot.USELIB_VARS:
+ _vars |= ccroot.USELIB_VARS[x]
+
+ for k in _vars:
+ lk = k.lower()
+ if k == 'INCLUDES': lk = 'includes'
+ if k == 'DEFKEYS': lk = 'defines'
+ if lk in kw:
+ val = kw[lk]
+ # remove trailing slash
+ if isinstance(val, str):
+ val = val.rstrip(os.path.sep)
+ self.env.append_unique(k + '_' + kw['uselib_store'], val)
+ return is_success
+
+@waflib.Configure.conf
+def define_with_comment(conf, define, value, comment=None, quote=True):
+ if comment is None:
+ return conf.define(define, value, quote)
+
+ assert define and isinstance(define, str)
+
+ comment_tbl = conf.env[DEFINE_COMMENTS] or {}
+ comment_tbl[define] = comment
+ conf.env[DEFINE_COMMENTS] = comment_tbl
+
+ return conf.define(define, value, quote)
+
+@waflib.Configure.conf
+def get_comment(self, key):
+ assert key and isinstance(key, str)
+
+ if key in self.env[DEFINE_COMMENTS]:
+ return self.env[DEFINE_COMMENTS][key]
+ return None
+
+@waflib.Configure.conf
+def define_cond(self, name, value, comment):
+ """Conditionally define a name.
+ Formally equivalent to: if value: define(name, 1) else: undefine(name)"""
+ if value:
+ self.define_with_comment(name, value, comment)
+ else:
+ self.undefine(name)
+
+@waflib.Configure.conf
+def get_config_header(self, defines=True, headers=False):
+ """
+ Create the contents of a ``config.h`` file from the defines and includes
+ set in conf.env.define_key / conf.env.include_key. No include guards are added.
+
+ :param defines: write the defines values
+ :type defines: bool
+ :param headers: write the headers
+ :type headers: bool
+ :return: the contents of a ``config.h`` file
+ :rtype: string
+ """
+ tpl = self.env["CONFIG_HEADER_TEMPLATE"] or "%(content)s"
+
+ lst = []
+ if headers:
+ for x in self.env[INCKEYS]:
+ lst.append('#include <%s>' % x)
+
+ if defines:
+ for x in self.env[DEFKEYS]:
+ if self.is_defined(x):
+ val = self.get_define(x)
+ cmt = self.get_comment(x)
+ if cmt is not None:
+ lst.append(cmt)
+ lst.append('#define %s %s\n' % (x, val))
+ else:
+ lst.append('/* #undef %s */\n' % x)
+ return tpl % {"content": "\n".join(lst)}
diff --git a/numpy/core/bscript b/numpy/core/bscript
index d85f634f9..b8baafe17 100644
--- a/numpy/core/bscript
+++ b/numpy/core/bscript
@@ -128,6 +128,71 @@ def signal_smp_checks(conf):
else:
NUMPYCONFIG_SYM.append(("NPY_NO_SMP", 0))
+def check_math_runtime(conf):
+ header_name = "Python.h math.h"
+ features = "c cprogram pyext"
+
+ mlibs = [None, "m", "cpml"]
+ mathlib = os.environ.get('MATHLIB')
+ if mathlib:
+ mlibs.insert(0, mathlib)
+
+ mlib = None
+ for lib in mlibs:
+ try:
+ if lib is None:
+ kw = {}
+ else:
+ kw = {"lib": lib}
+ st = conf.check_functions_at_once(["exp"], uselib_store="M", **kw)
+ mlib = lib or []
+ break
+ except waflib.Errors.ConfigurationError:
+ pass
+ if mlib is None:
+ raise waflib.Errors.ConfigurationError("No math lib found !")
+
+ # XXX: this is ugly: mathlib has nothing to do in a public header file
+ NUMPYCONFIG_SYM.append(('MATHLIB', ','.join(mlib)))
+
+ conf.check_functions_at_once(MANDATORY_FUNCS, use="M")
+
+ mfuncs = ('expl', 'expf', 'log1p', 'expm1', 'asinh', 'atanhf', 'atanhl',
+ 'rint', 'trunc')
+ conf.check_functions_at_once(mfuncs, use="M")
+
+ header_name = "Python.h math.h"
+ features = "c pyext"
+ for f in OPTIONAL_STDFUNCS_MAYBE:
+ try:
+ conf.check_declaration("HAVE_%s" % numpy.build_utils.sanitize_string(f),
+ header_name=header_name,
+ features=features)
+ OPTIONAL_STDFUNCS.remove(f)
+ except waflib.Errors.ConfigurationError:
+ pass
+
+ conf.check_functions_at_once(OPTIONAL_STDFUNCS,
+ features=features, mandatory=False, use="M")
+ conf.check_functions_at_once(C99_FUNCS_SINGLE,
+ features=features, mandatory=False, use="M")
+ conf.check_functions_at_once(C99_FUNCS_EXTENDED,
+ features=features, mandatory=False, use="M")
+
+ for f in ["isnan", "isinf", "signbit", "isfinite"]:
+ try:
+ conf.check_declaration("HAVE_DECL_%s" % f.upper(), header_name=header_name,
+ features=features)
+ NUMPYCONFIG_SYM.append(('DEFINE_NPY_HAVE_DECL_%s' % f.upper(),
+ '#define NPY_HAVE_DECL_%s' % f.upper()))
+ except waflib.Errors.ConfigurationError:
+ try:
+ conf.check_declaration(f, header_name=header_name, features=features)
+ NUMPYCONFIG_SYM.append(('DEFINE_NPY_HAVE_DECL_%s' % f.upper(),
+ '#define NPY_HAVE_DECL_%s' % f.upper()))
+ except waflib.Errors.ConfigurationError:
+ NUMPYCONFIG_SYM.append(('DEFINE_NPY_HAVE_DECL_%s' % f.upper(), ''))
+
@pre_configure()
def configure(context):
conf = context.waf_context
@@ -150,6 +215,7 @@ def configure(context):
type_checks(conf)
signal_smp_checks(conf)
+ check_math_runtime(conf)
conf.env["CONFIG_HEADER_TEMPLATE"] = """\
%(content)s
View Lorry Controller Status