-
Notifications
You must be signed in to change notification settings - Fork 139
/
graph_plot.py
62 lines (47 loc) · 1.52 KB
/
graph_plot.py
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
import matplotlib.pyplot as plt
from scipy import signal
import numpy as np
import scipy.fftpack
from scipy.signal import butter, lfilter
arr_red = []
arr_green = []
arr_blue = []
# frame_size = 300 #10 second of 30Hz video
# frame_buffer = []
# times = []
def butter_bandpass(lowcut, highcut, fs, order=5):
nyq = 0.5 * fs
low = lowcut / nyq
high = highcut / nyq
b, a = butter(order, [low, high], btype='band')
return b, a
def butter_bandpass_filter(data, lowcut, highcut, fs, order=5):
b, a = butter_bandpass(lowcut, highcut, fs, order=order)
y = lfilter(b, a, data)
return y
#read file signal.dat
with open("signal.dat") as f:
lines = f.readlines()
for i in range(lines.__len__()):
r,g,b = lines[i].split("%")
arr_red.append(float(r))
arr_green.append(float(g))
arr_blue.append(float(b))
green_detrended = signal.detrend(arr_blue)
L = len(arr_red)
bpf = butter_bandpass_filter(green_detrended,0.8,3,fs=30,order = 3)
even_times = np.linspace(0, L, L)
interpolated = np.interp(even_times, even_times, bpf)
interpolated = np.hamming(L)*interpolated
norm = interpolated/np.linalg.norm(interpolated)
raw = np.fft.rfft(norm*30)
freq = np.fft.rfftfreq(L, 1/30)*60
fft = np.abs(raw)**2
g = plt.figure("green")
ax2 = g.add_subplot(111)
ax2.set_title("band pass filter")
ax2.set_xlabel("time")
ax2.set_ylabel("magnitude")
plt.plot(freq,fft, color = "blue")
g.show()
input("Press Enter to exit...")