2020-04-12 16:44:29 +02:00
|
|
|
#!/usr/bin/env python3
|
|
|
|
|
|
|
|
|
|
import numpy as np
|
2020-04-13 15:59:34 +02:00
|
|
|
import zlib
|
2020-11-14 16:10:32 +01:00
|
|
|
import sys
|
|
|
|
|
import os
|
2020-04-12 16:44:29 +02:00
|
|
|
|
2021-06-02 18:56:46 +02:00
|
|
|
from view import stats, plot
|
2020-11-13 11:04:27 +01:00
|
|
|
|
2020-11-14 16:10:32 +01:00
|
|
|
scale = 1
|
|
|
|
|
if len(sys.argv) > 1:
|
|
|
|
|
os.chdir(sys.argv[1])
|
|
|
|
|
if len(sys.argv) > 2:
|
|
|
|
|
scale = int(sys.argv[2])
|
2020-11-14 12:05:52 +01:00
|
|
|
|
2020-11-14 16:10:32 +01:00
|
|
|
def load_map(name, dtype, shape):
|
|
|
|
|
dtype = np.dtype(dtype)
|
|
|
|
|
with open(name, 'rb') as f:
|
|
|
|
|
data = f.read()
|
2020-04-26 18:30:29 +02:00
|
|
|
if len(data) < shape[0]*shape[1]*dtype.itemsize:
|
|
|
|
|
data = zlib.decompress(data)
|
|
|
|
|
return np.frombuffer(data, dtype=dtype).reshape(shape)
|
|
|
|
|
|
2020-11-14 16:10:32 +01:00
|
|
|
shape = np.loadtxt('size', dtype='u4')
|
|
|
|
|
dem = load_map('dem', '>i2', shape)
|
|
|
|
|
lakes = load_map('lakes', '>i2', shape)
|
2020-04-26 18:30:29 +02:00
|
|
|
|
2020-11-14 16:10:32 +01:00
|
|
|
stats(dem, lakes, scale=scale)
|
|
|
|
|
plot(dem, lakes, scale=scale)
|
