# Copyright 2013-2023 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 Atmi(CMakePackage):
    """Asynchronous Task and Memory Interface, or ATMI, is a runtime framework
    and programming model for heterogeneous CPU-GPU systems. It provides a
    consistent, declarative API to create task graphs on CPUs and GPUs
    (integrated and discrete)."""

    homepage = "https://github.com/RadeonOpenCompute/atmi"
    git = "https://github.com/RadeonOpenCompute/atmi.git"
    url = "https://github.com/RadeonOpenCompute/atmi/archive/rocm-5.5.0.tar.gz"
    tags = ["rocm"]

    license("MIT")

    maintainers("srekolam", "renjithravindrankannath")
    version("5.5.1", sha256="6b3ee68433506315b55d093a4b47463916874fb6f3f602098eaff2ec283e69ab")
    version("5.5.0", sha256="b8bfd32e5c386f5169da62172964343f9b7fad207e0e74dd1093c7acf06d9811")
    version("5.4.3", sha256="243aae6614e5bd136a099102957a6d65a01434b620291349613ad63701868ef8")
    version("5.4.0", sha256="b5cce10d7099fecbb40a0d9c2f29a7675315471fe145212b375e37e4c8ba5618")
    version("5.3.3", sha256="cc1144e4939cea2944f6c72a21406b9dc5b56d933696494074c280df7469834a")
    version("5.3.0", sha256="dffc0eb0bc1617843e7f728dbd6c8b12326c5c8baa34369aa267aab40f5deb6a")
    version("5.2.3", sha256="5f66c59e668cf968e86b556a0a52ee0202d1b370d8406e291a874cbfd200ee17")
    version("5.2.1", sha256="6b33445aa67444c038cd756f855a58a72dd35db57e7b63da37fe78a8585b982b")
    version("5.2.0", sha256="33e77905a607734157d46c736c924c7c50b6b13f2b2ddbf711cb08e37f2efa4f")
    version("5.1.3", sha256="a43448d77705b2b07e1758ffe8035aa6ba146abc2167984e8cb0f1615797b341")
    version("5.1.0", sha256="6a758f5a8332e6774cd8e14a4e5ce05e43b1e05298d817b4068c35fa1793d333")
    version(
        "5.0.2",
        sha256="3aea040f5a246539ab118f2183cf3e802a21e0e6215a53025eda77f382341747",
        deprecated=True,
    )
    version(
        "5.0.0",
        sha256="208c1773170722b60b74357e264e698df5871e9d9d490d64011e6ea76750d9cf",
        deprecated=True,
    )
    version(
        "4.5.2",
        sha256="c235cfb8bdd89deafecf9123264217b8cc5577a5469e3e1f24587fa820d0792e",
        deprecated=True,
    )
    version(
        "4.5.0",
        sha256="64eeb0244cedae99db7dfdb365e0ad624106cc1090a531f94885ae81e254aabf",
        deprecated=True,
    )
    version(
        "4.3.1",
        sha256="4497fa6d33547b946e2a51619f2777ec36e9cff1b07fd534eb8a5ef0d8e30650",
        deprecated=True,
    )
    version(
        "4.3.0",
        sha256="1cbe0e9258ce7cce7b7ccc288335dffbac821ceb745c4f3fd48e2a258abada89",
        deprecated=True,
    )
    version(
        "4.2.0",
        sha256="c1c89c00d2dc3e764c63b2e51ff7fd5c06d5881ed56aed0adf639582d3389585",
        deprecated=True,
    )
    version(
        "4.1.0",
        sha256="b31849f86c79f90466a9d67f0a28a93c1675181e38e2a5f571ffc963e4b06f5f",
        deprecated=True,
    )
    version(
        "4.0.0",
        sha256="8a2e5789ee7165aff0f0669eecd23ac0a5c8a5bfbc1acd9380fe9a8ed5bffe3a",
        deprecated=True,
    )
    version(
        "3.10.0",
        sha256="387e87c622ec334d3ba7a2f4f015ea9a219712722f4c56c1ef572203d0d072ea",
        deprecated=True,
    )
    version(
        "3.9.0",
        sha256="0a305e85bab210dd9a0410aa01d46227e00b59141e4675c50d731ad1232ab828",
        deprecated=True,
    )
    version(
        "3.8.0",
        sha256="039f0c2b369d0dbc01000754893d9210828f4cb9b36c3e70da8c3819b131c933",
        deprecated=True,
    )
    version(
        "3.7.0",
        sha256="8df08489a10ee04cea911811393e0e7d91bd437fc1fd81a23a4e7ab924a974f3",
        deprecated=True,
    )
    version(
        "3.5.0",
        sha256="3fb57d2e583fab82bd0582d0c2bccff059ca91122c18ac49a7770a8bb041a37b",
        deprecated=True,
    )

    depends_on("cmake@3:", type="build")
    depends_on("rsync")

    for ver in [
        "3.5.0",
        "3.7.0",
        "3.8.0",
        "3.9.0",
        "3.10.0",
        "4.0.0",
        "4.1.0",
        "4.2.0",
        "4.3.0",
        "4.3.1",
        "4.5.0",
        "4.5.2",
        "5.0.0",
        "5.0.2",
        "5.1.0",
        "5.1.3",
        "5.2.0",
        "5.2.1",
        "5.2.3",
        "5.3.0",
        "5.3.3",
        "5.4.0",
        "5.4.3",
        "5.5.0",
        "5.5.1",
    ]:
        depends_on("comgr@" + ver, type="link", when="@" + ver)
        depends_on("hsa-rocr-dev@" + ver, type="link", when="@" + ver)
        depends_on("elf", type="link", when="@" + ver)

    for ver in ["5.5.0", "5.5.1"]:
        depends_on("rocm-core@" + ver, when="@" + ver)

    root_cmakelists_dir = "src"

    patch("0001-Remove-relative-link-paths-to-external-libraries.patch", when="@3.5.0")
    # Removing direct reference to /usr/bin/rysnc for rsync command.
    patch("0002-Remove-usr-bin-rsync-reference.patch", when="@4.0.0:5.0.0")
    # Reset the installation path and remove direct reference to rsync.
    patch("0002-Remove-usr-bin-rsync-reference-5.2.0.patch", when="@5.0.2:5.2.0")
    # Remove direct reference to /usr/bin/rsync path for rsync command
    patch(
        "0002-Remove-direct-reference-to-usr-bin-rysnc-for-rsync-cmd-5.2.1.patch", when="@5.2.1:"
    )

    def cmake_args(self):
        args = [self.define("ROCM_VERSION", self.spec.version)]
        return args

    @run_after("install")
    def install_stub(self):
        install("include/atmi_interop_hsa.h", self.prefix.include)