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

trunk/anuga_work/development/mem_time_tests/parameters/timestep/ex1.py

@@ -1 @@
-"""Simple water flow example using ANUGA
-Water flowing down a channel
-"""
 #------------------------------------------------------------------------------
 # Import necessary modules
@@ -10 +7 @@
 import random
 import os
+from anuga.abstract_2d_finite_volumes.util import add_directories
+from anuga.utilities import log
+
+#set up the variables for the temporary file directories
+home = os.getenv('INUNDATIONHOME')
+scenariodir = add_directories(home, ["data","mem_time_test", "parameters","timestep"])
+
+store ='store.txt'
+file_path_store = os.path.join(scenariodir, store)
+
+s = open(file_path_store,'r+') # read the timestep from the text file
+f = float(s.readline())
+s.close()
+
+#set up the variables for the log files and data directories
+scenariodirV = add_directories(home, ["data","mem_time_test", "parameters",
+                                       "timestep", "timestep-" + str(f)])
+log.log_filename = os.path.join(scenariodirV, "anuga.log")
+log._setup = False
+
+log.resource_usage_timing(prefix = 'BeforeSimulation') #get memory usage here
 #------------------------------------------------------------------------------
 # Setup computational domain
 #------------------------------------------------------------------------------
-from anuga.abstract_2d_finite_volumes.util import add_directories
+points, vertices, boundary = anuga.rectangular_cross(10,5,len1=10.0, len2=5.0) # Mesh
+domain = anuga.Domain(points, vertices, boundary) # Create domain
+domain.set_name('channel1') # Output name
+domain.set_datadir(scenariodirV) 
 
-home = os.getenv('INUNDATIONHOME')
-scenariodir = add_directories(home, ["data","mem_time_test", "parameters",
-                                     "timestep"])
-
-def runex(f):
-   points, vertices, boundary = anuga.rectangular_cross(10,5,
-   len1=10.0, len2=5.0) # Mesh
-   domain = anuga.Domain(points, vertices, boundary) # Create domain
-   domain.set_name('channel1') # Output name
-   domain.set_datadir(scenariodir) 
+log.resource_usage_timing(prefix = 'AfterMesh')  #get memory usage here
 #------------------------------------------------------------------------------
 # Setup initial conditions
 #------------------------------------------------------------------------------
-   def topography(x, y):
-       return -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', # Dry bed
-                           expression='elevation')
+def topography(x, y):
+    return -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', expression='elevation')
+
+log.resource_usage_timing(prefix='afterinitialconditions')#get memory usage here
 
 #------------------------------------------------------------------------------
 # Setup boundary conditions
 #------------------------------------------------------------------------------
-   Bi = anuga.Dirichlet_boundary([0.4, 0, 0]) # Inflow
-   Br = anuga.Reflective_boundary(domain) # Solid reflective wall
-   domain.set_boundary({'left': Bi, 'right': Br, 'top': Br, 'bottom': Br})
+Bi = anuga.Dirichlet_boundary([0.4, 0, 0]) # Inflow
+Br = anuga.Reflective_boundary(domain) # Solid reflective wall
+domain.set_boundary({'left': Bi, 'right': Br, 'top': Br, 'bottom': Br})
 
+log.resource_usage_timing(prefix='afterboundary') #get memory usage here
 #------------------------------------------------------------------------------
 # Evolve system through time
 #------------------------------------------------------------------------------
-   for t in domain.evolve(yieldstep=f, finaltime=40.0):
-      print domain.timestepping_statistics()
+for t in domain.evolve(yieldstep=f, finaltime=40.0):
+    print domain.timestepping_statistics()
 
+log.resource_usage_timing(prefix='aftersimulation') #get memory usage here