Process RDI data using COADS

adcp.py

#I have /Users/jklymak/codas/programs/: in my PYTHONPATH
from pycurrents.adcp.rdiraw import Multiread
# of course you need an ENX file from vmdas!
m = Multiread('March10-2015-03/ADCP029*.ENX','wh')
# ENX should already be in earth co-ordinates, but you may have to adjust the heading as below:
data = m.read()
from pycurrents.adcp.transform import heading_rotate
from pycurrents.adcp.transform import rdi_xyz_enu

dhead = -44.
data['earth']=heading_rotate(data['vel'],dhead)
data['btearth']=heading_rotate(data['bt_vel']*1.,dhead)
data['earth']=data['earth']-data['btearth'][:,newaxis,:]
### Plot
fig,axs=plt.subplots(1,2,sharex=True,sharey=True,figsize=(10,4))
for nn in range(2):
    pcm=axs.flatten()[nn].pcolormesh(data['dday'],data['dep'],  data['earth'][:,:,nn].T,rasterized=True,cmap=cm.RdBu_r)
    pcm.set_clim(np.array([-1.,1.])*0.6)
    axs[nn].plot(data['dday'],data['bt_depth'],linewidth=2,color='k')
axs[0].set_ylim([100,0])