Changeset 8326 for trunk/anuga_work/development/mem_time_tests/parameters/timelen-over-timestep/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/parameters/timelen-over-timestep/main.py (modified) (2 diffs)

trunk/anuga_work/development/mem_time_tests/parameters/timelen-over-timestep/main.py

@@ -1 @@
-"""Simple water flow example using ANUGA
-Water flowing down a channel
-"""
 #------------------------------------------------------------------------------
 # Import necessary modules
@@ -10 +7 @@
 import random
 import subprocess
-import ex1
 import csv
 import os
-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')
+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", "parameters",
                                      "timelen-over-timestep"])
 file = 'ex1.csv'
+file_path = os.path.join(scenariodir, file)
+spamWriter = csv.writer(open(file_path, 'wb'))
+spamWriter.writerow(['Time Length', 'Time Step' , 'Time Taken','Space Used'])
+store ='store.txt'
+file_path_store = os.path.join(scenariodir, store)
+storen ='storen.txt'
+file_path_storen = os.path.join(scenariodir, storen)
 
-log.log_filename = os.path.join(scenariodir, 'log.txt')
+#create each file
+e = open(file_path_store,'a') 
+e.close()
+e = open(file_path_storen,'a')
+e.close()
 
-file_path = os.path.join(scenariodir, file)
 
-spamWriter = csv.writer(open(file_path, 'wb'))
 
-spamWriter.writerow(['Time Length', 'Time Step' , 'Time Taken','Space Used'])
-
+# main loops that give the time length (m) and the time step (n)
 for m in range(1,1000,50):
 
     for n in range(1,1000,5):
-        n = n/1000.0
-        system_tools.MemoryUpdate()
-        x = time.clock()
-        ex1.runex(float(m),n)
-        y = time.clock() 
-        b = system_tools.MemoryUpdate()[0]
-        spamWriter.writerow([m,n ,(y-x),b])
-        a.append((n,y-x,b))
-print a
+        n = n/1000.0 #adjust it so its small enough to see the nature of the relationship
 
+        #write to file
+        s = open(file_path_store,'r+')
+        s.write(str(float(m)))
+        t = open(file_path_storen,'r+')
+        t.write(str(n))
+
+        x = time.clock()#time it
+        subprocess.call(['python2.5', 'ex1.py'])#run it
+        y = time.clock()#time it
+        spamWriter.writerow([m,n ,(y-x),'b'])#write the results
+
+print 'Done'
+
+analyse_log(scenariodir, 'metalog.csv')#get the memory usage statistics from the log files