# 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 PyAnuga(PythonPackage):
    """ANUGA (pronounced "AHnooGAH") is open-source software for the simulation
    of the shallow water equation, in particular it can be used to model
    tsunamis and floods."""

    homepage = "https://github.com/GeoscienceAustralia/anuga_core"
    url = "https://github.com/GeoscienceAustralia/anuga_core/archive/2.1.tar.gz"
    git = "https://github.com/GeoscienceAustralia/anuga_core.git"

    license("Apache-2.0")

    # The git main branch of the repo is now python3-only
    version("main", branch="main")

    # Non-versioned dependencies for Anuga main and future versions based on python@3.5:
    depends_on("python@3.5:", type=("build", "run"), when="@2.2:")
    depends_on("gdal+geos+python", type=("build", "run"), when="@2.2:")
    depends_on("py-future", type=("build", "run"), when="@2.2:")
    depends_on("py-matplotlib", type=("build", "run"), when="@2.2:")
    depends_on("py-numpy", type=("build", "run"), when="@2.2:")
    depends_on("py-setuptools", type=("build"), when="@2.2:")
    # Replaces pyar in python3 anuga:
    depends_on("py-mpi4py", type=("build", "run"), when="@2.2:")

    # Unversioned dependencies of the python2 and python3-based versions
    depends_on("py-cython", type=("build"))
    depends_on("py-netcdf4", type=("build", "run"))
    depends_on("py-dill", type=("build", "test"))
    depends_on("py-nose", type=("build", "test"))
    depends_on("py-scipy", type=("build", "test"))
    depends_on("py-triangle", type=("build", "test"))
    depends_on("mpi", type=("test", "run"))

    # https://github.com/GeoscienceAustralia/anuga_core/issues/247
    conflicts("%apple-clang@12:")

    def setup_run_environment(self, env):
        if self.run_tests:
            env.prepend_path("PATH", self.spec["mpi"].prefix.bin)

    install_time_test_callbacks = ["test", "installtest"]

    def installtest(self):
        python("runtests.py", "--no-build")