wmexplicitrungekuttascheme.h

Go to the documentation of this file.

#ifndef WMEXPLICITRUNGEKUTTASCHEME_H
#define WMEXPLICITRUNGEKUTTASCHEME_H
 
#include "warpxm/warpxm_config.h"
 
#include "array/array.h"
 
#include "hostactions/wmtemporalsolver.h"
 
// STL includes
#include <vector>
 
class WmUnstructuredPatch;
 
namespace wxm
{
namespace variable
{
class distributed_variable_t;
}
} // namespace wxm
 
 
class WmExplicitRungeKuttaScheme
{
public:
    // pair of variable/component
    typedef std::vector<std::pair<wxm::variable::distributed_variable_t*, size_t>>
        variables_type;
 
    WmExplicitRungeKuttaScheme() = default;
 
    virtual ~WmExplicitRungeKuttaScheme() = default;
 
    int getNumStages() const
    {
        return _numStages;
    }
    real getTimeUpdate(const int rkStage) const
    {
        return timeUpdates_[rkStage];
    }
    real getCFLLimit() const
    {
        return _cfl_max;
    }
 
    virtual void
    calc_stage(int stage,
               const std::vector<wxm::temporal_solver::temporal_var>& temporal_vars,
               std::vector<variables_type>& vars,
               real dt) const;
 
    void print_stage_definition(int stage) const;
 
protected:
    int stage_offset(int stage) const
    {
        return ((stage + 1) * (stage + 2) >> 1) - 1;
    }
 
    real _cfl_max = 1;
    int _numStages;
    std::vector<real> timeUpdates_;
    std::vector<real> multipliers_;
};
 
#endif // WMEXPLICITRUNGEKUTTASCHEME_H