-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathsetup.py
188 lines (175 loc) · 6.53 KB
/
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
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
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
import sys
import re
import itertools
from setuptools import setup, find_packages
from distutils.core import Extension
import numpy as np
import argparse
import os
from typing import Iterable
# Args parsing, gets the path to the library dirs
if '--path' in sys.argv:
index = sys.argv.index('--path')
sys.argv.pop(index) # Removes the '--path'
lib_dir = sys.argv.pop(index) # Returns the element after the '--path'
else:
print("Please provide a path to the required libraries")
exit(1)
armadir = os.path.join(lib_dir, 'armadillo-install')
blasdir = os.path.join(lib_dir, 'openblas-install')
lapackdir = os.path.join(lib_dir, 'lp-install')
cvdir = os.path.join(lib_dir, 'cv-install')
nbisdir = os.path.join(lib_dir, 'nbis-install')
config_file = os.path.join(os.path.dirname(__file__), 'pynger', 'config.py')
with open(config_file, 'w') as cfg:
cfg.write('# This is a configuration file generated by setup.py during Pynger\'s installation\n')
cfg.write('__NBIS_LIB__ = "{}"\n'.format(nbisdir))
# Find all the libraries
def find_libs(root: str, libs: dict):
""" Library linking helper.
Finds the given libraries and return their path in a format compatible with the distutils.core.Extension class.
Args:
root: path containing all the libraries to be loaded
libs: dictionary where the keys are (relative or absolute) path from root and values are lists of library names
Return:
The dictionary {libraries: ..., library_dirs: ..., extra_objects: ...}, to be used as
>>> Extension(..., **find_libs('root', ['path1', 'path2'], ['lib1', 'lib2']))
"""
ret = {
'libraries': [],
'library_dirs': [],
'extra_objects': []}
for d, pair in libs.items():
static_libraries, formats = pair
if os.path.isabs(d):
static_lib_dir = d
else:
static_lib_dir = os.path.join(root, d)
if sys.platform == 'win32':
ret['libraries'].extend(static_libraries)
ret['library_dirs'].append(static_lib_dir)
else: # POSIX
ret['extra_objects'].extend([os.path.join(static_lib_dir, fmt.format(lib)) for lib, fmt in zip(static_libraries, formats)])
return ret
def find_all_libs(root: str):
def get_all_static_libs_in_path(path):
lib_patt = re.compile('(?:lib)?(\\w+)\\.(?:lib|a)')
for dir, _, files in os.walk(path):
matches = filter(None, map(lib_patt.match, files))
files = list(set(map(lambda x: x[0], matches))) # use set to produce unique strings
if len(files) > 0:
yield list(map(lambda x: os.path.join(dir, x), files))
extra_objects = get_all_static_libs_in_path(root)
extra_objects = list(itertools.chain(*extra_objects))
return extra_objects
# NBIS Extension
extra_objects = find_all_libs(os.path.join(nbisdir, 'lib'))
if sys.platform.startswith('linux'):
extra_objects = ['-Wl,--start-group'] + extra_objects + ['-Wl,--end-group']
print('NBIS extra_objects:', extra_objects)
nbis_ext = Extension(
'pynger.fingerprint.nbis',
sources=[
'pynger/fingerprint/nbismodule/nbismodule.c',
'pynger/fingerprint/nbismodule/sgmnt.c',
'pynger/fingerprint/nbismodule/enhnc.c',
'pynger/fingerprint/nbismodule/rors.c',
'pynger/fingerprint/nbismodule/utils.c',
'pynger/fingerprint/nbismodule/mindtct.c'],
include_dirs=[
os.path.join(nbisdir, 'include'),
np.get_include()],
extra_objects=extra_objects
)
pani_ext = Extension(
'pynger.signal.pani',
sources=[
'pynger/signal/panimodule/panimodule.c',
'pynger/signal/panimodule/panigauss.c'],
include_dirs=[np.get_include()],
)
# OpenCV
ang_seg_args = []
ang_seg_link_args = []
if sys.platform != 'win32':
ang_seg_args += ['-std=gnu++14', '-Wextra']
ang_seg_link_args += ['-fPIC']
if sys.platform.startswith('linux'):
ang_seg_link_args += ['-Wl,--verbose']
extra_objects = find_all_libs(os.path.join(cvdir, 'lib'))
print('OpenCV extra_objects:', extra_objects)
# As the linking order matters, put the corelib at the end and the adelib right before
indices = []
### ADE
cv_adelib = [lib for lib in extra_objects if 'libade' in lib]
if len(cv_adelib) > 0:
cv_adelib_idx = extra_objects.index(cv_adelib[0])
indices += [cv_adelib_idx]
### CORE
cv_corelib = [lib for lib in extra_objects if 'core' in lib]
if len(cv_corelib) > 0:
cv_corelib_idx = extra_objects.index(cv_corelib[0])
indices += [cv_corelib_idx]
### LIBZ
cv_zlib = [lib for lib in extra_objects if 'libz' in lib]
if len(cv_zlib) > 0:
cv_zlib_idx = extra_objects.index(cv_zlib[0])
indices += [cv_zlib_idx]
### Reorder
indices = [k for k in range(len(extra_objects)) if k not in indices] + indices
extra_objects = [extra_objects[k] for k in indices]
# LAPACK
extra_objects += find_all_libs(os.path.join(lapackdir, 'lib'))
# Envelope with group construct
if sys.platform.startswith('linux'):
extra_objects = ['-Wl,--start-group'] + extra_objects + ['-Wl,--end-group']
print('OpenCV extra_compile_args:', ang_seg_args)
print('OpenCV extra_link_args:', ang_seg_link_args)
print('OpenCV extra_objects:', extra_objects)
ang_seg_ext = Extension(
'pynger.fingerprint.cangafris',
sources=[
'pynger/fingerprint/angafris_segmentation/Sources/AdaptiveThreshold.cpp',
'pynger/fingerprint/angafris_segmentation/Sources/ImageCropping.cpp',
'pynger/fingerprint/angafris_segmentation/Sources/ImageMaskSimplify.cpp',
'pynger/fingerprint/angafris_segmentation/Sources/ImageNormalization.cpp',
'pynger/fingerprint/angafris_segmentation/Sources/ImageRescale.cpp',
'pynger/fingerprint/angafris_segmentation/Sources/ImageSignificantMask.cpp',
'pynger/fingerprint/angafris_segmentation/Sources/myMathFunc.cpp',
'pynger/fingerprint/angafris_segmentation/Sources/TypesTraits.cpp',
'pynger/fingerprint/angafris_segmentation/Sources/ang_seg_wrapper.cpp',
'pynger/fingerprint/angafris_segmentation/ang_seg_module.cpp',
],
include_dirs=[
np.get_include(),
os.path.join(armadir, 'include'),
os.path.join(cvdir, 'include' if sys.platform == 'win32' else 'include/opencv4'),
],
extra_compile_args=ang_seg_args,
extra_link_args=ang_seg_link_args,
extra_objects=extra_objects,
)
# Load README file
with open("README.md", "r") as fh:
long_description = fh.read()
# Install the package
ext_modules = [pani_ext, ang_seg_ext]
if sys.platform != 'win32':
ext_modules += [nbis_ext]
setup(
name="pynger-DottD",
version="0.0.1",
author="Filippo Santarelli",
author_email="[email protected]",
description="A suite of utilities for Fingerprint Analysis",
long_description=long_description,
long_description_content_type="text/markdown",
url="https://github.com/pypa/sampleproject",
packages=find_packages(),
classifiers=[
"Programming Language :: Python :: 3",
"License :: OSI Approved :: MIT License",
"Operating System :: OS Independent",
],
ext_modules=ext_modules
)