source: anuga_work/development/demos/island.py @ 5442

"""Example of shallow water wave equation.

Island surrounded by water.
This example investigates onshore 'creep'

Creep is defined as water moving uphill over many timesteps.
The initial adjustment at time = 0 is not 'creep'

"""


#------------------------------------------------------------------------------
# Import necessary modules
#------------------------------------------------------------------------------

# Standard modules
from math import exp

# Application specific imports
from anuga.abstract_2d_finite_volumes.mesh_factory import rectangular
from anuga.shallow_water import Domain, Reflective_boundary, Dirichlet_boundary
from anuga.pmesh.mesh_interface import create_mesh_from_regions
from anuga.utilities.polygon import Polygon_function, read_polygon

from anuga.abstract_2d_finite_volumes.quantity import Quantity
from Numeric import allclose

#------------------------------------------------------------------------------
# Setup computational domain
#------------------------------------------------------------------------------
name = 'island'
mesh_filename =  name + '.tsh'
if True:
    #Create basic mesh
    create_mesh_from_regions( [[0,0], [100,0], [100,100], [0,100]],
                              boundary_tags = {'bottom': [0],
                                               'right': [1],
                                               'top': [2],
                                               'left': [3]},
                              maximum_triangle_area = 2,
                              filename =  mesh_filename,
                              interior_regions=[ ([[50,25], [70,25], [70,75], [50,75]], 10.0)]
                              #interior_holes=[[[50,25], [70,25], [70,75], [50,75]]],
                              )





# Select precomputed meshes here. In changeset:4725 there is a huge difference
# in timestepping characteristics depending on the mesh used.
# The mesh generated on windows takes 10 times as many timesteps.

#Create shallow water domain
domain = Domain(mesh_filename = mesh_filename)
domain.smooth = False #True
domain.set_name(name)
#domain.set_name('island_not_unique')
domain.default_order = 2

print domain.statistics()


#---------------------------
# Combinations with less creep
#---------------------------

domain.maximum_allowed_speed = 0 
domain.tight_slope_limiters = True



#------------------------------------------------------------------------------
# Setup initial conditions
#------------------------------------------------------------------------------

def island(x, y):
    z = 0*x
    for i in range(len(x)):
        z[i] = 20*exp( -((x[i]-50)**2 + (y[i]-50)**2)/100 )

        #z[i] += 0.5*exp( -((x[i]-10)**2 + (y[i]-10)**2)/50 )

    return z

def slump(x, y):
    z = 0*x
    for i in range(len(x)):
        z[i] -= 0.7*exp( -((x[i]-10)**2 + (y[i]-10)**2)/200 )

    return z

stage_value = 10.0
#domain.set_quantity('friction', 0.1)  #Honky dory
domain.set_quantity('friction', 0.01)     #Creep
domain.set_quantity('elevation', island)
domain.set_quantity('stage', stage_value)
domain.max_timestep = 0.01



#------------------------------------------------------------------------------
# Setup boundary conditions (all reflective)
#------------------------------------------------------------------------------

Br = Reflective_boundary(domain)
Bd = Dirichlet_boundary([stage_value, 0, 0])

domain.set_boundary({'left': Bd, 'right': Bd, 'top': Bd, 'bottom': Bd, 'exterior': Br})
domain.check_integrity()

#------------------------------------------------------------------------------
# Evolve system through time
#------------------------------------------------------------------------------

import time
for t in domain.evolve(yieldstep = 1, finaltime = 25):
    domain.write_time()
    #if allclose(t, 100):
    #    Q = domain.get_quantity('stage')
    #    Q_s = Quantity(domain)
    #    Q_s.set_values(slump)
    #    domain.set_quantity('stage', Q + Q_s)
    #print '    Volume: ', domain.get_quantity('stage').get_integral()

print 'Done'