Program Listing for File matrix_free_utils.h

↰ Return to documentation for file (include/immersx/algebra/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 <immersx/coupling/inclusions.h>

#include <fstream>

using namespace dealii;

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

namespace ImmersX
{


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

    MatrixFreeCouplingOperator(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;
  };



} // namespace ImmersX

#endif