forked from PythonOptimizers/NLP.py
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
120 lines (96 loc) · 3.8 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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
#!/usr/bin/env python
"""NLP.py: A Nonlinear Programming Environment in Python.
NLP.py is a programming environment that facilitates construction of
optimization algorithms by supplying a library of efficient building blocks.
D. Orban <dominique.orban@gerad.ca>
S. Arreckx <sylvain.arreckx@gmail.com>
"""
import os
import glob
from setuptools import setup # enables 'python setup.py develop'
from distutils.extension import Extension
import ConfigParser
import numpy as np
DOCLINES = __doc__.split("\n")
nlp_config = ConfigParser.SafeConfigParser()
nlp_config.read('site.cfg')
nlp_ext = []
# Optional AMPL extension
if nlp_config.has_section('ASL'):
# Read relevant AMPL-specific configuration options.
libampl_dir = nlp_config.get('ASL', 'asl_dir') # .split(os.pathsep)
libampl_libdir = os.path.join(libampl_dir, 'lib')
libampl_include = os.path.join(libampl_dir, os.path.join('include', 'asl'))
if not (os.path.isdir(libampl_libdir) and os.path.isdir(libampl_include)):
raise ValueError("Please check if ASL paths are correct:" +
"\n %s\n %s" % (libampl_libdir, libampl_include))
amplmodel_params = {}
amplmodel_params['library_dirs'] = [libampl_libdir]
amplmodel_params['include_dirs'] = [os.path.join('nlp', 'model', 'src'),
libampl_include,
np.get_include()]
amplmodel_params['libraries'] = ['asl']
amplmodel_src = [os.path.join('nlp', 'model', 'src', '_amplmodel.c'),
os.path.join('nlp', 'model', 'src', 'amplutils.c')]
nlp_ext.append(Extension(name="nlp.model._amplmodel",
sources=amplmodel_src,
**amplmodel_params))
#############################
# Strong Wolfe linesearches #
#############################
swls_params = {}
swls_params['include_dirs'] = [np.get_include()]
swls_src = [os.path.join('nlp', 'ls', 'src', '_strong_wolfe_linesearch.c')]
nlp_ext.append(Extension(name="nlp.ls._strong_wolfe_linesearch",
sources=swls_src,
**swls_params))
mswls_params = {}
mswls_params['include_dirs'] = [np.get_include()]
mswls_src = [os.path.join('nlp', 'ls', 'src',
'_modified_strong_wolfe_linesearch.c')]
nlp_ext.append(Extension(name="nlp.ls._modified_strong_wolfe_linesearch",
sources=mswls_src,
**mswls_params))
CLASSIFIERS = """\
Development Status :: 4 - Beta
Intended Audience :: Science/Research
Intended Audience :: Developers
License :: LGPL
Programming Language :: Python
Topic :: Software Development
Topic :: Scientific/Engineering
Operating System :: Microsoft :: Windows
Operating System :: POSIX
Operating System :: Unix
Operating System :: MacOS
"""
packages_list = ['nlp',
'nlp.model',
'nlp.ls',
'nlp.optimize',
'nlp.tools',
'nlp.tr']
scripts_list = glob.glob(os.path.join('nlp', 'drivers', 'nlp_*.py'))
setup(
name='nlp',
version="0.0.1",
author="Dominique Orban",
author_email="dominique.orban@gmail.com",
maintainer="NLP.py Developers",
maintainer_email="dominique.orban@gmail.com",
description=DOCLINES[0],
long_description="\n".join(DOCLINES[2:]),
url="https://github.com/PythonOptimizers/NLP.py.git",
download_url="https://github.com/PythonOptimizers/NLP.py.git",
license='LGPL',
platforms=["Windows", "Linux", "Solaris", "Mac OS-X", "Unix"],
classifiers=filter(None, CLASSIFIERS.split('\n')),
install_requires=['numpy'],
ext_modules=nlp_ext,
package_dir={"nlp": "nlp"},
packages=packages_list,
scripts=scripts_list,
tests_require=['pytest'],
setup_requires=['pytest-runner'],
zip_safe=False
)