Traits.hpp

#pragma once
 
#include <type_traits>
 
#include <petscmat.h>
#include <petscvec.h>
 
#include <dune/grid/common/partitionset.hh>
#include <amdis/linearalgebra/petsc/IndexDistribution.hpp>
#include <amdis/linearalgebra/petsc/MatrixNnzStructure.hpp>
#include <amdis/linearalgebra/petsc/MatrixBackend.hpp>
#include <amdis/linearalgebra/petsc/VectorBackend.hpp>
 
namespace AMDiS
{
  struct PetscTraits
  {
    template <class Basis>
    using IndexDist = PetscIndexDistribution_t<Basis>;
 
    template <class RowBasis, class ColBasis>
    struct Matrix
    {
      using SparsityPattern = MatrixNnzStructure;
 
      static_assert(std::is_same_v<IndexDist<RowBasis>, IndexDist<ColBasis>>);
 
      template <class Value>
      using Impl = PetscMatrix<IndexDist<RowBasis>>;
    };
 
    template <class Basis>
    struct Vector
    {
      template <class Value>
      using Impl = PetscVector<IndexDist<Basis>>;
    };
 
    using PartitionSet = Dune::Partitions::Interior;
  };
 
  using BackendTraits = PetscTraits;
 
} // end namespace AMDiS