import numpy as np 
import matplotlib.pyplot as plt 
import functions as r

##### Figure S2 #####
# load data for FigS2A
sims=np.load('coefC_tau_th.npy',allow_pickle=True)
# plot FigS2A
fig1,ax1,fig2,ax2,fig3,ax3,fig4,ax4,fig5,ax5,fig6,ax6=r.dif_2d(sims)
r.ns_region(sims,fig1,ax1,fig2,ax2,fig3,ax3,fig4,ax4,fig5,ax5,fig6,ax6)

# load data for FigS2B
sims=np.load('d_tau_th.npy',allow_pickle=True)
# plot FigS2B
fig1,ax1,fig2,ax2,fig3,ax3,fig4,ax4,fig5,ax5,fig6,ax6=r.dif_2d(sims)
r.ns_region(sims,fig1,ax1,fig2,ax2,fig3,ax3,fig4,ax4,fig5,ax5,fig6,ax6)


##### Figure S4 #####
# load data for FigS4B
sims=np.load('w1-w2v2.npy',allow_pickle=True)
# plot FigS4B
p1=1.0
p2=1.0
fig8, fig9, fig10, fig11, k = r.one_combi(sims,p1,p2)


# load data for FigS4D-E
sims=np.load('w1-w2v2.npy',allow_pickle=True)
# plot FigS4D-E
fig1,ax1,fig2,ax2,fig3,ax3,fig4,ax4,fig5,ax5,fig6,ax6=r.dif_2d(sims)
r.ns_region(sims,fig1,ax1,fig2,ax2,fig3,ax3,fig4,ax4,fig5,ax5,fig6,ax6)


##### Figure S8 #####
# load data for FigS8A-B-C
sims=np.load('coefC-LF_d-LF.npy',allow_pickle=True)
# plot FigS8A-B
fig1,ax1,fig2,ax2,fig3,ax3,fig4,ax4,fig5,ax5,fig6,ax6=r.dif_2d(sims)
r.ns_region(sims,fig1,ax1,fig2,ax2,fig3,ax3,fig4,ax4,fig5,ax5,fig6,ax6)
# plot FigS8C
p1=0.5
p2=1.7
fig8, fig9, fig10, fig11, k = r.one_combi(sims,p1,p2)

# load data for FigS8D-E-F
sims=np.load('coefC-LF_d-LF.npy',allow_pickle=True)
sims_tauth=np.load('coefC-LF_d-LF_15-tau_th.npy',allow_pickle=True)
# plot FigS8D-E
fig1,ax1,fig2,ax2,fig3,ax3,fig4,ax4,fig5,ax5,fig6,ax6=r.effect_tauth(sims, sims_tauth) 
r.ns_region(sims_tauth,fig1,ax1,fig2,ax2,fig3,ax3,fig4,ax4,fig5,ax5,fig6,ax6,sims)  
# plot FigS8F
p1=0.5
p2=1.7
fig8, fig9, fig10, fig11, k = r.one_combi(sims_tauth,p1,p2,sims)