123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161 |
- #ifndef BOOST_NUMERIC_ODEINT_INTEGRATE_DETAIL_INTEGRATE_ADAPTIVE_HPP_INCLUDED
- #define BOOST_NUMERIC_ODEINT_INTEGRATE_DETAIL_INTEGRATE_ADAPTIVE_HPP_INCLUDED
- #include <stdexcept>
- #include <boost/throw_exception.hpp>
- #include <boost/numeric/odeint/stepper/stepper_categories.hpp>
- #include <boost/numeric/odeint/stepper/controlled_step_result.hpp>
- #include <boost/numeric/odeint/integrate/max_step_checker.hpp>
- #include <boost/numeric/odeint/integrate/detail/integrate_const.hpp>
- #include <boost/numeric/odeint/util/bind.hpp>
- #include <boost/numeric/odeint/util/unwrap_reference.hpp>
- #include <boost/numeric/odeint/util/copy.hpp>
- #include <boost/numeric/odeint/util/detail/less_with_sign.hpp>
- #include <iostream>
- namespace boost {
- namespace numeric {
- namespace odeint {
- namespace detail {
- template< class Stepper , class System , class State , class Time , class Observer >
- size_t integrate_const(
- Stepper stepper , System system , State &start_state ,
- Time start_time , Time end_time , Time dt ,
- Observer observer , stepper_tag );
- template< class Stepper , class System , class State , class Time , class Observer >
- size_t integrate_adaptive(
- Stepper stepper , System system , State &start_state ,
- Time start_time , Time end_time , Time dt ,
- Observer observer , stepper_tag
- )
- {
- size_t steps = detail::integrate_const( stepper , system , start_state , start_time ,
- end_time , dt , observer , stepper_tag() );
- typename odeint::unwrap_reference< Observer >::type &obs = observer;
- typename odeint::unwrap_reference< Stepper >::type &st = stepper;
- Time end = start_time + dt*steps;
- if( less_with_sign( end , end_time , dt ) )
- {
- st.do_step( system , start_state , end , end_time - end );
- steps++;
- obs( start_state , end_time );
- }
- return steps;
- }
- template< class Stepper , class System , class State , class Time , class Observer >
- size_t integrate_adaptive(
- Stepper stepper , System system , State &start_state ,
- Time &start_time , Time end_time , Time &dt ,
- Observer observer , controlled_stepper_tag
- )
- {
- typename odeint::unwrap_reference< Observer >::type &obs = observer;
- typename odeint::unwrap_reference< Stepper >::type &st = stepper;
- failed_step_checker fail_checker;
- size_t count = 0;
- while( less_with_sign( start_time , end_time , dt ) )
- {
- obs( start_state , start_time );
- if( less_with_sign( end_time , static_cast<Time>(start_time + dt) , dt ) )
- {
- dt = end_time - start_time;
- }
- controlled_step_result res;
- do
- {
- res = st.try_step( system , start_state , start_time , dt );
- fail_checker();
- }
- while( res == fail );
- fail_checker.reset();
- ++count;
- }
- obs( start_state , start_time );
- return count;
- }
- template< class Stepper , class System , class State , class Time , class Observer >
- 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 odeint::unwrap_reference< Observer >::type &obs = observer;
- typename odeint::unwrap_reference< Stepper >::type &st = stepper;
- size_t count = 0;
- st.initialize( start_state , start_time , dt );
- while( less_with_sign( st.current_time() , end_time , st.current_time_step() ) )
- {
- while( less_eq_with_sign( static_cast<Time>(st.current_time() + st.current_time_step()) ,
- end_time ,
- st.current_time_step() ) )
- {
- obs( st.current_state() , st.current_time() );
- st.do_step( system );
- ++count;
- }
-
- st.initialize( st.current_state() , st.current_time() , static_cast<Time>(end_time - st.current_time()) );
- }
- obs( st.current_state() , st.current_time() );
-
- boost::numeric::odeint::copy( st.current_state() , start_state );
- return count;
- }
- }
- }
- }
- }
- #endif
|