import numpy as np
import matplotlib.pyplot as plt
import scipy.fftpack

data1 = np.loadtxt('D:/SKRIPSI FIX/data/afni/b7.csv',delimiter = ",")
oc3 = data1[0:256,3]
oc4 = data1[0:256,4]
data2 = np.loadtxt('D:/SKRIPSI FIX/data/afni/g7.csv',delimiter = ",")
fc3 = data2[0:256,3]
fc4 = data2[0:256,4]
data3 = np.loadtxt('D:/SKRIPSI FIX/data/afni/j7.csv',delimiter = ",")
pc3 = data3[0:256,3]
pc4 = data3[0:256,4]
# data4 = np.loadtxt('D:/SKRIPSI/raw data/umik/afni by umik/2.csv',delimiter = ",")
# ac3 = data4[0:1800,3]
# ac4 = data4[0:1800,4]

# Number of samplepoints
N = 256
# n = 1800
# sample spacing
T = 1.0 / 50
x = np.linspace(0.0, N*T, N)
# xn = np.linspace(0.0, n*T, n)
# y = np.sin(75.0 * 2.0*np.pi*x) + 0.5*np.sin(20.0 * 2.0*np.pi*x)
of = scipy.fftpack.fft(oc4)
pf = scipy.fftpack.fft(pc4)
ff = scipy.fftpack.fft(fc4)
# af = scipy.fftpack.fft(ac3)
xf = np.linspace(0.0, 1.0/(2.0*T), N/2)
# xfn = np.linspace(0.0, 1.0/(2.0*T), n/2)

fig, ax = plt.subplots()
ax.plot(xf, 2.0/N * np.abs(of[:N//2]), label = 'membuka')
ax.plot(xf, 2.0/N * np.abs(ff[:N//2]), label = 'menggenggam')
ax.plot(xf, 2.0/N * np.abs(pf[:N//2]), label = 'menjepit')
# ax.plot(xfn, 2.0/n * np.abs(af[:n//2]), label = 'all')
plt.title('Channel 4')
plt.xlabel('Frequency (Hz)')
plt.ylabel('Amplitude')
plt.legend()
plt.show()