# -*- coding: utf-8 -*-
# 運動エネルギー密度を計算し, 絵を描く

require "numru/ggraph"
include NumRu
include GGraph

filehead = "BS1998_"
filedir = "../time_000280000-000380000"
cuttime = 280000..380000

velx_file0 = "#{filedir}/#{filehead}VelX_rank000000.nc"
velx_file1 = "#{filedir}/#{filehead}VelX_rank000001.nc"
velx_file2 = "#{filedir}/#{filehead}VelX_rank000002.nc"
velx_file3 = "#{filedir}/#{filehead}VelX_rank000003.nc"
velx_file4 = "#{filedir}/#{filehead}VelX_rank000004.nc"
velx_file5 = "#{filedir}/#{filehead}VelX_rank000005.nc"
velx_var ="VelX"
#vely_file = "../time_000000000-004181000/BS1998-all_VelY.nc"
#vely_var ="VelY"
velz_file0 = "#{filedir}/#{filehead}VelZ_rank000000.nc"
velz_file1 = "#{filedir}/#{filehead}VelZ_rank000001.nc"
velz_file2 = "#{filedir}/#{filehead}VelZ_rank000002.nc"
velz_file3 = "#{filedir}/#{filehead}VelZ_rank000003.nc"
velz_file4 = "#{filedir}/#{filehead}VelZ_rank000004.nc"
velz_file5 = "#{filedir}/#{filehead}VelZ_rank000005.nc"
velz_var ="VelZ"
dens_file = "./restart_long-DensBZ.nc"
dens_var = "DensBZ"



DCL.gropn(2)

velx = GPhys::IO.open([velx_file0,velx_file1,velx_file2,velx_file3,velx_file4,velx_file5], "#{velx_var}")
velz = GPhys::IO.open([velz_file0,velz_file1,velz_file2,velz_file3,velz_file4,velz_file5], "#{velz_var}")
#vely = GPhys::IO.open("#{vely_file}", "#{vely_var}")
#velz = GPhys::IO.open("#{velz_file}", "#{velz_var}")
dens = GPhys::IO.open("#{dens_file}", "#{dens_var}")
time = GPhys::IO.open("#{velx_file0}", 't').val
x_dco = GPhys::IO.open("#{dens_file}", 'x').val
x_co = GPhys::IO.open("#{velx_file0}", 'x').val
z_co = GPhys::IO.open("#{velz_file0}", 'z').val

#dens = dens.mean('x').mean('y')
velx = velx.cut('t'=>cuttime)
velz = velz.cut('t'=>cuttime)
kener = 0.50 * ( velx*velx + velz*velz) * dens
kener = kener.mean('x').mean('y').mean('z')


#=begin
GGraph.set_axes('xtitle'=>'time','ytitle'=>'K')
GGraph.set_fig('viewport'=>[0.1,0.8,0.4,0.6])
GGraph.line( kener, true, 'title'=>'Kinetic energy density', 'max'=>20.0, 'min'=>0.0)

#GGraph.line( kenerdens.cut( 'y'=>100 ), true)
#=end
DCL.grcls