#from distutils.core import setup, Extension from setuptools import setup, Extension,find_packages from distutils.util import convert_path import glob import numpy import sys import os import os.path import re import pathlib here = pathlib.Path(__file__).parent.resolve() ROOT = here ROOT="" includes = [os.path.join(ROOT,"Include"),os.path.join(ROOT,"PrivateInclude"),os.path.join("PythonWrapper","cmsisdsp_pkg","src")] if sys.platform == 'win32': cflags = ["-DWIN","-DCMSISDSP","-DUNALIGNED_SUPPORT_DISABLE"] else: cflags = ["-Wno-attributes","-Wno-unused-function","-Wno-unused-variable","-Wno-implicit-function-declaration","-DCMSISDSP","-D__GNUC_PYTHON__"] transform = glob.glob(os.path.join(ROOT,"Source","TransformFunctions","*.c")) # Files are present when creating the source distribution # but they are not copied to the source distribution # When doing pip install those files are not prevent # and it should not fail try: transform.remove(os.path.join(ROOT,"Source","TransformFunctions","TransformFunctions.c")) transform.remove(os.path.join(ROOT,"Source","TransformFunctions","TransformFunctionsF16.c")) except: pass support = glob.glob(os.path.join(ROOT,"Source","SupportFunctions","*.c")) try: support.remove(os.path.join(ROOT,"Source","SupportFunctions","SupportFunctions.c")) support.remove(os.path.join(ROOT,"Source","SupportFunctions","SupportFunctionsF16.c")) except: pass fastmath = glob.glob(os.path.join(ROOT,"Source","FastMathFunctions","*.c")) try: fastmath.remove(os.path.join(ROOT,"Source","FastMathFunctions","FastMathFunctions.c")) except: pass filtering = glob.glob(os.path.join(ROOT,"Source","FilteringFunctions","*.c")) try: filtering.remove(os.path.join(ROOT,"Source","FilteringFunctions","FilteringFunctions.c")) filtering.remove(os.path.join(ROOT,"Source","FilteringFunctions","FilteringFunctionsF16.c")) except: pass matrix = glob.glob(os.path.join(ROOT,"Source","MatrixFunctions","*.c")) try: matrix.remove(os.path.join(ROOT,"Source","MatrixFunctions","MatrixFunctions.c")) matrix.remove(os.path.join(ROOT,"Source","MatrixFunctions","MatrixFunctionsF16.c")) except: pass statistics = glob.glob(os.path.join(ROOT,"Source","StatisticsFunctions","*.c")) try: statistics.remove(os.path.join(ROOT,"Source","StatisticsFunctions","StatisticsFunctions.c")) statistics.remove(os.path.join(ROOT,"Source","StatisticsFunctions","StatisticsFunctionsF16.c")) except: pass complexf = glob.glob(os.path.join(ROOT,"Source","ComplexMathFunctions","*.c")) try: complexf.remove(os.path.join(ROOT,"Source","ComplexMathFunctions","ComplexMathFunctions.c")) complexf.remove(os.path.join(ROOT,"Source","ComplexMathFunctions","ComplexMathFunctionsF16.c")) except: pass basic = glob.glob(os.path.join(ROOT,"Source","BasicMathFunctions","*.c")) try: basic.remove(os.path.join(ROOT,"Source","BasicMathFunctions","BasicMathFunctions.c")) basic.remove(os.path.join(ROOT,"Source","BasicMathFunctions","BasicMathFunctionsF16.c")) except: pass controller = glob.glob(os.path.join(ROOT,"Source","ControllerFunctions","*.c")) try: controller.remove(os.path.join(ROOT,"Source","ControllerFunctions","ControllerFunctions.c")) except: pass common = glob.glob(os.path.join(ROOT,"Source","CommonTables","*.c")) try: common.remove(os.path.join(ROOT,"Source","CommonTables","CommonTables.c")) common.remove(os.path.join(ROOT,"Source","CommonTables","CommonTablesF16.c")) except: pass interpolation = glob.glob(os.path.join(ROOT,"Source","InterpolationFunctions","*.c")) try: interpolation.remove(os.path.join(ROOT,"Source","InterpolationFunctions","InterpolationFunctions.c")) interpolation.remove(os.path.join(ROOT,"Source","InterpolationFunctions","InterpolationFunctionsF16.c")) except: pass quaternion = glob.glob(os.path.join(ROOT,"Source","QuaternionMathFunctions","*.c")) try: quaternion.remove(os.path.join(ROOT,"Source","QuaternionMathFunctions","QuaternionMathFunctions.c")) except: pass distance = glob.glob(os.path.join(ROOT,"Source","DistanceFunctions","*.c")) try: distance.remove(os.path.join(ROOT,"Source","DistanceFunctions","DistanceFunctions.c")) except: pass bayes = glob.glob(os.path.join(ROOT,"Source","BayesFunctions","*.c")) try: bayes.remove(os.path.join(ROOT,"Source","BayesFunctions","BayesFunctions.c")) except: pass svm = glob.glob(os.path.join(ROOT,"Source","SVMFunctions","*.c")) try: svm.remove(os.path.join(ROOT,"Source","SVMFunctions","SVMFunctions.c")) except: pass # Add dependencies transformMod = transform + common + basic + complexf + fastmath + matrix + statistics statisticsMod = statistics + common + fastmath + basic interpolationMod = interpolation + common filteringMod = filtering + common + support + fastmath + basic controllerMod = controller + common matrixMod = matrix supportMod = support complexfMod = complexf + fastmath + common + basic basicMod = basic quaternionMod = quaternion fastmathMod = basic + fastmath + common distanceMod = distance + common + basic + statistics + fastmath bayesMod = bayes + fastmath + common + statistics + basic svmMod = svm + fastmath + common + basic filteringMod.append(os.path.join("PythonWrapper","cmsisdsp_pkg","src","cmsisdsp_filtering.c")) matrixMod.append(os.path.join("PythonWrapper","cmsisdsp_pkg","src","cmsisdsp_matrix.c")) supportMod.append(os.path.join("PythonWrapper","cmsisdsp_pkg","src","cmsisdsp_support.c")) statisticsMod.append(os.path.join("PythonWrapper","cmsisdsp_pkg","src","cmsisdsp_statistics.c")) complexfMod.append(os.path.join("PythonWrapper","cmsisdsp_pkg","src","cmsisdsp_complexf.c")) basicMod.append(os.path.join("PythonWrapper","cmsisdsp_pkg","src","cmsisdsp_basic.c")) controllerMod.append(os.path.join("PythonWrapper","cmsisdsp_pkg","src","cmsisdsp_controller.c")) transformMod.append(os.path.join("PythonWrapper","cmsisdsp_pkg","src","cmsisdsp_transform.c")) interpolationMod.append(os.path.join("PythonWrapper","cmsisdsp_pkg","src","cmsisdsp_interpolation.c")) quaternionMod.append(os.path.join("PythonWrapper","cmsisdsp_pkg","src","cmsisdsp_quaternion.c")) fastmathMod.append(os.path.join("PythonWrapper","cmsisdsp_pkg","src","cmsisdsp_fastmath.c")) distanceMod.append(os.path.join("PythonWrapper","cmsisdsp_pkg","src","cmsisdsp_distance.c")) bayesMod.append(os.path.join("PythonWrapper","cmsisdsp_pkg","src","cmsisdsp_bayes.c")) svmMod.append(os.path.join("PythonWrapper","cmsisdsp_pkg","src","cmsisdsp_svm.c")) missing=set([ ]) def notf16(number): if re.search(r'f16',number): return(False) if re.search(r'F16',number): return(False) return(True) def isnotmissing(src): name=os.path.splitext(os.path.basename(src))[0] return(not (name in missing)) # If there are too many files, the linker command is failing on Windows. # So f16 functions are removed since they are not currently available in the wrapper. # A next version will have to structure this wrapper more cleanly so that the # build can work even with more functions filtering = list(filter(isnotmissing,list(filter(notf16, filteringMod)))) matrix = list(filter(isnotmissing,list(filter(notf16, matrixMod)))) support = list(filter(isnotmissing,list(filter(notf16, supportMod)))) statistics = list(filter(isnotmissing,list(filter(notf16, statisticsMod)))) complexf = list(filter(isnotmissing,list(filter(notf16, complexfMod)))) basic = list(filter(isnotmissing,list(filter(notf16, basicMod)))) controller = list(filter(isnotmissing,list(filter(notf16, controllerMod)))) transform = list(filter(isnotmissing,list(filter(notf16, transformMod)))) interpolation = list(filter(isnotmissing,list(filter(notf16, interpolationMod)))) quaternion = list(filter(isnotmissing,list(filter(notf16, quaternionMod)))) fastmath = list(filter(isnotmissing,list(filter(notf16, fastmathMod)))) distance = list(filter(isnotmissing,list(filter(notf16, distanceMod)))) bayes = list(filter(isnotmissing,list(filter(notf16, bayesMod)))) svm = list(filter(isnotmissing,list(filter(notf16, svmMod)))) #for l in filtering: # print(os.path.basename(l)) #quit() def mkModule(name,srcs,funcDir,newCflags=[]): localinc = os.path.join(ROOT,"Source",funcDir) return(Extension(name, sources = (srcs ) , include_dirs = [localinc] + includes + [numpy.get_include()], extra_compile_args = cflags + newCflags )) flagsForCommonWithoutFFT=["-DARM_DSP_CONFIG_TABLES", "-DARM_FAST_ALLOW_TABLES", "-DARM_ALL_FAST_TABLES"] moduleFiltering = mkModule('cmsisdsp_filtering',filtering,"FilteringFunctions",flagsForCommonWithoutFFT) moduleMatrix = mkModule('cmsisdsp_matrix',matrix,"MatrixFunctions") moduleSupport = mkModule('cmsisdsp_support',support,"SupportFunctions") moduleStatistics = mkModule('cmsisdsp_statistics',statistics,"StatisticsFunctions",flagsForCommonWithoutFFT) moduleComplexf= mkModule('cmsisdsp_complexf',complexf,"ComplexMathFunctions") moduleBasic = mkModule('cmsisdsp_basic',basic,"BasicMathFunctions") moduleController = mkModule('cmsisdsp_controller',controller,"ControllerFunctions",flagsForCommonWithoutFFT) moduleTransform = mkModule('cmsisdsp_transform',transform,"TransformFunctions") moduleInterpolation = mkModule('cmsisdsp_interpolation',interpolation,"InterpolationFunctions",flagsForCommonWithoutFFT) moduleQuaternion = mkModule('cmsisdsp_quaternion',quaternion,"QuaternionMathFunctions") moduleFastmath = mkModule('cmsisdsp_fastmath',fastmath,"FastMathFunctions",flagsForCommonWithoutFFT) moduleDistance = mkModule('cmsisdsp_distance',distance,"DistanceFunctions",flagsForCommonWithoutFFT) moduleBayes = mkModule('cmsisdsp_bayes',bayes,"BayesFunctions",flagsForCommonWithoutFFT) moduleSVM = mkModule('cmsisdsp_svm',svm,"SVMFunctions",flagsForCommonWithoutFFT) def build(): if sys.version_info.major < 3: print('setup.py: Error: This package only supports Python 3.', file=sys.stderr) sys.exit(1) main_ns = {} ver_path = convert_path(os.path.join("cmsisdsp","version.py")) with open(ver_path) as ver_file: exec(ver_file.read(), main_ns) setup (name = 'cmsisdsp', version = main_ns['__version__'], packages=["cmsisdsp","cmsisdsp.sdf","cmsisdsp.sdf.nodes","cmsisdsp.sdf.nodes.host","cmsisdsp.sdf.scheduler"], description = 'CMSIS-DSP Python API', long_description=open("PythonWrapper_README.md").read(), long_description_content_type='text/markdown', ext_modules = [moduleFiltering , moduleMatrix, moduleSupport, moduleStatistics, moduleComplexf, moduleBasic, moduleController, moduleTransform, moduleInterpolation, moduleQuaternion, moduleFastmath, moduleDistance, moduleBayes, moduleSVM ], include_package_data=True, author = 'Copyright (C) 2010-2022 ARM Limited or its affiliates. All rights reserved.', author_email = 'christophe.favergeon@arm.com', url="https://arm-software.github.io/CMSIS_5/DSP/html/index.html", python_requires='>=3.6', license="License :: OSI Approved :: Apache Software License", platforms=['any'], classifiers=[ "Programming Language :: Python :: 3", "Programming Language :: Python :: Implementation :: CPython", "Programming Language :: C", "License :: OSI Approved :: Apache Software License", "Operating System :: OS Independent", "Development Status :: 4 - Beta", "Topic :: Software Development :: Embedded Systems", "Topic :: Scientific/Engineering :: Mathematics", "Environment :: Console", "Intended Audience :: Developers", ], keywords=['development','dsp','cmsis','cmsis-dsp','Arm','signal processing','maths'], install_requires=['numpy>=1.19', 'networkx>=2.5', 'jinja2>= 2.0, <3.0', 'sympy>=1.6', 'markupsafe<2.1' ], project_urls={ # Optional 'Bug Reports': 'https://github.com/ARM-software/CMSIS_5/issues', 'Source': 'https://github.com/ARM-software/CMSIS_5/tree/develop/CMSIS/DSP', } ) build()