#!/usr/bin/env python
# Wenchang Yang (wenchang@princeton.edu)
# Fri Apr 16 11:05:03 EDT 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, matplotlib.pyplot as plt
#more imports
from fig_heatmap_corr_liz_lme import wyplot as plot_hu
from fig_heatmap_corr_liz_lmeMH import wyplot as plot_mh
from shared import labelax
#
if __name__ == '__main__':
    tt.check('end import')
#
#start from here
 
 
if __name__ == '__main__':
    from wyconfig import * #my plot settings
    fig, axes = plt.subplots(2, 1, figsize=(8,7), sharex=True)
    ax = axes[0]
    plot_hu(ax)

    ax = axes[1]
    plot_mh(ax)
    
    axes[0].set_xlabel('')
    fig.suptitle('Heatmaps of correlations between sedimentary reconstructions of TCs and LME HUs/MHs')
    for ax,label,title in zip(axes, list('ab'), ['LME HUs', 'LME MHs']):
        ax.set_title(title)
        #labelax(ax, label)

    #savefig
    if len(sys.argv) > 1 and sys.argv[1] == 'savefig':
        figname = __file__.replace('.py', f'.png')
        wysavefig(figname)
    tt.check(f'**Done**')
    plt.show()