File size: 2,286 Bytes
26225c5 |
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 |
#----------------------------------------------------------------------#
# distutils setup script for compiling cut-pursuit python extensions #
#----------------------------------------------------------------------#
"""
Compilation command: python setup.py build_ext
Hugo Raguet 2020
"""
from setuptools import setup, Extension
from distutils.command.build import build
import numpy
import shutil # for rmtree, os.rmdir can only remove _empty_ directory
import os
import re
### options ###
# compilation and linkage options
# MIN_OPS_PER_THREAD roughly controls parallelization, see doc in README.md
if os.name == 'nt': # windows
extra_compile_args = ["/std:c++11", "/openmp",
"-DMIN_OPS_PER_THREAD=10000"]
extra_link_args = ["/lgomp"]
elif os.name == 'posix': # linux
extra_compile_args = ["-std=c++11", "-fopenmp",
"-DMIN_OPS_PER_THREAD=10000"]
extra_link_args = ["-lgomp"]
else:
raise NotImplementedError('OS not yet supported.')
### auxiliary functions ###
class build_class(build):
def initialize_options(self):
build.initialize_options(self)
self.build_lib = "bin"
def run(self):
build_path = self.build_lib
def purge(dir, pattern):
for f in os.listdir(dir):
if re.search(pattern, f):
os.remove(os.path.join(dir, f))
### preprocessing ###
# ensure right working directory
tmp_work_dir = os.path.realpath(os.curdir)
os.chdir(os.path.realpath(os.path.dirname(__file__)))
if not os.path.exists("bin"):
os.mkdir("bin")
# remove previously compiled lib
purge("bin/", "grid_graph")
### compilation ###
name = "grid_graph"
mod = Extension(
name,
# list source files
["cpython/grid_graph_cpy.cpp",
"../src/edge_list_to_forward_star.cpp",
"../src/grid_to_graph.cpp"],
include_dirs=[numpy.get_include(), "../include"],
extra_compile_args=extra_compile_args,
extra_link_args=extra_link_args)
setup(name=name, ext_modules=[mod], cmdclass=dict(build=build_class))
### postprocessing ###
try:
shutil.rmtree("build") # remove temporary compilation products
except FileNotFoundError:
pass
os.chdir(tmp_work_dir) # get back to initial working directory
|