load "$NCARG_ROOT/lib/ncarg/nclscripts/csm/gsn_code.ncl"
load "$NCARG_ROOT/lib/ncarg/nclscripts/csm/gsn_csm.ncl"
load "$NCARG_ROOT/lib/ncarg/nclscripts/csm/gsn_code.ncl"
load "$NCARG_ROOT/lib/ncarg/nclscripts/wrf/WRFUserARW.ncl"

begin

; for use of high resolution map database, execute
; 'setenv NCARG_RANGS /network/rit/lab/atm419lab/zhou/rangs' (on snow)
; 'setenv NCARG_RANGS /atm419lab/zhou/rangs' (on ash, reed)
; before running the script

c="./cmap_shear.txt"
cmap=asciiread(c,(/72,3/),"float")
cmap=cmap/255

data=addfile("./shear_composite.nc","r")
shear=data->shear
shear!0="case"
shear!1="lat"
shear!2="lon"
printVarSummary(shear)
;print(num(ismissing(shear)))
;print(shear&lev)
;print(shear&lat)

shear&lat=fspan(-10,10,29)
shear&lon=fspan(-10,10,29)
;centlat_ind=ind_nearest_coord(center(case,0),lat,0)
;centlon_ind=ind_nearest_coord(center(case,1),lon,0)
;centlat=lat(centlat_ind)
;centlon=lon(centlon_ind)

dsizes=dimsizes(shear)
;print(dsizes)
dcases=dsizes(0)
dlat=dsizes(1)
dlon=dsizes(2)

shear_avg=new((/dlat,dlon/),"float")

do j=0,dlat-1,1
  do i=0,dlon-1,1
    shear_avg(j,i)=avg(shear(:,j,i))
  end do
end do

copy_VarMeta(shear(0,:,:),shear_avg)

wks=gsn_open_wks("png","composite")
gsn_define_colormap(wks,cmap)

res=True

res@gsnAddCyclic=False
res@gsnFrame=False
res@gsnMaximize=False
res@gsnStringFont=21
res@gsnLeftString="850-200hPa shear (m/s)"
res@gsnLeftStringOrthogonalPosF=0.01
res@gsnLeftStringFontHeightF=0.015
res@gsnRightString="composite of 83 cases"
;res@gsnRightString=name(case)+" @ "+sprinti("%10.0i",yyyymmddhh(case))
;res@gsnRightString="+"+sprinti("%0.2i",counter)+"h"
;res@gsnRightString="+"+sprintf("%4.1f",counter*0.5)+"h"
res@gsnRightStringOrthogonalPosF=0.01
res@gsnRightStringFontHeightF=0.015
;res@gsnScalarContour=True

res@vpXF=0.15
res@vpYF=0.95
res@vpHeightF=0.7
res@vpWidthF=0.7

res@trYMinF=-10
res@trYMaxF=10
res@trXMinF=-10
res@trXMaxF=10

;res@vcRefMagnitudeF=20
;res@vcRefLengthF=0.035
;res@vcGlyphStyle="CurlyVector"
;res@vcLineArrowThicknessF=3.0
;res@vcRefAnnoOrthogonalPosF=-0.079
;res@vcRefAnnoParallelPosF=1.0
;res@vcMinDistanceF=0.03
;res@vcMonoLineArrowColor=True
;res@vcLineArrowColor=0
;res@vcRefAnnoArrowUseVecColor=False
;res@vcRefAnnoArrowLineColor=1

res@cnFillOn=True
res@cnLevelSelectionMode="ExplicitLevels"
res@cnLevels=ispan(0,50,1)
res@cnFillOpacityF=1
res@cnLinesOn=False

res@lbBoxSeparatorLinesOn=False
res@lbLabelStride=10
res@lbLabelFontHeightF=0.015

res@pmLabelBarHeightF=0.1

;res@mpDataResolution="FinestResolution"
;res@mpDataBaseVersion="HighRes"
;res@mpLimitMode="LatLon"
;res@mpCenterLatF=centlat
;res@mpCenterLonF=centlon
;res@mpMinLatF=centlat-10
;res@mpMaxLatF=centlat+10
;res@mpMinLonF=centlon-10
;res@mpMaxLonF=centlon+10
;res@mpMonoFillColor=False
;res@mpFillColors=(/-1,-1,-1/)
;res@mpOutlineOn=True

res@tmXTOn                = False
res@tmYROn                = False

res@tmXBMode              = "Explicit"
res@tmXBValues            = ispan(-8,8,2)
res@tmXBLabels            = res@tmXBValues
;res@tmXBLabels            = (/"100~S~o~N~E","110~S~o~N~","120~S~o~N~E",\
;                              "130~S~o~N~","140~S~o~N~E","150~S~o~N~",\
;                              "160~S~o~N~E","170~S~o~N~","180~S~o~N~E"/)
res@tmXBMajorLengthF      = 0.01
res@tmXBMajorOutwardLengthF=0.0
res@tmXBMajorThicknessF   = 3.0
res@tmXBMinorOn           = False
res@tmXBMinorPerMajor     = 4
res@tmXBMinorLengthF      = 0.004
res@tmXBMinorOutwardLengthF=0.0
res@tmXBMinorThicknessF   = 2.0
res@tmXBLabelFont         = 21
res@tmXBLabelFontHeightF  = 0.015
res@tmXBLabelDeltaF       = -0.6

res@tmYLMode              = "Explicit"
res@tmYLValues            = ispan(-8,8,2)
res@tmYLLabels            = res@tmYLValues
;res@tmYLLabels            = (/"0~S~o~N~","10~S~o~N~N","20~S~o~N~",\
;                              "30~S~o~N~N","40~S~o~N~","50~S~o~N~N"/)
res@tmYLMajorLengthF      = 0.01
res@tmYLMajorOutwardLengthF=0.0
res@tmYLMajorThicknessF   = 3.0
res@tmYLMinorOn           = False
res@tmYLMinorPerMajor     = 4
res@tmYLMinorLengthF      = 0.004
res@tmYLMinorOutwardLengthF=0.0
res@tmYLMinorThicknessF   = 2.0
res@tmYLLabelFont         = 21
res@tmYLLabelFontHeightF  = 0.015
res@tmYLLabelDeltaF       = -0.3

plot=gsn_csm_contour(wks,shear_avg,res)

frame(wks)

end