Function sample_trajectory

tests/commit/test_throw.py:18–24

(pos, height, vel, angle, gravity=1.)

def sample_trajectory(pos, height, vel, angle, gravity=1.):
    hit, hit_time, height, vel_x, vel_y = simulate_hit(pos, height, vel, angle, gravity)
    def y(x):
        t = (x.vector[0] - pos) / vel_x
        y_ = height + vel_y * t - gravity / 2 * t ** 2
        return math.where((y_ > 0) & (t > 0), y_, math.NAN)
    return CenteredGrid(y, x=2000, bounds=Box(x=(min(pos.min, hit.min), max(pos.max, hit.max))))


BACKENDS = phi.detect_backends()