ENH: evaluate c99 complex funcs at build time

ewmoore · ewmoore · commit 50d6720dde48 · 2014-02-26T18:11:27.000-05:00
For all of the C99 complex function excepting the creal, cimag, cabs and
carg (which are either trivial or defined by a C89 function) we now run
a set of build time tests to evaluate if we want to use the system
version of these functions or our own.  These tests involve all of the
special values for these functions defined in Annxex G of the C99
standard as well as any tests that were existing in the numpy test
suite. These tests do not consider the sign of NAN.
diff --git a/numpy/core/setup.py b/numpy/core/setup.py
@@ -220,6 +220,29 @@ def check_prec(prec):
             else:
                 priv.extend([(fname2def(f), 1) for f in flist])
 
+            flist = [f + prec for f in C99_COMPLEX_FUNCS_CHECKED]
+            decl = dict([(f, True) for f in flist])
+            exists = []
+            if not config.check_funcs_once(flist, call=decl, decl=decl,
+                                           libraries=mathlibs):
+                for f in C99_COMPLEX_FUNCS_CHECKED:
+                    if config.check_func(f + prec, call=True, decl=True,
+                                         libraries=mathlibs):
+                        exists.append(f)
+            else:
+                exists.extend(C99_COMPLEX_FUNCS_CHECKED)
+                
+            if len(exists) > 0:
+                fp = open('./numpy/core/test_c99complex.c', 'r')
+                obody = fp.read()
+                fp.close()
+                precname = {'f':'FLOAT', '':'DOUBLE', 'l':'LONGDOUBLE'}[prec]
+            for f in exists:
+                print("ERIC1")
+                body = obody.replace('PYTESTPRECISION', precname).replace('PYTESTFUNC', f.upper())
+                if config.try_run(body, libraries=mathlibs):
+                    priv.append((fname2def(f + prec), 1))
+
         check_prec('')
         check_prec('f')
         check_prec('l')
diff --git a/numpy/core/setup_common.py b/numpy/core/setup_common.py
@@ -143,9 +143,10 @@ def check_api_version(apiversion, codegen_dir):
 
 C99_COMPLEX_TYPES = ['complex double', 'complex float', 'complex long double']
 
-C99_COMPLEX_FUNCS = ['creal', 'cimag', 'cabs', 'carg', 'cexp', 'clog', 'csqrt',
-                  'cpow', 'ccos', 'csin', 'ctan', 'ccosh', 'csinh', 'ctanh',
-                  'cacos', 'casin', 'catan', 'cacosh', 'casinh', 'catanh']
+C99_COMPLEX_FUNCS = ['creal', 'cimag', 'cabs', 'carg']
+C99_COMPLEX_FUNCS_CHECKED = ['cacos', 'casin', 'catan', 'cacosh', 'casinh',
+                             'catanh', 'ccos', 'csin', 'ctan', 'ccosh', 'csinh',
+                             'ctanh', 'cexp', 'clog', 'cpow', 'csqrt']
 
 def fname2def(name):
     return "HAVE_%s" % name.upper()
diff --git a/numpy/core/test_c99complex.c b/numpy/core/test_c99complex.c
