source: anuga_work/production/newcastle_2006/run_newcastle_slide.py @ 4186

"""Script for running a tsunami inundation scenario for Newcastle, NSW, Australia.

Source data such as elevation and boundary data is assumed to be available in
directories specified by project_slide.py
The output sww file is stored in project_slide.outputtimedir 

The scenario is defined by a triangular mesh created from project_slide.polygon,
the elevation data and a tsunami wave generated by s submarine mass failure.

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

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

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

# Related major packages
from anuga.shallow_water import Domain, Reflective_boundary, Dirichlet_boundary 
from anuga.shallow_water.data_manager import convert_dem_from_ascii2netcdf, dem2pts
from anuga.geospatial_data.geospatial_data import *
from anuga.abstract_2d_finite_volumes.util import start_screen_catcher, copy_code_files

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

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

# creates copy of code in output dir 
copy_code_files(project_slide.outputtimedir,__file__,dirname(project_slide.__file__)+sep+ project_slide.__name__+'.py' )
myid = 0
numprocs = 1
start_screen_catcher(project_slide.outputtimedir, myid, numprocs)

print 'USER:    ', project_slide.user

#-------------------------------------------------------------------------------
# Preparation of topographic data
#
# Convert ASC 2 DEM 2 PTS using source data and store result in source data
#-------------------------------------------------------------------------------

# filenames
nsw_dem_name = project_slide.nsw_dem_name
meshname = project_slide.meshname+'.msh'

# creates DEM from asc data
convert_dem_from_ascii2netcdf(nsw_dem_name, use_cache=True, verbose=True)

#creates pts file for onshore DEM
dem2pts(nsw_dem_name,
        easting_min=project_slide.eastingmin_nsw,
        easting_max=project_slide.eastingmax_nsw,
        northing_min=project_slide.northingmin_nsw,
        northing_max= project_slide.northingmax_nsw,
        use_cache=True, verbose=True)

print 'create offshore'
G11 = Geospatial_data(file_name = project_slide.offshore_dem_name2 + '.xya')+\
      Geospatial_data(file_name = project_slide.offshore_dem_name3 + '.xya')
G12 = Geospatial_data(file_name = project_slide.offshore_dem_name4 + '.xya')+\
      Geospatial_data(file_name = project_slide.offshore_dem_name5 + '.xya')+\
      Geospatial_data(file_name = project_slide.offshore_dem_name6 + '.xya')+\
      Geospatial_data(file_name = project_slide.offshore_dem_name7 + '.xya')+\
      Geospatial_data(file_name = project_slide.offshore_dem_name8 + '.xya')+\
      Geospatial_data(file_name = project_slide.offshore_dem_name9 + '.xya')
print 'create onshore'
G4 = Geospatial_data(file_name = project_slide.nsw_dem_name + '.pts')
print 'add'
G = G11.clip(Geospatial_data(project_slide.poly_surveyclip)) +\
    G12.clip(Geospatial_data(project_slide.polyAll)) +\
    (G4.clip(Geospatial_data(project_slide.polyAll)).clip_outside(Geospatial_data(project_slide.poly_surveyclip)))
print 'export points'
G.export_points_file(project_slide.combined_dem_name + '.pts')
#G.export_points_file(project_slide.combined_dem_name + '.xya')


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

from anuga.pmesh.mesh_interface import create_mesh_from_regions
remainder_res = 500000
local_res = 50000
newcastle_res = 1000
interior_regions = [[project_slide.poly_local, local_res],
                    [project_slide.poly_newcastle, newcastle_res]]

from caching import cache
_ = cache(create_mesh_from_regions,
          project_slide.polyAll,
           {'boundary_tags': {'e0': [0], 'e1': [1], 'e2': [2],
                              'e3': [3], 'e4':[4], 'e5': [5]},
           'maximum_triangle_area': remainder_res,
           'filename': meshname,
           'interior_regions': interior_regions},
          verbose = True, evaluate=False)
print 'created mesh'

#-------------------------------------------------------------------------------                                 
# Setup computational domain
#-------------------------------------------------------------------------------                                 
domain = Domain(meshname, use_cache = True, verbose = True)

print 'Number of triangles = ', len(domain)
print 'The extent is ', domain.get_extent()
print domain.statistics()

domain.set_name(project_slide.basename)
domain.set_datadir(project_slide.outputtimedir)
domain.set_quantities_to_be_stored(['stage', 'xmomentum', 'ymomentum'])
domain.set_minimum_storable_height(0.01)

#-------------------------------------------------------------------------------                                 
# Setup initial conditions
#-------------------------------------------------------------------------------

tide = 0.0
domain.set_quantity('stage', tide)
domain.set_quantity('friction', 0.0) 
domain.set_quantity('elevation', 
                    filename = project_slide.combined_dem_name + '.pts',
                    use_cache = True,
                    verbose = True,
                    alpha = 0.1
                    )

#-------------------------------------------------------------------------------
# Set up scenario (tsunami_source is a callable object used with set_quantity)
#-------------------------------------------------------------------------------
from smf import slide_tsunami

tsunami_source = slide_tsunami(length=project_slide.yacaaba_length,
                               width=project_slide.yacaaba_width,
                               depth=project_slide.yacaaba_depth,
                               slope=project_slide.yacaaba_slope,
                               thickness=project_slide.yacaaba_thickness, 
                               x0=project_slide.slide_origin_yacaaba_a[0], 
                               y0=project_slide.slide_origin_yacaaba_a[1], 
                               alpha=project_slide.yacaaba_alpha, 
                               domain=domain)

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

Br = Reflective_boundary(domain)
Bd = Dirichlet_boundary([tide,0,0])

domain.set_boundary( {'e0': Bd,  'e1': Bd, 'e2': Bd, 'e3': Bd,
                      'e4': Bd, 'e5': Bd} )


#-------------------------------------------------------------------------------                                 
# Evolve system through time
#-------------------------------------------------------------------------------
import time
t0 = time.time()
from Numeric import allclose
from anuga.abstract_2d_finite_volumes.quantity import Quantity

for t in domain.evolve(yieldstep = 30, finaltime = 5000): 
    domain.write_time()
    domain.write_boundary_statistics(tags = 'e14')
    stagestep = domain.get_quantity('stage') 

    if allclose(t, 30):
        slide = Quantity(domain)
        slide.set_values(tsunami_source)
        domain.set_quantity('stage', slide + stagestep)
    
print 'That took %.2f seconds' %(time.time()-t0)

print 'finished'