dragonpilot/selfdrive/can/parser_pyx_setup.py

import os
import subprocess
from distutils.core import Extension, setup  # pylint: disable=import-error,no-name-in-module

from Cython.Build import cythonize

from common.basedir import BASEDIR
from common.cython_hacks import BuildExtWithoutPlatformSuffix

sourcefiles = ['parser_pyx.pyx']
extra_compile_args = ["-std=c++11"]
ARCH = subprocess.check_output(["uname", "-m"], encoding='utf8').rstrip()  # pylint: disable=unexpected-keyword-arg

if ARCH == "aarch64":
  extra_compile_args += ["-Wno-deprecated-register"]

setup(name='CAN parser',
      cmdclass={'build_ext': BuildExtWithoutPlatformSuffix},
      ext_modules=cythonize(
        Extension(
          "parser_pyx",
          language="c++",
          sources=sourcefiles,
          extra_compile_args=extra_compile_args,
          include_dirs=[
            BASEDIR,
            os.path.join(BASEDIR, 'phonelibs', 'capnp-cpp/include'),
          ],
          extra_link_args=[
            os.path.join(BASEDIR, 'selfdrive', 'can', 'libdbc.so'),
          ],
        )
      ),
      nthreads=4,
)