#! /usr/bin/env python

# golf_trajectories.py

import numpy
import pylab, matplotlib

from numpy import pi

import cgolf

def get_U(x,z):
	return 0.

V0 = 70.
rpm0 = 3000.

# Get a series of trajectories and plot them
pylab.figure(figsize=(6,4))
for theta0 in [5,20,35]:
    x,y = cgolf.get_trajectory(V0,theta0,rpm0,get_U=get_U)
    pylab.plot(x,y,label='%.0f deg'%theta0)

pylab.xlabel('Distance (m)')
pylab.ylabel('Altitude (m)')
    
pylab.legend()
pylab.show()