#Test Name: SquareSheetHydrologyDCTwoLayers
import numpy as np
from model import *
from socket import gethostname
from triangle import *
from setmask import *
from parameterize import *
from transient import *
from setflowequation import *
from solve 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',gethostname(),'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=np.zeros((md.mesh.numberofvertices))
md.hydrology.spcsediment_head=float('NaN')*np.ones((md.mesh.numberofvertices))

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

md.initialization.epl_head=np.zeros((md.mesh.numberofvertices))
md.initialization.epl_thickness=np.ones((md.mesh.numberofvertices))
md.hydrology.spcepl_head=float('NaN')*np.ones((md.mesh.numberofvertices))
md.hydrology.mask_eplactive_node=np.zeros((md.mesh.numberofvertices))
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,'Transient')

sedvol=np.zeros(10)
eplvol=np.zeros(10)
time=np.arange(0.2,2.1,0.2)
store=md.constants.g*md.hydrology.sediment_porosity*md.materials.rho_freshwater*((md.hydrology.sediment_compressibility/md.hydrology.sediment_porosity)+md.hydrology.water_compressibility)
sedstore=20.0*store
for i in range(0,10):
	sedvol[i]=np.mean(md.results.TransientSolution[i].SedimentHead)*sedstore
	eplvol[i]=np.mean(md.results.TransientSolution[i].EplHead)*store*np.mean(md.results.TransientSolution[i].HydrologydcEplThickness)

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]