123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161 |
- /*
- [auto_generated]
- boost/numeric/odeint/integrate/detail/integrate_n_steps.hpp
- [begin_description]
- integrate steps implementation
- [end_description]
- Copyright 2012-2015 Mario Mulansky
- Copyright 2012 Christoph Koke
- Copyright 2012 Karsten Ahnert
- Distributed under the Boost Software License, Version 1.0.
- (See accompanying file LICENSE_1_0.txt or
- copy at http://www.boost.org/LICENSE_1_0.txt)
- */
- #ifndef BOOST_NUMERIC_ODEINT_INTEGRATE_DETAIL_INTEGRATE_N_STEPS_HPP_INCLUDED
- #define BOOST_NUMERIC_ODEINT_INTEGRATE_DETAIL_INTEGRATE_N_STEPS_HPP_INCLUDED
- #include <boost/numeric/odeint/util/unwrap_reference.hpp>
- #include <boost/numeric/odeint/stepper/stepper_categories.hpp>
- #include <boost/numeric/odeint/integrate/detail/integrate_adaptive.hpp>
- #include <boost/numeric/odeint/util/unit_helper.hpp>
- #include <boost/numeric/odeint/util/detail/less_with_sign.hpp>
- namespace boost {
- namespace numeric {
- namespace odeint {
- namespace detail {
- // forward declaration
- template< class Stepper , class System , class State , class Time , class Observer >
- size_t integrate_adaptive_checked(
- Stepper stepper , System system , State &start_state ,
- Time &start_time , Time end_time , Time &dt ,
- Observer observer, controlled_stepper_tag
- );
- /* basic version */
- template< class Stepper , class System , class State , class Time , class Observer>
- Time integrate_n_steps(
- Stepper stepper , System system , State &start_state ,
- Time start_time , Time dt , size_t num_of_steps ,
- Observer observer , stepper_tag )
- {
- typename odeint::unwrap_reference< Observer >::type &obs = observer;
- typename odeint::unwrap_reference< Stepper >::type &st = stepper;
- Time time = start_time;
- for( size_t step = 0; step < num_of_steps ; ++step )
- {
- obs( start_state , time );
- st.do_step( system , start_state , time , dt );
- // 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+1 ) * dt;
- }
- obs( start_state , time );
- return time;
- }
- /* controlled version */
- template< class Stepper , class System , class State , class Time , class Observer >
- Time integrate_n_steps(
- Stepper stepper , System system , State &start_state ,
- Time start_time , Time dt , size_t num_of_steps ,
- Observer observer , controlled_stepper_tag )
- {
- typename odeint::unwrap_reference< Observer >::type &obs = observer;
- Time time = start_time;
- Time time_step = dt;
- for( size_t step = 0; step < num_of_steps ; ++step )
- {
- obs( start_state , time );
- // integrate_adaptive_checked uses the given checker to throw if an overflow occurs
- detail::integrate_adaptive(stepper, system, start_state, time, static_cast<Time>(time + time_step), dt,
- null_observer(), controlled_stepper_tag());
- // 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+1) * time_step;
- }
- obs( start_state , time );
- return time;
- }
- /* dense output version */
- template< class Stepper , class System , class State , class Time , class Observer >
- 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;
- typename odeint::unwrap_reference< Stepper >::type &st = stepper;
- Time time = start_time;
- const Time end_time = start_time + static_cast< typename unit_value_type<Time>::type >(num_of_steps) * dt;
- st.initialize( start_state , time , dt );
- size_t step = 0;
- while( step < num_of_steps )
- {
- while( less_with_sign( time , st.current_time() , st.current_time_step() ) )
- {
- st.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( static_cast<Time>(st.current_time()+st.current_time_step()) ,
- end_time ,
- st.current_time_step() ) )
- {
- st.do_step( system );
- }
- else if( less_with_sign( st.current_time() , end_time , st.current_time_step() ) )
- { // do the last step ending exactly on the end point
- st.initialize( st.current_state() , st.current_time() , static_cast<Time>(end_time - st.current_time()) );
- st.do_step( system );
- }
- }
- // make sure we really end exactly where we should end
- while( st.current_time() < end_time )
- {
- if( less_with_sign( end_time ,
- static_cast<Time>(st.current_time()+st.current_time_step()) ,
- st.current_time_step() ) )
- st.initialize( st.current_state() , st.current_time() , static_cast<Time>(end_time - st.current_time()) );
- st.do_step( system );
- }
- // observation at final point
- obs( st.current_state() , end_time );
- return time;
- }
- }
- }
- }
- }
- #endif /* BOOST_NUMERIC_ODEINT_INTEGRATE_DETAIL_INTEGRATE_N_STEPS_HPP_INCLUDED */
|