#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Fri Sep  4 19:23:36 2020

#### Script Documentation Block  ####################################################
#
# Script Name:plot_windspd.py
#
# RFC Contact:
#
# Abstract:
# plot the GEFS ensemble Meteogram plot
#
# Input Data Source: GEFS  /com/gens/prod/gefs.yyyymmdd/
#
# Output Destination: $DATA
#
# Software Package Used: Python/3.6.3
                         Matplotlib/2.2.2
#
# Script History Log: First Implemented Sep 2020
# Original Code Development Contact: Li Xu, CPC, 301-683-1548

"""
import matplotlib
matplotlib.use('Agg')
import numpy as np
import matplotlib.pyplot as plt
import datetime

from meteogram import plot_meteorogram,rbin,read_stns,get_xi_yi

#  main execution part

D1=rbin('UGRD_ens.bin',(20,40,361,720))
Dcnt1=rbin('UGRD_cnt.bin',(40,361,720))

D2=rbin('VGRD_ens.bin',(20,40,361,720))
Dcnt2=rbin('VGRD_cnt.bin',(40,361,720))

D=np.sqrt(D1*D1+D2*D2)  # 2 wind speed
Dcnt=np.sqrt(Dcnt1*Dcnt1+Dcnt2*Dcnt2)

figsize=(10, 4)
fig1 = plt.figure(figsize=figsize)


with open('datecyc.txt','r') as f:
     dt,cyc=f.readline().split()

stns=read_stns()
for stn in stns:
    print(stn)
    stid, stname,country,lat,lon,elev,*_=stn
    xi,yi=get_xi_yi(lon,lat)
    data=D[:,:,yi,xi]
    t1=Dcnt[:,yi,xi]

    plot_meteorogram(fig1,data,t1,stn,
                     vname='Wind Speed (m/s)',
                     boxcolor='lightyellow',
                     dt=dt,
                     cyc=cyc,
                     pout='./all_plots/wind_speed')

plt.close()