b932c14008
* Bump the package API of the `builtin` repo to `v2.0`
* Move `var/spack/repos/builtin` -> `var/spack/repos/spack_repo/builtin`
* Move test repos `var/spack/repos/{builtin.mock,tutorial,...}` -> `var/spack/test_repos/`
* Update package dir names to v2 format (`-` -> `_` etc)
* Change absolute imports `from spack.pkg.builtin.my_pkg ...` to relative imports `from ..my_pkg.package ...`
Users who have a repo on top of builtin should change imports from
```python
from spack.pkg.builtin.my_pkg import MyPkg
```
to
```python
from spack_repo.builtin.packages.my_pkg.package import MyPkg
```
and can configure their editors with
```
PYTHONPATH=$spack/lib/spack:$spack/var/spack/repos
```
[skip-verify-checksums]
96 lines
3.8 KiB
Python
96 lines
3.8 KiB
Python
# Copyright Spack Project Developers. See COPYRIGHT file for details.
|
|
#
|
|
# SPDX-License-Identifier: (Apache-2.0 OR MIT)
|
|
|
|
import os
|
|
|
|
from spack.package import *
|
|
|
|
|
|
class Sw4(MakefilePackage):
|
|
"""SW4 implements substantial capabilities for 3-D seismic modeling."""
|
|
|
|
homepage = "https://github.com/geodynamics/sw4"
|
|
git = "https://github.com/geodynamics/sw4.git"
|
|
|
|
maintainers("houjun", "andersp")
|
|
|
|
license("GPL-2.0-or-later")
|
|
|
|
version("master", branch="master")
|
|
version("developer", branch="developer")
|
|
version("3.0", tag="v3.0", commit="13e6d431976f7fc49124c997bf87353aa7afd35e")
|
|
|
|
variant("openmp", default=True, description="Build with OpenMP")
|
|
variant("proj", default=True, description="Build with PROJ")
|
|
variant("hdf5", default=True, description="Build with HDF5")
|
|
variant("zfp", default=False, when="+hdf5", description="Build with ZFP")
|
|
variant("fftw", default=True, description="Build with FFTW")
|
|
variant("debug", default=False, description="Build with debugging symbols")
|
|
|
|
depends_on("cxx", type="build")
|
|
depends_on("fortran", type="build")
|
|
depends_on("mpi")
|
|
depends_on("blas")
|
|
depends_on("lapack")
|
|
depends_on("python")
|
|
depends_on("proj@9:", when="+proj")
|
|
depends_on("hdf5@1.14: +mpi", when="+hdf5")
|
|
depends_on("py-h5py", when="+hdf5")
|
|
depends_on("zfp", when="+zfp")
|
|
depends_on("h5z-zfp@develop", when="+zfp")
|
|
depends_on("fftw@3: +mpi", when="+fftw")
|
|
depends_on("llvm-openmp", when="+openmp %apple-clang")
|
|
|
|
def setup_build_environment(self, env: EnvironmentModifications) -> None:
|
|
env.set("CXX", self.spec["mpi"].mpicxx)
|
|
env.set("FC", self.spec["mpi"].mpifc)
|
|
# openmp is enabled by default
|
|
if self.spec.satisfies("~openmp"):
|
|
env.set("openmp", "no")
|
|
if self.spec.satisfies("+proj"):
|
|
env.set("proj", "yes")
|
|
env.set("SW4ROOT", self.spec["proj"].prefix)
|
|
if self.spec.satisfies("+hdf5"):
|
|
env.set("hdf5", "yes")
|
|
env.set("HDF5ROOT", self.spec["hdf5"].prefix)
|
|
if self.spec.satisfies("+zfp"):
|
|
env.set("zfp", "yes")
|
|
env.set("ZFPROOT", self.spec["zfp"].prefix)
|
|
env.set("H5ZROOT", self.spec["h5z-zfp"].prefix)
|
|
if self.spec.satisfies("+fftw"):
|
|
env.set("fftw", "yes")
|
|
env.set("FFTWHOME", self.spec["fftw"].prefix)
|
|
if self.spec.satisfies("+debug"):
|
|
env.set("debug", "yes")
|
|
env.set("EXTRA_LINK_FLAGS", "-lstdc++ -lm -ldl")
|
|
env.append_flags("EXTRA_LINK_FLAGS", self.spec["blas"].libs.ld_flags)
|
|
env.append_flags("EXTRA_LINK_FLAGS", self.spec["lapack"].libs.ld_flags)
|
|
if self.spec.satisfies("+openmp %apple-clang"):
|
|
env.append_flags("EXTRA_LINK_FLAGS", self.spec["llvm-openmp"].libs.ld_flags)
|
|
# From spack/trilinos
|
|
if (
|
|
self.spec.satisfies("%gcc")
|
|
or self.spec.satisfies("%clang")
|
|
or self.spec.satisfies("%apple-clang")
|
|
):
|
|
fc = Executable(self.compiler.fc)
|
|
libgfortran = fc("--print-file-name", "libgfortran." + dso_suffix, output=str).strip()
|
|
if libgfortran == "libgfortran." + dso_suffix:
|
|
libgfortran = fc("--print-file-name", "libgfortran.a", output=str).strip()
|
|
env.append_flags(
|
|
"EXTRA_LINK_FLAGS", "-L{0} -lgfortran".format(os.path.dirname(libgfortran))
|
|
)
|
|
|
|
def install(self, spec, prefix):
|
|
mkdir(prefix.bin)
|
|
if spec.satisfies("+openmp~debug"):
|
|
install("optimize_mp/sw4", prefix.bin)
|
|
elif spec.satisfies("+openmp+debug"):
|
|
install("debug_mp/sw4", prefix.bin)
|
|
elif spec.satisfies("~openmp~debug"):
|
|
install("optimize/sw4", prefix.bin)
|
|
elif spec.satisfies("~openmp+debug"):
|
|
install("debug/sw4", prefix.bin)
|
|
install_tree("pytest", prefix.test)
|