# Copyright 2013-2022 Lawrence Livermore National Security, LLC and other
# Spack Project Developers. See the top-level COPYRIGHT file for details.
#
# SPDX-License-Identifier: (Apache-2.0 OR MIT)
import os

from spack import *


class Turbine(AutotoolsPackage):
    """Turbine: The Swift/T runtime"""

    homepage = 'http://swift-lang.org/Swift-T'
    url      = 'https://swift-lang.github.io/swift-t-downloads/spack/turbine-1.3.0.tar.gz'
    git      = "https://github.com/swift-lang/swift-t.git"
    configure_directory = 'turbine/code'

    version('master', branch='master')
    version('1.3.0', sha256='9709e5dada91a7dce958a7967d6ff2bd39ccc9e7da62d05a875324b5089da393')
    version('1.2.3', sha256='a3156c7e0b39e166da3de8892f55fa5d535b0c99c87a9add067c801098fe51ba')

    variant('python', default=False,
            description='Enable calling python')
    variant('r', default=False,
            description='Enable calling R')
    variant('hdf5', default=False,
            description='Enable HDF5 support')

    depends_on('adlbx')
    depends_on('adlbx@master', when='@master')
    depends_on('adlbx@:0.9.2', when='@1.2.3:1.2.99')
    depends_on('tcl', type=('build', 'run'))
    depends_on('zsh', type=('build', 'run'))
    depends_on('swig', type='build')
    depends_on('python', when='+python')
    depends_on('r', when='+r')
    depends_on('r-rinside', when='+r')
    depends_on('hdf5', when='+hdf5')
    depends_on('mpi')
    depends_on('autoconf', type='build', when='@master')
    depends_on('automake', type='build', when='@master')
    depends_on('libtool', type='build', when='@master')
    depends_on('m4', type=('build', 'run'))

    def setup_build_environment(self, env):
        spec = self.spec

        env.set('CC', spec['mpi'].mpicc)
        env.set('CXX', spec['mpi'].mpicxx)
        env.set('CXXLD', spec['mpi'].mpicxx)

    @property
    def configure_directory(self):
        if self.version == Version('master'):
            return 'turbine/code'
        else:
            return '.'

    def configure_args(self):
        args = ['--with-c-utils=' + self.spec['exmcutils'].prefix,
                '--with-adlb='    + self.spec['adlbx'].prefix,
                '--with-tcl='     + self.spec['tcl'].prefix,
                '--with-mpi='     + self.spec['mpi'].prefix,
                '--disable-static-pkg']
        if '+hdf5' in self.spec:
            args.append('--with-hdf5=ON')
        else:
            args.append('--with-hdf5=OFF')
        if '+python' in self.spec:
            args.append('--with-python-exe={0}'.format(
                        self.spec['python'].command.path))
        if '+r' in self.spec:
            r_location = '{0}/rlib/R'.format(self.spec['r'].prefix)
            if not os.path.exists(r_location):
                rscript = which('Rscript')
                if rscript is not None:
                    r_location = rscript('-e', 'cat(R.home())', output=str)
                else:
                    msg = 'Could not locate Rscript on your PATH!'
                    raise RuntimeError(msg)
            args.append('--with-r={0}'.format(r_location))
        return args
