text: # 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.package import *
  
  
  class FenicsDolfinx(CMakePackage):
      """Next generation FEniCS problem solving environment"""
  
      homepage = "https://github.com/FEniCS/dolfinx"
      git = "https://github.com/FEniCS/dolfinx.git"
      url = "https://github.com/FEniCS/dolfinx/archive/v0.1.0.tar.gz"
      maintainers = ["chrisrichardson", "garth-wells", "nate-sime"]
  
      version("main", branch="main")
      version("0.4.1", sha256="68dcf29a26c750fcea5e02d8d58411e3b054313c3bf6fcbc1d0f08dd2851117f")
      version("0.3.0", sha256="4857d0fcb44a4e9bf9eb298ba5377abdee17a7ad0327448bdd06cce73d109bed")
      version("0.2.0", sha256="4c9b5a5c7ef33882c99299c9b4d98469fb7aa470a37a91bc5be3bb2fc5b863a4")
      version("0.1.0", sha256="0269379769b5b6d4d1864ded64402ecaea08054c2a5793c8685ea15a59af5e33")
  
      # Graph partitioner variants
      variant('partitioners',
              description='Graph partioning',
              default=('parmetis',),
              values=('kahip', 'parmetis', 'scotch'),
              multi=True,
              when='@0.4.0:')
      variant("kahip", default=False, when="@0.1.0:0.3.0", description="kahip support")
      variant("parmetis", default=False, when="@0.1.0:0.3.0",
              description="parmetis support")
  
      # Graph partitioner dependencies for @0.4.0:
      depends_on('kahip@3.12:', when="partitioners=kahip @main")
      depends_on('kahip@3.11', when="partitioners=kahip @:0.4.1")
      depends_on('parmetis', when="partitioners=parmetis")
      depends_on('scotch+mpi', when="partitioners=scotch")
  
      # Graph partitioner dependencies for "@0.1.0:0.3.0"
      depends_on('kahip', when="+kahip")
      depends_on('parmetis', when="+parmetis")
      depends_on('scotch+mpi', when="@0.1.0:0.3.0")
  
      variant("slepc", default=True, description="slepc support")
      variant("adios2", default=False, description="adios2 support")
  
      depends_on("cmake@3.18:", type="build")
      depends_on("pkgconfig", type="build")
      depends_on("mpi")
      depends_on("hdf5+mpi")
      depends_on("boost@1.7.0:+filesystem+program_options+timer")
  
      depends_on("petsc+mpi+shared")
      depends_on("petsc+mpi+shared@3.15.0:", when="@0.1.0")
  
      depends_on("xtl@0.7.2:")
      depends_on("xtensor@0.23.10:")
  
      depends_on("slepc", when="+slepc")
      depends_on("adios2+mpi", when="+adios2")
  
      depends_on("fenics-ufcx@main", when="@main")
      depends_on("fenics-ufcx@0.4.2", when="@0.4.1")
      depends_on("py-fenics-ffcx@0.3.0", type=("build", "run"), when="@0.3.0")
      depends_on("py-fenics-ffcx@0.2.0", type=("build", "run"), when="@0.2.0")
      depends_on("py-fenics-ffcx@0.1.0", type=("build", "run"), when="@0.1.0")
  
      depends_on("fenics-basix")
      depends_on("fenics-basix@main", when="@main")
      depends_on("fenics-basix@0.4.2", when="@0.4.1")
      depends_on("fenics-basix@0.3.0", when="@0.3.0")
      depends_on("fenics-basix@0.2.0", when="@0.2.0")
      depends_on("fenics-basix@0.1.0", when="@0.1.0")
  
      conflicts('%gcc@:8', msg='Improved C++17 support required')
  
      root_cmakelists_dir = "cpp"
  
      def cmake_args(self):
          args = [
              self.define('DOLFINX_SKIP_BUILD_TESTS', True),
              self.define_from_variant('DOLFINX_ENABLE_SLEPC', 'slepc'),
              self.define_from_variant('DOLFINX_ENABLE_ADIOS2', 'adios2'),
          ]
  
          if self.spec.satisfies('@0.4.0:'):
              args += [
                  self.define('DOLFINX_ENABLE_KAHIP', 'partitioners=kahip' in self.spec),
                  self.define('DOLFINX_ENABLE_PARMETIS',
                              'partitioners=parmetis' in self.spec),
                  self.define('DOLFINX_ENABLE_SCOTCH', 'partitioners=scotch' in self.spec)
              ]
  
          if self.spec.satisfies('@:0.3.0'):
              args.append(self.define_from_variant('DOLFINX_ENABLE_KAHIP', 'kahip'))
              args.append(self.define_from_variant('DOLFINX_ENABLE_PARMETIS', 'parmetis'))
              args.append(self.define('Python3_ROOT_DIR', self.spec['python'].home))
              args.append(self.define('Python3_FIND_STRATEGY', 'LOCATION'))
  
          return args
  