source: anuga_work/development/gong_2008/export_results.py @ 6868

import project, os
import sys
from anuga.shallow_water.data_manager import sww2dem
from anuga.shallow_water.data_manager import convert_dem_from_ascii2netcdf, dem2pts
from os import sep

time_dir = ''

cellsize = 25
name = project.outputdir+time_dir+sep+project.basename

var = [2,3,4] # depth, speed and elevation

for which_var in var:
    if which_var == 0:  # Stage
        outname = name + '_stage'
        quantityname = 'stage'

    if which_var == 1:  # Absolute Momentum
        outname = name + '_momentum'
        quantityname = '(xmomentum**2 + ymomentum**2)**0.5'  

    if which_var == 2:  # Depth
        outname = name + '_depth'
        quantityname = 'stage-elevation'  

    if which_var == 3:  # Speed
        outname = name + '_speed'
        quantityname = '(xmomentum**2 + ymomentum**2)**0.5/(stage-elevation+1.e-6/(stage-elevation))'  #Speed

    if which_var == 4:  # Elevation
        outname = name + '_elevation'
        quantityname = 'elevation'  #Elevation

    print 'start sww2dem'
    sww2dem(name, basename_out = outname,
                quantity = quantityname,
                #timestep = timestep,
                cellsize = cellsize,      
                easting_min = project.eastingmin,
                easting_max = project.eastingmax,
                northing_min = project.northingmin,
                northing_max = project.northingmax,        
                reduction = max, 
                verbose = True,
                format = 'asc')