# -*- coding:iso-8859-1 -*-
from scipy import *
import scipy.io.array_import
from scipy import gplt
from scipy import fftpack

tempdata = scipy.io.array_import.read_array('taches_solaires.dat')

annees=tempdata[:,0]
wolf=tempdata[:,1]
gplt.plot(annees,wolf,'title "Mesures" with linespoints')
gplt.xtitle('Années')
gplt.ytitle('Nombre de Wolf')
gplt.grid("off")
gplt.output('taches_solaires_temps.png','png medium transparent size 600 400')

Y=fft(wolf)
gplt.plot(Y.real,Y.imag,'title "Mesures" with points')
gplt.xtitle('Re(FFT)')
gplt.ytitle('Im(FFT)')
gplt.xaxis((-4000,2000))
gplt.grid("off")
gplt.output('taches_solaires_FFT.png','png medium transparent size 600 400')

n=len(Y)
puissance = abs(Y[1:(n/2)])**2
nyquist=1./2
freq=array(range(n/2))/(n/2.0)*nyquist
gplt.plot(freq[1:len(freq)], puissance,'title "Mesures" with linespoints')
gplt.xtitle('Fréquence [1/an]')
gplt.ytitle('|FFT|**2')
gplt.xaxis((0,0.20))
gplt.grid("off")
gplt.output('taches_solaires_freq.png','png medium transparent size 600 400')

periode=1./freq
gplt.plot(periode[1:len(periode)], puissance,'title "Mesures" with linespoints')
gplt.xaxis((0,40))
gplt.xtitle('Période [an]')
gplt.ytitle('|FFT|**2')
gplt.grid("off")
gplt.output('taches_solaires_period.png','png medium transparent size 600 400')