import os import sys import time from WBT.whitebox_tools import WhiteboxTools wbt = WhiteboxTools() work_dir = os.path.dirname(os.path.abspath(__file__)) wb_dir = work_dir + "/WBT" data_dir = work_dir + "/data" out_dir = work_dir + "/out" wbt.set_whitebox_dir(wb_dir) omr = '925' def run(): outfile = f'{out_dir}/basins/{omr}.tif' wbt.basins(f'{data_dir}/D8_Pointer/{omr}.dep', outfile) add_srs = f'gdal_edit.py -a_srs EPSG:3006 {outfile}' print(add_srs) os.system(add_srs) start = time.time() run() end = time.time() print(end - start)
import os import sys from WBT.whitebox_tools import WhiteboxTools wbt = WhiteboxTools() work_dir = os.path.dirname(os.path.abspath(__file__)) wbt.set_whitebox_dir(os.path.dirname(os.path.abspath(__file__)) + "/WBT/") def convert(wbt_work_dir): wbt.work_dir = wbt_work_dir wbt.convert_raster_format('824.dep', '824.tif') convert(os.path.join(work_dir, 'data', 'Breached_DEMs')) convert(os.path.join(work_dir, 'data', 'D8_Flowacc')) convert(os.path.join(work_dir, 'data', 'D8_Pointer'))