source: trunk/anuga_work/development/mem_time_tests/triangles/area/main.py @ 8326

#------------------------------------------------------------------------------
# Import necessary modules
#------------------------------------------------------------------------------
import anuga
import subprocess
import csv
import os
import time
import liststore
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')
host = os.getenv('HOST')
scenariodir = add_directories(home, ["data","mem_time_test", "triangles","area"])
file1 = 'ex1.csv'
file_path = os.path.join(scenariodir, file1)
storel ='storel.txt'
file_path_storel = os.path.join(scenariodir, storel)
storea ='storea.txt'
file_path_storea = os.path.join(scenariodir, storea)
storen ='storen.txt'
file_path_storen = os.path.join(scenariodir, storen)

#create and set up the files
spamWriter = csv.writer(open(file_path, 'wb'))
spamWriter.writerow(['Number Of Triangles' ,'Max Triangle Area', 'Extent', 'Space Used MB' , 'Time Taken s'])
e = open(file_path_storea,'a')
e.close()
e = open(file_path_storel,'a')
e.close()
e = open(file_path_storen,'a')
e.close()

# this is the main loops that assigns the maximum triangle area (m) and the map side length(l)

n = 4 #number of processors to use
for m in range(90,100,10):
     for l in range(100,2000,100):
        
        #write these values to file
        g = open(file_path_storel,'r+')
        g.write(str(l))
        h = open(file_path_storea,'r+')
        h.write(str(m))
        
        z = time.time()# time it

        #the different ways each host calls MPI properly
        if (host == 'cyclone.agso.gov.au'):
            subprocess.call(['mpirun', '-np', str(n), '-hostfile' ,'~/machinefiles/test.machines_cyclone', '-x','PYTHONPATH','-x','INUNDATIONHOME','python', 'runcairns.py']) 
        if (host == 'tornado.agso.gov.au'):
            subprocess.call(['mpirun', '-np', str(n), '-hostfile' ,'~/machinefiles/test.machines_tornado', '-x','PYTHONPATH','-x','INUNDATIONHOME','python', 'runcairns.py'])
        if (host == 'vayu1'):
            subprocess.call(['mpirun', '-np', str(n), '-x','PYTHONPATH','-x','INUNDATIONHOME','python', 'runcairns.py']) 
        if (host == 'rhe-compute1.ga.gov.au'):
            subprocess.call(['mpirun', '-np', str(n), '-x','PYTHONPATH','-x','INUNDATIONHOME','python2.6', 'runcairns.py'])    
        if (host == 'xe'):
            subprocess.call(['mpirun', '-np', str(n), '-x','PYTHONPATH','-x','INUNDATIONHOME','python', 'runcairns.py']) 
       
        y = time.time()# time it

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

        spamWriter.writerow([i,d,(l*l),'x' ,(y-z)])# record it

print 'Done'

analyse_log(scenariodir, 'metalog.csv') #get the memory statistics from the log files