comparison DEPENDENCIES/generic/include/boost/numeric/odeint/stepper/rosenbrock4_dense_output.hpp @ 16:2665513ce2d3

Add boost headers
author Chris Cannam
date Tue, 05 Aug 2014 11:11:38 +0100
parents
children c530137014c0
comparison
equal deleted inserted replaced
15:663ca0da4350 16:2665513ce2d3
1 /*
2 [auto_generated]
3 boost/numeric/odeint/stepper/rosenbrock4_dense_output.hpp
4
5 [begin_description]
6 Dense output for Rosenbrock 4.
7 [end_description]
8
9 Copyright 2009-2011 Karsten Ahnert
10 Copyright 2009-2011 Mario Mulansky
11
12 Distributed under the Boost Software License, Version 1.0.
13 (See accompanying file LICENSE_1_0.txt or
14 copy at http://www.boost.org/LICENSE_1_0.txt)
15 */
16
17
18 #ifndef BOOST_NUMERIC_ODEINT_STEPPER_ROSENBROCK4_DENSE_OUTPUT_HPP_INCLUDED
19 #define BOOST_NUMERIC_ODEINT_STEPPER_ROSENBROCK4_DENSE_OUTPUT_HPP_INCLUDED
20
21
22 #include <utility>
23
24 #include <boost/numeric/odeint/util/bind.hpp>
25
26 #include <boost/numeric/odeint/stepper/rosenbrock4_controller.hpp>
27 #include <boost/numeric/odeint/util/is_resizeable.hpp>
28
29
30 namespace boost {
31 namespace numeric {
32 namespace odeint {
33
34 template< class ControlledStepper >
35 class rosenbrock4_dense_output
36 {
37
38 public:
39
40 typedef ControlledStepper controlled_stepper_type;
41 typedef typename controlled_stepper_type::stepper_type stepper_type;
42 typedef typename stepper_type::value_type value_type;
43 typedef typename stepper_type::state_type state_type;
44 typedef typename stepper_type::wrapped_state_type wrapped_state_type;
45 typedef typename stepper_type::time_type time_type;
46 typedef typename stepper_type::deriv_type deriv_type;
47 typedef typename stepper_type::wrapped_deriv_type wrapped_deriv_type;
48 typedef typename stepper_type::resizer_type resizer_type;
49 typedef dense_output_stepper_tag stepper_category;
50
51 typedef rosenbrock4_dense_output< ControlledStepper > dense_output_stepper_type;
52
53 rosenbrock4_dense_output( const controlled_stepper_type &stepper = controlled_stepper_type() )
54 : m_stepper( stepper ) ,
55 m_x1() , m_x2() ,
56 m_current_state_x1( true ) ,
57 m_t() , m_t_old() , m_dt()
58 {
59 }
60
61
62
63 template< class StateType >
64 void initialize( const StateType &x0 , time_type t0 , time_type dt0 )
65 {
66 m_resizer.adjust_size( x0 , detail::bind( &dense_output_stepper_type::template resize_impl< StateType > , detail::ref( *this ) , detail::_1 ) );
67 get_current_state() = x0;
68 m_t = t0;
69 m_dt = dt0;
70 }
71
72 template< class System >
73 std::pair< time_type , time_type > do_step( System system )
74 {
75 const size_t max_count = 1000;
76
77 controlled_step_result res = fail;
78 m_t_old = m_t;
79 size_t count = 0;
80 do
81 {
82 res = m_stepper.try_step( system , get_current_state() , m_t , get_old_state() , m_dt );
83 if( count++ == max_count )
84 throw std::overflow_error( "rosenbrock4 : too much iterations!");
85 }
86 while( res == fail );
87 m_stepper.stepper().prepare_dense_output();
88 this->toggle_current_state();
89 return std::make_pair( m_t_old , m_t );
90 }
91
92
93 /*
94 * The two overloads are needed in order to solve the forwarding problem.
95 */
96 template< class StateOut >
97 void calc_state( time_type t , StateOut &x )
98 {
99 m_stepper.stepper().calc_state( t , x , get_old_state() , m_t_old , get_current_state() , m_t );
100 }
101
102 template< class StateOut >
103 void calc_state( time_type t , const StateOut &x )
104 {
105 m_stepper.stepper().calc_state( t , x , get_old_state() , m_t_old , get_current_state() , m_t );
106 }
107
108
109 template< class StateType >
110 void adjust_size( const StateType &x )
111 {
112 m_stepper.adjust_size( x );
113 resize_impl( x );
114 }
115
116
117
118
119 const state_type& current_state( void ) const
120 {
121 return get_current_state();
122 }
123
124 time_type current_time( void ) const
125 {
126 return m_t;
127 }
128
129 const state_type& previous_state( void ) const
130 {
131 return get_old_state();
132 }
133
134 time_type previous_time( void ) const
135 {
136 return m_t_old;
137 }
138
139 time_type current_time_step( void ) const
140 {
141 return m_dt;
142 }
143
144
145
146
147 private:
148
149 state_type& get_current_state( void )
150 {
151 return m_current_state_x1 ? m_x1.m_v : m_x2.m_v ;
152 }
153
154 const state_type& get_current_state( void ) const
155 {
156 return m_current_state_x1 ? m_x1.m_v : m_x2.m_v ;
157 }
158
159 state_type& get_old_state( void )
160 {
161 return m_current_state_x1 ? m_x2.m_v : m_x1.m_v ;
162 }
163
164 const state_type& get_old_state( void ) const
165 {
166 return m_current_state_x1 ? m_x2.m_v : m_x1.m_v ;
167 }
168
169 void toggle_current_state( void )
170 {
171 m_current_state_x1 = ! m_current_state_x1;
172 }
173
174
175 template< class StateIn >
176 bool resize_impl( const StateIn &x )
177 {
178 bool resized = false;
179 resized |= adjust_size_by_resizeability( m_x1 , x , typename is_resizeable<state_type>::type() );
180 resized |= adjust_size_by_resizeability( m_x2 , x , typename is_resizeable<state_type>::type() );
181 return resized;
182 }
183
184
185 controlled_stepper_type m_stepper;
186 resizer_type m_resizer;
187 wrapped_state_type m_x1 , m_x2;
188 bool m_current_state_x1;
189 time_type m_t , m_t_old , m_dt;
190 };
191
192
193
194 } // namespace odeint
195 } // namespace numeric
196 } // namespace boost
197
198
199 #endif // BOOST_NUMERIC_ODEINT_STEPPER_ROSENBROCK4_DENSE_OUTPUT_HPP_INCLUDED