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

#------------------------------------------------------------------------------
# Import necessary modules
#------------------------------------------------------------------------------
import anuga
import time
import sys
import os
from anuga.abstract_2d_finite_volumes.util import add_directories
from anuga.utilities import log

#set up variables to store the experiment data and the log file
length = sys.argv[1]
lower = sys.argv[2]

home = os.getenv('INUNDATIONHOME')
scenariodirV = add_directories(home, ["data","mem_time_test", "scenarios",
                                       "channelflow", "channelflow-" + str(lower) +"-"+ str(length)])
log.log_filename = os.path.join(scenariodirV, "anuga.log")
log._setup = False

log.timingInfo(msg=('length,'+str(length))) #write the variable to be measured to file
log.timingInfo(msg=('stepplace,'+str(lower))) #write the variable to be measured to file
log.timingInfo(msg=('beforetime,'+str(log.TimeStamp()))) #get the time at the beginning of the simulation

log.resource_usage_timing(prefix = 'beforesimulation')#get memory usage
#------------------------------------------------------------------------------
# Setup computational domain
#------------------------------------------------------------------------------
points, vertices, boundary = anuga.rectangular_cross(100,300,len1=length, len2=length) # Mesh
domain = anuga.Domain(points, vertices, boundary) # Create domain
domain.set_datadir(scenariodirV) 
domain.set_name('channel1.sww') # Output name

log.timingInfo(msg=('aftermeshtime,'+str(log.TimeStamp()))) #get the time at the beginning of the simulation
log.resource_usage_timing(prefix = 'aftermesh') #get memory usage
#--------------------------------------------------------------------------
# Setup Initial Conditions
#--------------------------------------------------------------------------
def topography(x, y):
    z = 0
    N = len(x)
    
    #makes a step 
    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', -10000000000000000000000.0)

log.resource_usage_timing(prefix='afterinitialconditions')#get memory usage
#------------------------------------------------------------------------------
# Setup boundary conditions
#------------------------------------------------------------------------------
Bi = anuga.Dirichlet_boundary([0.5, 10, 0]) # Inflow
Bo = anuga.Dirichlet_boundary([-0.5,0,0]) #outflow
Br = anuga.Reflective_boundary(domain) # Solid reflective wall
domain.set_boundary({'left': Bi, 'right': Bo, 'top': Br, 'bottom': Br})

log.resource_usage_timing(prefix='afterboundary')#get memory usage
#------------------------------------------------------------------------------
# Evolve system through time
#------------------------------------------------------------------------------
for t in domain.evolve(yieldstep=120, finaltime=3600):
    domain.write_time()

log.resource_usage_timing(prefix='aftersimulation') #get memory usage
log.timingInfo(msg=('aftertime,'+str(log.TimeStamp()))) #get the time at the end of the simulation