Changeset 8326 for trunk/anuga_work/development/mem_time_tests/scenarios/stage/ex1.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/stage/ex1.py (modified) (2 diffs)

trunk/anuga_work/development/mem_time_tests/scenarios/stage/ex1.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 time
@@ -11 +7 @@
 import os
 from anuga.abstract_2d_finite_volumes.util import add_directories
+from anuga.utilities import log
 
+#set up the variables for the temporary data storage
 home = os.getenv('INUNDATIONHOME')
 scenariodir = add_directories(home, ["data","mem_time_test", "scenarios",
                                      "stage"])
+store = 'store.txt'
+file_path_store = os.path.join(scenariodir, store)
+storen = 'storen.txt'
+file_path_storen = os.path.join(scenariodir, storen)
 
+#read the values from the text files
+f = open(file_path_store,'r+')
+l = float(f.readline())
+f.close()
+f = open(file_path_storen,'r+')
+tide = float(f.readline())
+f.close()
 
+#set up variables to store the output of the simulation and the log files
+scenariodirV = add_directories(home, ["data","mem_time_test", "scenarios",
+                                       "stage", "stage-" + str(tide) +"-"+ str(l)])
+log.log_filename = os.path.join(scenariodirV, "anuga.log")
+log._setup = False
 
+log.resource_usage_timing(prefix = 'BeforeSimulation')# get memory usage
 #------------------------------------------------------------------------------
 # Setup computational domain
 #------------------------------------------------------------------------------
-def runex(l,tide):
+points, vertices, boundary = anuga.rectangular_cross(100,300,
+                                                     len1=l, len2=(l/2)) # Mesh
+domain = anuga.Domain(points, vertices, boundary) # Create domain
+domain.set_datadir(scenariodirV) 
+domain.set_name('channel1') # Output name
 
-   points, vertices, boundary = anuga.rectangular_cross(50,50,
-                                                     len1=l, len2=(l/2)) # Mesh
-
+log.resource_usage_timing(prefix = 'AfterMesh')# get memory usage
 
 #--------------------------------------------------------------------------
-# Setup Domain only on processor 0
+# Setup initial conditions
 #--------------------------------------------------------------------------
+def topography(x, y):
+    return  0 #flat land
+domain.set_quantity('elevation', topography) # Use function for elevation
+domain.set_quantity('friction', 0.01) # Constant friction
+domain.set_quantity('stage', tide)
 
-   domain = anuga.Domain(points, vertices, boundary) # Create domain
-   domain.set_datadir(scenariodir) 
-   domain.set_name('channel1') # Output name
-
-   def topography(x, y):
-       return  0 #-x/10  linear bed slope
-   domain.set_quantity('elevation', topography) # Use function for elevation
-   domain.set_quantity('friction', 0.01) # Constant friction
-   domain.set_quantity('stage', tide)
-    
+log.resource_usage_timing(prefix='afterinitialconditions') # get memory usage
 
 #------------------------------------------------------------------------------
 # Setup boundary conditions
 #------------------------------------------------------------------------------
-#Bi = anuga.Dirichlet_boundary([0.4, 0, 0]) # Inflow 
-   Br = anuga.Reflective_boundary(domain) # Solid reflective wall
-   domain.set_boundary({'left': Br, 'right': Br, 'top': Br, 'bottom': Br})
+Br = anuga.Reflective_boundary(domain) # Solid reflective wall
+domain.set_boundary({'left': Br, 'right': Br, 'top': Br, 'bottom': Br})
+
+log.resource_usage_timing(prefix='afterboundary')# get memory usage
 
 #------------------------------------------------------------------------------
 # Evolve system through time
 #------------------------------------------------------------------------------
-   for t in domain.evolve(yieldstep=0.2, finaltime=40.0):
-       domain.write_time()
+for t in domain.evolve(yieldstep=0.2, finaltime=40.0):
+    domain.write_time()
 
-#domain.sww_merge()
-#finalize()
-
+log.resource_usage_timing(prefix='aftersimulation')# get memory usage