mtcontrib/mapgen_rivers
1
0
Fork 0
mirror of https://gitlab.com/gaelysam/mapgen_rivers.git synced 2024-12-29 12:20:41 +01:00
Code Issues Projects Releases Wiki Activity

Allow command-line options for Python processing

Browse Source
This commit is contained in:
Gael-de-Sailly 2020-11-14 14:30:36 +01:00
parent c33f2d9582
commit b90cecdaf7
1 changed files with 25 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
terrain_rivers.py
View File

@ -11,16 +11,30 @@ import settings
import view_map import view_map
# Always place in this script's parent directory ### READ SETTINGS
os.chdir(os.path.dirname(sys.argv[0]))
argc = len(sys.argv) argc = len(sys.argv)
config_file = 'terrain.conf' config_file = 'terrain.conf'
params_from_args = {}
if argc > 1: i = 1
config_file = sys.argv[1] while i < argc:
arg = sys.argv[i]
if arg[:2] == '--':
pname = arg[2:]
if i+1 < argc:
if pname == 'config':
config_file = sys.argv[i+1]
else:
params_from_args[pname] = sys.argv[i+1]
i += 2
else:
config_file = arg
i += 1
params = settings.read_config_file(config_file) params = settings.read_config_file(config_file)
params.update(params_from_args) # Params given from args prevail against conf file
print(params)
def get_setting(name, default): def get_setting(name, default):
if name in params: if name in params:
@ -43,6 +57,10 @@ flex_radius = float(get_setting('flex_radius', 20.0))
time = float(get_setting('time', 10.0)) time = float(get_setting('time', 10.0))
niter = int(get_setting('niter', 10)) niter = int(get_setting('niter', 10))
# Always place in this script's parent directory
os.chdir(os.path.dirname(sys.argv[0]))
### MAKE INITIAL TOPOGRAPHY
n = np.zeros((mapsize+1, mapsize+1)) n = np.zeros((mapsize+1, mapsize+1))
# Set noise parameters # Set noise parameters
@ -63,6 +81,7 @@ for x in range(mapsize+1):
nn = n*vscale + offset nn = n*vscale + offset
### COMPUTE LANDSCAPE EVOLUTION
# Initialize landscape evolution model # Initialize landscape evolution model
print('Initializing model') print('Initializing model')
model = EvolutionModel(nn, K=1, m=0.35, d=1, sea_level=0, flex_radius=flex_radius) model = EvolutionModel(nn, K=1, m=0.35, d=1, sea_level=0, flex_radius=flex_radius)
@ -97,6 +116,7 @@ offset_x, offset_y = bounds.twist(bx, by, bounds.get_fixed(model.dirs))
offset_x = np.clip(np.floor(offset_x * 256), -128, 127) offset_x = np.clip(np.floor(offset_x * 256), -128, 127)
offset_y = np.clip(np.floor(offset_y * 256), -128, 127) offset_y = np.clip(np.floor(offset_y * 256), -128, 127)
### SAVE OUTPUT
if not os.path.isdir('data'): if not os.path.isdir('data'):
os.mkdir('data') os.mkdir('data')
os.chdir('data') os.chdir('data')