dgpython-azurekinect/setup.py at master · isuthermography/dgpython-azurekinect · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
import os
import sys
import os.path
import shutil
import glob
import numpy as np
from setuptools import setup,Extension
from setuptools.command.build import build
from setuptools.command.install import install

from Cython.Build import cythonize
#import dataguzzler_python


ext_modules=cythonize(Extension("dgpython_azurekinect.kinect",
                                sources=["dgpython_azurekinect/kinect.pyx"],
                                libraries=["k4a","k4arecord"],
                                include_dirs=[np.get_include()]),
                      language_level=3)


class BuildCommand(build):
    user_options = build.user_options + [
        ('with-azurekinect=',None,'Path to Azure Kinect SDK')
    ]
    def initialize_options(self):
        build.initialize_options(self)
        self.with_azurekinect = None
        pass

    def finalize_options(self):
        if self.with_azurekinect is not None:
            #for ext in ext_modules:
            build_ext_cmd = self.distribution.get_command_obj("build_ext")
            if not hasattr(build_ext_cmd,"include_dirs") or build_ext_cmd.include_dirs is None:
                build_ext_cmd.include_dirs=[]
                pass
            build_ext_cmd.include_dirs.append(os.path.join(self.with_azurekinect,'include'))
            build_ext_cmd.include_dirs.append(os.path.join(self.with_azurekinect,'sdk','include'))
            if not hasattr(build_ext_cmd,"library_dirs") or build_ext_cmd.library_dirs is None:
                build_ext_cmd.library_dirs=[]
                pass

            build_ext_cmd.library_dirs.append(os.path.join(self.with_azurekinect,'lib'))
            if sys.platform=="win32":
                build_ext_cmd.library_dirs.append(os.path.join(self.with_azurekinect,'sdk','windows-desktop','amd64','release','lib'))
                pass
            if not hasattr(build_ext_cmd,"rpath") or build_ext_cmd.rpath is None:
                build_ext_cmd.rpath=[]
                pass
            if sys.platform != "win32":
                build_ext_cmd.rpath.insert(0,os.path.join(self.with_azurekinect,'lib'))
                build_ext_cmd.rpath.insert(0,"$ORIGIN")
                pass

            pass
        build.finalize_options(self)
        pass

    def run(self):
        #print("with_azurekinect=%s" % (self.with_azurekinect))
        #print("ext[0].include_dirs=%s" % (str(ext_modules[0].include_dirs)))
        build.run(self)
        pass
    pass


class InstallCommand(install):
    user_options = install.user_options + [
        ('with-azurekinect=',None,'Path to Azure Kinect SDK')
    ]
    def initialize_options(self):
        install.initialize_options(self)
        self.with_azurekinect = None
        pass

    def finalize_options(self):
        if self.with_azurekinect is not None:
            pass
        install.finalize_options(self)
        pass

    def run(self):
        #print("with_azurekinect=%s" % (self.with_azurekinect))
        #print("ext[0].include_dirs=%s" % (str(ext_modules[0].include_dirs)))
        install.run(self)
        if sys.platform=="win32":
            # install Azure Kinect DLL's in with our driver
            print("Installing Azure Kinect DLLs into %s" % (self.install_lib))
            dlldir = os.path.join(self.with_azurekinect,'sdk','windows-desktop','amd64','release','bin')
            depthengine_dlls = glob.glob(os.path.join(dlldir,"depthengine*.dll"))
            for depthengine_dll in depthengine_dlls:
                source = depthengine_dll
                destination = os.path.join(self.install_lib,"dgpython_azurekinect",os.path.split(depthengine_dll)[1])
                print("  %s -> %s" % (source,destination))
                shutil.copyfile(source,destination)
                pass
            for source in ["k4a.dll","k4a.pdb","k4arecord.dll","k4arecord.pdb"]:
                sourcepath = os.path.join(dlldir,source)
                destination = os.path.join(self.install_lib,"dgpython_azurekinect",source)
                print("  %s -> %s" % (sourcepath,destination))
                shutil.copyfile(sourcepath,destination)
                pass
            pass
        pass
    pass


setup(name="dgpython_azurekinect",
      description="Azure Kinect module for dgpython",
      author="Stephen D. Holland",
      url="http://thermal.cnde.iastate.edu",
      ext_modules=ext_modules,
      zip_safe=False,
      packages=["dgpython_azurekinect"],
      cmdclass = { 'build': BuildCommand, "install": InstallCommand })