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

path="/data/jra55/heyl/"
fu=addfile(path+"200808_ugrd.nc","r")
fv=addfile(path+"200808_vgrd.nc","r")
ft=addfile(path+"200808_tmp.nc","r")

u=fu->ugrd(0,:,:,:)
v=fv->vgrd(0,:,:,:)
t=ft->tmp(0,:,:,:)
printMinMax(u,0)
printMinMax(v,0)
printMinMax(t,0)


lat=u&g0_lat_1(0:48)
lev=u&lv_ISBL0(10:)

sf=5.0 ;;tofloat(getenv("sc_fact_strat"))
print(lat)
print(lev)


opt=True
;opt@magf=sf

epf=epflux(u(10:,0:48,:),v(10:,0:48,:),t(10:,0:48,:),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_jra")

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

res_vec@vfYArray=lev
res_vec@vfXArray=lat
res_vec@tiXAxisString=""
res_vec@tiYAxisString="Level"
res_vec@trYReverse=True
res_vec@vcRefMagnitudeF=8000
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
res_con@sfXArray=lat
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=(/1000,700,500,400,300,200,150,100/);lev
res_con@tmYLLabels=(/1000,700,500,400,300,200,150,100/);" "+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=300
res_con@cnMinLevelValF=-3000
res_con@cnMaxLevelValF=3000
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) JRA 2008/08/01"
plotvec1=gsn_vector(wks,Fy,Fz,res_vec)
plotcon1=gsn_csm_contour(wks,epdiv,res_con)
overlay(plotcon1,plotvec1)

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