explicit_low_storage_runge_kutta_integrator.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <deal.II/lac/la_parallel_vector.h>
 
#include <meltpooldg/time_integration/solution_history.hpp>
#include <meltpooldg/time_integration/time_integrator_base.hpp>
#include <meltpooldg/time_integration/time_integrator_data.hpp>
 
#include <functional>
#include <vector>
 
namespace MeltPoolDG::TimeIntegration
{
  inline static constexpr std::array<TimeIntegratorSchemes, 5> explicit_lsrk_supported_schemes{{
    TimeIntegratorSchemes::LSRK_stage_1_order_1,
    TimeIntegratorSchemes::LSRK_stage_3_order_3,
    TimeIntegratorSchemes::LSRK_stage_5_order_4,
    TimeIntegratorSchemes::LSRK_stage_7_order_4,
    TimeIntegratorSchemes::LSRK_stage_9_order_5,
  }};
 
  template <typename number>
  class LowStorageExplicitRungeKuttaIntegrator final : public TimeIntegratorBase<number>
  {
  public:
    using VectorType = dealii::LinearAlgebra::distributed::Vector<number>;
 
    using RhsFunctionType = std::function<void(number,
                                               number,
                                               VectorType &,
                                               const VectorType &,
                                               std::function<void(unsigned, unsigned)>)>;
 
 
 
    explicit LowStorageExplicitRungeKuttaIntegrator(
      const TimeIntegratorData<number> &time_integrator_data,
      const RhsFunctionType            &compute_rhs);
 
    unsigned int
    required_solution_history_size() const override;
 
    void
    reinit(const VectorType &vector_template) override;
 
    void
    reinit(const SolutionHistory<VectorType> &solution_history) override;
 
    void
    perform_time_step(const number                 current_time,
                      const number                 time_step,
                      SolutionHistory<VectorType> &solution_history,
                      const std::function<void(number, number, VectorType &, const VectorType &)>
                        &stage_pre_processing,
                      const std::function<void(number, number, VectorType &, const VectorType &)>
                        &stage_post_processing) override;
 
  private:
    std::vector<number> bi;
 
    std::vector<number> ai;
 
    std::vector<number> ci;
 
    unsigned int n_stages;
 
    dealii::LinearAlgebra::distributed::Vector<number> rk_register_ri;
 
    dealii::LinearAlgebra::distributed::Vector<number> rk_register_ki;
 
    RhsFunctionType compute_rhs;
  };
} // namespace MeltPoolDG::TimeIntegration