1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
|
"""Benchmarking of r.patch
@author Aaron Saw Min Sern
"""
from grass.exceptions import CalledModuleError
from grass.pygrass.modules import Module
from subprocess import DEVNULL
import grass.benchmark as bm
def main():
results = []
# Users can add more or modify existing reference maps
benchmark(7071, "r.patch_50M", results)
benchmark(14142, "r.patch_200M", results)
benchmark(10000, "r.patch_100M", results)
benchmark(20000, "r.patch_400M", results)
bm.nprocs_plot(results, filename="rpatch_benchmark.svg")
def benchmark(size, label, results):
references = ["r_patch_reference_map_{}".format(i) for i in range(4)]
output = "benchmark_r_patch"
generate_map(rows=size, cols=size, fname=references)
module = Module(
"r.patch",
input=references,
output=output,
run_=False,
stdout_=DEVNULL,
overwrite=True,
)
results.append(bm.benchmark_nprocs(module, label=label, max_nprocs=24, repeat=3))
for r in references:
Module("g.remove", quiet=True, flags="f", type="raster", name=r)
Module("g.remove", quiet=True, flags="f", type="raster", name=output)
def generate_map(rows, cols, fname):
temp = "r_patch_reference_tmp"
Module("g.region", flags="p", s=0, n=rows, w=0, e=cols, res=1)
# Generate using r.random.surface if r.surf.fractal fails
try:
print("Generating reference map using r.surf.fractal...")
for r in fname:
Module(
"r.surf.fractal",
output=temp,
overwrite=True,
)
# Make approximate half of the reference map to be null
Module(
"r.mapcalc",
expression=f"{r} = if({temp}, {temp}, 0, null())",
overwrite=True,
)
except CalledModuleError:
print("r.surf.fractal fails, using r.random.surface instead...")
for r in fname:
Module(
"r.random.surface",
high=255,
output=temp,
overwrite=True,
)
Module(
"r.mapcalc",
expression=f"{r} = if({temp} - 128, {temp}, 0, null())",
overwrite=True,
)
Module("g.remove", quiet=True, flags="f", type="raster", name=temp)
if __name__ == "__main__":
main()
|
