#! /usr/bin/env python
from    netCDF4               import    Dataset
from	numpy		      import	*
import  numpy                 as        np
import  matplotlib.pyplot     as        mpl
from matplotlib.cm import get_cmap
import pylab
import matplotlib.colors as mcolors
from FV3_utils import *

############################
folder="../"
filename1=folder+"diagfi_mean_A.nc"
var="ch4_gas" #variable
tint=[30,32] #Time must be as written in the input file
xarea="-180,179"

nc1=Dataset(filename1)

lat=nc1.variables["lat"][:]
lon=nc1.variables["lon"][:]
alt=nc1.variables["altitude"][:]
tim=nc1.variables["time_counter"][:]
############################

# def getvar(filename,var,tint,xarea):
#     myvar = pp(file=filename,var=var,t=tint,x=xarea,compute="mean").getf()
#     print((shape(myvar)))
    # return myvar

myvar=getvar(nc1,var,tint,tim)
myvar=myvar*28/16.*100.

mpl.figure(figsize=(20, 10))

font=26

#pal=rvb1 #get_cmap(name="RdYlBu_r")
#pal=get_cmap(name="Spectral_r")
pal=get_cmap(name="rainbow")

xticks=[-90,-60,-30,0,30,60,90]
#yticks=np.linspace(0,240,9)
alt=alt/1000.

mymin=0.1
mymax=4

# log
norm=mcolors.LogNorm()
lvls=np.logspace(np.log10(mymin),np.log10(mymax),16)
#titi=[1.e-14,1.e-13,1.e-12,1.e-11]
CF=mpl.contourf(lat, alt, myvar,levels=lvls,norm=norm,cmap=pal)
cbar=mpl.colorbar(CF, shrink=1, format="%.1f",extend='both')

for t in cbar.ax.get_yticklabels():
      t.set_fontsize(font)


vect=lvls
CS=mpl.contour(lat,alt,myvar,vect,colors='k',linewidths=0.5)
#### inline=1 : values over the line
mpl.clabel(CS, inline=1, fontsize=20, fmt='%1.1f',inline_spacing=1)


#lev=np.logspace(np.log10(0.1),np.log10(4),9)

#mpl.title('Latitude ='+str(tintstr[i]),fontsize=font)
mpl.ylabel('Altitude (km)',labelpad=10,fontsize=font)
mpl.xlabel('Latitude (deg)',labelpad=10, fontsize=font)
mpl.xticks(xticks,fontsize=font)
#mpl.xticks(fontsize=font)
#mpl.yticks(yticks,fontsize=font)
mpl.yticks(fontsize=font)
pylab.ylim([-4,200])

mpl.savefig('meanvmrch4.eps',dpi=200)
mpl.savefig('meanvmrch4.png',dpi=200)
mpl.show()