123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161 |
- /*
- [auto_generated]
- boost/numeric/odeint/integrate/detail/integrate_adaptive.hpp
- [begin_description]
- Default Integrate adaptive implementation.
- [end_description]
- Copyright 2011-2013 Karsten Ahnert
- Copyright 2011-2015 Mario Mulansky
- Copyright 2012 Christoph Koke
- 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_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 {
- // forward declaration
- 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 );
- /*
- * integrate_adaptive for simple stepper is basically an integrate_const + some last step
- */
- 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 ) )
- { //make a last step to end exactly at end_time
- st.do_step( system , start_state , end , end_time - end );
- steps++;
- obs( start_state , end_time );
- }
- return steps;
- }
- /*
- * integrate adaptive for controlled stepper
- */
- 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; // to throw a runtime_error if step size adjustment fails
- 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(); // check number of failed steps
- }
- while( res == fail );
- fail_checker.reset(); // if we reach here, the step was successful -> reset fail checker
- ++count;
- }
- obs( start_state , start_time );
- return count;
- }
- /*
- * integrate adaptive for dense output steppers
- *
- * step size control is used if the stepper supports it
- */
- 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() ) )
- { //make sure we don't go beyond the end_time
- obs( st.current_state() , st.current_time() );
- st.do_step( system );
- ++count;
- }
- // calculate time step to arrive exactly at end time
- st.initialize( st.current_state() , st.current_time() , static_cast<Time>(end_time - st.current_time()) );
- }
- obs( st.current_state() , st.current_time() );
- // overwrite start_state with the final point
- boost::numeric::odeint::copy( st.current_state() , start_state );
- return count;
- }
- } // namespace detail
- } // namespace odeint
- } // namespace numeric
- } // namespace boost
- #endif // BOOST_NUMERIC_ODEINT_INTEGRATE_DETAIL_INTEGRATE_ADAPTIVE_HPP_INCLUDED
|