Changeset 8326 for trunk/anuga_work/development/mem_time_tests/triangles/rectanglecross/runcairns.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/triangles/rectanglecross/runcairns.py (modified) (1 diff)

trunk/anuga_work/development/mem_time_tests/triangles/rectanglecross/runcairns.py

@@ -1 @@
-"""Script for running a tsunami inundation scenario for Cairns, QLD Australia.
-
-Source data such as elevation and boundary data is assumed to be available in
-directories specified by project.py
-The output sww file is stored in directory named after the scenario, i.e
-slide or fixed_wave.
-
-The scenario is defined by a triangular mesh created from project.polygon,
-the elevation data and a tsunami wave generated by a submarine mass failure.
-
-Geoscience Australia, 2004-present
-"""
-
-
-
 #------------------------------------------------------------------------------
 # Import necessary modules
 #------------------------------------------------------------------------------
-# Standard modules
 import os
 import time
 import sys
-import random
-
-# Related major packages
 import anuga
 from anuga.abstract_2d_finite_volumes.util import add_directories
+from anuga.utilities import log
 
+#set up the variables for the temporary data files
 home = os.getenv('INUNDATIONHOME')
 scenariodir = add_directories(home, ["data","mem_time_test", "triangles",
                                      "rectanglecross"])
+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 matrix size(number of triangles) and the map size from these files
+f = open(file_path_store,'r+')
+a = int(f.readline())
+f.close()
+f = open(file_path_storen,'r+') 
+l = float(f.readline())
+f.close()
 
-file = 'CAIRNS.sww'
-file_path = os.path.join(scenariodir, file)
+#set up the variables for the simulation out put and log files
+scenariodirV = add_directories(home, ["data","mem_time_test", "triangles",
+                                       "rectanglecross", "triangles-" + str(a) +"-"+ str(l)])
+log.log_filename = os.path.join(scenariodirV, "anuga.log")
+log._setup = False
 
-h = 'CAIRNS.msh'
-file_pathh = os.path.join(scenariodir, h)
-
-
-def runex(l,a):
-
-
+log.resource_usage_timing(prefix = 'BeforeSimulation')#get memory usage
 #------------------------------------------------------------------------------
-# Create the triangular mesh and domain based on 
-# overall clipping polygon with a tagged
-# boundary and interior regions as defined in project.py
+# Create the triangular mesh and domain
 #------------------------------------------------------------------------------
-    
-    points, vertices, boundary = anuga.rectangular_cross(a, a,len1 = l,len2 = l) # Basic mesh
-    domain = anuga.Domain(points, vertices, boundary) # Create domain
-
-    number=len(domain)
-
-                               
-#------------------------------------------------------------------------------
-# Setup parameters of computational domain
-#------------------------------------------------------------------------------
-    domain.set_name('CAIRNS.sww') # Name of sww file
-    domain.set_datadir(scenariodir)                       # Store sww output here
-
+points, vertices, boundary = anuga.rectangular_cross(a, a,len1 = l,len2 = l) # Basic mesh
+domain = anuga.Domain(points, vertices, boundary) # Create 
+domain.set_name('CAIRNS.sww') # Name of sww file
+domain.set_datadir(scenariodirV)# Store sww output here
+log.resource_usage_timing(prefix = 'AfterMesh') #get memory usage 
 #------------------------------------------------------------------------------
 # Setup initial conditions
 #------------------------------------------------------------------------------
 
-    def topography(x,y):
-        return 0.0
+#get the number of triangles
+number=len(domain)
 
-    tide = 100.0
-    friction = 0.0
-    domain.set_quantity('stage', tide)
-    domain.set_quantity('friction', friction) 
-    domain.set_quantity('elevation',topography,alpha=0.1)
+def topography(x,y):
+    return 0.0
 
+tide = 100.0
+friction = 0.0
+domain.set_quantity('stage', tide)
+domain.set_quantity('friction', friction) 
+domain.set_quantity('elevation',topography,alpha=0.1)
 
-
+log.resource_usage_timing(prefix='afterinitialconditions')#get memory usage
 #------------------------------------------------------------------------------
 # Setup boundary conditions
 #------------------------------------------------------------------------------
+Bi = anuga.Dirichlet_boundary([tide, 223.52, 0]) # inflow
+Bo = anuga.Dirichlet_boundary([-tide, 223.52, 0]) # outflow
+Br = anuga.Reflective_boundary(domain)
+domain.set_boundary({'right': Bo,'bottom': Br,'left': Bi,'top': Br})
 
-    Bi = anuga.Dirichlet_boundary([tide, 223.52, 0]) # inflow
-    Bo = anuga.Dirichlet_boundary([-tide, 223.52, 0]) # inflow
-    Bs = anuga.Transmissive_stage_zero_momentum_boundary(domain) # Neutral boundary
-    Br = anuga.Reflective_boundary(domain)
-#Bw = anuga.Time_boundary(domain=domain,function=lambda t: [(60<t<3660)*50, 0, 0])
-    domain.set_boundary({'right': Bo,
-                     'bottom': Br,
-                     'left': Bi,
-                     'top': Br})
-
+log.resource_usage_timing(prefix='afterboundary')#get memory usage
 #------------------------------------------------------------------------------
 # Evolve system through time
 #------------------------------------------------------------------------------
+for t in domain.evolve(yieldstep=120, finaltime=2000): 
+    print domain.timestepping_statistics()
 
-# Save every two mins leading up to wave approaching land
-    for t in domain.evolve(yieldstep=120, finaltime=2000): 
-        print domain.timestepping_statistics()
-    
-
-    return number
-
+log.resource_usage_timing(prefix='aftersimulation')#get memory usage
+   
+#write the number of triangles to the text file
+v = open(file_path_storea, 'r+')
+v.write(str(number))