mtcontrib/mapgen_rivers
1
0
Fork 0
mirror of https://gitlab.com/gaelysam/mapgen_rivers.git synced 2025-01-01 05:50:34 +01:00
Code Issues Projects Releases Wiki Activity

Add a small random number to DEM when calculating flow direction, to smooth flat areas

Browse Source
This commit is contained in:
Gael-de-Sailly 2020-04-11 14:27:56 +02:00
parent 8b78f6c5b4
commit ecba126983
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
rivermapper.py
View File

@ -18,11 +18,13 @@ neighbours_dirs = np.array([
neighbours_pattern = neighbours_dirs > 0 neighbours_pattern = neighbours_dirs > 0
def flow_dirs_lakes(dem): def flow_dirs_lakes(dem, random=0.0625):
(Y, X) = dem.shape (Y, X) = dem.shape
dem_margin = np.zeros((Y+2, X+2)) dem_margin = np.zeros((Y+2, X+2))
dem_margin[1:-1,1:-1] = dem dem_margin[1:-1,1:-1] = dem
if random > 0:
dem_margin += np.random.random(dem_margin.shape) * random
# Initialize: list map borders # Initialize: list map borders
borders = [] borders = []
@ -72,7 +74,7 @@ def accumulate(dirs, dem=None):
(Y, X) = dirs.shape (Y, X) = dirs.shape
dirs_margin = np.zeros((Y+2,X+2))-1 dirs_margin = np.zeros((Y+2,X+2))-1
dirs_margin[1:-1,1:-1] = dirs dirs_margin[1:-1,1:-1] = dirs
quantity = np.zeros((Y, X), dtype='u4') quantity = np.zeros((Y, X), dtype='i4')
def calculate_quantity(y, x): def calculate_quantity(y, x):
if quantity[y,x] > 0: if quantity[y,x] > 0: