sparse_vanka.h

// ---------------------------------------------------------------------
//
// Copyright (C) 1999 - 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.md at
// the top level directory of deal.II.
//
// ---------------------------------------------------------------------
 
#ifndef dealii_sparse_vanka_h
#define dealii_sparse_vanka_h
 
 
 
#include <deal.II/base/config.h>
 
#include <deal.II/base/multithread_info.h>
#include <deal.II/base/smartpointer.h>
 
#include <map>
#include <vector>
 
DEAL_II_NAMESPACE_OPEN
 
template <typename number>
class FullMatrix;
template <typename number>
class SparseMatrix;
template <typename number>
class Vector;
 
template <typename number>
class SparseVanka;
template <typename number>
class SparseBlockVanka;
 
template <typename number>
class SparseVanka
{
public:
  using size_type = types::global_dof_index;
 
  SparseVanka();
 
  SparseVanka(const SparseMatrix<number> &M,
              const std::vector<bool> &   selected,
              const bool                  conserve_memory = false,
              const unsigned int n_threads = MultithreadInfo::n_threads());
 
  ~SparseVanka();
 
  class AdditionalData
  {
  public:
    AdditionalData(const std::vector<bool> &selected,
                   const bool               conserve_memory = false,
                   const unsigned int n_threads = MultithreadInfo::n_threads());
 
    const std::vector<bool> &selected;
 
    const bool conserve_mem;
 
    const unsigned int n_threads;
  };
 
 
  void
  initialize(const SparseMatrix<number> &M,
             const AdditionalData &      additional_data);
 
  template <typename number2>
  void
  vmult(Vector<number2> &dst, const Vector<number2> &src) const;
 
  template <typename number2>
  void
  Tvmult(Vector<number2> &dst, const Vector<number2> &src) const;
 
  size_type
  m() const;
 
  size_type
  n() const;
 
protected:
  template <typename number2>
  void
  apply_preconditioner(Vector<number2> &              dst,
                       const Vector<number2> &        src,
                       const std::vector<bool> *const dof_mask = nullptr) const;
 
  std::size_t
  memory_consumption() const;
 
private:
  SmartPointer<const SparseMatrix<number>, SparseVanka<number>> matrix;
 
  bool conserve_mem;
 
  const std::vector<bool> *selected;
 
  unsigned int n_threads;
 
  mutable std::vector<SmartPointer<FullMatrix<float>, SparseVanka<number>>>
    inverses;
 
  size_type _m;
 
  size_type _n;
 
  void
  compute_inverses();
 
  void
  compute_inverses(const size_type begin, const size_type end);
 
  void
  compute_inverse(const size_type row, std::vector<size_type> &local_indices);
 
  template <typename T>
  friend class SparseBlockVanka;
};
 
 
 
template <typename number>
class SparseBlockVanka : public SparseVanka<number>
{
public:
  using size_type = types::global_dof_index;
 
  enum BlockingStrategy
  {
    index_intervals,
    adaptive
  };
 
  SparseBlockVanka(const SparseMatrix<number> &M,
                   const std::vector<bool> &   selected,
                   const unsigned int          n_blocks,
                   const BlockingStrategy      blocking_strategy,
                   const bool                  conserve_memory = false,
                   const unsigned int n_threads = MultithreadInfo::n_threads());
 
  template <typename number2>
  void
  vmult(Vector<number2> &dst, const Vector<number2> &src) const;
 
  std::size_t
  memory_consumption() const;
 
private:
  const unsigned int n_blocks;
 
  std::vector<std::vector<bool>> dof_masks;
 
  void
  compute_dof_masks(const SparseMatrix<number> &M,
                    const std::vector<bool> &   selected,
                    const BlockingStrategy      blocking_strategy);
};
 
/* ---------------------------------- Inline functions ------------------- */
 
#ifndef DOXYGEN
 
template <typename number>
inline typename SparseVanka<number>::size_type
SparseVanka<number>::m() const
{
  Assert(_m != 0, ExcNotInitialized());
  return _m;
}
 
template <typename number>
inline typename SparseVanka<number>::size_type
SparseVanka<number>::n() const
{
  Assert(_n != 0, ExcNotInitialized());
  return _n;
}
 
template <typename number>
template <typename number2>
inline void
SparseVanka<number>::Tvmult(Vector<number2> & /*dst*/,
                            const Vector<number2> & /*src*/) const
{
  AssertThrow(false, ExcNotImplemented());
}
 
#endif // DOXYGEN
 
DEAL_II_NAMESPACE_CLOSE
 
#endif