source: trunk/anuga_work/development/mem_time_tests/parameters/friction/ex1.py @ 8326

"""Simple water flow example using ANUGA
Water flowing down a channel
"""
#------------------------------------------------------------------------------
# Import necessary modules
#------------------------------------------------------------------------------
# Import standard shallow water domain and standard boundaries.
import anuga
import time
import os
from anuga.abstract_2d_finite_volumes.util import add_directories
from anuga.utilities import log

home = os.getenv('INUNDATIONHOME')
scenariodir = add_directories(home, ["data","mem_time_test", "parameters",
                                     "friction"])


#store = 'store.txt'
#file_path_store = os.path.join(scenariodir, store)

f = open(os.path.join(scenariodir, "store.txt"),'r+') # SQRT extent this is set
friction = float(f.readline())
f.close()

scenariodirV = add_directories(home, ["data","mem_time_test", "parameters",
                                     "friction","friction-" + str(friction)])

log.log_filename = os.path.join(scenariodirV, "anuga.log")
log._setup = False

log.resource_usage_timing(prefix = 'BeforeSimulation')

#------------------------------------------------------------------------------
# Setup computational domain
#------------------------------------------------------------------------------
points, vertices, boundary = anuga.rectangular_cross(100,300,len1=1000.0, len2=500.0) # Mesh
domain = anuga.Domain(points, vertices, boundary) # Create domain
domain.set_datadir(scenariodirV)
domain.set_name('channel1') # Output name

log.resource_usage_timing(prefix = 'AfterMesh')
   
#------------------------------------------------------------------------------
# Setup initial conditions
#------------------------------------------------------------------------------
def topography(x, y):
    return -x/10 # linear bed slope

log.resource_usage_timing(prefix = 'BeforeInitial')
domain.set_quantity('elevation', topography) # Use function for elevation
domain.set_quantity('friction', friction) # Constant friction
domain.set_quantity('stage', # Dry bed
                           expression='elevation')
log.resource_usage_timing(prefix = 'AfterInitial')

#------------------------------------------------------------------------------
# Setup boundary conditions
#------------------------------------------------------------------------------
log.resource_usage_timing(prefix = 'BeforeBoundary')
Bi = anuga.Dirichlet_boundary([0.4, 0, 0]) # Inflow
Br = anuga.Reflective_boundary(domain) # Solid reflective wall
domain.set_boundary({'left': Bi, 'right': Br, 'top': Br, 'bottom': Br})
log.resource_usage_timing(prefix = 'AfterBoundary')

#------------------------------------------------------------------------------
# Evolve system through time
#------------------------------------------------------------------------------
for t in domain.evolve(yieldstep=0.2, finaltime=40.0):
    print domain.timestepping_statistics()

log.resource_usage_timing(prefix = 'AfterSimulation')