RK4.h

#ifndef gridripper_odesolver_shoot_method_RK4_h
#define gridripper_odesolver_shoot_method_RK4_h

#include <gridripper/amr1d/FieldWrapper.h>
#include <gridripper/odesolver/shoot/ODEIntegrator.h>
#include <gridripper/tvalarray.h>

namespace gridripper { namespace odesolver { namespace shoot { namespace method {

class RK4: public ODEIntegrator
{
    tvalarray<GReal_t> tmpf1;
    tvalarray<GReal_t> tmpf2;
    tvalarray<GReal_t> tmpf3;
    tvalarray<GReal_t> tmpf4;

public:
    RK4(ODE* ode);

    void integrate(FieldWrapper& f, GReal_t x, GReal_t dx);

private:
    void integrate(GReal_t x, GReal_t dx, GReal_t* f0,
                   GReal_t* f1, GReal_t* f2, GReal_t* f3, GReal_t* f4);

    void step(GReal_t* f0, GReal_t* fin, GReal_t x, GReal_t* fout, GReal_t dx);

    void step4(GReal_t* f0, GReal_t* f1, GReal_t* f2, GReal_t* f3,
               GReal_t x, GReal_t* f4, GReal_t dx);
};

} } } } // namespace gridripper::odesolver::shoot::method

#endif /* gridripper_odesolver_shoot_method_RK4_h */