spack/var/spack/repos/builtin/packages/mpas-model/package.py
2022-01-14 22:50:21 -08:00

103 lines
3.7 KiB
Python

# Copyright 2013-2022 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 import *
class MpasModel(MakefilePackage):
"""The Model for Prediction Across Scales (MPAS) is a collaborative
project for developing atmosphere, ocean and other earth-system
simulation components for use in climate, regional climate and weather
studies."""
homepage = "https://mpas-dev.github.io/"
url = "https://github.com/MPAS-Dev/MPAS-Model/archive/v7.0.tar.gz"
maintainers = ['t-brown']
version('7.1', sha256='9b5c181b7d0163ae33d24d7a79ede6990495134b58cf4500ba5c8c94192102bc')
version('7.0', sha256='f898ce257e66cff9e29320458870570e55721d16cb000de7f2cc27de7fdef14f')
version('6.3', sha256='e7f1d9ebfeb6ada37d42a286aaedb2e69335cbc857049dc5c5544bb51e7a8db8')
version('6.2', sha256='2a81825a62a468bf5c56ef9d9677aa2eb88acf78d4f996cb49a7db98b94a6b16')
depends_on('mpi')
depends_on('parallelio')
patch('makefile.patch', when='@7.0')
parallel = False
resource(when='@6.2:6.3',
name='MPAS-Data',
git='https://github.com/MPAS-Dev/MPAS-Data.git',
commit='33561790de8b43087ab850be833f51a4e605f1bb')
resource(when='@7.0:',
name='MPAS-Data',
git='https://github.com/MPAS-Dev/MPAS-Data.git',
tag='v7.0')
def target(self, model, action):
spec = self.spec
satisfies = spec.satisfies
fflags = [self.compiler.openmp_flag]
cppflags = ['-D_MPI']
if satisfies('%gcc'):
fflags.extend([
'-ffree-line-length-none',
'-fconvert=big-endian',
'-ffree-form',
'-fdefault-real-8',
'-fdefault-double-8',
])
cppflags.append('-DUNDERSCORE')
elif satisfies('%fj'):
fflags.extend([
'-Free',
'-Fwide',
'-CcdRR8',
])
elif satisfies('%intel'):
fflags.extend([
'-r8',
'-convert big_endian',
'-FR',
])
cppflags.append('-DUNDERSCORE')
targets = [
'FC_PARALLEL={0}'.format(spec['mpi'].mpifc),
'CC_PARALLEL={0}'.format(spec['mpi'].mpicc),
'CXX_PARALLEL={0}'.format(spec['mpi'].mpicxx),
'FC_SERIAL={0}'.format(spack_fc),
'CC_SERIAL={0}'.format(spack_cc),
'CXX_SERIAL={0}'.format(spack_cxx),
'CFLAGS_OMP={0}'.format(self.compiler.openmp_flag),
'FFLAGS_OMP={0}'.format(' '.join(fflags)),
'CPPFLAGS={0}'.format(' '.join(cppflags)),
'PIO={0}'.format(spec['parallelio'].prefix),
'NETCDF={0}'.format(spec['netcdf-c'].prefix),
'NETCDFF={0}'.format(spec['netcdf-fortran'].prefix)
]
if satisfies('^parallelio+pnetcdf'):
targets.append(
'PNETCDF={0}'.format(spec['parallel-netcdf'].prefix)
)
targets.extend([
'USE_PIO2=true', 'CPP_FLAGS=-D_MPI', 'OPENMP=true',
'CORE={0}'.format(model), action
])
return targets
def build(self, spec, prefix):
copy_tree(join_path('MPAS-Data', 'atmosphere'),
join_path('src', 'core_atmosphere', 'physics'))
make(*self.target('init_atmosphere', 'all'))
mkdir('bin')
copy('init_atmosphere_model', 'bin')
make(*self.target('init_atmosphere', 'clean'))
make(*self.target('atmosphere', 'all'))
copy('atmosphere_model', 'bin')
def install(self, spec, prefix):
install_tree('bin', prefix.bin)