PrevUpHomeNext

Solar system

Gravitation and energy conservation
Define the system function

The next example in this tutorial is a simulation of the outer solar system, consisting of the sun, Jupiter, Saturn, Uranus, Neptune and Pluto.

solar_system

Each planet and of course the sun will be represented by mass points. The interaction force between each object is the gravitational force which can be written as

F​ij = -γ m​i m​j ( q​i - q​j ) / | q​i - q​j | 3

where γ is the gravitational constant, m​i and m​j are the masses and q​i and q​j are the locations of the two objects. The equations of motion are then

dq​i / dt = p​i

dp​i / dt = 1 / m​i Σ​ji F​ij

where p​i is the momenta of object i. The equations of motion can also be derived from the Hamiltonian

H = Σ​i p​i2 / ( 2 m​i ) + Σ​j V( q​i , q​j )

with the interaction potential V(q​i,q​j). The Hamiltonian equations give the equations of motion

dq​i / dt = dH / dp​i

dp​i / dt = -dH / dq​i

In time independent Hamiltonian system the energy and the phase space volume are conserved and special integration methods have to be applied in order to ensure these conservation laws. The odeint library provides classes for separable Hamiltonian systems, which can be written in the form H = Σ p​i2 / (2m​i) + H​q(q), where H​q(q) only depends on the coordinates. Although this functional form might look a bit arbitrary, it covers nearly all classical mechanical systems with inertia and without dissipation, or where the equations of motion can be written in the form dq​i / dt = p​i / m​i , dp​i / dt = f( q​i ).

[Note] Note

A short physical note: While the two-body-problem is known to be integrable, that means it can be solved with purely analytic techniques, already the three-body-problem is not solvable. This was found in the end of the 19th century by H. Poincare which led to the whole new subject of Chaos Theory.

To implement this system we define a 3D point type which will represent the space as well as the velocity. Therefore, we use the operators from Boost.Operators:

/*the point type */
template< class T , size_t Dim >
class point :
    boost::additive1< point< T , Dim > ,
    boost::additive2< point< T , Dim  > , T ,
    boost::multiplicative2< point< T , Dim > , T
    > > >
    {
    public:

        const static size_t dim = Dim;
        typedef T value_type;
        typedef point< value_type , dim > point_type;

        // ...
        // constructors

        // ...
        // operators

    private:

        T m_val[dim];
    };

    //...
    // more operators

The next step is to define a container type storing the values of q and p and to define system functions. As container type we use boost::array

// we simulate 5 planets and the sun
const size_t n = 6;

typedef point< double , 3 > point_type;
typedef boost::array< point_type , n > container_type;
typedef boost::array< double , n > mass_type;

The container_type is different from the state type of the ODE. The state type of the ode is simply a pair< container_type , container_type > since it needs the information about the coordinates and the momenta.

Next we define the system's equations. As we will use a stepper that accounts for the Hamiltonian (energy-preserving) character of the system, we have to define the rhs different from the usual case where it is just a single function. The stepper will make use of the separable character, which means the system will be defined by two objects representing f(p) = -dH/dq and g(q) = dH/dp:

const double gravitational_constant = 2.95912208286e-4;

struct solar_system_coor
{
    const mass_type &m_masses;

    solar_system_coor( const mass_type &masses ) : m_masses( masses ) { }

    void operator()( const container_type &p , container_type &dqdt ) const
    {
        for( size_t i=0 ; i<n ; ++i )
            dqdt[i] = p[i] / m_masses[i];
    }
};

struct solar_system_momentum
{
    const mass_type &m_masses;

    solar_system_momentum( const mass_type &masses ) : m_masses( masses ) { }

    void operator()( const container_type &q , container_type &dpdt ) const
    {
        const size_t n = q.size();
        for( size_t i=0 ; i<n ; ++i )
        {
            dpdt[i] = 0.0;
            for( size_t j=0 ; j<i ; ++j )
            {
                point_type diff = q[j] - q[i];
                double d = abs( diff );
                diff *= ( gravitational_constant * m_masses[i] * m_masses[j] / d / d / d );
                dpdt[i] += diff;
                dpdt[j] -= diff;

            }
        }
    }
};

In general a three body-system is chaotic, hence we can not expect that arbitrary initial conditions of the system will lead to a solution comparable with the solar system dynamics. That is we have to define proper initial conditions, which are taken from the book of Hairer, Wannier, Lubich [4] .

As mentioned above, we need to use some special integrators in order to conserve phase space volume. There is a well known family of such integrators, the so-called Runge-Kutta-Nystroem solvers, which we apply here in terms of a symplectic_rkn_sb3a_mclachlan stepper:

typedef symplectic_rkn_sb3a_mclachlan< container_type > stepper_type;
const double dt = 100.0;

integrate_const(
        stepper_type() ,
        make_pair( solar_system_coor( masses ) , solar_system_momentum( masses ) ) ,
        make_pair( boost::ref( q ) , boost::ref( p ) ) ,
        0.0 , 200000.0 , dt , streaming_observer( cout ) );

These integration routine was used to produce the above sketch of the solar system. Note, that there are two particularities in this example. First, the state of the symplectic stepper is not container_type but a pair of container_type. Hence, we must pass such a pair to the integrate function. Since, we want to pass them as references we can simply pack them into Boost.Ref. The second point is the observer, which is called with a state type, hence a pair of container_type. The reference wrapper is also passed, but this is not a problem at all:

struct streaming_observer
{
    std::ostream& m_out;

    streaming_observer( std::ostream &out ) : m_out( out ) { }

    template< class State >
    void operator()( const State &x , double t ) const
    {
        container_type &q = x.first;
        m_out << t;
        for( size_t i=0 ; i<q.size() ; ++i ) m_out << "\t" << q[i];
        m_out << "\n";
    }
};

[Tip] Tip

You can use C++11 lambda to create the observers

The full example can be found here: solar_system.cpp


PrevUpHomeNext