HERA_MapMaking_VisibilitySimulation/setup.py at master · JIANSHULI/HERA_MapMaking_VisibilitySimulation · 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
# from __future__ import absolute_import
# from distutils.core import setup, Extension
import os, glob, numpy
import os.path as op
# from src import version
import json
# from setuptools import Extension, Command
# from setuptools import setup
from distutils.core import setup, Extension


########## hera_dp_vs_mp version ###########
__version__ = '1.0.0'

def indir(dir, files): return [dir+f for f in files]
def globdir(dir, files):
    rv = []
    for f in files: rv += glob.glob(dir+f)
    return rv

setup(name = 'HERA_MapMaking_VisibilitySimulation',
    version = __version__,
    # version = version.version,
    description = __doc__,
    long_description = __doc__,
    license = 'GPL',
    author = 'Eric Yang, Jeff Zheng, Jianshu Li',
    author_email = '',
    url = 'https://github.com/JIANSHULI/HERA_MapMaking_VisibilitySimulation.git',
    package_dir = {'HERA_MapMaking_VisibilitySimulation':'src'},
    packages = ['HERA_MapMaking_VisibilitySimulation'],
    include_package_data = True,
    ext_modules = [
        Extension('HERA_MapMaking_VisibilitySimulation._boost_math',
            globdir('src/_boost_math/',
                ['*.cpp', '*.c', '*.cc']),
            include_dirs = ['src/_boost_math/include', '/usr/include', numpy.get_include()],
            extra_compile_args=['-Wno-write-strings', '-O3']
        ),
        Extension('HERA_MapMaking_VisibilitySimulation._Bulm',
            globdir('src/_Bulm/',
                ['*.cpp', '*.c', '*.cc']),
            include_dirs = ['src/_Bulm/include', '/usr/include', numpy.get_include()],
            extra_compile_args=['-Wno-write-strings', '-O3']
        ),
        Extension('HERA_MapMaking_VisibilitySimulation._omnical',
            ['src/_omnical/omnical_wrap.cpp', 'src/_omnical/omnical_redcal.cc'],
            # globdir('src/_omnical/',
            #    ['*.cpp', '*.c', '*.cc']),
            include_dirs=['src/_omnical/include', '/usr/include', numpy.get_include()],
            extra_compile_args=['-Wno-write-strings', '-O3']
        )
    ],
    scripts = glob.glob('scripts/*'),
)

from src import version
########## pyuvdata version ##########
data = [version.git_origin, version.git_hash, version.git_description, version.git_branch]
with open(op.join('src', 'GIT_INFO'), 'w') as outfile:
        json.dump(data, outfile)
with open(op.join('src', 'VERSION'), 'w') as outfile:
        json.dump(__version__, outfile)
try:
    with open('VERSION', 'w') as outfile:
        json.dump(__version__, outfile)
except:
    print('VERSION not saved to outside.')