# and a constant rain forcing. The full Jacobian is use and there is no dampening in the
# overland flow.
#--------------------------------------------------------------
from parflow import Run
from parflow.tools.fs import cp, mkdir, chdir, get_absolute_path
LWvdz = Run("LWvdz", __file__)
#---------------------------------------------------------
# Copying slope files
#---------------------------------------------------------
dir_name = get_absolute_path('test_output/LWvdz')
mkdir(dir_name)
chdir(dir_name)
cp('$PF_SRC/test/input/lw.1km.slope_x.10x.pfb')
cp('$PF_SRC/test/input/lw.1km.slope_y.10x.pfb')
#---------------------------------------------------------
LWvdz.FileVersion = 4
LWvdz.Process.Topology.P = 1
LWvdz.Process.Topology.Q = 1
LWvdz.Process.Topology.R = 1
#---------------------------------------------------------
# Computational Grid
#---------------------------------------------------------
hflow.Geom.lower_aquifer.Lower.Y = 0.0
hflow.Geom.lower_aquifer.Lower.Z = 0.0
hflow.Geom.lower_aquifer.Upper.X = 17.0
hflow.Geom.lower_aquifer.Upper.Y = 10.2
hflow.Geom.lower_aquifer.Upper.Z = 1.5
#-----------------------------------------------------------------------------
# Perm
#-----------------------------------------------------------------------------
hflow.Geom.Perm.Names = 'upper_aquifer lower_aquifer'
# we open a file, in this case from PEST to set upper and lower kg and sigma
cp('$PF_SRC/test/input/stats4.txt')
chdir('.')
file = open('stats4.txt', 'r')
lines = file.readlines()
kgu = float(lines[0])
varu = float(lines[1])
kgl = float(lines[2])
varl = float(lines[3])
## we use the parallel turning bands formulation in ParFlow to simulate
## GRF for upper and lower aquifer
hflow.Geom.upper_aquifer.Perm.Type = 'TurnBands'
hflow.Geom.upper_aquifer.Perm.LambdaX = 3.60
hflow.Geom.upper_aquifer.Perm.LambdaY = 3.60
hflow.Geom.upper_aquifer.Perm.LambdaZ = 0.19