import numpy as np
import matplotlib.pyplot as plt
import math
N = 10
# Trayectoria X
x = np.zeros(N+1)
Om_p0p0_x = np.zeros(N+1)
Om_m1p0_x = np.zeros(N+1)
Om_p1p0_x = np.zeros(N+1)
Om_p0m1_x = np.zeros(N+1)
Om_p0p1_x = np.zeros(N+1)
Om_p1p1_x = np.zeros(N+1)
Om_m1m1_x = np.zeros(N+1)
for iz in range(0,N+1):
x_x = 0.5*(iz/N)
x_y = 0.0*(iz/N)
x[iz] = math.sqrt(x_x*x_x+x_y+x_y)
Om_p0p0_x[iz] = math.sqrt((x_x+0)**2+(x_y+0)**2)
Om_m1p0_x[iz] = math.sqrt((x_x-1)**2+(x_y+0)**2)
Om_p1p0_x[iz] = math.sqrt((x_x+1)**2+(x_y+0)**2)
Om_p0m1_x[iz] = math.sqrt((x_x+0)**2+(x_y-1)**2)
Om_p0p1_x[iz] = math.sqrt((x_x+0)**2+(x_y+1)**2)
Om_p1p1_x[iz] = math.sqrt((x_x+1)**2+(x_y+1)**2)
Om_m1m1_x[iz] = math.sqrt((x_x-1)**2+(x_y-1)**2)
# Trayectoria M
m = np.zeros(N+1)
Om_p0p0_m = np.zeros(N+1)
Om_m1p0_m = np.zeros(N+1)
Om_p1p0_m = np.zeros(N+1)
Om_p0m1_m = np.zeros(N+1)
Om_p0p1_m = np.zeros(N+1)
Om_p1p1_m = np.zeros(N+1)
Om_m1m1_m = np.zeros(N+1)
for iz in range(0,N+1):
m_x = 0.5*(iz/N)
m_y = 0.5*(iz/N)
m[iz] = -math.sqrt(m_x*m_x+m_y*m_y)
Om_p0p0_m[iz] = math.sqrt((m_x+0)**2+(m_y+0)**2)
Om_m1p0_m[iz] = math.sqrt((m_x-1)**2+(m_y+0)**2)
Om_p1p0_m[iz] = math.sqrt((m_x+1)**2+(m_y+0)**2)
Om_p0m1_m[iz] = math.sqrt((m_x+0)**2+(m_y-1)**2)
Om_p0p1_m[iz] = math.sqrt((m_x+0)**2+(m_y+1)**2)
Om_p1p1_m[iz] = math.sqrt((m_x+1)**2+(m_y+1)**2)
Om_m1m1_m[iz] = math.sqrt((m_x-1)**2+(m_y-1)**2)
plt.plot(x,Om_p0p0_x,'-b',label='(0,0)')
plt.plot(x,Om_m1p0_x,'og',label='(-1,0)')
plt.plot(x,Om_p1p0_x,'-r',label='(+1,0)')
plt.plot(x,Om_p0m1_x,'-c',label='(0,-1)')
plt.plot(x,Om_p0p1_x,'oc',label='(0,+1)')
plt.plot(x,Om_p1p1_x,'-m',label='(+1,+1)')
plt.plot(x,Om_m1m1_x,'-y',label='(-1,-1)')
plt.plot(m,Om_p0p0_m,'-b',label='(0,0)')
plt.plot(m,Om_m1p0_m,'og',label='(-1,0)')
plt.plot(m,Om_p1p0_m,'-r',label='(+1,0)')
plt.plot(m,Om_p0m1_m,'-c',label='(0,-1)')
plt.plot(m,Om_p0p1_m,'oc',label='(0,+1)')
plt.plot(m,Om_p1p1_m,'-m',label='(+1,+1)')
plt.plot(m,Om_m1m1_m,'-y',label='(-1,-1)')
plt.legend()
plt.show()
