import os
import sys
#import pencil as pc
import numpy as np
import matplotlib
import matplotlib.pyplot as plt
from matplotlib import rc
matplotlib.rcParams['text.latex.preamble'] = [r'\boldmath']

#----runs------
#epsilon=0.0065
run0='../SW512condens0_coag_grav_turb_shima_f1em2_ngp_a24_L1em47_b/RDF/rel_vel_mom_pdfs/'
#epsilon=0.71
run1='../SW512condens0_coag_grav_turb_shima_f1em1_ngp_a24_L4em1/RDF/rel_vel_mom_pdfs/'
# load the moments
mom0_0= np.loadtxt(run0+'mom0.txt')
r_0=np.loadtxt(run0+'r.txt')
#
mom0_1= np.loadtxt(run1+'mom0.txt')
r_1=np.loadtxt(run1+'r.txt')
#plot
plt.close()
plt.loglog(r_0,mom0_0/r_0**2,'-o',color='b')
plt.loglog(r_1,mom0_1/r_1**2,'-o',color='r')
plt.rc('text', usetex=True)
plt.rc('font', family='Times')
plt.rc('xtick',labelsize=14)
plt.rc('ytick',labelsize=14)
plt.xlabel(r'$r/\eta$',fontsize=18)
plt.ylabel(r'$m_0/r^2$',fontsize=18)
plt.xlim([1.e-2,1.2e0])
plt.ylim([8.e-1,1.1e1])
plt.tight_layout(pad=0)
plt.subplots_adjust(left=0.11, right=.98, top=.98, bottom=0.10)
plt.text(5.e-2,5.5e0,r'$\epsilon=0.0065$',style='normal',fontsize=14,color='b')
plt.text(1.e-1,2.3e0,r'$\epsilon=0.71$',style='normal',fontsize=14,color='r')
plt.text(3.5e-1,9.e0,r'turb $\&$ grav',style='normal',fontsize=20,color='black')
plt.show()

#----save figure----
filename=sys.argv[0]
print(filename)
basename=os.path.basename(filename)
base=basename.split('.')[0]
plt.savefig(base+'.eps',dpi=300)
print('!mv'+' '+base+'.eps'+' '+'~/tex/nils/particles_coag/fig/')