setup.py
# -*- coding: utf-8 -*-
"""This module is needed for the installation of the package."""
import os
from setuptools import setup
from setuptools.command.install import install
import pkg_resources
import os
import subprocess
import sys
import warnings
version = {}
with open("smuthi/version.py") as fp:
exec(fp.read(), version)
__version__ = version['__version__']
class CustomInstallCommand(install):
"""Compile nfmds code."""
def run(self):
install.run(self)
# if Windows: try to install pywigxjpf now
if sys.platform.startswith('win'):
sys.stdout.write('Try to install pywigxjpf ... \n')
sys.stdout.flush()
try:
subprocess.check_call([sys.executable, "-m", "pip", "install", "pywigxjpf"], stdout=None, stderr=None)
except Exception as e:
warnings.warn('\n*****************************************************\n'
'pywigxjpf installation failed.\n'
'If you want to benefit from fast Wigner3j calculations:\n'
'Try to install manually by "pip install pywigxjpf".'
'\n*****************************************************\n',
UserWarning)
# compile nfmds if not built on readthedocs
if sys.platform.startswith('win'):
executable_ending = '.exe'
else:
executable_ending = '.out'
if not os.environ.get('READTHEDOCS'):
nfmds_sources_dirname = pkg_resources.resource_filename('smuthi.linearsystem.tmatrix.nfmds', 'NFM-DS')
# standard precision
sys.stdout.write('\nCompiling sources at ' + nfmds_sources_dirname + ' ...')
sys.stdout.flush()
os.chdir(nfmds_sources_dirname + '/TMATSOURCES')
sys.stdout.flush()
try:
subprocess.check_call(['gfortran', 'TAXSYM_SMUTHI.f90', '-o', 'TAXSYM_SMUTHI_DOUBLE_PREC' + executable_ending])
sys.stdout.write(' done.\n')
sys.stdout.flush()
except Exception as e:
warnings.warn('\n Compiling NFM-DS sources failed.\n', UserWarning)
# extended precision
sys.stdout.write('\nCompiling sources at ' + nfmds_sources_dirname + ' with extended precision ...')
os.rename('Parameters.f90', 'Parameters_standard.f90')
os.rename('Parameters_extended.f90', 'Parameters.f90')
try:
subprocess.check_call(['gfortran', 'TAXSYM_SMUTHI.f90', '-o', 'TAXSYM_SMUTHI_QUAD_PREC'
+ executable_ending])
sys.stdout.write(' done.\n')
sys.stdout.flush()
except Exception as e:
warnings.warn('\n Compiling NFM-DS sources with extended precision failed.\n', UserWarning)
os.rename('Parameters.f90', 'Parameters_extended.f90')
os.rename('Parameters_standard.f90', 'Parameters.f90')
def read(fname):
return open(os.path.join(os.path.dirname(__file__), fname)).read()
def get_requirements():
requirements = ['argparse',
'imageio',
'matplotlib',
'mpmath',
'numpy',
'numba',
'pyyaml',
'scipy',
'sympy',
'tqdm',
'h5py',
'pycparser']
if sys.platform.startswith('win'):
# skip pywigxjpf (to have it optional, if the user has no C compiler)
sys.stdout.write('Compiling from Windows machine. Skipping pywigxjpf for the moment. '
'I will try to install it in post processing.\n')
sys.stdout.flush()
else:
requirements.append('pywigxjpf')
return requirements
setup(
name="SMUTHI",
version=__version__,
author="Amos Egel",
author_email="amos.egel@gmail.com",
url='https://gitlab.com/AmosEgel/smuthi',
description="Light scattering by multiple particles in thin-film systems",
long_description=read('README.rst'),
packages=['smuthi',
'smuthi.fields',
'smuthi.linearsystem',
'smuthi.linearsystem.tmatrix',
'smuthi.linearsystem.tmatrix.nfmds',
'smuthi.linearsystem.particlecoupling',
'smuthi.postprocessing',
'smuthi.utility'],
cmdclass={'install': CustomInstallCommand,
'develop': CustomInstallCommand},
package_data={'smuthi.linearsystem.tmatrix.nfmds': ['NFM-DS/*.txt', 'NFM-DS/TMATSOURCES/*', 'NFM-DS/TMATFILES/*',
'NFM-DS/INPUTFILES/*.dat', 'NFM-DS/OUTPUTFILES/*'],
'smuthi': ['_data/*']},
install_requires=get_requirements(),
extras_require={'cuda': ['PyCuda']},
entry_points={'console_scripts': ['smuthi = smuthi.__main__:main']},
license='MIT',
)