Animations in EinsteinPy

Importing required modules

In [1]:
import numpy as np
In [2]:
from einsteinpy.geodesic import Timelike

from einsteinpy.plotting import StaticGeodesicPlotter

Defining various parameters

  • Initial position & momentum of the test partcle

  • Spin of the Kerr Black Hole in metric_params

  • Other solver parameters

Note that, we are working in M-Units ($G = c = M = 1$).

In [3]:
# Constant Radius Orbit

position = [4, np.pi / 3, 0.]

momentum = [0., 0.767851, 2.]

a = 0.99

steps = 400.

delta = 0.5

Calculating geodesic

In [4]:
geod = Timelike(

    metric="Kerr",

    metric_params=(a,),

    position=position,

    momentum=momentum,

    steps=steps,

    delta=delta,

    return_cartesian=True

)

Animating

In [5]:
%matplotlib nbagg

sgpl = StaticGeodesicPlotter()

sgpl.animate(geod, interval=1)

sgpl.show()

Saving animation as .gif

In [7]:
sgpl.ani.save('animation.gif', fps=60)
In [ ]: