trilinos_linear_operator.h

// ---------------------------------------------------------------------
//
// Copyright (C) 2016 - 2018 by the deal.II authors
//
// This file is part of the deal.II library.
//
// The deal.II library is free software; you can use it, redistribute
// it, and/or modify it under the terms of the GNU Lesser General
// Public License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
// The full text of the license can be found in the file LICENSE at
// the top level of the deal.II distribution.
//
// ---------------------------------------------------------------------

#ifndef dealii_trilinos_linear_operator_h
#define dealii_trilinos_linear_operator_h

#include <deal.II/base/config.h>

#if defined(DEAL_II_WITH_TRILINOS)

#include <deal.II/lac/linear_operator.h>
#include <deal.II/lac/block_linear_operator.h>

DEAL_II_NAMESPACE_OPEN

namespace TrilinosWrappers
{

  // Forward declarations:
  class SparseMatrix;
  class PreconditionBase;
  class BlockSparseMatrix;

  namespace internal
  {
    namespace LinearOperatorImplementation
    {
      class TrilinosPayload;
    }

    namespace BlockLinearOperatorImplementation
    {
      template <typename PayloadBlockType>
      class TrilinosBlockPayload;
    }
  }




  template <typename Range, typename Domain = Range,
            typename Matrix>
  inline LinearOperator<Range, Domain, TrilinosWrappers::internal::LinearOperatorImplementation::TrilinosPayload>
  linear_operator(const TrilinosWrappers::SparseMatrix &operator_exemplar, const Matrix &matrix)
  {
    typedef TrilinosWrappers::SparseMatrix OperatorExemplar;
    typedef TrilinosWrappers::internal::LinearOperatorImplementation::TrilinosPayload Payload;
    return ::linear_operator<Range, Domain, Payload, OperatorExemplar, Matrix>(operator_exemplar, matrix);
  }


  template <typename Range, typename Domain = Range>
  inline LinearOperator<Range, Domain, TrilinosWrappers::internal::LinearOperatorImplementation::TrilinosPayload>
  linear_operator(const TrilinosWrappers::SparseMatrix &matrix)
  {
    typedef TrilinosWrappers::SparseMatrix Matrix;
    typedef TrilinosWrappers::internal::LinearOperatorImplementation::TrilinosPayload Payload;
    return ::linear_operator<Range, Domain, Payload, Matrix, Matrix>(matrix, matrix);
  }





  template <typename Range,
            typename Domain = Range>
  inline BlockLinearOperator<Range, Domain, TrilinosWrappers::internal::BlockLinearOperatorImplementation::TrilinosBlockPayload<TrilinosWrappers::internal::LinearOperatorImplementation::TrilinosPayload> >
  block_operator(const TrilinosWrappers::BlockSparseMatrix &block_matrix)
  {
    typedef TrilinosWrappers::BlockSparseMatrix BlockMatrix;
    typedef TrilinosWrappers::internal::LinearOperatorImplementation::TrilinosPayload PayloadBlockType;
    typedef TrilinosWrappers::internal::BlockLinearOperatorImplementation::TrilinosBlockPayload<PayloadBlockType> BlockPayload;
    return ::block_operator<Range,Domain,BlockPayload,BlockMatrix>(block_matrix);
  }


  template <size_t m, size_t n,
            typename Range,
            typename Domain = Range>
  inline BlockLinearOperator<Range, Domain, TrilinosWrappers::internal::BlockLinearOperatorImplementation::TrilinosBlockPayload<TrilinosWrappers::internal::LinearOperatorImplementation::TrilinosPayload> >
  block_operator(const std::array<std::array<LinearOperator<typename Range::BlockType, typename Domain::BlockType, TrilinosWrappers::internal::LinearOperatorImplementation::TrilinosPayload>, n>, m> &ops)
  {
    typedef TrilinosWrappers::internal::LinearOperatorImplementation::TrilinosPayload PayloadBlockType;
    typedef TrilinosWrappers::internal::BlockLinearOperatorImplementation::TrilinosBlockPayload<PayloadBlockType> BlockPayload;
    return ::block_operator<m,n,Range,Domain,BlockPayload>(ops);
  }


  template <typename Range,
            typename Domain = Range>
  inline BlockLinearOperator<Range, Domain, TrilinosWrappers::internal::BlockLinearOperatorImplementation::TrilinosBlockPayload<TrilinosWrappers::internal::LinearOperatorImplementation::TrilinosPayload> >
  block_diagonal_operator(const TrilinosWrappers::BlockSparseMatrix &block_matrix)
  {
    typedef TrilinosWrappers::BlockSparseMatrix BlockMatrix;
    typedef TrilinosWrappers::internal::LinearOperatorImplementation::TrilinosPayload PayloadBlockType;
    typedef TrilinosWrappers::internal::BlockLinearOperatorImplementation::TrilinosBlockPayload<PayloadBlockType> BlockPayload;
    return ::block_diagonal_operator<Range, Domain, BlockPayload, BlockMatrix>(block_matrix);
  }


  template <size_t m, typename Range, typename Domain = Range>
  inline BlockLinearOperator<Range, Domain, TrilinosWrappers::internal::BlockLinearOperatorImplementation::TrilinosBlockPayload<TrilinosWrappers::internal::LinearOperatorImplementation::TrilinosPayload> >
  block_diagonal_operator(const std::array<LinearOperator<typename Range::BlockType, typename Domain::BlockType, TrilinosWrappers::internal::LinearOperatorImplementation::TrilinosPayload>, m> &ops)
  {
    typedef TrilinosWrappers::internal::LinearOperatorImplementation::TrilinosPayload PayloadBlockType;
    typedef TrilinosWrappers::internal::BlockLinearOperatorImplementation::TrilinosBlockPayload<PayloadBlockType> BlockPayload;
    return ::block_diagonal_operator<m,Range,Domain,BlockPayload>(ops);
  }


} // namespace TrilinosWrappers

DEAL_II_NAMESPACE_CLOSE

#endif // DEAL_II_WITH_TRILINOS
#endif