3 boost/numeric/odeint/stepper/rosenbrock4_dense_output.hpp
6 Dense output for Rosenbrock 4.
9 Copyright 2011-2012 Karsten Ahnert
10 Copyright 2011-2015 Mario Mulansky
11 Copyright 2012 Christoph Koke
13 Distributed under the Boost Software License, Version 1.0.
14 (See accompanying file LICENSE_1_0.txt or
15 copy at http://www.boost.org/LICENSE_1_0.txt)
19 #ifndef BOOST_NUMERIC_ODEINT_STEPPER_ROSENBROCK4_DENSE_OUTPUT_HPP_INCLUDED
20 #define BOOST_NUMERIC_ODEINT_STEPPER_ROSENBROCK4_DENSE_OUTPUT_HPP_INCLUDED
25 #include <boost/numeric/odeint/util/bind.hpp>
27 #include <boost/numeric/odeint/stepper/rosenbrock4_controller.hpp>
28 #include <boost/numeric/odeint/util/is_resizeable.hpp>
30 #include <boost/numeric/odeint/integrate/max_step_checker.hpp>
37 template< class ControlledStepper >
38 class rosenbrock4_dense_output
43 typedef ControlledStepper controlled_stepper_type;
44 typedef typename unwrap_reference< controlled_stepper_type >::type unwrapped_controlled_stepper_type;
45 typedef typename unwrapped_controlled_stepper_type::stepper_type stepper_type;
46 typedef typename stepper_type::value_type value_type;
47 typedef typename stepper_type::state_type state_type;
48 typedef typename stepper_type::wrapped_state_type wrapped_state_type;
49 typedef typename stepper_type::time_type time_type;
50 typedef typename stepper_type::deriv_type deriv_type;
51 typedef typename stepper_type::wrapped_deriv_type wrapped_deriv_type;
52 typedef typename stepper_type::resizer_type resizer_type;
53 typedef dense_output_stepper_tag stepper_category;
55 typedef rosenbrock4_dense_output< ControlledStepper > dense_output_stepper_type;
57 rosenbrock4_dense_output( const controlled_stepper_type &stepper = controlled_stepper_type() )
58 : m_stepper( stepper ) ,
60 m_current_state_x1( true ) ,
61 m_t() , m_t_old() , m_dt()
67 template< class StateType >
68 void initialize( const StateType &x0 , time_type t0 , time_type dt0 )
70 m_resizer.adjust_size( x0 , detail::bind( &dense_output_stepper_type::template resize_impl< StateType > , detail::ref( *this ) , detail::_1 ) );
71 get_current_state() = x0;
76 template< class System >
77 std::pair< time_type , time_type > do_step( System system )
79 unwrapped_controlled_stepper_type &stepper = m_stepper;
80 failed_step_checker fail_checker; // to throw a runtime_error if step size adjustment fails
81 controlled_step_result res = fail;
85 res = stepper.try_step( system , get_current_state() , m_t , get_old_state() , m_dt );
86 fail_checker(); // check for overflow of failed steps
89 stepper.stepper().prepare_dense_output();
90 this->toggle_current_state();
91 return std::make_pair( m_t_old , m_t );
96 * The two overloads are needed in order to solve the forwarding problem.
98 template< class StateOut >
99 void calc_state( time_type t , StateOut &x )
101 unwrapped_controlled_stepper_type &stepper = m_stepper;
102 stepper.stepper().calc_state( t , x , get_old_state() , m_t_old , get_current_state() , m_t );
105 template< class StateOut >
106 void calc_state( time_type t , const StateOut &x )
108 unwrapped_controlled_stepper_type &stepper = m_stepper;
109 stepper.stepper().calc_state( t , x , get_old_state() , m_t_old , get_current_state() , m_t );
113 template< class StateType >
114 void adjust_size( const StateType &x )
116 unwrapped_controlled_stepper_type &stepper = m_stepper;
117 stepper.adjust_size( x );
124 const state_type& current_state( void ) const
126 return get_current_state();
129 time_type current_time( void ) const
134 const state_type& previous_state( void ) const
136 return get_old_state();
139 time_type previous_time( void ) const
144 time_type current_time_step( void ) const
154 state_type& get_current_state( void )
156 return m_current_state_x1 ? m_x1.m_v : m_x2.m_v ;
159 const state_type& get_current_state( void ) const
161 return m_current_state_x1 ? m_x1.m_v : m_x2.m_v ;
164 state_type& get_old_state( void )
166 return m_current_state_x1 ? m_x2.m_v : m_x1.m_v ;
169 const state_type& get_old_state( void ) const
171 return m_current_state_x1 ? m_x2.m_v : m_x1.m_v ;
174 void toggle_current_state( void )
176 m_current_state_x1 = ! m_current_state_x1;
180 template< class StateIn >
181 bool resize_impl( const StateIn &x )
183 bool resized = false;
184 resized |= adjust_size_by_resizeability( m_x1 , x , typename is_resizeable<state_type>::type() );
185 resized |= adjust_size_by_resizeability( m_x2 , x , typename is_resizeable<state_type>::type() );
190 controlled_stepper_type m_stepper;
191 resizer_type m_resizer;
192 wrapped_state_type m_x1 , m_x2;
193 bool m_current_state_x1;
194 time_type m_t , m_t_old , m_dt;
199 } // namespace odeint
200 } // namespace numeric
204 #endif // BOOST_NUMERIC_ODEINT_STEPPER_ROSENBROCK4_DENSE_OUTPUT_HPP_INCLUDED