setup.py

# This code is licensed under the Apache License, Version 2.0. You may
# obtain a copy of this license in the LICENSE.txt file in the root directory
# of this source tree or at http://www.apache.org/licenses/LICENSE-2.0.
#
# Any modifications or derivative works of this code must retain this
# copyright notice, and modified files need to carry a notice indicating
# that they have been altered from the originals.

import os
import re
import shutil
import platform
import subprocess

from setuptools import setup, find_packages, Extension

with open("requirements.txt") as f:
    REQUIREMENTS = f.read().splitlines()

s_system = platform.system().lower()
if s_system == "windows":
    s_executable = "lindbladmpo.exe"
    s_target_os = "WINDOWS"
    print(
        "Note: Windows building of the solver is not currently supported.\n"
        "The installation guide in the pacakge directory contains instructions regarding solver\n"
        "building and local installation of cygwin, which is required for solver execution."
    )
else:
    s_executable = "lindbladmpo"
    if s_system == "darwin":
        s_target_os = "MACOS"
    else:
        s_target_os = "LINUX"

    process = subprocess.Popen(
        f"git clone https://github.com/ITensor/ITensor.git itensor3/", shell=True
    )
    exit_code = process.wait()
    if exit_code != 0:
        raise Exception("Cloning of ITensor repo using a git command failed.")
    shutil.copy("./src/options.mk", "itensor3/")

    exit_code = subprocess.call(f"cd itensor3 && make configure", shell=True)
    if exit_code != 0:
        pass

    exit_code = subprocess.call(
        f"cd itensor3/itensor && make build OS_TARGET={s_target_os}", shell=True
    )
    if exit_code != 0:
        pass
    process = subprocess.Popen(
        f"make -C ./src/ ITENSOR3_DIR=../itensor3 OS_TARGET={s_target_os}", shell=True
    )
    exit_code = process.wait()
    if exit_code != 0:
        pass
    shutil.copy(f"./bin/{s_executable}", "./lindbladmpo/")
    shutil.rmtree("itensor3")

# Read long description from README.
with open("README.md") as readme_file:
    README = re.sub(
        "<!--- long-description-skip-begin -->.*<!--- long-description-skip-end -->",
        "",
        readme_file.read(),
        flags=re.S | re.M,
    )

setup(
    name="lindbladmpo",
    version="0.2.1",
    description="A matrix-product-operators solver for the Lindblad master equation",
    long_description=README,
    long_description_content_type="text/markdown",
    url="https://github.com/haggaila/lindbladmpo",
    author="Gregoire Misguich, Haggai Landa, Gal Shmulovitz, Eldor Fadida, Dekel Meirom",
    # author_email="",
    license="Apache 2.0",
    classifiers=[
        "Environment :: Console",
        "License :: OSI Approved :: Apache Software License",
        "Intended Audience :: Developers",
        "Intended Audience :: Science/Research",
        # "Operating System :: Microsoft :: Windows",
        "Operating System :: MacOS",
        "Operating System :: POSIX :: Linux",
        "Programming Language :: Python :: 3.8",
        "Programming Language :: Python :: 3.9",
        "Topic :: Scientific/Engineering",
    ],
    keywords="mps mpo lindblad sdk quantum",
    packages=find_packages(exclude=["test*"]),
    install_requires=REQUIREMENTS,
    include_package_data=True,
    python_requires=">=3.8",
    extras_require={
        "visualization": ["matplotlib>=2.1"],
    },
    project_urls={
        "Source Code": "https://github.com/qiskit-community/lindbladmpo",
    },
    zip_safe=False,
)