#!/usr/bin/env python
# Wenchang Yang (wenchang@princeton.edu)
# Wed Jun 30 16:47:02 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
#
if __name__ == '__main__':
    tt.check('end import')
#
#start from here
ens = range(1,5+1)
for en in ens:
    print(f'{en = }')
    ifiles = f'/tigress/wenchang/MODEL_OUT/HistRCP45_tigercpu_intelmpi_18_576PE/en{en:02d}/POSTP/????0101.atmos_daily.nc'
    ifile = glob.glob(ifiles)[0]
    data_name = "t_ref_max"
    ofile = f'tmax_FLOR_HistRCP45_en{en:02d}.nc'
    xname = 'grid_xt'
    xlim = (360-125, 360-117)
    yname = 'grid_yt'
    ylim = (43, 54)
    
    x = xr.open_mfdataset(ifile)[xname].values
    ix = x.argsort()
    L = (x>=xlim[0]) & (x<=xlim[1])
    ixlim = ix[L][0], ix[L][-1]
    print(f'xlim: {xlim}; ixlim: {ixlim}')
    
    y = xr.open_mfdataset(ifile)[yname].values
    iy = y.argsort()
    L = (y>=ylim[0]) & (y<=ylim[1])
    iylim = iy[L][0], iy[L][-1]
    print(f'ylim: {ylim}; iylim: {iylim}')
    
    cmd = f'ncrcat -v {data_name} -d {xname},{ixlim[0]},{ixlim[1]} -d {yname},{iylim[0]},{iylim[1]} {ifiles} {ofile}'
    print(cmd, '...')
    s = os.system(cmd)
    if s==0:
        print('[OK]:', cmd)
 
 
if __name__ == '__main__':
    #from wyconfig import * #my plot settings
    
    #savefig
    if 'savefig' in sys.argv:
        figname = __file__.replace('.py', f'.png')
        wysavefig(figname)
    tt.check(f'**Done**')
    plt.show()