mirror of
https://github.com/boostorg/odeint.git
synced 2025-05-09 23:24:01 +00:00
65 lines
1.3 KiB
C++
65 lines
1.3 KiB
C++
/*
|
|
* odeint_rk4_phase_lattice.cpp
|
|
*
|
|
* Copyright 2011 Mario Mulansky
|
|
* Copyright 2012 Karsten Ahnert
|
|
*
|
|
* 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)
|
|
*/
|
|
|
|
|
|
#include <cmath>
|
|
|
|
#include <boost/array.hpp>
|
|
|
|
#include <boost/numeric/odeint/stepper/runge_kutta4_classic.hpp>
|
|
#include <boost/numeric/odeint/stepper/runge_kutta4.hpp>
|
|
#include <boost/numeric/odeint/algebra/array_algebra.hpp>
|
|
|
|
#include "rk_performance_test_case.hpp"
|
|
|
|
#include "phase_lattice.hpp"
|
|
|
|
const size_t N = 1024;
|
|
|
|
typedef boost::array< double , N > state_type;
|
|
typedef boost::numeric::odeint::runge_kutta4_classic< state_type , double , state_type , double , boost::numeric::odeint::array_algebra> rk4_odeint_type;
|
|
|
|
class odeint_wrapper
|
|
{
|
|
public:
|
|
void reset_init_cond()
|
|
{
|
|
for( size_t i = 0 ; i<N ; ++i )
|
|
m_x[i] = 2.0*3.1415927*rand() / RAND_MAX;
|
|
m_t = 0.0;
|
|
}
|
|
|
|
inline void do_step( const double dt )
|
|
{
|
|
m_stepper.do_step( phase_lattice<N>() , m_x , m_t , dt );
|
|
//m_t += dt;
|
|
}
|
|
|
|
double state( const size_t i ) const
|
|
{ return m_x[i]; }
|
|
|
|
private:
|
|
state_type m_x;
|
|
double m_t;
|
|
rk4_odeint_type m_stepper;
|
|
};
|
|
|
|
|
|
|
|
int main()
|
|
{
|
|
srand( 12312354 );
|
|
|
|
odeint_wrapper stepper;
|
|
|
|
run( stepper , 10000 , 1E-6 );
|
|
}
|