# -*- coding:  utf-8 -*-
"""
Created on Mond Mar 20 16:26:11 2017

@author: Sabloniere
"""

import math as m
import numpy as np
import matplotlib.pyplot as plt
import os

# on précise le répertoire dans lequel se trouve le fichier de données (chemin et nom du répertoire à adapter) :
os.chdir(r"C:\Users\Virginie\Desktop\MP\MPI\élec\TFD")
# rq avant le chemin 'r' signifie 'raw' i.e. 'brut' => pas de problème de lecture du chemin 

## on convertit le fichier texte de données en un tableau comportant 2 colonnes : 
# 1e colonne : temps : le temps en s 
# 2e colonne : U : la tension en V
t,U=np.loadtxt("créneau.txt",unpack=True)

N=len(t)
Te=t[1]-t[0] # période d'échantillonnage (en s)


## Tracer de la courbe U = f(t) pour visualiser l'évolution de la tension pendant l'acquisition
plt.plot(t,U)
plt.xlabel('t (s)')
plt.ylabel('U (V)')
plt.grid()
plt.show()

fourier = np.fft.rfft(U)
freq=np.fft.rfftfreq(N,d=Te)
amplfourier=np.abs(fourier)*2/N
plt.plot(freq,amplfourier)
plt.axis([0,0.1e6,-0.1,3])
plt.xlabel('f (Hz)')
plt.ylabel('amplitudes (V)')
plt.grid()
plt.show()