tileManagerDemo1.py

## @package python.demo.tileManagerDemo1
# 
# @brief Demo script for a simple tiling with tileManager
# 
# This script takes a number of ODM point clouds, tiles them into 2 rows and 2 columns, and creates a tile-wise DSM.
# Afterwards, the DSM is merged.
# 
# 
 
import sys, os
import glob
 
import pprint
 
import opals
from opals import DSM, Algebra
 
from opals.tools import tileManager
from opals.tools.doxygen import brief
from opals.tools.opalsargparse import PkgArgumentParser, Boolean, PathArgument
 
@opals.main
def main(options, logger):
    inp = []    #build list of input files (supporting wildcards)
    for i in options.i:
        inp.extend(glob.glob(str(i)))
    if len(inp) == 0:
        raise Exception("No input files found")
 
    inp_files = [{'files': inp, # List of Files (relative or full path, no */?)\n
                 'type': 'odm',  # either odm or raster\n
                 'name': 'pointcloud',  # any name that can be used as a dict key and a (windows) filename\n
                 'required': True}]
    tiling_concept = '2r2c'  # two rows, two columns
    naming_concept = 'LL_6'  # lower left coordinate, 6 digits, '_'-separated
    tempdir = 'temp'
 
    # prepare tileManager
    tm = tileManager.tileManager(logger, inp_files, tiling_concept, naming_concept, tempdir)
    # get number of elements - this already cuts the files!
    lix = len(tm)
 
    outfiles = []
    # iterate over the tiles (no more cutting needed here)
    for idx, tile in enumerate(tm):
        print("Now processing tile %s/%s:" % (idx, lix))
        pprint.pprint(tile)
        outfile = os.path.join(tempdir, tile['tile_name'], 'dsm.tif')
        # create a DSM
        dsm = DSM.DSM(inFile=tile['pointcloud'],
                      outFile=outfile)
        dsm.run()
        outfiles.append(outfile)
    # merge the output files into one mosaic
    alg = Algebra.Algebra(inFile=outfiles,
                          outFile='mosaic.tif',
                          formula='first(r)',
                          limit='union')
    alg.run()
 
if __name__ == '__main__':
    parser = PkgArgumentParser(usage="%(prog)s [options]\n" + brief(__doc__))
    parser.add_argument("-i", type=PathArgument, help="input files", remarks='mandatory', nargs='*' )
    options = parser.parse_args(sys.argv[1:])
    main(options)