-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
27 lines (19 loc) · 726 Bytes
/
setup.py
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
import os
from pathlib import Path
from setuptools import setup
from setuptools.command.build import build
class Build(build):
"""Custom build for setuptools to compile C++ shared libraries."""
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
def run(self):
self.build()
super().run()
def build(self):
current_dir = Path.cwd()
source_dir = Path(__file__).resolve().parent / "mlspm" / "_c"
os.chdir(source_dir)
os.system("g++ -fPIC -O3 -c matching.cpp peaks.cpp")
os.system("g++ -shared matching.o peaks.o -o mlspm_lib.so")
os.chdir(current_dir)
setup(cmdclass={"build": Build}, has_ext_modules=lambda: True)