setup.py

from setuptools import setup, Extension

def extra_compile_args():
    """Returns extra compiler args for platform."""
    import sys
    if sys.platform == 'win32':
        return ['/std:c++20', '/O2', '/DNDEBUG'] # for Visual Studio C++

    return ['-std=c++20', '-O3', '-DNDEBUG']

# Define the native extension
customalloc_extension = Extension(
    "customalloc",
    sources=["bigO/custom_alloc.cpp"],
    extra_compile_args=extra_compile_args(),
    include_dirs=["bigO/include"],
)

# Call setup
setup(
    name="bigO",
    version="0.0.1",
    description="Track asymptotic complexity in time and space.",
    long_description=open("README.md").read(),
    long_description_content_type="text/markdown",
    author="Emery Berger",
    author_email="emery.berger@gmail.com",
    url="https://github.com/plasma-umass/bigO",
    classifiers=[
        "Intended Audience :: Developers",
        "Intended Audience :: Science/Research",
        "Topic :: Software Development",
        "Topic :: Software Development :: Debuggers",
        "Programming Language :: Python",
        "Programming Language :: Python :: 3 :: Only",
        "Programming Language :: Python :: 3",
        "Programming Language :: Python :: 3.8",
        "Programming Language :: Python :: 3.9",
        "Programming Language :: Python :: 3.10",
        "Programming Language :: Python :: 3.11",
        "Programming Language :: Python :: 3.12",
        "License :: OSI Approved :: Apache Software License",
        "Operating System :: POSIX :: Linux",
        "Operating System :: MacOS :: MacOS X",
        "Operating System :: Microsoft :: Windows",
    ],
    python_requires=">=3.8",
    ext_modules=[customalloc_extension],
    packages=["bigO"],
    package_data={"bigO": ["*.cpp", "*.h"]},
)