source: trunk/anuga_work/development/mem_time_tests/triangles/rectanglecross/runcairns.py @ 8342

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

#set up the variables for the simulation out put and log files
a = int(sys.argv[1])
l = float(sys.argv[2])

home = os.getenv('INUNDATIONHOME')
scenariodirV = add_directories(home, ["data","mem_time_test", "triangles",
                                       "rectanglecross", "triangles-" + str(a) +"-"+ str(l)])
log.log_filename = os.path.join(scenariodirV, "anuga.log")
log._setup = False

log.timingInfo(msg=('matrixsize,'+str(a))) #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.resource_usage_timing(prefix = 'beforesimulation')#get memory usage
#------------------------------------------------------------------------------
# Create the triangular mesh and domain
#------------------------------------------------------------------------------
points, vertices, boundary = anuga.rectangular_cross(a, a,len1 = l,len2 = l) # Basic mesh
domain = anuga.Domain(points, vertices, boundary) # Create 
domain.set_name('CAIRNS.sww') # Name of sww file
domain.set_datadir(scenariodirV)# Store sww output here
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
#------------------------------------------------------------------------------
# Setup initial conditions
#------------------------------------------------------------------------------

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

def topography(x,y):
    return 0.0

tide = 100.0
friction = 0.0
domain.set_quantity('stage', tide)
domain.set_quantity('friction', friction) 
domain.set_quantity('elevation',topography,alpha=0.1)

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

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

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