/ proof-of-concept / benchcompress
Sign in
proof-of-concept / benchcompress
06875e9
Branches
mainbenchmark-results
benchcompress / benchcompress / setup.py
66 lines · 2.3 KBCodeBlameHistory
30a2f07initialJeremy Magland 1from setuptools import setup, find_packages
804780bspeed up markov with C++Jeremy Magland 2from setuptools.command.build_ext import build_ext
3from setuptools import Extension
4import os
5import sys
6import subprocess
8class CMakeExtension(Extension):
9 def __init__(self, name, sourcedir=""):
10 Extension.__init__(self, name, sources=[])
11 self.sourcedir = os.path.abspath(sourcedir)
13class CMakeBuild(build_ext):
14 def build_extension(self, ext):
15 extdir = os.path.abspath(os.path.dirname(self.get_ext_fullpath(ext.name)))
17 cmake_args = [
18 f"-DCMAKE_LIBRARY_OUTPUT_DIRECTORY={extdir}",
19 f"-DPYTHON_EXECUTABLE={sys.executable}"
20 ]
22 build_args = []
24 if not os.path.exists(self.build_temp):
25 os.makedirs(self.build_temp)
27 subprocess.check_call(["cmake", ext.sourcedir] + cmake_args, cwd=self.build_temp)
28 subprocess.check_call(["cmake", "--build", "."] + build_args, cwd=self.build_temp)
30a2f07initialJeremy Magland 29
30setup(
7884e06rename to benchcompressJeremy Magland 31 name="benchcompress",
30a2f07initialJeremy Magland 32 version="0.1.0",
33 packages=find_packages(where="src"),
34 package_dir={"": "src"},
804780bspeed up markov with C++Jeremy Magland 35 ext_modules=[
c1f542arename ansJeremy Magland 36 CMakeExtension("benchcompress.algorithms.ans.markov_reconstruct_cpp_ext",
37 sourcedir="src/benchcompress/algorithms/ans"),
38 CMakeExtension("benchcompress.algorithms.ans.markov_predict_cpp_ext",
39 sourcedir="src/benchcompress/algorithms/ans"),
40 CMakeExtension("benchcompress.algorithms.ans.get_run_lengths_cpp_ext",
41 sourcedir="src/benchcompress/algorithms/ans")
804780bspeed up markov with C++Jeremy Magland 42 ],
43 cmdclass={
44 "build_ext": CMakeBuild,
45 },
30a2f07initialJeremy Magland 46 python_requires=">=3.8",
9900d62formatJeremy Magland 47 entry_points={
48 "console_scripts": [
7884e06rename to benchcompressJeremy Magland 49 "benchcompress=benchcompress.cli:main",
9900d62formatJeremy Magland 50 ],
51 },
30a2f07initialJeremy Magland 52 author="Jeremy Magland",
93dd44areorganize datasets and algs, add seismicJeremy Magland 53 description="Benchmarking compression methods for scientific data arrays",
30a2f07initialJeremy Magland 54 long_description=open("README.md").read(),
55 long_description_content_type="text/markdown",
e892a17zia_benchmarkJeremy Magland 56 keywords="compression, signal processing",
30a2f07initialJeremy Magland 57 classifiers=[
58 "Development Status :: 3 - Alpha",
59 "Intended Audience :: Science/Research",
60 "Programming Language :: Python :: 3",
61 "Programming Language :: Python :: 3.8",
62 "Programming Language :: Python :: 3.9",
63 "Programming Language :: Python :: 3.10",
64 "Programming Language :: Python :: 3.11"
65 ],
↑↓ move↵openescclose