source: trunk/anuga_work/development/mem_time_tests/triangles/rectanglecross/main.py @ 8328

#------------------------------------------------------------------------------
# Import necessary modules
#------------------------------------------------------------------------------
import anuga
import subprocess
import csv
import os
import time
from anuga.abstract_2d_finite_volumes.util import add_directories
from anuga.utilities.log_analyser import analyse_log

#------------------------------------------------------------------------------
# Set up variables for the correct directories to store the output
#------------------------------------------------------------------------------
home = os.getenv('INUNDATIONHOME')
scenariodir = add_directories(home, ["data","mem_time_test", "triangles",
                                     "rectanglecross"])
storea ='storea.txt'
file_path_storea = os.path.join(scenariodir, storea)
meta = 'metalog.csv'
meta_path = os.path.join(scenariodir, meta)
final = 'final.csv'
final_path = os.path.join(scenariodir, final)
file = 'ex1.csv'
file_path = os.path.join(scenariodir, file)

#set up needed files
firstex1 = open(file_path, 'wb')
spamWriter = csv.writer(firstex1)
spamWriter.writerow(['Number of Triangles','Matrix Size','Extent(m^2)' , 'Time Taken(s)','Space Used'])
e = open(file_path_storea,'a')
e.close()

#these main loops assign the matrix size[number of triangles] (m) and the map side length(n)
for m in range(1,300,30):
    for n in range(1,1000,100):

        z = time.time() #time it
        subprocess.call(['python2.5', 'runcairns.py',str(m),str(n)]) #run the simulation
        y = time.time() #time it

        #read the number of triangles from this file
        f = open(file_path_storea,'r+')
        h = float(f.readline())
        f.close()

        spamWriter.writerow([h,m,(n*n),'x' ,(y-z)]) #record it
print 'DONE'

analyse_log(scenariodir, os.path.join(scenariodir,'metalog.csv')) #get the memory usage from the log files

#close ex1.csv so we can read from it for a different csv reader object
firstex1.close()

#merge the metalog useful memory info and the ex1 recorded info into one csv file named final
ex1 = csv.reader(open(file_path,'rb'))
metalog = csv.reader(open(meta_path,'rb'))
final = csv.writer(open(final_path,'wb'))


for row in ex1:
    d = metalog.next()
    final.writerow([row[0],row[1],d[0],d[3],d[6],d[9],d[12]])#,d[15],d[18]])