#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Thu Sep 16 19:57:21 2021

@author: Camaron.George
"""

import numpy as np
import netCDF4 as nc
from tide import Tide
import constituent as con
from datetime import datetime

path = '/scratch2/NCEPDEV/ohd/Camaron.George/'
inFile = 'Scripts/Tide/TideConstEastGulf.nc'
startPred = datetime(2018,9,20,13)#time zero for the prediction
hours = np.arange(60.0)#number of hours in the prediction

predTimes = Tide._times(startPred,hours)

data = nc.Dataset(path+inFile,'r')
amp = data.variables['amp'][:]
pha = data.variables['pha'][:]

wl = np.zeros((len(predTimes),amp.shape[0]))
cons = [c for c in con.noaa if c != con._Z0]
n = [3,34,0,2,5,29,25,1] #corresponds to position in list in constituent.py
cons = [cons[c] for c in n]
model = np.zeros(len(cons),dtype=Tide.dtype)
model['constituent'] = cons
for i in range(amp.shape[0]):
    model['amplitude'] = amp[i,:]
    model['phase'] = pha[i,:]

    tide = Tide(model=model,radians=False)
    wl[:,i] = (tide.at(predTimes))/100.0