# -*- 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',
)