#!/usr/bin/env python

''' Installation script for dipy package '''

import os

import sys

from os.path import join as pjoin

from glob import glob

# BEFORE importing distutils, remove MANIFEST. distutils doesn't properly

# update it when the contents of directories change.

if os.path.exists('MANIFEST'): os.remove('MANIFEST')

import numpy as np

# For some commands, use setuptools

if len(set(('develop', 'bdist_egg', 'bdist_rpm', 'bdist', 'bdist_dumb',

'bdist_wininst', 'install_egg_info', 'egg_info', 'easy_install',

)).intersection(sys.argv)) > 0:

# setup_egg imports setuptools setup, thus monkeypatching distutils.

from setup_egg import extra_setuptools_args

# Import distutils _after_ potential setuptools import above, and after removing

# MANIFEST

from distutils.core import setup

from distutils.extension import Extension

# extra_setuptools_args can be defined from the line above, but it can

# also be defined here because setup.py has been exec'ed from

# setup_egg.py.

if not 'extra_setuptools_args' in globals():

extra_setuptools_args = dict()

# Import build helpers

try:

from nisext.sexts import package_check, get_comrec_build

except ImportError:

raise RuntimeError('Need nisext package from nibabel installation'

' - please install nibabel first')

cmdclass = {'build_py': get_comrec_build('dipy')}

# Get version and release info, which is all stored in dipy/info.py

ver_file = os.path.join('dipy', 'info.py')

execfile(ver_file)

# We're running via setuptools - specify exta setuptools stuff

if 'setuptools' in sys.modules:

extra_setuptools_args['extras_require'] = dict(

doc=['Sphinx>=1.0'],

test=['nose>=0.10.1'],

)

# I removed numpy and scipy from install requires because easy_install seems

# to want to fetch these if they are already installed, meaning of course

# that there's a long fragile and unnecessary compile before the install

# finishes.

extra_setuptools_args['install_requires'] = [

'nibabel>=' + NIBABEL_MIN_VERSION,

]

# Do our own install time dependency checking. The dependency checks in

# setuptools above go into the egg so are useful for easy_install. These checks

# below run whenever we run setup.py - so - install or build via setup.py

package_check('numpy', NUMPY_MIN_VERSION)

package_check('scipy', SCIPY_MIN_VERSION)

package_check('nibabel', NIBABEL_MIN_VERSION)

# Cython is a build dependency

def _cython_version(pkg_name):

from Cython.Compiler.Version import version

return version

package_check('cython',

CYTHON_MIN_VERSION,

version_getter=_cython_version)

# we use cython to compile the modules

from Cython.Distutils import build_ext

cmdclass['build_ext'] = build_ext

EXTS = []

for modulename, other_sources in (

('dipy.reconst.recspeed', []),

('dipy.tracking.distances', []),

('dipy.tracking.vox2track', []),

('dipy.tracking.propspeed', [])):

pyx_src = pjoin(*modulename.split('.')) + '.pyx'

EXTS.append(Extension(modulename,[pyx_src] + other_sources,

include_dirs = [np.get_include()]))

def main(**extra_args):

)

#simple way to test what setup will do

#python setup.py install --prefix=/tmp

if __name__ == "__main__":

main(**extra_setuptools_args)