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

#### Script Documentation Block  ####################################################
#
# Script Name:plot_precip.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
D=rbin('APCP_ens.bin',(20,40,361,720))
Dcnt=rbin('APCP_cnt.bin',(40,361,720))


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='Precipitation (mm)',
                     boxcolor='lightgreen',
                     dt=dt,
                     cyc=cyc,
                     pout='./all_plots/precip')



plt.close()