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.'
		  'The installation guide in the pacakge directory contains instructions regarding solver '
		  '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'

	os.mkdir('../itensor3')
	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/')

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

# 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.1.0",
	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",
	# 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.6",
		"Programming Language :: Python :: 3.7",
		"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.6",
	extras_require={
		'visualization': ['matplotlib>=2.1'],
	},
	project_urls={
		"Source Code": "https://github.com/haggaila/lindbladmpo",
	},
	zip_safe=False
)