import os
import sys
import glob
import numpy as np
import matplotlib
import matplotlib.pyplot as plt
from matplotlib import rc
from IPython import embed
plt.rc('text', usetex=True)
plt.ion()
#
pT = []
files_luck = []
r10 = 10.
rlow = 0.1
num_history = 50
#------data------------
data1 = '../lucky_data/' 
data2 = '../lucky_data_180321/' 
data3 = '../lucky_data_180329/' 
data4 = '../lucky_data_180404/' 
#
files1 = glob.glob(data1+'lucky*.dat')
files2 = glob.glob(data2+'lucky*.dat')
files3 = glob.glob(data3+'lucky*.dat')
files4 = glob.glob(data4+'lucky*.dat')
files = files1+files2+files3+files4
print(len(files))
for i in range(0, len(files)):
  lucky = np.loadtxt(files[i])
  a2 = lucky[:,4]
  pT.append(lucky[0,0])
  files_luck.append(files[i])
print('realizations=', len(pT))
#
index = np.argsort(pT)
a = np.loadtxt(files_luck[index[-1]])
print('Lousy', files_luck[index[-1]])
t = a[:,0]
a1=a[:,3]
a2=a[:,4]

plt.close('all')
siz_text = 25
siz_tit = 30
siz_leg = 25
plt.rc('font', family='Times', size=siz_text)
plt.rc('xtick',labelsize=siz_text)
plt.rc('ytick',labelsize=siz_text)
plt.tight_layout(pad=0)
plt.subplots_adjust(left=0.145, right=.95, top=.975, bottom=0.18)
plt.tick_params(top='on', right='on', direction='in', which='both')
#xr = [0.,max(t)+2]
xr = [0.,3000]
#yr = [8.,50]
yr = [10.,50]
plt.plot(t,a1,'r', drawstyle='steps', label=r'slowest')
#plt.plot(t,a2,'og',markerfacecolor='none')
plt.xlim(xr)
plt.ylim(yr)
#plt.yticks(np.linspace(0,50,6))
plt.xlabel(r'$t \, [\rm{s}]$',fontsize=siz_tit)
plt.ylabel(r'$r\,[\mu\rm{m}]$',fontsize=siz_tit)
#
a = np.loadtxt(files_luck[index[0]])
print('Lucky', files_luck[index[0]])
t = a[:,0]
a1=a[:,3]
a2=a[:,4]
plt.plot(t,a1,'k',markerfacecolor='none',drawstyle='steps', label=r'fastest')
#plt.plot(t,a2,'.k',markerfacecolor='none')
#
leg = plt.legend(fontsize=siz_leg, loc='best', handlelength=1, frameon=False)
def color_legend_texts(leg):
    """Color legend texts based on color of corresponding lines"""
    for line, txt in zip(leg.get_lines(), leg.get_texts()):
        txt.set_color(line.get_color())  
color_legend_texts(leg)
#
# save the figure
filename=sys.argv[0]
print(filename)
basename=os.path.basename(filename)
base=basename.split('.')[0]
current_folder_path, current_folder_name = os.path.split(os.getcwd())
plt.savefig(base+'.eps',dpi=300)
print('!mv'+' '+base+'.eps'+' '+'~/tex/xiangyu/lucky/Figs/')