# 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)

import os

from spack.package import *


class HipExamples(Package):
    """Examples for HIP"""

    homepage = "https://github.com/ROCm-Developer-Tools/HIP-Examples/"
    git = "https://github.com/ROCm-Developer-Tools/HIP-Examples.git"
    url = "https://github.com/ROCm-Developer-Tools/HIP-Examples/archive/rocm-5.4.3.tar.gz"
    tags = ["rocm"]

    maintainers("srekolam", "renjithravindrankannath", "afzpatel")

    version("5.6.1", sha256="c1b5d30e387f869fae21170790ea3d604f7f0dba7771a9c096d9a5c2351dd001")
    version("5.6.0", sha256="b751a0cac938248f7ea0fbeaa9df35688357b54ddd13359e2842a770b7923dfe")
    version("5.5.1", sha256="c8522ef3f0804c85eef7e9efe2671f375b0d7f2100de85f55dcc2401efed6389")
    version("5.5.0", sha256="bea8a4155bbfbdb3bc1f83c22e4bd1214b1b4e1840b58dc7d37704620de5b103")
    version("5.4.3", sha256="053b8b7892e2929e3f90bd978d8bb1c9801e4803eadd7d97fc6692ce60af1d47")

    patch("0001-add-inc-and-lib-paths-to-openmp-helloworld.patch")
    patch("0002-add-fpic-compile-to-add4.patch")

    for ver in ["5.6.1", "5.6.0", "5.5.1", "5.5.0", "5.4.3"]:
        depends_on("hip@" + ver, when="@" + ver)
        depends_on("rocm-openmp-extras@" + ver, when="@" + ver)

    def install(self, spec, prefix):
        stage = os.getcwd()
        os.environ["ROCM_OPENMP_EXTRAS_DIR"] = self.spec["rocm-openmp-extras"].prefix
        os.environ["LD_LIBRARY_PATH"] = self.spec["rocm-openmp-extras"].prefix.lib
        for root, dirs, files in os.walk(stage):
            if "Makefile" in files:
                with working_dir(root, create=True):
                    make()
                    # itterate through the files in dir to find the newly built binary
                    for file in os.listdir("."):
                        if file not in files and os.path.isfile(file) and os.access(file, os.X_OK):
                            install(file, join_path(prefix, file))
                            if file == "RecursiveGaussian":
                                install(
                                    "RecursiveGaussian_Input.bmp",
                                    join_path(prefix, "RecursiveGaussian_Input.bmp"),
                                )

    def test_examples(self):
        """
        run all hip example binaries
        """

        os.environ["LD_LIBRARY_PATH"] = self.spec["rocm-openmp-extras"].prefix.lib
        test_dir = self.prefix
        for file_name in os.listdir(test_dir):
            file_path = join_path(test_dir, file_name)
            if os.path.isfile(file_path) and os.access(file_path, os.X_OK):
                with test_part(
                    self,
                    "test_example_{0}".format(file_name),
                    purpose="run installed {0}".format(file_name),
                ):
                    exe = which(file_path)
                    exe()