elementwise_operator.h

/*  ______________________________________________________________________
 *
 *  ExaDG - High-Order Discontinuous Galerkin for the Exa-Scale
 *
 *  Copyright (C) 2021 by the ExaDG authors
 *
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <https://www.gnu.org/licenses/>.
 *  ______________________________________________________________________
 */
 
#ifndef INCLUDE_EXADG_OPERATORS_ELEMENTWISE_OPERATOR_H_
#define INCLUDE_EXADG_OPERATORS_ELEMENTWISE_OPERATOR_H_
 
#include <exadg/solvers_and_preconditioners/solvers/elementwise_krylov_solvers.h>
 
namespace ExaDG
{
namespace Elementwise
{
template<int dim, typename Number, typename Operator>
class OperatorBase
{
public:
  OperatorBase(Operator const & operator_in) : op(operator_in), current_cell(1), problem_size(1)
  {
  }
 
  dealii::MatrixFree<dim, Number> const &
  get_matrix_free() const
  {
    return op.get_matrix_free();
  }
 
  unsigned int
  get_dof_index() const
  {
    return op.get_dof_index();
  }
 
  unsigned int
  get_quad_index() const
  {
    return op.get_quad_index();
  }
 
  void
  setup(unsigned int const cell, unsigned int const size)
  {
    current_cell = cell;
 
    problem_size = size;
  }
 
  unsigned int
  get_problem_size() const
  {
    return problem_size;
  }
 
  void
  vmult(dealii::VectorizedArray<Number> * dst, dealii::VectorizedArray<Number> * src) const
  {
    // set dst vector to zero
    Elementwise::vector_init(dst, problem_size);
 
    // evaluate block diagonal
    op.apply_add_block_diagonal_elementwise(current_cell, dst, src, problem_size);
  }
 
private:
  Operator const & op;
 
  unsigned int current_cell;
 
  unsigned int problem_size;
};
 
} // namespace Elementwise
} // namespace ExaDG
 
#endif /* INCLUDE_EXADG_OPERATORS_ELEMENTWISE_OPERATOR_H_ */