th0 = np.pi / 2  # primary direction of waves