import pycanum.main as pycan
import numpy as np
import matplotlib.pyplot as plt

f=40e3
fmod=1e3
m = 0.3
U = 1.0
tmax = 1/fmod

N = 5000
t = np.linspace(0,tmax,N)
te = t[1]
print(te)
fechant = 1/te
u = np.zeros(N)
i1 = 500
N1 = int(1/fmod/2/t[1])
t1 = t[i1:i1+N1]-t[i1]
u[i1:i1+N1] = U*(np.sin(2*np.pi*fmod*t1))*np.sin(2*np.pi*f*t1)


can = pycan.Sysam("SP5")
can.config_entrees([0,1,2],[2.0,2.0,2.0])
can.config_echantillon(te*10**6,N)
can.acquerir_avec_sorties(u,0)
t = can.temps()[0]
U = can.entrees()
Vin = U[0]
Vout = U[1]
Voutf = U[2]
can.fermer()

plt.figure(figsize=(16,6)) 
plt.plot(t*1e3,Vin,'k',label='Vin')
plt.plot(t*1e3,Vout,'r',label='Vout')
plt.plot(t*1e3,Voutf,'g',label='Voutf')
plt.xlabel('t (ms)')
plt.ylabel('Volts')
plt.legend(loc='upper right')
plt.grid()
plt.savefig("../../../../figures/sciphys/electro/enveloppe/paquet-fig1.png")
plt.savefig("../../../../figures/sciphys/electro/enveloppe/paquet-fig1.pdf")
plt.show()