FunctionHandles.h

Go to the documentation of this file.

 
#pragma once
 
#include "ODESolverTypes.h"
 
namespace MathLib
{
namespace ODE
{
namespace detail
{
 
class FunctionHandles
{
public:
    virtual bool call(const double t, double const* const y,
                      double* const ydot) = 0;
 
    virtual bool callJacobian(const double t,
                              double const* const y,
                              double* const ydot,
                              double* const jac) = 0;
 
    virtual bool hasJacobian() const = 0;
 
    virtual unsigned getNumberOfEquations() const = 0;
 
    virtual ~FunctionHandles() = default;
};
 
template <unsigned N>
struct FunctionHandlesImpl final : public FunctionHandles
{
    FunctionHandlesImpl(Function<N>& f, JacobianFunction<N>& df) : f(f), df(df)
    {
    }
 
    bool call(const double t, const double* const y,
              double* const ydot) override
    {
        if (f)
        {
            MappedVector<N> ydot_mapped{ydot};
            return f(t, MappedConstVector<N>{y}, ydot_mapped);
        }
        return false;
    }
 
    bool callJacobian(const double t, const double* const y, double* const ydot,
                      double* const jac) override
    {
        if (df)
        {
            MappedMatrix<N, N> jac_mapped{jac};
            return df(t,
                      MappedConstVector<N>{y},
                      MappedConstVector<N>{ydot},
                      jac_mapped);
        }
        return false;
    }
 
    bool hasJacobian() const override { return df != nullptr; }
    unsigned getNumberOfEquations() const override { return N; }
    Function<N> f;
    JacobianFunction<N> df;
};
 
 
}  // namespace detail
}  // namespace ODE
}  // namespace MathLib