From 30136bf60a68a18ad02dd8174868b1051e266a45 Mon Sep 17 00:00:00 2001 From: Gael-de-Sailly Date: Tue, 10 Nov 2020 13:19:20 +0100 Subject: [PATCH] Added scale (blocksize) parameter for view_map.py --- view_map.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/view_map.py b/view_map.py index 821c2b1..7808c3c 100755 --- a/view_map.py +++ b/view_map.py @@ -5,21 +5,21 @@ import zlib import matplotlib.colors as mcol import matplotlib.pyplot as plt -def view_map(dem, lakes, rivers): +def view_map(dem, lakes, rivers, scale): plt.subplot(1,3,1) - plt.pcolormesh(dem, cmap='viridis') + plt.pcolormesh(np.arange(dem.shape[0]+1)*scale, np.arange(dem.shape[1]+1)*scale, dem, cmap='viridis') plt.gca().set_aspect('equal', 'box') plt.colorbar(orientation='horizontal') plt.title('Raw elevation') plt.subplot(1,3,2) - plt.pcolormesh(lakes, cmap='viridis') + plt.pcolormesh(np.arange(lakes.shape[0]+1)*scale, np.arange(lakes.shape[1]+1)*scale, lakes, cmap='viridis') plt.gca().set_aspect('equal', 'box') plt.colorbar(orientation='horizontal') plt.title('Lake surface elevation') plt.subplot(1,3,3) - plt.pcolormesh(rivers, cmap='Blues', norm=mcol.LogNorm()) + plt.pcolormesh(np.arange(rivers.shape[0]+1)*scale, np.arange(rivers.shape[1]+1)*scale, rivers, cmap='Blues', norm=mcol.LogNorm()) plt.gca().set_aspect('equal', 'box') plt.colorbar(orientation='horizontal') plt.title('Rivers flux') @@ -29,8 +29,12 @@ def view_map(dem, lakes, rivers): if __name__ == "__main__": import sys import os + + scale = 1 if len(sys.argv) > 1: os.chdir(sys.argv[1]) + if len(sys.argv) > 2: + scale = int(sys.argv[2]) def load_map(name, dtype, shape): dtype = np.dtype(dtype) @@ -45,4 +49,4 @@ if __name__ == "__main__": lakes = load_map('lakes', '>i2', shape) rivers = load_map('rivers', '>u4', shape) - view_map(dem, lakes, rivers) + view_map(dem, lakes, rivers, scale)