mg_transfer_matrix_free.h

// ---------------------------------------------------------------------
//
// Copyright (C) 2016 - 2016 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__mg_transfer_matrix_free_h
#define dealii__mg_transfer_matrix_free_h

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

#include <deal.II/lac/parallel_vector.h>
#include <deal.II/multigrid/mg_base.h>
#include <deal.II/multigrid/mg_constrained_dofs.h>
#include <deal.II/base/mg_level_object.h>
#include <deal.II/multigrid/mg_transfer.h>
#include <deal.II/matrix_free/shape_info.h>

#include <deal.II/dofs/dof_handler.h>


DEAL_II_NAMESPACE_OPEN



template <int dim, typename Number>
class MGTransferMatrixFree : public MGLevelGlobalTransfer<parallel::distributed::Vector<Number> >
{
public:
  MGTransferMatrixFree ();

  MGTransferMatrixFree (const MGConstrainedDoFs &mg_constrained_dofs);

  virtual ~MGTransferMatrixFree ();

  void initialize_constraints (const MGConstrainedDoFs &mg_constrained_dofs);

  void clear ();

  void build (const DoFHandler<dim,dim> &mg_dof);

  virtual void prolongate (const unsigned int                           to_level,
                           parallel::distributed::Vector<Number>       &dst,
                           const parallel::distributed::Vector<Number> &src) const;

  virtual void restrict_and_add (const unsigned int from_level,
                                 parallel::distributed::Vector<Number>       &dst,
                                 const parallel::distributed::Vector<Number> &src) const;

  DeclException0(ExcNoProlongation);

  std::size_t memory_consumption () const;

private:

  unsigned int fe_degree;

  bool element_is_continuous;

  unsigned int n_components;

  unsigned int n_child_cell_dofs;

  std::vector<std::vector<unsigned int> > level_dof_indices;

  std::vector<std::vector<std::pair<unsigned int,unsigned int> > > parent_child_connect;

  std::vector<unsigned int> n_owned_level_cells;

  internal::MatrixFreeFunctions::ShapeInfo<Number> shape_info;

  mutable AlignedVector<VectorizedArray<Number> > evaluation_data;

  std::vector<AlignedVector<VectorizedArray<Number> > > weights_on_refined;

  std::vector<std::vector<std::vector<unsigned short> > > dirichlet_indices;

  template <int degree>
  void do_prolongate_add(const unsigned int                           to_level,
                         parallel::distributed::Vector<Number>       &dst,
                         const parallel::distributed::Vector<Number> &src) const;

  template <int degree>
  void do_restrict_add(const unsigned int                           from_level,
                       parallel::distributed::Vector<Number>       &dst,
                       const parallel::distributed::Vector<Number> &src) const;
};


DEAL_II_NAMESPACE_CLOSE

#endif