source: trunk/anuga_validation/performance_tests/run_cross_section_flow_profile.py @ 8288

"""Example of shallow water wave equation.

This version is for profiling of get_flow_through_cross_section
"""

######################
# Module imports
from anuga.shallow_water import Domain, Reflective_boundary
from anuga.shallow_water.data_manager import get_flow_through_cross_section
from mesh_factory import rectangular
from Numeric import array


######################
# Domain

N = 16   #Should take less than 0.1 s
#N = 64  #Should take less than 3s
#N = 128  #Should take less than 20s


print 'Creating domain'
#Create basic mesh
points, vertices, boundary = rectangular(N, N)

#Create shallow water domain
domain = Domain(points, vertices, boundary)
domain.set_default_order(2)        
domain.set_name('test_file')

print 'Setting initial conditions'

def slope(x, y):
    return -x

domain.set_quantity('elevation', slope)
domain.set_quantity('friction', 0.03)
domain.set_quantity('stage', expression = 'elevation + 0.5')



print 'Setting boundary conditions'
Br = Reflective_boundary(domain)
domain.set_boundary({'left': Br, 'right': Br, 'bottom': Br, 'top': Br})



######################
#Evolution

import time
for t in domain.evolve(yieldstep = 0.02, finaltime = 0.2):
    domain.write_time()

print 'Doing cross section'
t0 = time.time()

#--------------
# Cross section
swwfile = domain.get_name() + '.sww'

s = 't, Q = get_flow_through_cross_section(swwfile, [[0.5, 0.1],[0.7, 0.9]], verbose=True)'
import profile, pstats
FN = 'profile.dat'
profile.run(s, FN)

print 'That took %.2f seconds' %(time.time()-t0)

S = pstats.Stats(FN)
print S.sort_arg_dict_default

s = S.sort_stats('cumulative').print_stats(30)
#s = S.sort_stats('time').print_stats(30)
print s