#!/usr/bin/env python
# Wenchang Yang (wenchang@princeton.edu)
# Wed Jun 17 15:37:40 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_basins
maindir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
if maindir not in sys.path:
    sys.path.append(maindir)
from ERA5.data_seedxp_map_noLP import da as seedxp
#
if __name__ == '__main__':
    tt.check('end import')
#
#start from here
def get_seedxp_basinsum(basin='NA'):
    #basin = 'NA'
    bs = tc_basins()
    da = seedxp
    return da.where(bs.mask(da)==bs.map_keys(basin)).where(da.lat<30).sum(['lon', 'lat'])
 
if __name__ == '__main__':
    from wyconfig import * #my plot settings
    fig, ax = plt.subplots()
    basin = 'NA'
    figname = __file__.replace('.py', f'.{basin}.{tt.today()}.png')
    da = get_seedxp_basinsum(basin=basin)
    da.groupby('time.year').sum('time').plot()
    plt.savefig(figname)
    print('**saved**:', figname)

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