source: inundation/ga/storm_surge/pyvolution/flatbed.py @ 660

"""Example of shallow water wave equation.

Generate slope

"""

######################
# Module imports 
#
from os import sep, path
from mesh_factory import rectangular
from shallow_water import Domain, Reflective_boundary, Dirichlet_boundary,\
     Constant_height
from Numeric import array
from util import Polygon_function, read_polygon
from config import data_dir

#Create basic mesh
N = 20
points, vertices, boundary = rectangular(N, N, 100, 100)

#Create shallow water domain
domain = Domain(points, vertices, boundary)
domain.store = True
domain.set_name('polygons')
print "Output being written to " + data_dir + sep + \
              domain.filename + "." + domain.format
        

domain.default_order=2

#Set driving forces
manning = 0.07
manning = 0.0
inflow_level = 10.0
domain.set_quantity('friction', manning)




#Define polynomials
p0 = [[20,27], [30,25], [40,40], [20,40]]         
p1 = [[80,19], [90,20], [85,50], [80,55], [75,58], [70,60], [60,24]]         
p2 = read_polygon('testpoly.txt')


#Set elevation  
domain.set_quantity('elevation', 
        Polygon_function([(p0, 23), (p1,10), (p2,15)]))


          



######################
# Boundary conditions
Br = Reflective_boundary(domain)
Bd = Dirichlet_boundary([inflow_level,0.,0.])

domain.set_boundary({'left': Bd, 'right': Br, 'top': Br, 'bottom': Br})
domain.check_integrity()


######################
#Evolution
for t in domain.evolve(yieldstep = 1, finaltime = 1):
    domain.write_time()

print 'Done'