#!/usr/bin/python
import numpy as np
import matplotlib.pyplot as plt
import scipy.special as sp
import matplotlib.font_manager

col = ['orange', 'purple', 'deepskyblue', 'mediumseagreen']

X = np.arange(-5, 5, 0.01)

##                                                                              
## PDF                                                                          
##                                                                              

plt.rc('text', usetex = True)
plt.rc('font', family = 'serif', size = 12)

plt.clf()
plt.figure(figsize=(4,3.2))
plt.axes([0.14,0.12,0.83,0.82])
# plt.hold(True)

X = np.arange(-5, 5, 0.01)

A = []
for k,(x,g) in enumerate([(0,0.5),(0,1),(0,2),(-2,1)]):
    Y = ((X-x)/g)**2 + 1
    Y = 1/(np.pi*g*Y)
    a = plt.plot(X, Y, '-', color=col[k], lw=1.7)
    A.append(a)

plt.xlabel("$x$")
plt.ylabel("$P(x)$")
plt.ylim([0,0.7])
plt.tick_params(direction='in', top=True, right=True)

prop = matplotlib.font_manager.FontProperties(size=12)

# bx = 
plt.legend(("$x_0=0,\\, \\gamma=0.5$", "$x_0=0,\\,\\gamma=1$",\
                    "$x_0=0,\\, \\gamma=2$", "$x_0=-2,\\, \\gamma=1$"),
                numpoints=1, handlelength=0.75, handletextpad=0.5,\
                loc="upper right", frameon=False)
# bx.draw_frame(False)
plt.xlim(-5,5)

plt.savefig("cauchy_pdf.pdf")
plt.savefig("cauchy_pdf.svg")