kopia lustrzana https://github.com/OpenDroneMap/ODM
35 wiersze
1.7 KiB
Python
35 wiersze
1.7 KiB
Python
import os
|
|
from opendm import log
|
|
from opendm import system
|
|
from opendm import io
|
|
|
|
def generate_tiles(geotiff, output_dir, max_concurrency):
|
|
gdal2tiles = os.path.join(os.path.dirname(__file__), "gdal2tiles.py")
|
|
system.run('python3 "%s" --processes %s -z 5-21 -n -w none "%s" "%s"' % (gdal2tiles, max_concurrency, geotiff, output_dir))
|
|
|
|
def generate_orthophoto_tiles(geotiff, output_dir, max_concurrency):
|
|
try:
|
|
generate_tiles(geotiff, output_dir, max_concurrency)
|
|
except Exception as e:
|
|
log.ODM_WARNING("Cannot generate orthophoto tiles: %s" % str(e))
|
|
|
|
def generate_dem_tiles(geotiff, output_dir, max_concurrency):
|
|
relief_file = os.path.join(os.path.dirname(__file__), "color_relief.txt")
|
|
hsv_merge_script = os.path.join(os.path.dirname(__file__), "hsv_merge.py")
|
|
colored_dem = io.related_file_path(geotiff, postfix="color")
|
|
hillshade_dem = io.related_file_path(geotiff, postfix="hillshade")
|
|
colored_hillshade_dem = io.related_file_path(geotiff, postfix="colored_hillshade")
|
|
|
|
try:
|
|
system.run('gdaldem color-relief "%s" "%s" "%s" -alpha -co ALPHA=YES' % (geotiff, relief_file, colored_dem))
|
|
system.run('gdaldem hillshade "%s" "%s" -z 1.0 -s 1.0 -az 315.0 -alt 45.0' % (geotiff, hillshade_dem))
|
|
system.run('python3 "%s" "%s" "%s" "%s"' % (hsv_merge_script, colored_dem, hillshade_dem, colored_hillshade_dem))
|
|
generate_tiles(colored_hillshade_dem, output_dir, max_concurrency)
|
|
|
|
# Cleanup
|
|
for f in [colored_dem, hillshade_dem, colored_hillshade_dem]:
|
|
if os.path.isfile(f):
|
|
os.remove(f)
|
|
except Exception as e:
|
|
log.ODM_WARNING("Cannot generate DEM tiles: %s" % str(e))
|