warpx: new package (#9584)

1 files changed, 78 insertions, 0 deletions

diff --git a/var/spack/repos/builtin/packages/warpx/package.py b/var/spack/repos/builtin/packages/warpx/package.py
new file mode 100644
index 0000000000..7be45eb37c
--- /dev/null
+++ b/var/spack/repos/builtin/packages/warpx/package.py
@@ -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')