#!/usr/bin/env python
# Wenchang Yang (wenchang@princeton.edu)
# Fri Feb 12 12:23:53 EST 2021
if __name__ == '__main__':
    from misc.timer import Timer
    tt = Timer(f'start {__file__}')
import sys, os.path, os, glob, datetime
import xarray as xr, numpy as np, pandas as pd
import matplotlib.pyplot as plt
#more imports
from fig_lines_lmeTC_multicases import wyplot as plot_lines_lmeTC 
from fig_lines_lmeMDRrsst_multicases import wyplot as plot_lines_lmeMDRrsst
from fig_lines_lmeTROPsst_multicases import wyplot as plot_lines_lmeTROPsst
#from fig_heatmap_corr_liz_lme import plot_heatmap
from shared import labelax
#
if __name__ == '__main__':
    tt.check('end import')
#
#start from here
 
 
if __name__ == '__main__':
    from wyconfig import * #my plot settings
    figsize = 8,7
    fig, axes = plt.subplots(3, 1, figsize=figsize, dpi=100)

    ax = axes[0]
    plot_lines_lmeTC(ax, legend_ncol=3)
    #ax.set_title('LME TCs from various simulations', loc='left')
    ax.set_ylabel('LME Hurricane #')
    ax.set_xlabel('')
    labelax(ax, 'A'+' '*8, fontsize='large')
    
    ax = axes[1]
    plot_lines_lmeMDRrsst(ax, legend_ncol=3)
    labelax(ax, 'B'+' '*8, fontsize='large')
    ax.set_xlabel('')
    
    ax = axes[2]
    plot_lines_lmeTROPsst(ax, legend_ncol=3)
    ax.set_ylim(None, 26.5)
    ax.set_yticks(np.arange(25, 26.61, 0.4))
    labelax(ax, 'C'+' '*8, fontsize='large')
    
    figname = __file__.replace('.py', f'.png')
    if len(sys.argv) > 1 and sys.argv[1] == 'savefig':
        wysavefig(figname, dpi=128)
    tt.check(f'**Done**')
    plt.show()