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

import os

from spack.package import *
from spack.util.prefix import Prefix


class Pgi(Package):
    """PGI optimizing multi-core x64 compilers for Linux, MacOS & Windows
    with support for debugging and profiling of local MPI processes.

    Note: The PGI compilers are licensed software. You will need to create an
    account on the PGI homepage and download PGI yourself. Spack will search
    your current directory for the download tarball. Alternatively, add this
    file to a mirror so that Spack can find it. For instructions on how to
    set up a mirror, see https://spack.readthedocs.io/en/latest/mirrors.html"""

    homepage = "https://www.pgroup.com/"
    manual_download = True

    version("20.4", sha256="f3ecc2104b304cd5c8b20e3ffdb5da88f2b5f7cc148e8daf00561928a5cbbc2e")
    version("19.10", sha256="ac9db73ba80a66fe3bc875f63aaa9e16f54674a4e88b25416432430ba8cf203d")
    version("19.7", sha256="439692aeb51eff464b968c3bfed4536ed7bd3ba6f8174bc0ebe2219a78fe62ae")
    version("19.4", sha256="23eee0d4da751dd6f247d624b68b03538ebd172e63a053c41bb67013f07cf68e")
    version("19.1", sha256="3e05a6db2bf80b5d15f6ff83188f20cb89dc23e233417921e5c0822e7e57d34f")
    version("18.10", sha256="4b3ff83d2a13de6001bed599246eff8e63ef711b8952d4a9ee12efd666b3e326")
    version("18.4", sha256="81e0dcf6000b026093ece180d42d77854c23269fb8409cedcf51c674ca580a0f")
    version("17.10", sha256="9da8f869fb9b70c0c4423c903d40a9e22d54b997af359a43573c0841165cd1b6")
    version("17.4", sha256="115c212d526695fc116fe44f1e722793e60b6f7d1b341cd7e77a95da8e7f6c34")

    variant("network", default=True, description="Perform a network install")
    variant("single", default=False, description="Perform a single system install")
    variant(
        "nvidia", default=False, description="Enable installation of optional NVIDIA components"
    )
    variant("amd", default=False, description="Enable installation of optional AMD components")
    variant("java", default=False, description="Enable installation of Java Runtime Environment")
    variant("mpi", default=False, description="Enable installation of Open MPI")

    # Licensing
    license_required = True
    license_comment = "#"
    license_files = ["license.dat"]
    license_vars = ["PGROUPD_LICENSE_FILE", "LM_LICENSE_FILE"]
    license_url = "http://www.pgroup.com/doc/pgiinstall.pdf"

    def url_for_version(self, version):
        if int(str(version.up_to(1))) <= 17:
            return "file://{0}/pgilinux-20{1}-{2}-x86_64.tar.gz".format(
                os.getcwd(), version.up_to(1), version.joined
            )
        else:
            return "file://{0}/pgilinux-20{1}-{2}-x86-64.tar.gz".format(
                os.getcwd(), version.up_to(1), version.joined
            )

    def install(self, spec, prefix):
        # Enable the silent installation feature
        os.environ["PGI_SILENT"] = "true"
        os.environ["PGI_ACCEPT_EULA"] = "accept"
        os.environ["PGI_INSTALL_DIR"] = prefix

        if "+network" in spec and "~single" in spec:
            os.environ["PGI_INSTALL_TYPE"] = "network"
            os.environ["PGI_INSTALL_LOCAL_DIR"] = "%s/%s/share_objects" % (prefix, self.version)
        elif "+single" in spec and "~network" in spec:
            os.environ["PGI_INSTALL_TYPE"] = "single"
        else:
            msg = "You must choose either a network install or a single "
            msg += "system install.\nYou cannot choose both."
            raise RuntimeError(msg)

        if "+nvidia" in spec:
            os.environ["PGI_INSTALL_NVIDIA"] = "true"

        if "+amd" in spec:
            os.environ["PGI_INSTALL_AMD"] = "true"

        if "+java" in spec:
            os.environ["PGI_INSTALL_JAVA"] = "true"

        if "+mpi" in spec:
            os.environ["PGI_INSTALL_MPI"] = "true"

        # Run install script
        os.system("./install")

    def setup_run_environment(self, env):
        prefix = Prefix(join_path(self.prefix, "linux86-64", self.version))

        env.prepend_path("PATH", prefix.bin)
        env.prepend_path("MANPATH", prefix.man)
        env.prepend_path("LD_LIBRARY_PATH", prefix.lib)
        env.set("CC", join_path(prefix.bin, "pgcc"))
        env.set("CXX", join_path(prefix.bin, "pgc++"))
        env.set("F77", join_path(prefix.bin, "pgfortran"))
        env.set("FC", join_path(prefix.bin, "pgfortran"))

        if "+mpi" in self.spec:
            ompi_dir = os.listdir(prefix.mpi)[0]
            env.prepend_path("PATH", join_path(prefix.mpi, ompi_dir, "bin"))
            env.prepend_path("LD_LIBRARY_PATH", join_path(prefix.mpi, ompi_dir, "lib"))
            env.prepend_path("C_INCLUDE_PATH", join_path(prefix.mpi, ompi_dir, "include"))
            env.prepend_path("MANPATH", join_path(prefix.mpi, ompi_dir, "share/man"))