#!/usr/bin/env python
# Wenchang Yang (wenchang@princeton.edu)
# Tue Feb 21 10:40:04 EST 2023
if __name__ == '__main__':
    import sys,os
    from misc.timer import Timer
    tt = Timer(f'[{os.getcwd()}] start ' + ' '.join(sys.argv))
import sys, os.path, os, glob, datetime
import xarray as xr, numpy as np, pandas as pd, matplotlib.pyplot as plt
#more imports
#from misc import get_kws_from_argv
from misc.wysalem import get_world_shape
#
if __name__ == '__main__':
    tt.check('end import')
#
#start from here
shdf = get_world_shape('Nepal')
locname, lat0, lon0 = 'Lalitpur', 27.6588, 85.3247
latera5, lonera5 = 27.75, 85.25
latchirps, lonchirps = 27.625, 85.375
def wyplot_box(lonc, latc, size=0.25, ax=None, **kws):
    if ax is None:
        fig,ax = plt.subplots()
    lon1 = lonc - size/2
    lon2 = lonc + size/2
    lat1 = latc - size/2
    lat2 = latc + size/2
    lonvec = [lon1, lon2, lon2, lon1, lon1]
    latvec = [lat1, lat1, lat2, lat2, lat1]
   
    ax.plot(lonvec, latvec, **kws)
    

 
 
if __name__ == '__main__':
    from wyconfig import * #my plot settings
    from geoplots import mapplot
    shdf.boundary.plot()
    mapplot()
    
    ax = plt.gca()
    ax.plot(lon0, lat0, marker='x', label=locname)
    wyplot_box(lonera5, latera5, ax=ax, label='ERA5 nearest grid')
    wyplot_box(lonchirps, latchirps, ax=ax, label='CHIRPS nearest grid')
    ax.legend()


    
    #savefig
    if 'savefig' in sys.argv or 's' in sys.argv:
        figname = __file__.replace('.py', f'.png')
        if 'overwritefig' in sys.argv or 'o' in sys.argv:
            wysavefig(figname, overwritefig=True)
        else:
            wysavefig(figname)
    tt.check(f'**Done**')
    print()
    if 'notshowfig' in sys.argv or 'n' in sys.argv:
        pass
    else:
        if 'plt' in globals(): plt.show()