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

# This notebook is used to test the research_VENUS.py module.

# ##Test research_VENUS.py

# In[13]:


import matplotlib.pyplot as plt
import netCDF4 as nc
import numpy as np

from salishsea_tools import (
    nc_tools,
    viz_tools,
    tidetools,
)

from salishsea_tools.nowcast import figures
from salishsea_tools.nowcast import research_VENUS as V

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


# In[14]:


filec='/data/dlatorne/MEOPAR/SalishSea/nowcast/20may15/VENUS_central_gridded.nc'
filee='/data/dlatorne/MEOPAR/SalishSea/nowcast/20may15/VENUS_east_gridded.nc'
filet='/data/dlatorne/MEOPAR/SalishSea/nowcast/14may15/SalishSea_1h_20150514_20150514_grid_T.nc'
fileb='/data/dlatorne/MEOPAR/NEMO-forcing/grid/bathy_meter_SalishSea2.nc'

grid_c=nc.Dataset(filec)
grid_e=nc.Dataset(filee)
grid_t=nc.Dataset(filet)
grid_b=nc.Dataset(fileb)


# #Test the functions
# 

# In[15]:


station='Central'
fig = V.plot_vel_NE_gridded(station, grid_c, vmax=0.5)


# In[16]:


station='East'
fig = V.plot_vel_NE_gridded(station, grid_e)


# In[17]:


reload(V)
fig = V.VENUS_location(grid_b)


# In[18]:


reload(V)
fig = V.compare_VENUS('Central',grid_t, grid_b)


# In[ ]: