# -*- coding: utf-8 -*-
"""
Created on Thu Mar 16 10:40:10 2023

@author: regin
"""


import numpy as np
import matplotlib.pyplot as plt
from uncertainties import unumpy as unp


plt.rcParams.update({
  "text.usetex": True,
})

size=20.0

plt.rcParams['figure.figsize'] = (6.4, 4.8) 
plt.rcParams['axes.labelsize'] =size
plt.rcParams['xtick.labelsize'] =size
plt.rcParams['ytick.labelsize'] =size
plt.rcParams['xtick.direction'] = "in"
plt.rcParams['ytick.direction'] = "in"
plt.rcParams['legend.fontsize'] =size
plt.rcParams['legend.title_fontsize'] =size
plt.rcParams['legend.frameon'] = False
plt.rcParams['legend.labelspacing'] = 0.05
plt.rcParams['lines.linewidth'] = 2.0
plt.rcParams['lines.markersize'] = 5.0
plt.rcParams['font.family'] = 'lmodern' 

data=np.loadtxt('../data/data_trajectory.dat')
D=1
#a=1/(2*np.pi)
a=1
tau=a**2/D
d=a


t=data[:,0]
x1=data[:,1]
x1_ohne=data[:,2]
diff=data[:,3]
dt=0.01
s=0
e=1000
plt.xlabel(r'$t/\tau$')
plt.ylabel(r'$x/a$')
plt.plot(dt*t[s:e]/tau,np.array(x1_ohne[s:e])/d,label=r'$\Delta x_{\mathrm{f}}(t)$',color='#31a354')
plt.plot(dt*t[s:e]/tau,np.array(x1[s:e])/d,alpha=0.8,label='$\Delta x(t)$', color='#3182bd')
plt.plot(dt*t[s:e]/tau,diff[s:e]/d,label='$\delta x(t)$',color='#e6550d')
#plt.plot(t[s:e],a,label=r'$\vec{R}_{triangle}(t)-\vec{R}_{non}(t)$')
#plt.ylim(7,13)
plt.xscale('log')
#plt.yscale('log')
plt.tight_layout()
plt.legend(loc=0)
plt.show()


t=data[:,0]
x1=data[:,1]
x1_ohne=data[:,2]
diff=data[:,3]
dt=0.01
s=0
e=1000
plt.xlabel(r'$t/\tau$')
plt.ylabel(r'$x/a$')
plt.plot(dt*t[s:e]/tau,np.array(x1_ohne[s:e])/d,label=r'$\Delta x^{\mathrm{f}}(t)$',color='black')
plt.plot(dt*t[s:e]/tau,np.array(x1[s:e])/d,alpha=0.9,label='$\Delta x(t)$', color='blue')
plt.plot(dt*t[s:e]/tau,diff[s:e]/d,label=r'$\Delta x^{\mathrm{red}}(t)$',color='red')
#plt.plot(t[s:e],a,label=r'$\vec{R}_{triangle}(t)-\vec{R}_{non}(t)$')
#plt.ylim(7,13)
plt.tight_layout()
plt.legend()
plt.tick_params(axis='both', which='both', right='on',top='on',pad=6)
#plt.savefig('../plots/Fig_1.pdf')

plt.show()