Moving things to live under scipy

1 files changed, 232 insertions, 0 deletions

diff --git a/scipy/base/setup.py b/scipy/base/setup.py
new file mode 100644
index 000000000..3136f2874
--- /dev/null
+++ b/scipy/base/setup.py
@@ -0,0 +1,232 @@
+
+import imp
+import os
+from os.path import join
+from glob import glob
+from scipy.distutils.misc_util import Configuration,dot_join
+from distutils.dep_util import newer,newer_group
+
+from scipy.distutils.command.build import build
+
+def configuration(parent_package='',top_path=None):
+    config = Configuration('base',parent_package,top_path)
+    local_dir = config.local_path
+    codegen_dir = join(local_dir,'CodeGenerators')
+
+    generate_umath_py = join(codegen_dir,'generate_umath.py')
+    n = dot_join(config.name,'generate_umath')
+    generate_umath = imp.load_module('_'.join(n.split('.')),
+                                     open(generate_umath_py,'U'),generate_umath_py,
+                                     ('.py','U',1))
+
+    def generate_config_h(ext, build_dir):
+        target = join(build_dir,'config.h')
+        if newer(__file__,target):
+            config_cmd = config.get_config_cmd()
+            print 'Generating',target
+            #
+            tc = generate_testcode(target)
+            from distutils import sysconfig
+            python_include = sysconfig.get_python_inc()
+            result = config_cmd.try_run(tc,include_dirs=[python_include])
+            if not result:
+                raise "ERROR: Failed to test configuration"            
+            moredefs = []
+
+            #
+            mathlibs = []
+            tc = testcode_mathlib()
+            mathlibs_choices = [[],['m'],['cpml']]
+            mathlib = os.environ.get('MATHLIB')
+            if mathlib:
+                mathlibs_choices.insert(0,mathlib.split(','))
+            for libs in mathlibs_choices:
+                if config_cmd.try_run(tc,libraries=libs):
+                    mathlibs = libs
+                    break
+            else:
+                raise "math library missing; rerun setup.py after setting the MATHLIB env variable"
+            ext.libraries.extend(mathlibs)
+            moredefs.append(('MATHLIB',','.join(mathlibs)))
+
+            libs = mathlibs
+            if config_cmd.check_func('expl', libraries=libs, decl=1):
+                moredefs.append('HAVE_LONGDOUBLE_FUNCS')
+            if config_cmd.check_func('expf', libraries=libs, decl=1):
+                moredefs.append('HAVE_FLOAT_FUNCS')
+            if config_cmd.check_func('asinh', libraries=libs, decl=1):
+                moredefs.append('HAVE_INVERSE_HYPERBOLIC')
+
+            if moredefs:
+                target_f = open(target,'a')
+                for d in moredefs:
+                    if isinstance(d,str):
+                        target_f.write('#define %s\n' % (d))
+                    else:
+                        target_f.write('#define %s %s\n' % (d[0],d[1]))
+                target_f.close()
+        else:
+            mathlibs = []
+            target_f = open(target)
+            for line in target_f.readlines():
+                s = '#define MATHLIB'
+                if line.startswith(s):
+                    mathlibs.extend(line[len(s):].strip().split(','))
+            target_f.close()
+
+        ext.libraries.extend(mathlibs)
+        ext.include_dirs.append(os.path.dirname(target))
+        return target
+
+    def generate_array_api(ext,build_dir):
+        target = join(build_dir,'__multiarray_api.h')
+        script = join(codegen_dir,'generate_array_api.py')
+        if newer(script,target):
+            script = os.path.abspath(script)
+            old_cwd = os.getcwd()
+            os.chdir(build_dir)
+            print 'executing',script
+            execfile(script,{},{})
+            os.chdir(old_cwd)
+        return target
+
+    def generate_ufunc_api(ext,build_dir):
+        target = join(build_dir,'__ufunc_api.h')
+        script = join(codegen_dir,'generate_ufunc_api.py')
+        if newer(script,target):
+            script = os.path.abspath(script)
+            old_cwd = os.getcwd()
+            os.chdir(build_dir)
+            print 'executing',script
+            execfile(script,{},{})
+            os.chdir(old_cwd)
+        return target
+
+    def generate_umath_c(ext,build_dir):
+        target = join(build_dir,'__umath_generated.c')
+        script = generate_umath_py
+        if newer(script,target):
+            f = open(target,'w')
+            f.write(generate_umath.make_code(generate_umath.defdict,
+                                             generate_umath.__file__))
+            f.close()
+        return []
+
+    config.add_include_dirs('Include','Src')
+    config.add_headers(join('Include','scipy','base','*.h'))
+    from scipy.distutils.command.build_src import appendpath
+    print "****%s****" % config.local_path
+    config.add_include_dirs(appendpath('build/src',join(config.local_path,'Src')))
+
+    deps = [join('Src','umathmodule.c.src'),
+            join('Src','arrayobject.c'),
+            join('Src','arraymethods.c'),
+            join('Src','scalartypes.inc.src'),
+            join('Src','arraytypes.inc.src'),
+            ]
+
+    config.add_extension('multiarray',
+                         sources = [join('Src','multiarraymodule.c'),
+                                    generate_config_h,
+                                    generate_array_api,
+                                    join('Src','scalartypes.inc.src'),
+                                    join('Src','arraytypes.inc.src'),
+                                    join(codegen_dir,'generate_array_api.py'),
+                                    join('*.py')
+                                    ],
+                         depends = deps
+                         )
+
+    config.add_extension('umath',
+                         sources = [generate_config_h,
+                                    join('Src','umathmodule.c.src'),
+                                    generate_umath_c,
+                                    generate_ufunc_api,
+                                    join('Src','scalartypes.inc.src'),
+                                    join('Src','arraytypes.inc.src'),
+                                    ],
+                         depends = [join('Src','ufuncobject.c'),
+                                    generate_umath_py,
+                                    join(codegen_dir,'generate_ufunc_api.py')
+                                    ]+deps,
+                         )
+    config.add_include_dirs(appendpath('build/src',config.local_path))
+    config.add_extension('_compiled_base',
+                         sources=['_compiled_base.c'],
+                         )
+                         
+
+    return config
+
+def testcode_mathlib():
+    return """\
+/* check whether libm is broken */
+#include <math.h>
+int main(int argc, char *argv[])
+{
+  return exp(-720.) > 1.0;  /* typically an IEEE denormal */
+}
+"""
+
+
+def generate_testcode(target):
+    testcode = [r'''
+#include <Python.h>
+#include <limits.h>
+#include <stdio.h>
+
+int main(int argc, char **argv)
+{
+
+        FILE *fp;
+
+        fp = fopen("'''+target+'''","w");
+        ''']
+
+    c_size_test = r'''
+#ifndef %(sz)s
+          fprintf(fp,"#define %(sz)s %%d\n", sizeof(%(type)s));
+#else
+          fprintf(fp,"/* #define %(sz)s %%d */\n", %(sz)s);
+#endif
+'''
+    for sz, t in [('SIZEOF_SHORT', 'short'),
+                  ('SIZEOF_INT', 'int'),
+                  ('SIZEOF_LONG', 'long'),
+                  ('SIZEOF_FLOAT', 'float'),
+                  ('SIZEOF_DOUBLE', 'double'),
+                  ('SIZEOF_LONG_DOUBLE', 'long double'),
+                  ('SIZEOF_PY_INTPTR_T', 'Py_intptr_t'),
+                  ]:
+        testcode.append(c_size_test % {'sz' : sz, 'type' : t})
+
+    testcode.append('#ifdef PY_LONG_LONG')
+    testcode.append(c_size_test % {'sz' : 'SIZEOF_LONG_LONG',
+                                   'type' : 'PY_LONG_LONG'})
+    testcode.append(c_size_test % {'sz' : 'SIZEOF_PY_LONG_LONG',
+                                   'type' : 'PY_LONG_LONG'})
+    
+
+    testcode.append(r'''
+#else
+        fprintf(fp, "/* PY_LONG_LONG not defined */\n");
+#endif
+#ifndef CHAR_BIT
+          {
+             unsigned char var = 2;
+             int i=0;
+             while (var >= 2) {
+                     var = var << 1;
+                     i++;
+             }
+             fprintf(fp,"#define CHAR_BIT %d\n", i+1);
+          }
+#else
+          fprintf(fp, "/* #define CHAR_BIT %d */\n", CHAR_BIT);
+#endif
+          fclose(fp);
+          return 0;
+}
+''')
+    testcode = '\n'.join(testcode)    
+    return testcode