OrdinaryDifferentialEquation.hpp

浏览该文件的文档.

 
 
#pragma once
 
#include "AstGlobal.h"
 
AST_NAMESPACE_BEGIN
 
class AST_MATH_API OrdinaryDifferentialEquation
{
public:
    virtual ~OrdinaryDifferentialEquation() = default;
    
    virtual int getDimension() const = 0;
 
    virtual errc_t evaluate(const double* y, double* dy, double t) = 0;
 
    A_ALWAYS_INLINE 
    errc_t evaluate(const double t,const double* y, double* dy){return evaluate(y, dy, t);}
};
 
using ODE = OrdinaryDifferentialEquation;       
 
 
 
template<typename Func>
class ODEGeneric: public ODE
{
public:
    ODEGeneric(Func func, int dim) 
        : func_(std::move(func))
        , dim_(dim) 
    {}
    
    int getDimension() const override {
        return dim_;
    }
    
    errc_t evaluate(const double* y, double* dy, double t) override {
        return func_(y, dy, t);
    }
    
private:
    Func func_;
    int dim_;
};
 
 
template<typename Func>
ODEGeneric<Func> make_ode(Func func, int dim) {
    return ODEGeneric<Func>(func, dim);
}
 
template<typename Func>
ODEGeneric<Func> make_ode(int dim, Func func) {
    return ODEGeneric<Func>(func, dim);
}
 
AST_NAMESPACE_END