source: trunk/anuga_work/development/mem_time_tests/parameters/timelen/ex1.py @ 8427

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

#set up variables to store the results and the log files of each experiment
f = sys.argv[1]
g = float(f)/100.0
home = os.getenv('INUNDATIONHOME')
scenariodirV = add_directories(home, ["data","mem_time_test", "parameters",
                                       "timelength", "timelength-" + str(f)])
log.log_filename = os.path.join(scenariodirV, "anuga.log")
log._setup = False

log.timingInfo(msg=('timelength,'+str(f))) #write the variable to be measured to file
log.timingInfo(msg=('timestep,'+str(g))) #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 here

#------------------------------------------------------------------------------
# Create the domain and the mesh of the run
#------------------------------------------------------------------------------
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_name('channel1') # Output name
domain.set_datadir(scenariodirV)  

log.resource_usage_timing(prefix = 'aftermesh') #get memory usage here
log.timingInfo(msg=('aftermeshtime,'+str(log.TimeStamp()))) #get the time at the beginning of the simulation
#------------------------------------------------------------------------------
# Setup initial conditions
#------------------------------------------------------------------------------
def topography(x, y):
    return 0 # linear bed slope
domain.set_quantity('elevation', topography) # Use function for elevation
domain.set_quantity('friction', 0.01) # Constant friction
domain.set_quantity('stage', 10)

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

#------------------------------------------------------------------------------
# Setup boundary conditions
#------------------------------------------------------------------------------
Br = anuga.Reflective_boundary(domain) # Solid reflective wall
domain.set_boundary({'left': Br, 'right': Br, 'top': Br, 'bottom': Br})

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

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


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