import numpy as np
import matplotlib.pyplot as plt

data1 = np.genfromtxt('flur2_1.dat')[:,0]
# data2 = np.genfromtxt('flur2_3.dat')[:,0]

data1_x = np.arange(0, len(data1)) *0.001
# data2_x = np.arange(0, len(data2)) *0.003

mini = 331.4
maxi = 333.4

mini_ind = int(mini/0.001)
maxi_ind = int(maxi/0.001)
delta_ind = maxi_ind-mini_ind + 1	 # +1 because of counting

# mini_ind2 = mini_ind * 10 
# maxi_ind2 = maxi_ind * 10 
# delta_ind2 = maxi_ind2-mini_ind2 + 1 *10
# const = -3

plt.figure(figsize = (12,2))

plt.subplot(1, 1, 1)
plt.title("Fluorescence")
plt.plot(data1_x[mini_ind:maxi_ind]-mini, data1[mini_ind:maxi_ind], color='r')
# plt.plot(data2_x[mini_ind2+ const:maxi_ind2+ const], data2[mini_ind2:maxi_ind2]*10/30, color='b')
plt.ylim((0,40))
plt.ylabel("Count rate (ms$^{-1}$)")

xdat = np.array(data1_x[mini_ind:maxi_ind]-mini)
ydat = np.array(data1[mini_ind:maxi_ind])
np.savetxt("sample_fluo", np.stack((xdat,ydat),axis=1))

# mini = 58.03
# maxi = 59

# mini_ind = int(mini/0.03)
# maxi_ind = int(maxi/0.03)
# delta_ind = maxi_ind-mini_ind + 1	 # +1 because of counting

# mini_ind2 = mini_ind * 10 
# maxi_ind2 = maxi_ind * 10 
# delta_ind2 = maxi_ind2-mini_ind2 + 1 *10
# const = -3

# plt.subplot(2, 1, 2)
# plt.plot(data1_x[mini_ind:maxi_ind], data1[mini_ind:maxi_ind]/30, color='r')
# plt.plot(data2_x[mini_ind2+ const:maxi_ind2+ const], data2[mini_ind2:maxi_ind2]*10/30, color='b')
# plt.ylim((0,35))

# plt.plot(data1_x[delta_ind+mini_ind:delta_ind+maxi_ind], data1[delta_ind+mini_ind:delta_ind+maxi_ind]/30, color='r')
# plt.ylim((0,35))
# plt.plot(data2_x[delta_ind2+mini_ind2:delta_ind2+maxi_ind2], data2[delta_ind2+mini_ind2:delta_ind2+maxi_ind2]*10/30, color='b')
plt.ylabel("Count rate (ms$^{-1}$)")
plt.xlabel("Seconds (s)")
# plt.savefig('test.pdf')
plt.show()