#!/usr/bin/env python
# Wenchang Yang (wenchang@princeton.edu)
# Thu May 21 11:24:22 EDT 2020
if __name__ == '__main__':
    from misc.timer import Timer
    tt = Timer(f'start {__file__}')
import sys, os.path, os, glob
import xarray as xr, numpy as np, pandas as pd
#import matplotlib.pyplot as plt
#more imports
from xtc import tc_count
from misc.cim import cim, sem #confidence interval of the mean
maindir = '/tigress/wenchang/analysis/seedTC'
if maindir not in sys.path:
    sys.path.append(maindir)
from AM2p5.data_ntc import get_ntc
#
if __name__ == '__main__':
    tt.check('end import')
    
#start from here
source = 'AM2.5_amipHadISST'
basin = 'EP'
figname = f'data_ntc_{basin}_{source}_{tt.today()}.png'
 
if __name__ == '__main__':
    from wyconfig import * #my plot settings
    da = get_ntc(basin=basin)
    da.groupby('time.year').sum('time').plot(hue='en')
    plt.title(f'{basin} N_TC: {source}', loc='left')

    plt.savefig(figname)
    print('[saved]:', figname)

    tt.check(f'**Done**')
    plt.show()