#!/usr/bin/env python

# PyLPSolve is an object oriented wrapper for the open source LP

# solver lp_solve. Copyright (C) 2010 Hoyt Koepke.

#

# This library is free software; you can redistribute it and/or modify

# it under the terms of the GNU Lesser General Public License as

# published by the Free Software Foundation; either version 2.1 of

# the License, or (at your option) any later version.

#

# This library is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU

# Lesser General Public License for more details.

#

# You should have received a copy of the GNU Lesser General Public

# License along with this library; if not, write to the Free Software

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

################################################################################

# All the control parameters should go here

debug_mode_c_code = False

version = "0.11"

description="PyLPSolve: Object-oriented wrapper for the lpsolve linear programming solver."

author = "Hoyt Koepke"

author_email="hoytak@gmail.com"

name = 'pylpsolve'

scripts = []

url = ""

download_url = ""

long_description = \

"""

classifiers = [

]

source_directory_list = ['pylpsolve']

lpsolve_base = 'pylpsolve/lp_solve_5.5'

# Stuff for extension module stuff

compiler_args = ['-O3', '-DYY_NEVER_INTERACTIVE','-DPARSER_LP', '-DINVERSE_ACTIVE=INVERSE_LUSOL', '-DRoleIsExternalInvEngine']

link_args = ['-O3']

from glob import glob

from os.path import join

extra_library_dirs = []

extra_include_dirs = [join(lpsolve_base, d) for d in ['.', 'shared', 'bfp', 'bfp/bfp_LUSOL', 'bfp/bfp_LUSOL/LUSOL', 'colamd']]

library_includes = []

specific_libraries = {}

extra_sources = {'pylpsolve.pylpsolve' : [join(lpsolve_base, f) for f in

'lp_simplex.c', 'lp_SOS.c', 'lp_utils.c', 'yacc_read.c']]}

################################################################################

# Shouldn't have to adjust anything below this line...

import os

from os.path import split, join

from itertools import chain

import sys

from setuptools import setup, Extension

######################################################

# First have to see if we're authorized to use cython files, or if we

# should instead compile the included files

if "--cython" in sys.argv:

cython_mode = True

del sys.argv[sys.argv.index("--cython")]

else:

cython_mode = False

# Get all the cython files in the sub directories and in this directory

if cython_mode:

cython_files = dict( (d, glob(join(d, "*.pyx"))) for d in source_directory_list + ['.'])

else:

cython_files = {}

all_cython_files = set(chain(*cython_files.values()))

print "+++++++++++++++++++"

if cython_mode:

print "Cython Files Found: \n%s\n+++++++++++++++++++++" % ", ".join(sorted(all_cython_files))

else:

print "Cython support disabled; compiling extensions from pregenerated C sources."

print "To enable cython, run setup.py with the option --cython."

print "+++++++++++++++++++"

# Set the compiler arguments -- Add in the environment path stuff

ld_library_path = os.getenv("LD_LIBRARY_PATH")

if ld_library_path is not None:

lib_paths = ld_library_path.split(":")

else:

lib_paths = []

include_path = os.getenv("INCLUDE_PATH")

if include_path is not None:

include_paths = [p.strip() for p in include_path.split(":") if len(p.strip()) > 0]

else:

include_paths = []

# get all the c files that are not cythonized .pyx files.

c_files = dict( (d, [f for f in glob(join(d, "*.c"))

if (f[:-2] + '.pyx') not in all_cython_files])

for d in source_directory_list + ['.'])

for d, l in chain(((d, glob(join(d, "*.cxx"))) for d in source_directory_list + ['.']),

((d, glob(join(d, "*.cpp"))) for d in source_directory_list + ['.'])):

c_files[d] += l

print "C Extension Files Found: \n%s\n+++++++++++++++++++++" % ", ".join(sorted(chain(*c_files.values())))

# Collect all the python modules

def get_python_modules(f):

d, m = split(f[:f.rfind('.')])

return m if len(d) == 0 else d + "." + m

exclude_files = set(["setup.py"])

python_files = set(chain(* (list(glob(join(d, "*.py")) for d in source_directory_list) + [glob("*.py")])))

python_files -= exclude_files

python_modules = [get_python_modules(f) for f in python_files]

print "Relevant Python Files Found: \n%s\n+++++++++++++++++++++" % ", ".join(sorted(python_files))

if __name__ == '__main__':

# The rest is also shared with the setup.py file, in addition to

# this one, so

def get_include_dirs(m):

include_dir_list = extra_include_dirs + include_paths

stripped = (d.strip() for d in include_dir_list)

return [d for d in stripped if d]

def get_library_dirs(m):

stripped = (d.strip() for d in extra_library_dirs + lib_paths)

return [d for d in stripped if d]

def get_libraries(m):

return library_includes + (specific_libraries[m] if m in specific_libraries else [])

def get_extra_compile_args(m):

return compiler_args + (['-g', '-O0', '-DCYTHON_REFNANNY'] if debug_mode_c_code else [])

def get_extra_link_args(m):

return link_args + (['-g'] if debug_mode_c_code else [])

def get_extra_source_files(m):

return extra_sources[m] if m in extra_sources else []

############################################################

# Cython extension lists

def makeExtensionList(d, filelist):

return ext_modules

############################################################

# Now get all these ready to go

ext_modules = []

if cython_mode:

from Cython.Distutils import build_ext as _build_ext

ext_modules += list(chain(*list(makeExtensionList(d, l)

for d, l in cython_files.iteritems())))

else:

from setuptools.command.build_ext import build_ext as _build_ext

cmdclass = {}

class build_ext(_build_ext):

self.include_dirs.append(numpy.get_include())

ext_modules += list(chain(*list(makeExtensionList(d, l)

for d, l in c_files.iteritems())))

setup(

version = version,

description = description,

setup_requires=['numpy>=1.9,<2'],

install_requires=['numpy>=1.9,<2'],

author = author,

author_email = author_email,

name = name,

cmdclass={'build_ext' : build_ext},

ext_modules = ext_modules,

py_modules = python_modules,

scripts = scripts,

classifiers = classifiers,

url = url,

download_url = download_url

)