# -*- coding: utf-8 -*-
"""
Created on Mon Feb  4 22:50:02 2019

@author: cande
"""
import numpy as np #importo numpy
import matplotlib.pylab as plt #paquete para graficar
#%% DEFINIMOS PARÁMETROS. Pueden probar distintas modificaciones
L=1 #longitud de la cuerda
K=np.pi/L #defino k
X=np.arange(0,L,0.0001) #Elijo el intervalo donde calculo la función de onda

#%% Defino una función que será la que me dé la función de onda. La pongo en términos de los parámetros y variables
def FunciondeOnda(x,k,wt): 
    return np.sin(k*x)*np.cos(wt)+(1/3)*np.sin(3*k*x)*np.cos(3*wt)+ (1/5)*np.sin(5*k*x)*np.cos(5*wt)
#%%
V=[0,1/5,1/3,1/2,1] #Valores de w1t -a un factor pi- para los que me pide graficar 
for w in V:
    w1t=np.pi*w
    y=FunciondeOnda(X,K,w1t)
    plt.title("Función de onda") 
    plt.plot(X,y,label='w1t={}pi'.format(w))
    plt.xlabel('X*L')
    plt.ylabel("Cuerda")
    #plt.xlim(-.3,.3)
    plt.legend(loc=5)
plt.savefig("Cuerda.png")
#%% Veo la simetría alrededor de pi/2
V=[0,1/3,1/2,2/3,1] #Valores de w1t -a un factor pi- para los que me pide graficar 
for w in V:
    w1t=np.pi*w
    y=FunciondeOnda(X,K,w1t)
    plt.title("Función de onda") 
    plt.plot(X,y,label='w1t={}pi'.format(w))
    plt.xlabel('X*L')
    plt.ylabel("Cuerda")
    #plt.xlim(-.3,.3)
    plt.legend(loc=5)
plt.savefig("Cuerdas2.png")
#%% Veo simetría alrededor de pi
V=[0,1/3,1,5/3,2] #Valores de w1t -a un factor pi- para los que me pide graficar 
for w in V:
    w1t=np.pi*w
    y=FunciondeOnda(X,K,w1t)
    plt.title("Función de onda") 
    plt.plot(X,y,label='w1t={}pi'.format(w))
    plt.xlabel('X*L')
    plt.ylabel("Cuerda")
    #plt.xlim(-.3,.3)
    plt.legend(loc=5)
plt.savefig("Cuerdas3.png")