; ***********************************************

 ; using gsn_table for all

 ; output: line plot for each site (4 fields)

 ;         table for M_score

 ; ***********************************************

 load "$NCARG_ROOT/lib/ncarg/nclscripts/csm/gsn_code.ncl.test"

 load "$NCARG_ROOT/lib/ncarg/nclscripts/csm/gsn_csm.ncl.test"

 load "$NCARG_ROOT/lib/ncarg/nclscripts/csm/contributed.ncl"

 ;load "/fis/cgd/cseg/people/jeff/clamp/co2/metrics_table.ncl"

 ;************************************************

 begin

   plot_type     = "ps"

   plot_type_new = "png"

 ;************************************************

 ; read model data

 ;************************************************

 ; model_name = "06cn"

 ; film = "i01.06cn_1798-2004_MONS_climo.nc"

 ; model_name = "06casa"

 ; film = "i01.06casa_1798-2004_MONS_climo.nc"

 ; model_name = "10cn"

 ; film = "i01.10cn_1948-2004_MONS_climo.nc"

   model_name = "10casa"

   film = "i01.10casa_1948-2004_MONS_climo.nc"

 ;--------------------------------------------

   dirm = "/fis/cgd/cseg/people/jeff/clamp_data/model/"

   fm    = addfile(dirm+film,"r")

   xm    = fm->lon

   ym    = fm->lat

   nlat = dimsizes(ym)

   nlon = dimsizes(xm)

   nmon      = 12

   nfield    = 4

   data_mod0 = new ((/nfield,nmon,nlat,nlon/),float)

 ; change to unit of observed (u mol/m2/s)

 ; Model_units [=] gC/m2/s

 ; 12. = molecular weight of C

 ; u mol = 1e-6 mol 

   data = fm->NEE

   factor = 1e6 /12.

   data_mod0(0,:,:,:) = data(:,:,:) * factor

   delete (data)

   data = fm->NETRAD

   data_mod0(1,:,:,:) = data(:,:,:) 

   delete (data)

   data = fm->LATENT

   data_mod0(2,:,:,:) = data(:,:,:) 

   delete (data)

   data = fm->SENSIBLE

   data_mod0(3,:,:,:) = data(:,:,:) 

   delete (data)

 ; printVarSummary (data_mod0)

 ;************************************************

 ; read in data: observed

 ;************************************************

  station = (/"BOREAS_NSA_OBS" \

             ,"CastelPorziano" \

             ,"Hyytiala" \

             ,"Kaamanen" \

             ,"LBA_Tapajos_KM67" \

             ,"Lethbridge" \

             ,"Tharandt" \

             ,"Vielsalm" \

             /)

  year_ob = (/"1994-2004" \

             ,"1997-2003" \

             ,"1996-2003" \

             ,"2000-2003" \

             ,"2002-2005" \

             ,"1999-2004" \

             ,"1996-2003" \

             ,"1998-2003" \

             /)

  field   = (/"CO2 Flux" \

             ,"Net Radiation" \

             ,"Latent Heat" \

             ,"Sensible Heat" \

             /)

  nstation  = dimsizes(station)

  nmon      = 12

  nfield    = dimsizes(field)

  data_ob   = new ((/nstation, nfield, nmon/),float)

  lat_ob    = new ((/nstation/),float)

  lon_ob    = new ((/nstation/),float)

  diri_root  = "/fis/cgd/cseg/people/jeff/clamp_data/fluxnet/"

  do n = 0,nstation-1

     diri = diri_root + station(n)+"/"

     fili = station(n)+"_"+year_ob(n)+"_monthly.nc"

     g     = addfile (diri+fili,"r")

     lon_ob(n) = g->lon 

     lat_ob(n) = g->lat

     data      = g->CO2_FLUX

     data_ob(n,0,:) = dim_avg(data(month|:,year|:))

     delete (data)

     data      = g->RAD_FLUX

     data_ob(n,1,:) = dim_avg(data(month|:,year|:))

     delete (data)

     data      = g->LH_FLUX

     data_ob(n,2,:) = dim_avg(data(month|:,year|:))

     delete (data)

     data      = g->SH_FLUX

     data_ob(n,3,:) = dim_avg(data(month|:,year|:))

     delete (data)

     delete (g)

  end do

 ;print (lat_ob)

 ;print (lon_ob)

 ;printVarSummary (data_ob)

 ;************************************************************

 ; interpolate model data into observed station

 ; note: model is 0-360E, 90S-90N

 ;************************************************************

 ; to be able to handle observation at (-89.98,-24.80)

 ; print (ym(0))

   ym(0) = -90.  

   yy = linint2_points_Wrap(xm,ym,data_mod0,True,lon_ob,lat_ob,0)

   delete (data_mod0)

   yy!0 = "field"

   data_mod = yy(pts|:,field|:,time|:)

 ; printVarSummary (data_mod)

 ;************************************************************

 ; compute correlation coef and M score

 ;************************************************************

  score_max = 5.

  ccr     = new ((/nstation, nfield/),float)

  M_score = new ((/nstation, nfield/),float) 

  do n=0,nstation-1

  do m=0,nfield-1   

     ccr(n,m) = esccr(data_ob(n,m,:),data_mod(n,m,:),0)

     bias = sum(abs(data_mod(n,m,:)-data_ob(n,m,:))/(abs(data_mod(n,m,:))+abs(data_ob(n,m,:))))

     M_score(n,m) = (1. -(bias/nmon)) * score_max

  end do

  end do

 ;*******************************************************************

 ; for station line plot

 ;*******************************************************************

 ; for x-axis in xyplot

   mon = ispan(1,12,1)

   mon@long_name = "month"

   res                   = True               ; plot mods desired

   res@xyLineThicknesses = (/2.0,2.0/)        ; make 2nd lines thicker

   res@xyLineColors      = (/"blue","red"/)   ; line color (ob,model)

 ;-------------------------------------------------------------------------

 ; Add a boxed legend using the more simple method

   res@pmLegendDisplayMode    = "Always"

 ; res@pmLegendWidthF         = 0.1

   res@pmLegendWidthF         = 0.08

   res@pmLegendHeightF        = 0.06

 ; res@pmLegendOrthogonalPosF = -1.17

 ; res@pmLegendOrthogonalPosF = -1.00  ;(downward)

   res@pmLegendOrthogonalPosF = -0.30  ;(downward)

 ; res@pmLegendParallelPosF   =  0.18

   res@pmLegendParallelPosF   =  0.23  ;(rightward)

 ; res@lgPerimOn             = False

   res@lgLabelFontHeightF     = 0.015

   res@xyExplicitLegendLabels = (/"observed",model_name/)

 ;-------------------------------------------------------------------

 ; for panel plot

   res@gsnFrame     = False                   ; Do not draw plot 

   res@gsnDraw      = False                   ; Do not advance frame

   pres                            = True     ; panel plot mods desired

   pres@gsnPanelYWhiteSpacePercent = 5        ; increase white space around

                                              ; indiv. plots in panel

   pres@gsnMaximize                = True     ; fill the page

 ;-------------------------------------------------------------------

   plot_data   = new((/2,12/),float)

   plot_data!0 = "case"

   plot_data!1 = "month"

   do n = 0,nstation-1

 ;----------------------------

 ; for observed

     plot_name = station(n)+"_ob"    

     title = station(n)+"("+sprintf("%5.2f",lat_ob(n))+","+sprintf("%5.2f",lon_ob(n))+")"

     res@tiMainString = title

     wks = gsn_open_wks (plot_type,plot_name)

     plot=new(4,graphic)                        ; create graphic array   

     plot_data(0,:) = (/data_ob (n,0,:)/)

     plot_data@long_name = field(0)   

     plot(0)=gsn_csm_xy(wks,mon,plot_data(0,:),res)   ; create plot 1

     plot_data(0,:) = (/data_ob (n,1,:)/)

     plot_data@long_name = field(1)

     plot(1)=gsn_csm_xy(wks,mon,plot_data(0,:),res)   ; create plot 2

     plot_data(0,:) = (/data_ob (n,2,:)/)

     plot_data@long_name = field(2)   

     plot(2)=gsn_csm_xy(wks,mon,plot_data(0,:),res)   ; create plot 3

     plot_data(0,:) = (/data_ob (n,3,:)/)

     plot_data@long_name = field(3)

     plot(3)=gsn_csm_xy(wks,mon,plot_data(0,:),res)   ; create plot 4

     gsn_panel(wks,plot,(/2,2/),pres)                 ; create panel plot

     system("convert "+plot_name+"."+plot_type+" "+plot_name+"."+plot_type_new)

     system("rm "+plot_name+"."+plot_type)

     clear (wks)  

     delete (plot)

 ;----------------------------

 ; for model_vs_ob

     plot_name = station(n)+"_model_vs_ob"

     title = station(n)+"("+sprintf("%5.2f",lat_ob(n))+","+sprintf("%5.2f",lon_ob(n))+")"    

     res@tiMainString = title

     wks = gsn_open_wks (plot_type,plot_name)

     plot=new(4,graphic)                        ; create graphic array   

     plot_data(0,:) = (/data_ob (n,0,:)/)

     plot_data(1,:) = (/data_mod(n,0,:)/)

     plot_data@long_name = field(0)   

     plot(0)=gsn_csm_xy(wks,mon,plot_data,res)   ; create plot 1

     plot_data(0,:) = (/data_ob (n,1,:)/)

     plot_data(1,:) = (/data_mod(n,1,:)/)

     plot_data@long_name = field(1)

     plot(1)=gsn_csm_xy(wks,mon,plot_data,res)   ; create plot 2

     plot_data(0,:) = (/data_ob (n,2,:)/)

     plot_data(1,:) = (/data_mod(n,2,:)/)

     plot_data@long_name = field(2)   

     plot(2)=gsn_csm_xy(wks,mon,plot_data,res)   ; create plot 3

     plot_data(0,:) = (/data_ob (n,3,:)/)

     plot_data(1,:) = (/data_mod(n,3,:)/)

     plot_data@long_name = field(3)

     plot(3)=gsn_csm_xy(wks,mon,plot_data,res)   ; create plot 4

     gsn_panel(wks,plot,(/2,2/),pres)                 ; create panel plot

     system("convert "+plot_name+"."+plot_type+" "+plot_name+"."+plot_type_new)

     system("rm "+plot_name+"."+plot_type)

     clear (wks)  

     delete (plot)

  end do

 ;*******************************************************************

 ; for table of site score

 ;*******************************************************************

   table_length = 0.8 

 ; table header name

   table_header_name = "Site" 

 ; column (not including header column)

   col_header = (/"Latitude","Longitude","observed" \

                 ,"CO2 Flux","Net Radiation","Latent Heat" \

                 ,"Sensible Heat","Average" \

                 /)

   ncol = dimsizes(col_header) 

 ; row (not including header row) 

   nrow       = nstation + 1

   row_header = new ((/nrow/),string )

   row_header(0:nstation-1) = station(:)

   row_header(nrow-1)       = "All Sites" 

 ; Table header

   ncr1  = (/1,1/)               ; 1 row, 1 column

   x1    = (/0.005,0.15/)        ; Start and end X

   y1    = (/0.800,0.895/)       ; Start and end Y

   text1 = table_header_name

   res1               = True

   res1@txFontHeightF = 0.03

   res1@gsFillColor   = "CornFlowerBlue"

 ; Column header (equally space in x2)

   ncr2  = (/1,ncol/)            ; 1 rows, ncol columns

   x2    = (/x1(1),0.995/)       ; start from end of x1

   y2    = y1                    ; same as y1

   text2 = col_header

   res2               = True

   res2@txFontHeightF = 0.010

   res2@gsFillColor   = "Gray"

 ; Row header (equally space in y2)

   ncr3  = (/nrow,1/)                 ; nrow rows, 1 columns

   x3    = x1                         ; same as x1

   y3    = (/1.0-table_length,y1(0)/) ; end at start of y1

   text3 = row_header

   res3               = True

   res3@txFontHeightF = 0.010

   res3@gsFillColor   = "Gray"

 ; Main table body

   ncr4  = (/nrow,ncol/)           ; nrow rows, ncol columns

   x4    = x2                      ; Start and end x

   y4    = y3                      ; Start and end Y

   text4 = new((/nrow,ncol/),string)

   color_fill4           = new((/nrow,ncol/),string)

   color_fill4           = "white"

   color_fill4(:,ncol-1) = "grey"

   color_fill4(nrow-1,:) = "green"

   res4               = True       ; Set up resource list

 ; res4@gsnDebug      = True       ; Useful to print NDC row,col values used.

   res4@txFontHeightF = 0.015

   res4@gsFillColor   = color_fill4

   delete (color_fill4)

 ;-------------------------------------------------------------------

 ; for table value

   do n = 0,nrow-2

      text4(n,0) = sprintf("%5.2f", lat_ob(n))  

      text4(n,1) = sprintf("%5.2f", lon_ob(n))  

      text4(n,2) = year_ob(n)  

      text4(n,3) = sprintf("%5.2f", M_score(n,0))      ; CO2 Flux

      text4(n,4) = sprintf("%5.2f", M_score(n,1))      ; Net Radiation

      text4(n,5) = sprintf("%5.2f", M_score(n,2))      ; Latent Heat

      text4(n,6) = sprintf("%5.2f", M_score(n,3))      ; Sensible Heat

      text4(n,7) = sprintf("%5.2f", avg(M_score(n,:))) ; avg all fields    

   end do

 ;    for the last row

      M_co2 = avg(M_score(:,0))

      M_rad = avg(M_score(:,1))

      M_lh  = avg(M_score(:,2))

      M_sh  = avg(M_score(:,3))

      M_all = M_co2+ M_rad +M_lh + M_sh

      text4(nrow-1,0) = "-"

      text4(nrow-1,1) = "-"

      text4(nrow-1,2) = "-"

      text4(nrow-1,3) = sprintf("%5.2f", M_co2) ; avg all sites

      text4(nrow-1,4) = sprintf("%5.2f", M_rad) ; avg all sites

      text4(nrow-1,5) = sprintf("%5.2f", M_lh ) ; avg all sites

      text4(nrow-1,6) = sprintf("%5.2f", M_sh ) ; avg all sites

      text4(nrow-1,7) = sprintf("%5.2f", M_all) ; avg all sites

      print (M_co2)

      print (M_rad)

      print (M_lh )

      print (M_sh)

      print (M_all) 

 ;---------------------------------------------------------------------------

   plot_name = "table_site_score"

   wks = gsn_open_wks (plot_type,plot_name)

 ;------------------------------------------

 ; for table title

   gRes               = True

   gRes@txFontHeightF = 0.02

 ; gRes@txAngleF      = 90

   title_text = "Model  " + model_name +  "  M_Score"

   gsn_text_ndc(wks,title_text,0.50,0.95,gRes)

 ;------------------------------------------   

   gsn_table(wks,ncr1,x1,y1,text1,res1)

   gsn_table(wks,ncr2,x2,y2,text2,res2)

   gsn_table(wks,ncr3,x3,y3,text3,res3)

   gsn_table(wks,ncr4,x4,y4,text4,res4) 

   frame(wks)

   system("convert "+plot_name+"."+plot_type+" "+plot_name+"."+plot_type_new)

   system("rm "+plot_name+"."+plot_type)

 ;-------------------------------------------------------------------

   temp_name = "temp." + model_name

   system("mkdir -p " + temp_name)

   system("mv *.png " + temp_name)

   system("tar cf "+ temp_name +".tar " + temp_name)

 ;------------------------------------------------------------------- 

 end