CommLib/setup.py at master · DavidKWH/CommLib · 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
from setuptools import setup, find_packages
from setuptools.extension import Extension
from Cython.Build import cythonize
import numpy

# define extension modules
decoder_r1_2 = Extension('comm_ai.ldpc.ldpc_decoder_r1_2',
                   sources=['comm_ai/ldpc/ldpc_decoder_r1_2.pyx',
                            'comm_ai/ldpc/layered_ldpc_dec.c'],
                   define_macros=[('USE_CNODE_FCN','1'),
                                  ('CODE_RATE','1')],
                   libraries=['m']) # -lm
decoder_r5_8 = Extension('comm_ai.ldpc.ldpc_decoder_r5_8',
                   sources=['comm_ai/ldpc/ldpc_decoder_r5_8.pyx',
                            'comm_ai/ldpc/layered_ldpc_dec.c'],
                   define_macros=[('USE_CNODE_FCN','1'),
                                  ('CODE_RATE','2')],
                   libraries=['m']) # -lm
decoder_r3_4 = Extension('comm_ai.ldpc.ldpc_decoder_r3_4',
                   sources=['comm_ai/ldpc/ldpc_decoder_r3_4.pyx',
                            'comm_ai/ldpc/layered_ldpc_dec.c'],
                   define_macros=[('USE_CNODE_FCN','1'),
                                  ('CODE_RATE','3')],
                   libraries=['m']) # -lm
decoder_r13_16 = Extension('comm_ai.ldpc.ldpc_decoder_r13_16',
                   sources=['comm_ai/ldpc/ldpc_decoder_r13_16.pyx',
                            'comm_ai/ldpc/layered_ldpc_dec.c'],
                   define_macros=[('USE_CNODE_FCN','1'),
                                  ('CODE_RATE','4')],
                   libraries=['m']) # -lm
#min_sum_decoder = Extension('comm_ai.ldpc.ldpc_decoder_oms',
#                   sources=['comm_ai/ldpc/ldpc_decoder_oms.pyx',
#                            'comm_ai/ldpc/layered_ldpc_dec.c'],
#                   define_macros=[('USE_CNODE_FCN','1')],
#                   libraries=['m']) # -lm
#bp_decoder      = Extension('comm_ai.ldpc.ldpc_decoder_bp',
#                   sources=['comm_ai/ldpc/ldpc_decoder_bp.pyx',
#                            'comm_ai/ldpc/layered_ldpc_dec.c'],
#                   define_macros=[('USE_CNODE_FCN','2')],
#                   libraries=['m']) # -lm

extensions = [decoder_r1_2, decoder_r5_8, decoder_r3_4, decoder_r13_16]
#extensions = [min_sum_decoder, bp_decoder]

setup(name='comm_ai',
      version='0.3.0',
      packages=find_packages(),
      package_data={'': ['*.txt']},
      scripts=['bin/rclone.py'],
      install_requires=["numpy",
                        "Cython",
                        "importlib_resources ; python_version<'3.7'",
                        "google-api-python-client", # google-drive
                        "google-api-core",
                        "httplib2",
                        "oauth2client",
                        "jsonpickle",
                        "redis", # remote queue storage
                        "psutil",
                        "hotqueue"],
      ext_modules = cythonize(extensions, language_level=3),
      include_dirs=[numpy.get_include()],
      #license='LICENSE.txt',
      description='Physical layer communications package',
      #long_description=open('README.txt').read(),
      #url='http://pypi.python.org/pypi/CommLib/',
      author='David K. W. Ho',
      author_email='davidkwho@gmail.com',
      classifiers=[
          "Programming Language :: Python :: 3",
          "License :: OSI Approved :: MIT License",
          "Operating System :: POSIX :: Linux",
      ],
      )