Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
19 changes: 18 additions & 1 deletion setup.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,11 @@
import glob
from setuptools import setup
from pybind11.setup_helpers import Pybind11Extension, build_ext

sources = ['binder.cpp']
sources += glob.glob("tinytransfer/*.cpp")


# Convert distutils Windows platform specifiers to CMake -A arguments
PLAT_TO_CMAKE = {
"win32": "Win32",
Expand All @@ -9,10 +14,22 @@
"win-arm64": "ARM64",
}

# explicitly added heatshrink paths
sources += [
"tinytransfer/heatshrink/heatshrink.c",
"tinytransfer/heatshrink/heatshrink_decoder.c",
"tinytransfer/heatshrink/heatshrink_encoder.c",
]

ext_modules = [
Pybind11Extension(
"pytinytransfer",
['binder.cpp'],
sources,
#tinytransfer and heatshrink headers
include_dirs=[
"tinytransfer",
"tinytransfer/heatshrink",
],
),
]

Expand Down