ODEStateVectorCollector.hpp

 
 
#pragma once
 
#include "AstGlobal.h"
#include "AstMath/ODEStateObserver.hpp"
#include <vector>
 
AST_NAMESPACE_BEGIN
 
class AST_MATH_API ODEStateVectorCollector: public ODEStateObserver
{
public:
    ODEStateVectorCollector(int ndim);
    ~ODEStateVectorCollector() = default;
 
    EODEAction onStateUpdate(double* y, double& x, ODEIntegrator* integrator) override;
 
    std::vector<double>& x(){ return x_; };
    std::vector<std::vector<double>>& y(){ return y_; };
protected:
    int ndim_;
    std::vector<double> x_;
    std::vector<std::vector<double>> y_;
};
 
AST_NAMESPACE_END