# -*- coding: utf-8 -*-
import numpy as np
import matplotlib.pylab as plt
from matplotlib.font_manager import FontProperties
def smooth(signal, window_size):
extended_signal = signal[window_size:0:-1] + signal + signal[-1:-window_size:-1]
s = np.array(extended_signal)
w = np.hamming(window_size)
y = np.convolve(w/w.sum(), s, mode="same")
return y[window_size:-window_size+1]

temp_file = open("EDC_dD_temp_estim.csv")
data_rows = [x.split() for x in temp_file if not x.startswith("#") and len(x.split()) == 5]
temp_years = [float(x[1]) for x in data_rows]
raw_temp_vals = [float(x[3]) for x in data_rows]
temp_vals = list(smooth(raw_temp_vals[:4000], 50)) + list(smooth(raw_temp_vals[4000:6000], 10)) + raw_temp_vals[6000:]
co2_file1 = open("Vostok_CH4_CO2_age.csv")
data_rows = [x.split("\t") for x in co2_file1 if not x.startswith('"')]
co2_years1 = [float(x[0]) for x in data_rows if x[2] != "\n"]
co2_vals1 = [float(x[2].strip()) for x in data_rows if x[2] != "\n"]
co2_file2 = open("EDC_CO2_423-391_kyr_BP.csv")
data_rows = [x.split() for x in co2_file2 if not x.startswith('"')]
co2_years2 = [float(x[1]) for x in data_rows]
co2_vals2 = [float(x[2]) for x in data_rows]
co2_file3 = open("EDC_CO2_650-413_kyr_BP.csv")
data_rows = [x.split() for x in co2_file3 if not x.startswith('"')]
co2_years3 = [float(x[1]) for x in data_rows]
co2_vals3 = [float(x[2]) for x in data_rows]
co2_file4 = open("EDC99_CO2_bern.csv")
data_rows = [x.split() for x in co2_file4 if x.startswith("3")]
co2_years4 = [float(x[1]) for x in data_rows]
co2_vals4 = [float(x[2]) for x in data_rows]
plt.title("Temperature and CO$_2$ Records", size=28)
temp_ax = plt.axes()
temp_ax.yaxis.tick_left()
tline = plt.plot(temp_years, temp_vals, "red")
plt.ylabel(u"Temperature anomaly (°C)", size=16)
plt.xlabel("Thousands of Years Ago", size=16)
plt.xlim(800,0)
plt.ylim(-11,15)
co2_ax = plt.axes([0.125,0.1,0.775,0.8], frameon=False)
co2_ax.yaxis.tick_right()
co2_ax.yaxis.set_label_position("right")
plt.xticks([])
cline1 = plt.plot(co2_years1, co2_vals1, "#0000AA")
cline2 = plt.plot(co2_years2, co2_vals2, "#2288AA")
cline3 = plt.plot(co2_years3, co2_vals3, "#44AAFF")
cline4 = plt.plot(co2_years4, co2_vals4, "#88DDFF")
plt.ylabel("Atmospheric CO$_2$ (ppm)", size=16)
plt.xlim(800,0)
plt.ylim(160,420)
plt.annotate("Current CO$_2$ level", (2,386.36), (300,386.36), xycoords="data", arrowprops=dict(arrowstyle="->"))
plt.legend((tline, cline1, cline2, cline3, cline4), ("Temperature (EPICA Dome C)", "CO$_2$ (Vostok)", "CO$_2$ (EPICA Dome C)", "CO$_2$ (EPICA Dome C)", "CO$_2$ (EPICA Dome C)"), "upper left", prop=FontProperties(size=9), pad=0.1, handletextsep=0.005)
plt.savefig("py-co2-temp.svg")
plt.show()