Changeset 8326 for trunk/anuga_work/development/mem_time_tests/scenarios/channelflow/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/scenarios/channelflow/main.py (modified) (1 diff)

trunk/anuga_work/development/mem_time_tests/scenarios/channelflow/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 random
 import subprocess
-import ex1
 import csv
 import os
 import time
-from anuga.utilities import system_tools, log
 from anuga.abstract_2d_finite_volumes.util import add_directories
+from anuga.utilities.log_analyser import analyse_log
 
 #------------------------------------------------------------------------------
-# Setup computational domain
+# Set up variables for the correct directories to store the output
 #------------------------------------------------------------------------------
-a = []
-
-
 home = os.getenv('INUNDATIONHOME')
 scenariodir = add_directories(home, ["data","mem_time_test", "scenarios",
                                      "channelflow"])
-
-log.log_filename = os.path.join(scenariodir, 'log.txt')
-
+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)
 
+#set up the output files
 spamWriter = csv.writer(open(file_path, 'wb'))
 spamWriter.writerow(['Length','Stage','Time','Space'])
+e = open(file_path_store,'a')
+e.close()
+e = open(file_path_storen,'a')
+e.close()
 
+#the main loops that give the length of the inflow boundary (i) and the place to begin the step (n)
 for i in range(100,100000,10000):
+    for n in range(1,i/10,i/100):
+        
+        #adjust
+        i = i /100.0
+        n = n/10.0 
 
-    for n in range(1,i/10,i/100):
-        i = i /100.0
-        n = n/float(i) 
-        system_tools.MemoryUpdate()
-        z = time.time()
-        ex1.runex(n,i)
-        b = system_tools.MemoryUpdate()[0]
-        y = time.time()
-        spamWriter.writerow([i,(n/float(i)) ,(y-z), b])
-print a
+        #write these values to files
+        g = open(file_path_store,'r+')
+        g.write(str(i))
+        h = open(file_path_storen,'r+')
+        h.write(str(n))
 
+        z = time.time() #time it
+        subprocess.call(['python2.5', 'ex1.py'])#run the script
+        y = time.time() #time it
+
+        spamWriter.writerow([i,n,(y-z), 'b'])#record it
+print 'DONE'
+
+analyse_log(scenariodir, 'metalog.csv')#get the memory usage statistics from the various log files