import seaborn as sb
import pandas as pd
import scipy
import matplotlib.pyplot as plt

data_uy=pd.read_csv('data/m.csv')
#  head(data_uy)

correlation_result=scipy.stats.pearsonr(data_uy['u'], data_uy['y'])
print(correlation_result)

p = sb.lmplot(
		data=data_uy,
		x='u',
		y='y',
		fit_reg=True,
		height=8,
	)
p.set(title='Correlation UY')
p.savefig('res/uy-correlation-analysis.png')
plt.clf()

lp = sb.lineplot(data=data_uy['u'])
lp.set(
		title='Signal u',
		xlabel='kT',
		ylabel='u',
	)
fig = lp.get_figure()
fig.subplots_adjust(top=.97)
fig.savefig('res/signal_u.png')
plt.clf()

lp = sb.lineplot(data=data_uy['y'])
lp.set(
		title='Signal y',
		xlabel='kT',
		ylabel='y',
	)
fig = lp.get_figure()
fig.subplots_adjust(top=.97)
fig.savefig('res/signal_y.png')
plt.clf()

imf=pd.read_csv('data/impulse_func.csv')
lp = sb.lineplot(data=imf, legend=False, linewidth=2.5)
lp.set(
		title='Impulse Function Estimate',
		xlabel='time (s)',
		ylabel='amplitude',
	)
fig = lp.get_figure()
fig.subplots_adjust(top=.97)
fig.savefig('res/impulse_func_estimate.png')
plt.clf()

acr=pd.read_csv('data/autocorrelation_u.csv')
lp = sb.lineplot(data=acr, legend=False, linewidth=2.5)
lp.set(
		title='Autocorrelation U',
		xlabel='shift',
		ylabel='Ruu',
	)
fig = lp.get_figure()
fig.subplots_adjust(top=.97)
fig.savefig('res/autocorrelation_u.png')
plt.clf()

acr=pd.read_csv('data/autocorrelation_y.csv')
lp = sb.lineplot(data=acr, legend=False, linewidth=2.5)
lp.set(
		title='Autocorrelation Y',
		xlabel='shift',
		ylabel='Ryy',
	)
fig = lp.get_figure()
fig.subplots_adjust(top=.97)
fig.savefig('res/autocorrelation_y.png')
plt.clf()

mcr=pd.read_csv('data/mutual_correlation_uy.csv')
lp = sb.lineplot(data=mcr, legend=False, linewidth=2.5)
lp.set(
		title='Mutual correlation UY',
		xlabel='shift',
		ylabel='Ruy',
	)
fig = lp.get_figure()
fig.subplots_adjust(top=.97)
fig.savefig('res/mutual_correlation_uy.png')
plt.clf()

mcr=pd.read_csv('data/mutual_correlation_yu.csv')
lp = sb.lineplot(data=mcr, legend=False, linewidth=2.5)
lp.set(
		title='Mutual correlation YU',
		xlabel='shift',
		ylabel='Ryu',
	)
fig = lp.get_figure()
fig.subplots_adjust(top=.97)
fig.savefig('res/mutual_correlation_yu.png')
plt.clf()