warpx: new package (#9584)

This commit is contained in:
Dave 2018-10-19 04:31:22 -07:00 committed by Axel Huebl
parent 5f482b3c14
commit 343ad17383

View file

@ -0,0 +1,78 @@
# Copyright 2013-2018 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 Warpx(MakefilePackage):
"""WarpX is an advanced electromagnetic Particle-In-Cell code. It supports
many features including Perfectly-Matched Layers (PML) and mesh refinement.
In addition, WarpX is a highly-parallel and highly-optimized code and
features hybrid OpenMP/MPI parallelization, advanced vectorization
techniques and load balancing capabilities.
"""
homepage = "https://ecp-warpx.github.io/index.html"
url = "https://github.com/ECP-WarpX/WarpX"
version('master', git='https://github.com/ECP-WarpX/WarpX.git', tag='master')
version('dev', git='https://github.com/ECP-WarpX/WarpX.git', tag='dev')
depends_on('mpi', when='+mpi')
variant('dims',
default='3',
values=('1', '2', '3'),
multi=False,
description='Number of spatial dimensions')
variant('psatd', default=False, description='Enable PSATD solver')
variant('do_electrostatic', default=False, description='Include electrostatic solver')
variant('debug', default=False, description='Enable debugging features')
variant('tprof', default=False, description='Enable tiny profiling features')
variant('openmp', default=True, description='Enable OpenMP features')
resource(name='amrex',
git='https://github.com/AMReX-Codes/amrex.git',
tag='development',
destination='.')
resource(name='picsar',
git='https://bitbucket.org/berkeleylab/picsar.git',
tag='master',
destination='.')
def edit(self, spec, prefix):
comp = 'gcc'
vendors = {'%gcc': 'gcc', '%intel': 'intel'}
for key, value in vendors.items():
if self.spec.satisfies(key):
comp = value
def torf(s):
"Returns the string TRUE or FALSE"
return repr(s in spec).upper()
makefile = FileFilter('GNUmakefile')
makefile.filter('AMREX_HOME .*', 'AMREX_HOME = amrex')
makefile.filter('PICSAR_HOME .*', 'PICSAR_HOME = picsar')
makefile.filter('COMP .*', 'COMP = {0}'.format(comp))
makefile.filter('DIM .*',
'DIM = {0}'.format(int(spec.variants['dims'].value)))
makefile.filter('USE_PSATD .*',
'USE_PSATD = {0}'.format(torf('+psatd')))
makefile.filter('DO_ELECTROSTATIC .*',
'DO_ELECTROSTATIC = %s' % torf('+do_electrostatic'))
makefile.filter('USE_OMP .*',
'USE_OMP = {0}'.format(torf('+openmp')))
makefile.filter('DEBUG .*',
'DEBUG = {0}'.format(torf('+debug')))
makefile.filter('TINY_PROFILE .*',
'TINY_PROFILE = {0}'.format(torf('+tprof')))
makefile.filter('EBASE .*', 'EBASE = warpx')
def install(self, spec, prefix):
make('WarpxBinDir = {0}'.format(prefix.bin), 'all')