#!/usr/bin/env python
# Wenchang Yang (wenchang@princeton.edu)
# Fri Feb 12 13:45:34 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_map_ssta_LME_fullForcing import plot_sst_anom
#
if __name__ == '__main__':
    tt.check('end import')
#
#start from here
 
 
if __name__ == '__main__':
    from wyconfig import * #my plot settings
    from shared import labelax
    import cartopy.crs as ccrs, cartopy.feature as cfeature
    proj = ccrs.Robinson(central_longitude=180)
    dproj = ccrs.PlateCarree()
    land_color = 'k'
    figsize = (9,4)
    fig, axes = plt.subplots(2, 2, figsize=figsize, subplot_kw=dict(projection=proj))

    ax = axes[0,0]
    plt.sca(ax)
    plot_sst_anom('LME', 1277, ax=ax, transform=dproj, center=0, vmax=0.32, levels=17, extend='both')
    ax.add_feature(cfeature.LAND, color=land_color)

    ax = axes[0,1]
    plt.sca(ax)
    plot_sst_anom('LME', 1277, rSST=True, ax=ax, transform=dproj, center=0, vmax=0.32, levels=17, extend='both')
    ax.add_feature(cfeature.LAND, color=land_color)

    ax = axes[1,0]
    plt.sca(ax)
    plot_sst_anom('LME', 1980, ax=ax, transform=dproj, vmax=1, center=0, levels=21, extend='both')
    ax.add_feature(cfeature.LAND, color=land_color)

    ax = axes[1,1]
    plt.sca(ax)
    plot_sst_anom('LME', 1980, rSST=True, ax=ax, transform=dproj, vmax=1, center=0, levels=21, extend='both')
    ax.add_feature(cfeature.LAND, color=land_color)
    
    figname = __file__.replace('.py', f'.png')
    if len(sys.argv) > 1 and sys.argv[1] == 'savefig':
        wysavefig(figname)
    tt.check(f'**Done**')
    plt.show()