From 1b558aaa81d16c97b59aecfa7229119dadd4dbbb Mon Sep 17 00:00:00 2001
From: Robert Blake <blake14@llnl.gov>
Date: Sun, 18 Oct 2020 21:09:13 -0700
Subject: mpich: Add `external find` support (#18330)

* Adding external support to mpich

* Removing debugging print statement.
---
 var/spack/repos/builtin/packages/mpich/package.py | 106 ++++++++++++++++++++++
 1 file changed, 106 insertions(+)

diff --git a/var/spack/repos/builtin/packages/mpich/package.py b/var/spack/repos/builtin/packages/mpich/package.py
index b120762bf9..6d7df8ac9b 100644
--- a/var/spack/repos/builtin/packages/mpich/package.py
+++ b/var/spack/repos/builtin/packages/mpich/package.py
@@ -6,6 +6,7 @@
 from spack import *
 import os
 import sys
+import re
 
 
 class Mpich(AutotoolsPackage):
@@ -18,6 +19,8 @@ class Mpich(AutotoolsPackage):
     list_url = "http://www.mpich.org/static/downloads/"
     list_depth = 1
 
+    executables = ['^mpichversion$']
+
     version('develop', submodules=True)
     version('3.3.2', sha256='4bfaf8837a54771d3e4922c84071ef80ffebddbb6971a006038d91ee7ef959b9')
     version('3.3.1', sha256='fe551ef29c8eea8978f679484441ed8bb1d943f6ad25b63c235d4b9243d551e5')
@@ -175,6 +178,109 @@ spack package at this time.''',
     conflicts('+pci', when='@:3.2~hydra')
     conflicts('+libxml2', when='@:3.2~hydra')
 
+    @classmethod
+    def determine_version(cls, exe):
+        output = Executable(exe)(output=str, error=str)
+        match = re.search(r'MPICH Version:\s+(\S+)', output)
+        return match.group(1) if match else None
+
+    @classmethod
+    def determine_variants(cls, exes, version):
+        def get_spack_compiler_spec(path):
+            spack_compilers = spack.compilers.find_compilers([path])
+            actual_compiler = None
+            # check if the compiler actually matches the one we want
+            for spack_compiler in spack_compilers:
+                if os.path.dirname(spack_compiler.cc) == path:
+                    actual_compiler = spack_compiler
+                    break
+            return actual_compiler.spec if actual_compiler else None
+
+        def is_enabled(text):
+            if text in set(['t', 'true', 'enabled', 'enable', 'with',
+                            'yes', '1']):
+                return True
+            return False
+
+        def is_disabled(text):
+            if text in set(['f', 'false', 'disabled', 'disable',
+                            'without', 'no', '0']):
+                return True
+            return False
+
+        results = []
+        for exe in exes:
+            variants = ''
+            output = Executable(exe)(output=str, error=str)
+            if re.search(r'--with-hwloc-prefix=embedded', output):
+                variants += '~hwloc'
+
+            if re.search(r'--with-pm=hydra', output):
+                variants += '+hydra'
+            else:
+                variants += '~hydra'
+
+            match = re.search(r'--(\S+)-romio', output)
+            if match and is_enabled(match.group(1)):
+                variants += '+romio'
+            elif match and is_disabled(match.group(1)):
+                variants += '~romio'
+
+            if re.search(r'--with-ibverbs', output):
+                variants += '+verbs'
+            elif re.search(r'--without-ibverbs', output):
+                variants += '~verbs'
+
+            match = re.search(r'--enable-wrapper-rpath=(\S+)', output)
+            if match and is_enabled(match.group(1)):
+                variants += '+wrapperrpath'
+            match = re.search(r'--enable-wrapper-rpath=(\S+)', output)
+            if match and is_disabled(match.group(1)):
+                variants += '~wrapperrpath'
+
+            if re.search(r'--disable-fortran', output):
+                variants += '~fortran'
+
+            match = re.search(r'--with-slurm=(\S+)', output)
+            if match and is_enabled(match.group(1)):
+                variants += '+slurm'
+
+            if re.search(r'--enable-libxml2', output):
+                variants += '+libxml2'
+            elif re.search(r'--disable-libxml2', output):
+                variants += '~libxml2'
+
+            if re.search(r'--with-thread-package=argobots', output):
+                variants += '+argobots'
+
+            if re.search(r'--with-pmi=no', output):
+                variants += ' pmi=off'
+            elif re.search(r'--with-pmi=simple', output):
+                variants += ' pmi=pmi'
+            elif re.search(r'--with-pmi=pmi2/simple', output):
+                variants += ' pmi=pmi2'
+            elif re.search(r'--with-pmix', output):
+                variants += ' pmi=pmix'
+
+            match = re.search(r'MPICH Device:\s+(\S+)', output)
+            if match:
+                if match.group(1) == 'ch3:nemesis':
+                    variants += ' device=ch3'
+                else:
+                    variants += ' device=' + match.group(1)
+
+            match = re.search(r'--with-device=ch.\S+(ucx|ofi|mxm|tcp)', output)
+            if match:
+                variants += ' netmod=' + match.group(1)
+
+            match = re.search(r'MPICH CC:\s+(\S+)', output)
+            compiler_spec = get_spack_compiler_spec(
+                os.path.dirname(match.group(1)))
+            if compiler_spec:
+                variants += '%' + str(compiler_spec)
+            results.append(variants)
+        return results
+
     def setup_build_environment(self, env):
         env.unset('F90')
         env.unset('F90FLAGS')
-- 
cgit v1.2.3-70-g09d2