#  *******      DEFINCICIONES     *****************
import pyodide_js
await pyodide_js.loadPackage('pandas')
import pandas as pd
await pyodide_js.loadPackage('matplotlib')
import matplotlib.pyplot as plt
import math as mt
import numpy as np

# await pyodide_js.loadPackage('requests')
# from datetime import datetime
# import time

# import shutil
# from pyscript import fetch
#import datetime as dt
# ************** DIBUJOS    *******************
# from pyscript import display


# ************** datos   *******************

t_values = []
y_values = []
t1_values = []
y1_values = []
t2_values = []
y2_values = []
t3_values = []
y3_values = []
t4_values = []
y4_values = []

ma=1
ri=100

deini=0.2
veini=1.8

# ************** Calculos   *******************

periodo=2*3.14159*mt.sqrt(ma/ri)
omega = mt.sqrt(ri/ma)


#   ************ AMORTIGUADOR 1 ***************
radamo=0.05

for t in np.arange(0, 2,0.01):
    y = (deini*np.cos(omega*t) + (((veini+radamo*omega*deini)/omega)*np.sin(omega*t)))*mt.exp(-1*radamo*omega*t)
    t_values.append(t)
    y_values.append(y)
    
#    print(t,y) 

plt.xlabel('TIEMPO (seg)')
plt.ylabel('DEFORMACION (cm)')
plt.title('Comparison of the Dynamic Response for Various Dampers')

plt.plot(t_values, y_values, label='5%')


#   ************ FIN 1 ***************


#   ************ AMORTIGUADOR 2 ***************
radamo=0.10

for t in np.arange(0, 2,0.01):
    y = (deini*np.cos(omega*t) + (((veini+radamo*omega*deini)/omega)*np.sin(omega*t)))*mt.exp(-1*radamo*omega*t)
    t1_values.append(t)
    y1_values.append(y)
    

plt.plot(t1_values, y1_values, label='10%')

#   ************ FIN 2 ***************

#   ************ AMORTIGUADOR 3 ***************
radamo=0.25

for t in np.arange(0, 2,0.01):
    y = (deini*np.cos(omega*t) + (((veini+radamo*omega*deini)/omega)*np.sin(omega*t)))*mt.exp(-1*radamo*omega*t)
    t2_values.append(t)
    y2_values.append(y)
    

plt.plot(t2_values, y2_values, label='25%')

#   ************ FIN 3 ***************


#   ************ AMORTIGUADOR 4 ***************
radamo=0.50

for t in np.arange(0, 2,0.01):
    y = (deini*np.cos(omega*t) + (((veini+radamo*omega*deini)/omega)*np.sin(omega*t)))*mt.exp(-1*radamo*omega*t)
    t3_values.append(t)
    y3_values.append(y)
    

plt.plot(t3_values, y3_values, label='50%')

#   ************ FIN 4 ***************


#   ************ AMORTIGUADOR 5 ***************
radamo=1

for t in np.arange(0, 2,0.01):
    y = (deini*np.cos(omega*t) + (((veini+radamo*omega*deini)/omega)*np.sin(omega*t)))*mt.exp(-1*radamo*omega*t)
    t4_values.append(t)
    y4_values.append(y)
    

plt.plot(t4_values, y4_values, label='100%')
plt.legend()
plt.axhline(y=0, color='r', linestyle='--', label='Línea horizontal')
plt.show()

#   ************ FIN 5 ***************