# 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 Ucx(AutotoolsPackage, CudaPackage):
    """a communication library implementing high-performance messaging for
    MPI/PGAS frameworks"""

    homepage = "http://www.openucx.org"
    url      = "https://github.com/openucx/ucx/releases/download/v1.3.1/ucx-1.3.1.tar.gz"
    git      = "https://github.com/openucx/ucx.git"

    maintainers = ['hppritcha']

    # Current
    version('1.11.2', sha256='deebf86a5344fc2bd9e55449f88c650c4514928592807c9bc6fe4190e516c6df')
    version('1.11.1', sha256='29338cad18858517f96b46ff83bdd259a5899e274792cebd269717c660aa86fd')
    version('1.11.0', sha256='b7189b69fe0e16e3c03784ef674e45687a9c520750bd74a45125c460ede37647')
    version('1.10.1', sha256='ae9a108af6842ca135e7ec9b6131469adf9f1e50f899349fafcc69a215368bc9')
    version('1.10.0', sha256='b885e24b1b94724c03cb213c355381e98df1e2d1fd7f633cf8055b6dd05db92d')
    version('1.9-dev', branch='v1.9.x')
    version('1.9.0', sha256='a7a2c8841dc0d5444088a4373dc9b9cc68dbffcd917c1eba92ca8ed8e5e635fb')
    version('1.8.1', sha256='a48820cb8d0761b5ccf3e7ba03a7c8c1dde6276017657178829e07ffc35b556a')
    version('1.8.0', sha256='e400f7aa5354971c8f5ac6b881dc2846143851df868088c37d432c076445628d')
    version('1.7.0', sha256='6ab81ee187bfd554fe7e549da93a11bfac420df87d99ee61ffab7bb19bdd3371')
    version('1.6.1', sha256='1425648aa03f5fa40e4bc5c4a5a83fe0292e2fe44f6054352fbebbf6d8f342a1')
    version('1.6.0', sha256='360e885dd7f706a19b673035a3477397d100a02eb618371697c7f3ee4e143e2c')
    version('1.5.2', sha256='1a333853069860e86ba69b8d071ccc9871209603790e2b673ec61f8086913fad')
    version('1.5.1', sha256='567119cd80ad2ae6968ecaa4bd1d2a80afadd037ccc988740f668de10d2fdb7e')
    version('1.5.0', sha256='84f6e4fa5740afebb9b1c8bb405c07206e58c56f83120dcfcd8dc89e4b7d7458')
    version('1.4.0', sha256='99891a98476bcadc6ac4ef9c9f083bc6ffb188a96b3c3bc89c8bbca64de2c76e')

    # Still supported
    version('1.3.1', sha256='e058c8ec830d2f50d9db1e3aaaee105cd2ad6c1e6df20ae58b9b4179de7a8992')
    version('1.3.0', sha256='71e69e6d78a4950cc5a1edcbe59bf7a8f8e38d59c9f823109853927c4d442952')
    version('1.2.2', sha256='914d10fee8f970d4fb286079dd656cf8a260ec7d724d5f751b3109ed32a6da63')
    version('1.2.1', sha256='fc63760601c03ff60a2531ec3c6637e98f5b743576eb410f245839c84a0ad617')
    version('1.2.0', sha256='1e1a62d6d0f89ce59e384b0b5b30b416b8fd8d7cedec4182a5319d0dfddf649c')

    variant('thread_multiple', default=False,
            description='Enable thread support in UCP and UCT')
    variant('optimizations', default=True,
            description='Enable optimizations')
    variant('logging', default=False,
            description='Enable logging')
    variant('debug', default=False,
            description='Enable debugging')
    variant('assertions', default=False,
            description='Enable assertions')
    variant('parameter_checking', default=False,
            description='Enable parameter checking')
    variant('pic', default=True,
            description='Builds with PIC support')
    variant('java', default=False,
            description='Builds with Java bindings')
    variant('gdrcopy', default=False,
            description='Enable gdrcopy support')
    variant('knem', default=False,
            description='Enable KNEM support')
    variant('xpmem', default=False,
            description='Enable XPMEM support')
    variant('cma', default=False,
            description="nable Cross Memory Attach")
    variant('rc', default=False,
            description="Compile with IB Reliable Connection support")
    variant('dc', default=False,
            description="Compile with IB Dynamic Connection support")
    variant('ud', default=False,
            description="Compile with IB Unreliable Datagram support")
    variant('mlx5-dv', default=False,
            description="Compile with mlx5 Direct Verbs support")
    variant('ib-hw-tm', default=False,
            description="Compile with IB Tag Matching support")
    variant('dm', default=False,
            description="Compile with Device Memory support")
    variant('cm', default=False,
            description="Compile with IB Connection Manager support")

    depends_on('numactl')
    depends_on('rdma-core')
    depends_on('pkgconfig', type='build')
    depends_on('java@8', when='+java')
    depends_on('maven', when='+java')
    depends_on('gdrcopy', when='@1.7:+gdrcopy')
    depends_on('gdrcopy@1.3', when='@:1.6+gdrcopy')
    conflicts('+gdrcopy', when='~cuda',
              msg='gdrcopy currently requires cuda support')
    depends_on('xpmem', when='+xpmem')
    depends_on('knem', when='+knem')
    depends_on('binutils+ld', when='%aocc', type='build')

    configure_abs_path = 'contrib/configure-release'

    @when('@1.9-dev')
    def autoreconf(self, spec, prefix):
        Executable('./autogen.sh')()

    def configure_args(self):
        spec = self.spec
        config_args = []

        if '+thread_multiple' in spec:
            config_args.append('--enable-mt')
        else:
            config_args.append('--disable-mt')

        if '+cma' in spec:
            config_args.append('--enable-cma')
        else:
            config_args.append('--disable-cma')

        if '+paramter_checking' in spec:
            config_args.append('--enable-params-check')
        else:
            config_args.append('--disable-params-check')

        # Activate SIMD based on properties of the target
        if 'avx' in self.spec.target:
            config_args.append('--with-avx')
        else:
            config_args.append('--without-avx')

        config_args.extend(self.enable_or_disable('optimizations'))
        config_args.extend(self.enable_or_disable('assertions'))
        config_args.extend(self.enable_or_disable('logging'))

        config_args.extend(self.with_or_without('pic'))
        config_args.extend(self.with_or_without('rc'))
        config_args.extend(self.with_or_without('ud'))
        config_args.extend(self.with_or_without('dc'))
        config_args.extend(self.with_or_without('mlx5-dv'))
        config_args.extend(self.with_or_without('ib-hw-tm'))
        config_args.extend(self.with_or_without('dm'))
        config_args.extend(self.with_or_without('cm'))
        config_args.extend(self.with_or_without('java',
                                                activation_value='prefix'))
        config_args.extend(self.with_or_without('cuda',
                                                activation_value='prefix'))
        config_args.extend(self.with_or_without('gdrcopy',
                                                activation_value='prefix'))
        config_args.extend(self.with_or_without('knem',
                                                activation_value='prefix'))
        config_args.extend(self.with_or_without('xpmem',
                                                activation_value='prefix'))

        # lld doesn't support '-dynamic-list-data'
        if '%aocc' in spec:
            config_args.append('LDFLAGS=-fuse-ld=bfd')

        return config_args