source: anuga_work/production/pt_hedland_2008/build_pt_hedland.py @ 6540

"""
Script for building the elevation data to run a tsunami inundation scenario
for pt_hedland, WA, Australia.

Input: elevation data from project.py
Output: pts file stored in project.topographies_dir 
The run_pt_hedland.py is reliant on the output of this script.

"""

#------------------------------------------------------------------------------
# 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.data_manager import convert_dem_from_ascii2netcdf, dem2pts
from anuga.geospatial_data.geospatial_data import *
from anuga.shallow_water.data_manager import start_screen_catcher, copy_code_files

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

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

copy_code_files(project.output_build_time_dir,__file__, 
               dirname(project.__file__)+sep+ project.__name__+'.py' )

start_screen_catcher(project.output_build_time_dir)

print 'USER:    ', project.user

#-------------------------------------------------------------------------------
# Preparation of topographic data
#
# Convert ASC 2 DEM 2 PTS using source data and store result in source data
# Do for coarse and fine data
# Fine pts file to be clipped to area of interest
#-------------------------------------------------------------------------------
print "project.poly_all", project.poly_all
print "project.combined_dir_name", project.combined_dir_name

# input elevation directory filenames
onshore_in_dir_name = project.onshore_in_dir_name
coast_in_dir_name = project.coast_in_dir_name
offshore_in_dir_name = project.offshore_in_dir_name


# output elevation directory filenames
onshore_dir_name = project.onshore_dir_name
coast_dir_name = project.coast_dir_name
offshore_dir_name = project.offshore_dir_name


# creates DEM from asc data
print "creates DEMs from asc data"
convert_dem_from_ascii2netcdf(onshore_in_dir_name, basename_out=onshore_dir_name, use_cache=True, verbose=True)


# creates pts file for onshore DEM
print "creates pts file for onshore DEM"
dem2pts(onshore_dir_name ,use_cache=True,verbose=True)

# creates pts file for island DEM


# create onshore pts files
print'create Geospatial data1 objects from topographies'
G1 = Geospatial_data(file_name = onshore_dir_name + '.pts')


# create coastal and offshore pts files
print'create Geospatial data2 objects from topographies'
G_coast = Geospatial_data(file_name = coast_in_dir_name)
print'create Geospatial data3 objects from topographies'
G_off = Geospatial_data(file_name = offshore_in_dir_name)

#-------------------------------------------------------------------------------
# Combine, clip and export dataset 
#-------------------------------------------------------------------------------

print'add all geospatial objects'
G = G1 + G_coast + G_off

print'clip combined geospatial object by bounding polygon'
G_clipped = G.clip(project.poly_all)

print'export combined DEM file'
if access(project.topographies_dir,F_OK) == 0:
    mkdir (project.topographies_dir)
G_clipped.export_points_file(project.combined_dir_name + '.pts')
#G_clipped.export_points_file(project.combined_dir_name + '.txt') #Use for comparision in ARC