1 | """Simple water flow example using ANUGA |
---|
2 | Water flowing down a channel |
---|
3 | """ |
---|
4 | #------------------------------------------------------------------------------ |
---|
5 | # Import necessary modules |
---|
6 | #------------------------------------------------------------------------------ |
---|
7 | # Import standard shallow water domain and standard boundaries. |
---|
8 | import anuga |
---|
9 | import time |
---|
10 | import random |
---|
11 | import subprocess |
---|
12 | import ex1 |
---|
13 | import csv |
---|
14 | import os |
---|
15 | from anuga.utilities import system_tools, log |
---|
16 | from anuga.abstract_2d_finite_volumes.util import add_directories |
---|
17 | #------------------------------------------------------------------------------ |
---|
18 | # Setup computational domain |
---|
19 | #------------------------------------------------------------------------------ |
---|
20 | a = [] |
---|
21 | home = os.getenv('INUNDATIONHOME') |
---|
22 | |
---|
23 | |
---|
24 | scenariodir = add_directories(home, ["data","mem_time_test", "parameters", |
---|
25 | "timelen-over-timestep"]) |
---|
26 | file = 'ex1.csv' |
---|
27 | |
---|
28 | log.log_filename = os.path.join(scenariodir, 'log.txt') |
---|
29 | |
---|
30 | file_path = os.path.join(scenariodir, file) |
---|
31 | |
---|
32 | spamWriter = csv.writer(open(file_path, 'wb')) |
---|
33 | |
---|
34 | spamWriter.writerow(['Time Length', 'Time Step' , 'Time Taken','Space Used']) |
---|
35 | |
---|
36 | for m in range(1,1000,50): |
---|
37 | |
---|
38 | for n in range(1,1000,5): |
---|
39 | n = n/1000.0 |
---|
40 | system_tools.MemoryUpdate() |
---|
41 | x = time.clock() |
---|
42 | ex1.runex(float(m),n) |
---|
43 | y = time.clock() |
---|
44 | b = system_tools.MemoryUpdate()[0] |
---|
45 | spamWriter.writerow([m,n ,(y-x),b]) |
---|
46 | a.append((n,y-x,b)) |
---|
47 | print a |
---|
48 | |
---|