distarray/setup.py at master · minrk/distarray · GitHub

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
# encoding: utf-8

__docformat__ = "restructuredtext en"

#----------------------------------------------------------------------------
#  Copyright (C) 2008  The IPython Development Team
#
#  Distributed under the terms of the BSD License.  The full license is in
#  the file COPYING, distributed as part of this software.
#----------------------------------------------------------------------------

#----------------------------------------------------------------------------
# Imports
#----------------------------------------------------------------------------

from distutils.core import setup
from mpidistutils import Distribution, Extension, Executable
from mpidistutils import config, build, build_ext
from mpidistutils import build_exe, install_exe, clean_exe
import mpi4py
import numpy

#--------- -------------------------------------------------------------------
# Metadata
#----------------------------------------------------------------------------

metadata = {
    'name'             : 'distarray',
    'version'          : '0.1',
    'description'      : 'Distributed Memory Arrays for Python',
    'keywords'         : 'parallel mpi distributed array',
    'license'          : 'New BSD',
    'author'           : 'Brian E. Granger',
    'author_email'     : 'ellisonbg@gmail.com',
    }

# See if FFTW_DIR is set
import os
fftw_dir = os.getenv("FFTW_DIR")

#----------------------------------------------------------------------------
# Extension modules
#----------------------------------------------------------------------------

def find_ext_modules():
    import sys

    maps = Extension(
        name='distarray.core.maps_fast',
        sources=['distarray/core/maps_fast.c']
    )
    # This extension shows how to call mpi4py's C layer using Cython
    mpi_test = Extension(
        name='distarray.mpi.tests.helloworld',
        sources=['distarray/mpi/tests/helloworld.c'],
        include_dirs = [mpi4py.get_include()]
    )

    allext = [maps, mpi_test]
    if not fftw_dir is None:
        py_fftw = Extension(
            name = 'distarray.fft.py_fftw',
            library_dirs = [fftw_dir+"/lib"],
            include_dirs = [
                fftw_dir+"/include",
                mpi4py.get_include(),
                numpy.get_include()],
            libraries = ['fftw3_mpi', 'fftw3', 'fftw3f_mpi','fftw3f'],
            sources = ['distarray/fft/py_fftw.c'],
        )
        allext.append(py_fftw)
        print "FFTW found, including distarray.fft"

    return allext

def find_headers():
    # allheaders = ['mpi/ext/libmpi.h']
    return []

def find_executables():
    return []

def find_packages():
    packages= [
        'distarray',
        'distarray.tests',
        'distarray.core',
        'distarray.core.tests',
        'distarray.mpi',
        'distarray.mpi.tests',
        'distarray.random',
        'distarray.random.tests',
        'distarray.linalg',
        'distarray.linalg.tests'
    ]

    if not fftw_dir is None:
        packages.extend([
            'distarray.fft',
            'distarray.fft.tests'
        ])

    return packages


#----------------------------------------------------------------------------
# Setup
#----------------------------------------------------------------------------


def main():
    setup(packages = find_packages(),
          package_data = {'distarray' : ['include/*.pxi']},
          headers = find_headers(),
          ext_modules = find_ext_modules(),
          executables = find_executables(),
          distclass = Distribution,
          cmdclass = {'config'      : config,
                      'build'       : build,
                      'build_ext'   : build_ext,
                      'build_exe'   : build_exe,
                      'clean_exe'   : clean_exe,
                      'install_exe' : install_exe,
                      },
          **metadata)

if __name__ == '__main__':
    # hack distutils.sysconfig to eliminate debug flags
    from distutils import sysconfig
    cvars = sysconfig.get_config_vars()
    cflags = cvars.get('OPT')
    if cflags:
        cflags = cflags.split()
        for flag in ('-g', '-g3'):
            if flag in cflags:
                cflags.remove(flag)
        cvars['OPT'] = str.join(' ', cflags)
    # and now call main
    main()