#Test Name: SquareSheetHydrologyDCTwoLayers
import numpy
from model import *
from triangle import *
from setmask import *
from parameterize import *
from transient import *
from setflowequation import *
from EnumDefinitions import *
from solve import *
from MatlabFuncs import *

from generic import generic

md=triangle(model(),'../Exp/Square.exp',100000.)
md=setmask(md,'','')
md=parameterize(md,'../Par/SquareSheetConstrained.py')
md.transient=transient.setallnullparameters(md.transient)
md.transient.ishydrology=True
md.transient.issmb=True
md=setflowequation(md,'SSA','all')
md.cluster=generic('name',oshostname(),'np',1)
md.hydrology=hydrologydc()
md.hydrology=md.hydrology.initialize(md)

md.hydrology.isefficientlayer=1
md.hydrology.sedimentlimit_flag=1
md.hydrology.sedimentlimit=800.0
md.initialization.sediment_head=numpy.zeros((md.mesh.numberofvertices,1))
md.hydrology.spcsediment_head=float('NaN')*numpy.ones((md.mesh.numberofvertices,1))

md.basalforcings.groundedice_melting_rate = 2.0*numpy.ones((md.mesh.numberofvertices,1))
md.basalforcings.floatingice_melting_rate = 0.0*numpy.ones((md.mesh.numberofvertices,1))
md.hydrology.sediment_transmitivity= 3.0*numpy.ones((md.mesh.numberofvertices,1))

md.initialization.epl_head=numpy.zeros((md.mesh.numberofvertices,1))
md.initialization.epl_thickness=numpy.ones((md.mesh.numberofvertices,1))
md.hydrology.spcepl_head=float('NaN')*numpy.ones((md.mesh.numberofvertices,1))
md.hydrology.mask_eplactive_node=numpy.zeros((md.mesh.numberofvertices,1))
md.hydrology.epl_conductivity=30
md.hydrology.epl_initial_thickness=1
md.hydrology.epl_colapse_thickness=1.0e-3
md.hydrology.epl_thick_comp=1
md.hydrology.epl_max_thickness=1
md.timestepping.time_step=0.2
md.timestepping.final_time=2.0

md=solve(md,'TransientSolution')

field_names     =['SedimentWaterHead1','EplWaterHead1','SedimentHeadResidual1', \
						'SedimentWaterHead4','EplWaterHead4','SedimentHeadResidual4', \
						'SedimentWaterHead5','EplWaterHead5','SedimentHeadResidual5', \
						'SedimentWaterHead9','EplWaterHead9','SedimentHeadResidual9']
field_tolerances=[ \
	1e-13, 1e-13, 1e-13, \
	1e-13, 1e-13, 1e-13, \
	1e-13, 5e-12, 1e-11, \
	1e-13, 5e-12, 1e-11]
field_values=[md.results.TransientSolution[0].SedimentHead, \
							md.results.TransientSolution[0].EplHead, \
							md.results.TransientSolution[0].SedimentHeadResidual, \
							md.results.TransientSolution[3].SedimentHead, \
							md.results.TransientSolution[3].EplHead, \
							md.results.TransientSolution[3].SedimentHeadResidual, \
							md.results.TransientSolution[4].SedimentHead, \
							md.results.TransientSolution[4].EplHead, \
							md.results.TransientSolution[4].SedimentHeadResidual, \
							md.results.TransientSolution[8].SedimentHead, \
							md.results.TransientSolution[8].EplHead, \
							md.results.TransientSolution[8].SedimentHeadResidual]