source: branches/2016/dev_merge_2016/NEMOGCM/CONFIG/WAD_TEST_CASES/EXP00/matpoly2.py @ 7514

import numpy as np
import netCDF4

import matplotlib.pyplot as plt
import matplotlib
from matplotlib.patches import Polygon
from matplotlib.collections import PatchCollection
from netCDF4 import Dataset
import sys

print sys.argv[1]
pathin =  sys.argv[1]

nx = 34 
ny = 1
nt = 24
print nx
print ny


fw = Dataset(pathin)
ssh = fw.variables['sossheig'][:,:,:]
bat = fw.variables['gdepw_0'][0,0,:,:]
#bat = fw.variables['ht_wd'][0,:,:]
#vot = fw.variables['votemper'][:,0,:,:]
vot = fw.variables['sosaline'][:,:,:]
print ssh.shape
print "bat"
print bat.shape 
print "vot"
print vot.shape 
nt = ssh.shape[0]
nx = ssh.shape[2]
print nx,nt
fw.close()

bat = -1.*bat
batmin = np.amin(bat)
batmax = np.amax(bat)
brange = batmax - batmin
tol = 0.1*brange
print batmin,batmax,' ho'

#for t in range(300,313):

for t in range(0,nt,3):
 wadfr = "wadfr{:0>4d}.png".format(t)
 t24  = np.mod(t,24)
 dy   = np.int(t/24.0)

 tfac = 18.0/3600.0
 t24  = np.int(np.mod(t*tfac,24))
 dy   = np.int(t*tfac/24.0)
 mn   = np.int(np.rint((np.mod(t*tfac,24) - t24 )*60))
 hour  = "t={:0>2d}:{:0>2d}:{:0>2d} ".format(dy,t24,mn)
 hour2  = "  (days:hrs:mins)"
 batpts = np.zeros((nx+4,2))
 sshpts = np.zeros((2*nx,2))
 votpts = np.zeros((nx,2))
 #codes = np.ones(nx,int)
 #codes[0] = path.Path.MOVETO
 #codes[1:] = path.Path.LINETO
 for pt in range(nx):
    batpts[pt+2,0] = pt
    batpts[pt+2,1] = bat[0,pt]
    sshpts[pt,0] = pt
    sshpts[pt,1] = ssh[t,0,pt]
    votpts[pt,0] = pt
    votpts[pt,1] = np.minimum(36.,vot[t,0,pt])
    votpts[pt,1] = np.maximum(30.0,votpts[pt,1])
    votpts[pt,1] = batmin +0.2*brange + (votpts[pt,1]-30.)*brange/6.0
 batpts[nx+1,1] = batmax
 batpts[nx+2,0] = nx-1
 batpts[nx+2,1] = batmax
 batpts[nx+3,0] = nx-1
 batpts[nx+3,1] = batmin
 batpts[0,0] = 0.0
 batpts[0,1] = batmin
 batpts[1,0] = 0.0
 batpts[1,1] = batmax
 batpts[2,1] = batmax
 sshpts[nx-1,0] = nx-1
 sshpts[nx-1,1] = sshpts[nx-2,1]
 sshpts[0,0] = 0.0
 #sshpts[0,1] = batmin
 votpts[nx-2,0] = nx-1
 votpts[nx-1,0] = nx-1
 votpts[0,0] = 0.0
 votpts[nx-1,1] = batmax + tol
 votpts[0,1] = batmax + tol
 sshpts[0,1] = sshpts[1,1]
 #for pt in range(nx):
 #  print pt,votpts[pt,0],votpts[pt,1]
 for pt in range(nx):
    sshpts[pt+nx,0]=batpts[nx+1-pt,0]
    sshpts[pt+nx,1]=batpts[nx+1-pt,1]
 

 xs, ys = zip(*votpts)
 fig, ax = plt.subplots()
 patches = []

 polygon = Polygon(batpts, True)
 patches.append(polygon)
 polygon2 = Polygon(sshpts, True)
 patches.append(polygon2)

 p = PatchCollection(patches, cmap=matplotlib.cm.jet, alpha=0.4)
 
 #colors = 100*np.random.rand(len(patches))
 colors = (4,3)
 p.set_array(np.array(colors))

 ax.set_ylim([-10., 4.0])
 ax.set_xlim([0., 50.0])
 ax.add_collection(p)
 ax.plot(xs,ys, '--', color='black', ms=10)

 plt.annotate(hour,xy=(2,batmin+0.1*brange))
 plt.annotate(hour2,xy=(2,batmin+0.05*brange))
 plt.savefig(wadfr)