BENTO: add complex checks.

2 files changed, 48 insertions, 0 deletions

diff --git a/numpy/build_utils/__init__.py b/numpy/build_utils/__init__.py
index 9789e09d9..02fa4e2e3 100644
--- a/numpy/build_utils/__init__.py
+++ b/numpy/build_utils/__init__.py
@@ -106,6 +106,33 @@ int main()
         return ret == 0
     return ret
 
+@waflib.Configure.conf
+def check_type(self, type_name, **kw):
+    code = r"""
+int main() {
+	if ((%(type_name)s *) 0)
+        return 0;
+	if (sizeof (%(type_name)s))
+        return 0;
+}
+""" % {"type_name": type_name}
+
+    kw["code"] = to_header(kw) + code
+    kw["msg"] = "Checking for type %r" % type_name
+    kw["errmsg"] = "not found"
+    kw["okmsg"] = "yes"
+
+    validate_arguments(self, kw)
+    try_compile(self, kw)
+    ret = kw["success"]
+
+    kw["define_name"] = "HAVE_%s" % sanitize_string(type_name)
+    kw["define_comment"] = "/* Set to 1 if %s is defined. */" % type_name
+    self.post_check(**kw)
+    if not kw.get('execute', False):
+        return ret == 0
+    return ret
+
 def do_binary_search(conf, type_name, kw):
     code = """\
 typedef %(type)s waf_check_sizeof_type;
diff --git a/numpy/core/bscript b/numpy/core/bscript
index b0b45d720..6bd5c9c87 100644
--- a/numpy/core/bscript
+++ b/numpy/core/bscript
@@ -193,6 +193,26 @@ def check_math_runtime(conf):
             except waflib.Errors.ConfigurationError:
                 NUMPYCONFIG_SYM.append(('DEFINE_NPY_HAVE_DECL_%s' % f.upper(), ''))
 
+def check_complex(conf):
+    if conf.check_header("complex.h"):
+        NUMPYCONFIG_SYM.append(('DEFINE_NPY_USE_C99_COMPLEX',
+                                '#define NPY_USE_C99_COMPLEX 1'))
+        for t in C99_COMPLEX_TYPES:
+            try:
+                conf.check_type(t, header_name='complex.h')
+                NUMPYCONFIG_SYM.append(('DEFINE_NPY_HAVE_%s' % numpy.build_utils.sanitize_string(t),
+                                        '#define NPY_HAVE_%s' % numpy.build_utils.sanitize_string(t)))
+            except waflib.Errors.ConfigurationError:
+                NUMPYCONFIG_SYM.append(('DEFINE_NPY_HAVE_%s' % numpy.build_utils.sanitize_string(t), ''))
+
+        for prec in ["", "f", "l"]:
+            flist = [f + prec for f in C99_COMPLEX_FUNCS]
+            conf.check_functions_at_once(flist, use="M")
+    else:
+        NUMPYCONFIG_SYM.append(('DEFINE_NPY_USE_C99_COMPLEX', ''))
+        for t in C99_COMPLEX_TYPES:
+            NUMPYCONFIG_SYM.append(('DEFINE_NPY_HAVE_%s' % numpy.build_utils.sanitize_string(t), ''))
+
 @pre_configure()
 def configure(context):
     conf = context.waf_context
@@ -217,6 +237,7 @@ def configure(context):
     signal_smp_checks(conf)
     check_math_runtime(conf)
     numpy.build_utils.check_inline(conf)
+    check_complex(conf)
 
     conf.env["CONFIG_HEADER_TEMPLATE"] = """\
 %(content)s