180 lines
6.0 KiB
Plaintext
180 lines
6.0 KiB
Plaintext
/*
|
|
[auto_generated]
|
|
boost/numeric/odeint/integrate/detail/integrate_times.hpp
|
|
|
|
[begin_description]
|
|
Default integrate times implementation.
|
|
[end_description]
|
|
|
|
Copyright 2011-2015 Mario Mulansky
|
|
Copyright 2012 Karsten Ahnert
|
|
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_TIMES_HPP_INCLUDED
|
|
#define BOOST_NUMERIC_ODEINT_INTEGRATE_DETAIL_INTEGRATE_TIMES_HPP_INCLUDED
|
|
|
|
#include <stdexcept>
|
|
|
|
#include <boost/config.hpp>
|
|
#include <boost/throw_exception.hpp>
|
|
#include <boost/numeric/odeint/util/unwrap_reference.hpp>
|
|
#include <boost/numeric/odeint/stepper/controlled_step_result.hpp>
|
|
#include <boost/numeric/odeint/util/detail/less_with_sign.hpp>
|
|
#include <boost/numeric/odeint/integrate/max_step_checker.hpp>
|
|
|
|
|
|
namespace boost {
|
|
namespace numeric {
|
|
namespace odeint {
|
|
namespace detail {
|
|
|
|
|
|
|
|
/*
|
|
* integrate_times for simple stepper
|
|
*/
|
|
template<class Stepper, class System, class State, class TimeIterator, class Time, class Observer>
|
|
size_t integrate_times(
|
|
Stepper stepper , System system , State &start_state ,
|
|
TimeIterator start_time , TimeIterator end_time , Time dt ,
|
|
Observer observer , stepper_tag
|
|
)
|
|
{
|
|
typedef typename odeint::unwrap_reference< Stepper >::type stepper_type;
|
|
typedef typename odeint::unwrap_reference< Observer >::type observer_type;
|
|
|
|
stepper_type &st = stepper;
|
|
observer_type &obs = observer;
|
|
typedef typename unit_value_type<Time>::type time_type;
|
|
|
|
size_t steps = 0;
|
|
Time current_dt = dt;
|
|
while( true )
|
|
{
|
|
Time current_time = *start_time++;
|
|
obs( start_state , current_time );
|
|
if( start_time == end_time )
|
|
break;
|
|
while( less_with_sign( current_time , static_cast<time_type>(*start_time) , current_dt ) )
|
|
{
|
|
current_dt = min_abs( dt , *start_time - current_time );
|
|
st.do_step( system , start_state , current_time , current_dt );
|
|
current_time += current_dt;
|
|
steps++;
|
|
}
|
|
}
|
|
return steps;
|
|
}
|
|
|
|
/*
|
|
* integrate_times for controlled stepper
|
|
*/
|
|
template< class Stepper , class System , class State , class TimeIterator , class Time , class Observer >
|
|
size_t integrate_times(
|
|
Stepper stepper , System system , State &start_state ,
|
|
TimeIterator start_time , TimeIterator 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;
|
|
typedef typename unit_value_type<Time>::type time_type;
|
|
|
|
failed_step_checker fail_checker; // to throw a runtime_error if step size adjustment fails
|
|
size_t steps = 0;
|
|
while( true )
|
|
{
|
|
Time current_time = *start_time++;
|
|
obs( start_state , current_time );
|
|
if( start_time == end_time )
|
|
break;
|
|
while( less_with_sign( current_time , static_cast<time_type>(*start_time) , dt ) )
|
|
{
|
|
// adjust stepsize to end up exactly at the observation point
|
|
Time current_dt = min_abs( dt , *start_time - current_time );
|
|
if( st.try_step( system , start_state , current_time , current_dt ) == success )
|
|
{
|
|
++steps;
|
|
// successful step -> reset the fail counter, see #173
|
|
fail_checker.reset();
|
|
// continue with the original step size if dt was reduced due to observation
|
|
dt = max_abs( dt , current_dt );
|
|
}
|
|
else
|
|
{
|
|
fail_checker(); // check for possible overflow of failed steps in step size adjustment
|
|
dt = current_dt;
|
|
}
|
|
}
|
|
}
|
|
return steps;
|
|
}
|
|
|
|
/*
|
|
* integrate_times for dense output stepper
|
|
*/
|
|
template< class Stepper , class System , class State , class TimeIterator , class Time , class Observer >
|
|
size_t integrate_times(
|
|
Stepper stepper , System system , State &start_state ,
|
|
TimeIterator start_time , TimeIterator 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;
|
|
|
|
typedef typename unit_value_type<Time>::type time_type;
|
|
|
|
if( start_time == end_time )
|
|
return 0;
|
|
|
|
TimeIterator last_time_iterator = end_time;
|
|
--last_time_iterator;
|
|
Time last_time_point = static_cast<time_type>(*last_time_iterator);
|
|
|
|
st.initialize( start_state , *start_time , dt );
|
|
obs( start_state , *start_time++ );
|
|
|
|
size_t count = 0;
|
|
while( start_time != end_time )
|
|
{
|
|
while( ( start_time != end_time ) && less_eq_with_sign( static_cast<time_type>(*start_time) , st.current_time() , st.current_time_step() ) )
|
|
{
|
|
st.calc_state( *start_time , start_state );
|
|
obs( start_state , *start_time );
|
|
start_time++;
|
|
}
|
|
|
|
// we have not reached the end, do another real step
|
|
if( less_eq_with_sign( st.current_time() + st.current_time_step() ,
|
|
last_time_point ,
|
|
st.current_time_step() ) )
|
|
{
|
|
st.do_step( system );
|
|
++count;
|
|
}
|
|
else if( start_time != end_time )
|
|
{ // do the last step ending exactly on the end point
|
|
st.initialize( st.current_state() , st.current_time() , last_time_point - st.current_time() );
|
|
st.do_step( system );
|
|
++count;
|
|
}
|
|
}
|
|
return count;
|
|
}
|
|
|
|
|
|
} // namespace detail
|
|
} // namespace odeint
|
|
} // namespace numeric
|
|
} // namespace boost
|
|
|
|
|
|
#endif // BOOST_NUMERIC_ODEINT_INTEGRATE_DETAIL_INTEGRATE_ADAPTIVE_HPP_INCLUDED
|