Program Listing for File reduced_poisson.h

Return to documentation for file (include/immersx/physics/reduced_poisson.h)

// ---------------------------------------------------------------------
//
// Copyright (C) 2026 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 dealii_reduced_poisson_h
#define dealii_reduced_poisson_h

#include <deal.II/base/function.h>
#include <deal.II/base/parsed_convergence_table.h>
#include <deal.II/base/quadrature_lib.h>
#include <deal.II/base/timer.h>

#include <deal.II/lac/block_linear_operator.h>
#include <deal.II/lac/generic_linear_algebra.h>
#include <deal.II/lac/la_parallel_block_vector.h>
#include <deal.II/lac/linear_operator.h>
#include <deal.II/lac/linear_operator_tools.h>
// #define MATRIX_FREE_PATH

#define FORCE_USE_OF_TRILINOS
#include <deal.II/base/conditional_ostream.h>
#include <deal.II/base/index_set.h>
#include <deal.II/base/parameter_acceptor.h>
#include <deal.II/base/parsed_function.h>
#include <deal.II/base/utilities.h>

#include <deal.II/distributed/grid_refinement.h>
#include <deal.II/distributed/solution_transfer.h>
#include <deal.II/distributed/tria.h>

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

#include <deal.II/fe/fe_nothing.h>
#include <deal.II/fe/fe_q.h>
#include <deal.II/fe/fe_system.h>
#include <deal.II/fe/fe_values.h>
#include <deal.II/fe/mapping_fe_field.h>
#include <deal.II/fe/mapping_q.h>

#include <deal.II/grid/grid_generator.h>
#include <deal.II/grid/grid_in.h>
#include <deal.II/grid/grid_tools.h>
#include <deal.II/grid/manifold_lib.h>

#include <deal.II/lac/affine_constraints.h>
#include <deal.II/lac/dynamic_sparsity_pattern.h>
#include <deal.II/lac/full_matrix.h>
#include <deal.II/lac/petsc_precondition.h>
#include <deal.II/lac/petsc_solver.h>
#include <deal.II/lac/petsc_sparse_matrix.h>
#include <deal.II/lac/petsc_vector.h>
#include <deal.II/lac/solver_cg.h>
#include <deal.II/lac/solver_gmres.h>
#include <deal.II/lac/solver_minres.h>
#include <deal.II/lac/sparsity_tools.h>
#include <deal.II/lac/vector.h>

#include <deal.II/numerics/matrix_tools.h>

#ifdef MATRIX_FREE_PATH
#  include <deal.II/matrix_free/operators.h>

#  include <deal.II/multigrid/mg_coarse.h>
#  include <deal.II/multigrid/mg_matrix.h>
#  include <deal.II/multigrid/mg_smoother.h>
#  include <deal.II/multigrid/mg_tools.h>
#  include <deal.II/multigrid/mg_transfer_matrix_free.h>
#  include <deal.II/multigrid/multigrid.h>
#endif

#include <deal.II/numerics/data_out.h>
#include <deal.II/numerics/error_estimator.h>
#include <deal.II/numerics/vector_tools.h>

#include <deal.II/opencascade/manifold_lib.h>
#include <deal.II/opencascade/utilities.h>

#include <immersx/algebra/matrix_free_utils.h>
#include <immersx/coupling/reduced_coupling.h>

#ifdef DEAL_II_WITH_OPENCASCADE
#  include <TopoDS.hxx>
#endif
#include <cmath>
#include <fstream>
#include <iostream>
#include <memory>


namespace ImmersX
{
  namespace LA
  {
#if defined(DEAL_II_WITH_PETSC) && !defined(DEAL_II_PETSC_WITH_COMPLEX) && \
  !(defined(DEAL_II_WITH_TRILINOS) && defined(FORCE_USE_OF_TRILINOS))
    using namespace dealii::LinearAlgebraPETSc;
#  define USE_PETSC_LA
#elif defined(DEAL_II_WITH_TRILINOS)
    using namespace dealii::LinearAlgebraTrilinos;
#else
#  error DEAL_II_WITH_PETSC or DEAL_II_WITH_TRILINOS required
#endif
  } // namespace LA

  template <int spacedim, int reduced_dim = 1, int cross_section_dim = 2>
  class ReducedPoissonParameters : public ParameterAcceptor
  {
  public:
    ReducedPoissonParameters();

    std::string                   output_directory = ".";
    std::string                   output_name      = "solution";
    unsigned int                  fe_degree        = 1;
    std::list<types::boundary_id> dirichlet_ids{0};
    std::string                   name_of_grid        = "hyper_cube";
    std::string                   arguments_for_grid  = "-1: 1: false";
    std::string                   refinement_strategy = "fixed_fraction";
    double                        coarsening_fraction = 0.0;
    double                        refinement_fraction = 0.3;
    unsigned int                  n_refinement_cycles = 1;
    unsigned int                  max_cells           = 20000;
    mutable ParameterAcceptorProxy<Functions::ParsedFunction<spacedim>> rhs;
    mutable ParameterAcceptorProxy<Functions::ParsedFunction<spacedim>> bc;

    mutable ParameterAcceptorProxy<ReductionControl> inner_control;
    mutable ParameterAcceptorProxy<ReductionControl> outer_control;

    bool        output_results_before_solving = false;
    bool        estimate_condition_number     = false;
    std::string solver_name                   = "AL";
    bool        assemble_full_AL_system       = false;

    mutable ParsedConvergenceTable convergence_table;

    ReducedCouplingParameters<reduced_dim, cross_section_dim, spacedim, 1>
      reduced_coupling_parameters;
  };


  template <int dim,
            int spacedim          = dim,
            int reduced_dim       = 1,
            int cross_section_dim = 2>
  class ReducedPoisson : public EnableObserverPointer
  {
  public:
    ReducedPoisson(
      const ReducedPoissonParameters<spacedim, reduced_dim, cross_section_dim>
        &par);
    void
    make_grid();
    void
    setup_fe();
    void
    setup_dofs();
#ifndef MATRIX_FREE_PATH
    void
    assemble_poisson_system();
#else
    void
                          assemble_rhs();
#endif
    void
    assemble_coupling_system();

    void
    run();

    void
    solve();

    unsigned int
    n_reduced_dofs() const;

    double
    coupling_matrix_frobenius_norm() const;

    double
    bulk_solution_l2_norm() const;

    double
    multiplier_solution_l2_norm() const;

    bool
    bulk_solution_is_finite() const;

    bool
    multiplier_solution_is_finite() const;

    void
    refine_and_transfer();

    std::string
    output_solution() const;

    void
    output_results() const;

    void
    print_parameters() const;

  private:
    const ReducedPoissonParameters<spacedim, reduced_dim, cross_section_dim>
                                                  &par;
    MPI_Comm                                       mpi_communicator;
    ConditionalOStream                             pcout;
    mutable TimerOutput                            computing_timer;
    parallel::distributed::Triangulation<spacedim> tria;
    std::unique_ptr<FiniteElement<spacedim>>       fe;

    std::unique_ptr<Quadrature<spacedim>> quadrature;

    DoFHandler<spacedim> dh;

    ReducedCoupling<reduced_dim, cross_section_dim, spacedim, 1>
      reduced_coupling;

    std::vector<IndexSet> owned_dofs;
    std::vector<IndexSet> relevant_dofs;

    AffineConstraints<double> constraints;

    LA::MPI::SparseMatrix coupling_matrix;
    LA::MPI::SparseMatrix coupling_matrix_transpose;
    LA::MPI::SparseMatrix inclusion_matrix;
    MappingQ<spacedim>    mapping;
#ifdef MATRIX_FREE_PATH
    using VectorType      = LinearAlgebra::distributed::Vector<double>;
    using BlockVectorType = LinearAlgebra::distributed::BlockVector<double>;
    std::unique_ptr<MatrixFreeCouplingOperator<spacedim, double, 1>>
                                                       coupling_operator;
    MatrixFreeOperators::LaplaceOperator<spacedim, -1> stiffness_matrix;
    using LevelMatrixType = MatrixFreeOperators::LaplaceOperator<
      spacedim,
      -1,
      -1,
      1,
      LinearAlgebra::distributed::Vector<float>>;
    MGLevelObject<LevelMatrixType> mg_matrices;
    MGConstrainedDoFs              mg_constrained_dofs;
#else
    LA::MPI::SparseMatrix stiffness_matrix;
    using VectorType      = LA::MPI::Vector;
    using BlockVectorType = LA::MPI::BlockVector;
#endif

    BlockVectorType                                 solution;
    BlockVectorType                                 locally_relevant_solution;
    BlockVectorType                                 system_rhs;
    std::vector<std::vector<BoundingBox<spacedim>>> global_bounding_boxes;
    unsigned int                                    cycle = 0;
  };


} // namespace ImmersX

#endif