GPU-Tracking-code/test_naff.py at master · skostogl/GPU-Tracking-code · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
import numpy as np
import matplotlib.pyplot as plt

from modules.tracker import *
from modules.naff import *
from modules.grid import *
from modules.tune_resonances import *
from modules.FMA import *
from modules.NAFF import *
from modules.fft_naff import *

############ Tune#####################
data=[]
data_init =[]
data_f1   =[]
data_final=[]
#t=np.arange(0.0,0.1,0.0001)
t=np.arange(0.0,10,0.001)
#t=np.arange(0.0,10,0.1)
for i in t:
      #data.append(100.0*p.sin(0.5*2.0*.141592*i))
      #data.append((0.1*np.cos(2*3.141592*300.0*i)))
      data.append((0.1*np.cos(2*3.141592*300.0*i))+ ((0.5*np.cos(2*3.141592*400.0*i)))+(0.6*np.cos(2*3.141592*350.0*i)))
coord=Vec_cpp()
zero=Vec_cpp()
coord.extend(i for i in data)
zero.extend((0)*i for i in data)
naff=NAFF()
tune_all=naff.get_f(coord,zero)
#print tune_all
#tune_all=get_f1(coord,zero)
for i in tune_all:
    print i
###print tune_all
plt.plot(t,data,linestyle='-')
plt.show()
############################################## LHC DATA ##############################3
f = open('signal.dat','r')
for line in f:
    parts = line.split()
    data_init.append( complex (float(parts[0]), float(parts[1])) )
#    data_final.append( complex (float(parts[4]), float(parts[5])) )
#    data_f1.append( complex (float(parts[2]), float(parts[3])) )
#t_final  = np.arange( len(data_final) )
#t_f1 = np.arange(len(data_f1))
t_init = np.arange(len(data_init))
#sp_final = np.fft.fft(data_final)
#sp_f1= np.fft.fft(data_f1)
sp_init= np.fft.fft(data_init)
#freq_final  = np.fft.fftfreq(t_final.shape[-1])
#freq_f1 = np.fft.fftfreq(t_f1.shape[-1])
freq_init = np.fft.fftfreq(t_init.shape[-1])
plt.plot(freq_init,(sp_init))
plt.xlabel(r'Frequency', fontsize=20)
plt.ylabel(r'Amplitude', fontsize=20)
plt.xticks(fontsize=20)
plt.yticks(fontsize=20)
plt.title('FFT for initial signal')
plt.tight_layout()
plt.grid(True)
plt.show()
#plt.plot(freq_f1,(sp_f1))
#plt.xlabel(r'Frequency', fontsize=20)
#plt.ylabel(r'Amplitude', fontsize=20)
#plt.xticks(fontsize=20)
#plt.yticks(fontsize=20)
#plt.title('FFT for data that will be subtracted C++')
#plt.tight_layout()
#plt.grid(True)
#plt.show()
#
#plt.plot(freq_final,(sp_final))
#plt.xlabel(r'Frequency', fontsize=20)
#plt.ylabel(r'Amplitude', fontsize=20)
#plt.xticks(fontsize=20)
#plt.yticks(fontsize=20)
#plt.title('FFT for final signal')
#plt.tight_layout()
#plt.grid(True)
#plt.show()
#
#plt.plot(data_init)
##plt.show()
##plt.plot(data_f1)
#plt.plot(data_final,ms=4,linewidth=4,color='r')
#plt.show()
#quit()
#