python-cnorxz/setup.py

#!/usr/bin/env python

#import numpy
import setuptools
from distutils.core import setup, Extension
import sysconfig
import subprocess

libname = "cnorxz"
author = "Christian Zimmermann"
author_email = "chizeta@f3l.de"
path_to_cnorxz = "/home/chizeta/repos/cnorxz/install"
expected_cnorxz_version = "1.0.0" # dummy for now; in the future this is supposed to be the c++ version which is guaranteed to be compatible with py cnorxz

git_hash = subprocess.run(['git','rev-parse','HEAD'],stdout=subprocess.PIPE).stdout.decode('ascii')[:-1]
git_tags = subprocess.run(['git', 'tag', '-l', '--sort=refname', 'v*'],stdout=subprocess.PIPE).stdout.decode('ascii').split()
cnorxz_version = subprocess.run([path_to_cnorxz+"/bin/cnorxz-config",'--version'],stdout=subprocess.PIPE).stdout.decode('ascii')[:-1]

assert len(git_tags) != 0, "no version tags found"
git_tag = git_tags[-1]
version = git_tag[1:] + 'dev' 
git_hash_tag = subprocess.run(['git','rev-list','-n','1',git_tag],stdout=subprocess.PIPE).stdout.decode('ascii')[:-1]
if git_hash_tag == git_hash and expected_cnorxz_version != cnorxz_version:
    version = git_tag[1:]
        
inc_dirs = list()
if path_to_cnorxz != "":
    inc_dirs.append( path_to_cnorxz + "/include/cnorxz" )
inc_dirs.append( "cnorxz/core/include" )
#inc_dirs.append( numpy.get_include() )
inc_dirs.append( "/home/chizeta/my_env/lib/python3.11/site-packages/numpy/core/include" )
    
lib_dirs = list()
lib_dirs.append( "/usr/lib" )
lib_dirs.append( "/usr/local/lib" )
if path_to_cnorxz != "":
    inc_dirs.append( path_to_cnorxz + "/lib" )
lib_dirs.append( "/home/chizeta/repos/cnorxz/install/lib" )

extra_compile_args = sysconfig.get_config_var('CFLAGS').split()
cnorxz_flags = subprocess.run([path_to_cnorxz+"/bin/cnorxz-config",'--flags'],stdout=subprocess.PIPE).stdout.decode('ascii').split()
cnorxz_flags.remove("-Werror")
extra_compile_args += cnorxz_flags

default_extension_args = dict(
    language = "c++",
    include_dirs = inc_dirs,
    libraries = [libname],
    library_dirs = lib_dirs,
    extra_compile_args = extra_compile_args,
)

if __name__ == "__main__":
    print("Include directories:")
    for d in inc_dirs:
        print("\t",d)
    print("Library directories:")
    for d in lib_dirs:
        print("\t",d)

    #extensions = list()
    #extensions.append(
    #    setuptools.Extension(
    #        #name = libname + ".core",
    #        name = libname,
    #        sources = [
    #            "cnorxz/core/core.cpp",
    #            "cnorxz/core/lib/array_wrapper.cpp"
    #        ],
    #        define_macros=[('HAVE_CEREAL',None)],
    #        **default_extension_args
    #    )
    #)
    # append further extensions (cereal, hdf5, mpi ...) here
    
    setup(
        name = libname,
        packages = setuptools.find_packages(),
        version = version,
        include_dirs = inc_dirs,
        ext_modules = [Extension(
            'cnorxz', ['cnorxz/core/core.cpp','cnorxz/core/lib/array_wrapper.cpp'],
            define_macros=[('HAVE_CEREAL',None)],
            **default_extension_args
        )],
        #ext_modules = Cython.Build.cythonize(extensions),
        author = author,
        author_email = author_email,
    )