def mean_motion(orbit, tofs, **kwargs):
k = orbit.attractor.k.to(u.km**3 / u.s**2).value
r0 = orbit.r.to(u.km).value
v0 = orbit.v.to(u.km / u.s).value
if not hasattr(tofs, '__len__'):
return mean_motion_fast(k, r0, v0, tofs)
results = [mean_motion_fast(k, r0, v0, tof) for tof in tofs]
return [result[0] for result in results], [result[1] for result in results]
def mean_motion(orbit, tofs, **kwargs):
k = orbit.attractor.k.to(u.km ** 3 / u.s ** 2).value
r0 = orbit.r.to(u.km).value
v0 = orbit.v.to(u.km / u.s).value
if not hasattr(tofs, '__len__'):
return mean_motion_fast(k, r0, v0, tofs)
results = [mean_motion_fast(k, r0, v0, tof) for tof in tofs]
return [result[0] for result in results], [result[1] for result in results]
def mean_motion(k, r, v, tofs, **kwargs):
"""Propagates orbit using Cowell's formulation.
Parameters
----------
k : ~astropy.units.Quantity
Standard gravitational parameter of the attractor.
r : ~astropy.units.Quantity
Position vector.
v : ~astropy.units.Quantity
Velocity vector.
tofs : ~astropy.units.Quantity
Array of times to propagate.
Returns
-------
rr : ~astropy.units.Quantity
Propagated position vectors.
vv : ~astropy.units.Quantity
Propagated velocity vectors.
"""
k = k.to(u.km**3 / u.s**2).value
r0 = r.to(u.km).value
v0 = v.to(u.km / u.s).value
tofs = tofs.to(u.s).value
results = [mean_motion_fast(k, r0, v0, tof) for tof in tofs]
# TODO: Rewrite to avoid iterating twice
return (
[result[0] for result in results] * u.km,
[result[1] for result in results] * u.km / u.s,
)
def mean_motion(k, r, v, tofs, **kwargs):
"""Propagates orbit using Cowell's formulation.
Parameters
----------
k : ~astropy.units.Quantity
Standard gravitational parameter of the attractor.
r : ~astropy.units.Quantity
Position vector.
v : ~astropy.units.Quantity
Velocity vector.
tofs : ~astropy.units.Quantity
Array of times to propagate.
Returns
-------
rr : ~astropy.units.Quantity
Propagated position vectors.
vv : ~astropy.units.Quantity
Propagated velocity vectors.
"""
k = k.to(u.km ** 3 / u.s ** 2).value
r0 = r.to(u.km).value
v0 = v.to(u.km / u.s).value
tofs = tofs.to(u.s).value
results = [mean_motion_fast(k, r0, v0, tof) for tof in tofs]
# TODO: Rewrite to avoid iterating twice
return (
[result[0] for result in results] * u.km,
[result[1] for result in results] * u.km / u.s,
)