source: trunk/anuga_work/development/mem_time_tests/triangles/fromregions/main.py @ 8303

"""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 subprocess
import runcairns
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", "triangles",
                                     "fromregions"])
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(['Number Of Triangles' ,'Max Triangle Area', 'Extent', 'Space Used MB' , 'Time Taken s'])



def runlist(r):
    for n in range(1,2000,100):
        system_tools.MemoryUpdate()
        z = time.time()
        h = runcairns.runex(n,r)
        y = time.time()
        x = system_tools.MemoryUpdate()[0]
        spamWriter.writerow([h,r,(n*n),x ,(y-z)])

for m in range(1,100,10):
    runlist(m)

print 'DONE'