diff options
Diffstat (limited to 'DSP/PythonWrapper/setup.py')
| -rw-r--r-- | DSP/PythonWrapper/setup.py | 85 |
1 files changed, 85 insertions, 0 deletions
diff --git a/DSP/PythonWrapper/setup.py b/DSP/PythonWrapper/setup.py new file mode 100644 index 0000000..7884518 --- /dev/null +++ b/DSP/PythonWrapper/setup.py @@ -0,0 +1,85 @@ +from distutils.core import setup, Extension +import glob +import numpy +import config +import sys +import os +from config import ROOT + +includes = [os.path.join(ROOT,"Include"),os.path.join("cmsisdsp_pkg","src")] + +if sys.platform == 'win32': + cflags = ["-DWIN",config.cflags,"-DUNALIGNED_SUPPORT_DISABLE"] + # Custom because a customized arm_math.h is required to build on windows + # since the visual compiler and the win platform are + # not supported by default in arm_math.h +else: + cflags = ["-Wno-unused-variable","-Wno-implicit-function-declaration",config.cflags] + +transform = glob.glob(os.path.join(ROOT,"Source","TransformFunctions","*.c")) +#transform.remove(os.path.join(ROOT,"Source","TransformFunctions","arm_dct4_init_q15.c")) +#transform.remove(os.path.join(ROOT,"Source","TransformFunctions","arm_rfft_init_q15.c")) +transform.remove(os.path.join(ROOT,"Source","TransformFunctions","TransformFunctions.c")) + +support = glob.glob(os.path.join(ROOT,"Source","SupportFunctions","*.c")) +support.remove(os.path.join(ROOT,"Source","SupportFunctions","SupportFunctions.c")) + +fastmath = glob.glob(os.path.join(ROOT,"Source","FastMathFunctions","*.c")) +fastmath.remove(os.path.join(ROOT,"Source","FastMathFunctions","FastMathFunctions.c")) + +filtering = glob.glob(os.path.join(ROOT,"Source","FilteringFunctions","*.c")) +filtering.remove(os.path.join(ROOT,"Source","FilteringFunctions","FilteringFunctions.c")) + +matrix = glob.glob(os.path.join(ROOT,"Source","MatrixFunctions","*.c")) +matrix.remove(os.path.join(ROOT,"Source","MatrixFunctions","MatrixFunctions.c")) + +statistics = glob.glob(os.path.join(ROOT,"Source","StatisticsFunctions","*.c")) +statistics.remove(os.path.join(ROOT,"Source","StatisticsFunctions","StatisticsFunctions.c")) + +complexf = glob.glob(os.path.join(ROOT,"Source","ComplexMathFunctions","*.c")) +complexf.remove(os.path.join(ROOT,"Source","ComplexMathFunctions","ComplexMathFunctions.c")) + +basic = glob.glob(os.path.join(ROOT,"Source","BasicMathFunctions","*.c")) +basic.remove(os.path.join(ROOT,"Source","BasicMathFunctions","BasicMathFunctions.c")) + +controller = glob.glob(os.path.join(ROOT,"Source","ControllerFunctions","*.c")) +controller.remove(os.path.join(ROOT,"Source","ControllerFunctions","ControllerFunctions.c")) + +common = glob.glob(os.path.join(ROOT,"Source","CommonTables","*.c")) +common.remove(os.path.join(ROOT,"Source","CommonTables","CommonTables.c")) + +#modulesrc = glob.glob(os.path.join("cmsisdsp_pkg","src","*.c")) +modulesrc = [] +modulesrc.append(os.path.join("cmsisdsp_pkg","src","fftinit.c")) +modulesrc.append(os.path.join("cmsisdsp_pkg","src","cmsismodule.c")) + +module1 = Extension(config.extensionName, + sources = (support + + fastmath + + filtering + + matrix + + statistics + + complexf + + basic + + controller + + transform + + modulesrc + + common + ) + , + include_dirs = includes + [numpy.get_include()], + #extra_compile_args = ["-Wno-unused-variable","-Wno-implicit-function-declaration",config.cflags] + extra_compile_args = cflags + ) + +setup (name = config.setupName, + version = '0.0.1', + description = config.setupDescription, + ext_modules = [module1], + author = 'Copyright (C) 2010-2019 ARM Limited or its affiliates. All rights reserved.', + url="https://github.com/ARM-software/CMSIS_5", + classifiers=[ + "Programming Language :: Python", + "License :: OSI Approved :: Apache Software License", + "Operating System :: OS Independent", + ])
\ No newline at end of file |