Official ARM version: v5.6.0HEADmaster

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