RKPlus.h

Go to the documentation of this file.

#ifndef RKPLUS_H
#define RKPLUS_H

#include "timeInt.h"


/*
    This header files prvides the API for a number of explicit RK integrators,
  all of which are taken from Gottlieb, Ketchson & Shu (2009), and are strong
  stability preserving.

    Differently to the previous RK integrators, we combine the source and flux
  approximations into the RHS and treat as an ODE, as opposed to operator
  splitting methods. As a result ALL these integrators can be used as id for
  models with sources.

  The RK2, RKSplit and RKSplit2 integrators are now deprecated. Now choose from
  the following integrators:
  RK2(2,2), RK3(3,3), RK4(4,5), RK4(4,10)
*/









class RKPlus : public TimeIntegrator
{
  public:

    double * fluxCont; 


    RKPlus(Data * data, Model * model, Bcs * bcs, FluxMethod * fluxMethod, ModelExtension * modelExtension = NULL);

    virtual ~RKPlus();


    virtual void rhs(double * cons, double * prims, double * aux, double * rhsVec);

};



class RK2B : public RKPlus
{
  public:


    double
    *u1cons,
    *u1prims,
    *u1aux,         
    *rhs1,
    *rhs2;


    RK2B(Data * data, Model * model, Bcs * bcs, FluxMethod * fluxMethod, ModelExtension * modelExtension = NULL);

    virtual ~RK2B();


    void stage1(double * cons, double * prims, double * aux, double dt);


    void stage2(double * cons, double * prims, double * aux, double dt);


    void step(double * cons, double * prims, double * aux, double dt=0);

};


class RK3 : public RKPlus
{
  public:

    double
    *u1cons,
    *u1prims,
    *u1aux,
    *u2cons,
    *u2prims,         
    *u2aux,
    *rhs1,
    *rhs2,
    *rhs3;


    RK3(Data * data, Model * model, Bcs * bcs, FluxMethod * fluxMethod, ModelExtension * modelExtension = NULL);

    virtual ~RK3();


    void stage1(double * cons, double * prims, double * aux, double dt);


    void stage2(double * cons, double * prims, double * aux, double dt);


    void stage3(double * cons, double * prims, double * aux, double dt);


    void step(double * cons, double * prims, double * aux, double dt=0);

};


class RK4 : public RKPlus
{
  public:


    double
    *u1cons, *u1prims, *u1aux,
    *u2cons, *u2prims, *u2aux,
    *u3cons, *u3prims, *u3aux,          
    *u4cons, *u4prims, *u4aux,
    *rhs1, *rhs2, *rhs3,
    *rhs4, *rhs5;


    RK4(Data * data, Model * model, Bcs * bcs, FluxMethod * fluxMethod, ModelExtension * modelExtension = NULL);

    virtual ~RK4();


    void stage1(double * cons, double * prims, double * aux, double dt);


    void stage2(double * cons, double * prims, double * aux, double dt);


    void stage3(double * cons, double * prims, double * aux, double dt);


    void stage4(double * cons, double * prims, double * aux, double dt);


    void stage5(double * cons, double * prims, double * aux, double dt);


    void step(double * cons, double * prims, double * aux, double dt=0);

};



class RK4_10 : public RKPlus
{
  public:

    // Need some work arrays
    double
    *u1cons, *u1prims, *u1aux,
    *u2cons, *u2prims, *u2aux,         
    *rhs1;


    RK4_10(Data * data, Model * model, Bcs * bcs, FluxMethod * fluxMethod, ModelExtension * modelExtension = NULL);

    virtual ~RK4_10();


    void prepare1(double * cons, double * prims, double * aux);


    void prepare2(double * cons, double * prims, double * aux);


    void stageRepeat(double * cons, double * prims, double * aux, double dt);


    void stageFinal(double * cons, double * prims, double * aux, double dt);


    void step(double * cons, double * prims, double * aux, double dt=0);

};


#endif