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

# # Test `pt_atkinson_tide` Module
# 
# Render figure object produced by the `nowcast.figures.publish.pt_atkinson_tide` module.
# Provides data for visual testing to confirm that refactoring has not adversely changed figure for web page.
# 
# Set-up and function call replicates as nearly as possible what is done in the `nowcast.workers.make_plots` worker.

# Notebooks like this should be developed in a
# [Nowcast Figures Development Environment](https://salishsea-nowcast.readthedocs.io/en/latest/figures/fig_dev_env.html)
# so that all of the necessary dependency packages are installed.
# The development has to be done on a workstation that has the Salish Sea Nowcast system `/results/` parition mounted.

# In[1]:


import io
from pathlib import Path

import arrow
import netCDF4 as nc
import yaml

from nowcast.figures.publish import pt_atkinson_tide


# In[2]:


get_ipython().run_line_magic('matplotlib', 'inline')


# In[3]:


config = '''
    timezone: Canada/Pacific

    ssh:
        tidal_predictions: /SalishSeaCast/tidal-predictions/

    run:
        results_archive:
            forecast: /results/SalishSea/forecast.201812/
            forecast2: /results/SalishSea/forecast2.201812/
'''

config = yaml.safe_load(io.StringIO(config))


# In[4]:


run_date = arrow.get('2020-02-09')
run_type = 'forecast'

dmy = run_date.format('DDMMMYY').lower()
start_day = {
    'nowcast': run_date.format('YYYYMMDD'),
    'forecast': run_date.shift(days=+1).format('YYYYMMDD'),
}
end_day = {
    'nowcast': run_date.format('YYYYMMDD'),
    'forecast': run_date.shift(days=+2).format('YYYYMMDD'),
}

results_home = Path(config['run']['results_archive'][run_type])
results_dir = results_home/dmy


# In[5]:


grid_T_hr = nc.Dataset(
    str(results_dir/'SalishSea_1h_{0}_{1}_grid_T.nc'
        .format(start_day[run_type], end_day[run_type])))
tidal_predictions = config['ssh']['tidal_predictions']


# In[6]:


get_ipython().run_cell_magic('timeit', '-n1 -r1', "\n# Refactored rendering of figure\n\nfrom importlib import reload\nfrom nowcast.figures import website_theme\nreload(pt_atkinson_tide)\nreload(website_theme)\n\nfig = pt_atkinson_tide.make_figure(\n    grid_T_hr, tidal_predictions, config['timezone'], theme=website_theme)\n")


# In[6]: