传递参数以在C ++中提升odeint

时间:2016-11-23 20:21:56

标签: c++ boost parameters odeint integrator

This回答是有帮助的,但我想知道如何将不同类型的多个参数传递给ODE模型,可能在结构中。对于我的直接用例,我需要能够传递一个std::array<double, 6>,两个std::vector<std::vector<double>>和两个double标量,总共传递四个参数。在链接示例中以及harmonic_oscillator.cpp中,只有一个double传递参数。感谢。

这是我需要传递给ODE力模型并在速率方程中使用的结构示例。

struct T
{
    std::array<double, 6> IC;
    double S;
    double M;
    std::vector<std::vector<double>> C;
    std::vector<std::vector<double>> WT;
};

1 个答案:

答案 0 :(得分:1)

我相信我已经提出了一个可行的结构解决方案,但我不确定它是否有任何变量/内存范围no-no。这是一个例子:

#include <vector>
#include <boost/numeric/odeint.hpp>

// define structure
struct T
{
    std::array<double, 6> IC;
    double                S;
};

// force model
class harm_osc
{
    struct T T1;

public:
    harm_osc(struct T G) : T1(G) {}

    void operator() ( const std::vector< double > &x , std::vector< double > &dxdt , const double /* t */ )
    {
        dxdt[0] = x[1];
        dxdt[1] = -x[0] - T1.IC[0]*x[1] + T1.S;
    }
};

// print integrated state solution
void write_solution( const std::vector< double > &x , const double t )
{
    printf("%-6.2f %-6.2f %-6.2f\n", t, x[0], x[1]);
}

// problem setup
int main()
{

    std::vector< double > x(2);
    x[0] = 1.0;
    x[1] = 0.0;

    struct T T2;

    T2.IC = {0.15, 0.15, 0.15, 0.15, 0.15, 0.15};
    T2.S  = 0.0;

    harm_osc ho(T2);
    boost::numeric::odeint::integrate(ho, x, 0.0, 10.0, 0.1, write_solution);

}