source: anuga_work/production/perth_2006/run_perth.py @ 4152

"""Script for running tsunami inundation scenario for Perth, WA, 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 project.output_run_time_dir 

The scenario is defined by a triangular mesh created from project.polygon,
the elevation data and a simulated tsunami generated with URS code.

Ole Nielsen and Duncan Gray, GA - 2005 and Jane Sexton, Nick Bartzis, GA - 2006
"""

#------------------------------------------------------------------------------
# Import necessary modules
#------------------------------------------------------------------------------

# Standard modules
from os import sep
from os.path import dirname, basename
from os import mkdir, access, F_OK
from shutil import copy
import time
import sys

# Related major packages
from anuga.shallow_water import Domain
from anuga.shallow_water import Dirichlet_boundary
from anuga.shallow_water import File_boundary
from anuga.shallow_water import Reflective_boundary
from Numeric import allclose

from anuga.pmesh.mesh_interface import create_mesh_from_regions
from anuga.geospatial_data.geospatial_data import *
from anuga.abstract_2d_finite_volumes.util import start_screen_catcher, copy_code_files
from anuga_parallel.parallel_api import distribute, numprocs, myid, barrier
from anuga.utilities.polygon import plot_polygons, polygon_area

# Application specific imports
import project                 # Definition of file names and polygons

#------------------------------------------------------------------------------
# Copy scripts to time stamped output directory and capture screen
# output to file
#------------------------------------------------------------------------------

start_screen_catcher(project.output_run_time_dir, myid, numprocs)

# filenames
boundaries_name = project.scenario
meshes_dir_name = project.meshes_dir_name+'.msh'
#boundaries_time_dir_name = project.boundaries_time_dir_name
boundaries_dir_name = project.boundaries_dir_name

tide = project.tide

# creates copy of code in output dir
if myid == 0:
    copy_code_files(project.output_run_time_dir,__file__, 
                 dirname(project.__file__)+sep+ project.__name__+'.py' )
barrier()

print 'USER: ', project.user
print 'min triangles', project.trigs_min,
print 'Note: This is generally about 20% less than the final amount'

#--------------------------------------------------------------------------
# Create the triangular mesh based on overall clipping polygon with a
# tagged
# boundary and interior regions defined in project.py along with
# resolutions (maximal area of per triangle) for each polygon
#--------------------------------------------------------------------------

if myid == 0:

    print 'start create mesh from regions'
#    interior_regions = [[project.poly_pos20_neg20, project.res_pos20_neg20],
#                    [project.poly_cbd, project.res_cbd],
#                    [project.poly_penguin, project.res_penguin]]    
#                    [project.poly_interior, 1000]]       

#    from anuga.utilities.polygon import plot_polygons
#    figname = project.output_run_time_dir + 'poly_pic'
#    plot_polygons([project.poly_coast,project.poly_pipeline,project.poly_facility,
#               project.bounding_polygon],
#               figname,
#               verbose = True)
#    import sys; sys.exit()

    print 'start create mesh from regions'
    create_mesh_from_regions(project.poly_all,
                         boundary_tags={'back': [4], 'side': [0,3],
                                        'ocean': [1, 2]},
                         maximum_triangle_area=project.res_poly_all,
                         interior_regions=project.interior_regions,
#                         filename=meshes_time_dir_name,
                         filename=meshes_dir_name,
                         use_cache=True,
                         verbose=True)

# to sync all processors are ready 
barrier()

#-------------------------------------------------------------------------
# Setup computational domain
#-------------------------------------------------------------------------
print 'Setup computational domain'
#domain = Domain(meshes_time_dir_name, use_cache=True, verbose=True)
domain = Domain(meshes_dir_name, use_cache=True, verbose=True)
print domain.statistics()
'''
print 'starting to create boundary conditions'
boundaries_in_dir_name = project.boundaries_in_dir_name

from anuga.shallow_water.data_manager import urs2sww, ferret2sww

# put above distribute
print 'boundary file is: ',boundaries_dir_name
from caching import cache
if myid == 0:
    cache(ferret2sww,
          (boundaries_in_dir_name,
    #       boundaries_time_dir_name), 
           boundaries_dir_name), 
          {'verbose': True,
           'minlat': project.south_boundary,
           'maxlat': project.north_boundary,
           'minlon': project.west_boundary,
           'maxlon': project.east_boundary,
#           'minlat': project.south,
#           'maxlat': project.north,
#           'minlon': project.west,
#           'maxlon': project.east,
           'mint': 0, 'maxt': 35100,
           'origin': domain.geo_reference.get_origin(),
           'mean_stage': project.tide,
#           'zscale': 1,                 #Enhance tsunami
           'fail_on_NaN': False},
           verbose = True,
           )
barrier()
'''

#-------------------------------------------------------------------------
# Setup initial conditions
#-------------------------------------------------------------------------
if myid == 0:

    print 'Setup initial conditions'

    domain.set_quantity('stage', tide)
    domain.set_quantity('friction', 0.01) 
    #combined_time_dir_name = project.topographies_dir+build_time+project.combined_name
    print 'Start Set quantity'

    domain.set_quantity('elevation', 
#                    filename = project.combined_dir_name+'.txt',
                    filename = project.combined_smaller_name_dir+'.txt',
                    use_cache = True,
                    verbose = True,
                    alpha = 0.1)
    print 'Finished Set quantity'
barrier()

#------------------------------------------------------
# Distribute domain to implement parallelism !!!
#------------------------------------------------------

if numprocs > 1:
    domain=distribute(domain)

#------------------------------------------------------
# Set domain parameters
#------------------------------------------------------

domain.set_name(project.scenario_name)
domain.set_datadir(project.output_run_time_dir)
domain.set_default_order(2) # Apply second order scheme
domain.set_minimum_storable_height(0.01) # Don't store anything less than 1cm
domain.set_store_vertices_uniquely(False)
domain.set_quantities_to_be_stored(['stage', 'xmomentum', 'ymomentum'])
domain.set_maximum_allowed_speed(0.1) # Allow a little runoff (0.1 is OK)

#-------------------------------------------------------------------------
# Setup boundary conditions 
#-------------------------------------------------------------------------
print 'Available boundary tags', domain.get_boundary_tags()

print 'Reading Boundary file'
print 'domain id', id(domain)
#Bf = File_boundary(boundaries_dir_name + '.sww',
#                  domain, time_thinning=5, use_cache=True, verbose=True)

print 'finished reading boundary file'
Br = Reflective_boundary(domain)
Bd = Dirichlet_boundary([tide,0,0])
Bo = Dirichlet_boundary([tide+5.0,0,0])

print'set_boundary'
domain.set_boundary({'back': Br,
                     'side': Bd,
                     'ocean': Bd}) 
print'finish set boundary'

#----------------------------------------------------------------------------
# Evolve system through time
#----------------------------------------------------------------------------

t0 = time.time()

for t in domain.evolve(yieldstep = 60, finaltime = 10000):
    domain.write_time()
    domain.write_boundary_statistics(tags = 'ocean')
    if allclose(t, 120):
        domain.set_boundary({'back': Br, 'side': Bd, 'ocean': Bo})

    if allclose(t, 1020):
        domain.set_boundary({'back': Br, 'side': Bd, 'ocean': Bd})

    
print 'That took %.2f seconds' %(time.time()-t0)