IPKM/python-pygmx
3
0
Fork 0
Code Issues Pull Requests Activity

Locate gromacs lib in subdirectoies of LD_LIBRARY_PATH

Browse Source
This commit is contained in:
Niels Müller
2016-03-16 15:07:53 +01:00
parent 21e6594c07
commit cced8be730
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
setup.py
View File

@ -5,10 +5,17 @@ from Cython.Build import cythonize
import numpy import numpy
include_dirs = [numpy.get_include(), 'gromacs/src','gromacs/src/external/tng_io/include',] def locate_lib(directory, lib):
flib = 'lib{}.so'.format(lib)
for root, dirs, files in os.walk(directory):
if flib in files:
return root
include_dirs = [numpy.get_include(), 'gromacs/src', 'gromacs/src/external/tng_io/include']
library_dirs = [] library_dirs = []
if 'LD_LIBRARY_PATH' in os.environ: if 'LD_LIBRARY_PATH' in os.environ:
lib = os.environ['LD_LIBRARY_PATH'].split(':')[0] lib = locate_lib(os.environ['LD_LIBRARY_PATH'].split(':')[0], 'gromacs')
library_dirs.insert(0, lib) library_dirs.insert(0, lib)
include = lib.replace('/lib', '/include') include = lib.replace('/lib', '/include')
if os.path.exists(include): if os.path.exists(include):