停止在 odeint 中与僵硬 ode 的集成

Question

我想用 odeint 解决一个僵硬的 ode。我在关注this（rosenbrock4_dense_output stepper），但是我的函数增长很快，所以如果 x(t)>xMAX，我想停止积分。

在这个question 中，他们有一个解决方案，但由于我是 C++ 新手，我不知道在使用 rosenbrock4_dense_output 步进器时如何实现它。

我想看看如何专门为 rosenbrock4_dense_output 步进器编写此代码。

Answer 1

目前，使用 odeint 不容易做到这一点。如果您可以使用范围库here，您可以结合for_each 和find_if 算法。

否则需要自己写循环，也不是那么难，应该类似这样：

auto stepper = make_dense_output< rosenbrock4< double > >( 1.0e-12 , 1.0e-12 );
auto ode = make_pair( stiff_system() , stiff_system_jacobi() );

double t = 0.0;
double const end_time = 50.0;
double const dt = 0.01;
vector_type x( 2 , 1.0 );
const double y_min = 1.0;

stepper.initialize( x , t , dt );
cout << t << " " << x[0] << " " << x[1] << endl; // or some other output
t += dt;
while( t < end_time )
{
    if( t > stepper.current_time() )
    {
        // perform a real step
        stepper.do_step( ode );
    }
    else  
    {
        // perform a dense output step
        stepper.calc_state( t , x );
        cout << t << " " << x[0] << " " << x[1] << endl; // or some other output
        t += dt;
    }
    if( x[1] < y_min ) // or some other condition
    {
        cout << "Bound reached." << endl;
        break;
    }
}