source: trunk/anuga_work/development/mem_time_tests/scenarios/stage/main.py @ 8304

"""Simple water flow example using ANUGA
Water flowing down a channel
"""
#------------------------------------------------------------------------------
# Import necessary modules
#------------------------------------------------------------------------------
# Import standard shallow water domain and standard boundaries.
import anuga
import random
import subprocess
import ex1
import csv
import os
import time
from anuga.utilities import system_tools, log
from anuga.abstract_2d_finite_volumes.util import add_directories


#------------------------------------------------------------------------------
# Setup computational domain
#------------------------------------------------------------------------------
a = []

home = os.getenv('INUNDATIONHOME')
scenariodir = add_directories(home, ["data","mem_time_test", "scenarios",
                                     "stage"])

log.log_filename = os.path.join(scenariodir, 'log.txt')
file = 'ex1.csv'
file_path = os.path.join(scenariodir, file)

spamWriter = csv.writer(open(file_path, 'wb'))
spamWriter.writerow(['Stage','Extent','Time','Space'])


def onevar(b):
    for n in range(1,100,10):
        system_tools.MemoryUpdate()
        z = time.time()
        ex1.runex(b,n)
        y = time.time()
        a = system_tools.MemoryUpdate()[0]
        spamWriter.writerow([n ,((b**2)/2),(y-z),a])
    print a

for m in range(100,1000,100):
    onevar(m)