source: inundation/pyvolution/run_profile.py @ 2126

Last change on this file since 2126 was 1629, checked in by steve, 19 years ago
File size: 1.4 KB
Line 
1"""Example of shallow water wave equation.
2
3This version is for profiling
4(The original)
5"""
6
7######################
8# Module imports
9#
10from shallow_water import Domain, Reflective_boundary, Dirichlet_boundary,\
11     Transmissive_boundary, Time_boundary,\
12     Weir_simple as Weir, Constant_height
13
14from mesh_factory import rectangular
15from Numeric import array
16
17
18######################
19# Domain
20#
21
22N = 128
23
24print 'Creating domain'
25#Create basic mesh
26points, vertices, boundary = rectangular(N, N)
27
28#Create shallow water domain
29domain = Domain(points, vertices, boundary)
30domain.default_order = 2
31#domain.visualise = True
32
33
34print 'Field values'
35def slope(x, y):
36    return -x
37
38domain.set_quantity('elevation', slope)
39domain.set_quantity('friction', 0.3)
40
41
42# Boundary conditions
43#
44Br = Reflective_boundary(domain)
45domain.set_boundary({'left': Br, 'right': Br, 'bottom': Br, 'top': Br})
46
47
48######################
49#Initial condition
50domain.set_quantity('stage', Constant_height(slope, 0.3))
51
52domain.check_integrity()
53
54
55######################
56#Evolution
57
58import time
59t0 = time.time()
60
61s = 'for t in domain.evolve(yieldstep = 0.02, finaltime = 0.2): domain.write_time()'
62
63
64
65import profile, pstats
66FN = 'profile.dat'
67
68profile.run(s, FN)
69
70print 'That took %.2f seconds' %(time.time()-t0)
71
72S = pstats.Stats(FN)
73#S.sort_stats('time').print_stats(20)
74s = S.sort_stats('cumulative').print_stats(30)
75
76print s
Note: See TracBrowser for help on using the repository browser.