Add benchmark script

raster/r.resamp.filter/benchmark/benchmark_r_resamp_filter.py

@@ -0,0 +1,58 @@
+"""Benchmarking of r.resamp.filter
+
+@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.resamp.filter_50M", results)
+    benchmark(10000, "r.resamp.filter_100M", results)
+    benchmark(14142, "r.resamp.filter_200M", results)
+    benchmark(20000, "r.resamp.filter_400M", results)
+
+    bm.nprocs_plot(results)
+
+
+def benchmark(size, label, results):
+    reference = "r_resamp_filter_reference_map"
+    output = "benchmark_r_resamp_filter_nprocs"
+
+    generate_map(rows=size, cols=size, fname=reference)
+    module = Module(
+        "r.resamp.filter",
+        input=reference,
+        filter=["lanczos3", "sinc"],
+        output=output,
+        radius=[10, 10],
+        memory=500,
+        run_=False,
+        stdout_=DEVNULL,
+        overwrite=True,
+    )
+    results.append(bm.benchmark_nprocs(module, label=label, max_nprocs=16, repeat=3))
+    Module("g.remove", quiet=True, flags="f", type="raster", name=reference)
+    Module("g.remove", quiet=True, flags="f", type="raster", name=output)
+
+
+def generate_map(rows, cols, fname):
+    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...")
+        Module("r.surf.fractal", output=fname)
+    except CalledModuleError:
+        print("r.surf.fractal fails, using r.random.surface instead...")
+        Module("r.random.surface", output=fname)
+
+
+if __name__ == "__main__":
+    main()