SDF_utilities/setup.py at main · dholland88/SDF_utilities · 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
import os
import sys
from sysconfig import get_path
from setuptools import setup, Extension
import subprocess as sp
try:
    import numpy
    gotnumpy = True
except:
    gotnumpy = False

def get_numpy_dir():
    if gotnumpy:
        numpydir = '/nonexistent'
        incdir = '/nonexistent'
        if numpy.__file__:
            numpydir = os.path.dirname(numpy.__file__)
            incdir = os.path.join(numpydir, 'core', 'include')
        for r, d, fl in os.walk(incdir):
            for f in fl:
                if f == 'arrayobject.h':
                    return os.path.realpath(os.path.join(r, '..'))
        for r, d, fl in os.walk(numpydir):
            for f in fl:
                if f == 'arrayobject.h':
                    return os.path.realpath(os.path.join(r, '..'))
    sys.path.insert(0, get_path('platlib'))
    for path in sys.path:
        for r, d, fl in os.walk(path):
            for f in fl:
                if f == 'arrayobject.h':
                    return os.path.realpath(os.path.join(r, '..'))
    return None


sdfdir = os.path.join(os.environ['PWD'], '..', 'C')

srcfiles = ['sdf_python.c']

incdirs = [get_numpy_dir()] + [os.path.join(sdfdir, 'include')]
libdirs = [os.path.join(sdfdir, 'lib')]


def get_version():
    import subprocess as sp

    commit_id = "UNKNOWN"
    version = "UNKNOWN"

    sp.call("sh gen_commit_string.sh .", shell=True)

    with open("commit_info.h") as f:
        lines = f.readlines()
        for l in lines:
            if l.find('SDF_COMMIT_ID') != -1:
                commit_id = l.split('"')[1]
                version = commit_id.lstrip('v').split('-')[0]
            elif l.find('SDF_COMMIT_DATE') != -1:
                commit_date = l.split('"')[1]

    with open("sdf_helper/_version.py", "w") as f:
        f.write('__version__ = "{}"\n'.format(version))
        f.write('__commit_id__ = "{}"\n'.format(commit_id))
        f.write('__commit_date__ = "{}"\n'.format(commit_date))

    return commit_id

get_version()

setup(
    name='sdf',
    version='1.0',
    url='http://github.com/Warwick-Plasma/SDF.git',
    description='Python module for processing SDF files',
    packages=['sdf_helper'],
    py_modules=['sdf_legacy'],
    ext_modules=[
        Extension(
            'sdf',
            srcfiles,
            include_dirs=incdirs,
            library_dirs=libdirs,
            libraries=['sdfc']
        )
    ],
)