import pandas as pd
from matplotlib import pyplot as plt

from lib.lib import read_sp, aggregate_sp_fd

data_2d_sp = read_sp("../data/rust-sticking-probability")
data_2d_sp_low_p = read_sp("../data/rust-sp-low-p")
full_data = pd.concat([data_2d_sp_low_p, data_2d_sp])
sp_fd_data = aggregate_sp_fd(full_data)

# %%
plt.fill_between(sp_fd_data.index, sp_fd_data.fd - sp_fd_data.fd_std, sp_fd_data.fd + sp_fd_data.fd_std, alpha=0.2, label=f"Standard error band")
plt.plot(sp_fd_data.index, sp_fd_data.fd, color='tab:blue', label='NF, fd mean, seeds = 100')

plt.xlabel("$p_{stick}$")
plt.ylabel("$fd$")
plt.legend()
# plt.show()
# plt.loglog(sp_fd_data.index, sp_fd_data.fd, color='tab:blue', label='NF, fd mean, seeds = 100')
# plt.show()

# %%
plt.savefig('../figures/sp-fd.svg')
plt.savefig('../figures/sp-fd.png')
plt.show()