source: anuga_validation/okushiri_2005/run_okushiri_parallel.py @ 3929

"""Validation of the AnuGA implementation of the shallow water wave equation.

This script sets up LWRU2 benchmark with initial condition stated

See also

http://www.cee.cornell.edu/longwave/index.cfm?page=benchmark&problem=2

Depth at western boundary is d = 13.5 cm

This a parallel version.
"""

# Module imports
from Numeric import array, zeros, Float, allclose

from anuga.shallow_water import Domain
from anuga.shallow_water import Reflective_boundary
from anuga.shallow_water import Transmissive_Momentum_Set_Stage_boundary
from anuga.abstract_2d_finite_volumes.util import file_function

from anuga_parallel.parallel_api import myid, numprocs, distribute

import project


#-------------------------
# Create Domain
#-------------------------
domain = Domain(project.mesh_filename, use_cache=True, verbose=True)

#-------------------------
# Initial Conditions
#-------------------------
domain.set_quantity('friction', 0.0)
domain.set_quantity('stage', 0.0)
domain.set_quantity('elevation',
                    filename = project.bathymetry_filename[:-4] + '.pts',
                    alpha = 0.02,                    
                    verbose = True,
                    use_cache = True)


#-------------------------
# Distribute domain
#-------------------------
domain = distribute(domain)

# Parameters
domain.set_name('okushiri_parallel')
domain.set_minimum_storable_height(0.001)
domain.set_default_order(2)
domain.set_maximum_allowed_speed(0.1) # Allow a little runoff (0.1 is OK)

# Set old (pre Sep 2006) defaults for limiters
domain.beta_w      = 0.9
domain.beta_w_dry  = 0.9
domain.beta_uh     = 0.9
domain.beta_uh_dry = 0.9
domain.beta_vh     = 0.9
domain.beta_vh_dry = 0.9


#------------------------------------------------------------------------------
# Setup boundary conditions
# (MUST currently happen after domain has been distributed)
#------------------------------------------------------------------------------

Br = Reflective_boundary(domain)      # Solid reflective wall

function = file_function(project.boundary_filename[:-4] + '.tms',
                         domain, 
                         verbose=True)

Bts = Transmissive_Momentum_Set_Stage_boundary(domain, function)

domain.set_boundary({'wave': Bts,  # Bind this one later
                     'wall': Br})


#-------------------------
# Evolve through time
#-------------------------
import time
t0 = time.time()

for t in domain.evolve(yieldstep = 0.05, finaltime = 22.5):
    domain.write_time()

if myid == 0:
    print 'That took %.2f seconds' %(time.time()-t0)