# Copyright 2013-2021 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)


from spack import *


class Hpx5(AutotoolsPackage):
    """The HPX-5 Runtime System. HPX-5 (High Performance ParalleX) is an
    open source, portable, performance-oriented runtime developed at
    CREST (Indiana University). HPX-5 provides a distributed
    programming model allowing programs to run unmodified on systems
    from a single SMP to large clusters and supercomputers with
    thousands of nodes. HPX-5 supports a wide variety of Intel and ARM
    platforms. It is being used by a broad range of scientific
    applications enabling scientists to write code that performs and
    scales better than contemporary runtimes."""
    homepage = "http://hpx.crest.iu.edu"
    url      = "https://github.com/adk9/hpx5/archive/v3.1.0.tar.gz"

    version('4.1.0',     sha256='3f01009f5e517c8dfca266dabb49894d688db5adce09608fb1c877263605a9f8')
    version('4.0.0',     sha256='e35b1161566a65ffbd875c1413ea97a84be0c7b528a3dee99f5e250b2aecbd19')
    version('3.1.0',     sha256='359d457a26b87abb415605911d791ce0ff6edbb064bc40b0f830960f8f612b84')
    version('3.0.0',     sha256='10f14ba198a32787cee05962e346bafb922f74a5135fb09a1ba8c32a1e942800')
    version('2.2.0',     sha256='e34c7513a287d517e67cce5aa3011474c48718e7860c3860ba1290c702be28a8')
    version('2.1.0',     sha256='675826f669eeb3eab40947715af8c8495e2b3d299223372431dc01c1f7d5d616')
    version('2.0.0',     sha256='0278728557b6684aeb86228f44d548ac809302f05a0b9c8b433cdd157629e384')

    # Don't second-guess what compiler we are using on Cray
    patch("configure.patch", when='@4.0.0')

    variant('cuda', default=False, description='Enable CUDA support')
    variant('cxx11', default=False, description='Enable C++11 hpx++ interface')
    variant('debug', default=False, description='Build debug version of HPX-5')
    variant('instrumentation', default=False, description='Enable instrumentation (may affect performance)')
    variant('metis', default=False, description='Enable METIS support')
    variant('mpi', default=False, description='Enable MPI support')
    variant('opencl', default=False, description='Enable OpenCL support')
    variant('photon', default=False, description='Enable Photon support')
    variant('pic', default=True, description='Produce position-independent code')

    depends_on("autoconf", type='build')
    depends_on("automake", type='build')
    depends_on("hwloc")
    depends_on("hwloc +cuda", when='+cuda')
    # Note: We could disable CUDA support via "hwloc ~cuda"
    depends_on("jemalloc")
    # depends_on("libffi")
    depends_on("libtool", type='build')
    # depends_on("lz4")   # hpx5 always builds its own lz4
    depends_on("m4", type='build')
    depends_on("metis", when='+metis')
    depends_on("mpi", when='+mpi')
    depends_on("mpi", when='+photon')
    depends_on("opencl", when='+opencl')
    # depends_on("papi")
    depends_on("pkgconfig", type='build')

    configure_directory = "hpx"
    build_directory = "spack-build"

    def configure_args(self):
        spec = self.spec
        args = [
            '--enable-agas',          # make this a variant?
            '--enable-jemalloc',      # make this a variant?
            '--enable-percolation',   # make this a variant?
            # '--enable-rebalancing',   # this seems broken
            '--with-hwloc=hwloc',
            '--with-jemalloc=jemalloc',
            # Spack's libffi installs its headers strangely,
            # leading to problems
            '--with-libffi=contrib',
            # '--with-papi=papi',   # currently disabled in HPX
        ]

        if '+cxx11' in spec:
            args += ['--enable-hpx++']

        if '+debug' in spec:
            args += ['--enable-debug']

        if '+instrumentation' in spec:
            args += ['--enable-instrumentation']

        if '+mpi' in spec or '+photon' in spec:
            # photon requires mpi
            args += ['--enable-mpi']
            # Choose pkg-config name for MPI library
            if '^openmpi' in spec:
                args += ['--with-mpi=ompi-cxx']
            elif '^mpich' in spec:
                args += ['--with-mpi=mpich']
            elif '^mvapich2' in spec:
                args += ['--with-mpi=mvapich2-cxx']
            else:
                args += ['--with-mpi=system']

        # METIS does not support pkg-config; HPX will pick it up automatically
        # if '+metis' in spec:
        #     args += ['--with-metis=???']

        if '+opencl' in spec:
            args += ['--enable-opencl']
            if '^pocl' in spec:
                args += ['--with-opencl=pocl']
            else:
                args += ['--with-opencl=system']

        if '+photon' in spec:
            args += ['--enable-photon']

        if '+pic' in spec:
            args += ['--with-pic']

        return args
