explicit_solver.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: © 2025 PRISMS Center at the University of Michigan
// SPDX-License-Identifier: GNU Lesser General Public Version 2.1
 
#pragma once
 
#include <prismspf/core/timer.h>
#include <prismspf/core/types.h>
 
#include <prismspf/solvers/mf_operator.h>
#include <prismspf/solvers/solver_base.h>
 
#include <prismspf/config.h>
 
PRISMS_PF_BEGIN_NAMESPACE
 
template <unsigned int dim, unsigned int degree, typename number>
class SolveContext;

template <unsigned int dim, unsigned int degree, typename number>
class ExplicitSolver : public SolverBase<dim, degree, number>
{
  using SolverBase<dim, degree, number>::solutions;
  using SolverBase<dim, degree, number>::solve_context;
  using SolverBase<dim, degree, number>::solve_block;
 
public:
  ExplicitSolver(SolveBlock                               _solve_block,
                 const SolveContext<dim, degree, number> &_solve_context)
    : SolverBase<dim, degree, number>(_solve_block, _solve_context)
  {}
 
  void
  init(const std::list<SolveBlock> &all_solve_blocks) override
  {
    SolverBase<dim, degree, number>::init(all_solve_blocks);
    // Initialize rhs_operator
    rhs_operator.init(solve_context->get_pde_operator(),
                      &PDEOperatorBase<dim, degree, number>::compute_rhs,
                      solve_context->get_field_attributes(),
                      solve_context->get_solution_indexer(),
                      solve_context->get_matrix_free_manager(),
                      solve_context->get_simulation_timer(),
                      solve_block,
                      solve_block.dependencies_rhs);
    rhs_operator.set_scaling_diagonal(
      true,
      solve_context->get_invm_manager().get_invm(solve_context->get_field_attributes(),
                                                 solve_block.field_indices));
  }

  void
  solve_impl() override
  {
    // Zero out the ghosts
    Timer::start_section("Zero ghosts");
    solutions.zero_out_ghosts();
    Timer::end_section("Zero ghosts");
 
    rhs_operator.compute_operator(solutions.get_solution_full_vector());
 
    // Apply constraints
    solutions.apply_constraints();
 
    // Update the ghosts
    Timer::start_section("Update ghosts");
    solutions.update_ghosts();
    Timer::end_section("Update ghosts");
  }
 
private:
  MFOperator<dim, degree, number> rhs_operator;
};
 
PRISMS_PF_END_NAMESPACE