"""
Simpl is an open source library for sinusoidal modelling written in C/C++ and
Python, and making use of Scientific Python (SciPy). The aim of this project
is to tie together many of the existing sinusoidal modelling implementations
into a single unified system with a consistent API, as well as providing
implementations of some recently published sinusoidal modelling algorithms,
many of which have yet to be released in software. Simpl is primarily intended
as a tool for other researchers in the field, allowing them to easily combine,
compare and contrast many of the published analysis/synthesis algorithms.
"""
import os
import glob
from distutils.core import setup
from distutils.extension import Extension
from Cython.Distutils import build_ext

# -----------------------------------------------------------------------------
# Global
# -----------------------------------------------------------------------------

# detect platform
platform = os.uname()[0] if hasattr(os, 'uname') else 'Windows'

# get numpy include directory
try:
    import numpy
    try:
        numpy_include = numpy.get_include()
    except AttributeError:
        numpy_include = numpy.get_numpy_include()
except ImportError:
    print "Error: Numpy was not found."
    exit(1)

macros = []
link_args = []
include_dirs = ['simpl', 'src/simpl', 'src/sms', 'src/sndobj',
                'src/loris', numpy_include, '/usr/local/include']
libs = ['m', 'fftw3', 'gsl', 'gslcblas']
sources = []

# -----------------------------------------------------------------------------
# SndObj Library
# -----------------------------------------------------------------------------

sndobj_sources = """
    SndObj.cpp SndIO.cpp FFT.cpp PVA.cpp IFGram.cpp SinAnal.cpp
    SinSyn.cpp AdSyn.cpp ReSyn.cpp HarmTable.cpp HammingTable.cpp
    """.split()
sndobj_sources = map(lambda x: 'src/sndobj/' + x, sndobj_sources)
sources.extend(sndobj_sources)

# -----------------------------------------------------------------------------
# SMS
# -----------------------------------------------------------------------------
sms_sources = """
    OOURA.c cepstrum.c peakContinuation.c soundIO.c tables.c
    fileIO.c peakDetection.c spectralApprox.c transforms.c
    filters.c residual.c spectrum.c windows.c SFMT.c fixTracks.c
    sineSynth.c stocAnalysis.c harmDetection.c sms.c synthesis.c
    analysis.c modify.c
    """.split()
sms_sources = map(lambda x: 'src/sms/' + x, sms_sources)
sources.extend(sms_sources)

# -----------------------------------------------------------------------------
# Loris
# -----------------------------------------------------------------------------
loris_sources = glob.glob(os.path.join('src', 'loris', '*.C'))
sources.extend(loris_sources)

# -----------------------------------------------------------------------------
# Base
# -----------------------------------------------------------------------------
base = Extension(
    "simpl.base",
    sources=["simpl/base.pyx", "src/simpl/base.cpp"],
    include_dirs=include_dirs,
    language="c++"
)

# -----------------------------------------------------------------------------
# Peak Detection
# -----------------------------------------------------------------------------
peak_detection = Extension(
    "simpl.peak_detection",
    sources=sources + ["simpl/peak_detection.pyx",
                       "src/simpl/peak_detection.cpp",
                       "src/simpl/base.cpp"],
    include_dirs=include_dirs,
    libraries=libs,
    extra_compile_args=['-DMERSENNE_TWISTER', '-DHAVE_FFTW3_H'],
    language="c++"
)

# -----------------------------------------------------------------------------
# Partial Tracking
# -----------------------------------------------------------------------------
partial_tracking = Extension(
    "simpl.partial_tracking",
    sources=["simpl/partial_tracking.pyx", "src/simpl/partial_tracking.cpp",
             "src/simpl/base.cpp"],
    include_dirs=include_dirs,
    language="c++"
)


# -----------------------------------------------------------------------------
# Synthesis
# -----------------------------------------------------------------------------
synthesis = Extension(
    "simpl.synthesis",
    sources=["simpl/synthesis.pyx", "src/simpl/synthesis.cpp",
             "src/simpl/base.cpp"],
    include_dirs=include_dirs,
    language="c++"
)


# -----------------------------------------------------------------------------
# Residual
# -----------------------------------------------------------------------------
residual = Extension(
    "simpl.residual",
    sources=["simpl/residual.pyx", "src/simpl/residual.cpp",
             "src/simpl/base.cpp"],
    include_dirs=include_dirs,
    language="c++"
)

# -----------------------------------------------------------------------------
# Package
# -----------------------------------------------------------------------------

doc_lines = __doc__.split("\n")

setup(
    name='simpl',
    description=doc_lines[0],
    long_description="\n".join(doc_lines[2:]),
    url='http://simplsound.sourceforge.net',
    download_url='http://simplsound.sourceforge.net',
    license='GPL',
    author='John Glover',
    author_email='j@johnglover.net',
    platforms=["Linux", "Mac OS-X", "Unix"],
    version='0.3',
    ext_modules=[base, peak_detection, partial_tracking, synthesis, residual],
    cmdclass={'build_ext': build_ext},
    packages=['simpl', 'simpl.plot']
)