Changeset 8326 for trunk/anuga_work/development/mem_time_tests/triangles/rectanglecross/main.py

Timestamp:

Jan 31, 2012, 10:54:12 AM (12 years ago)

Author:

pittj

Message:

formatted the experiment scripts

File:

• trunk/anuga_work/development/mem_time_tests/triangles/rectanglecross/main.py (modified) (1 diff)

trunk/anuga_work/development/mem_time_tests/triangles/rectanglecross/main.py

@@ -1 @@
-"""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.abstract_2d_finite_volumes.util import add_directories
+from anuga.utilities.log_analyser import analyse_log
 
-from anuga.utilities import system_tools, log
-from anuga.abstract_2d_finite_volumes.util import add_directories
-
+#------------------------------------------------------------------------------
+# 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"])
-
-log.log_filename = os.path.join(scenariodir, 'log.txt')
-
+storea ='storea.txt'
+file_path_storea = os.path.join(scenariodir, storea)
+store ='store.txt'
+file_path_store = os.path.join(scenariodir, store)
+storen ='storen.txt'
+file_path_storen = os.path.join(scenariodir, storen)
 file = 'ex1.csv'
 file_path = os.path.join(scenariodir, file)
 
-#------------------------------------------------------------------------------
-# Setup computational domain
-#------------------------------------------------------------------------------
-a = []
+#create and set up the output files
 spamWriter = csv.writer(open(file_path, 'wb'))
 spamWriter.writerow(['Number Of Triangles' ,'MxM Rectangular Mesh', 'Extent', 'Space Used MB' , 'Time Taken s'])
+e = open(file_path_storea,'a')
+e.close()
+e = open(file_path_store,'a')
+e.close()
+e = open(file_path_storen,'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,500,10):
+    for n in range(1,1000,100):
 
+        #write these values to file
+        g = open(file_path_store,'r+')
+        g.write(str(m))
+        h = open(file_path_storen,'r+')
+        h.write(str(n))
 
-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)])
+        z = time.time() #time it
+        subprocess.call(['python2.5', 'runcairns.py']) #run the simulation
+        y = time.time() #time it
 
-for m in range(1,100,10):
-    runlist(m)
+        #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, 'metalog.csv')#get the memory statistics from the log files
+