from RK4Stepper import RK4Stepper
from ODESolver import ODESolver
import numpy as np
def getStartingState(zcoord):
return SystemState(
EarthPosition=[
-3.1617821569319294E10, 1.436603570949401E11, 117565.96043321176
],
EarthVelocity=[
-29581.545053209236, -6515.282025813395, -0.010617114855559093
],
AsteroidPosition=[-1.59458654e+11, -1.94077604e+11, 6.61733278e+09],
AsteroidVelocity=[13224.513110762178, -14246.151046842608, zcoord])
space = Space()
stepper = RK4Stepper()
solver = ODESolver(stepper, space)
def printer(time, state, iteration):
print(state)
solver.registerOutput(printer)
hit = solver.integrate(0, 365 * 24 * 60 * 60, 5 * 60, 10E-6,
getStartingState(1000))
