;;Usage: plot one day from ncep and jra, to compare the result
;;Author: heyl 20161130


fu=addfile("uwnd.2008.nc","r")
fv=addfile("vwnd.2008.nc","r")
ft=addfile("air.2008.nc","r")

index=day_of_year(2008,8,1)
u=fu->uwnd(index,:,:,:)
v=fv->vwnd(index,:,:,:)
t=ft->air(index,:,:,:)
printMinMax(u,0)
printMinMax(v,0)
printMinMax(t,0)


lat=u&lat(0:24)
lev=u&level(0:11)
sf=5.0 ;;tofloat(getenv("sc_fact_strat"))
print(lat)
print(lev)


opt=True
;opt@magf=sf

epf=epflux(u(0:11,0:24,:),v(0:11,0:24,:),t(0:11,0:24,:),lev,lat,opt)
;printVarSummary(epf)
Fy=epf[0]
Fz=epf[1]
epdiv=epf[2]

printVarSummary(Fy)
printMinMax(Fy,0)
printVarSummary(Fz)
printMinMax(Fz,0)
printVarSummary(epdiv)
printMinMax(epdiv,0)



;;;;;================plot ====================
wks=gsn_open_wks("eps","compare_ncep")

res_vec=True
;res_vec@gsnMaximize=True
res_vec@gsnDraw=False
res_vec@gsnFrame=False

res_vec@vfYArray=lev(0:11)
res_vec@vfXArray=lat(0:24)
res_vec@tiXAxisString=""
res_vec@tiYAxisString="Level"
res_vec@trYReverse=True
res_vec@vcRefMagnitudeF=80
res_vec@vcRefLengthF=0.05
res_vec@vcMonoLineArrowColor=True
res_vec@vcRefAnnoOn=True
res_vec@vcRefAnnoOrthogonalPosF=-0.1
;res_vec@vcRefAnnoParallelPosF=
res_vec@vcRefAnnoPerimOn=False
res_vec@vcRefAnnoString1="80m/s"
res_vec@vcRefAnnoFontHeightF=0.015
res_vec@vcRefAnnoString2On=False

;res_vec@pmLabelBarDisplayMode="Always"
;res_vec@pmLabelBarWidthF=0.08
;res_vec@lbPerimOn=False
res_vec@gsnYAxisIrregular2Log=True
res_vec@tiYAxisFontHeightF=0.02
res_vec@tmXBLabelFontHeightF=0.015
res_vec@tmYLLabelFontHeightF=0.015
res_vec@tmYLMode="Explicit"
res_vec@tmYLValues=lev
res_vec@tmYLLabels=" "+toint(lev)
res_vec@tmYLLabels(1)=""  ;;no 925
res_vec@tmYLLabels(2)=""    ;; 850
res_vec@tmYLLabels(4)=""    ;; 600
res_vec@tmYLLabels(8)=""    ;; 250

;res_vec@vpWidthF=0.6
;res_vec@vpHeightF=0.6
;res_vec@vcLevelSelectionMode="ManualLevels"
;res_vec@vcLevelSpacingF=25
;res_vec@vcMinLevelValF=0
;res_vec@vcMaxLevelValF=400
res_vec@vcMinDistanceF=0.00875

;;;;contour for epdiv
res_con=True
res_con@gsnDraw=False
res_con@gsnFrame=False
res_con@sfYArray=lev(0:11)
res_con@sfXArray=lat(0:24)
res_con@trYReverse=True
res_con@gsnYAxisIrregular2Log=True
res_con@cnFillOn=True
res_con@cnLinesOn=False
res_con@cnLineLabelsOn=False
res_con@cnInfoLabelOn=False
res_con@cnSmoothingOn=True
res_con@tiMainString=""
res_con@tmYLMode="Explicit"
res_con@tmYLValues=lev
res_con@tmYLLabels=" "+toint(lev)
res_con@tmYLLabels(1)=""
res_con@tmYLLabels(2)=""
res_con@tmYLLabels(4)=""
res_con@tmYLLabels(8)=""
res_con@tmXBMode="Explicit"
res_con@tmXBValues=ispan(30,90,10)
res_con@tmXBLabels=ispan(30,90,10)+"~S~o~N~N"
res_con@tmXBLabelFontHeightF=0.015
res_con@tmYLLabelFontHeightF=0.015
res_con@tmXBMajorLengthF=-0.0075
res_con@tmYLMajorLengthF=-0.0075
res_con@lbLabelBarOn=True
res_con@cnLevelSelectionMode="ManualLevels"
res_con@cnLevelSpacingF=50
res_con@cnMinLevelValF=-500
res_con@cnMaxLevelValF=500
cold_color=(/10,20,30,33,34,36,38,40,44,48,50/)
warm_color=ispan(51,101,5)
res_con@cnFillColors=array_append_record(cold_color,warm_color,0)
res_con@pmLabelBarHeightF=0.08
res_con@pmLabelBarWidthF=0.5
;res_con@pmLabelBarOrthogonalPosF=0.1 ;;up down
res_con@pmLabelBarParallelPosF=0.4
res_con@lbLabelFontHeightF=0.015


gsn_define_colormap(wks,"BlWhRe")
res_con@gsnLeftString="(a) NCEP 2008/08/01"
plotvec1=gsn_vector(wks,Fy(0:11,0:24),Fz(0:11,0:24),res_vec)
plotcon1=gsn_csm_contour(wks,epdiv(0:11,0:24),res_con)
overlay(plotcon1,plotvec1)

gsn_panel(wks,(/plotcon1/),(/1,1/),False)