raster_average.py 文件源码

def mean(input_dir, out_dir, size, func, partitions, mask):
    input_dir = abspath(input_dir)
    if isdir(input_dir):
        log.info('Reading tifs from {}'.format(input_dir))
        tifs = glob.glob(join(input_dir, '*.tif'))
    else:
        assert isfile(input_dir)
        tifs = [input_dir]

    process_tifs = np.array_split(tifs, mpiops.chunks)[mpiops.chunk_index]

    for tif in process_tifs:
        log.info('Starting to average {}'.format(basename(tif)))
        treat_file(tif, out_dir, size, func, partitions, mask)
        log.info('Finished averaging {}'.format(basename(tif)))