Program Listing for File matrix_free_utils.h

↰ Return to documentation for file (include/matrix_free_utils.h)

// ---------------------------------------------------------------------
//
// Copyright (C) 2024 by Luca Heltai
//
// This file is part of the ImmersX application, based on
// the deal.II library.
//
// The ImmersX application is free software; you can use
// it, redistribute it, and/or modify it under the terms of the Apache-2.0
// License WITH LLVM-exception as published by the Free Software Foundation;
// either version 3.0 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
// of the ImmersX distribution.
//
// ---------------------------------------------------------------------

#ifndef rdlm_mf_utils
#define rdlm_mf_utils


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

#include <deal.II/fe/fe_q.h>

#include <deal.II/lac/la_parallel_vector.h>

#include <deal.II/matrix_free/fe_point_evaluation.h>

#include <inclusions.h>

#include <fstream>

using namespace dealii;

#if DEAL_II_VERSION_GTE(9, 7, 0)
#else
#  include <deal.II/base/smartpointer.h>
template <typename T, typename P = void>
using ObserverPointer = SmartPointer<T, P>;
#endif



template <int dim, typename number, int n_components = 1>
class CouplingOperator
{
public:
  using VectorType = LinearAlgebra::distributed::Vector<number>;

  CouplingOperator(
    const Inclusions<dim>           &inclusions,
    const DoFHandler<dim>           &dof_handler,
    const AffineConstraints<number> &constraints = AffineConstraints<number>(),
    const MappingQ<dim>             &mapping     = MappingQ1<dim>(),
    const FiniteElement<dim>        &fe          = FE_Q<dim>(1));

  void
  initialize_dof_vector(VectorType &vec) const;

  void
  vmult(VectorType &dst, const VectorType &src) const;

  void
  Tvmult(VectorType &dst, const VectorType &src) const;

  void
  vmult_add(VectorType &dst, const VectorType &src) const;

  void
  Tvmult_add(VectorType &dst, const VectorType &src) const;


private:
  Utilities::MPI::RemotePointEvaluation<dim> rpe;
  ObserverPointer<const Mapping<dim>> mapping;
  ObserverPointer<const FiniteElement<dim>> fe;
  ObserverPointer<const DoFHandler<dim>> dof_handler;
  ObserverPointer<const Inclusions<dim>> inclusions;
  ObserverPointer<const AffineConstraints<number>> constraints;
  const unsigned int n_coefficients;
};



#endif