GPy/setup.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

#===============================================================================
# Copyright (c) 2012 - 2014, GPy authors (see AUTHORS.txt).
# Copyright (c) 2014, James Hensman, Max Zwiessele
# Copyright (c) 2015, Max Zwiessele
#
# All rights reserved.
# 
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
# 
# * Redistributions of source code must retain the above copyright notice, this
#   list of conditions and the following disclaimer.
# 
# * 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.
# 
# * Neither the name of GPy 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.
#===============================================================================

from __future__ import print_function
import os
import sys
from setuptools import setup, Extension
import numpy as np


def read(fname):
    try:
        with open(os.path.join(os.path.dirname(__file__), fname)) as f:
            return unicode(f.read(), 'utf-8')
    except NameError:
        #python 3
        with open(os.path.join(os.path.dirname(__file__), fname)) as f:
            return f.read()

def read_to_rst(fname):
    try:
        import pypandoc
        #print 'Warning in installation: For rst formatting in pypi, consider installing pypandoc for conversion'
        with open('README.rst', 'w') as f:
            f.write(pypandoc.convert('README.md', 'rst'))
    except ImportError:
        return read(fname)

version_dummy = {}
exec(read('GPy/__version__.py'), version_dummy)
__version__ = version_dummy['__version__']
del version_dummy

#Mac OS X Clang doesn't support OpenMP at the current time.
#This detects if we are building on a Mac
def ismac():
    return sys.platform[:6] == 'darwin'

if ismac():
    compile_flags = [ '-O3', ]
    link_args = []
else:
    compile_flags = [ '-fopenmp', '-O3', ]
    link_args = ['-lgomp']

ext_mods = [Extension(name='GPy.kern._src.stationary_cython',
                      sources=['GPy/kern/_src/stationary_cython.c',
                               'GPy/kern/_src/stationary_utils.c'],
                      include_dirs=[np.get_include(),'.'],
                      extra_compile_args=compile_flags,
                      extra_link_args = link_args),
            Extension(name='GPy.util.choleskies_cython',
                      sources=['GPy/util/choleskies_cython.c'],
                      include_dirs=[np.get_include(),'.'],
                      extra_link_args = link_args,
                      extra_compile_args=compile_flags),
            Extension(name='GPy.util.linalg_cython',
                      sources=['GPy/util/linalg_cython.c'],
                      include_dirs=[np.get_include(),'.'],
                      extra_compile_args=compile_flags),
            Extension(name='GPy.kern._src.coregionalize_cython',
                      sources=['GPy/kern/_src/coregionalize_cython.c'],
                      include_dirs=[np.get_include(),'.'],
                      extra_compile_args=compile_flags)]

setup(name = 'GPy',
      version = __version__,
      author = read('AUTHORS.txt'),
      author_email = "gpy.authors@gmail.com",
      description = ("The Gaussian Process Toolbox"),
      license = "BSD 3-clause",
      keywords = "machine-learning gaussian-processes kernels",
      url = "http://sheffieldml.github.com/GPy/",
      ext_modules = ext_mods,
      packages = ["GPy",
                  "GPy.core",
                  "GPy.core.parameterization", 
                  "GPy.kern",
                  "GPy.kern._src",
                  "GPy.kern._src.psi_comp", 
                  "GPy.models",
                  "GPy.inference",
                  "GPy.inference.optimization",
                  "GPy.inference.mcmc",
                  "GPy.inference.latent_function_inference",
                  "GPy.likelihoods", 
                  "GPy.mappings",
                  "GPy.examples",
                  "GPy.testing",
                  "GPy.util", 
                  "GPy.plotting",
                  "GPy.plotting.gpy_plot",
                  "GPy.plotting.matplot_dep", 
                  "GPy.plotting.matplot_dep.controllers",
                  "GPy.plotting.plotly_dep", 
                  ],
      package_dir={'GPy': 'GPy'},
      package_data = {'GPy': ['defaults.cfg', 'installation.cfg',
                              'util/data_resources.json',
                              'util/football_teams.json',
                              'plotting/plotting_tests/baseline/*.png'
                              ]},
      include_package_data = True,
      py_modules = ['GPy.__init__'],
      test_suite = 'GPy.testing',
      long_description=read_to_rst('README.md'),
      install_requires=['numpy>=1.7', 'scipy>=0.16', 'six'],
      extras_require = {'docs':['matplotlib >=1.3','Sphinx','IPython'],'optional':['mpi4py']},
      classifiers=['License :: OSI Approved :: BSD License',
                   'Natural Language :: English',
                   'Operating System :: MacOS :: MacOS X',
                   'Operating System :: Microsoft :: Windows',
                   'Operating System :: POSIX :: Linux',
                   'Programming Language :: Python :: 2.7',
                   'Programming Language :: Python :: 3.3',
                   'Programming Language :: Python :: 3.4',
                   'Programming Language :: Python :: 3.5',
                   ]
      )


# Check config files and settings:
local_file = os.path.abspath(os.path.join(os.path.dirname(__file__), 'GPy', 'installation.cfg'))
home = os.getenv('HOME') or os.getenv('USERPROFILE')
user_file = os.path.join(home,'.config', 'GPy', 'user.cfg')

print("")
if not os.path.exists(user_file):
    # Does an old config exist?
    old_user_file = os.path.join(home,'.gpy_user.cfg')
    if os.path.exists(old_user_file):
        # Move it to new location:
        print("GPy: Found old config file, moving to new location {}".format(user_file))
        os.rename(old_user_file, user_file)                
    else:
        # No config file exists, save informative stub to user config folder:
        print("GPy: Saving user configuration file to {}".format(user_file))
        if not os.path.exists(os.path.dirname(user_file)):
            os.makedirs(os.path.dirname(user_file))
        with open(user_file, 'w') as f:
            with open(local_file, 'r') as l:
                tmp = l.read()
                f.write(tmp)
else:
    print("GPy: User configuration file at location {}".format(user_file))