from read_baseline_data import *
from IPython import embed
import matplotlib.pyplot as plt
import numpy as np
from IPython import embed

## beat

data_dir = '../data'
dataset = '2018-11-09-ad-invivo-1'
inch_factor = 2.54

time, eod = read_baseline_eod(os.path.join(data_dir, dataset))
eod_norm = eod - np.mean(eod)
eod_norm = eod_norm[1000:2000]

x = np.arange(0., len(eod_norm))
y = np.sin(x*2*np.pi*550)
ampl = eod_norm + y

time_axis = np.arange(len(ampl))


fig, ax = plt.subplots(figsize=(20/inch_factor, 10/inch_factor))
plt.plot(time_axis, ampl)
ax.set_xlabel("time [ms]", fontsize = 22)
plt.xticks(fontsize = 18)
ax.set_ylabel("eod amplitude [mV]", fontsize = 22)
plt.yticks(fontsize = 18)
ax.spines["top"].set_visible(False)
ax.spines["right"].set_visible(False)
fig.tight_layout()

#plt.show()
plt.savefig('beat.png')