import numpy as np
import matplotlib
matplotlib.rcParams.update({'font.size': 16})
import matplotlib.pyplot as plt
from scipy.signal import savgol_filter
# import seaborn
# seaborn.set_style("whitegrid", {'axes.grid' : False})

data = np.genfromtxt('mcmc_fitted_components_170ns.dat',
	names=True
	)
time = data['time']*1e6 # ns
signal = data['signal']*1e3 # raw data, mV
one_signal = data['one_signal']*1e3
two_signal = data['two_signal']*1e3

plt.figure(figsize=(7, 6))
plt.plot(time,signal,label='data',color='grey')
plt.plot(time,one_signal+two_signal,label='Bayesian fit', color='black')
plt.plot(time,one_signal,label='Bayesian fit 1st photon', color='blue')
plt.plot(time,two_signal,label='Bayesian fit 2nd photon',color='red')
plt.xlim(0,1.5)
# plt.xlim(0,0.2)
plt.xlabel(r'time ($\mu$s)')
plt.ylabel('TES signal (mV)')
# plt.legend()
plt.tight_layout
plt.savefig('mcmc_fitted_components_170ns.eps')
# plt.figure()
# plt.plot(time[1:],savgol_filter(np.diff(signal),51,1))
# plt.plot(time[1:],np.diff(signal))
# plt.xlim(0,1.5)
plt.show()