Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #3240 from arianepaola/feature/setup_py
Shogun Python package using setup.py
- Loading branch information
Showing
1 changed file
with
333 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,333 @@ | ||
#!/usr/bin/env python | ||
# -*- coding: utf-8 -*- | ||
|
||
# | ||
# Copyright (c) 2016, Shogun Toolbox Foundation | ||
# All rights reserved. | ||
# | ||
# Redistribution and use in source and binary forms, with or without | ||
# modification, are permitted provided that the following conditions are met: | ||
# 1. Redistributions of source code must retain the above copyright notice, | ||
# this list of conditions and the following disclaimer. | ||
# | ||
# 2. Redistributions in binary form must reproduce the above copyright notice, | ||
# this list of conditions and the following disclaimer in the documentation | ||
# and/or other materials provided with the distribution. | ||
# | ||
# 3. Neither the name of the copyright holder nor the names of its | ||
# contributors may be used to endorse or promote products derived from this | ||
# software without specific prior written permission. | ||
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" | ||
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE | ||
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE | ||
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE | ||
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR | ||
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF | ||
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS | ||
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN | ||
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) | ||
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE | ||
# POSSIBILITY OF SUCH DAMAGE. | ||
# | ||
# Written (W) 2016 Ariane Paola Gomes | ||
# | ||
|
||
from setuptools import find_packages, setup as shogun_setup | ||
import distutils.command.build | ||
import setuptools.command.install | ||
import distutils.spawn | ||
import glob | ||
import re | ||
import os | ||
import sys | ||
|
||
setup_py_location = os.path.abspath(os.path.dirname(__file__)) | ||
|
||
shogun_build_directory = os.path.join(setup_py_location, 'build') | ||
shogun_generated_install = os.path.join(shogun_build_directory, 'install') | ||
shogun_versionstring_h = os.path.abspath('src/shogun/lib/versionstring.h') | ||
shogun_python_packages_location = None | ||
|
||
shogun_completed_cmake = False | ||
shogun_completed_make = False | ||
shogun_completed_make_install = False | ||
show_debug_information = True | ||
|
||
with open(os.path.join(setup_py_location, 'README.md')) as f: | ||
readme = f.read() | ||
|
||
|
||
def shogun_bootstrap(): | ||
global shogun_completed_cmake | ||
global shogun_completed_make | ||
global shogun_completed_make_install | ||
global shogun_python_packages_location | ||
|
||
print("Bootstrapping Shogun") | ||
|
||
if shogun_cmake(): | ||
shogun_completed_cmake = True | ||
else: | ||
print('Error running CMake') | ||
|
||
if shogun_make(): | ||
shogun_completed_make = True | ||
else: | ||
print('Error running make') | ||
|
||
if shogun_make_install(): | ||
shogun_completed_make_install = True | ||
else: | ||
print('Error running make install') | ||
|
||
if shogun_preconditions_met(): | ||
print("Setting Shogun Python packages location") | ||
shogun_python_packages_location = glob.glob(os.path.join(shogun_generated_install, 'lib/*/dist-packages'))[0] | ||
|
||
print("Bootstrapping Shogun successfully completed!") | ||
|
||
shogun_debug_information() | ||
else: | ||
print("Shogun bootstrapping failed!") | ||
print("Please make sure that you have cmake and make installed.") | ||
|
||
sys.exit() | ||
|
||
|
||
def shogun_preconditions_met(): | ||
print("Verifying Shogun preconditions") | ||
|
||
if show_debug_information: | ||
print("Shogun build environment completed tasks: cmake: [%s] - make: [%s] - make install: [%s]" % (shogun_completed_cmake, shogun_completed_make, shogun_completed_make_install)) | ||
|
||
return shogun_completed_cmake and shogun_completed_make and shogun_completed_make_install | ||
|
||
|
||
def shogun_preconditions(): | ||
if not shogun_preconditions_met(): | ||
shogun_bootstrap() | ||
|
||
return shogun_preconditions_met() | ||
|
||
|
||
def shogun_debug_information(): | ||
if show_debug_information: | ||
print("Shogun generated installation location %s" % shogun_generated_install) | ||
print("Shogun Python package location: %s" % shogun_python_packages_location) | ||
print("Shogun version string location: %s" % shogun_versionstring_h) | ||
|
||
|
||
def parse_shogun_version(version_header): | ||
shogun_version_pattern = re.compile(ur'#define MAINVERSION \"([0-9]\.[0-9]\.[0-9])\"') | ||
|
||
with open(version_header, 'r') as f: | ||
content = f.read() | ||
|
||
matches = re.findall(shogun_version_pattern, content) | ||
|
||
if len(matches): | ||
return matches[0] | ||
else: | ||
return 'undefined' | ||
|
||
|
||
def shogun_cmake(arguments=None): | ||
print("Running CMake") | ||
|
||
if arguments is None: | ||
arguments='-DPythonModular=ON -DENABLE_TESTING=OFF -DCMAKE_INSTALL_PREFIX=install' | ||
|
||
if distutils.spawn.find_executable('cmake') is not None: | ||
print('CMake arguments: %s ' % arguments) | ||
print('Creating build directory: %s' % shogun_build_directory) | ||
distutils.dir_util.mkpath(shogun_build_directory) | ||
|
||
os.chdir(shogun_build_directory) | ||
|
||
try: | ||
distutils.spawn.spawn(['cmake'] + arguments.split() + ['..']) | ||
|
||
except distutils.spawn.DistutilsExecError: | ||
print('CMake error.') | ||
return False | ||
|
||
finally: | ||
os.chdir(os.path.abspath(setup_py_location)) | ||
|
||
return True | ||
|
||
else: | ||
print('CMake is required to build shogun!') | ||
return False | ||
|
||
|
||
def shogun_make(arguments=None): | ||
print("Running make") | ||
|
||
if arguments is None: | ||
arguments='all' | ||
|
||
if distutils.spawn.find_executable('make') is not None: | ||
print('make arguments: %s ' % arguments) | ||
|
||
os.chdir(shogun_build_directory) | ||
|
||
try: | ||
distutils.spawn.spawn(['make'] + arguments.split()) | ||
|
||
except distutils.spawn.DistutilsExecError: | ||
print('make error.') | ||
return False | ||
|
||
finally: | ||
os.chdir(os.path.abspath(setup_py_location)) | ||
|
||
return True | ||
|
||
else: | ||
print('make is required to build shogun!') | ||
return False | ||
|
||
|
||
def shogun_make_install(): | ||
return shogun_make(arguments='install') | ||
|
||
|
||
def get_shogun_version(): | ||
print("Retrieving Shogun version") | ||
|
||
if shogun_preconditions(): | ||
shogun_version = parse_shogun_version(shogun_versionstring_h) | ||
|
||
if show_debug_information: | ||
print('The Shogun version is %s' % shogun_version) | ||
|
||
return shogun_version | ||
|
||
|
||
def python_package_path(package_path): | ||
print("Generating destination Python package path") | ||
directories = package_path.split(os.sep) | ||
destination_path = os.path.join(directories[-2], directories[-1]) | ||
|
||
if show_debug_information: | ||
print("Shogun destination Python package path: %s" % destination_path) | ||
|
||
return destination_path | ||
|
||
|
||
def shogun_packages(): | ||
if not shogun_preconditions_met(): | ||
shogun_bootstrap() | ||
|
||
return find_packages(where=shogun_python_packages_location) | ||
|
||
|
||
def shogun_package_directories(): | ||
package_directories = dict() | ||
|
||
if not shogun_preconditions_met(): | ||
shogun_bootstrap() | ||
|
||
package_directories[''] = shogun_python_packages_location | ||
|
||
return package_directories | ||
|
||
|
||
def shogun_data_files(): | ||
data_files = list() | ||
libshogun_files = glob.glob(os.path.join(shogun_generated_install, 'lib/libshogun*')) | ||
modshogun_so_destination = os.path.join('lib', python_package_path(shogun_python_packages_location)) | ||
modshogun_so_file = glob.glob(os.path.join(shogun_python_packages_location, '_modshogun.so'))[0] | ||
|
||
# appending data files | ||
data_files.append(('lib', libshogun_files)) | ||
data_files.append((modshogun_so_destination, [modshogun_so_file])) | ||
|
||
if show_debug_information: | ||
print('Shogun Python package data files:') | ||
for data_file_content in data_files: | ||
print('|->[%s]' % data_file_content[0]) | ||
|
||
for data_file in data_file_content[1]: | ||
print(' |--> %s' % os.path.basename(data_file)) | ||
|
||
return data_files | ||
|
||
|
||
# https://docs.python.org/2/distutils/apiref.html#creating-a-new-distutils-command | ||
class ShogunBuild(distutils.command.build.build): | ||
user_options = distutils.command.build.build.user_options + [('cmake=', None, 'Specify CMake arguments.')] | ||
build_base = '' | ||
build_lib = '' | ||
build_scripts = '' | ||
plat_name = '' | ||
|
||
def initialize_options(self): | ||
self.cmake = None | ||
|
||
def finalize_options(self): | ||
pass | ||
|
||
def run(self): | ||
print('Running Package build') | ||
|
||
if not shogun_preconditions_met: | ||
shogun_cmake(self.cmake) | ||
shogun_make() | ||
shogun_make_install() | ||
|
||
# Command.sub_commands | ||
|
||
|
||
class ShogunInstall(setuptools.command.install.install): | ||
def run(self): | ||
print('Running Package install') | ||
|
||
if shogun_preconditions(): | ||
self.do_egg_install() | ||
|
||
|
||
shogun_setup( | ||
name = "shogun", | ||
version = get_shogun_version(), | ||
|
||
description = 'The Shogun Machine Learning Toolbox', | ||
long_description = readme, | ||
url = 'http://www.shogun-toolbox.org/', | ||
|
||
author = 'Ariane Paola Gomes', | ||
author_email = 'arianepaola@users.noreply.github.com', | ||
|
||
license = 'The GNU General Public License v3.0', | ||
|
||
# https://pypi.python.org/pypi?%3Aaction=list_classifiers | ||
classifiers = [ | ||
'Development Status :: 5 - Production/Stable', | ||
'Intended Audience :: Developers', | ||
'Intended Audience :: Education', | ||
'Intended Audience :: Information Technology', | ||
'Intended Audience :: Science/Research', | ||
'Topic :: Software Development :: Libraries', | ||
# Python 2 and Python 3 support | ||
'Programming Language :: Python :: 2', | ||
'Programming Language :: Python :: 3' | ||
], | ||
|
||
keywords = 'Unified and efficient Machine Learning', | ||
|
||
zip_safe = False, | ||
|
||
# Shogun bootstrapping build and install | ||
cmdclass = {'build': ShogunBuild, 'install': ShogunInstall}, | ||
|
||
# Shogun package content | ||
packages = shogun_packages(), | ||
package_dir = shogun_package_directories(), | ||
py_modules =['modshogun'], | ||
data_files = shogun_data_files(), | ||
|
||
# Shogun dependencies | ||
install_requires = ['numpy'] | ||
) | ||
|