MFEM version 3.1 (#749)

MFEM is a free, lightweight, scalable C++ library for finite element methods. Includes lapack, hypre, metis, suite-sparse, mpi variants & tests for serial and parallel versions.
2016-04-26 10:00:24 -07:00 · 2016-04-26 10:00:24 -07:00 · f70046e6b7
commit f70046e6b7
parent ed16bd133a
1 changed files with 125 additions and 0 deletions
--- a/var/spack/repos/builtin/packages/mfem/package.py
+++ b/var/spack/repos/builtin/packages/mfem/package.py
@ -0,0 +1,125 @@
+from spack import *
+import glob, string
+
+class Mfem(Package):
+    """Free, lightweight, scalable C++ library for finite element methods."""
+
+    homepage = 'http://www.mfem.org'
+    url      = 'https://github.com/mfem/mfem'
+
+#    version('3.1', git='https://github.com/mfem/mfem.git',
+#            commit='dbae60fe32e071989b52efaaf59d7d0eb2a3b574')
+
+    version('3.1', '841ea5cf58de6fae4de0f553b0e01ebaab9cd9c67fa821e8a715666ecf18fc57',
+            url='http://goo.gl/xrScXn', expand=False)
+
+    variant('metis', default=False, description='Activate support for metis')
+    variant('hypre', default=False, description='Activate support for hypre')
+    variant('suite-sparse', default=False,
+            description='Activate support for SuiteSparse')
+    variant('mpi', default=False, description='Activate support for MPI')
+    variant('lapack', default=False, description='Activate support for LAPACK')
+    variant('debug', default=False, description='Build debug version')
+
+    depends_on('blas', when='+lapack')
+    depends_on('lapack', when='+lapack')
+
+    depends_on('mpi', when='+mpi')
+    depends_on('metis', when='+mpi')
+    depends_on('hypre', when='+mpi')
+
+    depends_on('hypre', when='+hypre')
+
+    depends_on('metis@4:', when='+metis')
+
+    depends_on('suite-sparse', when='+suite-sparse')
+    depends_on('blas', when='+suite-sparse')
+    depends_on('lapack', when='+suite-sparse')
+    depends_on('metis@5:', when='+suite-sparse ^suite-sparse@4.5:')
+    depends_on('cmake', when='^metis@5:')
+
+    def check_variants(self, spec):
+        if '+mpi' in spec and ('+hypre' not in spec or '+metis' not in spec):
+            raise InstallError('mfem+mpi must be built with +hypre ' +
+                               'and +metis!')
+        if '+suite-sparse' in spec and ('+metis' not in spec or
+                                        '+lapack' not in spec):
+            raise InstallError('mfem+suite-sparse must be built with ' +
+            '+metis and +lapack!')
+        if 'metis@5:' in spec and '%clang' in spec and ('^cmake %gcc' not in spec):
+            raise InstallError('To work around CMake bug with clang, must ' +
+                               'build mfem with mfem[+variants] %clang ' +
+                               '^cmake %gcc to force CMake to build with gcc')
+        return
+
+    def install(self, spec, prefix):
+        self.check_variants(spec)
+
+        options = ['PREFIX=%s' % prefix]
+
+        if '+lapack' in spec:
+            lapack_lib = '-L{0} -llapack -L{1} -lblas'.format(
+                spec['lapack'].prefix.lib, spec['blas'].prefix.lib)
+            options.extend(['MFEM_USE_LAPACK=YES',
+                            'LAPACK_OPT=-I%s' % spec['lapack'].prefix.include,
+                            'LAPACK_LIB=%s' % lapack_lib])
+
+        if '+hypre' in spec:
+            options.extend(['HYPRE_DIR=%s' % spec['hypre'].prefix,
+                            'HYPRE_OPT=-I%s' % spec['hypre'].prefix.include,
+                            'HYPRE_LIB=-L%s' % spec['hypre'].prefix.lib +
+                            ' -lHYPRE'])
+
+        if '+metis' in spec:
+            metis_lib = '-L%s -lmetis' % spec['metis'].prefix.lib
+            if spec['metis'].satisfies('@5:'):
+                metis_str = 'MFEM_USE_METIS_5=YES'
+            else:
+                metis_str = 'MFEM_USE_METIS_5=NO'
+            options.extend([metis_str,
+                            'METIS_DIR=%s' % spec['metis'].prefix,
+                            'METIS_OPT=-I%s' % spec['metis'].prefix.include,
+                            'METIS_LIB=%s' % metis_lib])
+
+        if '+mpi' in spec: options.extend(['MFEM_USE_MPI=YES'])
+
+        if '+suite-sparse' in spec:
+            ssp = spec['suite-sparse'].prefix
+            ss_lib = '-L%s' % ssp.lib
+            ss_lib += (' -lumfpack -lcholmod -lcolamd -lamd -lcamd' +
+                        ' -lccolamd -lsuitesparseconfig')
+
+            no_librt_archs = ['darwin-i686', 'darwin-x86_64']
+            no_rt = any(map(lambda a: spec.satisfies('='+a), no_librt_archs))
+            if not no_rt: ss_lib += ' -lrt'
+            ss_lib += (' ' + metis_lib + ' ' + lapack_lib)
+
+            options.extend(['MFEM_USE_SUITESPARSE=YES',
+                            'SUITESPARSE_DIR=%s' % ssp,
+                            'SUITESPARSE_OPT=-I%s' % ssp.include,
+                            'SUITESPARSE_LIB=%s' % ss_lib])
+
+        if '+debug' in spec: options.extend(['MFEM_DEBUG=YES'])
+
+        # Dirty hack to cope with URL redirect
+        tgz_file = string.split(self.url,'/')[-1]
+        tar = which('tar')
+        tar('xzvf', tgz_file)
+        cd(glob.glob('mfem*')[0])
+        # End dirty hack to cope with URL redirect
+
+        make('config', *options)
+        make('all')
+
+        # Run a small test before installation
+        args = ['-m', join_path('data','star.mesh'), '--no-visualization']
+        if '+mpi' in spec:
+            Executable(join_path(spec['mpi'].prefix.bin,
+                                 'mpirun'))('-np',
+                                            '4',
+                                            join_path('examples','ex1p'),
+                                            *args)
+        else:
+            Executable(join_path('examples', 'ex1'))(*args)
+
+        make('install')