"""
===========
Plot wedges
===========

This example shows how you can plot wedge and comet lines on normal axes.
"""
import matplotlib.pyplot as plt
import numpy as np

##############################################################################
# Generate random data
# --------------------
# First let's generate some random data
np.random.seed(42)
x = np.random.uniform(0, 120, 10)
y = np.random.uniform(0, 80, 10)
x_end = np.random.uniform(0, 120, 10)
y_end = np.random.uniform(0, 80, 10)

##############################################################################
# Plot using wedges
# -----------------
# This shows how to plot the lines using wedges. The code was donated by 
# `Jon Ollington <https://twitter.com/jonollington>`_
fig, ax = plt.subplots()
ax.set_xlim(0, 120)
ax.set_ylim(0, 80)
for i in range(len(x)):
    ax.annotate('', xy=(x[i], y[i]), xytext=(x_end[i], y_end[i]), zorder=2,
                arrowprops=dict(arrowstyle="wedge,tail_width=1", linewidth=1,
                                fc='red', ec='black', alpha=0.4),
                )

##############################################################################
# Plot using lines
# ----------------
# You can also use mplsoccer's lines function on normal axes. 
from mplsoccer import lines

fig, ax = plt.subplots()
ax.set_xlim(0, 120)
ax.set_ylim(0, 80)
lc1 = lines(x, y, x_end, y_end, color='red', comet=True, transparent=True,
            alpha_start=0.1, alpha_end=0.4, ax=ax)

##############################################################################
# Plot using cmap
# ---------------
# You can also use plot using a color map using mplsoccer's lines function.

fig, ax = plt.subplots()
ax.set_xlim(0, 120)
ax.set_ylim(0, 80)
lc2 = lines(x, y, x_end, y_end, cmap='viridis', comet=True, transparent=True,
            alpha_start=0.1, alpha_end=0.4, ax=ax)

plt.show()  # If you are using a Jupyter notebook you do not need this line