#
# dcpam の計算結果もしくは NCEP データの月平均・東西平均分布を描画する
#
# * 2011/09/30 ulwrftop の変数の扱いを修正
# * 2011/09/27 viewport の綴りを直した
# * 2011/09/22 x, y 軸の値を指定. NCEP のデータも扱えるように編集
# * 2011/09/21 OSR も描画するように編集
#
#
require "numru/ggraph"
include NumRu

######################################

def clim_mean( iyrs, iyre, imon, gphys, data )

  if data == "dcpam" then

     dataNumOfDay = 4
     daysOfMonth  = [31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]

     daysofyear = 0
     daysOfMonth.each do |i|
       daysofyear += i
     end

     time = gphys.coord('time').val

     for iyr in iyrs..iyre
       t = daysofyear*(iyr-1)
       for i in 1..imon-1
         t = t + daysOfMonth[i-1]
       end

       ts = t
       te = t + daysOfMonth[imon-1]

       ts = ts * dataNumOfDay
       te = te * dataNumOfDay - 1

       if iyr == iyrs
         gphystmp = gphys.cut('time'=>time[ts]..time[te]).mean('time')
       else
         gphystmp = gphystmp + gphys.cut('time'=>time[ts]..time[te]).mean('time')
       end
         gphystmp = gphystmp / ( iyre - iyrs + 1 )
     end

  elsif data == "NCEP" then

    hoursOfDay = 24
    daysofyear = 0

    time = gphys.coord('time').val

    for iyr in iyrs..iyre
 
=begin
      ulu1 = iyr % 400
      ulu2 = iyr % 100
      ulu3 = iyr % 4
      #p ulu1, ulu2, ulu3

      if ulu1 == 0 then

        daysOfMonth = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
 
      elsif ulu2 == 0 then

        daysOfMonth = [31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]

      elsif ulu3 == 0 then
 
        daysOfMonth = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
      
      else

        daysOfMonth = [31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]

      end

      daysOfMonth.each do |i|
      daysofyear += i
      end

      for i in 1..imon-1
        t = t + daysOfMonth[i-1]
      end
=end

     t  = (iyr - 1948) * 12

     t  = t + (imon - 1)

     ts = t
 
#      p iyrs, imon, ts

      if iyr == iyrs
        gphystmp = gphys.cut('time'=>time[ts])
      else
        gphystmp = gphystmp + gphys.cut('time'=>time[ts])
      end
    end

     gphystmp = gphystmp / ( iyre - iyrs + 1 )

  end

  return gphystmp

end

######################################

iyrs = ARGV[2].to_i
iyre = ARGV[3].to_i

imon = ARGV[4].to_i

title = ARGV[5]

dir    = ARGV[0]

data = ARGV[6]

#vname1 = 'SurfTemp'
#vname2 = 'Rain'
#vname3 = 'SoilMoist'
#vname4 = 'SurfSnow'

vname1 = ARGV[1]
vname1_longname = vname1 + "top"

if data == "dcpam" then

  gphys1 = GPhys::NetCDF_IO.open(dir+'/'+vname1+".nc", vname1) 
  #gphys2 = GPhys::NetCDF_IO.open(dir+'/'+vname2+".nc", vname2)
  #gphys3 = GPhys::NetCDF_IO.open(dir+'/'+vname3+".nc", vname3)
  #gphys4 = GPhys::NetCDF_IO.open(dir+'/'+vname4+".nc", vname4)

  tm = gphys1.shape[2] # number of elements for time      dimension
  time = gphys1.coord('time').val

elsif data == "NCEP" then

  gphys1 = GPhys::NetCDF_IO.open(dir+'/'+vname1_longname + ".mon.mean"+".nc", vname1) 
  #gphys2 = GPhys::NetCDF_IO.open(dir+'/'+vname2+".nc", vname2)
  #gphys3 = GPhys::NetCDF_IO.open(dir+'/'+vname3+".nc", vname3)

  tm = gphys1.shape[0] # number of elements for time      dimension
  time = gphys1.coord('time').val

end 

#< DCLのオープンと設定 >
DCL.gropn(2)
#DCL.sldiv('y',2,2)           # 2x2に画面分割, 'y'=yoko: 左上→右上→左下...
DCL.sgpset('lcntl', false)   # 制御文字を解釈しない
DCL.sgpset('lfull',true)     # 全画面表示
DCL.uzfact(0.75)             # 座標軸の文字列サイズを 0.75 倍
DCL.sgpset('lfprop',true)    # プロポーショナルフォントを使う

#< GGraph による 描画 >
#GGraph.set_fig 'itr'=>10, 'viewport'=>[0.15,0.85,0.1,0.6]

GGraph.set_fig 'viewport'=>[0.15,0.7,0.1,0.5] # 描画領域 (xmin, xmax, ymin, ymax)
GGraph.set_axes('xlabelint'=>30) # x 軸に値を打つ間隔

gphystmp = clim_mean( iyrs, iyre, imon, gphys1, data ).mean('lon')

if vname1 == 'SurfTemp'

# first panel
  GGraph.line( gphystmp )
#  GGraph.tone( gphystmp, true, 
#             'lev'=>[210,220,230,240,250,260,270,280,290,300],
#             # レベル＆パターンを陽に指定
#             'pat'=>[20999,30999,40999,45999,50999,55999,60999,70999,80999,85999,90999], 
#             # パタンの方が1つ多→±∞まで
#             'map_axes'=>true )
#  #GGraph.contour( gphystmp, false, 'lev'=>[200,210,220,230,240,250,260,270,280,290,300], 'index'=>3 )      # 参考まで

elsif vname1 == 'Rain'

  # second panel
  gphystmp       = gphystmp / 2.5e6
  gphystmp.units = 'kg m-2 s-1'
  GGraph.line( gphystmp )
#  GGraph.tone( gphystmp, true, 
#             'lev'=>[0,1e-5,2e-5,4e-5,8e-5,12e-5,16e-5,20e-5,24e-5,28e-5,32e-5],
#             # レベル＆パターンを陽に指定
#             'pat'=>[1,20999,30999,35999,40999,50999,60999,65999,70999,75999,80999,85999], 
#             # パタンの方が1つ多→±∞まで
#             'map_axes'=>true )
#  #GGraph.contour( gphystmp, false, 'lev'=>[0,10,20,30,40,50,60,70,80,90,100,110,120,130,140,150], 'index'=>3 )      # 参考まで

elsif vname1 == 'SoilMoist'


  # third panel
  GGraph.line( gphystmp )
#  GGraph.tone( gphystmp, true, 
#             'lev'=>[0,10,20,30,40,50,60,70,80,90,100,110,120,130,140,150],
#             # レベル＆パターンを陽に指定
#             'pat'=>[15999,20999,25999,30999,35999,40999,45999,50999,55999,60999,65999,70999,75999,80999,85999,90999,95999],
#             # パタンの方が1つ多→±∞まで
#             'map_axes'=>true )
#  #GGraph.contour( gphystmp, false, 'lev'=>[0,10,20,30,40,50,60,70,80,90,100,110,120,130,140,150], 'index'=>3 )      # 参考まで

elsif vname1 == 'SurfSnow'

  # forth panel
  GGraph.line( gphystmp )
#  GGraph.tone( gphystmp, true, 
#             'lev'=>[0,10,20,30,40,50,60,70,80,90,100,110,120,130,140,150],
#             # レベル＆パターンを陽に指定
#             'pat'=>[15999,20999,25999,30999,35999,40999,45999,50999,55999,60999,65999,70999,75999,80999,85999,90999,95999], 
#             # パタンの方が1つ多→±∞まで
#             'map_axes'=>true )
#  #GGraph.contour( gphystmp, false, 'lev'=>[0,10,20,30,40,50,60,70,80,90,100,110,120,130,140,150], 'index'=>3 )      # 参考まで

elsif vname1 == 'ulwrf' || vname1 == 'OLRA'

  # forth panel

  GGraph.set_fig('window'=>[-90,90,140,300])     # 軸の最大/最小値を設定. [x 軸最小, x 軸最大, y 軸最小, y 軸最大]
  GGraph.line( gphystmp )
#  GGraph.tone( gphystmp, true, 
#             'lev'=>[100,120,140,160,180,200,220,240,260,280,300],
#             # レベル＆パターンを陽に指定
#             'pat'=>[10999,15999,20999,25999,30999,35999,40999,50999,60999,70999,80999,90999],
#             # パタンの方が1つ多→±∞まで
#             'map_axes'=>true )
#  #GGraph.contour( gphystmp, false, 'lev'=>[0,10,20,30,40,50,60,70,80,90,100,110,120,130,140,150], 'index'=>3 )      # 参考まで

elsif vname1 == 'OSR'

  # forth panel

   GGraph.set_fig('window'=>[-90,90,0,350])     # 軸の最大/最小値を設定. [x 軸最小, x 軸最大, y 軸最小, y 軸最大]
  GGraph.line( gphystmp )
#  GGraph.tone( gphystmp, true, 
#             'lev'=>[100,120,140,160,180,200,220,240,260,280,300],
#             # レベル＆パターンを陽に指定
#             'pat'=>[10999,15999,20999,25999,30999,35999,40999,50999,60999,70999,80999,90999],
#             # パタンの方が1つ多→±∞まで
#             'map_axes'=>true )
#  #GGraph.contour( gphystmp, false, 'lev'=>[0,10,20,30,40,50,60,70,80,90,100,110,120,130,140,150], 'index'=>3 )      # 参考まで

end

#GGraph.color_bar


DCL::uxmttl('T', ' ', 1.0)
DCL::uxmttl('T', ' ', 1.0)
DCL::uxmttl('T', title, -1.0)

DCL.grcls

