import os

import sys

from skbuild import setup

from textwrap import dedent

if __name__ == '__main__':

try:

import cpuinfo

cpu_info = cpuinfo.get_cpu_info()

except Exception:

# newer cpuinfo versions fail to import on unsupported architectures

cpu_info = None

########### Check versions ##########

def exit_with_error(message):

print('ERROR: %s' % message)

sys.exit(1)

########### End of checks ##########

# Read the long_description from README.rst

with open('README.rst') as f:

long_description = f.read()

# Blosc version

with open('VERSION') as f:

VERSION = f.read().strip()

# Create the version.py file

with open('blosc/version.py', 'w') as f:

f.write('__version__ = "%s"\n' % VERSION)

def cmake_bool(cond):

return 'ON' if cond else 'OFF'

classifiers = dedent("""\

""")

setup(name = "blosc",

version = VERSION,

description = 'Blosc data compressor',

long_description = long_description,

classifiers = [c for c in classifiers.split("\n") if c],

python_requires=">=3.9",

author = 'The Blosc development team',

author_email = 'blosc@blosc.org',

maintainer = 'The Blosc development team',

maintainer_email = 'blosc@blosc.org',

url = 'https://github.com/blosc/python-blosc',

license = 'https://opensource.org/licenses/BSD-3-Clause',

platforms = ['any'],

cmake_args = (

['-DUSE_SYSTEM_BLOSC:BOOL=ON'] if int(os.environ.get('USE_SYSTEM_BLOSC', '0'))

else [

'-DUSE_SYSTEM_BLOSC:BOOL=OFF',

'-DDEACTIVATE_SSE2:BOOL=%s' % cmake_bool(('DISABLE_BLOSC_SSE2' in os.environ) or

(cpu_info is None) or

('flags' in cpu_info and 'sse2' not in cpu_info['flags'])),

'-DDEACTIVATE_AVX2:BOOL=%s' % cmake_bool(('DISABLE_BLOSC_AVX2' in os.environ) or

(cpu_info is None) or

('flags' in cpu_info and 'avx2' not in cpu_info['flags'])),

'-DDEACTIVATE_LZ4:BOOL=%s' % cmake_bool(not int(os.environ.get('INCLUDE_LZ4', '1'))),

# Snappy is disabled by default

'-DDEACTIVATE_SNAPPY:BOOL=%s' % cmake_bool(not int(os.environ.get('INCLUDE_SNAPPY', '0'))),

'-DDEACTIVATE_ZLIB:BOOL=%s' % cmake_bool(not int(os.environ.get('INCLUDE_ZLIB', '1'))),

'-DDEACTIVATE_ZSTD:BOOL=%s' % cmake_bool(not int(os.environ.get('INCLUDE_ZSTD', '1'))),

]),

setup_requires=['scikit-build'],

tests_require=['numpy', 'psutil'],

packages = ['blosc'],

)

elif __name__ == '__mp_main__':

# This occurs from `cpuinfo 4.0.0` using multiprocessing to interrogate the

# CPUID flags

# https://github.com/workhorsy/py-cpuinfo/issues/108

pass