import komod
import Nio
from IPython.display import Image

%pylab inline

ff =Nio.open_file('MIT_output_2d.nc')
area = ff.variables['AREAtave'][:]
lat  = ff.variables['latitude'][:]
lon  = ff.variables['longitude'][:]
komod.arctpl(lon, lat, area[0,:,:])

!convert -scale 50% output.ps output.png

Image(filename='output.png')

komod.arctpl(lon, lat, area[0,:,:], region='FramStAnna')

!convert -scale 50% output.ps output.png
Image(filename='output.png')

komod.arctpl(lon, lat, area[0,:,:], region='Global', minLon=10, maxLon=50 , minLat=40 , maxLat=75 )

!convert -scale 50% output.ps output.png
Image(filename='output.png')

!wget ftp://ftp.cdc.noaa.gov/Datasets/ncep.reanalysis.dailyavgs/surface/air.sig995.2012.nc
!cdo seasmean air.sig995.2012.nc air.sig995.2012_sm.nc

ncep = Nio.open_file('air.sig995.2012_sm.nc', 'r')
lat_ncep = ncep.variables['lat'][:]
lon_ncep = ncep.variables['lon'][:]
ncep_air = ncep.variables['air'][:]
lon_ncep2, lat_ncep2 = np.meshgrid(lon_ncep, lat_ncep)

komod.arctpl(lon_ncep2, lat_ncep2, ((ncep_air[:,:,:]*0.01)+512.81)-273.15, add_cyclic=True)

Image(filename='5plots.png')

komod.arctpltnc('MIT_output_2d.nc', 'AREAtave', lon='longitude', lat='latitude', showfig=False)

!convert -scale 50% output.ps output.png
Image(filename='output.png')

komod.arctpltnc('MIT_output_3d.nc', 'Ttave', lon='longitude', lat='latitude', llevel='z', levon=True)

Image(filename='many_levels.png')

komod.pltgrd(lon,lat, region = 'FramStAnna')

!convert -scale 50% -rotate -90 grid.ps grid.png
Image(filename='grid.png')

komod.pltgrd(lon,lat, region = 'FramStAnna', every=5)
!convert -scale 50% -rotate -90 grid.ps grid.png
Image(filename='grid.png')

komod.pltgrdnc('air.sig995.2012.nc', 'air')

!convert -scale 50% -rotate -90 grid.ps grid.png
Image(filename='grid.png')

komod.pltgrd_line(lon, lat, lon1=30, lat1=80, lon2=30, lat2=85, every=3, region = 'FramStAnna')  

!convert -scale 50%  grid_line.ps grid_line.png
Image(filename='grid_line.png')

import Nio

f3d = Nio.open_file('MIT_output_3d.nc', 'r')

lat3 = f3d.variables['latitude'][:]
lon3 = f3d.variables['longitude'][:]
temp = f3d.variables['Ttave'][:]
lev  = f3d.variables['z'][:] # depths

data_prof, x_kilometers, m_grid, n_grid = komod.get_transect(lat3, lon3, temp, lon1=30, lat1=80, lon2=30, lat2=85)

komod.plt_transect(distances=x_kilometers , levels=lev, vvalues=data_prof)

!convert -scale 50% -rotate -90 output.ps output.png
Image(filename='output.png')