#!/usr/bin/env python
# Wenchang Yang (wenchang@princeton.edu)
# Tue Nov 17 14:56:34 EST 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
#
if __name__ == '__main__':
    tt.check('end import')
#
#start from here
location = 'New York'
lat, lon = 40.7128, 360 - 74.0060 # New York lat/lon
q2m = xr.open_dataarray('/tigress/wenchang/data/era5/analysis_wy/2m_specific_humidity/daily/clim/era5.q2m.w1clim1981-2010.nc')
t2m = xr.open_dataarray('/tigress/wenchang/data/era5/analysis_wy/2m_temperature/daily/clim/era5.t2m.w1clim1981-2010.nc')
uv = xr.open_dataarray('/tigress/wenchang/data/era5/analysis_wy/mean_surface_downward_uv_radiation_flux/daily/clim/era5.msdwuvrf.w1clim1981-2010.nc')

 
if __name__ == '__main__':
    from wyconfig import * #my plot settings
    figname = __file__.replace('.py', f'_{tt.today()}.png')
    plt.close()
    plt.figure(figsize=(6.8, 6.8*9/16))

    q2m.mean('longitude').assign_attrs(units='kg/kg').plot.contourf(y='latitude', levels=20)
    q2m.mean('longitude').idxmax('week').plot(y='latitude', ls='', marker='.', markersize=None, color='C0', label='week of max q2m(shading)', fillstyle='none')

    uv.mean('longitude').plot.contour(y='latitude', levels=20, cmap='turbo')
    uv.mean('longitude').idxmax('week').plot(y='latitude', ls='', marker='*', markersize=None, color='C1', label='week of max uv(contours)', fillstyle='none')

    plt.legend(bbox_to_anchor=(1,1), loc='lower right', ncol=2)

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

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