source: trunk/anuga_work/development/mem_time_tests/linearregression/area/pruncairns.py

#------------------------------------------------------------------------------
# Import necessary modules
#------------------------------------------------------------------------------
import os
import time
import sys
import anuga
from anuga_parallel import distribute, numprocs, myid
from anuga.abstract_2d_finite_volumes.util import add_directories
from anuga.utilities import log

#set up the variables for the output data and the log files
sidelength = float(sys.argv[3])
timestep = float(sys.argv[1])
finaltime2 = float(sys.argv[2])
depth = (sidelength * sidelength * 0.0004)
velocity = 10
maxtrianglearea = 20

host = os.getenv('HOST')
home = os.getenv('INUNDATIONHOME')
scenariodirV = add_directories(home, ["data","mem_time_test","linearregression","prun",str(host), "variables-" +str(sidelength)+"-"+ str(timestep) +"-"+ str(finaltime2)])
h = 'CAIRNS.msh'
file_pathh = os.path.join(scenariodirV, h)
log.log_filename = os.path.join(scenariodirV, "anuga.log")
log._setup = False

log.timingInfo(msg=('sidelength,'+str(sidelength))) #write the variable to be measured to file
log.timingInfo(msg=('timestep,'+str(timestep))) #write the variable to be measured to file
log.timingInfo(msg=('finaltime,'+str(finaltime2))) #write the variable to be measured to file
log.timingInfo(msg=('depthofwater,'+str(depth))) #write the variable to be measured to file
log.timingInfo(msg=('velocity,'+str(velocity))) #write the variable to be measured to file
log.timingInfo(msg=('maxtrianglearea,'+str(maxtrianglearea))) #write the variable to be measured to file
log.timingInfo(msg=('percentageofwatercover,'+str(100.0))) #write the variable to be measured to file
log.timingInfo(msg=('extent,'+str(sidelength * sidelength))) #write the variable to be measured to file
log.timingInfo(msg=('numberofcpus,'+str(numprocs))) #write the variable to be measured to file
log.timingInfo(msg=('host,'+str(os.getenv('HOST')))) #write the variable to be measured to file
log.timingInfo(msg=('myid,'+str(myid)) #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 on one processor
#------------------------------------------------------------------------------
if myid == 0:
    domain = anuga.create_domain_from_regions([(0.0,0.0),(sidelength,sidelength),(0.0,sidelength),(sidelength,0.0)],
                                    boundary_tags={'top': [0],
                                                   'right': [1],
                                                   'bottom': [2],
                                                   'left': [3]},
                                    maximum_triangle_area=maxtrianglearea,
                                    mesh_filename=file_pathh
                                    )
    #get the number of triangles
    n = len(domain)
    log.timingInfo(msg=('numberoftriangles,'+str(n))) #write the variable to be measured to file

else:
    domain = None
 

#parallel   
domain = distribute(domain)                               

domain.set_name('CAIRNS.sww') # Name of sww file
domain.set_datadir(scenariodirV)# Store sww output here

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):
    return 0.0

domain.set_quantity('stage', depth)
domain.set_quantity('friction', 0) 
domain.set_quantity('elevation',topography,alpha=0.1)

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

#------------------------------------------------------------------------------
# Setup boundary conditions
#------------------------------------------------------------------------------

Bi = anuga.Dirichlet_boundary([depth, velocity, 0]) # inflow
Bo = anuga.Dirichlet_boundary([-depth, velocity, 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=timestep, finaltime=finaltime2): 
    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