forked from CPJKU/madmom
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
97 lines (84 loc) · 3.3 KB
/
setup.py
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
#!/usr/bin/env python
# encoding: utf-8
"""
This file contains the setup for setuptools to distribute everything as a
(PyPI) package.
"""
from setuptools import setup, find_packages
from distutils.extension import Extension
from Cython.Build import cythonize, build_ext
import glob
import numpy as np
# define version
version = '0.17.dev0'
# define which extensions to compile
include_dirs = [np.get_include()]
extensions = [
Extension('madmom.audio.comb_filters', ['madmom/audio/comb_filters.pyx'],
include_dirs=include_dirs),
Extension('madmom.features.beats_crf', ['madmom/features/beats_crf.pyx'],
include_dirs=include_dirs),
Extension('madmom.ml.hmm', ['madmom/ml/hmm.pyx'],
include_dirs=include_dirs),
Extension('madmom.ml.nn.layers', ['madmom/ml/nn/layers.py'],
include_dirs=include_dirs),
]
# define scripts to be installed by the PyPI package
scripts = glob.glob('bin/*')
# define the models to be included in the PyPI package
package_data = ['models/LICENSE',
'models/README.rst',
'models/beats/201[56]/*',
'models/chords/*/*',
'models/chroma/*/*',
'models/downbeats/*/*',
'models/key/2018/*',
'models/notes/*/*',
'models/onsets/*/*',
'models/patterns/*/*',
]
# some PyPI metadata
classifiers = ['Development Status :: 3 - Alpha',
'Programming Language :: Python :: 2.7',
'Programming Language :: Python :: 3.5',
'Programming Language :: Python :: 3.6',
'Programming Language :: Python :: 3.7',
'Environment :: Console',
'License :: OSI Approved :: BSD License',
'License :: Free for non-commercial use',
'Topic :: Multimedia :: Sound/Audio :: Analysis',
'Topic :: Scientific/Engineering :: Artificial Intelligence']
# requirements
requirements = ['numpy>=1.13.4',
'scipy>=0.16',
'cython>=0.25',
'mido>=1.2.6',
]
# docs to be included
try:
long_description = open('README.rst', encoding='utf-8').read()
long_description += '\n' + open('CHANGES.rst', encoding='utf-8').read()
except TypeError:
long_description = open('README.rst').read()
long_description += '\n' + open('CHANGES.rst').read()
# the actual setup routine
setup(name='madmom',
version=version,
description='Python audio signal processing library',
long_description=long_description,
author='Department of Computational Perception, Johannes Kepler '
'University, Linz, Austria and Austrian Research Institute for '
'Artificial Intelligence (OFAI), Vienna, Austria',
author_email='[email protected]',
url='https://github.com/CPJKU/madmom',
license='BSD, CC BY-NC-SA',
packages=find_packages(exclude=['tests', 'docs']),
ext_modules=cythonize(extensions),
package_data={'madmom': package_data},
exclude_package_data={'': ['tests', 'docs']},
scripts=scripts,
install_requires=requirements,
cmdclass={'build_ext': build_ext},
setup_requires=['pytest-runner'],
tests_require=['pytest'],
classifiers=classifiers)