source: trunk/anuga_work/development/mem_time_tests/triangles/fromregions/runcairns.py @ 8326

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

# set up the variables for the temporary files
home = os.getenv('INUNDATIONHOME')
scenariodir = add_directories(home, ["data","mem_time_test", "triangles",
                                     "fromregions"])
store ='store.txt'
file_path_store = os.path.join(scenariodir, store)
storen ='storen.txt'
file_path_storen = os.path.join(scenariodir, storen)
storea = 'storea.txt'
file_path_storea = os.path.join(scenariodir, storea)

#read the maximum triangle area and map side length from the text files
f = open(file_path_store,'r+')
a = float(f.readline())
f.close()
f = open(file_path_storen,'r+')
l = float(f.readline())
f.close()

#set up the variables for the simulation output and the log files
scenariodirV = add_directories(home, ["data","mem_time_test", "triangles",
                                       "fromregions", "triangles-" + str(a) +"-"+ str(l)])
h = 'CAIRNS.msh'
file_pathh = os.path.join(scenariodirV, h)
log.log_filename = os.path.join(scenariodirV, "anuga.log")
log._setup = False

log.resource_usage_timing(prefix = 'BeforeSimulation') #get memory usage
#------------------------------------------------------------------------------
# Create the triangular mesh and domain
#------------------------------------------------------------------------------
domain = anuga.create_domain_from_regions([(0.0,0.0),(l,l),(0.0,l),(l,0.0)],
                                    boundary_tags={'top': [0],
                                                   'right': [1],
                                                   'bottom': [2],
                                                   'left': [3]},
                                    maximum_triangle_area=a,
                                    mesh_filename=file_pathh
                                    )                            
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
#------------------------------------------------------------------------------
# Setup initial conditions
#------------------------------------------------------------------------------

#get the number of triangles
number=len(domain)

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]) # inflow
Bs = anuga.Transmissive_stage_zero_momentum_boundary(domain) # Neutral boundary
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
  
#write the number of triangles to file
i = open(file_path_storea, 'r+')
i.write(str(number))