size_t integrate_adaptive( Stepper stepper , System system , State &start_state , Time start_time , Time end_time , Time dt , Observer observer , dense_output_stepper_tag ) { typename omplext_odeint::unwrap_reference< Observer >::type &obs = observer; size_t count = 0; stepper.initialize( start_state , start_time , dt ); while( less_with_sign( stepper.current_time() , end_time , stepper.current_time_step() ) ) { while( less_eq_with_sign( stepper.current_time() + stepper.current_time_step() , end_time , stepper.current_time_step() ) ) { //make sure we don't go beyond the end_time obs( stepper.current_state() , stepper.current_time() ); stepper.do_step( system ); ++count; } stepper.initialize( stepper.current_state() , stepper.current_time() , end_time - stepper.current_time() ); } obs( stepper.current_state() , stepper.current_time() ); // overwrite start_state with the final point boost::numeric::omplext_odeint::copy( stepper.current_state() , start_state ); return count; }
Time integrate_n_steps( Stepper stepper , System system , State &start_state , Time start_time , Time dt , size_t num_of_steps , Observer observer , dense_output_stepper_tag ) { typename odeint::unwrap_reference< Observer >::type &obs = observer; Time time = start_time; const Time end_time = start_time + static_cast< typename unit_value_type<Time>::type >(num_of_steps) * dt; stepper.initialize( start_state , time , dt ); size_t step = 0; while( step < num_of_steps ) { while( less_with_sign( time , stepper.current_time() , stepper.current_time_step() ) ) { stepper.calc_state( time , start_state ); obs( start_state , time ); ++step; // direct computation of the time avoids error propagation happening when using time += dt // we need clumsy type analysis to get boost units working here time = start_time + static_cast< typename unit_value_type<Time>::type >(step) * dt; } // we have not reached the end, do another real step if( less_with_sign( stepper.current_time()+stepper.current_time_step() , end_time , stepper.current_time_step() ) ) { stepper.do_step( system ); } else if( less_with_sign( stepper.current_time() , end_time , stepper.current_time_step() ) ) { // do the last step ending exactly on the end point stepper.initialize( stepper.current_state() , stepper.current_time() , end_time - stepper.current_time() ); stepper.do_step( system ); } } while( stepper.current_time() < end_time ) { if( less_with_sign( end_time , stepper.current_time()+stepper.current_time_step() , stepper.current_time_step() ) ) stepper.initialize( stepper.current_state() , stepper.current_time() , end_time - stepper.current_time() ); stepper.do_step( system ); } // observation at end point, only if we ended exactly on the end-point (or above due to finite precision) obs( stepper.current_state() , end_time ); return time; }
size_t integrate_adaptive( Stepper stepper , System system , State &start_state , Time start_time , Time end_time , Time dt , Observer observer , dense_output_stepper_tag ) { typename boost::unwrap_reference< Observer >::type &obs = observer; size_t count = 0; stepper.initialize( start_state , start_time , dt ); while( stepper.current_time() < end_time ) { while( stepper.current_time() + stepper.current_time_step() <= end_time ) { //make sure we don't go beyond the end_time obs( stepper.current_state() , stepper.current_time() ); stepper.do_step( system ); ++count; } stepper.initialize( stepper.current_state() , stepper.current_time() , end_time - stepper.current_time() ); } obs( stepper.current_state() , stepper.current_time() ); return count; }