Остановить интеграцию в odeint с жесткой одой

Question

Я хочу решить жесткую оду с odeint. Я следовал 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; 
    } 
}