Using Frames

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

import pyorb

from astropy.utils import iers
from astropy.time import Time
iers.conf.auto_download = False

from sorts.propagator import SGP4
from sorts import frames

prop = SGP4()
orb = pyorb.Orbit(
    M0 = pyorb.M_earth,
    direct_update=True,
    auto_update=True,
    degrees = True,
    a=7000e3,
    e=0,
    i=69,
    omega=0,
    Omega=0,
    anom=0,
)


t = np.linspace(0,3600*24.0,num=5000)
mjd0 = Time(53005.0, format='mjd', scale='utc')
times = Time(mjd0 + t/(3600*24.0), format='mjd', scale='utc')

states_teme = prop.propagate(t, orb.cartesian[:,0], epoch=mjd0, A=1.0, C_R = 1.0, C_D = 1.0)

fig = plt.figure(figsize=(15,15))

for ind, frame_name in enumerate(['TEME', 'ITRS', 'ICRS', 'GCRS']):

    states_conv = frames.convert(
        times,
        states_teme,
        in_frame='TEME',
        out_frame=frame_name,
    )

    ax = fig.add_subplot(221 + ind, projection='3d')
    ax.plot(states_conv[0,:], states_conv[1,:], states_conv[2,:],"-b")
    ax.set_title(f'In: TEME, out: {frame_name}')


plt.show()