=begin

=exner-cont.rb

エクスナー関数のコンター図を描く Ruby スクリプト

==USAGE

  exner-cont.rb [option]

==option

  -xt : x-t コンター図を描く

  -zt : z-t コンター図を描く

  --ps : ポストスクリプトファイルに出力する. dcl.ps というファイル名になる.

  --without-tone : カラートーンを描かない.

  --without-contour : コンターを描かない.

=end

require "numru/gphys"
require "numru/dcl"
require "/home/kitamo/lib/ruby/colorbar" # カラーバーモジュール

include NumRu

##### configure ###########################

p xinterval = 1 # x 方向に格子点を間引く割合
p yinterval = 1 # y 方向に格子点を間引く割合

xmin = 0.0      # x 座標の最小値
xmax = 10000.0  # x 座標の最大値
ymin = 0.0      # y 座標の最小値
ymax = 20.0	# y 座標の最大値

valmin = 0.998  # 変数の最小値
valmax = 1.008  # 変数の最大値

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

# ファイルオープン

filename = ARGV[0]
varname = "Exner"
gphys0 = GPhys::NetCDF_IO.open(filename,varname)

# 変数配列, 座標軸情報を取り出す

if ARGV.index("-xt")
  gphys0 = gphys0[0..-1,0,0..-1]
  x_title = "x"
  x_unit = "m"
  y_title = "t"
  y_unit = "s"
elsif ARGV.index("-zt")
  gphys0 = gphys0[0,0..-1,0..-1]
  x_title = "z"
  x_unit = "m"
  y_title = "t"
  y_unit = "s"
else
  gphys0 = gphys0[0..-1,0,0..-1]
end

data0 = gphys0.data.val
xaxis = gphys0.coord(0).val
yaxis = gphys0.coord(1).val

ysize = yaxis.shape[0]-1
xsize = xaxis.shape[0]-1

xaxis1 = NArray.sfloat((xsize+1)/1).fill(0.0)
yaxis1 = NArray.sfloat((ysize+1)/1).fill(0.0)
data1 = NArray.sfloat(xsize+1, (ysize+1)/1).fill(0.0)

# 描画に用いる格子を間引く 

0.upto(xsize) {|i|
  if (i % xinterval == 0)
    xaxis1[i/xinterval] = xaxis[i]
  end
}
0.upto(ysize) {|i|
  if (i % yinterval == 0)
    yaxis1[i/yinterval] = yaxis[i]
    0.upto(xsize) {|j|
        if (j % xinterval == 0)
        data1[j/xinterval,i/yinterval] = data0[j,i]
      end 
    }
  end
}

# 出力形式を決定

if ARGV.index("--ps") then
  DCL::gropn(2)
elsif
  DCL::gropn(4)
end

# 描画範囲の指定

vxmin = 0.2
vxmax = 0.8
vymin = 0.2
vymax = 0.8

DCL::sglset("LCNTL", false )
DCL::udlset("LMSG", false )
DCL::uglset("LMSG", false )
DCL::gllset("LMISS", true )

# 手動でカラートーンを指定
#
#cltstep = 10
#delta = (valmax - valmin) / cltstep
#for k in 1..10
#  tlev1 = valmin + (k-1)*delta
#  tlev2 = tlev1 + delta
#  ipat = (13+8*k)*1000 + 999
#  DCL::uestlv(tlev1,tlev2,ipat)
#end

DCL::grfrm
DCL::grswnd(xmin,xmax,ymin,ymax)
DCL::grsvpt(vxmin,vxmax,vymin,vymax)
DCL::grstrn(1)
DCL::grstrf

DCL::sglset("LCLIP", true )

# 格子点の指定

DCL::uwsgxa(xaxis1)
DCL::uwsgya(yaxis1)

# カラートーン描画

unless ARGV.index("--without-tone")
  DCL::ueitlv
  DCL::uegtla(valmin, valmax, 0 )
  DCL::uetone(data1)
end

# コンター描画

unless ARGV.index("--without-contour")
  DCL::udgcla(valmin, valmax, 0 )
  DCL::udcntz(data1)
end

DCL::ussttl(x_title, x_unit, y_title, y_unit)
DCL::usdaxs
DCL::sglset("LCLIP", false )
i=0
DCL::sgtxzr(vxmax+0.01,vymax-0.03*i,"z=5050 m",0.02,0,-1,1)
DCL::uzrset("ROFFXT", 0.06)
title = "Exner(1)"
DCL::uxsttl("t", title, 0 )


# カラーバーモジュールを読み込む

MyTool::colorbar(valmin,valmax)


DCL::grcls