adamstark@51: # setup.py
adamstark@51: # build command : python setup.py build build_ext --inplace
adamstark@51: from numpy.distutils.core import setup, Extension
adamstark@110: import os, numpy, platform
adamstark@110: 
adamstark@110: if platform.machine() == 'arm64':
adamstark@110:       include_path = '/opt/homebrew/include'
adamstark@110:       library_path = '/opt/homebrew/lib'
adamstark@110: else:
adamstark@110:       include_path = '/usr/local/include'
adamstark@110:       library_path = '/usr/local/lib'
adamstark@51: 
adamstark@51: name = 'btrack'
adamstark@110: sources = [
adamstark@110:       'btrack_python_module.cpp',
adamstark@110:       '../../src/OnsetDetectionFunction.cpp',
adamstark@110:       '../../src/BTrack.cpp'
adamstark@110:       ]
adamstark@51: 
adamstark@93: sources.append ('../../libs/kiss_fft130/kiss_fft.c')
adamstark@93: 
adamstark@51: include_dirs = [
adamstark@110:       numpy.get_include(),
adamstark@110:       include_path
adamstark@110: ]
adamstark@51: 
adamstark@93: include_dirs.append ('../../libs/kiss_fft130')
adamstark@93: 
adamstark@110: setup(name = 'BTrack',
adamstark@51:       include_dirs = include_dirs,
adamstark@110:       ext_modules = [Extension(name, sources,libraries = ['fftw3','samplerate'],library_dirs = [library_path],define_macros=[
adamstark@94:                          ('USE_FFTW', None)])]
adamstark@51:       )