# Copyright 2013-2024 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.package import *


class Arrayfire(CMakePackage, CudaPackage):
    """ArrayFire is a high performance software library for parallel computing
    with an easy-to-use API. Its array based function set makes parallel
    programming more accessible."""

    homepage = "https://arrayfire.org/docs/index.htm"
    git = "https://github.com/arrayfire/arrayfire.git"
    maintainers("umar456")

    license("FreeImage")

    version("master")
    version("3.8.1", commit="823e8e399fe8c120c6ec7ec75f09e6106b3074ca", tag="v3.8.1")
    version(
        "3.7.3", commit="59ac7b980d1ae124aae914fb29cbf086c948954d", submodules=True, tag="v3.7.3"
    )
    version(
        "3.7.2", commit="218dd2c99300e77496239ade76e94b0def65d032", submodules=True, tag="v3.7.2"
    )
    version(
        "3.7.0", commit="fbea2aeb6f7f2d277dcb0ab425a77bb18ed22291", submodules=True, tag="v3.7.0"
    )

    variant("forge", default=False, description="Enable graphics library")
    variant("opencl", default=False, description="Enable OpenCL backend")

    depends_on("boost@1.70:")
    depends_on("fftw-api@3:")
    depends_on("blas")
    depends_on("cuda@7.5:", when="+cuda")
    depends_on("cudnn", when="+cuda")

    depends_on("opencl", when="+opencl")
    depends_on("pocl+icd", when="^[virtuals=opencl] pocl")

    # TODO add more opencl backends:
    # currently only Cuda backend is enabled
    # https://github.com/arrayfire/arrayfire/wiki/Build-Instructions-for-Linux#opencl-backend-dependencies

    depends_on("fontconfig", when="+forge")
    depends_on("glfw@3.1.4:", when="+forge")

    conflicts("cuda_arch=none", when="+cuda", msg="CUDA architecture is required")

    @property
    def libs(self):
        query_parameters = self.spec.last_query.extra_parameters

        libraries = []
        if "cpu" in query_parameters:
            libraries.append("libafcpu")
        if "cuda" in query_parameters and "+cuda" in self.spec:
            libraries.append("libafcuda")
        if "opencl" in query_parameters and "+opencl" in self.spec:
            libraries.append("libafopencl")
        if not query_parameters or "unified" in query_parameters:
            libraries.append("libaf")

        return find_libraries(libraries, root=self.prefix, recursive=True)

    def cmake_args(self):
        args = []
        args.extend(
            [
                self.define_from_variant("AF_BUILD_CUDA", "cuda"),
                self.define_from_variant("AF_BUILD_FORGE", "forge"),
                self.define_from_variant("AF_BUILD_OPENCL", "opencl"),
                self.define("BUILD_TESTING", self.run_tests),
            ]
        )

        if "+cuda" in self.spec:
            arch_list = [
                "{}.{}".format(arch[:-1], arch[-1])
                for arch in self.spec.variants["cuda_arch"].value
            ]
            args.append(self.define("CUDA_architecture_build_targets", arch_list))

        if self.spec["blas"].name in INTEL_MATH_LIBRARIES:
            if self.version >= Version("3.8.0"):
                args.append(self.define("AF_COMPUTE_LIBRARY", "Intel-MKL"))
            else:
                args.append(self.define("USE_CPU_MKL", True))
                args.append(self.define("USE_OPENCL_MKL", True))
            if "%intel" not in self.spec:
                args.append(self.define("MKL_THREAD_LAYER", "GNU OpenMP"))
        else:
            if self.version >= Version("3.8.0"):
                args.append(self.define("AF_COMPUTE_LIBRARY", "FFTW/LAPACK/BLAS"))
            else:
                args.append(self.define("USE_CPU_MKL", False))
                args.append(self.define("USE_OPENCL_MKL", False))

        return args