OdeSolver Class Reference

#include <ode_solver.h>

Public Member Functions
	OdeSolver (std::string method, double t_start, double rtol, double atol)
	~OdeSolver ()
void	reset (double t_start, double rtol, double atol)
template<class Functor , class AfterStep >
void	step_to (double t_stop, double &t, std::vector< double > &y, Functor &derivs, AfterStep &after_step)
int	get_fn_evals ()

Public Attributes
struct {
double   abs_tol = 1e-8
double   rel_tol = 1e-8
}	control

Private Attributes
SolverType	type
void *	solver = nullptr
int	nfe_cumm = 0

Detailed Description

Definition at line 14 of file ode_solver.h.

Constructor & Destructor Documentation

OdeSolver()

OdeSolver::OdeSolver ( std::string  method, double  t_start, double  rtol, double  atol  )

inline

Definition at line 27 of file ode_solver.h.

                                                                         {
        if (method == "rk45ck") type = ODE_RKCK45;
        else if (method == "lsoda") type = ODE_LSODA;
        else {
            throw std::runtime_error("Fatal: Unknown ODE method " + method); //exit(1);
        }
        reset(t_start, rtol, atol);
    }

Here is the call graph for this function:

~OdeSolver()

OdeSolver::~OdeSolver ( )

inline

Definition at line 36 of file ode_solver.h.

                {
        if      (type == ODE_RKCK45) delete static_cast<RKCK45*>(solver);   
        else if (type == ODE_LSODA)  delete static_cast<LSODA*>(solver);    
    }

Member Function Documentation

get_fn_evals()

int OdeSolver::get_fn_evals ( )

inline

Definition at line 92 of file ode_solver.h.

                      {
        if      (type == ODE_RKCK45) return static_cast<RKCK45*>(solver)->get_fn_evals();   
        else if (type == ODE_LSODA)  return nfe_cumm;
        else return -1;
    }

reset()

void OdeSolver::reset ( double  t_start, double  rtol, double  atol  )

inline

Definition at line 42 of file ode_solver.h.

                                                        {
        nfe_cumm = 0;
        control.abs_tol = atol;
        control.rel_tol = rtol;
        if (type == ODE_RKCK45){
            RKCK45* sol = static_cast<RKCK45*>(solver);
            delete sol; sol = nullptr;
            solver = new RKCK45(t_start, rtol, 1e-8);
        }
        else if (type == ODE_LSODA){
            LSODA* sol = static_cast<LSODA*>(solver);
            delete sol; sol = nullptr;
            solver = new LSODA();
        }
    }

Here is the caller graph for this function:

step_to()

template<class Functor , class AfterStep >

void OdeSolver::step_to ( double  t_stop, double &  t, std::vector< double > &  y, Functor &  derivs, AfterStep &  after_step  )

inline

Definition at line 60 of file ode_solver.h.

                                                                                                       {
        if (t_stop == t || y.size() == 0) return;
        
        if (type == ODE_RKCK45){
            RKCK45 * sol = static_cast<RKCK45*>(solver);
            sol->Step_to(t_stop, t, y, derivs, after_step);
        }
        else if (type == ODE_LSODA ){
            LSODA* sol = static_cast<LSODA*>(solver);
            sol->set_istate(1); // forces re-initialization
            sol->lsoda_update(derivs, after_step, y.size(), y, &t, t_stop, nullptr, control.rel_tol, control.abs_tol);
            if(sol->get_istate() <= 0) {
                std::cerr << "LSODA Error: istate = " << sol->get_istate() << std::endl;
                return;
            }
            nfe_cumm += sol->get_fncalls();
        }
        
        //else if (type == ODE_LSODA ){
            //LSODA * sol = new LSODA();
            //sol->lsoda_update(derivs, y.size(), y, &t, t_stop, nullptr, control.rel_tol, control.abs_tol);
            //if(sol->get_istate() <= 0) {
                //std::cerr << "LSODA Error: istate = " << sol->get_istate() << std::endl;
                //return;
            //}
            //nfe_cumm += sol->get_fncalls();
            //delete sol;
        //}
    }

Here is the call graph for this function:

Here is the caller graph for this function:

Member Data Documentation

abs_tol

double OdeSolver::abs_tol = 1e-8

Definition at line 22 of file ode_solver.h.

control

struct { ... } OdeSolver::control

nfe_cumm

int OdeSolver::nfe_cumm = 0

private

Definition at line 18 of file ode_solver.h.

rel_tol

double OdeSolver::rel_tol = 1e-8

Definition at line 23 of file ode_solver.h.

solver

void* OdeSolver::solver = nullptr

private

Definition at line 17 of file ode_solver.h.

type

SolverType OdeSolver::type

private

Definition at line 16 of file ode_solver.h.

---

The documentation for this class was generated from the following file:

• include/ode_solver.h