source: trunk/anuga_work/development/mem_time_tests/parameters/timelen-over-timestep/ex1.py @ 8326

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

#set up variables to store the temporary data
home = os.getenv('INUNDATIONHOME')
scenariodir = add_directories(home, ["data","mem_time_test", "parameters",
                                     "timelen-over-timestep"])
store ='store.txt'
file_path_store = os.path.join(scenariodir, store)
storen ='storen.txt'
file_path_storen = os.path.join(scenariodir, storen)

#get the time length and time step from the text files
f = open(file_path_store,'r+') 
g = float(f.readline())
f.close()
f = open(file_path_storen,'r+')
h = float(f.readline())
f.close()

#set up variables to store the data and the log files
scenariodirV = add_directories(home, ["data","mem_time_test", "parameters",
                                       "timelen-over-timestep", "TT-" + str(g) +"-"+ str(h)])
log.log_filename = os.path.join(scenariodirV, "anuga.log")
log._setup = False


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

#------------------------------------------------------------------------------
#Create Domain and Mesh
#------------------------------------------------------------------------------
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

#------------------------------------------------------------------------------
# Setup initial conditions
#------------------------------------------------------------------------------
def topography(x, y):
    return -x/10 # linear bed slope

domain.set_quantity('elevation', topography) # Use function for elevation
domain.set_quantity('friction', 0.01) # Constant friction
domain.set_quantity('stage', expression='elevation')
log.resource_usage_timing(prefix='afterinitialconditions')  #get memory usage here

#------------------------------------------------------------------------------
# Setup boundary conditions
#------------------------------------------------------------------------------
Bi = anuga.Dirichlet_boundary([0.4, 0, 0]) # Inflow
Br = anuga.Reflective_boundary(domain) # Solid reflective wall
domain.set_boundary({'left': Bi, '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=h, finaltime=g):
    print domain.timestepping_statistics()

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