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

trunk/anuga_work/development/mem_time_tests/scenarios/vel2/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 files
 home = os.getenv('INUNDATIONHOME')
-scenariodir = add_directories(home, ["data","mem_time_test", "triangles",
-                                     "serial"])
+scenariodir = add_directories(home, ["data","mem_time_test", "scenarios",
+                                     "velocity"])
+store ='store.txt'
+file_path_store = os.path.join(scenariodir, store)
+storen = 'storen.txt'
+file_path_storen = os.path.join(scenariodir, storen)
+storea = 'storea.txt'
+file_path_storea = os.path.join(scenariodir, storea)
 
+# read the velocity and the map side length from the text files
+f = open(file_path_store,'r+')
+k = float(f.readline())
+f.close()
+f = open(file_path_storen,'r+')
+l = float(f.readline())
+f.close()
+
+#set up variables for the simulation output and the log files
+scenariodirV = add_directories(home, ["data","mem_time_test", "scenarios",
+                                       "velocity", "velocity-" + str(k) +"-"+ 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(k,l):
+points, vertices, boundary = anuga.rectangular_cross(100,300, len1=1000, len2=500) # Mesh
+domain = anuga.Domain(points, vertices, boundary) # Create domain
+domain.set_name('channel1') # Output name
+domain.set_datadir(scenariodirV) 
 
-   points, vertices, boundary = anuga.rectangular_cross(100,100,
-                                                       len1=l, len2=l) # Mesh
+log.resource_usage_timing(prefix = 'AfterMesh')  #get memory usage 
+
+#get the number of triangles
+number = len(domain)
 
 #--------------------------------------------------------------------------
-# Setup Domain only on processor 0
+# Setup initial conditions
 #--------------------------------------------------------------------------
-   domain = anuga.Domain(points, vertices, boundary) # Create domain
-   domain.set_name('channel1') # Output name
-   domain.set_datadir(scenariodir) 
-   
-   number = len(domain)
+def topography(x, y):
+    return  0.0
+domain.set_quantity('elevation', topography) # Use function for elevation
+domain.set_quantity('friction', 0.01) # Constant friction
+domain.set_quantity('stage', -1000.0)
 
-   def topography(x, y):
-       return  0.0
-
-   domain.set_quantity('elevation', topography) # Use function for elevation
-   domain.set_quantity('friction', 0.01) # Constant friction
-   domain.set_quantity('stage', -1000.0)
+log.resource_usage_timing(prefix='afterinitialconditions')#get memory usage
 
 #------------------------------------------------------------------------------
 # Setup boundary conditions
 #------------------------------------------------------------------------------
-   Bi = anuga.Dirichlet_boundary([0.5, k, 0]) # Inflow 
-   Br = anuga.Reflective_boundary(domain) # Solid reflective wall
-   Bo = anuga.Dirichlet_boundary([-0.5, 0, 0]) # Outflow
-   domain.set_boundary({'left': Bi, 'right': Bo, 'top': Br, 'bottom': Br})
+Bi = anuga.Dirichlet_boundary([0.5, k, 0]) # Inflow 
+Br = anuga.Reflective_boundary(domain) # Solid reflective wall
+Bo = anuga.Dirichlet_boundary([-0.5, 0, 0]) # Outflow
+domain.set_boundary({'left': Bi, 'right': Bo, '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()
 
-   return number
+log.resource_usage_timing(prefix='aftersimulation')  #get memory usage
 
+#write the number of triangles to the file
+c = open(file_path_storea,'r+')
+c.write(str(number))