%matplotlib inline
from matplotlib import pylab
import matplotlib.pyplot as plt
import numpy as np
import scipy.io
from datetime import datetime, timedelta

data_east = scipy.io.loadmat('RDIADCP150WH17457_20131015T000000.378Z-53E05.mat')

for key in data_east.keys():
    print key

data_east['adcp'].dtype

PPe = data_east['adcp']['pressure'][0][0]
print PP[600]
vele = data_east['adcp']['velocity'][0][0]
uvele = data_east['adcp']['u'][0][0]
vvele = data_east['adcp']['v'][0][0]
erange = data_east['adcp']['range'][0][0]
etime = data_east['adcp']['time'][0][0]
speede = np.sqrt(uvele*uvele+vvele*vvele)
print uvele[100,30], erange.shape, etime.shape
print etime[100][0]
ii = 1800
print datetime.fromordinal(int(etime[ii][0])) + timedelta(days=etime[ii][0]%1) - timedelta(days = 366)

bins = 85
dse = np.zeros((bins,24))
# find median for each hour in each bin
for id in np.arange(bins):
    for j in np.arange(24):
        dse[id,j] = np.median(speede[j*ii:j*ii+ii,id])
print dse.shape
plt.plot(dse[2],'b',dse[12],'r',dse[22],'g',dse[32],'m',dse[42],'k')
plt.title('East Station')
print 170-erange[2],170-erange[12],170-erange[22],170-erange[32],170-erange[42]

data_central = scipy.io.loadmat('RDIADCP150WH8497_20131015T000005.419Z-53E04.mat')

PPc = data_central['adcp']['pressure'][0][0]
print PP[600]
velc = data_central['adcp']['velocity'][0][0]
uvelc = data_central['adcp']['u'][0][0]
vvelc = data_central['adcp']['v'][0][0]
crange = data_central['adcp']['range'][0][0]
ctime = data_central['adcp']['time'][0][0]
speedc = np.sqrt(uvelc*uvelc+vvelc*vvelc)
print uvelc[100,30], crange.shape, ctime.shape
print ctime[100][0]
ii = 600
print datetime.fromordinal(int(ctime[ii][0])) + timedelta(days=ctime[ii][0]%1) - timedelta(days = 366)

bins = 60
dsc = np.zeros((bins,24))
# find median for each hour in each bin
for id in np.arange(bins):
    for j in np.arange(24):
        dsc[id,j] = np.median(speedc[j*ii:j*ii+ii,id])
print dsc.shape
plt.plot(dsc[2],'b',dsc[12],'r',dsc[22],'g',dsc[32],'m',dsc[42],'k')
plt.title('Central Station')
print 300-crange[2],300-crange[12],300-crange[22],300-crange[32],300-crange[42]