#!/usr/bin/env python
# coding: utf-8

# In[78]:


import panel as pn
import pandas as pd
import numpy as np
from datetime import datetime, timedelta

#pn.extension(sizing_mode='stretch_width')


# In[ ]:


dates_anl = pd.date_range(start='2020-12-01 00:00', end='2020-12-15 00:00', freq='1D')

def get_date_fct(date_15d):
    date_15d += timedelta(days=15)
    #return pd.date_range(start=date_anl, end=date_15d, freq='6H')
    return date_15d

URL_FMT = ('http://ftp1.cptec.inpe.br/pesquisa/das/carlos.bastarz/oensMB09/exps/{exp}/prod/{prod}/{datei}/{var}{datei}{datef}.png') 
VARS = ['psnm', 'geop', 'temp']
EXPS = ['gnu_singularity_m128p_p64p', 'gnu_egeon_m128p_p64p', 'intel_egeon_m128p_p64p'] 
PRODS = ['spread', 'cluster']

#fcst_hour = ['00H', '06H', '12H' , '18H', '24H']

# build widgets
image_rows = []
for v in range(1, len(VARS)):
    for e in range(1, len(EXPS)):
        for p in range(1, len(PRODS)):
            image_row = pn.Row()
            fcst_hour = f"{6 * v:02d}H"
            #print(v, e, p, fcst_hour)
            for var in VARS:
                for exp in EXPS:
                    for prod in PRODS:
                        datei = dates_anl[0]
                        datef = get_date_fct(datei)
                        url = URL_FMT.format(var=var, v=v, exp=exp, e=e, prod=prod, p=p, datei=datei.strftime('%Y%m%d%H'), datef=datef.strftime('%Y%m%d%H'))
                        image_row.append(pn.pane.PNG(url))
                    image_rows.append((fcst_hour, image_row))  
  
# build tabs
tabs = pn.Tabs(*image_rows, align="center")

# build player
player = pn.widgets.Player(
    end=len(image_rows) - 1,
    loop_policy="loop",
    align="center",
)
player.jslink(tabs, value="active", bidirectional=True)

# create dashboard
pn.Column(tabs, player)


# In[ ]:


dates_anl = pd.date_range(start='2020-12-01 00:00', end='2020-12-15 00:00', freq='1D')
#datei = dates_anl[0]
datef = dates_anl[0]

#URL_FMT = ('http://ftp1.cptec.inpe.br/pesquisa/das/carlos.bastarz/oensMB09/exps/{exp}/prod/{prod}/{datei}/{var}{datei}{datef}.png') 
VARS = ['psnm', 'geop', 'temp']
EXPS = ['gnu_singularity_m128p_p64p', 'gnu_egeon_m128p_p64p', 'intel_egeon_m128p_p64p'] 
PRODS = ['spread', 'cluster']

URL_FMT = ('http://ftp1.cptec.inpe.br/pesquisa/das/carlos.bastarz/oensMB09/exps/gnu_singularity_m128p_p64p/prod/spread/2020120100/geop2502020120100{datef}.png')

image_rows = []

for i in np.arange(0,366,6):
    image_row = pn.Row()
    fcst_hour = f"{i:03d}H"
    datef += timedelta(hours=6)
    url = URL_FMT.format(datef=datef.strftime('%Y%m%d%H'))
    image_row.append(pn.pane.PNG(url))
    image_rows.append((fcst_hour, image_row))
    
tabs = pn.Tabs(*image_rows, align='center', tabs_location='left')

# build player
player = pn.widgets.Player(end=len(image_rows) - 1, loop_policy='loop', align='center')

player.jslink(tabs, value='active', bidirectional=True)

# create dashboard
pn.Column(tabs, player).show()


# In[63]:


exps = ['gnu_singularity_m128p_p64p', 'gnu_egeon_m128p_p64p', 'intel_egeon_m128p_p64p'] 
prods = ['spread', 'cluster']
dates = pd.date_range(start='2020-12-01 00:00', end='2020-12-15 00:00', freq='1D')

vars_spread = ['psnm', 'geop250', 'geop500', 'temp850']
vars_spaguetti = ['geop500', 'temp850', 'temp925', 'temp1000']
vars_cluster = ['precpsnm1000', 'precvento250', 'precvento925', 'temp700', 'temp850', 'temp925', 'temp1000', 'tems1000', 'zgeo200', 'zgeo500']
vars_probagr = ['semana 1', 'semana 2', 'semana 3']
vars_perturbations = ['temp250', 'temp500', 'temp850', 'uvel250', 'uvel500', 'uvel850', 'vvel250', 'vvel500', 'vvel850']

regs_spaguetti = ['gl', 'as']

date = pn.widgets.Select(name='Data', options=list(dates))
exp = pn.widgets.Select(name='Experimento', options=list(exps))
var_spread = pn.widgets.Select(name='Variável', options=list(vars_spread))
var_spaguetti = pn.widgets.Select(name='Variável', options=list(vars_spaguetti))
var_cluster = pn.widgets.Select(name='Variável', options=list(vars_cluster))
var_probagr = pn.widgets.Select(name='Variável', options=list(vars_probagr))
var_perturbations = pn.widgets.Select(name='Variável', options=list(vars_perturbations))

reg_spaguetti = pn.widgets.Select(name='Região', options=list(regs_spaguetti))

url_fmt_spread = ('http://ftp1.cptec.inpe.br/pesquisa/das/carlos.bastarz/oensMB09/exps/{exp}/prod/{prod}/{datei}/{var}{datei}{datef}.png')
url_fmt_spaguetti = ('http://ftp1.cptec.inpe.br/pesquisa/das/carlos.bastarz/oensMB09/exps/{exp}/prod/{prod}/{datei}/spt{reg}{var}{datei}{datef}.png')
url_fmt_cluster = ('http://ftp1.cptec.inpe.br/pesquisa/das/carlos.bastarz/oensMB09/exps/{exp}/prod/{prod}/{datei}/{prod}{var}{datei}{datef}.png')
url_fmt_probability = ('http://ftp1.cptec.inpe.br/pesquisa/das/carlos.bastarz/oensMB09/exps/{exp}/prod/{prod}/{datei}/prec{datei}{datef}.png')
url_fmt_probagr = ('http://ftp1.cptec.inpe.br/pesquisa/das/carlos.bastarz/oensMB09/exps/{exp}/prod/{prod}/{datei}/prec_agric_large{num}.png')
url_fmt_perturbations = ('http://ftp1.cptec.inpe.br/pesquisa/das/carlos.bastarz/oensMB09/exps/{exp}/prod/{prod}/{datei}/perturbations{var}_{datei}{datei}.png')
url_fmt_chievol = ('http://ftp1.cptec.inpe.br/pesquisa/das/carlos.bastarz/oensMB09/exps/{exp}/prod/{prod}/{datei}/chi_evol{datei}{datef}.png')

@pn.depends(date, exp, var_spread)
def fig_spread(date, exp, var_spread):

    datei = date
    datef = date
    
    image_rows = []
    
    #for i in np.arange(0,120,6):
    for i in np.arange(0,360,24):
        image_row = pn.Row()
        fcst_hour = f"{i:03d}H"
        #datef += timedelta(hours=6)
        datef += timedelta(hours=24)
        url = url_fmt_spread.format(exp=exp, var=var_spread, prod='spread', datei=datei.strftime('%Y%m%d%H'), datef=datef.strftime('%Y%m%d%H'))
        image_row.append(pn.pane.PNG(url))
        image_rows.append((fcst_hour, image_row))
    
    tabs = pn.Tabs(*image_rows, align='center')
    player = pn.widgets.Player(end=len(image_rows) - 1, loop_policy='loop', align='center')
    player.jslink(tabs, value='active', bidirectional=True)

    return pn.Column(tabs, player)
      
@pn.depends(date, exp, var_spaguetti, reg_spaguetti)
def fig_spaguetti(date, exp, var_spaguetti, reg_spaguetti):

    datei = date
    datef = date
    
    image_rows = []
    
    for i in np.arange(0,360,24):
        image_row = pn.Row()
        fcst_hour = f"{i:03d}H"
        datef += timedelta(hours=24)
        url = url_fmt_spaguetti.format(exp=exp, var=var_spaguetti, reg=reg_spaguetti, prod='spaguetti', datei=datei.strftime('%Y%m%d%H'), datef=datef.strftime('%Y%m%d%H'))
        image_row.append(pn.pane.PNG(url))
        image_rows.append((fcst_hour, image_row))
    
    tabs = pn.Tabs(*image_rows, align='center')
    player = pn.widgets.Player(end=len(image_rows) - 1, loop_policy='loop', align='center')
    player.jslink(tabs, value='active', bidirectional=True)

    return pn.Column(tabs, player)
    
@pn.depends(date, exp, var_cluster)
def fig_cluster(date, exp, var_cluster):

    datei = date
    datef = date
    
    image_rows = []
    
    for i in np.arange(0,360,24):
        image_row = pn.Row()
        fcst_hour = f"{i:03d}H"
        datef += timedelta(hours=24)
        url = url_fmt_cluster.format(exp=exp, var=var_cluster, prod='cluster', datei=datei.strftime('%Y%m%d%H'), datef=datef.strftime('%Y%m%d%H'))
        image_row.append(pn.pane.PNG(url))
        image_rows.append((fcst_hour, image_row))
    
    tabs = pn.Tabs(*image_rows, align='center')
    player = pn.widgets.Player(end=len(image_rows) - 1, loop_policy='loop', align='center')
    player.jslink(tabs, value='active', bidirectional=True)

    return pn.Column(tabs, player)    
    
@pn.depends(date, exp)
def fig_probability(date, exp):

    datei = date
    datef = date
    
    image_rows = []
    
    for i in np.arange(0,360,24):
        image_row = pn.Row()
        fcst_hour = f"{i:03d}H"
        datef += timedelta(hours=24)
        url = url_fmt_probability.format(exp=exp, prod='probability', datei=datei.strftime('%Y%m%d%H'), datef=datef.strftime('%Y%m%d%H'))
        image_row.append(pn.pane.PNG(url))
        image_rows.append((fcst_hour, image_row))
    
    tabs = pn.Tabs(*image_rows, align='center')
    player = pn.widgets.Player(end=len(image_rows) - 1, loop_policy='loop', align='center')
    player.jslink(tabs, value='active', bidirectional=True)

    return pn.Column(tabs, player)      
    
@pn.depends(date, exp, var_probagr)
def fig_probagr(date, exp, var_probagr):

    datei = date
    
    if var_probagr == 'semana 1': num = '5'
    if var_probagr == 'semana 2': num = '10'
    if var_probagr == 'semana 3': num = '15'
    
    image_row = pn.Row()
    url = url_fmt_probagr.format(exp=exp, prod='probagr', var=var_probagr, datei=datei.strftime('%Y%m%d%H'), num=num)
    image_row.append(pn.pane.PNG(url))

    return pn.Column(image_row)      

@pn.depends(date, exp, var_perturbations)
def fig_perturbations(date, exp, var_perturbations):

    datei = date
    
    image_row = pn.Row()
    url = url_fmt_perturbations.format(exp=exp, var=var_perturbations, prod='perturbations', datei=datei.strftime('%Y%m%d%H'))
    image_row.append(pn.pane.PNG(url))

    return pn.Column(image_row)   

@pn.depends(date, exp)
def fig_chievol(date, exp):

    datei = date
    datef = date + timedelta(hours=360)
    
    image_row = pn.Row()
    url = url_fmt_chievol.format(exp=exp, prod='chievol', datei=datei.strftime('%Y%m%d%H'), datef=datef.strftime('%Y%m%d%H'))
    image_row.append(pn.pane.PNG(url))

    return pn.Column(image_row) 

prod_tabs = pn.Tabs(
    ('Spread', pn.Row(pn.Column(var_spread, width=300), pn.Spacer(width=100), fig_spread)), 
    ('Spaguetti', pn.Row(pn.Column(var_spaguetti, reg_spaguetti, width=300), pn.Spacer(width=100), fig_spaguetti)), 
    ('Cluster', pn.Row(pn.Column(var_cluster, width=300), pn.Spacer(width=100), fig_cluster)),
    ('Probability', pn.Row(fig_probability)),
    ('Probagr', pn.Row(pn.Column(var_probagr, width=300), pn.Spacer(width=100), fig_probagr)),
    ('Perturbations', pn.Row(pn.Column(var_perturbations, width=300), pn.Spacer(width=100), fig_perturbations)),
    ('Chievol', pn.Row(fig_chievol)),
    align='center')
    
card_parameters = pn.Card('**Geral**', exp, date, 
                          title='Parâmetros', collapsed=False)
    
pn.template.FastListTemplate(
    site="oensMB09", title="Produtos", sidebar=[card_parameters],
    main=[prod_tabs], 
).show();    


# In[88]:


exps = ['gnu_singularity_m128p_p64p', 'gnu_egeon_m128p_p64p', 'intel_egeon_m128p_p64p'] 
prods = ['spread', 'cluster']
dates = pd.date_range(start='2020-12-01 00:00', end='2020-12-15 00:00', freq='1D')
products = ['spread', 'spaguetti', 'cluster', 'probability', 'probagr', 'perturbations', 'chievol']

vars_spread = ['psnm', 'geop250', 'geop500', 'temp850']
vars_spaguetti = ['geop500', 'temp850', 'temp925', 'temp1000']
vars_cluster = ['precpsnm1000', 'precvento250', 'precvento925', 'temp700', 'temp850', 'temp925', 'temp1000', 'tems1000', 'zgeo200', 'zgeo500']
vars_probagr = ['semana 1', 'semana 2', 'semana 3']
vars_perturbations = ['temp250', 'temp500', 'temp850', 'uvel250', 'uvel500', 'uvel850', 'vvel250', 'vvel500', 'vvel850']

regs_spaguetti = ['gl', 'as']

prod = pn.widgets.Select(name='Produto', options=products)
date = pn.widgets.Select(name='Data', options=list(dates))
exp = pn.widgets.Select(name='Experimento', options=list(exps))
var_spread = pn.widgets.Select(name='Variável', options=list(vars_spread))
var_spaguetti = pn.widgets.Select(name='Variável', options=list(vars_spaguetti))
var_cluster = pn.widgets.Select(name='Variável', options=list(vars_cluster))
var_probagr = pn.widgets.Select(name='Variável', options=list(vars_probagr))
var_perturbations = pn.widgets.Select(name='Variável', options=list(vars_perturbations))

reg_spaguetti = pn.widgets.Select(name='Região', options=list(regs_spaguetti))

url_fmt_spread = ('http://ftp1.cptec.inpe.br/pesquisa/das/carlos.bastarz/oensMB09/exps/{exp}/prod/{prod}/{datei}/{var}{datei}{datef}.png')
url_fmt_spaguetti = ('http://ftp1.cptec.inpe.br/pesquisa/das/carlos.bastarz/oensMB09/exps/{exp}/prod/{prod}/{datei}/spt{reg}{var}{datei}{datef}.png')
url_fmt_cluster = ('http://ftp1.cptec.inpe.br/pesquisa/das/carlos.bastarz/oensMB09/exps/{exp}/prod/{prod}/{datei}/{prod}{var}{datei}{datef}.png')
url_fmt_probability = ('http://ftp1.cptec.inpe.br/pesquisa/das/carlos.bastarz/oensMB09/exps/{exp}/prod/{prod}/{datei}/prec{datei}{datef}.png')
url_fmt_probagr = ('http://ftp1.cptec.inpe.br/pesquisa/das/carlos.bastarz/oensMB09/exps/{exp}/prod/{prod}/{datei}/prec_agric_large{num}.png')
url_fmt_perturbations = ('http://ftp1.cptec.inpe.br/pesquisa/das/carlos.bastarz/oensMB09/exps/{exp}/prod/{prod}/{datei}/perturbations{var}_{datei}{datei}.png')
url_fmt_chievol = ('http://ftp1.cptec.inpe.br/pesquisa/das/carlos.bastarz/oensMB09/exps/{exp}/prod/{prod}/{datei}/chi_evol{datei}{datef}.png')

@pn.depends(date, exp, var_spread)
def fig_spread(date, exp, var_spread):

    datei = date
    datef = date
    
    image_rows = []
    
    #for i in np.arange(0,120,6):
    for i in np.arange(0,360,24):
        image_row = pn.Row()
        fcst_hour = f"{i:03d}H"
        #datef += timedelta(hours=6)
        datef += timedelta(hours=24)
        url = url_fmt_spread.format(exp=exp, var=var_spread, prod='spread', datei=datei.strftime('%Y%m%d%H'), datef=datef.strftime('%Y%m%d%H'))
        image_row.append(pn.pane.PNG(url))
        image_rows.append((fcst_hour, image_row))
    
    tabs = pn.Tabs(*image_rows, align='center')
    player = pn.widgets.Player(end=len(image_rows) - 1, loop_policy='loop', align='center')
    player.jslink(tabs, value='active', bidirectional=True)

    return pn.Column(tabs, player)
      
@pn.depends(date, exp, var_spaguetti, reg_spaguetti)
def fig_spaguetti(date, exp, var_spaguetti, reg_spaguetti):

    datei = date
    datef = date
    
    image_rows = []
    
    for i in np.arange(0,360,24):
        image_row = pn.Row()
        fcst_hour = f"{i:03d}H"
        datef += timedelta(hours=24)
        url = url_fmt_spaguetti.format(exp=exp, var=var_spaguetti, reg=reg_spaguetti, prod='spaguetti', datei=datei.strftime('%Y%m%d%H'), datef=datef.strftime('%Y%m%d%H'))
        image_row.append(pn.pane.PNG(url))
        image_rows.append((fcst_hour, image_row))
    
    tabs = pn.Tabs(*image_rows, align='center')
    player = pn.widgets.Player(end=len(image_rows) - 1, loop_policy='loop', align='center')
    player.jslink(tabs, value='active', bidirectional=True)

    return pn.Column(tabs, player)
    
@pn.depends(date, exp, var_cluster)
def fig_cluster(date, exp, var_cluster):

    datei = date
    datef = date
    
    image_rows = []
    
    for i in np.arange(0,360,24):
        image_row = pn.Row()
        fcst_hour = f"{i:03d}H"
        datef += timedelta(hours=24)
        url = url_fmt_cluster.format(exp=exp, var=var_cluster, prod='cluster', datei=datei.strftime('%Y%m%d%H'), datef=datef.strftime('%Y%m%d%H'))
        image_row.append(pn.pane.PNG(url))
        image_rows.append((fcst_hour, image_row))
    
    tabs = pn.Tabs(*image_rows, align='center')
    player = pn.widgets.Player(end=len(image_rows) - 1, loop_policy='loop', align='center')
    player.jslink(tabs, value='active', bidirectional=True)

    return pn.Column(tabs, player)    
    
@pn.depends(date, exp)
def fig_probability(date, exp):

    datei = date
    datef = date
    
    image_rows = []
    
    for i in np.arange(0,360,24):
        image_row = pn.Row()
        fcst_hour = f"{i:03d}H"
        datef += timedelta(hours=24)
        url = url_fmt_probability.format(exp=exp, prod='probability', datei=datei.strftime('%Y%m%d%H'), datef=datef.strftime('%Y%m%d%H'))
        image_row.append(pn.pane.PNG(url))
        image_rows.append((fcst_hour, image_row))
    
    tabs = pn.Tabs(*image_rows, align='center')
    player = pn.widgets.Player(end=len(image_rows) - 1, loop_policy='loop', align='center')
    player.jslink(tabs, value='active', bidirectional=True)

    return pn.Column(tabs, player)      
    
@pn.depends(date, exp, var_probagr)
def fig_probagr(date, exp, var_probagr):

    datei = date
    
    if var_probagr == 'semana 1': num = '5'
    if var_probagr == 'semana 2': num = '10'
    if var_probagr == 'semana 3': num = '15'
    
    image_row = pn.Row()
    url = url_fmt_probagr.format(exp=exp, prod='probagr', var=var_probagr, datei=datei.strftime('%Y%m%d%H'), num=num)
    image_row.append(pn.pane.PNG(url))

    return pn.Column(image_row)      

@pn.depends(date, exp, var_perturbations)
def fig_perturbations(date, exp, var_perturbations):

    datei = date
    
    image_row = pn.Row()
    url = url_fmt_perturbations.format(exp=exp, var=var_perturbations, prod='perturbations', datei=datei.strftime('%Y%m%d%H'))
    image_row.append(pn.pane.PNG(url))

    return pn.Column(image_row)   

@pn.depends(date, exp)
def fig_chievol(date, exp):

    datei = date
    datef = date + timedelta(hours=360)
    
    image_row = pn.Row()
    url = url_fmt_chievol.format(exp=exp, prod='chievol', datei=datei.strftime('%Y%m%d%H'), datef=datef.strftime('%Y%m%d%H'))
    image_row.append(pn.pane.PNG(url))

    return pn.Column(image_row) 

#prod_tabs = pn.Tabs(
#    ('Spread', pn.Row(pn.Column(var_spread, width=300), pn.Spacer(width=100), fig_spread)), 
#    ('Spaguetti', pn.Row(pn.Column(var_spaguetti, reg_spaguetti, width=300), pn.Spacer(width=100), fig_spaguetti)), 
#    ('Cluster', pn.Row(pn.Column(var_cluster, width=300), pn.Spacer(width=100), fig_cluster)),
#    ('Probability', pn.Row(fig_probability)),
#    ('Probagr', pn.Row(pn.Column(var_probagr, width=300), pn.Spacer(width=100), fig_probagr)),
#    ('Perturbations', pn.Row(pn.Column(var_perturbations, width=300), pn.Spacer(width=100), fig_perturbations)),
#    ('Chievol', pn.Row(fig_chievol)),
#    align='center')
 
@pn.depends(prod)    
def get_prod(prod):
    if prod == 'spread': return pn.Row(pn.Row(var_spread, width=300), fig_spread)
    if prod == 'spaguetti': return pn.Row(pn.Column(var_spaguetti, reg_spaguetti, width=300), fig_spaguetti)
    if prod == 'cluster': return pn.Row(pn.Row(var_cluster, width=300), fig_cluster)
    if prod == 'probability': return pn.Row(fig_probability)
    if prod == 'probagr': return pn.Row(pn.Row(var_probagr, width=300), fig_probagr)
    if prod == 'perturbations': return pn.Row(pn.Row(var_perturbations, width=300), fig_perturbations)
    if prod == 'chievol': return pn.Row(fig_chievol)
      
card_parameters = pn.Card('**Geral**', exp, date, prod, title='Parâmetros', collapsed=False)
    
#pn.template.FastListTemplate(
#    site="oensMB09", title="Produtos", sidebar=[card_parameters],
#    main=[get_prod], 
#).show();    

pn.Row(pn.Row(pn.Column(exp, date, prod), width=300), get_prod, width=1500).servable();


# In[ ]: