Mercurial > public > mercurial-scm > hg-stable
diff contrib/python-zstandard/setup_zstd.py @ 30822:b54a2984cdd4
zstd: vendor python-zstandard 0.6.0
Commit 63c68d6f5fc8de4afd9bde81b13b537beb4e47e8 from
https://github.com/indygreg/python-zstandard is imported without
modifications (other than removing unwanted files).
This includes minor performance and feature improvements. It also
changes the vendored zstd library from 1.1.1 to 1.1.2.
# no-check-commit
| author | Gregory Szorc <gregory.szorc@gmail.com> |
|---|---|
| date | Sat, 14 Jan 2017 19:41:43 -0800 |
| parents | b86a448a2965 |
| children | c32454d69b85 |
line wrap: on
line diff
--- a/contrib/python-zstandard/setup_zstd.py Sat Jan 14 20:05:15 2017 +0530 +++ b/contrib/python-zstandard/setup_zstd.py Sat Jan 14 19:41:43 2017 -0800 @@ -16,15 +16,24 @@ 'common/zstd_common.c', 'compress/fse_compress.c', 'compress/huf_compress.c', - 'compress/zbuff_compress.c', 'compress/zstd_compress.c', 'decompress/huf_decompress.c', - 'decompress/zbuff_decompress.c', 'decompress/zstd_decompress.c', 'dictBuilder/divsufsort.c', 'dictBuilder/zdict.c', )] +zstd_sources_legacy = ['zstd/%s' % p for p in ( + 'deprecated/zbuff_compress.c', + 'deprecated/zbuff_decompress.c', + 'legacy/zstd_v01.c', + 'legacy/zstd_v02.c', + 'legacy/zstd_v03.c', + 'legacy/zstd_v04.c', + 'legacy/zstd_v05.c', + 'legacy/zstd_v06.c', + 'legacy/zstd_v07.c' +)] zstd_includes = [ 'c-ext', @@ -35,6 +44,11 @@ 'zstd/dictBuilder', ] +zstd_includes_legacy = [ + 'zstd/deprecated', + 'zstd/legacy', +] + ext_sources = [ 'zstd.c', 'c-ext/compressiondict.c', @@ -51,14 +65,27 @@ 'c-ext/dictparams.c', ] +zstd_depends = [ + 'c-ext/python-zstandard.h', +] -def get_c_extension(name='zstd'): + +def get_c_extension(support_legacy=False, name='zstd'): """Obtain a distutils.extension.Extension for the C extension.""" root = os.path.abspath(os.path.dirname(__file__)) sources = [os.path.join(root, p) for p in zstd_sources + ext_sources] + if support_legacy: + sources.extend([os.path.join(root, p) for p in zstd_sources_legacy]) + include_dirs = [os.path.join(root, d) for d in zstd_includes] + if support_legacy: + include_dirs.extend([os.path.join(root, d) for d in zstd_includes_legacy]) + + depends = [os.path.join(root, p) for p in zstd_depends] # TODO compile with optimizations. return Extension(name, sources, - include_dirs=include_dirs) + include_dirs=include_dirs, + depends=depends, + extra_compile_args=["-DZSTD_LEGACY_SUPPORT=1"] if support_legacy else [])