def gravity_function(t): g0 = numpy.array([c(t), s(t), 9.0]) g1 = numpy.array([-s(t), c(t), 0.0]) g2 = numpy.array([-c(t), -s(t), 0.0]) return numpy.concatenate([g0, g1, g2]) gravity = gravity_function(0) t0 = 0 y0 = x_t start = timeit.default_timer() Controller = Vector_Thrust_Controller(gravity_function) out = Controller.output(x_t, gravity) stop = timeit.default_timer() print "time = " + str(stop - start) numpy.set_printoptions(precision=15) for i in range(6): print(out[i]) delta_t = 0.001 print '\n' start = timeit.default_timer() x_t_dt = Controller.predict_x(x_t, t0, delta_t) stop = timeit.default_timer() print "time = " + str(stop - start)
def __init__(self, gravity_function_handle): self.VT_Ctrll = Vector_Thrust_Controller(gravity_function_handle)