geofwi/setup.py at main · aaspip/geofwi · 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
#!/usr/bin/env python
# -*- encoding: utf8 -*-
# import glob
# import inspect
# import io
# import os

from setuptools import setup
import numpy, os
# from torch.utils.cpp_extension import (BuildExtension, CppExtension,
#                                        CUDAExtension)
# from torch import cuda

long_description = """
Source code: https://github.com/aaspip/geofwi""".strip()

# export CMAKE_CXX_COMPILER='gcc-mp-14'
os.environ["CXX"] = "g++-mp-14"
os.environ["CFLAGS"] = '-Wall -Wextra -pedantic -fPIC -Ofast -DDW_ACCURACY=2 -DDW_DTYPE=float'

# from distutils.core import Extension
# scalar_module = Extension('scalar', sources=['geofwi/deepwave/scalar.c'],
#                                                 include_dirs=[numpy.get_include()])

def read(*names, **kwargs):
    return io.open(
        os.path.join(os.path.dirname(__file__), *names),
        encoding=kwargs.get("encoding", "utf8")).read()

# from numpy.distutils.core import setup
setup(
    name="geofwi",
    version="0.0.0.1",
    license='MIT License',
    description="GeoFWI: A lightweight velocity model dataset for benchmarking full waveform inversion using deep learning",
    long_description=long_description,
    author="geofwi developing team",
    author_email="chenyk2016@gmail.com",
    url="https://github.com/aaspip/geofwi",
    packages=['geofwi'],
    include_package_data=True,
    zip_safe=False,
    classifiers=[
        # complete classifier list:
        # http://pypi.python.org/pypi?%3Aaction=list_classifiers
        "Development Status :: 4 - Beta",
        "Intended Audience :: Developers",
        "Intended Audience :: Science/Research",
        "Operating System :: Unix",
        "Operating System :: POSIX",
        "Operating System :: Microsoft :: Windows",
        "Programming Language :: Python",
        "Programming Language :: Python :: 3.11",
        "Programming Language :: Python :: Implementation :: CPython",
        "Topic :: Scientific/Engineering",
        "Topic :: Scientific/Engineering :: Physics",
        "License :: OSI Approved :: MIT License"
    ],
    keywords=[
        "seismology", "seismic imaging", "computational seismology", "AI for science"
    ],
    install_requires=[
        "numpy", "scipy", "matplotlib", "torch"
    ],
    extras_require={
        "docs": ["sphinx", "ipython", "runipy"]
    },
    ext_modules=[]
)