source: trunk/anuga_work/development/mem_time_tests/scenarios/channelflow/ex1.py @ 8304

"""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 random
import os
from anuga.abstract_2d_finite_volumes.util import add_directories

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

#------------------------------------------------------------------------------
# Setup computational domain
#------------------------------------------------------------------------------
def runex(lower,length):
   points, vertices, boundary = anuga.rectangular_cross(50,50,
                                                     len1=length, len2=length) # Mesh

#--------------------------------------------------------------------------
# Setup Domain only on processor 0
#--------------------------------------------------------------------------
   domain = anuga.Domain(points, vertices, boundary) # Create domain
   domain.set_datadir(scenariodir) 
   domain.set_name('channel1.sww') # Output name

   def topography(x, y):
      z = -x/10
      N = len(x)
      for i in range(N):
         if y[i] > lower:
            z[i] += 1
              
      return  z

   domain.set_quantity('elevation', topography) # Use function for elevation
   domain.set_quantity('friction', 0.01) # Constant friction
   domain.set_quantity('stage', -1000.0)


#------------------------------------------------------------------------------
# Setup boundary conditions
#------------------------------------------------------------------------------
   Bi = anuga.Dirichlet_boundary([0.5, 1, 0]) # Inflow
   Bo = anuga.Dirichlet_boundary([-100,0,0]) 
   Br = anuga.Reflective_boundary(domain) # Solid reflective wall
   domain.set_boundary({'left': Bi, 'right': Bo, 'top': Br, 'bottom': Br})

#------------------------------------------------------------------------------
# Evolve system through time
#------------------------------------------------------------------------------
   for t in domain.evolve(yieldstep=0.2, finaltime=40.0):
     #if myid == 0:
       domain.write_time()

#domain.sww_merge()
#finalize()