source: inundation/ga/storm_surge/pyvolution/flatbed_compare.py @ 474

"""Example of shallow water wave equation.

Comparison of flatbed between mark 2(Grohm) and mark 3

"""

######################
# Module imports 
#
from mesh_factory import rectangular
from shallow_water import Domain, Reflective_boundary, Dirichlet_boundary,\
     Transmissive_boundary, Constant_height
from Numeric import array

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

#Create shallow water domain
domain = Domain(points, vertices, boundary)

if N < 20:
    domain.visualise = True
else:
    domain.visualise = False
    domain.store = True
    domain.format = 'sww'
    domain.filename = 'compare_py3'

#domain.visualise = False
domain.smooth = False
domain.default_order = 1


print 'Field values'
domain.set_quantity('elevation', 0.0)
domain.set_quantity('friction', 1)


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

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

print domain.quantities['elevation'].centroid_values[:4]
print domain.quantities['friction'].centroid_values[:4]

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

#print domain.quantities['level'].centroid_values
#print domain.quantities['xmomentum'].centroid_values
#print domain.quantities['ymomentum'].centroid_values    

#print 'R'
#print domain.quantities['level'].edge_values