source: trunk/anuga_work/development/mem_time_tests/scenarios/vel2/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 for the simulation output and the log files
k= sys.argv[1]
l= float(sys.argv[2])

home = os.getenv('INUNDATIONHOME')
scenariodirV = add_directories(home, ["data","mem_time_test", "scenarios",
                                       "velocity", "velocity-" + str(k) +"-"+ str(l)])

log.timingInfo(msg=('velocity,'+str(k))) #write the variable to be measured to file
log.timingInfo(msg=('extent,'+str(l**2))) #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.log_filename = os.path.join(scenariodirV, "anuga.log")
log._setup = False

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

log.resource_usage_timing(prefix = 'aftermesh')  #get memory usage 
log.timingInfo(msg=('aftermeshtime,'+str(log.TimeStamp()))) #get the time at the beginning of the simulation

#get the number of triangles
number = len(domain)
log.timingInfo(msg=('numberoftriangles'+str(number))) #write the variable to be measured to file

#--------------------------------------------------------------------------
# Setup initial conditions
#--------------------------------------------------------------------------
def topography(x, y):
    return  0.0
domain.set_quantity('elevation', topography) # Use function for elevation
domain.set_quantity('friction', 0.01) # Constant friction
domain.set_quantity('stage', -1000.0)

log.resource_usage_timing(prefix='afterinitialconditions')#get memory usage

#------------------------------------------------------------------------------
# Setup boundary conditions
#------------------------------------------------------------------------------
Bi = anuga.Dirichlet_boundary([0.5, k, 0]) # Inflow 
Br = anuga.Reflective_boundary(domain) # Solid reflective wall
Bo = anuga.Dirichlet_boundary([-0.5, 0, 0]) # Outflow
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=0.2, finaltime=40.0):
    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