Animations in EinsteinPy

Importing required modules

import numpy as np

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

• Other solver parameters

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

# Constant Radius Orbit
position = [4, np.pi / 3, 0.]
momentum = [0., 0.767851, 2.]
a = 0.99
end_lambda = 200.
step_size = 0.5

Calculating geodesic, using the Julia back-end

geod = Timelike(
    position=position,
    momentum=momentum,
    a=a,
    end_lambda=end_lambda,
    step_size=step_size,
    return_cartesian=True,
    julia=True
)

Animating

%matplotlib nbagg
sgpl = StaticGeodesicPlotter()
sgpl.animate(geod, interval=1)
sgpl.show()

Saving animation as .gif

sgpl.ani.save('animation.gif', writer='imagemagick', fps=60)